Merge branch '1.0' of ssh://main.carlh.net/home/carl/git/libdcp into 1.0
[libdcp.git] / tools / dcpdiff.cc
1 /*
2     Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include "dcp.h"
21 #include "exceptions.h"
22 #include "common.h"
23 #include "mxf.h"
24 #include <getopt.h>
25 #include <boost/optional.hpp>
26 #include <boost/shared_ptr.hpp>
27 #include <boost/filesystem.hpp>
28 #include <iostream>
29 #include <list>
30
31 using std::list;
32 using std::cerr;
33 using std::cout;
34 using std::string;
35 using boost::shared_ptr;
36 using boost::optional;
37 using boost::dynamic_pointer_cast;
38 using namespace dcp;
39
40 static bool verbose = false;
41
42 static void
43 help (string n)
44 {
45         cerr << "Syntax: " << n << " [OPTION] <DCP> <DCP>\n"
46              << "  -V, --version                show libdcp version\n"
47              << "  -h, --help                   show this help\n"
48              << "  -v, --verbose                be verbose\n"
49              << "  -n, --mxf-filenames          allow differing MXF filenames\n"
50              << "      --cpl-annotation-texts   allow differing CPL annotation texts\n"
51              << "  -m, --mean-pixel             maximum allowed mean pixel error (default 5)\n"
52              << "  -s, --std-dev-pixel          maximum allowed standard deviation of pixel error (default 5)\n"
53              << "      --key                    hexadecimal key to use to decrypt MXFs\n"
54              << "  -k, --keep-going             carry on in the event of errors, if possible\n"
55              << "      --ignore-missing-assets  ignore missing asset files\n"
56              << "\n"
57              << "The <DCP>s are the DCP directories to compare.\n"
58              << "Comparison is of metadata and content, ignoring timestamps\n"
59              << "and differing UUIDs.\n";
60 }
61
62 void
63 note (NoteType t, string n)
64 {
65         if (t == DCP_ERROR || verbose) {
66                 cout << " " << n << "\n";
67         }
68 }
69
70 DCP *
71 load_dcp (boost::filesystem::path path, bool keep_going, bool ignore_missing_assets, optional<string> key)
72 {
73         DCP* dcp = 0;
74         try {
75                 dcp = new DCP (path);
76                 DCP::ReadErrors errors;
77                 dcp->read (keep_going, &errors);
78                 filter_errors (errors, ignore_missing_assets);
79                 for (DCP::ReadErrors::const_iterator i = errors.begin(); i != errors.end(); ++i) {
80                         cerr << (*i)->what() << "\n";
81                 }
82
83                 if (key) {
84                         list<shared_ptr<Asset> > assets = dcp->assets ();
85                         for (list<shared_ptr<Asset> >::const_iterator i = assets.begin(); i != assets.end(); ++i) {
86                                 shared_ptr<MXF> mxf = dynamic_pointer_cast<MXF> (*i);
87                                 if (mxf) {
88                                         mxf->set_key (Key (key.get ()));
89                                 }
90                         }
91                 }
92
93         } catch (FileError& e) {
94                 cerr << "Could not read DCP " << path.string() << "; " << e.what() << " " << e.filename() << "\n";
95                 exit (EXIT_FAILURE);
96         }
97
98         return dcp;
99 }
100
101 int
102 main (int argc, char* argv[])
103 {
104         EqualityOptions options;
105         options.max_mean_pixel_error = 5;
106         options.max_std_dev_pixel_error = 5;
107         options.reel_hashes_can_differ = true;
108         bool keep_going = false;
109         bool ignore_missing_assets = false;
110         optional<string> key;
111         
112         int option_index = 0;
113         while (1) {
114                 static struct option long_options[] = {
115                         { "version", no_argument, 0, 'V'},
116                         { "help", no_argument, 0, 'h'},
117                         { "verbose", no_argument, 0, 'v'},
118                         { "mxf-filenames", no_argument, 0, 'n'},
119                         { "mean-pixel", required_argument, 0, 'm'},
120                         { "std-dev-pixel", required_argument, 0, 's'},
121                         { "keep-going", no_argument, 0, 'k'},
122                         /* From here we're using random capital letters for the short option */
123                         { "ignore-missing-assets", no_argument, 0, 'A'},
124                         { "cpl-annotation-texts", no_argument, 0, 'C'},
125                         { "key", required_argument, 0, 'D'},
126                         { 0, 0, 0, 0 }
127                 };
128
129                 int c = getopt_long (argc, argv, "Vhvnm:s:kACD:", long_options, &option_index);
130
131                 if (c == -1) {
132                         break;
133                 }
134
135                 switch (c) {
136                 case 'V':
137                         cout << "dcpdiff version " << LIBDCP_VERSION << "\n";
138                         exit (EXIT_SUCCESS);
139                 case 'h':
140                         help (argv[0]);
141                         exit (EXIT_SUCCESS);
142                 case 'v':
143                         verbose = true;
144                         break;
145                 case 'n':
146                         options.mxf_filenames_can_differ = true;
147                         break;
148                 case 'm':
149                         options.max_mean_pixel_error = atof (optarg);
150                         break;
151                 case 's':
152                         options.max_std_dev_pixel_error = atof (optarg);
153                         break;
154                 case 'k':
155                         keep_going = true;
156                         break;
157                 case 'A':
158                         ignore_missing_assets = true;
159                         break;
160                 case 'B':
161                 case 'C':
162                         options.cpl_annotation_texts_can_differ = true;
163                         break;
164                 case 'D':
165                         key = string (optarg);
166                         break;
167                 }
168         }
169
170         if (argc <= optind || argc > (optind + 2)) {
171                 help (argv[0]);
172                 exit (EXIT_FAILURE);
173         }
174
175         if (!boost::filesystem::exists (argv[optind])) {
176                 cerr << argv[0] << ": DCP " << argv[optind] << " not found.\n";
177                 exit (EXIT_FAILURE);
178         }
179
180         if (!boost::filesystem::exists (argv[optind + 1])) {
181                 cerr << argv[0] << ": DCP " << argv[optind + 1] << " not found.\n";
182                 exit (EXIT_FAILURE);
183         }
184
185         DCP* a = load_dcp (argv[optind], keep_going, ignore_missing_assets, key);
186         DCP* b = load_dcp (argv[optind + 1], keep_going, ignore_missing_assets, key);
187
188         /* I think this is just below the LSB at 16-bits (ie the 8th most significant bit at 24-bit) */
189         options.max_audio_sample_error = 255;
190
191         bool const equals = a->equals (*b, options, boost::bind (note, _1, _2));
192
193         exit (equals ? EXIT_SUCCESS : EXIT_FAILURE);
194 }