X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=test%2Fffmpeg_encoder_test.cc;h=c15973faed4ae9ef5b437da39fd825007bbdcd37;hp=e6cff9b8fdfb57df583e57e3d7342a56ff8f9bb5;hb=9dcfb0c57bf107626dcfc94c37ad14acc9045e3b;hpb=fde6445d7e08b74366955246ddd4629bfd2b5914 diff --git a/test/ffmpeg_encoder_test.cc b/test/ffmpeg_encoder_test.cc index e6cff9b8f..c15973fae 100644 --- a/test/ffmpeg_encoder_test.cc +++ b/test/ffmpeg_encoder_test.cc @@ -1,5 +1,5 @@ /* - Copyright (C) 2017 Carl Hetherington + Copyright (C) 2017-2021 Carl Hetherington This file is part of DCP-o-matic. @@ -18,211 +18,412 @@ */ + #include "lib/ffmpeg_encoder.h" #include "lib/film.h" #include "lib/ffmpeg_content.h" +#include "lib/image_content.h" +#include "lib/video_content.h" #include "lib/audio_content.h" -#include "lib/text_subtitle_content.h" +#include "lib/string_text_file_content.h" #include "lib/ratio.h" #include "lib/transcode_job.h" #include "lib/dcp_content.h" -#include "lib/subtitle_content.h" +#include "lib/text_content.h" +#include "lib/compose.hpp" +#include "lib/content_factory.h" #include "test.h" #include -using boost::shared_ptr; -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mov) +using std::string; +using std::shared_ptr; +using std::make_shared; +using boost::optional; +using namespace dcpomatic; + + +static void +ffmpeg_content_test (int number, boost::filesystem::path content, ExportFormat format) { - shared_ptr film = new_test_film ("ffmpeg_transcoder_basic_test_mov"); - film->set_name ("ffmpeg_transcoder_basic_test"); - shared_ptr c (new FFmpegContent (film, "test/data/test.mp4")); - film->set_container (Ratio::from_id ("185")); + Cleanup cl; + + string name = "ffmpeg_encoder_"; + string extension; + switch (format) { + case ExportFormat::H264_AAC: + name += "h264"; + extension = "mp4"; + break; + case ExportFormat::PRORES: + name += "prores"; + extension = "mov"; + break; + case ExportFormat::H264_PCM: + case ExportFormat::SUBTITLES_DCP: + BOOST_REQUIRE (false); + } + + name = String::compose("%1_test%2", name, number); + + auto c = make_shared(content); + shared_ptr film = new_test_film2 (name, {c}, &cl); + film->set_name (name); film->set_audio_channels (6); - film->examine_and_add_content (c); - BOOST_REQUIRE (!wait_for_jobs ()); - - shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test.mov", FFmpegEncoder::FORMAT_PRORES, false); + film->write_metadata (); + auto job = make_shared(film); + auto file = boost::filesystem::path("build") / "test" / String::compose("%1.%2", name, extension); + cl.add (file); + FFmpegEncoder encoder (film, job, file, format, false, false, false, 23); encoder.go (); + + cl.run (); +} + + +/** Red / green / blue MP4 -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test1) +{ + ffmpeg_content_test (1, "test/data/test.mp4", ExportFormat::PRORES); +} + + +/** Dolby Aurora trailer VOB -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test2) +{ + ffmpeg_content_test (2, TestPaths::private_data() / "dolby_aurora.vob", ExportFormat::PRORES); +} + + +/** Sintel trailer -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test3) +{ + ffmpeg_content_test (3, TestPaths::private_data() / "Sintel_Trailer1.480p.DivX_Plus_HD.mkv", ExportFormat::PRORES); +} + + +/** Big Buck Bunny trailer -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test4) +{ + ffmpeg_content_test (4, TestPaths::private_data() / "big_buck_bunny_trailer_480p.mov", ExportFormat::PRORES); } -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mp4) + +/** Still image -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test5) { - shared_ptr film = new_test_film ("ffmpeg_transcoder_basic_test_mp4"); - film->set_name ("ffmpeg_transcoder_basic_test"); - shared_ptr c (new FFmpegContent (film, "test/data/test.mp4")); + auto film = new_test_film ("ffmpeg_encoder_prores_test5"); + film->set_name ("ffmpeg_encoder_prores_test5"); + auto c = make_shared(TestPaths::private_data() / "bbc405.png"); film->set_container (Ratio::from_id ("185")); film->set_audio_channels (6); film->examine_and_add_content (c); BOOST_REQUIRE (!wait_for_jobs ()); - shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test.mp4", FFmpegEncoder::FORMAT_H264, false); + c->video->set_length (240); + + film->write_metadata (); + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test5.mov", ExportFormat::PRORES, false, false, false, 23); encoder.go (); } -/** Simplest possible export subtitle case: just the subtitles */ -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_h264_1) + +/** Subs -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test6) { - shared_ptr film = new_test_film ("ffmpeg_encoder_test_subs_h264_1"); - film->set_name ("ffmpeg_encoder_test_subs_h264_1"); + auto film = new_test_film ("ffmpeg_encoder_prores_test6"); + film->set_name ("ffmpeg_encoder_prores_test6"); film->set_container (Ratio::from_id ("185")); film->set_audio_channels (6); - shared_ptr s (new TextSubtitleContent (film, "test/data/subrip2.srt")); + auto s = make_shared("test/data/subrip2.srt"); film->examine_and_add_content (s); BOOST_REQUIRE (!wait_for_jobs ()); - s->subtitle->set_colour (dcp::Colour (255, 255, 0)); - s->subtitle->set_shadow (true); - s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255)); + s->only_text()->set_colour (dcp::Colour (255, 255, 0)); + s->only_text()->set_effect (dcp::Effect::SHADOW); + s->only_text()->set_effect_colour (dcp::Colour (0, 255, 255)); film->write_metadata(); - shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_h264_1.mp4", FFmpegEncoder::FORMAT_H264, false); + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test6.mov", ExportFormat::PRORES, false, false, false, 23); encoder.go (); } -/** Slightly more complicated example with longer subs and a video to overlay */ -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_h264_2) + +/** Video + subs -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test7) { - shared_ptr film = new_test_film ("ffmpeg_encoder_test_subs_h264_2"); - film->set_name ("ffmpeg_encoder_test_subs_h264_2"); + auto film = new_test_film ("ffmpeg_encoder_prores_test7"); + film->set_name ("ffmpeg_encoder_prores_test7"); film->set_container (Ratio::from_id ("185")); film->set_audio_channels (6); - shared_ptr c (new FFmpegContent (film, "test/data/test.mp4")); + auto c = make_shared("test/data/test.mp4"); film->examine_and_add_content (c); BOOST_REQUIRE (!wait_for_jobs ()); - shared_ptr s (new TextSubtitleContent (film, "test/data/subrip.srt")); + auto s = make_shared("test/data/subrip.srt"); film->examine_and_add_content (s); BOOST_REQUIRE (!wait_for_jobs ()); - s->subtitle->set_colour (dcp::Colour (255, 255, 0)); - s->subtitle->set_shadow (true); - s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255)); - film->write_metadata(); + s->only_text()->set_colour (dcp::Colour (255, 255, 0)); + s->only_text()->set_effect (dcp::Effect::SHADOW); + s->only_text()->set_effect_colour (dcp::Colour (0, 255, 255)); - shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_h264_2.mp4", FFmpegEncoder::FORMAT_H264, false); + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test7.mov", ExportFormat::PRORES, false, false, false, 23); encoder.go (); } -/** Simplest possible export subtitle case: just the subtitles */ -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_prores_1) + +/** Red / green / blue MP4 -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test1) { - shared_ptr film = new_test_film ("ffmpeg_encoder_test_subs_prores_1"); - film->set_name ("ffmpeg_encoder_test_subs_prores_1"); + ffmpeg_content_test(1, "test/data/test.mp4", ExportFormat::H264_AAC); +} + + +/** Just subtitles -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test2) +{ + auto film = new_test_film ("ffmpeg_encoder_h264_test2"); + film->set_name ("ffmpeg_encoder_h264_test2"); film->set_container (Ratio::from_id ("185")); film->set_audio_channels (6); - shared_ptr s (new TextSubtitleContent (film, "test/data/subrip2.srt")); + auto s = make_shared("test/data/subrip2.srt"); film->examine_and_add_content (s); BOOST_REQUIRE (!wait_for_jobs ()); - s->subtitle->set_colour (dcp::Colour (255, 255, 0)); - s->subtitle->set_shadow (true); - s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255)); + s->only_text()->set_colour (dcp::Colour (255, 255, 0)); + s->only_text()->set_effect (dcp::Effect::SHADOW); + s->only_text()->set_effect_colour (dcp::Colour (0, 255, 255)); film->write_metadata(); - shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_prores_1.mov", FFmpegEncoder::FORMAT_PRORES, false); + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_h264_test2.mp4", ExportFormat::H264_AAC, false, false, false, 23); encoder.go (); } -/** Slightly more complicated example with longer subs and a video to overlay */ -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_test_subs_prores_2) + +/** Video + subs -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test3) { - shared_ptr film = new_test_film ("ffmpeg_encoder_test_subs_prores_2"); - film->set_name ("ffmpeg_encoder_test_subs_prores_2"); + auto film = new_test_film ("ffmpeg_encoder_h264_test3"); + film->set_name ("ffmpeg_encoder_h264_test3"); film->set_container (Ratio::from_id ("185")); film->set_audio_channels (6); - shared_ptr c (new FFmpegContent (film, "test/data/test.mp4")); + auto c = make_shared("test/data/test.mp4"); film->examine_and_add_content (c); - BOOST_REQUIRE (!wait_for_jobs ()); + BOOST_REQUIRE (!wait_for_jobs()); - shared_ptr s (new TextSubtitleContent (film, "test/data/subrip.srt")); + auto s = make_shared("test/data/subrip.srt"); film->examine_and_add_content (s); BOOST_REQUIRE (!wait_for_jobs ()); - s->subtitle->set_colour (dcp::Colour (255, 255, 0)); - s->subtitle->set_shadow (true); - s->subtitle->set_effect_colour (dcp::Colour (0, 255, 255)); + s->only_text()->set_colour (dcp::Colour (255, 255, 0)); + s->only_text()->set_effect (dcp::Effect::SHADOW); + s->only_text()->set_effect_colour (dcp::Colour (0, 255, 255)); + film->write_metadata(); - shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_test_subs_prores_2.mov", FFmpegEncoder::FORMAT_PRORES, false); + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_h264_test3.mp4", ExportFormat::H264_AAC, false, false, false, 23); encoder.go (); } -/** Test a bug with export of scope-in-flat DCP content */ -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_bug_test_scope) + +/** Scope-in-flat DCP -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test4) { - shared_ptr film = new_test_film2("ffmpeg_encoder_bug_test_scope"); - film->examine_and_add_content(shared_ptr(new DCPContent(film, "test/data/scope_dcp"))); + auto film = new_test_film2("ffmpeg_encoder_h264_test4", {make_shared("test/data/scope_dcp")}); BOOST_REQUIRE(!wait_for_jobs()); film->set_container(Ratio::from_id("185")); - shared_ptr job(new TranscodeJob(film)); - FFmpegEncoder encoder(film, job, "build/test/ffmpeg_encoder_bug_test_scope.mp4", FFmpegEncoder::FORMAT_H264, false); + auto job = make_shared(film); + FFmpegEncoder encoder(film, job, "build/test/ffmpeg_encoder_h264_test4.mp4", ExportFormat::H264_AAC, false, false, false, 23); encoder.go(); } -BOOST_AUTO_TEST_CASE (ffmpeg_encoder_basic_test_mixdown) + +/** Test mixdown from 5.1 to stereo */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test5) { - shared_ptr film = new_test_film ("ffmpeg_transcoder_basic_test_mixdown"); - film->set_name ("ffmpeg_transcoder_basic_test"); + auto film = new_test_film ("ffmpeg_transcoder_h264_test5"); + film->set_name ("ffmpeg_transcoder_h264_test5"); film->set_container (Ratio::from_id ("185")); film->set_audio_channels (6); - shared_ptr L (new FFmpegContent (film, "test/data/L.wav")); + auto L = make_shared("test/data/L.wav"); film->examine_and_add_content (L); - shared_ptr R (new FFmpegContent (film, "test/data/R.wav")); + auto R = make_shared("test/data/R.wav"); film->examine_and_add_content (R); - shared_ptr C (new FFmpegContent (film, "test/data/C.wav")); + auto C = make_shared("test/data/C.wav"); film->examine_and_add_content (C); - shared_ptr Ls (new FFmpegContent (film, "test/data/Ls.wav")); + auto Ls = make_shared("test/data/Ls.wav"); film->examine_and_add_content (Ls); - shared_ptr Rs (new FFmpegContent (film, "test/data/Rs.wav")); + auto Rs = make_shared("test/data/Rs.wav"); film->examine_and_add_content (Rs); - shared_ptr Lfe (new FFmpegContent (film, "test/data/Lfe.wav")); + auto Lfe = make_shared("test/data/Lfe.wav"); film->examine_and_add_content (Lfe); BOOST_REQUIRE (!wait_for_jobs ()); AudioMapping map (1, MAX_DCP_AUDIO_CHANNELS); - L->set_position (DCPTime::from_seconds (0)); + L->set_position (film, DCPTime::from_seconds(0)); map.make_zero (); map.set (0, 0, 1); L->audio->set_mapping (map); - R->set_position (DCPTime::from_seconds (1)); + R->set_position (film, DCPTime::from_seconds(1)); map.make_zero (); map.set (0, 1, 1); R->audio->set_mapping (map); - C->set_position (DCPTime::from_seconds (2)); + C->set_position (film, DCPTime::from_seconds(2)); map.make_zero (); map.set (0, 2, 1); C->audio->set_mapping (map); - Lfe->set_position (DCPTime::from_seconds (3)); + Lfe->set_position (film, DCPTime::from_seconds(3)); map.make_zero (); map.set (0, 3, 1); Lfe->audio->set_mapping (map); - Ls->set_position (DCPTime::from_seconds (4)); + Ls->set_position (film, DCPTime::from_seconds(4)); map.make_zero (); map.set (0, 4, 1); Ls->audio->set_mapping (map); - Rs->set_position (DCPTime::from_seconds (5)); + Rs->set_position (film, DCPTime::from_seconds(5)); map.make_zero (); map.set (0, 5, 1); Rs->audio->set_mapping (map); + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_h264_test5.mp4", ExportFormat::H264_AAC, true, false, false, 23); + encoder.go (); + + check_ffmpeg ("build/test/ffmpeg_encoder_h264_test5.mp4", "test/data/ffmpeg_encoder_h264_test5.mp4", 1); +} + + +/** Test export of a VF */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test6) +{ + auto film = new_test_film2 ("ffmpeg_encoder_h264_test6_ov"); + film->examine_and_add_content (make_shared(TestPaths::private_data() / "bbc405.png")); + BOOST_REQUIRE (!wait_for_jobs()); + make_and_verify_dcp (film); + + auto film2 = new_test_film2 ("ffmpeg_encoder_h264_test6_vf"); + auto ov = make_shared("build/test/ffmpeg_encoder_h264_test6_ov/" + film->dcp_name(false)); + film2->examine_and_add_content (ov); + BOOST_REQUIRE (!wait_for_jobs()); + ov->set_reference_video (true); + auto subs = content_factory("test/data/subrip.srt").front(); + film2->examine_and_add_content (subs); + BOOST_REQUIRE (!wait_for_jobs()); + for (auto i: subs->text) { + i->set_use (true); + } + + auto job = make_shared (film2); + FFmpegEncoder encoder (film2, job, "build/test/ffmpeg_encoder_h264_test6_vf.mp4", ExportFormat::H264_AAC, true, false, false, 23); + encoder.go (); +} + + +/** Test export of a 3D DCP in a 2D project */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_3d_dcp_to_h264) +{ + auto dcp = make_shared(TestPaths::private_data() / "xm"); + auto film2 = new_test_film2 ("ffmpeg_encoder_3d_dcp_to_h264_export", {dcp}); + + auto job = make_shared (film2); + FFmpegEncoder encoder (film2, job, "build/test/ffmpeg_encoder_3d_dcp_to_h264.mp4", ExportFormat::H264_AAC, true, false, false, 23); + encoder.go (); +} + + +/** Test export of a 3D DCP in a 2D project */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test7) +{ + auto L = make_shared(TestPaths::private_data() / "bbc405.png"); + auto R = make_shared(TestPaths::private_data() / "bbc405.png"); + auto film = new_test_film2 ("ffmpeg_encoder_h264_test7_data", {L, R}); + L->video->set_frame_type (VideoFrameType::THREE_D_LEFT); + L->set_position (film, DCPTime()); + R->video->set_frame_type (VideoFrameType::THREE_D_RIGHT); + R->set_position (film, DCPTime()); + film->set_three_d (true); + make_and_verify_dcp (film); + + auto dcp = make_shared(film->dir(film->dcp_name())); + auto film2 = new_test_film2 ("ffmpeg_encoder_h264_test7_export", {dcp}); + + auto job = make_shared (film2); + FFmpegEncoder encoder (film2, job, "build/test/ffmpeg_encoder_h264_test7.mp4", ExportFormat::H264_AAC, true, false, false, 23); + encoder.go (); +} + + +/** Stereo project with mixdown-to-stereo set */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test8) +{ + shared_ptr film = new_test_film2("ffmpeg_encoder_h264_test4"); + film->examine_and_add_content(shared_ptr(new DCPContent("test/data/scope_dcp"))); + BOOST_REQUIRE(!wait_for_jobs()); + film->set_audio_channels (2); + + shared_ptr job(new TranscodeJob(film)); + FFmpegEncoder encoder(film, job, "build/test/ffmpeg_encoder_h264_test8.mp4", ExportFormat::H264_AAC, true, false, false, 23); + encoder.go(); +} + + +/** 7.1/HI/VI (i.e. 12-channel) project */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test9) +{ + shared_ptr film = new_test_film ("ffmpeg_encoder_prores_test9"); + film->set_name ("ffmpeg_encoder_prores_test9"); + shared_ptr c (new ImageContent(TestPaths::private_data() / "bbc405.png")); + film->set_container (Ratio::from_id ("185")); + film->set_audio_channels (12); + + film->examine_and_add_content (c); + BOOST_REQUIRE (!wait_for_jobs ()); + + c->video->set_length (240); + + film->write_metadata (); shared_ptr job (new TranscodeJob (film)); - FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_basic_test_mixdown.mp4", FFmpegEncoder::FORMAT_H264, true); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test9.mov", ExportFormat::H264_AAC, false, false, false, 23); encoder.go (); +} - /* Skip the first video packet when checking as it contains x264 options which can vary between machines - (e.g. number of threads used for encoding). - */ - check_ffmpeg ("build/test/ffmpeg_encoder_basic_test_mixdown.mp4", "test/data/ffmpeg_encoder_basic_test_mixdown.mp4"); + +/** DCP -> Prores with crop */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_from_dcp_with_crop) +{ + auto dcp = make_shared("test/data/import_dcp_test2"); + auto film = new_test_film2 ("ffmpeg_encoder_prores_from_dcp_with_crop", { dcp }); + dcp->video->set_left_crop (32); + dcp->video->set_right_crop (32); + film->write_metadata (); + + auto job = make_shared(film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_from_dcp_with_crop.mov", ExportFormat::PRORES, false, false, false, 23); + encoder.go (); +} + + +/** DCP -> H264 with crop */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_from_dcp_with_crop) +{ + auto dcp = make_shared("test/data/import_dcp_test2"); + auto film = new_test_film2 ("ffmpeg_encoder_h264_from_dcp_with_crop", { dcp }); + dcp->video->set_left_crop (32); + dcp->video->set_right_crop (32); + film->write_metadata (); + + auto job = make_shared(film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_from_dcp_with_crop.mov", ExportFormat::H264_AAC, false, false, false, 23); + encoder.go (); }