X-Git-Url: https://git.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=test%2Fburnt_subtitle_test.cc;h=0fffe4913e52bf76e96698e1e7724eda99502c2d;hp=fc5b885780d11dc71d9b6b36b0449b3efd8b06fa;hb=6d686ea45f5cd01a0d11f92a903ac77779ad8562;hpb=80400212e939dc2f3b987cb6df57709929aa5178 diff --git a/test/burnt_subtitle_test.cc b/test/burnt_subtitle_test.cc index fc5b88578..0fffe4913 100644 --- a/test/burnt_subtitle_test.cc +++ b/test/burnt_subtitle_test.cc @@ -1,52 +1,72 @@ /* - Copyright (C) 2014 Carl Hetherington + Copyright (C) 2014-2021 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ /** @file test/burnt_subtitle_test.cc * @brief Test the burning of subtitles into the DCP. + * @ingroup feature */ -#include -#include "lib/text_subtitle_content.h" -#include "lib/dcp_subtitle_content.h" +#include "lib/plain_text_content.h" +#include "lib/dcp_text_content.h" #include "lib/film.h" #include "lib/ratio.h" #include "lib/dcp_content_type.h" +#include "lib/text_content.h" +#include "lib/dcp_content.h" +#include "lib/content_factory.h" +#include "lib/config.h" +#include "lib/log_entry.h" #include "test.h" +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include #include using std::cout; -using boost::shared_ptr; +using std::map; +using std::shared_ptr; +using std::dynamic_pointer_cast; +using std::make_shared; +using namespace dcpomatic; /** Build a small DCP with no picture and a single subtitle overlaid onto it from a SubRip file */ BOOST_AUTO_TEST_CASE (burnt_subtitle_test_subrip) { - shared_ptr film = new_test_film ("burnt_subtitle_test_subrip"); + auto film = new_test_film ("burnt_subtitle_test_subrip"); film->set_container (Ratio::from_id ("185")); film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR")); film->set_name ("frobozz"); - shared_ptr content (new TextSubtitleContent (film, "test/data/subrip2.srt")); - content->set_use_subtitles (true); - content->set_burn_subtitles (true); - film->examine_and_add_content (content, true); - wait_for_jobs (); - film->make_dcp (); - wait_for_jobs (); + auto content = make_shared(film, "test/data/subrip2.srt"); + content->subtitle->set_use (true); + content->subtitle->set_burn (true); + film->examine_and_add_content (content); + BOOST_REQUIRE (!wait_for_jobs()); + make_and_verify_dcp (film); check_dcp ("test/data/burnt_subtitle_test_subrip", film->dir (film->dcp_name ())); } @@ -54,17 +74,63 @@ BOOST_AUTO_TEST_CASE (burnt_subtitle_test_subrip) /** Build a small DCP with no picture and a single subtitle overlaid onto it from a DCP XML file */ BOOST_AUTO_TEST_CASE (burnt_subtitle_test_dcp) { - shared_ptr film = new_test_film ("burnt_subtitle_test_dcp"); + auto film = new_test_film ("burnt_subtitle_test_dcp"); film->set_container (Ratio::from_id ("185")); film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR")); film->set_name ("frobozz"); - film->set_burn_subtitles (true); - shared_ptr content (new DCPSubtitleContent (film, "test/data/dcp_sub.xml")); - content->set_use_subtitles (true); - film->examine_and_add_content (content, true); - wait_for_jobs (); - film->make_dcp (); - wait_for_jobs (); + auto content = make_shared(film, "test/data/dcp_sub.xml"); + content->subtitle->set_use (true); + film->examine_and_add_content (content); + BOOST_REQUIRE (!wait_for_jobs()); + make_and_verify_dcp (film); check_dcp ("test/data/burnt_subtitle_test_dcp", film->dir (film->dcp_name ())); } + +/** Burn some subtitles into an existing DCP to check the colour conversion */ +BOOST_AUTO_TEST_CASE (burnt_subtitle_test_onto_dcp) +{ + auto film = new_test_film ("burnt_subtitle_test_onto_dcp"); + film->set_container (Ratio::from_id ("185")); + film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR")); + film->set_name ("frobozz"); + film->examine_and_add_content (content_factory(film, "test/data/flat_black.png").front()); + BOOST_REQUIRE (!wait_for_jobs()); + make_and_verify_dcp (film); + + Config::instance()->set_log_types (Config::instance()->log_types() | LogEntry::TYPE_DEBUG_ENCODE); + auto film2 = new_test_film ("burnt_subtitle_test_onto_dcp2"); + film2->set_container (Ratio::from_id ("185")); + film2->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR")); + film2->set_name ("frobozz"); + auto background_dcp = make_shared(film2, film->dir(film->dcp_name())); + film2->examine_and_add_content (background_dcp); + auto sub = dynamic_pointer_cast ( + content_factory(film2, "test/data/subrip2.srt").front() + ); + sub->subtitle->set_burn (true); + sub->subtitle->set_outline (true); + film2->examine_and_add_content (sub); + BOOST_REQUIRE (!wait_for_jobs()); + make_and_verify_dcp (film2); + + BOOST_CHECK (background_dcp->position() == DCPTime()); + BOOST_CHECK (sub->position() == DCPTime()); + + dcp::DCP dcp (film2->dir (film2->dcp_name ())); + dcp.read (); + BOOST_REQUIRE_EQUAL (dcp.cpls().size(), 1); + BOOST_REQUIRE_EQUAL (dcp.cpls().front()->reels().size(), 1); + BOOST_REQUIRE (dcp.cpls().front()->reels().front()->main_picture()); + BOOST_REQUIRE (dcp.cpls().front()->reels().front()->main_picture()->asset()); + auto pic = dynamic_pointer_cast ( + dcp.cpls().front()->reels().front()->main_picture() + )->mono_asset(); + BOOST_REQUIRE (pic); + auto frame = pic->start_read()->get_frame(12); + auto xyz = frame->xyz_image (); + BOOST_CHECK_EQUAL (xyz->size().width, 1998); + BOOST_CHECK_EQUAL (xyz->size().height, 1080); + + check_dcp ("test/data/burnt_subtitle_test_onto_dcp", film->dir(film->dcp_name())); +}