Simpole DCP recovery utility (dcprecover) added.

[libdcp.git] / tools / dcprecover.cc

diff --git a/tools/dcprecover.cc b/tools/dcprecover.cc
new file mode 100644 (file)
index 0000000..b747a5d
--- /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;
+}