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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
/*
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/lexical_cast.hpp>
#include <boost/filesystem.hpp>
#include "check_hashes_job.h"
#include "options.h"
#include "log.h"
#include "job_manager.h"
#include "ab_transcode_job.h"
#include "transcode_job.h"
#include "film.h"
#include "exceptions.h"
using std::string;
using std::stringstream;
using std::ifstream;
using boost::shared_ptr;
CheckHashesJob::CheckHashesJob (shared_ptr<Film> f, shared_ptr<const Options> o, shared_ptr<Job> req)
: Job (f, req)
, _opt (o)
, _bad (0)
{
}
string
CheckHashesJob::name () const
{
return String::compose ("Check hashes of %1", _film->name());
}
void
CheckHashesJob::run ()
{
_bad = 0;
if (!_film->dcp_length()) {
throw EncodeError ("cannot check hashes of a DCP with unknown length");
}
int const N = _film->dcp_length().get();
for (int i = 0; i < N; ++i) {
string const j2k_file = _opt->frame_out_path (i, false);
string const hash_file = j2k_file + ".md5";
if (!boost::filesystem::exists (j2k_file)) {
_film->log()->log (String::compose ("Frame %1 has a missing J2K file.", i));
boost::filesystem::remove (hash_file);
++_bad;
} else if (!boost::filesystem::exists (hash_file)) {
_film->log()->log (String::compose ("Frame %1 has a missing hash file.", i));
boost::filesystem::remove (j2k_file);
++_bad;
} else {
ifstream ref (hash_file.c_str ());
string hash;
ref >> hash;
if (hash != md5_digest (j2k_file)) {
_film->log()->log (String::compose ("Frame %1 has wrong hash; deleting.", i));
boost::filesystem::remove (j2k_file);
boost::filesystem::remove (hash_file);
++_bad;
}
}
set_progress (float (i) / N);
}
if (_bad) {
shared_ptr<Job> tc;
if (_film->dcp_ab()) {
tc.reset (new ABTranscodeJob (_film, _opt, shared_from_this()));
} else {
tc.reset (new TranscodeJob (_film, _opt, shared_from_this()));
}
JobManager::instance()->add_after (shared_from_this(), tc);
JobManager::instance()->add_after (tc, shared_ptr<Job> (new CheckHashesJob (_film, _opt, tc)));
}
set_progress (1);
set_state (FINISHED_OK);
}
string
CheckHashesJob::status () const
{
stringstream s;
s << Job::status ();
if (overall_progress() > 0) {
if (_bad == 0) {
s << "; no bad frames found";
} else if (_bad == 1) {
s << "; 1 bad frame found";
} else {
s << "; " << _bad << " bad frames found";
}
}
return s.str ();
}
|
