Simpole DCP recovery utility (dcprecover) added.

2 files changed, 164 insertions, 2 deletions

diff --git a/tools/dcprecover.cc b/tools/dcprecover.cc
new file mode 100644
index 00000000..b747a5da
--- /dev/null
+++ b/tools/dcprecover.cc
@@ -0,0 +1,162 @@
+/*
+    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 "dcp.h"
+#include "cpl.h"
+#include "exceptions.h"
+#include "asset_factory.h"
+#include "reel_asset.h"
+#include <getopt.h>
+#include <libxml++/libxml++.h>
+#include <boost/filesystem.hpp>
+#include <boost/foreach.hpp>
+#include <iostream>
+
+using std::cerr;
+using std::cout;
+using std::string;
+using std::list;
+using boost::shared_ptr;
+using boost::optional;
+
+static void
+help (string n)
+{
+	cerr << "Syntax: " << n << " [OPTION] <DCP>]\n"
+	     << "  -h, --help         show this help\n"
+	     << "  -o, --output       output DCP directory\n";
+}
+
+void progress (float f)
+{
+	cout << (f * 100) << "%               \r";
+}
+
+int
+main (int argc, char* argv[])
+{
+	int option_index = 0;
+	optional<boost::filesystem::path> output;
+	while (true) {
+		struct option long_options[] = {
+			{ "help", no_argument, 0, 'h' },
+			{ "output", required_argument, 0, 'o' },
+			{ 0, 0, 0, 0 }
+		};
+
+		int c = getopt_long (argc, argv, "ho:", long_options, &option_index);
+
+		if (c == -1) {
+			break;
+		}
+
+		switch (c) {
+		case 'h':
+			help (argv[0]);
+			exit (EXIT_SUCCESS);
+		case 'o':
+			output = optarg;
+			break;
+		}
+	}
+
+	if (optind >= argc) {
+		help (argv[0]);
+		exit (EXIT_FAILURE);
+	}
+
+	boost::filesystem::path dcp_dir = argv[optind];
+
+	/* Try to read it and report errors */
+
+	dcp::DCP dcp (dcp_dir);
+	dcp::DCP::ReadErrors errors;
+	try {
+		dcp.read (true, &errors, true);
+	} catch (dcp::DCPReadError& e) {
+		cout << "Error:" <<  e.what() << "\n";
+	}
+
+	BOOST_FOREACH (shared_ptr<dcp::DCPReadError> i, errors) {
+		cout << "Error: " << i->what() << "\n";
+	}
+
+	/* Look for a CPL */
+
+	shared_ptr<dcp::CPL> cpl;
+	for (boost::filesystem::directory_iterator i(dcp_dir); i != boost::filesystem::directory_iterator(); ++i) {
+		if (i->path().extension() == ".xml") {
+			try {
+				cpl.reset(new dcp::CPL(i->path()));
+			} catch (dcp::DCPReadError& e) {
+				cout << "Error: " << e.what() << "\n";
+			} catch (xmlpp::parse_error& e) {
+				cout << "Error: " << e.what() << "\n";
+			}
+		}
+	}
+
+	if (cpl) {
+		cout << "Got a CPL!\n";
+
+		if (!output) {
+			cerr << "No output directory specified.\n";
+			exit(1);
+		}
+
+		/* Read all MXF assets */
+		list<shared_ptr<dcp::Asset> > assets;
+		for (boost::filesystem::directory_iterator i(dcp_dir); i != boost::filesystem::directory_iterator(); ++i) {
+			if (i->path().extension() == ".mxf") {
+				try {
+					shared_ptr<dcp::Asset> asset = dcp::asset_factory(i->path(), true);
+					asset->set_file (*output / i->path().filename());
+					cout << "Hashing " << i->path().filename() << "\n";
+					asset->hash (&progress);
+					cout << "100%                     \n";
+					assets.push_back (asset);
+				} catch (dcp::DCPReadError& e) {
+					cout << "Error: " << e.what() << "\n";
+				}
+			}
+		}
+
+		dcp::DCP fixed (*output);
+		fixed.add (cpl);
+		fixed.resolve_refs (assets);
+		fixed.write_xml (dcp::INTEROP);
+		cout << "Fixed XML files written to " << output->string() << "\n";
+	}
+
+	return 0;
+}
diff --git a/tools/wscript b/tools/wscript
index 96c31fdc..99f5207e 100644
--- a/tools/wscript
+++ b/tools/wscript
@@ -1,5 +1,5 @@
 #
-#    Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
+#    Copyright (C) 2012-2018 Carl Hetherington <cth@carlh.net>
 #
 #    This file is part of libdcp.
 #
@@ -44,7 +44,7 @@ def build(bld):
     obj.source = 'dcpinfo.cc common.cc'
     obj.target = 'dcpinfo'
 
-    for f in ['dumpsub', 'decryptmxf', 'kdm', 'thumb']:
+    for f in ['dumpsub', 'decryptmxf', 'kdm', 'thumb', 'recover']:
         obj = bld(features='cxx cxxprogram')
         obj.use = ['libdcp%s' % bld.env.API_VERSION]
         obj.uselib = 'OPENJPEG CXML OPENMP ASDCPLIB_CTH BOOST_FILESYSTEM LIBXML++ XMLSEC1 OPENSSL'