Fix build on OS X.
[libdcp.git] / src / cpl.cc
index fd7056809009efb30ab5c9265f09ed336f49b046..29214ca6b4c9f3458a485f837c1e3cd3dcba8184 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2013 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
 
 */
 
-#include <fstream>
+#include <libxml/parser.h>
 #include "cpl.h"
 #include "parse/cpl.h"
 #include "util.h"
-#include "picture_asset.h"
+#include "mono_picture_asset.h"
+#include "stereo_picture_asset.h"
 #include "sound_asset.h"
 #include "subtitle_asset.h"
 #include "parse/asset_map.h"
 #include "reel.h"
 #include "metadata.h"
+#include "signer.h"
+#include "exceptions.h"
+#include "compose.hpp"
 
 using std::string;
 using std::stringstream;
-using std::ofstream;
 using std::ostream;
 using std::list;
+using std::pair;
+using std::make_pair;
 using boost::shared_ptr;
 using boost::lexical_cast;
+using boost::optional;
 using namespace libdcp;
 
-CPL::CPL (string directory, string name, ContentKind content_kind, int length, int frames_per_second)
+CPL::CPL (boost::filesystem::path directory, string name, ContentKind content_kind, int length, int frames_per_second)
        : _directory (directory)
        , _name (name)
        , _content_kind (content_kind)
        , _length (length)
        , _fps (frames_per_second)
 {
-       _uuid = make_uuid ();
+       _id = make_uuid ();
 }
 
 /** Construct a CPL object from a XML file.
  *  @param directory The directory containing this CPL's DCP.
  *  @param file The CPL XML filename.
- *  @param asset_map The corresponding asset map.
+ *  @param asset_maps AssetMaps to look for assets in.
  *  @param require_mxfs true to throw an exception if a required MXF file does not exist.
  */
-CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMap> asset_map, bool require_mxfs)
+CPL::CPL (boost::filesystem::path directory, string file, list<PathAssetMap> asset_maps, bool require_mxfs)
        : _directory (directory)
        , _content_kind (FEATURE)
        , _length (0)
@@ -72,7 +78,10 @@ CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMa
        _name = cpl->annotation_text;
        _content_kind = cpl->content_kind;
 
-       for (list<shared_ptr<libdcp::parse::Reel> >::iterator i = cpl->reels.begin(); i != cpl->reels.end(); ++i) {
+       /* Trim urn:uuid: off the front */
+       _id = cpl->id.substr (9);
+
+       for (list<shared_ptr<parse::Reel> >::iterator i = cpl->reels.begin(); i != cpl->reels.end(); ++i) {
 
                shared_ptr<parse::Picture> p;
 
@@ -98,14 +107,18 @@ CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMa
                if (!(*i)->asset_list->main_stereoscopic_picture && p->edit_rate == p->frame_rate) {
 
                        try {
-                               picture.reset (new MonoPictureAsset (
-                                                      _directory,
-                                                      asset_map->asset_from_id (p->id)->chunks.front()->path
-                                                      )
-                                       );
+                               pair<string, shared_ptr<const parse::AssetMapAsset> > asset = asset_from_id (asset_maps, p->id);
+
+                               picture.reset (new MonoPictureAsset (asset.first, asset.second->chunks.front()->path));
 
+                               picture->read ();
+                               picture->set_edit_rate (_fps);
                                picture->set_entry_point (p->entry_point);
                                picture->set_duration (p->duration);
+                               if (p->key_id.length() > 9) {
+                                       /* Trim urn:uuid: */
+                                       picture->set_key_id (p->key_id.substr (9));
+                               }
                        } catch (MXFFileError) {
                                if (require_mxfs) {
                                        throw;
@@ -114,16 +127,18 @@ CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMa
                        
                } else {
                        try {
-                               picture.reset (new StereoPictureAsset (
-                                                      _directory,
-                                                      asset_map->asset_from_id (p->id)->chunks.front()->path,
-                                                      _fps,
-                                                      p->duration
-                                                      )
-                                       );
+                               pair<string, shared_ptr<const parse::AssetMapAsset> > asset = asset_from_id (asset_maps, p->id);
 
+                               picture.reset (new StereoPictureAsset (asset.first, asset.second->chunks.front()->path));
+
+                               picture->read ();
+                               picture->set_edit_rate (_fps);
                                picture->set_entry_point (p->entry_point);
                                picture->set_duration (p->duration);
+                               if (p->key_id.length() > 9) {
+                                       /* Trim urn:uuid: */
+                                       picture->set_key_id (p->key_id.substr (9));
+                               }
                                
                        } catch (MXFFileError) {
                                if (require_mxfs) {
@@ -136,14 +151,18 @@ CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMa
                if ((*i)->asset_list->main_sound) {
                        
                        try {
-                               sound.reset (new SoundAsset (
-                                                    _directory,
-                                                    asset_map->asset_from_id ((*i)->asset_list->main_sound->id)->chunks.front()->path
-                                                    )
-                                       );
-
-                               sound->set_entry_point ((*i)->asset_list->main_sound->entry_point);
-                               sound->set_duration ((*i)->asset_list->main_sound->duration);
+                               pair<string, shared_ptr<const parse::AssetMapAsset> > asset = asset_from_id (asset_maps, (*i)->asset_list->main_sound->id);
+                       
+                               sound.reset (new SoundAsset (asset.first, asset.second->chunks.front()->path));
+                               shared_ptr<parse::MainSound> s = (*i)->asset_list->main_sound;
+
+                               sound->read ();
+                               sound->set_entry_point (s->entry_point);
+                               sound->set_duration (s->duration);
+                               if (s->key_id.length() > 9) {
+                                       /* Trim urn:uuid: */
+                                       sound->set_key_id (s->key_id.substr (9));
+                               }
                        } catch (MXFFileError) {
                                if (require_mxfs) {
                                        throw;
@@ -153,11 +172,9 @@ CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMa
 
                if ((*i)->asset_list->main_subtitle) {
                        
-                       subtitle.reset (new SubtitleAsset (
-                                               _directory,
-                                               asset_map->asset_from_id ((*i)->asset_list->main_subtitle->id)->chunks.front()->path
-                                               )
-                               );
+                       pair<string, shared_ptr<const parse::AssetMapAsset> > asset = asset_from_id (asset_maps, (*i)->asset_list->main_subtitle->id);
+
+                       subtitle.reset (new SubtitleAsset (asset.first, asset.second->chunks.front()->path));
 
                        subtitle->set_entry_point ((*i)->asset_list->main_subtitle->entry_point);
                        subtitle->set_duration ((*i)->asset_list->main_subtitle->duration);
@@ -168,44 +185,60 @@ CPL::CPL (string directory, string file, shared_ptr<const libdcp::parse::AssetMa
 }
 
 void
-CPL::add_reel (shared_ptr<const Reel> reel)
+CPL::add_reel (shared_ptr<Reel> reel)
 {
        _reels.push_back (reel);
 }
 
 void
-CPL::write_xml (XMLMetadata const & metadata) const
+CPL::write_xml (bool interop, XMLMetadata const & metadata, shared_ptr<const Signer> signer) const
 {
        boost::filesystem::path p;
        p /= _directory;
        stringstream s;
-       s << _uuid << "_cpl.xml";
+       s << _id << "_cpl.xml";
        p /= s.str();
 
        xmlpp::Document doc;
-       xmlpp::Element* root = doc.create_root_node ("CompositionPlaylist", "http://www.smpte-ra.org/schemas/429-7/2006/CPL");
-       root->add_child("Id")->add_child_text ("urn:uuid:" + _uuid);
+       xmlpp::Element* root;
+       if (interop) {
+               root = doc.create_root_node ("CompositionPlaylist", "http://www.digicine.com/PROTO-ASDCP-CPL-20040511#");
+       } else {
+               root = doc.create_root_node ("CompositionPlaylist", "http://www.smpte-ra.org/schemas/429-7/2006/CPL");
+       }
+
+       if (signer) {
+               root->set_namespace_declaration ("http://www.w3.org/2000/09/xmldsig#", "dsig");
+       }
+       
+       root->add_child("Id")->add_child_text ("urn:uuid:" + _id);
        root->add_child("AnnotationText")->add_child_text (_name);
        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("ContentTitleText")->add_child_text (_name);
        root->add_child("ContentKind")->add_child_text (content_kind_to_string (_content_kind));
        {
                xmlpp::Node* cv = root->add_child ("ContentVersion");
-               cv->add_child ("Id")->add_child_text ("urn:uri:" + _uuid + "_" + metadata.issue_date);
-               cv->add_child ("LabelText")->add_child_text (_uuid + "_" + metadata.issue_date);
+               cv->add_child ("Id")->add_child_text ("urn:uri:" + _id + "_" + metadata.issue_date);
+               cv->add_child ("LabelText")->add_child_text (_id + "_" + metadata.issue_date);
        }
        root->add_child("RatingList");
 
-       xmlpp::Node* reel_list = root->add_child ("ReelList");
+       xmlpp::Element* reel_list = root->add_child ("ReelList");
        
-       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
-               (*i)->write_to_cpl (reel_list);
+       for (list<shared_ptr<Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+               (*i)->write_to_cpl (reel_list, interop);
        }
 
-       doc.write_to_file_formatted (p.string (), "UTF-8");
+       if (signer) {
+               signer->sign (root, interop);
+       }
 
-       _digest = make_digest (p.string ());
+       /* This must not be the _formatted version otherwise signature digests will be wrong */
+       doc.write_to_file (p.string (), "UTF-8");
+
+       _digest = make_digest (p.string (), 0);
        _length = boost::filesystem::file_size (p.string ());
 }
 
@@ -213,7 +246,7 @@ void
 CPL::write_to_pkl (xmlpp::Node* node) const
 {
        xmlpp::Node* asset = node->add_child ("Asset");
-       asset->add_child("Id")->add_child_text ("urn:uuid:" + _uuid);
+       asset->add_child("Id")->add_child_text ("urn:uuid:" + _id);
        asset->add_child("Hash")->add_child_text (_digest);
        asset->add_child("Size")->add_child_text (lexical_cast<string> (_length));
        asset->add_child("Type")->add_child_text ("text/xml");
@@ -223,7 +256,7 @@ list<shared_ptr<const Asset> >
 CPL::assets () const
 {
        list<shared_ptr<const Asset> > a;
-       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+       for (list<shared_ptr<Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
                if ((*i)->main_picture ()) {
                        a.push_back ((*i)->main_picture ());
                }
@@ -242,10 +275,10 @@ void
 CPL::write_to_assetmap (xmlpp::Node* node) const
 {
        xmlpp::Node* asset = node->add_child ("Asset");
-       asset->add_child("Id")->add_child_text ("urn:uuid:" + _uuid);
+       asset->add_child("Id")->add_child_text ("urn:uuid:" + _id);
        xmlpp::Node* chunk_list = asset->add_child ("ChunkList");
        xmlpp::Node* chunk = chunk_list->add_child ("Chunk");
-       chunk->add_child("Path")->add_child_text (_uuid + "_cpl.xml");
+       chunk->add_child("Path")->add_child_text (_id + "_cpl.xml");
        chunk->add_child("VolumeIndex")->add_child_text ("1");
        chunk->add_child("Offset")->add_child_text("0");
        chunk->add_child("Length")->add_child_text(lexical_cast<string> (_length));
@@ -269,22 +302,22 @@ CPL::equals (CPL const & other, EqualityOptions opt, boost::function<void (NoteT
        }
 
        if (_fps != other._fps) {
-               note (ERROR, "frames per second differ");
+               note (ERROR, String::compose ("frames per second differ (%1 vs %2)", _fps, other._fps));
                return false;
        }
 
        if (_length != other._length) {
-               note (ERROR, "lengths differ");
-               return false;
+               stringstream s;
+               note (ERROR, String::compose ("lengths differ (%1 vs %2)", _length, other._length));
        }
 
        if (_reels.size() != other._reels.size()) {
-               note (ERROR, "reel counts differ");
+               note (ERROR, String::compose ("reel counts differ (%1 vs %2)", _reels.size(), other._reels.size()));
                return false;
        }
        
-       list<shared_ptr<const Reel> >::const_iterator a = _reels.begin ();
-       list<shared_ptr<const Reel> >::const_iterator b = other._reels.begin ();
+       list<shared_ptr<Reel> >::const_iterator a = _reels.begin ();
+       list<shared_ptr<Reel> >::const_iterator b = other._reels.begin ();
        
        while (a != _reels.end ()) {
                if (!(*a)->equals (*b, opt, note)) {
@@ -296,3 +329,45 @@ CPL::equals (CPL const & other, EqualityOptions opt, boost::function<void (NoteT
 
        return true;
 }
+
+/** @return true if we have any encrypted content */
+bool
+CPL::encrypted () const
+{
+       for (list<shared_ptr<Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+               if ((*i)->encrypted ()) {
+                       return true;
+               }
+       }
+
+       return false;
+}
+
+void
+CPL::add_kdm (KDM const & kdm)
+{
+       for (list<shared_ptr<Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+               (*i)->add_kdm (kdm);
+       }
+}
+
+pair<string, shared_ptr<const parse::AssetMapAsset> >
+CPL::asset_from_id (list<PathAssetMap> asset_maps, string id) const
+{
+       for (list<PathAssetMap>::const_iterator i = asset_maps.begin(); i != asset_maps.end(); ++i) {
+               shared_ptr<parse::AssetMapAsset> a = i->second->asset_from_id (id);
+               if (a) {
+                       return make_pair (i->first, a);
+               }
+       }
+
+       return make_pair ("", shared_ptr<const parse::AssetMapAsset> ());
+}
+
+void
+CPL::set_mxf_keys (Key key)
+{
+       for (list<shared_ptr<Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+               (*i)->set_mxf_keys (key);
+       }
+}