#include <boost/lexical_cast.hpp>
#include <boost/filesystem.hpp>
#include "check_hashes_job.h"
-#include "film_state.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 namespace std;
-using namespace boost;
+using std::string;
+using std::stringstream;
+using std::ifstream;
+using boost::shared_ptr;
-CheckHashesJob::CheckHashesJob (shared_ptr<const FilmState> s, shared_ptr<const Options> o, Log* l)
- : Job (s, o, l)
+CheckHashesJob::CheckHashesJob (shared_ptr<Film> f, DecodeOptions o, shared_ptr<Job> req)
+ : Job (f, req)
+ , _decode_opt (o)
, _bad (0)
{
string
CheckHashesJob::name () const
{
- stringstream s;
- s << "Check hashes of " << _fs->name;
- return s.str ();
+ return String::compose ("Check hashes of %1", _film->name());
}
void
CheckHashesJob::run ()
{
_bad = 0;
+
+ if (!_film->dcp_intrinsic_duration()) {
+ throw EncodeError ("cannot check hashes of a DCP with unknown intrinsic duration");
+ }
- for (int i = 0; i < _fs->length; ++i) {
- string const j2k_file = _opt->frame_out_path (i, false);
- string const hash_file = j2k_file + ".md5";
-
- ifstream ref (hash_file.c_str ());
- string hash;
- ref >> hash;
-
- if (hash != md5_digest (j2k_file)) {
- _log->log ("Frame " + lexical_cast<string> (i) + " has wrong hash; deleting.");
- filesystem::remove (j2k_file);
- filesystem::remove (hash_file);
+ int const N = _film->dcp_intrinsic_duration().get();
+ for (int i = 0; i < N; ++i) {
+ string const j2k_file = _film->frame_out_path (i, false);
+ string const hash_file = _film->hash_out_path (i, false);
+
+ 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) / _fs->length);
+ set_progress (float (i) / N);
}
if (_bad) {
shared_ptr<Job> tc;
- if (_fs->dcp_ab) {
- tc.reset (new ABTranscodeJob (_fs, _opt, _log));
+ if (_film->dcp_ab()) {
+ tc.reset (new ABTranscodeJob (_film, _decode_opt, shared_from_this()));
} else {
- tc.reset (new TranscodeJob (_fs, _opt, _log));
+ tc.reset (new TranscodeJob (_film, _decode_opt, shared_from_this()));
}
JobManager::instance()->add_after (shared_from_this(), tc);
- JobManager::instance()->add_after (tc, shared_ptr<Job> (new CheckHashesJob (_fs, _opt, _log)));
+ JobManager::instance()->add_after (tc, shared_ptr<Job> (new CheckHashesJob (_film, _decode_opt, tc)));
}
set_progress (1);