X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Fffmpeg_encoder_test.cc;h=9bce836c3fb4d3c814912ddedc9c888683b256fd;hb=981ffdc9db4a84c614b3186a0fb27136fe6feb37;hp=c96e19f9294d4e2f2a07b200dd5438917ae1e4a2;hpb=6b9c53001a63342a86899c60779a96edf96f369a;p=dcpomatic.git diff --git a/test/ffmpeg_encoder_test.cc b/test/ffmpeg_encoder_test.cc index c96e19f92..9bce836c3 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,28 +18,382 @@ */ + #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/string_text_file_content.h" #include "lib/ratio.h" #include "lib/transcode_job.h" +#include "lib/dcp_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) +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) +{ + 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->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); +} + + +/** Still image -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test5) +{ + 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 ()); + + 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 (); +} + + +/** Subs -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test6) +{ + 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); + + auto s = make_shared("test/data/subrip2.srt"); + film->examine_and_add_content (s); + BOOST_REQUIRE (!wait_for_jobs ()); + 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(); + + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test6.mov", ExportFormat::PRORES, false, false, false, 23); + encoder.go (); +} + + +/** Video + subs -> Prores */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_prores_test7) +{ + 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); + + auto c = make_shared("test/data/test.mp4"); + film->examine_and_add_content (c); + BOOST_REQUIRE (!wait_for_jobs ()); + + auto s = make_shared("test/data/subrip.srt"); + film->examine_and_add_content (s); + BOOST_REQUIRE (!wait_for_jobs ()); + 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)); + + auto job = make_shared (film); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test7.mov", ExportFormat::PRORES, false, false, false, 23); + encoder.go (); +} + + +/** Red / green / blue MP4 -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test1) +{ + 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); + + auto s = make_shared("test/data/subrip2.srt"); + film->examine_and_add_content (s); + BOOST_REQUIRE (!wait_for_jobs ()); + 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(); + + 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 (); +} + + +/** Video + subs -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test3) +{ + 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); + + auto c = make_shared("test/data/test.mp4"); + film->examine_and_add_content (c); + BOOST_REQUIRE (!wait_for_jobs()); + + auto s = make_shared("test/data/subrip.srt"); + film->examine_and_add_content (s); + BOOST_REQUIRE (!wait_for_jobs ()); + 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(); + + 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 (); +} + + +/** Scope-in-flat DCP -> H264 */ +BOOST_AUTO_TEST_CASE (ffmpeg_encoder_h264_test4) +{ + 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")); + + 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(); +} + + +/** 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"); - film->set_name ("ffmpeg_transcoder_basic_test"); - shared_ptr c (new FFmpegContent (film, "test/data/test.mp4")); + 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); + auto L = make_shared("test/data/L.wav"); + film->examine_and_add_content (L); + auto R = make_shared("test/data/R.wav"); + film->examine_and_add_content (R); + auto C = make_shared("test/data/C.wav"); + film->examine_and_add_content (C); + auto Ls = make_shared("test/data/Ls.wav"); + film->examine_and_add_content (Ls); + auto Rs = make_shared("test/data/Rs.wav"); + film->examine_and_add_content (Rs); + 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 (film, DCPTime::from_seconds(0)); + map.make_zero (); + map.set (0, 0, 1); + L->audio->set_mapping (map); + R->set_position (film, DCPTime::from_seconds(1)); + map.make_zero (); + map.set (0, 1, 1); + R->audio->set_mapping (map); + C->set_position (film, DCPTime::from_seconds(2)); + map.make_zero (); + map.set (0, 2, 1); + C->audio->set_mapping (map); + Lfe->set_position (film, DCPTime::from_seconds(3)); + map.make_zero (); + map.set (0, 3, 1); + Lfe->audio->set_mapping (map); + Ls->set_position (film, DCPTime::from_seconds(4)); + map.make_zero (); + map.set (0, 4, 1); + Ls->audio->set_mapping (map); + 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() / "XMenDarkPhoenix_TLR-12-3D_F_DE-XX_DE_51_2K_TCFG_20190227_TM_IOP-3D_OV"); + 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); - wait_for_jobs (); + 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/test.mov", FFmpegEncoder::FORMAT_PRORES); + FFmpegEncoder encoder (film, job, "build/test/ffmpeg_encoder_prores_test9.mov", ExportFormat::H264_AAC, false, false, false, 23); encoder.go (); }