Fix various bugs in subtitle/ccap verification.
[libdcp.git] / src / smpte_subtitle_asset.cc
index 1ce64e5c72141631e52ddc0cc9a7ddf3143078d3..a55b91ae8477e7242d9248666d59bac10500fc20 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012-2019 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of libdcp.
 
     files in the program, then also delete it here.
 */
 
+
 /** @file  src/smpte_subtitle_asset.cc
- *  @brief SMPTESubtitleAsset class.
+ *  @brief SMPTESubtitleAsset class
  */
 
+
 #include "smpte_subtitle_asset.h"
 #include "smpte_load_font_node.h"
 #include "exceptions.h"
 #include <asdcp/KM_util.h>
 #include <asdcp/KM_log.h>
 #include <libxml++/libxml++.h>
-#include <boost/foreach.hpp>
 #include <boost/algorithm/string.hpp>
 
+
 using std::string;
 using std::list;
 using std::vector;
 using std::map;
 using std::shared_ptr;
+using std::dynamic_pointer_cast;
+using std::make_shared;
 using boost::split;
 using boost::is_any_of;
 using boost::shared_array;
-using std::dynamic_pointer_cast;
 using boost::optional;
 using boost::starts_with;
 using namespace dcp;
 
+
 static string const subtitle_smpte_ns = "http://www.smpte-ra.org/schemas/428-7/2010/DCST";
 
+
 SMPTESubtitleAsset::SMPTESubtitleAsset ()
-       : MXF (SMPTE)
-       , _intrinsic_duration (0)
+       : MXF (Standard::SMPTE)
        , _edit_rate (24, 1)
        , _time_code_rate (24)
-       , _xml_id (make_uuid ())
+       , _xml_id (make_uuid())
 {
 
 }
 
-/** Construct a SMPTESubtitleAsset by reading an MXF or XML file.
- *  @param file Filename.
- */
+
 SMPTESubtitleAsset::SMPTESubtitleAsset (boost::filesystem::path file)
        : SubtitleAsset (file)
 {
-       shared_ptr<cxml::Document> xml (new cxml::Document ("SubtitleReel"));
+       auto xml = make_shared<cxml::Document>("SubtitleReel");
 
-       shared_ptr<ASDCP::TimedText::MXFReader> reader (new ASDCP::TimedText::MXFReader ());
-       Kumu::Result_t r = Kumu::RESULT_OK;
+       auto reader = make_shared<ASDCP::TimedText::MXFReader>();
+       auto r = Kumu::RESULT_OK;
        {
                ASDCPErrorSuspender sus;
                r = reader->OpenRead (_file->string().c_str ());
        }
-       if (!ASDCP_FAILURE (r)) {
+       if (!ASDCP_FAILURE(r)) {
                /* MXF-wrapped */
                ASDCP::WriterInfo info;
                reader->FillWriterInfo (info);
@@ -101,13 +103,13 @@ SMPTESubtitleAsset::SMPTESubtitleAsset (boost::filesystem::path file)
                        reader->ReadTimedTextResource (_raw_xml);
                        xml->read_string (_raw_xml);
                        parse_xml (xml);
-                       read_mxf_descriptor (reader, shared_ptr<DecryptionContext> (new DecryptionContext (optional<Key>(), SMPTE)));
+                       read_mxf_descriptor (reader, make_shared<DecryptionContext>(optional<Key>(), Standard::SMPTE));
                }
        } else {
                /* Plain XML */
                try {
                        _raw_xml = dcp::file_to_string (file);
-                       xml.reset (new cxml::Document ("SubtitleReel"));
+                       xml = make_shared<cxml::Document>("SubtitleReel");
                        xml->read_file (file);
                        parse_xml (xml);
                        _id = _xml_id = remove_urn_uuid (xml->string_child ("Id"));
@@ -116,7 +118,7 @@ SMPTESubtitleAsset::SMPTESubtitleAsset (boost::filesystem::path file)
                                ReadError (
                                        String::compose (
                                                "Failed to read subtitle file %1; MXF failed with %2, XML failed with %3",
-                                               file, static_cast<int> (r), e.what ()
+                                               file, static_cast<int>(r), e.what()
                                                )
                                        )
                                );
@@ -125,11 +127,11 @@ SMPTESubtitleAsset::SMPTESubtitleAsset (boost::filesystem::path file)
                /* Try to read PNG files from the same folder that the XML is in; the wisdom of this is
                   debatable, at best...
                */
-               BOOST_FOREACH (shared_ptr<Subtitle> i, _subtitles) {
-                       shared_ptr<SubtitleImage> im = dynamic_pointer_cast<SubtitleImage>(i);
+               for (auto i: _subtitles) {
+                       auto im = dynamic_pointer_cast<SubtitleImage>(i);
                        if (im && im->png_image().size() == 0) {
                                /* Even more dubious; allow <id>.png or urn:uuid:<id>.png */
-                               boost::filesystem::path p = file.parent_path() / String::compose("%1.png", im->id());
+                               auto p = file.parent_path() / String::compose("%1.png", im->id());
                                if (boost::filesystem::is_regular_file(p)) {
                                        im->read_png_file (p);
                                } else if (starts_with (im->id(), "urn:uuid:")) {
@@ -140,17 +142,19 @@ SMPTESubtitleAsset::SMPTESubtitleAsset (boost::filesystem::path file)
                                }
                        }
                }
+               _standard = Standard::SMPTE;
        }
 
        /* Check that all required image data have been found */
-       BOOST_FOREACH (shared_ptr<Subtitle> i, _subtitles) {
-               shared_ptr<SubtitleImage> im = dynamic_pointer_cast<SubtitleImage>(i);
+       for (auto i: _subtitles) {
+               auto im = dynamic_pointer_cast<SubtitleImage>(i);
                if (im && im->png_image().size() == 0) {
                        throw MissingSubtitleImageError (im->id());
                }
        }
 }
 
+
 void
 SMPTESubtitleAsset::parse_xml (shared_ptr<cxml::Document> xml)
 {
@@ -164,7 +168,7 @@ SMPTESubtitleAsset::parse_xml (shared_ptr<cxml::Document> xml)
        _language = xml->optional_string_child ("Language");
 
        /* This is supposed to be two numbers, but a single number has been seen in the wild */
-       string const er = xml->string_child ("EditRate");
+       auto const er = xml->string_child ("EditRate");
        vector<string> er_parts;
        split (er_parts, er, is_any_of (" "));
        if (er_parts.size() == 1) {
@@ -177,24 +181,24 @@ SMPTESubtitleAsset::parse_xml (shared_ptr<cxml::Document> xml)
 
        _time_code_rate = xml->number_child<int> ("TimeCodeRate");
        if (xml->optional_string_child ("StartTime")) {
-               _start_time = Time (xml->string_child ("StartTime"), _time_code_rate);
+               _start_time = Time (xml->string_child("StartTime"), _time_code_rate);
        }
 
        /* Now we need to drop down to xmlpp */
 
-       list<ParseState> ps;
-       xmlpp::Node::NodeList c = xml->node()->get_children ();
-       for (xmlpp::Node::NodeList::const_iterator i = c.begin(); i != c.end(); ++i) {
-               xmlpp::Element const * e = dynamic_cast<xmlpp::Element const *> (*i);
+       vector<ParseState> ps;
+       for (auto i: xml->node()->get_children()) {
+               auto const e = dynamic_cast<xmlpp::Element const *>(i);
                if (e && e->get_name() == "SubtitleList") {
-                       parse_subtitles (e, ps, _time_code_rate, SMPTE);
+                       parse_subtitles (e, ps, _time_code_rate, Standard::SMPTE);
                }
        }
 
        /* Guess intrinsic duration */
-       _intrinsic_duration = latest_subtitle_out().as_editable_units (_edit_rate.numerator / _edit_rate.denominator);
+       _intrinsic_duration = latest_subtitle_out().as_editable_units_ceil(_edit_rate.numerator / _edit_rate.denominator);
 }
 
+
 void
 SMPTESubtitleAsset::read_mxf_descriptor (shared_ptr<ASDCP::TimedText::MXFReader> reader, shared_ptr<DecryptionContext> dec)
 {
@@ -204,7 +208,7 @@ SMPTESubtitleAsset::read_mxf_descriptor (shared_ptr<ASDCP::TimedText::MXFReader>
        /* Load fonts and images */
 
        for (
-               ASDCP::TimedText::ResourceList_t::const_iterator i = descriptor.ResourceList.begin();
+               auto i = descriptor.ResourceList.begin();
                i != descriptor.ResourceList.end();
                ++i) {
 
@@ -213,7 +217,7 @@ SMPTESubtitleAsset::read_mxf_descriptor (shared_ptr<ASDCP::TimedText::MXFReader>
                reader->ReadAncillaryResource (i->ResourceID, buffer, dec->context(), dec->hmac());
 
                char id[64];
-               Kumu::bin2UUIDhex (i->ResourceID, ASDCP::UUIDlen, id, sizeof (id));
+               Kumu::bin2UUIDhex (i->ResourceID, ASDCP::UUIDlen, id, sizeof(id));
 
                shared_array<uint8_t> data (new uint8_t[buffer.Size()]);
                memcpy (data.get(), buffer.RoData(), buffer.Size());
@@ -221,7 +225,7 @@ SMPTESubtitleAsset::read_mxf_descriptor (shared_ptr<ASDCP::TimedText::MXFReader>
                switch (i->Type) {
                case ASDCP::TimedText::MT_OPENTYPE:
                {
-                       list<shared_ptr<SMPTELoadFontNode> >::const_iterator j = _load_font_nodes.begin ();
+                       auto j = _load_font_nodes.begin();
                        while (j != _load_font_nodes.end() && (*j)->urn != id) {
                                ++j;
                        }
@@ -233,7 +237,7 @@ SMPTESubtitleAsset::read_mxf_descriptor (shared_ptr<ASDCP::TimedText::MXFReader>
                }
                case ASDCP::TimedText::MT_PNG:
                {
-                       list<shared_ptr<Subtitle> >::const_iterator j = _subtitles.begin ();
+                       auto j = _subtitles.begin();
                        while (j != _subtitles.end() && ((!dynamic_pointer_cast<SubtitleImage>(*j)) || dynamic_pointer_cast<SubtitleImage>(*j)->id() != id)) {
                                ++j;
                        }
@@ -248,17 +252,17 @@ SMPTESubtitleAsset::read_mxf_descriptor (shared_ptr<ASDCP::TimedText::MXFReader>
                }
        }
 
-       /* Get intrinsic duration */
        _intrinsic_duration = descriptor.ContainerDuration;
 }
 
+
 void
 SMPTESubtitleAsset::set_key (Key key)
 {
        /* See if we already have a key; if we do, and we have a file, we'll already
           have read that file.
        */
-       bool const had_key = static_cast<bool> (_key);
+       auto const had_key = static_cast<bool>(_key);
 
        MXF::set_key (key);
 
@@ -272,8 +276,8 @@ SMPTESubtitleAsset::set_key (Key key)
 
        /* Our data was encrypted; now we can decrypt it */
 
-       shared_ptr<ASDCP::TimedText::MXFReader> reader (new ASDCP::TimedText::MXFReader ());
-       Kumu::Result_t r = reader->OpenRead (_file->string().c_str ());
+       auto reader = make_shared<ASDCP::TimedText::MXFReader>();
+       auto r = reader->OpenRead (_file->string().c_str ());
        if (ASDCP_FAILURE (r)) {
                boost::throw_exception (
                        ReadError (
@@ -282,37 +286,40 @@ SMPTESubtitleAsset::set_key (Key key)
                        );
        }
 
-       shared_ptr<DecryptionContext> dec (new DecryptionContext (key, SMPTE));
+       auto dec = make_shared<DecryptionContext>(key, Standard::SMPTE);
        reader->ReadTimedTextResource (_raw_xml, dec->context(), dec->hmac());
-       shared_ptr<cxml::Document> xml (new cxml::Document ("SubtitleReel"));
+       auto xml = make_shared<cxml::Document>("SubtitleReel");
        xml->read_string (_raw_xml);
        parse_xml (xml);
        read_mxf_descriptor (reader, dec);
 }
 
-list<shared_ptr<LoadFontNode> >
+
+vector<shared_ptr<LoadFontNode>>
 SMPTESubtitleAsset::load_font_nodes () const
 {
-       list<shared_ptr<LoadFontNode> > lf;
-       copy (_load_font_nodes.begin(), _load_font_nodes.end(), back_inserter (lf));
+       vector<shared_ptr<LoadFontNode>> lf;
+       copy (_load_font_nodes.begin(), _load_font_nodes.end(), back_inserter(lf));
        return lf;
 }
 
+
 bool
 SMPTESubtitleAsset::valid_mxf (boost::filesystem::path file)
 {
        ASDCP::TimedText::MXFReader reader;
        Kumu::DefaultLogSink().UnsetFilterFlag(Kumu::LOG_ALLOW_ALL);
-       Kumu::Result_t r = reader.OpenRead (file.string().c_str ());
+       auto r = reader.OpenRead (file.string().c_str ());
        Kumu::DefaultLogSink().SetFilterFlag(Kumu::LOG_ALLOW_ALL);
        return !ASDCP_FAILURE (r);
 }
 
+
 string
 SMPTESubtitleAsset::xml_as_string () const
 {
        xmlpp::Document doc;
-       xmlpp::Element* root = doc.create_root_node ("dcst:SubtitleReel");
+       auto root = doc.create_root_node ("dcst:SubtitleReel");
        root->set_namespace_declaration (subtitle_smpte_ns, "dcst");
        root->set_namespace_declaration ("http://www.w3.org/2001/XMLSchema", "xs");
 
@@ -331,25 +338,25 @@ SMPTESubtitleAsset::xml_as_string () const
        root->add_child("EditRate", "dcst")->add_child_text (_edit_rate.as_string ());
        root->add_child("TimeCodeRate", "dcst")->add_child_text (raw_convert<string> (_time_code_rate));
        if (_start_time) {
-               root->add_child("StartTime", "dcst")->add_child_text (_start_time.get().as_string (SMPTE));
+               root->add_child("StartTime", "dcst")->add_child_text(_start_time.get().as_string(Standard::SMPTE));
        }
 
-       BOOST_FOREACH (shared_ptr<SMPTELoadFontNode> i, _load_font_nodes) {
-               xmlpp::Element* load_font = root->add_child("LoadFont", "dcst");
+       for (auto i: _load_font_nodes) {
+               auto load_font = root->add_child("LoadFont", "dcst");
                load_font->add_child_text ("urn:uuid:" + i->urn);
                load_font->set_attribute ("ID", i->id);
        }
 
-       subtitles_as_xml (root->add_child ("SubtitleList", "dcst"), _time_code_rate, SMPTE);
+       subtitles_as_xml (root->add_child("SubtitleList", "dcst"), _time_code_rate, Standard::SMPTE);
 
        return doc.write_to_string ("UTF-8");
 }
 
-/** Write this content to a MXF file */
+
 void
 SMPTESubtitleAsset::write (boost::filesystem::path p) const
 {
-       EncryptionContext enc (key(), SMPTE);
+       EncryptionContext enc (key(), Standard::SMPTE);
 
        ASDCP::WriterInfo writer_info;
        fill_writer_info (&writer_info, _id);
@@ -360,8 +367,8 @@ SMPTESubtitleAsset::write (boost::filesystem::path p) const
 
        /* Font references */
 
-       BOOST_FOREACH (shared_ptr<dcp::SMPTELoadFontNode> i, _load_font_nodes) {
-               list<Font>::const_iterator j = _fonts.begin ();
+       for (auto i: _load_font_nodes) {
+               auto j = _fonts.begin();
                while (j != _fonts.end() && j->load_id != i->id) {
                        ++j;
                }
@@ -377,8 +384,8 @@ SMPTESubtitleAsset::write (boost::filesystem::path p) const
 
        /* Image subtitle references */
 
-       BOOST_FOREACH (shared_ptr<Subtitle> i, _subtitles) {
-               shared_ptr<SubtitleImage> si = dynamic_pointer_cast<SubtitleImage>(i);
+       for (auto i: _subtitles) {
+               auto si = dynamic_pointer_cast<SubtitleImage>(i);
                if (si) {
                        ASDCP::TimedText::TimedTextResourceDescriptor res;
                        unsigned int c;
@@ -411,8 +418,8 @@ SMPTESubtitleAsset::write (boost::filesystem::path p) const
 
        /* Font payload */
 
-       BOOST_FOREACH (shared_ptr<dcp::SMPTELoadFontNode> i, _load_font_nodes) {
-               list<Font>::const_iterator j = _fonts.begin ();
+       for (auto i: _load_font_nodes) {
+               auto j = _fonts.begin();
                while (j != _fonts.end() && j->load_id != i->id) {
                        ++j;
                }
@@ -422,7 +429,7 @@ SMPTESubtitleAsset::write (boost::filesystem::path p) const
                        buffer.SetData (data_copy.data(), data_copy.size());
                        buffer.Size (j->data.size());
                        r = writer.WriteAncillaryResource (buffer, enc.context(), enc.hmac());
-                       if (ASDCP_FAILURE (r)) {
+                       if (ASDCP_FAILURE(r)) {
                                boost::throw_exception (MXFFileError ("could not write font to timed text resource", p.string(), r));
                        }
                }
@@ -430,8 +437,8 @@ SMPTESubtitleAsset::write (boost::filesystem::path p) const
 
        /* Image subtitle payload */
 
-       BOOST_FOREACH (shared_ptr<Subtitle> i, _subtitles) {
-               shared_ptr<SubtitleImage> si = dynamic_pointer_cast<SubtitleImage>(i);
+       for (auto i: _subtitles) {
+               auto si = dynamic_pointer_cast<SubtitleImage>(i);
                if (si) {
                        ASDCP::TimedText::FrameBuffer buffer;
                        buffer.SetData (si->png_image().data(), si->png_image().size());
@@ -455,23 +462,23 @@ SMPTESubtitleAsset::equals (shared_ptr<const Asset> other_asset, EqualityOptions
                return false;
        }
 
-       shared_ptr<const SMPTESubtitleAsset> other = dynamic_pointer_cast<const SMPTESubtitleAsset> (other_asset);
+       auto other = dynamic_pointer_cast<const SMPTESubtitleAsset>(other_asset);
        if (!other) {
-               note (DCP_ERROR, "Subtitles are in different standards");
+               note (NoteType::ERROR, "Subtitles are in different standards");
                return false;
        }
 
-       list<shared_ptr<SMPTELoadFontNode> >::const_iterator i = _load_font_nodes.begin ();
-       list<shared_ptr<SMPTELoadFontNode> >::const_iterator j = other->_load_font_nodes.begin ();
+       auto i = _load_font_nodes.begin();
+       auto j = other->_load_font_nodes.begin();
 
        while (i != _load_font_nodes.end ()) {
                if (j == other->_load_font_nodes.end ()) {
-                       note (DCP_ERROR, "<LoadFont> nodes differ");
+                       note (NoteType::ERROR, "<LoadFont> nodes differ");
                        return false;
                }
 
                if ((*i)->id != (*j)->id) {
-                       note (DCP_ERROR, "<LoadFont> nodes differ");
+                       note (NoteType::ERROR, "<LoadFont> nodes differ");
                        return false;
                }
 
@@ -480,63 +487,65 @@ SMPTESubtitleAsset::equals (shared_ptr<const Asset> other_asset, EqualityOptions
        }
 
        if (_content_title_text != other->_content_title_text) {
-               note (DCP_ERROR, "Subtitle content title texts differ");
+               note (NoteType::ERROR, "Subtitle content title texts differ");
                return false;
        }
 
        if (_language != other->_language) {
-               note (DCP_ERROR, String::compose("Subtitle languages differ (`%1' vs `%2')", _language.get_value_or("[none]"), other->_language.get_value_or("[none]")));
+               note (NoteType::ERROR, String::compose("Subtitle languages differ (`%1' vs `%2')", _language.get_value_or("[none]"), other->_language.get_value_or("[none]")));
                return false;
        }
 
        if (_annotation_text != other->_annotation_text) {
-               note (DCP_ERROR, "Subtitle annotation texts differ");
+               note (NoteType::ERROR, "Subtitle annotation texts differ");
                return false;
        }
 
        if (_issue_date != other->_issue_date) {
                if (options.issue_dates_can_differ) {
-                       note (DCP_NOTE, "Subtitle issue dates differ");
+                       note (NoteType::NOTE, "Subtitle issue dates differ");
                } else {
-                       note (DCP_ERROR, "Subtitle issue dates differ");
+                       note (NoteType::ERROR, "Subtitle issue dates differ");
                        return false;
                }
        }
 
        if (_reel_number != other->_reel_number) {
-               note (DCP_ERROR, "Subtitle reel numbers differ");
+               note (NoteType::ERROR, "Subtitle reel numbers differ");
                return false;
        }
 
        if (_edit_rate != other->_edit_rate) {
-               note (DCP_ERROR, "Subtitle edit rates differ");
+               note (NoteType::ERROR, "Subtitle edit rates differ");
                return false;
        }
 
        if (_time_code_rate != other->_time_code_rate) {
-               note (DCP_ERROR, "Subtitle time code rates differ");
+               note (NoteType::ERROR, "Subtitle time code rates differ");
                return false;
        }
 
        if (_start_time != other->_start_time) {
-               note (DCP_ERROR, "Subtitle start times differ");
+               note (NoteType::ERROR, "Subtitle start times differ");
                return false;
        }
 
        return true;
 }
 
+
 void
 SMPTESubtitleAsset::add_font (string load_id, dcp::ArrayData data)
 {
        string const uuid = make_uuid ();
        _fonts.push_back (Font(load_id, uuid, data));
-       _load_font_nodes.push_back (shared_ptr<SMPTELoadFontNode> (new SMPTELoadFontNode (load_id, uuid)));
+       _load_font_nodes.push_back (make_shared<SMPTELoadFontNode>(load_id, uuid));
 }
 
+
 void
 SMPTESubtitleAsset::add (shared_ptr<Subtitle> s)
 {
        SubtitleAsset::add (s);
-       _intrinsic_duration = latest_subtitle_out().as_editable_units (_edit_rate.numerator / _edit_rate.denominator);
+       _intrinsic_duration = latest_subtitle_out().as_editable_units_ceil(_edit_rate.numerator / _edit_rate.denominator);
 }