wip: add AssetMap and fix everything up.dcp-editor

1 files changed, 204 insertions, 0 deletions

diff --git a/src/asset_map.cc b/src/asset_map.cc
new file mode 100644
index 00000000..763d348e
--- /dev/null
+++ b/src/asset_map.cc
@@ -0,0 +1,204 @@
+/*
+    Copyright (C) 2022 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 "asset_map.h"
+#include "dcp_assert.h"
+#include "raw_convert.h"
+#include "util.h"
+#include "warnings.h"
+LIBDCP_DISABLE_WARNINGS
+#include <libxml++/libxml++.h>
+LIBDCP_ENABLE_WARNINGS
+#include <boost/algorithm/string.hpp>
+
+
+using std::string;
+using std::vector;
+using boost::algorithm::starts_with;
+using namespace dcp;
+
+
+static string const assetmap_interop_ns = "http://www.digicine.com/PROTO-ASDCP-AM-20040311#";
+static string const assetmap_smpte_ns   = "http://www.smpte-ra.org/schemas/429-9/2007/AM";
+
+
+AssetMap::AssetMap(boost::filesystem::path file)
+	: _file(file)
+{
+	cxml::Document doc("AssetMap");
+	doc.read_file(file);
+
+	if (doc.namespace_uri() == assetmap_interop_ns) {
+		_standard = Standard::INTEROP;
+	} else if (doc.namespace_uri() == assetmap_smpte_ns) {
+		_standard = Standard::SMPTE;
+	} else {
+		boost::throw_exception(XMLError("Unrecognised AssetMap namespace " + doc.namespace_uri()));
+	}
+
+	_annotation_text = doc.optional_string_child("AnnotationText");
+	_issue_date = doc.string_child("IssueDate");
+	_issuer = doc.string_child("Issuer");
+	_creator = doc.string_child("Creator");
+
+	for (auto asset: doc.node_child("AssetList")->node_children("Asset")) {
+		_assets.push_back(Asset(asset, _file->parent_path(), _standard));
+	}
+}
+
+
+vector<boost::filesystem::path>
+AssetMap::pkl_paths() const
+{
+	auto paths = std::vector<boost::filesystem::path>();
+	for (auto asset: _assets) {
+		if (asset.pkl()) {
+			paths.push_back(asset.path());
+		}
+	}
+	return paths;
+}
+
+
+std::map<std::string, boost::filesystem::path>
+AssetMap::paths() const
+{
+	auto paths = std::map<string, boost::filesystem::path>();
+	for (auto asset: _assets) {
+		paths[asset.id()] = asset.path();
+	}
+	return paths;
+}
+
+
+void
+AssetMap::add_asset(string id, boost::filesystem::path path, bool pkl)
+{
+	_assets.push_back(Asset(id, path, pkl));
+}
+
+
+void
+AssetMap::write_xml(boost::filesystem::path file)
+{
+	xmlpp::Document doc;
+	xmlpp::Element* root;
+
+	switch (_standard) {
+	case Standard::INTEROP:
+		root = doc.create_root_node("AssetMap", assetmap_interop_ns);
+		break;
+	case Standard::SMPTE:
+		root = doc.create_root_node("AssetMap", assetmap_smpte_ns);
+		break;
+	default:
+		DCP_ASSERT (false);
+	}
+
+	root->add_child("Id")->add_child_text("urn:uuid:" + _id);
+	if (_annotation_text) {
+		root->add_child("AnnotationText")->add_child_text(*_annotation_text);
+	}
+
+	switch (_standard) {
+	case Standard::INTEROP:
+		root->add_child("VolumeCount")->add_child_text("1");
+		root->add_child("IssueDate")->add_child_text(_issue_date);
+		root->add_child("Issuer")->add_child_text(_issuer);
+		root->add_child("Creator")->add_child_text(_creator);
+		break;
+	case Standard::SMPTE:
+		root->add_child("Creator")->add_child_text(_creator);
+		root->add_child("VolumeCount")->add_child_text ("1");
+		root->add_child("IssueDate")->add_child_text(_issue_date);
+		root->add_child("Issuer")->add_child_text(_issuer);
+		break;
+	default:
+		DCP_ASSERT (false);
+	}
+
+	auto const dcp_root_directory = file.parent_path();
+
+	auto asset_list = root->add_child("AssetList");
+
+	for (auto asset: _assets) {
+		auto node = asset_list->add_child("Asset");
+		node->add_child("Id")->add_child_text("urn:uuid:" + asset.id());
+		if (asset.pkl()) {
+			node->add_child("PackingList")->add_child_text("true");
+		}
+		auto chunk_list = node->add_child("ChunkList");
+		auto chunk = chunk_list->add_child("Chunk");
+
+		auto relative_path = relative_to_root(dcp_root_directory, asset.path());
+		DCP_ASSERT(relative_path);
+
+		chunk->add_child("Path")->add_child_text(relative_path->generic_string());
+		chunk->add_child("VolumeIndex")->add_child_text("1");
+		chunk->add_child("Offset")->add_child_text("0");
+		chunk->add_child("Length")->add_child_text(raw_convert<string>(boost::filesystem::file_size(asset.path())));
+	}
+
+	doc.write_to_file_formatted(file.string(), "UTF-8");
+	_file = file;
+}
+
+
+AssetMap::Asset::Asset(cxml::ConstNodePtr node, boost::filesystem::path root, dcp::Standard standard)
+	: Object(remove_urn_uuid(node->string_child("Id")))
+{
+	if (node->node_child("ChunkList")->node_children("Chunk").size() != 1) {
+		boost::throw_exception(XMLError("unsupported asset chunk count"));
+	}
+
+	auto path_from_xml = node->node_child("ChunkList")->node_child("Chunk")->string_child("Path");
+	if (starts_with(path_from_xml, "file://")) {
+		path_from_xml = path_from_xml.substr(7);
+	}
+
+	_path = root / path_from_xml;
+
+	switch (standard) {
+		case Standard::INTEROP:
+			_pkl = static_cast<bool>(node->optional_node_child("PackingList"));
+			break;
+		case Standard::SMPTE:
+			{
+				auto pkl_bool = node->optional_string_child("PackingList");
+				_pkl = pkl_bool && *pkl_bool == "true";
+				break;
+			}
+	}
+}
+