2 Copyright (C) 2017-2018 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
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.
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.
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/>.
21 #include "lib/ffmpeg_encoder.h"
23 #include "lib/ffmpeg_content.h"
24 #include "lib/audio_content.h"
25 #include "lib/text_subtitle_content.h"
26 #include "lib/ratio.h"
27 #include "lib/transcode_job.h"
28 #include "lib/dcp_content.h"
29 #include "lib/subtitle_content.h"
31 #include <boost/test/unit_test.hpp>
33 using boost::shared_ptr;
35 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mov)
37 shared_ptr<Film> film = new_test_film ("ffmpeg_transcoder_basic_test_mov");
38 film->set_name ("ffmpeg_transcoder_basic_test");
39 shared_ptr<FFmpegContent> c (new FFmpegContent (film, "test/data/test.mp4"));
40 film->set_container (Ratio::from_id ("185"));
41 film->set_audio_channels (6);
43 film->examine_and_add_content (c);
44 BOOST_REQUIRE (!wait_for_jobs ());
46 shared_ptr<Job> job (new TranscodeJob (film));
47 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test.mov", FFmpegEncoder::FORMAT_PRORES, false);
51 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mp4)
53 shared_ptr<Film> film = new_test_film ("ffmpeg_transcoder_basic_test_mp4");
54 film->set_name ("ffmpeg_transcoder_basic_test");
55 shared_ptr<FFmpegContent> c (new FFmpegContent (film, "test/data/test.mp4"));
56 film->set_container (Ratio::from_id ("185"));
57 film->set_audio_channels (6);
59 film->examine_and_add_content (c);
60 BOOST_REQUIRE (!wait_for_jobs ());
62 shared_ptr<Job> job (new TranscodeJob (film));
63 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test.mp4", FFmpegEncoder::FORMAT_H264, false);
67 /** Simplest possible export subtitle case: just the subtitles */
68 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_h264_1)
70 shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_h264_1");
71 film->set_name ("ffmpeg_encoder_test_subs_h264_1");
72 film->set_container (Ratio::from_id ("185"));
73 film->set_audio_channels (6);
75 shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip2.srt"));
76 film->examine_and_add_content (s);
77 BOOST_REQUIRE (!wait_for_jobs ());
78 s->subtitle->set_colour (dcp::Colour (255, 255, 0));
79 s->subtitle->set_effect (dcp::SHADOW);
80 s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
81 film->write_metadata();
83 shared_ptr<Job> job (new TranscodeJob (film));
84 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_h264_1.mp4", FFmpegEncoder::FORMAT_H264, false);
88 /** Slightly more complicated example with longer subs and a video to overlay */
89 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_h264_2)
91 shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_h264_2");
92 film->set_name ("ffmpeg_encoder_test_subs_h264_2");
93 film->set_container (Ratio::from_id ("185"));
94 film->set_audio_channels (6);
96 shared_ptr<FFmpegContent> c (new FFmpegContent (film, "test/data/test.mp4"));
97 film->examine_and_add_content (c);
98 BOOST_REQUIRE (!wait_for_jobs ());
100 shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip.srt"));
101 film->examine_and_add_content (s);
102 BOOST_REQUIRE (!wait_for_jobs ());
103 s->subtitle->set_colour (dcp::Colour (255, 255, 0));
104 s->subtitle->set_effect (dcp::SHADOW);
105 s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
106 film->write_metadata();
108 shared_ptr<Job> job (new TranscodeJob (film));
109 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_h264_2.mp4", FFmpegEncoder::FORMAT_H264, false);
113 /** Simplest possible export subtitle case: just the subtitles */
114 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_prores_1)
116 shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_prores_1");
117 film->set_name ("ffmpeg_encoder_test_subs_prores_1");
118 film->set_container (Ratio::from_id ("185"));
119 film->set_audio_channels (6);
121 shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip2.srt"));
122 film->examine_and_add_content (s);
123 BOOST_REQUIRE (!wait_for_jobs ());
124 s->subtitle->set_colour (dcp::Colour (255, 255, 0));
125 s->subtitle->set_effect (dcp::SHADOW);
126 s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
127 film->write_metadata();
129 shared_ptr<Job> job (new TranscodeJob (film));
130 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_prores_1.mov", FFmpegEncoder::FORMAT_PRORES, false);
134 /** Slightly more complicated example with longer subs and a video to overlay */
135 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_prores_2)
137 shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_prores_2");
138 film->set_name ("ffmpeg_encoder_test_subs_prores_2");
139 film->set_container (Ratio::from_id ("185"));
140 film->set_audio_channels (6);
142 shared_ptr<FFmpegContent> c (new FFmpegContent (film, "test/data/test.mp4"));
143 film->examine_and_add_content (c);
144 BOOST_REQUIRE (!wait_for_jobs ());
146 shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip.srt"));
147 film->examine_and_add_content (s);
148 BOOST_REQUIRE (!wait_for_jobs ());
149 s->subtitle->set_colour (dcp::Colour (255, 255, 0));
150 s->subtitle->set_effect (dcp::SHADOW);
151 s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
153 shared_ptr<Job> job (new TranscodeJob (film));
154 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_prores_2.mov", FFmpegEncoder::FORMAT_PRORES, false);
158 /** Test a bug with export of scope-in-flat DCP content */
159 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_bug_test_scope)
161 shared_ptr<Film> film = new_test_film2("ffmpeg_encoder_bug_test_scope");
162 film->examine_and_add_content(shared_ptr<DCPContent>(new DCPContent(film, "test/data/scope_dcp")));
163 BOOST_REQUIRE(!wait_for_jobs());
165 film->set_container(Ratio::from_id("185"));
167 shared_ptr<Job> job(new TranscodeJob(film));
168 FFmpegEncoder encoder(film, job, "build/test/ffmpeg_encoder_bug_test_scope.mp4", FFmpegEncoder::FORMAT_H264, false);
172 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mixdown)
174 shared_ptr<Film> film = new_test_film ("ffmpeg_transcoder_basic_test_mixdown");
175 film->set_name ("ffmpeg_transcoder_basic_test");
176 film->set_container (Ratio::from_id ("185"));
177 film->set_audio_channels (6);
179 shared_ptr<FFmpegContent> L (new FFmpegContent (film, "test/data/L.wav"));
180 film->examine_and_add_content (L);
181 shared_ptr<FFmpegContent> R (new FFmpegContent (film, "test/data/R.wav"));
182 film->examine_and_add_content (R);
183 shared_ptr<FFmpegContent> C (new FFmpegContent (film, "test/data/C.wav"));
184 film->examine_and_add_content (C);
185 shared_ptr<FFmpegContent> Ls (new FFmpegContent (film, "test/data/Ls.wav"));
186 film->examine_and_add_content (Ls);
187 shared_ptr<FFmpegContent> Rs (new FFmpegContent (film, "test/data/Rs.wav"));
188 film->examine_and_add_content (Rs);
189 shared_ptr<FFmpegContent> Lfe (new FFmpegContent (film, "test/data/Lfe.wav"));
190 film->examine_and_add_content (Lfe);
191 BOOST_REQUIRE (!wait_for_jobs ());
193 AudioMapping map (1, MAX_DCP_AUDIO_CHANNELS);
195 L->set_position (DCPTime::from_seconds (0));
198 L->audio->set_mapping (map);
199 R->set_position (DCPTime::from_seconds (1));
202 R->audio->set_mapping (map);
203 C->set_position (DCPTime::from_seconds (2));
206 C->audio->set_mapping (map);
207 Lfe->set_position (DCPTime::from_seconds (3));
210 Lfe->audio->set_mapping (map);
211 Ls->set_position (DCPTime::from_seconds (4));
214 Ls->audio->set_mapping (map);
215 Rs->set_position (DCPTime::from_seconds (5));
218 Rs->audio->set_mapping (map);
220 shared_ptr<Job> job (new TranscodeJob (film));
221 FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test_mixdown.mp4", FFmpegEncoder::FORMAT_H264, true);
224 check_ffmpeg ("build/test/ffmpeg_encoder_basic_test_mixdown.mp4", "test/data/ffmpeg_encoder_basic_test_mixdown.mp4", 1);