Fix signing of KDMs.

[libdcp.git] / src / reel.cc

diff --git a/src/reel.cc b/src/reel.cc
index 86533ea2e8e26a204b45bd42bfd9d1dacf943515..51aa560c706297a0658c5744808c1cbc9698d2c3 100644 (file)
--- a/src/reel.cc
+++ b/src/reel.cc
@@ -17,36 +17,38 @@
 
 */
 
+#include <libxml++/nodes/element.h>
 #include "reel.h"
 #include "util.h"
 #include "picture_asset.h"
 #include "sound_asset.h"
 #include "subtitle_asset.h"
+#include "kdm.h"
 
-using namespace std;
+using std::string;
+using std::list;
+using std::cout;
+using boost::shared_ptr;
 using namespace libdcp;
 
 void
-Reel::write_to_cpl (ostream& s) const
+Reel::write_to_cpl (xmlpp::Element* node, bool interop) const
 {
-       s << "    <Reel>\n"
-         << "      <Id>urn:uuid:" << make_uuid() << "</Id>\n"
-         << "      <AssetList>\n";
+       xmlpp::Element* reel = node->add_child ("Reel");
+       reel->add_child("Id")->add_child_text ("urn:uuid:" + make_uuid());
+       xmlpp::Element* asset_list = reel->add_child ("AssetList");
        
        if (_main_picture) {
-               _main_picture->write_to_cpl (s);
+               _main_picture->write_to_cpl (asset_list, interop);
        }
 
        if (_main_sound) {
-               _main_sound->write_to_cpl (s);
+               _main_sound->write_to_cpl (asset_list, interop);
        }
 
        if (_main_subtitle) {
-               _main_subtitle->write_to_cpl (s);
+               _main_subtitle->write_to_cpl (asset_list, interop);
        }
-
-       s << "      </AssetList>\n"
-         << "    </Reel>\n";
 }
        
 bool
@@ -82,3 +84,34 @@ Reel::equals (boost::shared_ptr<const Reel> other, EqualityOptions opt, boost::f
        return true;
 }
 
+bool
+Reel::encrypted () const
+{
+       return ((_main_picture && _main_picture->encrypted ()) || (_main_sound && _main_sound->encrypted ()));
+}
+
+void
+Reel::add_kdm (KDM const & kdm)
+{
+       list<KDMKey> keys = kdm.keys ();
+       
+       for (list<KDMKey>::iterator i = keys.begin(); i != keys.end(); ++i) {
+               if (i->key_id() == _main_picture->key_id()) {
+                       _main_picture->set_key (i->key ());
+               }
+               if (i->key_id() == _main_sound->key_id()) {
+                       _main_sound->set_key (i->key ());
+               }
+       }
+}
+
+void
+Reel::set_mxf_keys (Key key)
+{
+       _main_picture->set_key (key);
+       if (_main_sound) {
+               _main_sound->set_key (key);
+       }
+
+       /* XXX: subtitle asset? */
+}