More enum class additions.
[dcpomatic.git] / test / subtitle_reel_test.cc
1 /*
2     Copyright (C) 2019-2020 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/image_content.h"
24 #include "lib/dcp_subtitle_content.h"
25 #include "lib/text_content.h"
26 #include "lib/video_content.h"
27 #include "test.h"
28 #include <dcp/dcp.h>
29 #include <dcp/cpl.h>
30 #include <dcp/reel.h>
31 #include <dcp/interop_subtitle_asset.h>
32 #include <dcp/reel_closed_caption_asset.h>
33 #include <dcp/reel_subtitle_asset.h>
34 #include <boost/test/unit_test.hpp>
35
36
37 using std::list;
38 using std::string;
39 using std::shared_ptr;
40 using std::make_shared;
41 using boost::optional;
42
43
44 /* Check that timings are done correctly for multi-reel DCPs with PNG subs */
45 BOOST_AUTO_TEST_CASE (subtitle_reel_test)
46 {
47         auto film = new_test_film2 ("subtitle_reel_test");
48         film->set_interop (true);
49         auto red_a = make_shared<ImageContent>("test/data/flat_red.png");
50         auto red_b = make_shared<ImageContent>("test/data/flat_red.png");
51         auto sub_a = make_shared<DCPSubtitleContent>("test/data/png_subs/subs.xml");
52         auto sub_b = make_shared<DCPSubtitleContent>("test/data/png_subs/subs.xml");
53
54         film->examine_and_add_content (red_a);
55         film->examine_and_add_content (red_b);
56         film->examine_and_add_content (sub_a);
57         film->examine_and_add_content (sub_b);
58
59         BOOST_REQUIRE (!wait_for_jobs());
60
61         red_a->set_position (film, dcpomatic::DCPTime());
62         red_a->video->set_length (240);
63         sub_a->set_position (film, dcpomatic::DCPTime());
64         red_b->set_position (film, dcpomatic::DCPTime::from_seconds(10));
65         red_b->video->set_length (240);
66         sub_b->set_position (film, dcpomatic::DCPTime::from_seconds(10));
67
68         film->set_reel_type (ReelType::BY_VIDEO_CONTENT);
69
70         film->make_dcp ();
71         BOOST_REQUIRE (!wait_for_jobs());
72
73         dcp::DCP dcp ("build/test/subtitle_reel_test/" + film->dcp_name());
74         dcp.read ();
75         BOOST_REQUIRE_EQUAL (dcp.cpls().size(), 1U);
76         shared_ptr<dcp::CPL> cpl = dcp.cpls().front();
77
78         auto reels = cpl->reels ();
79         BOOST_REQUIRE_EQUAL (reels.size(), 2U);
80         auto i = reels.begin ();
81         BOOST_REQUIRE ((*i)->main_subtitle());
82         BOOST_REQUIRE ((*i)->main_subtitle()->asset());
83         auto A = std::dynamic_pointer_cast<dcp::InteropSubtitleAsset>((*i)->main_subtitle()->asset());
84         BOOST_REQUIRE (A);
85         ++i;
86         BOOST_REQUIRE ((*i)->main_subtitle());
87         BOOST_REQUIRE ((*i)->main_subtitle()->asset());
88         auto B = std::dynamic_pointer_cast<dcp::InteropSubtitleAsset>((*i)->main_subtitle()->asset());
89         BOOST_REQUIRE (B);
90
91         BOOST_REQUIRE_EQUAL (A->subtitles().size(), 1U);
92         BOOST_REQUIRE_EQUAL (B->subtitles().size(), 1U);
93
94         /* These times should be the same as they are should be offset from the start of the reel */
95         BOOST_CHECK (A->subtitles().front()->in() == B->subtitles().front()->in());
96 }
97
98
99
100 /** Check that with a SMPTE DCP if we have subtitles in one reel, all reels have a
101  *  SubtitleAsset (even if it's empty); SMPTE Bv2.1 section 8.3.1.
102  */
103 BOOST_AUTO_TEST_CASE (subtitle_in_all_reels_test)
104 {
105         auto film = new_test_film2 ("subtitle_in_all_reels_test");
106         film->set_interop (false);
107         film->set_sequence (false);
108         film->set_reel_type (ReelType::BY_VIDEO_CONTENT);
109         for (int i = 0; i < 3; ++i) {
110                 auto video = content_factory("test/data/flat_red.png").front();
111                 film->examine_and_add_content (video);
112                 BOOST_REQUIRE (!wait_for_jobs());
113                 video->video->set_length (15 * 24);
114                 video->set_position (film, dcpomatic::DCPTime::from_seconds(15 * i));
115         }
116         auto subs = content_factory("test/data/15s.srt").front();
117         film->examine_and_add_content (subs);
118         BOOST_REQUIRE (!wait_for_jobs());
119         film->make_dcp ();
120         BOOST_REQUIRE (!wait_for_jobs());
121
122         dcp::DCP dcp ("build/test/subtitle_in_all_reels_test/" + film->dcp_name());
123         dcp.read ();
124         BOOST_REQUIRE_EQUAL (dcp.cpls().size(), 1U);
125         auto cpl = dcp.cpls()[0];
126         BOOST_REQUIRE_EQUAL (cpl->reels().size(), 3U);
127
128         for (auto i: cpl->reels()) {
129                 BOOST_CHECK (i->main_subtitle());
130         }
131 }
132
133
134 /** Check that with a SMPTE DCP if we have closed captions in one reel, all reels have a
135  *  ClosedCaptionAssets for the same set of tracks (even if they are empty); SMPTE Bv2.1 section 8.3.1.
136  */
137 BOOST_AUTO_TEST_CASE (closed_captions_in_all_reels_test)
138 {
139         auto film = new_test_film2 ("closed_captions_in_all_reels_test");
140         film->set_interop (false);
141         film->set_sequence (false);
142         film->set_reel_type (ReelType::BY_VIDEO_CONTENT);
143
144         for (int i = 0; i < 3; ++i) {
145                 auto video = content_factory("test/data/flat_red.png").front();
146                 film->examine_and_add_content (video);
147                 BOOST_REQUIRE (!wait_for_jobs());
148                 video->video->set_length (15 * 24);
149                 video->set_position (film, dcpomatic::DCPTime::from_seconds(15 * i));
150         }
151
152         auto ccap1 = content_factory("test/data/15s.srt").front();
153         film->examine_and_add_content (ccap1);
154         BOOST_REQUIRE (!wait_for_jobs());
155         ccap1->text.front()->set_type (TextType::CLOSED_CAPTION);
156         ccap1->text.front()->set_dcp_track (DCPTextTrack("Test", "de-DE"));
157
158         auto ccap2 = content_factory("test/data/15s.srt").front();
159         film->examine_and_add_content (ccap2);
160         BOOST_REQUIRE (!wait_for_jobs());
161         ccap2->text.front()->set_type (TextType::CLOSED_CAPTION);
162         ccap2->text.front()->set_dcp_track (DCPTextTrack("Other", "en-GB"));
163
164         film->make_dcp ();
165         BOOST_REQUIRE (!wait_for_jobs());
166
167         dcp::DCP dcp ("build/test/closed_captions_in_all_reels_test/" + film->dcp_name());
168         dcp.read ();
169         BOOST_REQUIRE_EQUAL (dcp.cpls().size(), 1U);
170         auto cpl = dcp.cpls().front();
171         BOOST_REQUIRE_EQUAL (cpl->reels().size(), 3U);
172
173         for (auto i: cpl->reels()) {
174                 BOOST_REQUIRE_EQUAL (i->closed_captions().size(), 2U);
175                 auto first = i->closed_captions().front()->language();
176                 auto second = i->closed_captions().back()->language();
177                 BOOST_REQUIRE (first);
178                 BOOST_REQUIRE (second);
179                 BOOST_CHECK (
180                         (*first == "en-GB" && *second == "de-DE") ||
181                         (*first == "de-DE" && *second == "en-GB")
182                         );
183         }
184 }