Merge branch 'master' into trimming
[dcpomatic.git] / test / test.cc
1 /*
2     Copyright (C) 2012 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 <fstream>
21 #include <iostream>
22 #include <boost/filesystem.hpp>
23 #include <boost/algorithm/string/predicate.hpp>
24 #include <boost/date_time.hpp>
25 #include "format.h"
26 #include "film.h"
27 #include "filter.h"
28 #include "job_manager.h"
29 #include "util.h"
30 #include "exceptions.h"
31 #include "delay_line.h"
32 #include "image.h"
33 #include "log.h"
34 #include "dcp_video_frame.h"
35 #include "config.h"
36 #include "server.h"
37 #include "cross.h"
38 #include "job.h"
39 #include "subtitle.h"
40 #include "scaler.h"
41 #include "ffmpeg_decoder.h"
42 #include "sndfile_decoder.h"
43 #include "trimmer.h"
44 #define BOOST_TEST_DYN_LINK
45 #define BOOST_TEST_MODULE dvdomatic_test
46 #include <boost/test/unit_test.hpp>
47
48 using std::string;
49 using std::list;
50 using std::stringstream;
51 using std::vector;
52 using boost::shared_ptr;
53 using boost::thread;
54 using boost::dynamic_pointer_cast;
55
56 void
57 setup_test_config ()
58 {
59         Config::instance()->set_num_local_encoding_threads (1);
60         Config::instance()->set_servers (vector<ServerDescription*> ());
61         Config::instance()->set_server_port (61920);
62         Config::instance()->set_default_dci_metadata (DCIMetadata ());
63 }
64
65 boost::filesystem::path
66 test_film_dir (string name)
67 {
68         boost::filesystem::path p;
69         p /= "build";
70         p /= "test";
71         p /= name;
72         return p;
73 }
74
75 shared_ptr<Film>
76 new_test_film (string name)
77 {
78         boost::filesystem::path p = test_film_dir (name);
79         if (boost::filesystem::exists (p)) {
80                 boost::filesystem::remove_all (p);
81         }
82         
83         return shared_ptr<Film> (new Film (p.string(), false));
84 }
85
86
87 /* Check that Image::make_black works, and doesn't use values which crash
88    sws_scale().
89 */
90 BOOST_AUTO_TEST_CASE (make_black_test)
91 {
92         /* This needs to happen in the first test */
93         dvdomatic_setup ();
94
95         libdcp::Size in_size (512, 512);
96         libdcp::Size out_size (1024, 1024);
97
98         list<AVPixelFormat> pix_fmts;
99         pix_fmts.push_back (AV_PIX_FMT_RGB24);
100         pix_fmts.push_back (AV_PIX_FMT_YUV420P);
101         pix_fmts.push_back (AV_PIX_FMT_YUV422P10LE);
102         pix_fmts.push_back (AV_PIX_FMT_YUV444P9LE);
103         pix_fmts.push_back (AV_PIX_FMT_YUV444P9BE);
104         pix_fmts.push_back (AV_PIX_FMT_YUV444P10LE);
105         pix_fmts.push_back (AV_PIX_FMT_YUV444P10BE);
106         pix_fmts.push_back (AV_PIX_FMT_UYVY422);
107
108         int N = 0;
109         for (list<AVPixelFormat>::const_iterator i = pix_fmts.begin(); i != pix_fmts.end(); ++i) {
110                 boost::shared_ptr<Image> foo (new SimpleImage (*i, in_size, true));
111                 foo->make_black ();
112                 boost::shared_ptr<Image> bar = foo->scale_and_convert_to_rgb (out_size, 0, Scaler::from_id ("bicubic"), true);
113                 
114                 uint8_t* p = bar->data()[0];
115                 for (int y = 0; y < bar->size().height; ++y) {
116                         uint8_t* q = p;
117                         for (int x = 0; x < bar->line_size()[0]; ++x) {
118                                 BOOST_CHECK_EQUAL (*q++, 0);
119                         }
120                         p += bar->stride()[0];
121                 }
122
123                 ++N;
124         }
125 }
126
127 shared_ptr<AudioBuffers> trimmer_test_last;
128
129 void
130 trimmer_test_helper (shared_ptr<AudioBuffers> audio)
131 {
132         trimmer_test_last = audio;
133 }
134
135 /** Test the audio handling of the Trimmer */
136 BOOST_AUTO_TEST_CASE (trimmer_test)
137 {
138         Trimmer trimmer (shared_ptr<Log> (), 25, 75, 200, 48000, 25, 25);
139
140         trimmer.Audio.connect (bind (&trimmer_test_helper, _1));
141
142         /* 21 video frames-worth of audio frames; should be completely stripped */
143         trimmer_test_last.reset ();
144         shared_ptr<AudioBuffers> audio (new AudioBuffers (6, 21 * 1920));
145         trimmer.process_audio (audio);
146         BOOST_CHECK (trimmer_test_last == 0);
147
148         /* 42 more video frames-worth, 4 should be stripped from the start */
149         audio.reset (new AudioBuffers (6, 42 * 1920));
150         trimmer.process_audio (audio);
151         BOOST_CHECK (trimmer_test_last);
152         BOOST_CHECK_EQUAL (trimmer_test_last->frames(), 38 * 1920);
153
154         /* 42 more video frames-worth, should be kept as-is */
155         trimmer_test_last.reset ();
156         audio.reset (new AudioBuffers (6, 42 * 1920));
157         trimmer.process_audio (audio);
158         BOOST_CHECK (trimmer_test_last);
159         BOOST_CHECK_EQUAL (trimmer_test_last->frames(), 42 * 1920);
160
161         /* 25 more video frames-worth, 5 should be trimmed from the end */
162         trimmer_test_last.reset ();
163         audio.reset (new AudioBuffers (6, 25 * 1920));
164         trimmer.process_audio (audio);
165         BOOST_CHECK (trimmer_test_last);
166         BOOST_CHECK_EQUAL (trimmer_test_last->frames(), 20 * 1920);
167
168         /* Now some more; all should be trimmed */
169         trimmer_test_last.reset ();
170         audio.reset (new AudioBuffers (6, 100 * 1920));
171         trimmer.process_audio (audio);
172         BOOST_CHECK (trimmer_test_last == 0);
173 }
174
175
176 BOOST_AUTO_TEST_CASE (film_metadata_test)
177 {
178         setup_test_config ();
179
180         string const test_film = "build/test/film_metadata_test";
181         
182         if (boost::filesystem::exists (test_film)) {
183                 boost::filesystem::remove_all (test_film);
184         }
185
186         BOOST_CHECK_THROW (new Film (test_film, true), OpenFileError);
187         
188         shared_ptr<Film> f (new Film (test_film, false));
189         f->_dci_date = boost::gregorian::from_undelimited_string ("20130211");
190         BOOST_CHECK (f->format() == 0);
191         BOOST_CHECK (f->dcp_content_type() == 0);
192         BOOST_CHECK (f->filters ().empty());
193
194         f->set_name ("fred");
195         BOOST_CHECK_THROW (f->set_content ("jim"), OpenFileError);
196         f->set_dcp_content_type (DCPContentType::from_pretty_name ("Short"));
197         f->set_format (Format::from_nickname ("Flat"));
198         f->set_left_crop (1);
199         f->set_right_crop (2);
200         f->set_top_crop (3);
201         f->set_bottom_crop (4);
202         vector<Filter const *> f_filters;
203         f_filters.push_back (Filter::from_id ("pphb"));
204         f_filters.push_back (Filter::from_id ("unsharp"));
205         f->set_filters (f_filters);
206         f->set_trim_start (42);
207         f->set_trim_end (99);
208         f->set_dcp_ab (true);
209         f->write_metadata ();
210
211         stringstream s;
212         s << "diff -u test/metadata.ref " << test_film << "/metadata";
213         BOOST_CHECK_EQUAL (::system (s.str().c_str ()), 0);
214
215         shared_ptr<Film> g (new Film (test_film, true));
216
217         BOOST_CHECK_EQUAL (g->name(), "fred");
218         BOOST_CHECK_EQUAL (g->dcp_content_type(), DCPContentType::from_pretty_name ("Short"));
219         BOOST_CHECK_EQUAL (g->format(), Format::from_nickname ("Flat"));
220         BOOST_CHECK_EQUAL (g->crop().left, 1);
221         BOOST_CHECK_EQUAL (g->crop().right, 2);
222         BOOST_CHECK_EQUAL (g->crop().top, 3);
223         BOOST_CHECK_EQUAL (g->crop().bottom, 4);
224         vector<Filter const *> g_filters = g->filters ();
225         BOOST_CHECK_EQUAL (g_filters.size(), 2);
226         BOOST_CHECK_EQUAL (g_filters.front(), Filter::from_id ("pphb"));
227         BOOST_CHECK_EQUAL (g_filters.back(), Filter::from_id ("unsharp"));
228         BOOST_CHECK_EQUAL (g->trim_start(), 42);
229         BOOST_CHECK_EQUAL (g->trim_end(), 99);
230         BOOST_CHECK_EQUAL (g->dcp_ab(), true);
231         
232         g->write_metadata ();
233         BOOST_CHECK_EQUAL (::system (s.str().c_str ()), 0);
234 }
235
236 BOOST_AUTO_TEST_CASE (stream_test)
237 {
238         FFmpegAudioStream a ("ffmpeg 4 44100 1 hello there world", boost::optional<int> (1));
239         BOOST_CHECK_EQUAL (a.id(), 4);
240         BOOST_CHECK_EQUAL (a.sample_rate(), 44100);
241         BOOST_CHECK_EQUAL (a.channel_layout(), 1);
242         BOOST_CHECK_EQUAL (a.name(), "hello there world");
243         BOOST_CHECK_EQUAL (a.to_string(), "ffmpeg 4 44100 1 hello there world");
244
245         SndfileStream e ("external 44100 1", boost::optional<int> (1));
246         BOOST_CHECK_EQUAL (e.sample_rate(), 44100);
247         BOOST_CHECK_EQUAL (e.channel_layout(), 1);
248         BOOST_CHECK_EQUAL (e.to_string(), "external 44100 1");
249
250         SubtitleStream s ("5 a b c", boost::optional<int> (1));
251         BOOST_CHECK_EQUAL (s.id(), 5);
252         BOOST_CHECK_EQUAL (s.name(), "a b c");
253
254         shared_ptr<AudioStream> ff = audio_stream_factory ("ffmpeg 4 44100 1 hello there world", boost::optional<int> (1));
255         shared_ptr<FFmpegAudioStream> cff = dynamic_pointer_cast<FFmpegAudioStream> (ff);
256         BOOST_CHECK (cff);
257         BOOST_CHECK_EQUAL (cff->id(), 4);
258         BOOST_CHECK_EQUAL (cff->sample_rate(), 44100);
259         BOOST_CHECK_EQUAL (cff->channel_layout(), 1);
260         BOOST_CHECK_EQUAL (cff->name(), "hello there world");
261         BOOST_CHECK_EQUAL (cff->to_string(), "ffmpeg 4 44100 1 hello there world");
262
263         shared_ptr<AudioStream> fe = audio_stream_factory ("external 44100 1", boost::optional<int> (1));
264         BOOST_CHECK_EQUAL (fe->sample_rate(), 44100);
265         BOOST_CHECK_EQUAL (fe->channel_layout(), 1);
266         BOOST_CHECK_EQUAL (fe->to_string(), "external 44100 1");
267 }
268
269 BOOST_AUTO_TEST_CASE (format_test)
270 {
271         Format::setup_formats ();
272         
273         Format const * f = Format::from_nickname ("Flat");
274         BOOST_CHECK (f);
275         BOOST_CHECK_EQUAL (f->ratio_as_integer(shared_ptr<const Film> ()), 185);
276         
277         f = Format::from_nickname ("Scope");
278         BOOST_CHECK (f);
279         BOOST_CHECK_EQUAL (f->ratio_as_integer(shared_ptr<const Film> ()), 239);
280 }
281
282 /* Test VariableFormat-based scaling of content */
283 BOOST_AUTO_TEST_CASE (scaling_test)
284 {
285         shared_ptr<Film> film (new Film (test_film_dir ("scaling_test").string(), false));
286
287         /* 4:3 ratio */
288         film->set_size (libdcp::Size (320, 240));
289
290         /* This format should preserve aspect ratio of the source */
291         Format const * format = Format::from_id ("var-185");
292
293         /* We should have enough padding that the result is 4:3,
294            which would be 1440 pixels.
295         */
296         BOOST_CHECK_EQUAL (format->dcp_padding (film), (1998 - 1440) / 2);
297         
298         /* This crops it to 1.291666667 */
299         film->set_left_crop (5);
300         film->set_right_crop (5);
301
302         /* We should now have enough padding that the result is 1.29166667,
303            which would be 1395 pixels.
304         */
305         BOOST_CHECK_EQUAL (format->dcp_padding (film), rint ((1998 - 1395) / 2.0));
306 }
307
308 BOOST_AUTO_TEST_CASE (util_test)
309 {
310         string t = "Hello this is a string \"with quotes\" and indeed without them";
311         vector<string> b = split_at_spaces_considering_quotes (t);
312         vector<string>::iterator i = b.begin ();
313         BOOST_CHECK_EQUAL (*i++, "Hello");
314         BOOST_CHECK_EQUAL (*i++, "this");
315         BOOST_CHECK_EQUAL (*i++, "is");
316         BOOST_CHECK_EQUAL (*i++, "a");
317         BOOST_CHECK_EQUAL (*i++, "string");
318         BOOST_CHECK_EQUAL (*i++, "with quotes");
319         BOOST_CHECK_EQUAL (*i++, "and");
320         BOOST_CHECK_EQUAL (*i++, "indeed");
321         BOOST_CHECK_EQUAL (*i++, "without");
322         BOOST_CHECK_EQUAL (*i++, "them");
323 }
324
325 class NullLog : public Log
326 {
327 public:
328         void do_log (string) {}
329 };
330
331 void
332 do_positive_delay_line_test (int delay_length, int data_length)
333 {
334         shared_ptr<NullLog> log (new NullLog);
335         
336         DelayLine d (log, 6, delay_length);
337         shared_ptr<AudioBuffers> data (new AudioBuffers (6, data_length));
338
339         int in = 0;
340         int out = 0;
341         int returned = 0;
342         int zeros = 0;
343         
344         for (int i = 0; i < 64; ++i) {
345                 for (int j = 0; j < data_length; ++j) {
346                         for (int c = 0; c < 6; ++c ) {
347                                 data->data(c)[j] = in;
348                                 ++in;
349                         }
350                 }
351
352                 /* This only works because the delay line modifies the parameter */
353                 d.process_audio (data);
354                 returned += data->frames ();
355
356                 for (int j = 0; j < data->frames(); ++j) {
357                         if (zeros < delay_length) {
358                                 for (int c = 0; c < 6; ++c) {
359                                         BOOST_CHECK_EQUAL (data->data(c)[j], 0);
360                                 }
361                                 ++zeros;
362                         } else {
363                                 for (int c = 0; c < 6; ++c) {
364                                         BOOST_CHECK_EQUAL (data->data(c)[j], out);
365                                         ++out;
366                                 }
367                         }
368                 }
369         }
370
371         BOOST_CHECK_EQUAL (returned, 64 * data_length);
372 }
373
374 void
375 do_negative_delay_line_test (int delay_length, int data_length)
376 {
377         shared_ptr<NullLog> log (new NullLog);
378
379         DelayLine d (log, 6, delay_length);
380         shared_ptr<AudioBuffers> data (new AudioBuffers (6, data_length));
381
382         int in = 0;
383         int out = -delay_length * 6;
384         int returned = 0;
385         
386         for (int i = 0; i < 256; ++i) {
387                 data->set_frames (data_length);
388                 for (int j = 0; j < data_length; ++j) {
389                         for (int c = 0; c < 6; ++c) {
390                                 data->data(c)[j] = in;
391                                 ++in;
392                         }
393                 }
394
395                 /* This only works because the delay line modifies the parameter */
396                 d.process_audio (data);
397                 returned += data->frames ();
398
399                 for (int j = 0; j < data->frames(); ++j) {
400                         for (int c = 0; c < 6; ++c) {
401                                 BOOST_CHECK_EQUAL (data->data(c)[j], out);
402                                 ++out;
403                         }
404                 }
405         }
406
407         returned += -delay_length;
408         BOOST_CHECK_EQUAL (returned, 256 * data_length);
409 }
410
411 BOOST_AUTO_TEST_CASE (delay_line_test)
412 {
413         do_positive_delay_line_test (64, 128);
414         do_positive_delay_line_test (128, 64);
415         do_positive_delay_line_test (3, 512);
416         do_positive_delay_line_test (512, 3);
417
418         do_positive_delay_line_test (0, 64);
419
420         do_negative_delay_line_test (-64, 128);
421         do_negative_delay_line_test (-128, 64);
422         do_negative_delay_line_test (-3, 512);
423         do_negative_delay_line_test (-512, 3);
424 }
425
426 BOOST_AUTO_TEST_CASE (md5_digest_test)
427 {
428         string const t = md5_digest ("test/md5.test");
429         BOOST_CHECK_EQUAL (t, "15058685ba99decdc4398c7634796eb0");
430
431         BOOST_CHECK_THROW (md5_digest ("foobar"), OpenFileError);
432 }
433
434 BOOST_AUTO_TEST_CASE (paths_test)
435 {
436         shared_ptr<Film> f = new_test_film ("paths_test");
437         f->set_directory ("build/test/a/b/c/d/e");
438
439         f->_content = "/foo/bar/baz";
440         BOOST_CHECK_EQUAL (f->content_path(), "/foo/bar/baz");
441         f->_content = "foo/bar/baz";
442         BOOST_CHECK_EQUAL (f->content_path(), "build/test/a/b/c/d/e/foo/bar/baz");
443 }
444
445 void
446 do_remote_encode (shared_ptr<DCPVideoFrame> frame, ServerDescription* description, shared_ptr<EncodedData> locally_encoded)
447 {
448         shared_ptr<EncodedData> remotely_encoded;
449         BOOST_CHECK_NO_THROW (remotely_encoded = frame->encode_remotely (description));
450         BOOST_CHECK (remotely_encoded);
451         
452         BOOST_CHECK_EQUAL (locally_encoded->size(), remotely_encoded->size());
453         BOOST_CHECK (memcmp (locally_encoded->data(), remotely_encoded->data(), locally_encoded->size()) == 0);
454 }
455
456 BOOST_AUTO_TEST_CASE (client_server_test)
457 {
458         shared_ptr<Image> image (new SimpleImage (PIX_FMT_RGB24, libdcp::Size (1998, 1080), true));
459         uint8_t* p = image->data()[0];
460         
461         for (int y = 0; y < 1080; ++y) {
462                 uint8_t* q = p;
463                 for (int x = 0; x < 1998; ++x) {
464                         *q++ = x % 256;
465                         *q++ = y % 256;
466                         *q++ = (x + y) % 256;
467                 }
468                 p += image->stride()[0];
469         }
470
471         shared_ptr<Image> sub_image (new SimpleImage (PIX_FMT_RGBA, libdcp::Size (100, 200), true));
472         p = sub_image->data()[0];
473         for (int y = 0; y < 200; ++y) {
474                 uint8_t* q = p;
475                 for (int x = 0; x < 100; ++x) {
476                         *q++ = y % 256;
477                         *q++ = x % 256;
478                         *q++ = (x + y) % 256;
479                         *q++ = 1;
480                 }
481                 p += sub_image->stride()[0];
482         }
483
484         shared_ptr<Subtitle> subtitle (new Subtitle (Position (50, 60), sub_image));
485
486         shared_ptr<FileLog> log (new FileLog ("build/test/client_server_test.log"));
487
488         shared_ptr<DCPVideoFrame> frame (
489                 new DCPVideoFrame (
490                         image,
491                         subtitle,
492                         libdcp::Size (1998, 1080),
493                         0,
494                         0,
495                         1,
496                         Scaler::from_id ("bicubic"),
497                         0,
498                         24,
499                         "",
500                         0,
501                         200000000,
502                         log
503                         )
504                 );
505
506         shared_ptr<EncodedData> locally_encoded = frame->encode_locally ();
507         BOOST_ASSERT (locally_encoded);
508         
509         Server* server = new Server (log);
510
511         new thread (boost::bind (&Server::run, server, 2));
512
513         /* Let the server get itself ready */
514         dvdomatic_sleep (1);
515
516         ServerDescription description ("localhost", 2);
517
518         list<thread*> threads;
519         for (int i = 0; i < 8; ++i) {
520                 threads.push_back (new thread (boost::bind (do_remote_encode, frame, &description, locally_encoded)));
521         }
522
523         for (list<thread*>::iterator i = threads.begin(); i != threads.end(); ++i) {
524                 (*i)->join ();
525         }
526
527         for (list<thread*>::iterator i = threads.begin(); i != threads.end(); ++i) {
528                 delete *i;
529         }
530 }
531
532 BOOST_AUTO_TEST_CASE (make_dcp_test)
533 {
534         shared_ptr<Film> film = new_test_film ("make_dcp_test");
535         film->set_name ("test_film2");
536         film->set_content ("../../../test/test.mp4");
537         film->set_format (Format::from_nickname ("Flat"));
538         film->set_dcp_content_type (DCPContentType::from_pretty_name ("Test"));
539         film->make_dcp ();
540         film->write_metadata ();
541
542         while (JobManager::instance()->work_to_do ()) {
543                 dvdomatic_sleep (1);
544         }
545         
546         BOOST_CHECK_EQUAL (JobManager::instance()->errors(), false);
547 }
548
549 /** Test Film::have_dcp().  Requires the output from make_dcp_test above */
550 BOOST_AUTO_TEST_CASE (have_dcp_test)
551 {
552         boost::filesystem::path p = test_film_dir ("make_dcp_test");
553         Film f (p.string ());
554         BOOST_CHECK (f.have_dcp());
555
556         p /= f.dcp_name();
557         p /= f.dcp_video_mxf_filename();
558         boost::filesystem::remove (p);
559         BOOST_CHECK (!f.have_dcp ());
560 }
561
562 BOOST_AUTO_TEST_CASE (make_dcp_with_range_test)
563 {
564         shared_ptr<Film> film = new_test_film ("make_dcp_with_range_test");
565         film->set_name ("test_film3");
566         film->set_content ("../../../test/test.mp4");
567         film->examine_content ();
568         film->set_format (Format::from_nickname ("Flat"));
569         film->set_dcp_content_type (DCPContentType::from_pretty_name ("Test"));
570         film->set_trim_end (42);
571         film->make_dcp ();
572
573         while (JobManager::instance()->work_to_do() && !JobManager::instance()->errors()) {
574                 dvdomatic_sleep (1);
575         }
576
577         BOOST_CHECK_EQUAL (JobManager::instance()->errors(), false);
578 }
579
580 /* Test best_dcp_frame_rate and FrameRateConversion */
581 BOOST_AUTO_TEST_CASE (best_dcp_frame_rate_test)
582 {
583         /* Run some tests with a limited range of allowed rates */
584         
585         std::list<int> afr;
586         afr.push_back (24);
587         afr.push_back (25);
588         afr.push_back (30);
589         Config::instance()->set_allowed_dcp_frame_rates (afr);
590
591         int best = best_dcp_frame_rate (60);
592         FrameRateConversion frc = FrameRateConversion (60, best);
593         BOOST_CHECK_EQUAL (best, 30);
594         BOOST_CHECK_EQUAL (frc.skip, true);
595         BOOST_CHECK_EQUAL (frc.repeat, false);
596         BOOST_CHECK_EQUAL (frc.change_speed, false);
597         
598         best = best_dcp_frame_rate (50);
599         frc = FrameRateConversion (50, best);
600         BOOST_CHECK_EQUAL (best, 25);
601         BOOST_CHECK_EQUAL (frc.skip, true);
602         BOOST_CHECK_EQUAL (frc.repeat, false);
603         BOOST_CHECK_EQUAL (frc.change_speed, false);
604
605         best = best_dcp_frame_rate (48);
606         frc = FrameRateConversion (48, best);
607         BOOST_CHECK_EQUAL (best, 24);
608         BOOST_CHECK_EQUAL (frc.skip, true);
609         BOOST_CHECK_EQUAL (frc.repeat, false);
610         BOOST_CHECK_EQUAL (frc.change_speed, false);
611         
612         best = best_dcp_frame_rate (30);
613         frc = FrameRateConversion (30, best);
614         BOOST_CHECK_EQUAL (best, 30);
615         BOOST_CHECK_EQUAL (frc.skip, false);
616         BOOST_CHECK_EQUAL (frc.repeat, false);
617         BOOST_CHECK_EQUAL (frc.change_speed, false);
618
619         best = best_dcp_frame_rate (29.97);
620         frc = FrameRateConversion (29.97, best);
621         BOOST_CHECK_EQUAL (best, 30);
622         BOOST_CHECK_EQUAL (frc.skip, false);
623         BOOST_CHECK_EQUAL (frc.repeat, false);
624         BOOST_CHECK_EQUAL (frc.change_speed, true);
625         
626         best = best_dcp_frame_rate (25);
627         frc = FrameRateConversion (25, best);
628         BOOST_CHECK_EQUAL (best, 25);
629         BOOST_CHECK_EQUAL (frc.skip, false);
630         BOOST_CHECK_EQUAL (frc.repeat, false);
631         BOOST_CHECK_EQUAL (frc.change_speed, false);
632
633         best = best_dcp_frame_rate (24);
634         frc = FrameRateConversion (24, best);
635         BOOST_CHECK_EQUAL (best, 24);
636         BOOST_CHECK_EQUAL (frc.skip, false);
637         BOOST_CHECK_EQUAL (frc.repeat, false);
638         BOOST_CHECK_EQUAL (frc.change_speed, false);
639
640         best = best_dcp_frame_rate (14.5);
641         frc = FrameRateConversion (14.5, best);
642         BOOST_CHECK_EQUAL (best, 30);
643         BOOST_CHECK_EQUAL (frc.skip, false);
644         BOOST_CHECK_EQUAL (frc.repeat, true);
645         BOOST_CHECK_EQUAL (frc.change_speed, true);
646
647         best = best_dcp_frame_rate (12.6);
648         frc = FrameRateConversion (12.6, best);
649         BOOST_CHECK_EQUAL (best, 25);
650         BOOST_CHECK_EQUAL (frc.skip, false);
651         BOOST_CHECK_EQUAL (frc.repeat, true);
652         BOOST_CHECK_EQUAL (frc.change_speed, true);
653
654         best = best_dcp_frame_rate (12.4);
655         frc = FrameRateConversion (12.4, best);
656         BOOST_CHECK_EQUAL (best, 25);
657         BOOST_CHECK_EQUAL (frc.skip, false);
658         BOOST_CHECK_EQUAL (frc.repeat, true);
659         BOOST_CHECK_EQUAL (frc.change_speed, true);
660
661         best = best_dcp_frame_rate (12);
662         frc = FrameRateConversion (12, best);
663         BOOST_CHECK_EQUAL (best, 24);
664         BOOST_CHECK_EQUAL (frc.skip, false);
665         BOOST_CHECK_EQUAL (frc.repeat, true);
666         BOOST_CHECK_EQUAL (frc.change_speed, false);
667
668         /* Now add some more rates and see if it will use them
669            in preference to skip/repeat.
670         */
671
672         afr.push_back (48);
673         afr.push_back (50);
674         afr.push_back (60);
675         Config::instance()->set_allowed_dcp_frame_rates (afr);
676
677         best = best_dcp_frame_rate (60);
678         frc = FrameRateConversion (60, best);
679         BOOST_CHECK_EQUAL (best, 60);
680         BOOST_CHECK_EQUAL (frc.skip, false);
681         BOOST_CHECK_EQUAL (frc.repeat, false);
682         BOOST_CHECK_EQUAL (frc.change_speed, false);
683         
684         best = best_dcp_frame_rate (50);
685         frc = FrameRateConversion (50, best);
686         BOOST_CHECK_EQUAL (best, 50);
687         BOOST_CHECK_EQUAL (frc.skip, false);
688         BOOST_CHECK_EQUAL (frc.repeat, false);
689         BOOST_CHECK_EQUAL (frc.change_speed, false);
690
691         best = best_dcp_frame_rate (48);
692         frc = FrameRateConversion (48, best);
693         BOOST_CHECK_EQUAL (best, 48);
694         BOOST_CHECK_EQUAL (frc.skip, false);
695         BOOST_CHECK_EQUAL (frc.repeat, false);
696         BOOST_CHECK_EQUAL (frc.change_speed, false);
697
698         /* Check some out-there conversions (not the best) */
699         
700         frc = FrameRateConversion (14.99, 24);
701         BOOST_CHECK_EQUAL (frc.skip, false);
702         BOOST_CHECK_EQUAL (frc.repeat, true);
703         BOOST_CHECK_EQUAL (frc.change_speed, true);
704
705         /* Check some conversions with limited DCP targets */
706
707         afr.clear ();
708         afr.push_back (24);
709         Config::instance()->set_allowed_dcp_frame_rates (afr);
710
711         best = best_dcp_frame_rate (25);
712         frc = FrameRateConversion (25, best);
713         BOOST_CHECK_EQUAL (best, 24);
714         BOOST_CHECK_EQUAL (frc.skip, false);
715         BOOST_CHECK_EQUAL (frc.repeat, false);
716         BOOST_CHECK_EQUAL (frc.change_speed, true);
717 }
718
719 BOOST_AUTO_TEST_CASE (audio_sampling_rate_test)
720 {
721         std::list<int> afr;
722         afr.push_back (24);
723         afr.push_back (25);
724         afr.push_back (30);
725         Config::instance()->set_allowed_dcp_frame_rates (afr);
726
727         shared_ptr<Film> f = new_test_film ("audio_sampling_rate_test");
728         f->set_source_frame_rate (24);
729         f->set_dcp_frame_rate (24);
730
731         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 48000, 0)));
732         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 48000);
733
734         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 44100, 0)));
735         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 48000);
736
737         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 80000, 0)));
738         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 96000);
739
740         f->set_source_frame_rate (23.976);
741         f->set_dcp_frame_rate (best_dcp_frame_rate (23.976));
742         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 48000, 0)));
743         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 47952);
744
745         f->set_source_frame_rate (29.97);
746         f->set_dcp_frame_rate (best_dcp_frame_rate (29.97));
747         BOOST_CHECK_EQUAL (f->dcp_frame_rate (), 30);
748         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 48000, 0)));
749         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 47952);
750
751         f->set_source_frame_rate (25);
752         f->set_dcp_frame_rate (24);
753         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 48000, 0)));
754         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 50000);
755
756         f->set_source_frame_rate (25);
757         f->set_dcp_frame_rate (24);
758         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 44100, 0)));
759         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), 50000);
760
761         /* Check some out-there conversions (not the best) */
762         
763         f->set_source_frame_rate (14.99);
764         f->set_dcp_frame_rate (25);
765         f->set_content_audio_stream (shared_ptr<AudioStream> (new FFmpegAudioStream ("a", 42, 16000, 0)));
766         /* The FrameRateConversion within target_audio_sample_rate should choose to double-up
767            the 14.99 fps video to 30 and then run it slow at 25.
768         */
769         BOOST_CHECK_EQUAL (f->target_audio_sample_rate(), rint (48000 * 2 * 14.99 / 25));
770 }
771
772 class TestJob : public Job
773 {
774 public:
775         TestJob (shared_ptr<Film> f)
776                 : Job (f)
777         {
778
779         }
780
781         void set_finished_ok () {
782                 set_state (FINISHED_OK);
783         }
784
785         void set_finished_error () {
786                 set_state (FINISHED_ERROR);
787         }
788
789         void run ()
790         {
791                 while (1) {
792                         if (finished ()) {
793                                 return;
794                         }
795                 }
796         }
797
798         string name () const {
799                 return "";
800         }
801 };
802
803 BOOST_AUTO_TEST_CASE (job_manager_test)
804 {
805         shared_ptr<Film> f;
806
807         /* Single job */
808         shared_ptr<TestJob> a (new TestJob (f));
809
810         JobManager::instance()->add (a);
811         dvdomatic_sleep (1);
812         BOOST_CHECK_EQUAL (a->running (), true);
813         a->set_finished_ok ();
814         dvdomatic_sleep (2);
815         BOOST_CHECK_EQUAL (a->finished_ok(), true);
816 }
817
818 BOOST_AUTO_TEST_CASE (compact_image_test)
819 {
820         SimpleImage* s = new SimpleImage (PIX_FMT_RGB24, libdcp::Size (50, 50), false);
821         BOOST_CHECK_EQUAL (s->components(), 1);
822         BOOST_CHECK_EQUAL (s->stride()[0], 50 * 3);
823         BOOST_CHECK_EQUAL (s->line_size()[0], 50 * 3);
824         BOOST_CHECK (s->data()[0]);
825         BOOST_CHECK (!s->data()[1]);
826         BOOST_CHECK (!s->data()[2]);
827         BOOST_CHECK (!s->data()[3]);
828
829         /* copy constructor */
830         SimpleImage* t = new SimpleImage (*s);
831         BOOST_CHECK_EQUAL (t->components(), 1);
832         BOOST_CHECK_EQUAL (t->stride()[0], 50 * 3);
833         BOOST_CHECK_EQUAL (t->line_size()[0], 50 * 3);
834         BOOST_CHECK (t->data()[0]);
835         BOOST_CHECK (!t->data()[1]);
836         BOOST_CHECK (!t->data()[2]);
837         BOOST_CHECK (!t->data()[3]);
838         BOOST_CHECK (t->data() != s->data());
839         BOOST_CHECK (t->data()[0] != s->data()[0]);
840         BOOST_CHECK (t->line_size() != s->line_size());
841         BOOST_CHECK (t->line_size()[0] == s->line_size()[0]);
842         BOOST_CHECK (t->stride() != s->stride());
843         BOOST_CHECK (t->stride()[0] == s->stride()[0]);
844
845         /* assignment operator */
846         SimpleImage* u = new SimpleImage (PIX_FMT_YUV422P, libdcp::Size (150, 150), true);
847         *u = *s;
848         BOOST_CHECK_EQUAL (u->components(), 1);
849         BOOST_CHECK_EQUAL (u->stride()[0], 50 * 3);
850         BOOST_CHECK_EQUAL (u->line_size()[0], 50 * 3);
851         BOOST_CHECK (u->data()[0]);
852         BOOST_CHECK (!u->data()[1]);
853         BOOST_CHECK (!u->data()[2]);
854         BOOST_CHECK (!u->data()[3]);
855         BOOST_CHECK (u->data() != s->data());
856         BOOST_CHECK (u->data()[0] != s->data()[0]);
857         BOOST_CHECK (u->line_size() != s->line_size());
858         BOOST_CHECK (u->line_size()[0] == s->line_size()[0]);
859         BOOST_CHECK (u->stride() != s->stride());
860         BOOST_CHECK (u->stride()[0] == s->stride()[0]);
861
862         delete s;
863         delete t;
864         delete u;
865 }
866
867 BOOST_AUTO_TEST_CASE (aligned_image_test)
868 {
869         SimpleImage* s = new SimpleImage (PIX_FMT_RGB24, libdcp::Size (50, 50), true);
870         BOOST_CHECK_EQUAL (s->components(), 1);
871         /* 160 is 150 aligned to the nearest 32 bytes */
872         BOOST_CHECK_EQUAL (s->stride()[0], 160);
873         BOOST_CHECK_EQUAL (s->line_size()[0], 150);
874         BOOST_CHECK (s->data()[0]);
875         BOOST_CHECK (!s->data()[1]);
876         BOOST_CHECK (!s->data()[2]);
877         BOOST_CHECK (!s->data()[3]);
878
879         /* copy constructor */
880         SimpleImage* t = new SimpleImage (*s);
881         BOOST_CHECK_EQUAL (t->components(), 1);
882         BOOST_CHECK_EQUAL (t->stride()[0], 160);
883         BOOST_CHECK_EQUAL (t->line_size()[0], 150);
884         BOOST_CHECK (t->data()[0]);
885         BOOST_CHECK (!t->data()[1]);
886         BOOST_CHECK (!t->data()[2]);
887         BOOST_CHECK (!t->data()[3]);
888         BOOST_CHECK (t->data() != s->data());
889         BOOST_CHECK (t->data()[0] != s->data()[0]);
890         BOOST_CHECK (t->line_size() != s->line_size());
891         BOOST_CHECK (t->line_size()[0] == s->line_size()[0]);
892         BOOST_CHECK (t->stride() != s->stride());
893         BOOST_CHECK (t->stride()[0] == s->stride()[0]);
894
895         /* assignment operator */
896         SimpleImage* u = new SimpleImage (PIX_FMT_YUV422P, libdcp::Size (150, 150), false);
897         *u = *s;
898         BOOST_CHECK_EQUAL (u->components(), 1);
899         BOOST_CHECK_EQUAL (u->stride()[0], 160);
900         BOOST_CHECK_EQUAL (u->line_size()[0], 150);
901         BOOST_CHECK (u->data()[0]);
902         BOOST_CHECK (!u->data()[1]);
903         BOOST_CHECK (!u->data()[2]);
904         BOOST_CHECK (!u->data()[3]);
905         BOOST_CHECK (u->data() != s->data());
906         BOOST_CHECK (u->data()[0] != s->data()[0]);
907         BOOST_CHECK (u->line_size() != s->line_size());
908         BOOST_CHECK (u->line_size()[0] == s->line_size()[0]);
909         BOOST_CHECK (u->stride() != s->stride());
910         BOOST_CHECK (u->stride()[0] == s->stride()[0]);
911
912         delete s;
913         delete t;
914         delete u;
915 }
916