29e8ad05ce7c8021f7b3f57cbe5e21b4a704d074
[dcpomatic.git] / test / required_disk_space_test.cc
1 /*
2     Copyright (C) 2016 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "lib/content_factory.h"
22 #include "lib/film.h"
23 #include "lib/dcp_content.h"
24 #include "test.h"
25 #include <boost/test/unit_test.hpp>
26
27 using boost::shared_ptr;
28 using boost::dynamic_pointer_cast;
29
30 void check_within_n (int64_t a, int64_t b, int64_t n)
31 {
32         BOOST_CHECK (abs (a - b) <= n);
33 }
34
35
36 BOOST_AUTO_TEST_CASE (required_disk_space_test)
37 {
38         shared_ptr<Film> film = new_test_film ("required_disk_space_test");
39         film->set_j2k_bandwidth (100000000);
40         film->set_audio_channels (6);
41         shared_ptr<Content> content_a = content_factory (film, "test/data/flat_blue.png");
42         film->examine_and_add_content (content_a);
43         shared_ptr<DCPContent> content_b = dynamic_pointer_cast<DCPContent> (content_factory (film, "test/data/burnt_subtitle_test_dcp"));
44         film->examine_and_add_content (content_b);
45         wait_for_jobs ();
46         film->write_metadata ();
47
48         check_within_n (
49                 film->required_disk_space(),
50                 289LL * (100000000 / 8) / 24 +  // video
51                 289LL * 48000 * 6 * 3 / 24 +    // audio
52                 65536,                          // extra
53                 16
54                 );
55
56         content_b->set_reference_video (true);
57
58         check_within_n (
59                 film->required_disk_space(),
60                 240LL * (100000000 / 8) / 24 +  // video
61                 289LL * 48000 * 6 * 3 / 24 +    // audio
62                 65536,                          // extra
63                 16
64                 );
65
66         content_b->set_reference_audio (true);
67
68         check_within_n (
69                 film->required_disk_space(),
70                 240LL * (100000000 / 8) / 24 +  // video
71                 240LL * 48000 * 6 * 3 / 24 +    // audio
72                 65536,                          // extra
73                 16
74                 );
75 }