Support more than one closed caption asset per reel.

[libdcp.git] / src / verify.cc

diff --git a/src/verify.cc b/src/verify.cc
index 76d7c18d401a4c27f72f0177e419abea61d41854..db65595451347145ff56f60c1dc2f5a758eb16e1 100644 (file)
--- a/src/verify.cc
+++ b/src/verify.cc
@@ -33,19 +33,58 @@
 
 #include "verify.h"
 #include "dcp.h"
+#include "cpl.h"
+#include "reel.h"
+#include "reel_picture_asset.h"
+#include "reel_sound_asset.h"
 #include "exceptions.h"
 #include <boost/foreach.hpp>
 #include <list>
 #include <vector>
+#include <iostream>
 
 using std::list;
 using std::vector;
+using std::string;
+using std::cout;
 using boost::shared_ptr;
+using boost::optional;
+using boost::function;
 
 using namespace dcp;
 
+enum Result {
+       RESULT_GOOD,
+       RESULT_CPL_PKL_DIFFER,
+       RESULT_BAD
+};
+
+static Result
+verify_asset (shared_ptr<DCP> dcp, shared_ptr<ReelAsset> reel_asset, function<void (float)> progress)
+{
+       string const actual_hash = reel_asset->asset_ref()->hash(progress);
+
+       shared_ptr<PKL> pkl = dcp->pkl();
+       /* We've read this DCP in so it must have a PKL */
+       DCP_ASSERT (pkl);
+
+       shared_ptr<Asset> asset = reel_asset->asset_ref().asset();
+       string const pkl_hash = pkl->hash (reel_asset->asset_ref()->id());
+
+       optional<string> cpl_hash = reel_asset->hash();
+       if (cpl_hash && *cpl_hash != pkl_hash) {
+               return RESULT_CPL_PKL_DIFFER;
+       }
+
+       if (actual_hash != pkl_hash) {
+               return RESULT_BAD;
+       }
+
+       return RESULT_GOOD;
+}
+
 list<VerificationNote>
-dcp::verify (vector<boost::filesystem::path> directories)
+dcp::verify (vector<boost::filesystem::path> directories, function<void (string, optional<boost::filesystem::path>)> stage, function<void (float)> progress)
 {
        list<VerificationNote> notes;
 
@@ -54,14 +93,50 @@ dcp::verify (vector<boost::filesystem::path> directories)
                dcps.push_back (shared_ptr<DCP> (new DCP (i)));
        }
 
-       BOOST_FOREACH (shared_ptr<DCP> i, dcps) {
+       BOOST_FOREACH (shared_ptr<DCP> dcp, dcps) {
+               stage ("Checking DCP", dcp->directory());
                DCP::ReadErrors errors;
                try {
-                       i->read (true, &errors);
+                       dcp->read (true, &errors);
                } catch (DCPReadError& e) {
-                       notes.push_back (VerificationNote (VerificationNote::ERROR, e.what ()));
+                       notes.push_back (VerificationNote (VerificationNote::VERIFY_ERROR, e.what ()));
                } catch (XMLError& e) {
-                       notes.push_back (VerificationNote (VerificationNote::ERROR, e.what ()));
+                       notes.push_back (VerificationNote (VerificationNote::VERIFY_ERROR, e.what ()));
+               }
+
+               BOOST_FOREACH (shared_ptr<CPL> cpl, dcp->cpls()) {
+                       stage ("Checking CPL", cpl->file());
+                       BOOST_FOREACH (shared_ptr<Reel> reel, cpl->reels()) {
+                               stage ("Checking reel", optional<boost::filesystem::path>());
+                               if (reel->main_picture()) {
+                                       stage ("Checking picture asset hash", reel->main_picture()->asset()->file());
+                                       Result const r = verify_asset (dcp, reel->main_picture(), progress);
+                                       switch (r) {
+                                       case RESULT_BAD:
+                                               notes.push_back (VerificationNote (VerificationNote::VERIFY_ERROR, "Picture asset hash is incorrect."));
+                                               break;
+                                       case RESULT_CPL_PKL_DIFFER:
+                                               notes.push_back (VerificationNote (VerificationNote::VERIFY_ERROR, "PKL and CPL hashes differ for picture asset."));
+                                               break;
+                                       default:
+                                               break;
+                                       }
+                               }
+                               if (reel->main_sound()) {
+                                       stage ("Checking sound asset hash", reel->main_sound()->asset()->file());
+                                       Result const r = verify_asset (dcp, reel->main_sound(), progress);
+                                       switch (r) {
+                                       case RESULT_BAD:
+                                               notes.push_back (VerificationNote (VerificationNote::VERIFY_ERROR, "Sound asset hash is incorrect."));
+                                               break;
+                                       case RESULT_CPL_PKL_DIFFER:
+                                               notes.push_back (VerificationNote (VerificationNote::VERIFY_ERROR, "PKL and CPL hashes differ for sound asset."));
+                                               break;
+                                       default:
+                                               break;
+                                       }
+                               }
+                       }
                }
        }