1 files changed, 79 insertions, 0 deletions
diff --git a/src/lib/check_hashes_job.cc b/src/lib/check_hashes_job.cc
new file mode 100644
index 000000000..87eb40d14
--- /dev/null
+++ b/src/lib/check_hashes_job.cc
@@ -0,0 +1,79 @@
+/*
+    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 "film_state.h"
+#include "options.h"
+#include "log.h"
+
+using namespace std;
+using namespace boost;
+
+CheckHashesJob::CheckHashesJob (shared_ptr<const FilmState> s, shared_ptr<const Options> o, Log* l)
+	: Job (s, o, l)
+	, _bad (0)
+{
+
+}
+
+string
+CheckHashesJob::name () const
+{
+	stringstream s;
+	s << "Check hashes of " << _fs->name;
+	return s.str ();
+}
+
+void
+CheckHashesJob::run ()
+{
+	_bad = 0;
+	
+	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);
+			++_bad;
+		}
+
+		set_progress (float (i) / _fs->length);
+	}
+
+	set_progress (1);
+	set_state (FINISHED_OK);
+}
+
+string
+CheckHashesJob::status () const
+{
+	stringstream s;
+	s << Job::status () << "; " << _bad << " bad frames found";
+	return s.str ();
+}