summaryrefslogtreecommitdiff
path: root/src/lib/dvd.cc
blob: 19b59b5881daa7949b82a92047a60e02164c764b (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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/*
    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>
#include "dvd.h"

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 "";
}

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

	list<DVDTitle> titles;
	
	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 ());
				if (b == 0) {
					continue;
				}

				list<DVDTitle>::iterator j = titles.begin ();
				while (j != titles.end() && j->number != a) {
					++j;
				}

				if (j == titles.end ()) {
					titles.push_back (DVDTitle (a, size));
				} else {
					j->size += size;
				}
			}
		}
	}

	titles.sort ();
					
	return titles;
}


bool
operator< (DVDTitle const & a, DVDTitle const & b)
{
	return a.number < b.number;
}