Use OpenMP when comparing picture assets.
[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              << "      --cpl-annotation-texts   allow differing CPL annotation texts\n"
50              << "      --reel-annotation-texts  allow differing reel annotation texts\n"
51              << "  -a, --annotation-texts       allow different CPL and reel annotation texts\n"
52              << "  -m, --mean-pixel             maximum allowed mean pixel error (default 5)\n"
53              << "  -s, --std-dev-pixel          maximum allowed standard deviation of pixel error (default 5)\n"
54              << "      --key                    hexadecimal key to use to decrypt MXFs\n"
55              << "  -k, --keep-going             carry on in the event of errors, if possible\n"
56              << "      --ignore-missing-assets  ignore missing asset files\n"
57              << "\n"
58              << "The <DCP>s are the DCP directories to compare.\n"
59              << "Comparison is of metadata and content, ignoring timestamps\n"
60              << "and differing UUIDs.\n";
61 }
62
63 void
64 note (NoteType t, string n)
65 {
66         if (t == DCP_ERROR || verbose) {
67                 cout << " " << n << "\n";
68         }
69 }
70
71 DCP *
72 load_dcp (boost::filesystem::path path, bool keep_going, bool ignore_missing_assets, optional<string> key)
73 {
74         DCP* dcp = 0;
75         try {
76                 dcp = new DCP (path);
77                 DCP::ReadErrors errors;
78                 dcp->read (keep_going, &errors);
79                 filter_errors (errors, ignore_missing_assets);
80                 for (DCP::ReadErrors::const_iterator i = errors.begin(); i != errors.end(); ++i) {
81                         cerr << (*i)->what() << "\n";
82                 }
83
84                 if (key) {
85                         list<shared_ptr<Asset> > assets = dcp->assets ();
86                         for (list<shared_ptr<Asset> >::const_iterator i = assets.begin(); i != assets.end(); ++i) {
87                                 shared_ptr<MXF> mxf = dynamic_pointer_cast<MXF> (*i);
88                                 if (mxf) {
89                                         mxf->set_key (Key (key.get ()));
90                                 }
91                         }
92                 }
93
94         } catch (FileError& e) {
95                 cerr << "Could not read DCP " << path.string() << "; " << e.what() << " " << e.filename() << "\n";
96                 exit (EXIT_FAILURE);
97         }
98
99         return dcp;
100 }
101
102 int
103 main (int argc, char* argv[])
104 {
105         EqualityOptions options;
106         options.max_mean_pixel_error = 5;
107         options.max_std_dev_pixel_error = 5;
108         options.reel_hashes_can_differ = true;
109         options.reel_annotation_texts_can_differ = false;
110         options.keep_going = false;
111         bool ignore_missing_assets = false;
112         optional<string> key;
113
114         int option_index = 0;
115         while (1) {
116                 static struct option long_options[] = {
117                         { "version", no_argument, 0, 'V'},
118                         { "help", no_argument, 0, 'h'},
119                         { "verbose", no_argument, 0, 'v'},
120                         { "mean-pixel", required_argument, 0, 'm'},
121                         { "std-dev-pixel", required_argument, 0, 's'},
122                         { "keep-going", no_argument, 0, 'k'},
123                         { "annotation-texts", no_argument, 0, 'a'},
124                         /* From here we're using random capital letters for the short option */
125                         { "ignore-missing-assets", no_argument, 0, 'A'},
126                         { "cpl-annotation-texts", no_argument, 0, 'C'},
127                         { "key", required_argument, 0, 'D'},
128                         { "reel-annotation-texts", no_argument, 0, 'E'},
129                         { 0, 0, 0, 0 }
130                 };
131
132                 int c = getopt_long (argc, argv, "Vhvm:s:kaACD:E", long_options, &option_index);
133
134                 if (c == -1) {
135                         break;
136                 }
137
138                 switch (c) {
139                 case 'V':
140                         cout << "dcpdiff version " << LIBDCP_VERSION << "\n";
141                         exit (EXIT_SUCCESS);
142                 case 'h':
143                         help (argv[0]);
144                         exit (EXIT_SUCCESS);
145                 case 'v':
146                         verbose = 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                         options.keep_going = true;
156                         break;
157                 case 'a':
158                         options.cpl_annotation_texts_can_differ = options.reel_annotation_texts_can_differ = true;
159                         break;
160                 case 'A':
161                         ignore_missing_assets = true;
162                         break;
163                 case 'B':
164                 case 'C':
165                         options.cpl_annotation_texts_can_differ = true;
166                         break;
167                 case 'D':
168                         key = string (optarg);
169                         break;
170                 case 'E':
171                         options.reel_annotation_texts_can_differ = true;
172                         break;
173                 }
174         }
175
176         if (argc <= optind || argc > (optind + 2)) {
177                 help (argv[0]);
178                 exit (EXIT_FAILURE);
179         }
180
181         if (!boost::filesystem::exists (argv[optind])) {
182                 cerr << argv[0] << ": DCP " << argv[optind] << " not found.\n";
183                 exit (EXIT_FAILURE);
184         }
185
186         if (!boost::filesystem::exists (argv[optind + 1])) {
187                 cerr << argv[0] << ": DCP " << argv[optind + 1] << " not found.\n";
188                 exit (EXIT_FAILURE);
189         }
190
191         DCP* a = load_dcp (argv[optind], options.keep_going, ignore_missing_assets, key);
192         DCP* b = load_dcp (argv[optind + 1], options.keep_going, ignore_missing_assets, key);
193
194         /* I think this is just below the LSB at 16-bits (ie the 8th most significant bit at 24-bit) */
195         options.max_audio_sample_error = 255;
196
197         bool const equals = a->equals (*b, options, boost::bind (note, _1, _2));
198
199         exit (equals ? EXIT_SUCCESS : EXIT_FAILURE);
200 }