test/ffmpeg_encoder_test.cc

   1 /*
   2     Copyright (C) 2017 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/ffmpeg_encoder.h"
  22 #include "lib/film.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"
  30 #include "test.h"
  31 #include <boost/test/unit_test.hpp>
  32
  33 using boost::shared_ptr;
  34
  35 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mov)
  36 {
  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);
  42
  43         film->examine_and_add_content (c);
  44         BOOST_REQUIRE (!wait_for_jobs ());
  45
  46         shared_ptr<Job> job (new TranscodeJob (film));
  47         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test.mov", FFmpegEncoder::FORMAT_PRORES, false);
  48         encoder.go ();
  49 }
  50
  51 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mp4)
  52 {
  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);
  58
  59         film->examine_and_add_content (c);
  60         BOOST_REQUIRE (!wait_for_jobs ());
  61
  62         shared_ptr<Job> job (new TranscodeJob (film));
  63         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test.mp4", FFmpegEncoder::FORMAT_H264, false);
  64         encoder.go ();
  65 }
  66
  67 /** Simplest possible export subtitle case: just the subtitles */
  68 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_h264_1)
  69 {
  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);
  74
  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_shadow (true);
  80         s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
  81
  82         shared_ptr<Job> job (new TranscodeJob (film));
  83         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_h264_1.mp4", FFmpegEncoder::FORMAT_H264, false);
  84         encoder.go ();
  85 }
  86
  87 /** Slightly more complicated example with longer subs and a video to overlay */
  88 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_h264_2)
  89 {
  90         shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_h264_2");
  91         film->set_name ("ffmpeg_encoder_test_subs_h264_2");
  92         film->set_container (Ratio::from_id ("185"));
  93         film->set_audio_channels (6);
  94
  95         shared_ptr<FFmpegContent> c (new FFmpegContent (film, "test/data/test.mp4"));
  96         film->examine_and_add_content (c);
  97         BOOST_REQUIRE (!wait_for_jobs ());
  98
  99         shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip.srt"));
 100         film->examine_and_add_content (s);
 101         BOOST_REQUIRE (!wait_for_jobs ());
 102         s->subtitle->set_colour (dcp::Colour (255, 255, 0));
 103         s->subtitle->set_shadow (true);
 104         s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
 105
 106         shared_ptr<Job> job (new TranscodeJob (film));
 107         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_h264_2.mp4", FFmpegEncoder::FORMAT_H264, false);
 108         encoder.go ();
 109 }
 110
 111 /** Simplest possible export subtitle case: just the subtitles */
 112 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_prores_1)
 113 {
 114         shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_prores_1");
 115         film->set_name ("ffmpeg_encoder_test_subs_prores_1");
 116         film->set_container (Ratio::from_id ("185"));
 117         film->set_audio_channels (6);
 118
 119         shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip2.srt"));
 120         film->examine_and_add_content (s);
 121         BOOST_REQUIRE (!wait_for_jobs ());
 122         s->subtitle->set_colour (dcp::Colour (255, 255, 0));
 123         s->subtitle->set_shadow (true);
 124         s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
 125
 126         shared_ptr<Job> job (new TranscodeJob (film));
 127         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_prores_1.mov", FFmpegEncoder::FORMAT_PRORES, false);
 128         encoder.go ();
 129 }
 130
 131 /** Slightly more complicated example with longer subs and a video to overlay */
 132 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_prores_2)
 133 {
 134         shared_ptr<Film> film = new_test_film ("ffmpeg_encoder_test_subs_prores_2");
 135         film->set_name ("ffmpeg_encoder_test_subs_prores_2");
 136         film->set_container (Ratio::from_id ("185"));
 137         film->set_audio_channels (6);
 138
 139         shared_ptr<FFmpegContent> c (new FFmpegContent (film, "test/data/test.mp4"));
 140         film->examine_and_add_content (c);
 141         BOOST_REQUIRE (!wait_for_jobs ());
 142
 143         shared_ptr<TextSubtitleContent> s (new TextSubtitleContent (film, "test/data/subrip.srt"));
 144         film->examine_and_add_content (s);
 145         BOOST_REQUIRE (!wait_for_jobs ());
 146         s->subtitle->set_colour (dcp::Colour (255, 255, 0));
 147         s->subtitle->set_shadow (true);
 148         s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255));
 149
 150         shared_ptr<Job> job (new TranscodeJob (film));
 151         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_prores_2.mov", FFmpegEncoder::FORMAT_PRORES, false);
 152         encoder.go ();
 153 }
 154
 155 /** Test a bug with export of scope-in-flat DCP content */
 156 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_bug_test_scope)
 157 {
 158         shared_ptr<Film> film = new_test_film2("ffmpeg_encoder_bug_test_scope");
 159         film->examine_and_add_content(shared_ptr<DCPContent>(new DCPContent(film, "test/data/scope_dcp")));
 160         BOOST_REQUIRE(!wait_for_jobs());
 161
 162         film->set_container(Ratio::from_id("185"));
 163
 164         shared_ptr<Job> job(new TranscodeJob(film));
 165         FFmpegEncoder encoder(film, job, "build/test/ffmpeg_encoder_bug_test_scope.mp4", FFmpegEncoder::FORMAT_H264, false);
 166         encoder.go();
 167 }
 168
 169 BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mixdown)
 170 {
 171         shared_ptr<Film> film = new_test_film ("ffmpeg_transcoder_basic_test_mixdown");
 172         film->set_name ("ffmpeg_transcoder_basic_test");
 173         film->set_container (Ratio::from_id ("185"));
 174         film->set_audio_channels (6);
 175
 176         shared_ptr<FFmpegContent> L (new FFmpegContent (film, "test/data/L.wav"));
 177         film->examine_and_add_content (L);
 178         shared_ptr<FFmpegContent> R (new FFmpegContent (film, "test/data/R.wav"));
 179         film->examine_and_add_content (R);
 180         shared_ptr<FFmpegContent> C (new FFmpegContent (film, "test/data/C.wav"));
 181         film->examine_and_add_content (C);
 182         shared_ptr<FFmpegContent> Ls (new FFmpegContent (film, "test/data/Ls.wav"));
 183         film->examine_and_add_content (Ls);
 184         shared_ptr<FFmpegContent> Rs (new FFmpegContent (film, "test/data/Rs.wav"));
 185         film->examine_and_add_content (Rs);
 186         shared_ptr<FFmpegContent> Lfe (new FFmpegContent (film, "test/data/Lfe.wav"));
 187         film->examine_and_add_content (Lfe);
 188         BOOST_REQUIRE (!wait_for_jobs ());
 189
 190         AudioMapping map (1, MAX_DCP_AUDIO_CHANNELS);
 191
 192         L->set_position (DCPTime::from_seconds (0));
 193         map.make_zero ();
 194         map.set (0, 0, 1);
 195         L->audio->set_mapping (map);
 196         R->set_position (DCPTime::from_seconds (1));
 197         map.make_zero ();
 198         map.set (0, 1, 1);
 199         R->audio->set_mapping (map);
 200         C->set_position (DCPTime::from_seconds (2));
 201         map.make_zero ();
 202         map.set (0, 2, 1);
 203         C->audio->set_mapping (map);
 204         Lfe->set_position (DCPTime::from_seconds (3));
 205         map.make_zero ();
 206         map.set (0, 3, 1);
 207         Lfe->audio->set_mapping (map);
 208         Ls->set_position (DCPTime::from_seconds (4));
 209         map.make_zero ();
 210         map.set (0, 4, 1);
 211         Ls->audio->set_mapping (map);
 212         Rs->set_position (DCPTime::from_seconds (5));
 213         map.make_zero ();
 214         map.set (0, 5, 1);
 215         Rs->audio->set_mapping (map);
 216
 217         shared_ptr<Job> job (new TranscodeJob (film));
 218         FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test_mixdown.mp4", FFmpegEncoder::FORMAT_H264, true);
 219         encoder.go ();
 220
 221         /* Skip the first video packet when checking as it contains x264 options which can vary between machines
 222            (e.g. number of threads used for encoding).
 223         */
 224         check_ffmpeg ("build/test/ffmpeg_encoder_basic_test_mixdown.mp4", "test/data/ffmpeg_encoder_basic_test_mixdown.mp4");
 225 }