X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fdcp.cc;h=c7634b5d3385d1a9cafd5b847dffd41dd4540d66;hb=27e74ac5cf717233914fd017eb9ca67ee7c4af96;hp=27ae9636da7ee3a4cff317acc929f1a29518fa51;hpb=4de6f6e4f4df4cc9b28b98188d36082b26e81345;p=libdcp.git

diff --git a/src/dcp.cc b/src/dcp.cc
index 27ae9636..c7634b5d 100644
--- a/src/dcp.cc
+++ b/src/dcp.cc
@@ -25,246 +25,272 @@
 #include <fstream>
 #include <iomanip>
 #include <cassert>
+#include <iostream>
 #include <boost/filesystem.hpp>
+#include <boost/algorithm/string.hpp>
+#include <boost/lexical_cast.hpp>
+#include <libxml++/libxml++.h>
 #include "dcp.h"
 #include "asset.h"
 #include "sound_asset.h"
 #include "picture_asset.h"
+#include "subtitle_asset.h"
 #include "util.h"
 #include "metadata.h"
-
-using namespace std;
-using namespace boost;
+#include "exceptions.h"
+#include "parse/pkl.h"
+#include "parse/asset_map.h"
+#include "reel.h"
+#include "cpl.h"
+
+using std::string;
+using std::list;
+using std::stringstream;
+using std::ofstream;
+using std::ostream;
+using boost::shared_ptr;
+using boost::lexical_cast;
 using namespace libdcp;
 
-DCP::DCP (string directory, string name, ContentType content_type, int fps, int length)
+DCP::DCP (string directory)
 	: _directory (directory)
-	, _name (name)
-	, _content_type (content_type)
-	, _fps (fps)
-	, _length (length)
 {
-	
+	boost::filesystem::create_directories (directory);
 }
 
 void
-DCP::add_sound_asset (vector<string> const & files)
+DCP::write_xml (XMLMetadata const & metadata) const
 {
-	filesystem::path p;
-	p /= _directory;
-	p /= "audio.mxf";
-	_assets.push_back (shared_ptr<SoundAsset> (new SoundAsset (files, p.string(), &Progress, _fps, _length)));
+	for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+		(*i)->write_xml (metadata);
+	}
+
+	string pkl_uuid = make_uuid ();
+	string pkl_path = write_pkl (pkl_uuid, metadata);
+	
+	write_volindex ();
+	write_assetmap (pkl_uuid, boost::filesystem::file_size (pkl_path), metadata);
 }
 
-void
-DCP::add_sound_asset (sigc::slot<string, Channel> get_path, int channels)
+std::string
+DCP::write_pkl (string pkl_uuid, XMLMetadata const & metadata) const
 {
-	filesystem::path p;
+	assert (!_cpls.empty ());
+	
+	boost::filesystem::path p;
 	p /= _directory;
-	p /= "audio.mxf";
-	_assets.push_back (shared_ptr<SoundAsset> (new SoundAsset (get_path, p.string(), &Progress, _fps, _length, channels)));
+	stringstream s;
+	s << pkl_uuid << "_pkl.xml";
+	p /= s.str();
+
+	xmlpp::Document doc;
+	xmlpp::Element* root = doc.create_root_node ("PackingList", "http://www.smpte-ra.org/schemas/429-8/2007/PKL");
+
+	root->add_child("Id")->add_child_text ("urn:uuid:" + pkl_uuid);
+	/* XXX: this is a bit of a hack */
+	root->add_child("AnnotationText")->add_child_text (_cpls.front()->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);
+
+	xmlpp::Node* asset_list = root->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);
+	}
+
+	for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+		(*i)->write_to_pkl (asset_list);
+	}
+
+	doc.write_to_file_formatted (p.string (), "UTF-8");
+	return p.string ();
 }
 
 void
-DCP::add_picture_asset (vector<string> const & files, int width, int height)
+DCP::write_volindex () const
 {
-	filesystem::path p;
+	boost::filesystem::path p;
 	p /= _directory;
-	p /= "video.mxf";
-	_assets.push_back (shared_ptr<PictureAsset> (new PictureAsset (files, p.string(), &Progress, _fps, _length, width, height)));
+	p /= "VOLINDEX.xml";
+
+	xmlpp::Document doc;
+	xmlpp::Element* root = doc.create_root_node ("VolumeIndex", "http://www.smpte-ra.org/schemas/429-9/2007/AM");
+	root->add_child("Index")->add_child_text ("1");
+	doc.write_to_file_formatted (p.string (), "UTF-8");
 }
 
 void
-DCP::add_picture_asset (sigc::slot<string, int> get_path, int width, int height)
+DCP::write_assetmap (string pkl_uuid, int pkl_length, XMLMetadata const & metadata) const
 {
-	filesystem::path p;
+	boost::filesystem::path p;
 	p /= _directory;
-	p /= "video.mxf";
-	_assets.push_back (shared_ptr<PictureAsset> (new PictureAsset (get_path, p.string(), &Progress, _fps, _length, width, height)));
+	p /= "ASSETMAP.xml";
+
+	xmlpp::Document doc;
+	xmlpp::Element* root = doc.create_root_node ("AssetMap", "http://www.smpte-ra.org/schemas/429-9/2007/AM");
+
+	root->add_child("Id")->add_child_text ("urn:uuid:" + make_uuid());
+	root->add_child("Creator")->add_child_text (metadata.creator);
+	root->add_child("VolumeCount")->add_child_text ("1");
+	root->add_child("IssueDate")->add_child_text (metadata.issue_date);
+	root->add_child("Issuer")->add_child_text (metadata.issuer);
+	xmlpp::Node* asset_list = root->add_child ("AssetList");
+
+	xmlpp::Node* asset = asset_list->add_child ("Asset");
+	asset->add_child("Id")->add_child_text ("urn:uuid:" + pkl_uuid);
+	asset->add_child("PackingList")->add_child_text ("true");
+	xmlpp::Node* chunk_list = asset->add_child ("ChunkList");
+	xmlpp::Node* chunk = chunk_list->add_child ("Chunk");
+	chunk->add_child("Path")->add_child_text (pkl_uuid + "_pkl.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> (pkl_length));
+	
+	for (list<shared_ptr<const CPL> >::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
+		(*i)->write_to_assetmap (asset_list);
+	}
+
+	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 (asset_list);
+	}
+
+	doc.write_to_file_formatted (p.string (), "UTF-8");
 }
 
+
 void
-DCP::write_xml () const
+DCP::read (bool require_mxfs)
 {
-	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);
+	Files files;
+
+	shared_ptr<parse::AssetMap> asset_map;
+	try {
+		boost::filesystem::path p = _directory;
+		p /= "ASSETMAP";
+		if (boost::filesystem::exists (p)) {
+			asset_map.reset (new libdcp::parse::AssetMap (p.string ()));
+		} else {
+			p = _directory;
+			p /= "ASSETMAP.xml";
+			if (boost::filesystem::exists (p)) {
+				asset_map.reset (new libdcp::parse::AssetMap (p.string ()));
+			} else {
+				boost::throw_exception (DCPReadError ("could not find AssetMap file"));
+			}
+		}
+		
+	} catch (FileError& e) {
+		boost::throw_exception (FileError ("could not load AssetMap file", files.asset_map));
+	}
 
-	string pkl_uuid = make_uuid ();
-	string pkl_path = write_pkl (pkl_uuid, cpl_uuid, cpl_digest, cpl_length);
-	
-	write_volindex ();
-	write_assetmap (cpl_uuid, cpl_length, pkl_uuid, filesystem::file_size (pkl_path));
-}
+	for (list<shared_ptr<libdcp::parse::AssetMapAsset> >::const_iterator i = asset_map->assets.begin(); i != asset_map->assets.end(); ++i) {
+		if ((*i)->chunks.size() != 1) {
+			boost::throw_exception (XMLError ("unsupported asset chunk count"));
+		}
 
-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());
+		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;
+		}
+
+		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 {
+				boost::throw_exception (DCPReadError ("duplicate PKLs found"));
+			}
+		}
+	}
 	
-	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_type_string (_content_type) << "</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";
-
-	cpl << "    <Reel>\n"
-	    << "      <Id>urn:uuid:" << make_uuid() << "</Id>\n"
-	    << "      <AssetList>\n";
-
-	for (list<shared_ptr<Asset> >::const_iterator i = _assets.begin(); i != _assets.end(); ++i) {
-		(*i)->write_to_cpl (cpl);
+	if (files.cpls.empty ()) {
+		boost::throw_exception (FileError ("no CPL files found", ""));
 	}
 
-	cpl << "      </AssetList>\n"
-	    << "    </Reel>\n"
-	    << "  </ReelList>\n"
-	    << "</CompositionPlaylist>\n";
+	if (files.pkl.empty ()) {
+		boost::throw_exception (FileError ("no PKL file found", ""));
+	}
 
-	return p.string ();
+	shared_ptr<parse::PKL> pkl;
+	try {
+		pkl.reset (new parse::PKL (files.pkl));
+	} catch (FileError& e) {
+		boost::throw_exception (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)));
+	}
 }
 
-std::string
-DCP::write_pkl (string pkl_uuid, string cpl_uuid, string cpl_digest, int cpl_length) const
+bool
+DCP::equals (DCP const & other, EqualityOptions opt, boost::function<void (NoteType, string)> note) const
 {
-	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<Asset> >::const_iterator i = _assets.begin(); i != _assets.end(); ++i) {
-		(*i)->write_to_pkl (pkl);
+	if (_cpls.size() != other._cpls.size()) {
+		note (ERROR, "CPL counts differ");
+		return false;
 	}
 
-	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";
+	list<shared_ptr<const CPL> >::const_iterator a = _cpls.begin ();
+	list<shared_ptr<const CPL> >::const_iterator b = other._cpls.begin ();
 
-	pkl << "  </AssetList>\n"
-	    << "</PackingList>\n";
+	while (a != _cpls.end ()) {
+		if (!(*a)->equals (*b->get(), opt, note)) {
+			return false;
+		}
+		++a;
+		++b;
+	}
 
-	return p.string ();
+	return true;
 }
 
+
 void
-DCP::write_volindex () const
+DCP::add_cpl (shared_ptr<CPL> cpl)
 {
-	filesystem::path p;
-	p /= _directory;
-	p /= "VOLINDEX.xml";
-	ofstream vi (p.string().c_str());
-
-	vi << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
-	   << "<VolumeIndex xmlns=\"http://www.smpte-ra.org/schemas/429-9/2007/AM\">\n"
-	   << "  <Index>1</Index>\n"
-	   << "</VolumeIndex>\n";
+	_cpls.push_back (cpl);
 }
 
-void
-DCP::write_assetmap (string cpl_uuid, int cpl_length, string pkl_uuid, int pkl_length) const
+class AssetComparator
 {
-	filesystem::path p;
-	p /= _directory;
-	p /= "ASSETMAP.xml";
-	ofstream am (p.string().c_str());
-
-	am << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
-	   << "<AssetMap xmlns=\"http://www.smpte-ra.org/schemas/429-9/2007/AM\">\n"
-	   << "  <Id>urn:uuid:" << make_uuid() << "</Id>\n"
-	   << "  <Creator>" << Metadata::instance()->creator << "</Creator>\n"
-	   << "  <VolumeCount>1</VolumeCount>\n"
-	   << "  <IssueDate>" << Metadata::instance()->issue_date << "</IssueDate>\n"
-	   << "  <Issuer>" << Metadata::instance()->issuer << "</Issuer>\n"
-	   << "  <AssetList>\n";
-
-	am << "    <Asset>\n"
-	   << "      <Id>urn:uuid:" << pkl_uuid << "</Id>\n"
-	   << "      <PackingList>true</PackingList>\n"
-	   << "      <ChunkList>\n"
-	   << "        <Chunk>\n"
-	   << "          <Path>" << pkl_uuid << "_pkl.xml</Path>\n"
-	   << "          <VolumeIndex>1</VolumeIndex>\n"
-	   << "          <Offset>0</Offset>\n"
-	   << "          <Length>" << pkl_length << "</Length>\n"
-	   << "        </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<Asset> >::const_iterator i = _assets.begin(); i != _assets.end(); ++i) {
-		(*i)->write_to_assetmap (am);
+public:
+	bool operator() (shared_ptr<const Asset> a, shared_ptr<const Asset> b) {
+		return a->uuid() < b->uuid();
 	}
+};
 
-	am << "  </AssetList>\n"
-	   << "</AssetMap>\n";
-}
-
-string
-DCP::content_type_string (ContentType type)
+list<shared_ptr<const Asset> >
+DCP::assets () const
 {
-	switch (type) {
-	case FEATURE:
-		return "feature";
-	case SHORT:
-		return "short";
-	case TRAILER:
-		return "trailer";
-	case TEST:
-		return "test";
-	case TRANSITIONAL:
-		return "transitional";
-	case RATING:
-		return "rating";
-	case TEASER:
-		return "teaser";
-	case POLICY:
-		return "policy";
-	case PUBLIC_SERVICE_ANNOUNCEMENT:
-		return "psa";
-	case ADVERTISEMENT:
-		return "advertisement";
+	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);
 	}
 
-	assert (false);
+	a.sort ();
+	a.unique ();
+	return a;
 }
-		
+