Merge master
[libdcp.git] / src / dcp.cc
index 767ab401c8e62f250fef76a44702b36b42947c49..b3048bae0475fe8b54d7da778c049864d9f4d8ad 100644 (file)
 #include <cassert>
 #include <iostream>
 #include <boost/filesystem.hpp>
+#include <boost/lexical_cast.hpp>
+#include <boost/algorithm/string.hpp>
 #include <libxml++/libxml++.h>
+#include <xmlsec/xmldsig.h>
+#include <xmlsec/app.h>
 #include "dcp.h"
 #include "asset.h"
 #include "sound_asset.h"
 #include "util.h"
 #include "metadata.h"
 #include "exceptions.h"
-#include "cpl.h"
-#include "pkl.h"
+#include "cpl_file.h"
+#include "pkl_file.h"
 #include "asset_map.h"
 #include "reel.h"
 
-using namespace std;
-using namespace boost;
+using std::string;
+using std::list;
+using std::stringstream;
+using std::ofstream;
+using std::ostream;
+using boost::shared_ptr;
 using namespace libdcp;
 
-DCP::DCP (string directory, string name, ContentKind content_kind, int fps, int length)
+DCP::DCP (string directory)
        : _directory (directory)
-       , _name (name)
-       , _content_kind (content_kind)
-       , _fps (fps)
-       , _length (length)
-{
-       
-}
-
-void
-DCP::add_reel (shared_ptr<const Reel> reel)
 {
-       _reels.push_back (reel);
+       boost::filesystem::create_directories (directory);
 }
 
 void
-DCP::write_xml () const
+DCP::write_xml (shared_ptr<Encryption> crypt) const
 {
-       string cpl_uuid = make_uuid ();
-       string cpl_path = write_cpl (cpl_uuid);
-       int cpl_length = filesystem::file_size (cpl_path);
-       string cpl_digest = make_digest (cpl_path, 0);
+       for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+               (*i)->write_xml (crypt);
+       }
 
        string pkl_uuid = make_uuid ();
-       string pkl_path = write_pkl (pkl_uuid, cpl_uuid, cpl_digest, cpl_length);
+       string pkl_path = write_pkl (pkl_uuid, crypt);
        
        write_volindex ();
-       write_assetmap (cpl_uuid, cpl_length, pkl_uuid, filesystem::file_size (pkl_path));
-}
-
-string
-DCP::write_cpl (string cpl_uuid) const
-{
-       filesystem::path p;
-       p /= _directory;
-       stringstream s;
-       s << cpl_uuid << "_cpl.xml";
-       p /= s.str();
-       ofstream cpl (p.string().c_str());
-       
-       cpl << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
-           << "<CompositionPlaylist xmlns=\"http://www.smpte-ra.org/schemas/429-7/2006/CPL\">\n"
-           << "  <Id>urn:uuid:" << cpl_uuid << "</Id>\n"
-           << "  <AnnotationText>" << _name << "</AnnotationText>\n"
-           << "  <IssueDate>" << Metadata::instance()->issue_date << "</IssueDate>\n"
-           << "  <Creator>" << Metadata::instance()->creator << "</Creator>\n"
-           << "  <ContentTitleText>" << _name << "</ContentTitleText>\n"
-           << "  <ContentKind>" << content_kind_to_string (_content_kind) << "</ContentKind>\n"
-           << "  <ContentVersion>\n"
-           << "    <Id>urn:uri:" << cpl_uuid << "_" << Metadata::instance()->issue_date << "</Id>\n"
-           << "    <LabelText>" << cpl_uuid << "_" << Metadata::instance()->issue_date << "</LabelText>\n"
-           << "  </ContentVersion>\n"
-           << "  <RatingList/>\n"
-           << "  <ReelList>\n";
-
-       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
-               (*i)->write_to_cpl (cpl);
-       }
-
-       cpl << "      </AssetList>\n"
-           << "    </Reel>\n"
-           << "  </ReelList>\n"
-           << "</CompositionPlaylist>\n";
-
-       return p.string ();
+       write_assetmap (pkl_uuid, boost::filesystem::file_size (pkl_path));
 }
 
 std::string
-DCP::write_pkl (string pkl_uuid, string cpl_uuid, string cpl_digest, int cpl_length) const
+DCP::write_pkl (string pkl_uuid, shared_ptr<Encryption> crypt) const
 {
-       filesystem::path p;
+       assert (!_cpls.empty ());
+       
+       boost::filesystem::path p;
        p /= _directory;
        stringstream s;
        s << pkl_uuid << "_pkl.xml";
        p /= s.str();
-       ofstream pkl (p.string().c_str());
-
-       pkl << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
-           << "<PackingList xmlns=\"http://www.smpte-ra.org/schemas/429-8/2007/PKL\">\n"
-           << "  <Id>urn:uuid:" << pkl_uuid << "</Id>\n"
-           << "  <AnnotationText>" << _name << "</AnnotationText>\n"
-           << "  <IssueDate>" << Metadata::instance()->issue_date << "</IssueDate>\n"
-           << "  <Issuer>" << Metadata::instance()->issuer << "</Issuer>\n"
-           << "  <Creator>" << Metadata::instance()->creator << "</Creator>\n"
-           << "  <AssetList>\n";
 
-       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
-               (*i)->write_to_pkl (pkl);
+       xmlpp::Document doc;
+       xmlpp::Element* pkl = doc.create_root_node("PackingList", "http://www.smpte-ra.org/schemas/429-8/2007/PKL");
+       if (crypt) {
+               pkl->set_namespace_declaration ("http://www.w3.org/2000/09/xmldsig#", "dsig");
        }
 
-       pkl << "    <Asset>\n"
-           << "      <Id>urn:uuid:" << cpl_uuid << "</Id>\n"
-           << "      <Hash>" << cpl_digest << "</Hash>\n"
-           << "      <Size>" << cpl_length << "</Size>\n"
-           << "      <Type>text/xml</Type>\n"
-           << "    </Asset>\n";
+       pkl->add_child("Id")->add_child_text ("urn:uuid:" + pkl_uuid);
+       /* XXX: this is a bit of a hack */
+       pkl->add_child("AnnotationText")->add_child_text(_cpls.front()->name());
+       pkl->add_child("IssueDate")->add_child_text (Metadata::instance()->issue_date);
+       pkl->add_child("Issuer")->add_child_text (Metadata::instance()->issuer);
+       pkl->add_child("Creator")->add_child_text (Metadata::instance()->creator);
+
+       {
+               xmlpp::Element* asset_list = pkl->add_child("AssetList");
+               list<shared_ptr<const Asset> > a = assets ();
+               for (list<shared_ptr<const Asset> >::const_iterator i = a.begin(); i != a.end(); ++i) {
+                       (*i)->write_to_pkl (asset_list);
+               }
 
-       pkl << "  </AssetList>\n"
-           << "</PackingList>\n";
+               for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+                       (*i)->write_to_pkl (asset_list);
+               }
+       }
+
+       if (crypt) {
+               sign (pkl, crypt->certificates, crypt->signer_key);
+       }
+               
+       doc.write_to_file_formatted (p.string(), "UTF-8");
 
        return p.string ();
 }
@@ -152,7 +121,7 @@ DCP::write_pkl (string pkl_uuid, string cpl_uuid, string cpl_digest, int cpl_len
 void
 DCP::write_volindex () const
 {
-       filesystem::path p;
+       boost::filesystem::path p;
        p /= _directory;
        p /= "VOLINDEX.xml";
        ofstream vi (p.string().c_str());
@@ -164,9 +133,9 @@ DCP::write_volindex () const
 }
 
 void
-DCP::write_assetmap (string cpl_uuid, int cpl_length, string pkl_uuid, int pkl_length) const
+DCP::write_assetmap (string pkl_uuid, int pkl_length) const
 {
-       filesystem::path p;
+       boost::filesystem::path p;
        p /= _directory;
        p /= "ASSETMAP.xml";
        ofstream am (p.string().c_str());
@@ -192,20 +161,13 @@ DCP::write_assetmap (string cpl_uuid, int cpl_length, string pkl_uuid, int pkl_l
           << "        </Chunk>\n"
           << "      </ChunkList>\n"
           << "    </Asset>\n";
-
-       am << "    <Asset>\n"
-          << "      <Id>urn:uuid:" << cpl_uuid << "</Id>\n"
-          << "      <ChunkList>\n"
-          << "        <Chunk>\n"
-          << "          <Path>" << cpl_uuid << "_cpl.xml</Path>\n"
-          << "          <VolumeIndex>1</VolumeIndex>\n"
-          << "          <Offset>0</Offset>\n"
-          << "          <Length>" << cpl_length << "</Length>\n"
-          << "        </Chunk>\n"
-          << "      </ChunkList>\n"
-          << "    </Asset>\n";
        
-       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+       for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+               (*i)->write_to_assetmap (am);
+       }
+
+       list<shared_ptr<const Asset> > a = assets ();
+       for (list<shared_ptr<const Asset> >::const_iterator i = a.begin(); i != a.end(); ++i) {
                (*i)->write_to_assetmap (am);
        }
 
@@ -214,55 +176,173 @@ DCP::write_assetmap (string cpl_uuid, int cpl_length, string pkl_uuid, int pkl_l
 }
 
 
-DCP::DCP (string directory)
-       : _directory (directory)
+void
+DCP::read (bool require_mxfs)
 {
        Files files;
-       scan (files, directory);
 
-       if (files.cpl.empty ()) {
-               throw FileError ("no CPL file found", "");
+       shared_ptr<AssetMap> asset_map;
+       try {
+               boost::filesystem::path p = _directory;
+               p /= "ASSETMAP";
+               if (boost::filesystem::exists (p)) {
+                       asset_map.reset (new AssetMap (p.string ()));
+               } else {
+                       p = _directory;
+                       p /= "ASSETMAP.xml";
+                       if (boost::filesystem::exists (p)) {
+                               asset_map.reset (new AssetMap (p.string ()));
+                       } else {
+                               throw DCPReadError ("could not find AssetMap file");
+                       }
+               }
+               
+       } catch (FileError& e) {
+               throw FileError ("could not load AssetMap file", files.asset_map);
        }
 
-       if (files.pkl.empty ()) {
-               throw FileError ("no PKL file found", "");
-       }
+       for (list<shared_ptr<AssetMapAsset> >::const_iterator i = asset_map->assets.begin(); i != asset_map->assets.end(); ++i) {
+               if ((*i)->chunks.size() != 1) {
+                       throw XMLError ("unsupported asset chunk count");
+               }
 
-       if (files.asset_map.empty ()) {
-               throw FileError ("no AssetMap file found", "");
-       }
+               boost::filesystem::path t = _directory;
+               t /= (*i)->chunks.front()->path;
+               
+               if (boost::algorithm::ends_with (t.string(), ".mxf") || boost::algorithm::ends_with (t.string(), ".ttf")) {
+                       continue;
+               }
 
-       /* Read the XML */
-       shared_ptr<CPL> cpl;
-       try {
-               cpl.reset (new CPL (files.cpl));
-       } catch (FileError& e) {
-               throw FileError ("could not load CPL file", files.cpl);
+               xmlpp::DomParser* p = new xmlpp::DomParser;
+               try {
+                       p->parse_file (t.string());
+               } catch (std::exception& e) {
+                       delete p;
+                       continue;
+               }
+
+               string const root = p->get_document()->get_root_node()->get_name ();
+               delete p;
+
+               if (root == "CompositionPlaylist") {
+                       files.cpls.push_back (t.string());
+               } else if (root == "PackingList") {
+                       if (files.pkl.empty ()) {
+                               files.pkl = t.string();
+                       } else {
+                               throw DCPReadError ("duplicate PKLs found");
+                       }
+               }
+       }
+       
+       if (files.cpls.empty ()) {
+               throw FileError ("no CPL files found", "");
        }
 
-       shared_ptr<PKL> pkl;
-       try {
-               pkl.reset (new PKL (files.pkl));
-       } catch (FileError& e) {
-               throw FileError ("could not load PKL file", files.pkl);
+       if (files.pkl.empty ()) {
+               throw FileError ("no PKL file found", "");
        }
 
-       shared_ptr<AssetMap> asset_map;
+       shared_ptr<PKLFile> pkl;
        try {
-               asset_map.reset (new AssetMap (files.asset_map));
+               pkl.reset (new PKLFile (files.pkl));
        } catch (FileError& e) {
-               throw FileError ("could not load AssetMap file", files.asset_map);
+               throw FileError ("could not load PKL file", files.pkl);
        }
 
        /* Cross-check */
        /* XXX */
 
+       for (list<string>::iterator i = files.cpls.begin(); i != files.cpls.end(); ++i) {
+               _cpls.push_back (shared_ptr<CPL> (new CPL (_directory, *i, asset_map, require_mxfs)));
+       }
+}
+
+bool
+DCP::equals (DCP const & other, EqualityOptions opt, list<string>& notes) const
+{
+       if (_cpls.size() != other._cpls.size()) {
+               notes.push_back ("CPL counts differ");
+               return false;
+       }
+
+       list<shared_ptr<const CPL> >::const_iterator a = _cpls.begin ();
+       list<shared_ptr<const CPL> >::const_iterator b = other._cpls.begin ();
+
+       while (a != _cpls.end ()) {
+               if (!(*a)->equals (*b->get(), opt, notes)) {
+                       return false;
+               }
+               ++a;
+               ++b;
+       }
+
+       return true;
+}
+
+
+void
+DCP::add_cpl (shared_ptr<CPL> cpl)
+{
+       _cpls.push_back (cpl);
+}
+
+class AssetComparator
+{
+public:
+       bool operator() (shared_ptr<const Asset> a, shared_ptr<const Asset> b) {
+               return a->uuid() < b->uuid();
+       }
+};
+
+list<shared_ptr<const Asset> >
+DCP::assets () const
+{
+       list<shared_ptr<const Asset> > a;
+       for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+               list<shared_ptr<const Asset> > t = (*i)->assets ();
+               a.merge (t);
+       }
+
+       a.sort (AssetComparator ());
+       a.unique ();
+       return a;
+}
+
+CPL::CPL (string 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 ();
+}
+
+/** 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 require_mxfs true to throw an exception if a required MXF file does not exist.
+ */
+CPL::CPL (string directory, string file, shared_ptr<const AssetMap> asset_map, bool require_mxfs)
+       : _directory (directory)
+       , _content_kind (FEATURE)
+       , _length (0)
+       , _fps (0)
+{
+       /* Read the XML */
+       shared_ptr<CPLFile> cpl;
+       try {
+               cpl.reset (new CPLFile (file));
+       } catch (FileError& e) {
+               throw FileError ("could not load CPL file", file);
+       }
+       
        /* Now cherry-pick the required bits into our own data structure */
        
        _name = cpl->annotation_text;
        _content_kind = cpl->content_kind;
-       _length = 0;
-       _fps = 0;
 
        for (list<shared_ptr<CPLReel> >::iterator i = cpl->reels.begin(); i != cpl->reels.end(); ++i) {
 
@@ -274,162 +354,373 @@ DCP::DCP (string directory)
                        p = (*i)->asset_list->main_stereoscopic_picture;
                }
                
-               assert (_fps == 0 || _fps == p->frame_rate.numerator);
-               _fps = p->frame_rate.numerator;
+               _fps = p->edit_rate.numerator;
                _length += p->duration;
 
-               string n = pkl->asset_from_id (p->id)->original_file_name;
-               if (n.empty ()) {
-                       n = p->annotation_text;
-               }
-               
                shared_ptr<PictureAsset> picture;
                shared_ptr<SoundAsset> sound;
                shared_ptr<SubtitleAsset> subtitle;
-               
-               if ((*i)->asset_list->main_picture) {
 
-                       picture.reset (new MonoPictureAsset (
-                                              _directory,
-                                              n,
-                                              _fps,
-                                              (*i)->asset_list->main_picture->duration
-                                              )
-                               );
-                       
-               } else if ((*i)->asset_list->main_stereoscopic_picture) {
+               /* Some rather twisted logic to decide if we are 3D or not;
+                  some DCPs give a MainStereoscopicPicture to indicate 3D, others
+                  just have a FrameRate twice the EditRate and apparently
+                  expect you to divine the fact that they are hence 3D.
+               */
+
+               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,
+                                                      _fps,
+                                                      (*i)->asset_list->main_picture->entry_point,
+                                                      (*i)->asset_list->main_picture->duration
+                                                      )
+                                       );
+                       } catch (MXFFileError) {
+                               if (require_mxfs) {
+                                       throw;
+                               }
+                       }
                        
-                       picture.reset (new StereoPictureAsset (
-                                              _directory,
-                                              n,
-                                              _fps,
-                                              (*i)->asset_list->main_stereoscopic_picture->duration
-                                              )
-                               );
+               } else {
+
+                       try {
+                               picture.reset (new StereoPictureAsset (
+                                                      _directory,
+                                                      asset_map->asset_from_id (p->id)->chunks.front()->path,
+                                                      _fps,
+                                                      p->entry_point,
+                                                      p->duration
+                                                      )
+                                       );
+                       } catch (MXFFileError) {
+                               if (require_mxfs) {
+                                       throw;
+                               }
+                       }
                        
                }
                
                if ((*i)->asset_list->main_sound) {
                        
-                       n = pkl->asset_from_id ((*i)->asset_list->main_sound->id)->original_file_name;
-                       if (n.empty ()) {
-                               n = (*i)->asset_list->main_sound->annotation_text;
+                       try {
+                               sound.reset (new SoundAsset (
+                                                    _directory,
+                                                    asset_map->asset_from_id ((*i)->asset_list->main_sound->id)->chunks.front()->path,
+                                                    _fps,
+                                                    (*i)->asset_list->main_sound->entry_point,
+                                                    (*i)->asset_list->main_sound->duration
+                                                    )
+                                       );
+                       } catch (MXFFileError) {
+                               if (require_mxfs) {
+                                       throw;
+                               }
                        }
-                       
-                       sound.reset (new SoundAsset (
-                                            _directory,
-                                            n,
-                                            _fps,
-                                            (*i)->asset_list->main_sound->duration
-                                            )
-                               );
                }
 
-               assert (files.subtitles.size() < 2);
-
-               if (!files.subtitles.empty ()) {
-                       string const l = files.subtitles.front().substr (_directory.length ());
-                       subtitle.reset (new SubtitleAsset (_directory, l));
+               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
+                                               )
+                               );
                }
-
+                       
                _reels.push_back (shared_ptr<Reel> (new Reel (picture, sound, subtitle)));
        }
 }
 
+void
+CPL::add_reel (shared_ptr<const Reel> reel)
+{
+       _reels.push_back (reel);
+}
 
 void
-DCP::scan (Files& files, string directory) const
+CPL::write_xml (shared_ptr<Encryption> crypt) const
 {
-       for (filesystem::directory_iterator i = filesystem::directory_iterator(directory); i != filesystem::directory_iterator(); ++i) {
-               
-               string const t = i->path().string ();
+       boost::filesystem::path p;
+       p /= _directory;
+       stringstream s;
+       s << _uuid << "_cpl.xml";
+       p /= s.str();
 
-               if (filesystem::is_directory (*i)) {
-                       scan (files, t);
-                       continue;
-               }
+       xmlpp::Document doc;
+       xmlpp::Element* cpl = doc.create_root_node("CompositionPlaylist", "http://www.smpte-ra.org/schemas/429-7/2006/CPL");
 
-               if (ends_with (t, ".mxf") || ends_with (t, ".ttf")) {
-                       continue;
-               }
+       if (crypt) {
+               cpl->set_namespace_declaration ("http://www.w3.org/2000/09/xmldsig#", "dsig");
+       }
 
-               xmlpp::DomParser* p = new xmlpp::DomParser;
+       cpl->add_child("Id")->add_child_text ("urn:uuid:" + _uuid);
+       cpl->add_child("AnnotationText")->add_child_text (_name);
+       cpl->add_child("IssueDate")->add_child_text (Metadata::instance()->issue_date);
+       cpl->add_child("Creator")->add_child_text (Metadata::instance()->creator);
+       cpl->add_child("ContentTitleText")->add_child_text (_name);
+       cpl->add_child("ContentKind")->add_child_text (content_kind_to_string (_content_kind));
+
+       {
+               xmlpp::Element* cv = cpl->add_child ("ContentVersion");
+               cv->add_child("Id")->add_child_text ("urn:uri:" + _uuid + "_" + Metadata::instance()->issue_date);
+               cv->add_child("LabelText")->add_child_text (_uuid + "_" + Metadata::instance()->issue_date);
+       }
 
-               try {
-                       p->parse_file (t);
-               } catch (std::exception& e) {
-                       delete p;
-                       continue;
-               }
-               
-               if (!p) {
-                       delete p;
-                       continue;
-               }
+       cpl->add_child("RatingList");
 
-               string const root = p->get_document()->get_root_node()->get_name ();
-               delete p;
-               
-               if (root == "CompositionPlaylist") {
-                       if (files.cpl.empty ()) {
-                               files.cpl = t;
-                       } else {
-                               throw DCPReadError ("duplicate CPLs found");
-                       }
-               } else if (root == "PackingList") {
-                       if (files.pkl.empty ()) {
-                               files.pkl = t;
-                       } else {
-                               throw DCPReadError ("duplicate PKLs found");
-                       }
-               } else if (root == "AssetMap") {
-                       if (files.asset_map.empty ()) {
-                               files.asset_map = t;
-                       } else {
-                               throw DCPReadError ("duplicate AssetMaps found");
-                       }
-                       files.asset_map = t;
-               } else if (root == "DCSubtitle") {
-                       files.subtitles.push_back (t);
-               }
+       xmlpp::Element* reel_list = cpl->add_child("ReelList");
+       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+               (*i)->write_to_cpl (reel_list);
+       }
+
+       if (crypt) {
+               sign (cpl, crypt->certificates, crypt->signer_key);
        }
+
+       doc.write_to_file_formatted (p.string(), "UTF-8");
+
+       _digest = make_digest (p.string ());
+       _length = boost::filesystem::file_size (p.string ());
 }
 
+void
+CPL::write_to_pkl (xmlpp::Element* p) const
+{
+       xmlpp::Element* asset = p->add_child("Asset");
+       asset->add_child("Id")->add_child_text("urn:uuid:" + _uuid);
+       asset->add_child("Hash")->add_child_text(_digest);
+       asset->add_child("Size")->add_child_text(boost::lexical_cast<string> (_length));
+       asset->add_child("Type")->add_child_text("text/xml");
+}
 
-list<string>
-DCP::equals (DCP const & other, EqualityOptions opt) const
+list<shared_ptr<const Asset> >
+CPL::assets () const
 {
-       list<string> notes;
-       
-       if (opt.flags & LIBDCP_METADATA) {
-               if (_name != other._name) {
-                       notes.push_back ("names differ");
-               }
-               if (_content_kind != other._content_kind) {
-                       notes.push_back ("content kinds differ");
+       list<shared_ptr<const Asset> > a;
+       for (list<shared_ptr<const Reel> >::const_iterator i = _reels.begin(); i != _reels.end(); ++i) {
+               if ((*i)->main_picture ()) {
+                       a.push_back ((*i)->main_picture ());
                }
-               if (_fps != other._fps) {
-                       notes.push_back ("frames per second differ");
+               if ((*i)->main_sound ()) {
+                       a.push_back ((*i)->main_sound ());
                }
-               if (_length != other._length) {
-                       notes.push_back ("lengths differ");
+               if ((*i)->main_subtitle ()) {
+                       a.push_back ((*i)->main_subtitle ());
                }
        }
 
+       return a;
+}
+
+void
+CPL::write_to_assetmap (ostream& s) const
+{
+       s << "    <Asset>\n"
+         << "      <Id>urn:uuid:" << _uuid << "</Id>\n"
+         << "      <ChunkList>\n"
+         << "        <Chunk>\n"
+         << "          <Path>" << _uuid << "_cpl.xml</Path>\n"
+         << "          <VolumeIndex>1</VolumeIndex>\n"
+         << "          <Offset>0</Offset>\n"
+         << "          <Length>" << _length << "</Length>\n"
+         << "        </Chunk>\n"
+         << "      </ChunkList>\n"
+         << "    </Asset>\n";
+}
+       
+       
+       
+bool
+CPL::equals (CPL const & other, EqualityOptions opt, list<string>& notes) const
+{
+       if (_name != other._name) {
+               notes.push_back ("names differ");
+               return false;
+       }
+
+       if (_content_kind != other._content_kind) {
+               notes.push_back ("content kinds differ");
+               return false;
+       }
+
+       if (_fps != other._fps) {
+               notes.push_back ("frames per second differ");
+               return false;
+       }
+
+       if (_length != other._length) {
+               notes.push_back ("lengths differ");
+               return false;
+       }
+
        if (_reels.size() != other._reels.size()) {
                notes.push_back ("reel counts differ");
+               return false;
        }
        
        list<shared_ptr<const Reel> >::const_iterator a = _reels.begin ();
        list<shared_ptr<const Reel> >::const_iterator b = other._reels.begin ();
        
        while (a != _reels.end ()) {
-               list<string> n = (*a)->equals (*b, opt);
-               notes.merge (n);
+               if (!(*a)->equals (*b, opt, notes)) {
+                       return false;
+               }
                ++a;
                ++b;
        }
 
-       return notes;
+       return true;
 }
 
+shared_ptr<xmlpp::Document>
+CPL::make_kdm (
+       CertificateChain const & certificates,
+       string const & signer_key,
+       shared_ptr<const Certificate> recipient_cert,
+       boost::posix_time::ptime from,
+       boost::posix_time::ptime until
+       ) const
+{
+       shared_ptr<xmlpp::Document> doc (new xmlpp::Document);
+       xmlpp::Element* root = doc->create_root_node ("DCinemaSecurityMessage");
+       root->set_namespace_declaration ("http://www.smpte-ra.org/schemas/430-3/2006/ETM", "");
+       root->set_namespace_declaration ("http://www.w3.org/2000/09/xmldsig#", "ds");
+       root->set_namespace_declaration ("http://www.w3.org/2001/04/xmlenc#", "enc");
+
+       {
+               xmlpp::Element* authenticated_public = root->add_child("AuthenticatedPublic");
+               authenticated_public->set_attribute("Id", "ID_AuthenticatedPublic");
+               xmlAddID (0, doc->cobj(), (const xmlChar *) "ID_AuthenticatedPublic", authenticated_public->get_attribute("Id")->cobj());
+               
+               authenticated_public->add_child("MessageId")->add_child_text("urn:uuid:" + make_uuid());
+               authenticated_public->add_child("MessageType")->add_child_text("http://www.smpte-ra.org/430-1/2006/KDM#kdm-key-type");
+               authenticated_public->add_child("AnnotationText")->add_child_text(Metadata::instance()->product_name);
+               authenticated_public->add_child("IssueDate")->add_child_text(Metadata::instance()->issue_date);
+
+               {
+                       xmlpp::Element* signer = authenticated_public->add_child("Signer");
+                       signer->add_child("X509IssuerName", "ds")->add_child_text (
+                               Certificate::name_for_xml (recipient_cert->issuer())
+                               );
+                       signer->add_child("X509SerialNumber", "ds")->add_child_text (
+                               recipient_cert->serial()
+                               );
+               }
+
+               {
+                       xmlpp::Element* required_extensions = authenticated_public->add_child("RequiredExtensions");
+
+                       {
+                               xmlpp::Element* kdm_required_extensions = required_extensions->add_child("KDMRequiredExtensions");
+                               kdm_required_extensions->set_namespace_declaration ("http://www.smpte-ra.org/schemas/430-1/2006/KDM");
+                               {
+                                       xmlpp::Element* recipient = kdm_required_extensions->add_child("Recipient");
+                                       {
+                                               xmlpp::Element* serial_element = recipient->add_child("X509IssuerSerial");
+                                               serial_element->add_child("X509IssuerName", "ds")->add_child_text (
+                                                       Certificate::name_for_xml (recipient_cert->issuer())
+                                                       );
+                                               serial_element->add_child("X509SerialNumber", "ds")->add_child_text (
+                                                       recipient_cert->serial()
+                                                       );
+                                       }
+
+                                       recipient->add_child("X509SubjectName")->add_child_text (Certificate::name_for_xml (recipient_cert->subject()));
+                               }
+
+                               kdm_required_extensions->add_child("CompositionPlaylistId")->add_child_text("urn:uuid:" + _uuid);
+                               kdm_required_extensions->add_child("ContentTitleText")->add_child_text(_name);
+                               kdm_required_extensions->add_child("ContentAuthenticator")->add_child_text(certificates.leaf()->thumbprint());
+                               kdm_required_extensions->add_child("ContentKeysNotValidBefore")->add_child_text("XXX");
+                               kdm_required_extensions->add_child("ContentKeysNotValidAfter")->add_child_text("XXX");
+
+                               {
+                                       xmlpp::Element* authorized_device_info = kdm_required_extensions->add_child("AuthorizedDeviceInfo");
+                                       authorized_device_info->add_child("DeviceListIdentifier")->add_child_text("urn:uuid:" + make_uuid());
+                                       authorized_device_info->add_child("DeviceListDescription")->add_child_text(recipient_cert->subject());
+                                       {
+                                               xmlpp::Element* device_list = authorized_device_info->add_child("DeviceList");
+                                               device_list->add_child("CertificateThumbprint")->add_child_text(recipient_cert->thumbprint());
+                                       }
+                               }
+
+                               {
+                                       xmlpp::Element* key_id_list = kdm_required_extensions->add_child("KeyIdList");
+                                       list<shared_ptr<const Asset> > a = assets();
+                                       for (list<shared_ptr<const Asset> >::iterator i = a.begin(); i != a.end(); ++i) {
+                                               /* XXX: non-MXF assets? */
+                                               shared_ptr<const MXFAsset> mxf = boost::dynamic_pointer_cast<const MXFAsset> (*i);
+                                               if (mxf) {
+                                                       mxf->add_typed_key_id (key_id_list);
+                                               }
+                                       }
+                               }
+
+                               {
+                                       xmlpp::Element* forensic_mark_flag_list = kdm_required_extensions->add_child("ForensicMarkFlagList");
+                                       forensic_mark_flag_list->add_child("ForensicMarkFlag")->add_child_text ( 
+                                               "http://www.smpte-ra.org/430-1/2006/KDM#mrkflg-picture-disable"
+                                               );
+                                       forensic_mark_flag_list->add_child("ForensicMarkFlag")->add_child_text ( 
+                                               "http://www.smpte-ra.org/430-1/2006/KDM#mrkflg-audio-disable"
+                                               );
+                               }
+                       }
+               }
+                                        
+               authenticated_public->add_child("NonCriticalExtensions");
+       }
+
+       {
+               xmlpp::Element* authenticated_private = root->add_child("AuthenticatedPrivate");
+               authenticated_private->set_attribute ("Id", "ID_AuthenticatedPrivate");
+               xmlAddID (0, doc->cobj(), (const xmlChar *) "ID_AuthenticatedPrivate", authenticated_private->get_attribute("Id")->cobj());
+               {
+                       xmlpp::Element* encrypted_key = authenticated_private->add_child ("EncryptedKey", "enc");
+                       {
+                               xmlpp::Element* encryption_method = encrypted_key->add_child ("EncryptionMethod", "enc");
+                               encryption_method->set_attribute ("Algorithm", "http://www.w3.org/2001/04/xmlenc#rsa-oaep-mgf1p");
+                               encryption_method->add_child("DigestMethod", "ds")->set_attribute("Algorithm", "http://www.w3.org/2000/09/xmldsig#sha1");
+                       }
+
+                       xmlpp::Element* cipher_data = authenticated_private->add_child ("CipherData", "enc");
+                       cipher_data->add_child("CipherValue", "enc")->add_child_text("XXX");
+               }
+       }
+       
+       /* XXX: x2 one for each mxf? */
+
+       {
+               xmlpp::Element* signature = root->add_child("Signature", "ds");
+               
+               {
+                       xmlpp::Element* signed_info = signature->add_child("SignedInfo", "ds");
+                       signed_info->add_child("CanonicalizationMethod", "ds")->set_attribute(
+                               "Algorithm", "http://www.w3.org/TR/2001/REC-xml-c14n-20010315#WithComments"
+                               );
+                       signed_info->add_child("SignatureMethod", "ds")->set_attribute(
+                               "Algorithm", "http://www.w3.org/2001/04/xmldsig-more#rsa-sha256"
+                               );
+                       {
+                               xmlpp::Element* reference = signed_info->add_child("Reference", "ds");
+                               reference->set_attribute("URI", "#ID_AuthenticatedPublic");
+                               reference->add_child("DigestMethod", "ds")->set_attribute("Algorithm", "http://www.w3.org/2001/04/xmlenc#sha256");
+                               reference->add_child("DigestValue", "ds");
+                       }
+                       
+                       {                               
+                               xmlpp::Element* reference = signed_info->add_child("Reference", "ds");
+                               reference->set_attribute("URI", "#ID_AuthenticatedPrivate");
+                               reference->add_child("DigestMethod", "ds")->set_attribute("Algorithm", "http://www.w3.org/2001/04/xmlenc#sha256");
+                               reference->add_child("DigestValue", "ds");
+                       }
+               }
+               
+               add_signature_value (signature, certificates, signer_key, "ds");
+       }
+
+       return doc;
+}