Use an enum class for Marker.
[libdcp.git] / src / dcp.cc
index 52598610a808defa0031f8379e75c4cc89b01dc5..ca7313a535cee7784d923a39b5d4099010de9daf 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of libdcp.
 
 #include "font_asset.h"
 #include "pkl.h"
 #include "asset_factory.h"
+#include "verify.h"
 #include <asdcp/AS_DCP.h>
 #include <xmlsec/xmldsig.h>
 #include <xmlsec/app.h>
 #include <libxml++/libxml++.h>
 #include <boost/filesystem.hpp>
 #include <boost/algorithm/string.hpp>
-#include <boost/foreach.hpp>
 
 using std::string;
 using std::list;
@@ -73,9 +73,10 @@ using std::cout;
 using std::make_pair;
 using std::map;
 using std::cerr;
+using std::make_shared;
 using std::exception;
-using boost::shared_ptr;
-using boost::dynamic_pointer_cast;
+using std::shared_ptr;
+using std::dynamic_pointer_cast;
 using boost::optional;
 using boost::algorithm::starts_with;
 using namespace dcp;
@@ -95,36 +96,33 @@ DCP::DCP (boost::filesystem::path directory)
        _directory = boost::filesystem::canonical (_directory);
 }
 
-/** Call this instead of throwing an exception if the error can be tolerated */
-template<class T> void
-survivable_error (bool keep_going, dcp::DCP::ReadErrors* errors, T const & e)
-{
-       if (keep_going) {
-               if (errors) {
-                       errors->push_back (shared_ptr<T> (new T (e)));
-               }
-       } else {
-               throw e;
-       }
-}
-
+/** Read a DCP.  This method does not do any deep checking of the DCP's validity, but
+ *  if it comes across any bad things it will do one of two things.
+ *
+ *  Errors that are so serious that they prevent the method from working will result
+ *  in an exception being thrown.  For example, a missing ASSETMAP means that the DCP
+ *  can't be read without a lot of guesswork, so this will throw.
+ *
+ *  Errors that are not fatal will be added to notes, if it's non-0.  For example,
+ *  if the DCP contains a mixture of Interop and SMPTE elements this will result
+ *  in a note being added to the vector.
+ */
 void
-DCP::read (bool keep_going, ReadErrors* errors, bool ignore_incorrect_picture_mxf_type)
+DCP::read (vector<dcp::VerificationNote>* notes, bool ignore_incorrect_picture_mxf_type)
 {
        /* Read the ASSETMAP and PKL */
 
-       boost::filesystem::path asset_map_file;
        if (boost::filesystem::exists (_directory / "ASSETMAP")) {
-               asset_map_file = _directory / "ASSETMAP";
+               _asset_map = _directory / "ASSETMAP";
        } else if (boost::filesystem::exists (_directory / "ASSETMAP.xml")) {
-               asset_map_file = _directory / "ASSETMAP.xml";
+               _asset_map = _directory / "ASSETMAP.xml";
        } else {
-               boost::throw_exception (DCPReadError (String::compose ("could not find AssetMap file in `%1'", _directory.string())));
+               boost::throw_exception (ReadError(String::compose("Could not find ASSETMAP nor ASSETMAP.xml in '%1'", _directory.string())));
        }
 
        cxml::Document asset_map ("AssetMap");
 
-       asset_map.read_file (asset_map_file);
+       asset_map.read_file (_asset_map.get());
        if (asset_map.namespace_uri() == assetmap_interop_ns) {
                _standard = INTEROP;
        } else if (asset_map.namespace_uri() == assetmap_smpte_ns) {
@@ -133,14 +131,14 @@ DCP::read (bool keep_going, ReadErrors* errors, bool ignore_incorrect_picture_mx
                boost::throw_exception (XMLError ("Unrecognised Assetmap namespace " + asset_map.namespace_uri()));
        }
 
-       list<shared_ptr<cxml::Node> > asset_nodes = asset_map.node_child("AssetList")->node_children ("Asset");
+       auto asset_nodes = asset_map.node_child("AssetList")->node_children ("Asset");
        map<string, boost::filesystem::path> paths;
-       list<boost::filesystem::path> pkl_paths;
-       BOOST_FOREACH (shared_ptr<cxml::Node> i, asset_nodes) {
+       vector<boost::filesystem::path> pkl_paths;
+       for (auto i: asset_nodes) {
                if (i->node_child("ChunkList")->node_children("Chunk").size() != 1) {
                        boost::throw_exception (XMLError ("unsupported asset chunk count"));
                }
-               string p = i->node_child("ChunkList")->node_child("Chunk")->string_child ("Path");
+               auto p = i->node_child("ChunkList")->node_child("Chunk")->string_child ("Path");
                if (starts_with (p, "file://")) {
                        p = p.substr (7);
                }
@@ -154,7 +152,7 @@ DCP::read (bool keep_going, ReadErrors* errors, bool ignore_incorrect_picture_mx
                        break;
                case SMPTE:
                {
-                       optional<string> pkl_bool = i->optional_string_child("PackingList");
+                       auto pkl_bool = i->optional_string_child("PackingList");
                        if (pkl_bool && *pkl_bool == "true") {
                                pkl_paths.push_back (p);
                        } else {
@@ -169,58 +167,82 @@ DCP::read (bool keep_going, ReadErrors* errors, bool ignore_incorrect_picture_mx
                boost::throw_exception (XMLError ("No packing lists found in asset map"));
        }
 
-       BOOST_FOREACH (boost::filesystem::path i, pkl_paths) {
-               _pkls.push_back (shared_ptr<PKL>(new PKL(_directory / i)));
+       for (auto i: pkl_paths) {
+               _pkls.push_back (make_shared<PKL>(_directory / i));
        }
 
-       /* Read all the assets from the asset map */
+       /* Now we have:
+            paths - map of files in the DCP that are not PKLs; key is ID, value is path.
+            _pkls - PKL objects for each PKL.
+
+          Read all the assets from the asset map.
+        */
 
        /* Make a list of non-CPL/PKL assets so that we can resolve the references
           from the CPLs.
        */
-       list<shared_ptr<Asset> > other_assets;
-
-       for (map<string, boost::filesystem::path>::const_iterator i = paths.begin(); i != paths.end(); ++i) {
-               boost::filesystem::path path = _directory / i->second;
+       vector<shared_ptr<Asset>> other_assets;
+
+       for (auto i: paths) {
+               auto path = _directory / i.second;
+
+               if (i.second.empty()) {
+                       /* I can't see how this is valid, but it's
+                          been seen in the wild with a DCP that
+                          claims to come from ClipsterDCI 5.10.0.5.
+                       */
+                       if (notes) {
+                               notes->push_back (VerificationNote(VerificationNote::VERIFY_WARNING, VerificationNote::EMPTY_ASSET_PATH));
+                       }
+                       continue;
+               }
 
-               if (!boost::filesystem::exists (path)) {
-                       survivable_error (keep_going, errors, MissingAssetError (path));
+               if (!boost::filesystem::exists(path)) {
+                       if (notes) {
+                               notes->push_back (VerificationNote(VerificationNote::VERIFY_ERROR, VerificationNote::MISSING_ASSET, path));
+                       }
                        continue;
                }
 
+               /* Find the <Type> for this asset from the PKL that contains the asset */
                optional<string> pkl_type;
-               BOOST_FOREACH (shared_ptr<PKL> j, _pkls) {
-                       pkl_type = j->type(i->first);
+               for (auto j: _pkls) {
+                       pkl_type = j->type(i.first);
                        if (pkl_type) {
                                break;
                        }
                }
 
-               DCP_ASSERT (pkl_type);
+               if (!pkl_type) {
+                       /* This asset is in the ASSETMAP but not mentioned in any PKL so we don't
+                        * need to worry about it.
+                        */
+                       continue;
+               }
 
                if (*pkl_type == CPL::static_pkl_type(*_standard) || *pkl_type == InteropSubtitleAsset::static_pkl_type(*_standard)) {
-                       xmlpp::DomParser* p = new xmlpp::DomParser;
+                       auto p = new xmlpp::DomParser;
                        try {
                                p->parse_file (path.string());
                        } catch (std::exception& e) {
                                delete p;
-                               throw DCPReadError(String::compose("XML error in %1", path.string()), e.what());
+                               throw ReadError(String::compose("XML error in %1", path.string()), e.what());
                        }
 
-                       string const root = p->get_document()->get_root_node()->get_name ();
+                       auto const root = p->get_document()->get_root_node()->get_name ();
                        delete p;
 
                        if (root == "CompositionPlaylist") {
-                               shared_ptr<CPL> cpl (new CPL (path));
-                               if (_standard && cpl->standard() && cpl->standard().get() != _standard.get()) {
-                                       survivable_error (keep_going, errors, MismatchedStandardError ());
+                               auto cpl = make_shared<CPL>(path);
+                               if (_standard && cpl->standard() && cpl->standard().get() != _standard.get() && notes) {
+                                       notes->push_back (VerificationNote(VerificationNote::VERIFY_ERROR, VerificationNote::MISMATCHED_STANDARD));
                                }
                                _cpls.push_back (cpl);
                        } else if (root == "DCSubtitle") {
-                               if (_standard && _standard.get() == SMPTE) {
-                                       survivable_error (keep_going, errors, MismatchedStandardError ());
+                               if (_standard && _standard.get() == SMPTE && notes) {
+                                       notes->push_back (VerificationNote(VerificationNote::VERIFY_ERROR, VerificationNote::MISMATCHED_STANDARD));
                                }
-                               other_assets.push_back (shared_ptr<InteropSubtitleAsset> (new InteropSubtitleAsset (path)));
+                               other_assets.push_back (make_shared<InteropSubtitleAsset>(path));
                        }
                } else if (
                        *pkl_type == PictureAsset::static_pkl_type(*_standard) ||
@@ -231,23 +253,32 @@ DCP::read (bool keep_going, ReadErrors* errors, bool ignore_incorrect_picture_mx
 
                        other_assets.push_back (asset_factory(path, ignore_incorrect_picture_mxf_type));
                } else if (*pkl_type == FontAsset::static_pkl_type(*_standard)) {
-                       other_assets.push_back (shared_ptr<FontAsset> (new FontAsset (i->first, path)));
+                       other_assets.push_back (make_shared<FontAsset>(i.first, path));
                } else if (*pkl_type == "image/png") {
                        /* It's an Interop PNG subtitle; let it go */
                } else {
-                       throw DCPReadError (String::compose("Unknown asset type %1 in PKL", *pkl_type));
+                       throw ReadError (String::compose("Unknown asset type %1 in PKL", *pkl_type));
                }
        }
 
-       BOOST_FOREACH (shared_ptr<CPL> i, cpls ()) {
-               i->resolve_refs (other_assets);
+       resolve_refs (other_assets);
+
+       /* While we've got the ASSETMAP lets look and see if this DCP refers to things that are not in its ASSETMAP */
+       if (notes) {
+               for (auto i: cpls()) {
+                       for (auto j: i->reel_mxfs()) {
+                               if (!j->asset_ref().resolved() && paths.find(j->asset_ref().id()) == paths.end()) {
+                                       notes->push_back (VerificationNote(VerificationNote::VERIFY_WARNING, VerificationNote::EXTERNAL_ASSET, j->asset_ref().id()));
+                               }
+                       }
+               }
        }
 }
 
 void
-DCP::resolve_refs (list<shared_ptr<Asset> > assets)
+DCP::resolve_refs (vector<shared_ptr<Asset>> assets)
 {
-       BOOST_FOREACH (shared_ptr<CPL> i, cpls ()) {
+       for (auto i: cpls()) {
                i->resolve_refs (assets);
        }
 }
@@ -255,8 +286,8 @@ DCP::resolve_refs (list<shared_ptr<Asset> > assets)
 bool
 DCP::equals (DCP const & other, EqualityOptions opt, NoteHandler note) const
 {
-       list<shared_ptr<CPL> > a = cpls ();
-       list<shared_ptr<CPL> > b = other.cpls ();
+       auto a = cpls ();
+       auto b = other.cpls ();
 
        if (a.size() != b.size()) {
                note (DCP_ERROR, String::compose ("CPL counts differ: %1 vs %2", a.size(), b.size()));
@@ -265,8 +296,8 @@ DCP::equals (DCP const & other, EqualityOptions opt, NoteHandler note) const
 
        bool r = true;
 
-       BOOST_FOREACH (shared_ptr<CPL> i, a) {
-               list<shared_ptr<CPL> >::const_iterator j = b.begin ();
+       for (auto i: a) {
+               auto j = b.begin();
                while (j != b.end() && !(*j)->equals (i, opt, note)) {
                        ++j;
                }
@@ -280,7 +311,7 @@ DCP::equals (DCP const & other, EqualityOptions opt, NoteHandler note) const
 }
 
 void
-DCP::add (boost::shared_ptr<CPL> cpl)
+DCP::add (std::shared_ptr<CPL> cpl)
 {
        _cpls.push_back (cpl);
 }
@@ -288,8 +319,8 @@ DCP::add (boost::shared_ptr<CPL> cpl)
 bool
 DCP::encrypted () const
 {
-       BOOST_FOREACH (shared_ptr<CPL> i, cpls ()) {
-               if (i->encrypted ()) {
+       for (auto i: cpls()) {
+               if (i->encrypted()) {
                        return true;
                }
        }
@@ -304,10 +335,10 @@ DCP::encrypted () const
 void
 DCP::add (DecryptedKDM const & kdm)
 {
-       list<DecryptedKDMKey> keys = kdm.keys ();
+       auto keys = kdm.keys ();
 
-       BOOST_FOREACH (shared_ptr<CPL> i, cpls ()) {
-               BOOST_FOREACH (DecryptedKDMKey const & j, kdm.keys ()) {
+       for (auto i: cpls()) {
+               for (auto const& j: kdm.keys()) {
                        if (j.cpl_id() == i->id()) {
                                i->add (kdm);
                        }
@@ -352,9 +383,12 @@ DCP::write_volindex (Standard standard) const
 }
 
 void
-DCP::write_assetmap (Standard standard, string pkl_uuid, boost::filesystem::path pkl_path, XMLMetadata metadata) const
+DCP::write_assetmap (
+       Standard standard, string pkl_uuid, boost::filesystem::path pkl_path,
+       string issuer, string creator, string issue_date, string annotation_text
+       ) const
 {
-       boost::filesystem::path p = _directory;
+       auto p = _directory;
 
        switch (standard) {
        case INTEROP:
@@ -382,42 +416,43 @@ DCP::write_assetmap (Standard standard, string pkl_uuid, boost::filesystem::path
        }
 
        root->add_child("Id")->add_child_text ("urn:uuid:" + make_uuid());
-       root->add_child("AnnotationText")->add_child_text (metadata.annotation_text);
+       root->add_child("AnnotationText")->add_child_text (annotation_text);
 
        switch (standard) {
        case INTEROP:
                root->add_child("VolumeCount")->add_child_text ("1");
-               root->add_child("IssueDate")->add_child_text (metadata.issue_date);
-               root->add_child("Issuer")->add_child_text (metadata.issuer);
-               root->add_child("Creator")->add_child_text (metadata.creator);
+               root->add_child("IssueDate")->add_child_text (issue_date);
+               root->add_child("Issuer")->add_child_text (issuer);
+               root->add_child("Creator")->add_child_text (creator);
                break;
        case SMPTE:
-               root->add_child("Creator")->add_child_text (metadata.creator);
+               root->add_child("Creator")->add_child_text (creator);
                root->add_child("VolumeCount")->add_child_text ("1");
-               root->add_child("IssueDate")->add_child_text (metadata.issue_date);
-               root->add_child("Issuer")->add_child_text (metadata.issuer);
+               root->add_child("IssueDate")->add_child_text (issue_date);
+               root->add_child("Issuer")->add_child_text (issuer);
                break;
        default:
                DCP_ASSERT (false);
        }
 
-       xmlpp::Node* asset_list = root->add_child ("AssetList");
+       auto asset_list = root->add_child ("AssetList");
 
-       xmlpp::Node* asset = asset_list->add_child ("Asset");
+       auto asset = asset_list->add_child ("Asset");
        asset->add_child("Id")->add_child_text ("urn:uuid:" + pkl_uuid);
        asset->add_child("PackingList")->add_child_text ("true");
-       xmlpp::Node* chunk_list = asset->add_child ("ChunkList");
-       xmlpp::Node* chunk = chunk_list->add_child ("Chunk");
+       auto chunk_list = asset->add_child ("ChunkList");
+       auto chunk = chunk_list->add_child ("Chunk");
        chunk->add_child("Path")->add_child_text (pkl_path.filename().string());
        chunk->add_child("VolumeIndex")->add_child_text ("1");
        chunk->add_child("Offset")->add_child_text ("0");
        chunk->add_child("Length")->add_child_text (raw_convert<string> (boost::filesystem::file_size (pkl_path)));
 
-       BOOST_FOREACH (shared_ptr<Asset> i, assets ()) {
+       for (auto i: assets()) {
                i->write_to_assetmap (asset_list, _directory);
        }
 
        doc.write_to_file_formatted (p.string (), "UTF-8");
+       _asset_map = p;
 }
 
 /** Write all the XML files for this DCP.
@@ -428,12 +463,15 @@ DCP::write_assetmap (Standard standard, string pkl_uuid, boost::filesystem::path
 void
 DCP::write_xml (
        Standard standard,
-       XMLMetadata metadata,
+       string issuer,
+       string creator,
+       string issue_date,
+       string annotation_text,
        shared_ptr<const CertificateChain> signer,
        NameFormat name_format
        )
 {
-       BOOST_FOREACH (shared_ptr<CPL> i, cpls ()) {
+       for (auto i: cpls()) {
                NameFormat::Map values;
                values['t'] = "cpl";
                i->write_xml (_directory / (name_format.get(values, "_" + i->id() + ".xml")), standard, signer);
@@ -442,9 +480,9 @@ DCP::write_xml (
        shared_ptr<PKL> pkl;
 
        if (_pkls.empty()) {
-               pkl.reset (new PKL (standard, metadata.annotation_text, metadata.issue_date, metadata.issuer, metadata.creator));
+               pkl = make_shared<PKL>(standard, annotation_text, issue_date, issuer, creator);
                _pkls.push_back (pkl);
-               BOOST_FOREACH (shared_ptr<Asset> i, assets ()) {
+               for (auto i: assets()) {
                        i->add_to_pkl (pkl, _directory);
                }
         } else {
@@ -453,14 +491,14 @@ DCP::write_xml (
 
        NameFormat::Map values;
        values['t'] = "pkl";
-       boost::filesystem::path pkl_path = _directory / name_format.get(values, "_" + pkl->id() + ".xml");
+       auto pkl_path = _directory / name_format.get(values, "_" + pkl->id() + ".xml");
        pkl->write (pkl_path, signer);
 
        write_volindex (standard);
-       write_assetmap (standard, pkl->id(), pkl_path, metadata);
+       write_assetmap (standard, pkl->id(), pkl_path, issuer, creator, issue_date, annotation_text);
 }
 
-list<shared_ptr<CPL> >
+vector<shared_ptr<CPL>>
 DCP::cpls () const
 {
        return _cpls;
@@ -470,22 +508,33 @@ DCP::cpls () const
  *  an exception is thrown if they are found.
  *  @return All assets (including CPLs).
  */
-list<shared_ptr<Asset> >
+vector<shared_ptr<Asset>>
 DCP::assets (bool ignore_unresolved) const
 {
-       list<shared_ptr<Asset> > assets;
-       BOOST_FOREACH (shared_ptr<CPL> i, cpls ()) {
+       vector<shared_ptr<Asset>> assets;
+       for (auto i: cpls()) {
                assets.push_back (i);
-               BOOST_FOREACH (shared_ptr<const ReelMXF> j, i->reel_mxfs()) {
+               for (auto j: i->reel_mxfs()) {
                        if (ignore_unresolved && !j->asset_ref().resolved()) {
                                continue;
                        }
-                       shared_ptr<Asset> o = j->asset_ref().asset ();
-                       assets.push_back (o);
-                       /* More Interop special-casing */
-                       shared_ptr<InteropSubtitleAsset> sub = dynamic_pointer_cast<InteropSubtitleAsset> (o);
-                       if (sub) {
-                               sub->add_font_assets (assets);
+
+                       auto const id = j->asset_ref().id();
+                       auto already_got = false;
+                       for (auto k: assets) {
+                               if (k->id() == id) {
+                                       already_got = true;
+                               }
+                       }
+
+                       if (!already_got) {
+                               auto o = j->asset_ref().asset();
+                               assets.push_back (o);
+                               /* More Interop special-casing */
+                               auto sub = dynamic_pointer_cast<InteropSubtitleAsset>(o);
+                               if (sub) {
+                                       sub->add_font_assets (assets);
+                               }
                        }
                }
        }
@@ -498,7 +547,7 @@ vector<boost::filesystem::path>
 DCP::directories_from_files (vector<boost::filesystem::path> files)
 {
        vector<boost::filesystem::path> d;
-       BOOST_FOREACH (boost::filesystem::path i, files) {
+       for (auto i: files) {
                if (i.filename() == "ASSETMAP" || i.filename() == "ASSETMAP.xml") {
                        d.push_back (i.parent_path ());
                }