Read PKL when reading DCP.

1 files changed, 114 insertions, 0 deletions

diff --git a/src/pkl.cc b/src/pkl.cc
new file mode 100644
index 00000000..236313aa
--- /dev/null
+++ b/src/pkl.cc
@@ -0,0 +1,114 @@
+/*
+    Copyright (C) 2018 Carl Hetherington <cth@carlh.net>
+
+    This file is part of libdcp.
+
+    libdcp is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    libdcp is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with libdcp.  If not, see <http://www.gnu.org/licenses/>.
+
+    In addition, as a special exception, the copyright holders give
+    permission to link the code of portions of this program with the
+    OpenSSL library under certain conditions as described in each
+    individual source file, and distribute linked combinations
+    including the two.
+
+    You must obey the GNU General Public License in all respects
+    for all of the code used other than OpenSSL.  If you modify
+    file(s) with this exception, you may extend this exception to your
+    version of the file(s), but you are not obligated to do so.  If you
+    do not wish to do so, delete this exception statement from your
+    version.  If you delete this exception statement from all source
+    files in the program, then also delete it here.
+*/
+
+#include "pkl.h"
+#include "exceptions.h"
+#include "util.h"
+#include "raw_convert.h"
+#include <libxml++/libxml++.h>
+#include <boost/foreach.hpp>
+
+using std::string;
+using boost::shared_ptr;
+using namespace dcp;
+
+static string const pkl_interop_ns = "http://www.digicine.com/PROTO-ASDCP-PKL-20040311#";
+static string const pkl_smpte_ns   = "http://www.smpte-ra.org/schemas/429-8/2007/PKL";
+
+PKL::PKL (boost::filesystem::path file)
+{
+	cxml::Document pkl ("PackingList");
+	pkl.read_file (file);
+
+	if (pkl.namespace_uri() == pkl_interop_ns) {
+		_standard = INTEROP;
+	} else if (pkl.namespace_uri() == pkl_smpte_ns) {
+		_standard = SMPTE;
+	} else {
+		boost::throw_exception (XMLError ("Unrecognised packing list namesapce " + pkl.namespace_uri()));
+	}
+
+	_id = remove_urn_uuid (pkl.string_child ("Id"));
+	_annotation_text = pkl.optional_string_child ("AnnotationText");
+	_issue_date = pkl.string_child ("IssueDate");
+	_issuer = pkl.string_child ("Issuer");
+	_creator = pkl.string_child ("Creator");
+
+	BOOST_FOREACH (cxml::ConstNodePtr i, pkl.node_child("AssetList")->node_children("Asset")) {
+		_asset_list.push_back (shared_ptr<Asset> (new Asset (i)));
+	}
+}
+
+void
+PKL::add_asset (std::string id, boost::optional<std::string> annotation_text, std::string hash, int64_t size, std::string type)
+{
+	_asset_list.push_back (shared_ptr<Asset> (new Asset (id, annotation_text, hash, size, type)));
+}
+
+void
+PKL::write (boost::filesystem::path file, shared_ptr<const CertificateChain> signer) const
+{
+	xmlpp::Document doc;
+	xmlpp::Element* pkl;
+	if (_standard == INTEROP) {
+		pkl = doc.create_root_node("PackingList", pkl_interop_ns);
+	} else {
+		pkl = doc.create_root_node("PackingList", pkl_smpte_ns);
+	}
+
+	pkl->add_child("Id")->add_child_text ("urn:uuid:" + _id);
+	if (_annotation_text) {
+		pkl->add_child("AnnotationText")->add_child_text (*_annotation_text);
+	}
+	pkl->add_child("IssueDate")->add_child_text (_issue_date);
+	pkl->add_child("Issuer")->add_child_text (_issuer);
+	pkl->add_child("Creator")->add_child_text (_creator);
+
+	xmlpp::Element* asset_list = pkl->add_child("AssetList");
+	BOOST_FOREACH (shared_ptr<Asset> i, _asset_list) {
+		xmlpp::Element* asset = asset_list->add_child("Asset");
+		asset->add_child("Id")->add_child_text ("urn:uuid:" + i->id());
+		if (i->annotation_text) {
+			asset->add_child("AnnotationText")->add_child_text (*i->annotation_text);
+		}
+		asset->add_child("Hash")->add_child_text (i->hash);
+		asset->add_child("Size")->add_child_text (raw_convert<string> (i->size));
+		asset->add_child("Type")->add_child_text (i->type);
+	}
+
+	if (signer) {
+		signer->sign (pkl, _standard);
+	}
+
+	doc.write_to_file (file.string(), "UTF-8");
+}