summaryrefslogtreecommitdiff
path: root/src/lib/dvd.cc
blob: 629ba1ac8c21bb2ebb0736248c0a6c13e58050a8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include <fstream>
#include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp>

using namespace std;
using namespace boost;

string
find_dvd ()
{
	ifstream f ("/etc/mtab");
	while (f.good ()) {
		string s;
		getline (f, s);
		vector<string> b;
		split (b, s, is_any_of (" "));
		if (b.size() >= 3 && b[2] == "udf") {
			replace_all (b[1], "\\040", " ");
			return b[1];
		}
	}

	return "";
}

vector<uint64_t>
dvd_titles (string dvd)
{
	filesystem::path video (dvd);
	video /= "VIDEO_TS";

	vector<uint64_t> sizes;
	
	for (filesystem::directory_iterator i = filesystem::directory_iterator (video); i != filesystem::directory_iterator(); ++i) {
#if BOOST_FILESYSTEM_VERSION == 3		
		string const n = filesystem::path(*i).filename().generic_string();
#else		
		string const n = filesystem::path(*i).filename();
#endif
		if (starts_with (n, "VTS_") && ends_with (n, ".VOB")) {
			uint64_t const size = filesystem::file_size (filesystem::path (*i));
			vector<string> p;
			split (p, n, is_any_of ("_."));
			if (p.size() == 4) {
				int const a = atoi (p[1].c_str ());
				int const b = atoi (p[2].c_str ());
				while (a >= int (sizes.size())) {
					sizes.push_back (0);
				}

				if (b > 0) {
					sizes[a] += size;
				}
			}
		}
	}
					
	return sizes;		
}