Default to SMPTE for tests.
[dcpomatic.git] / test / dcp_subtitle_test.cc
index ef733d370d5bde31b15f0fa56c6e646a7171d024..7b7308a00737a09faec226c02037d470c4ed3bd2 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2014 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2014-2016 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -29,6 +29,7 @@
 #include "lib/ratio.h"
 #include "lib/dcp_decoder.h"
 #include "lib/dcp_content_type.h"
+#include "lib/dcp_subtitle_decoder.h"
 #include "lib/subtitle_content.h"
 #include "lib/content_subtitle.h"
 #include "lib/subtitle_decoder.h"
@@ -50,7 +51,7 @@ BOOST_AUTO_TEST_CASE (dcp_subtitle_test)
        film->examine_and_add_content (content);
        wait_for_jobs ();
 
-       BOOST_CHECK_EQUAL (content->full_length(), DCPTime::from_seconds (2));
+       BOOST_CHECK_EQUAL (content->full_length().get(), DCPTime::from_seconds(2).get());
 
        content->subtitle->set_use (true);
        content->subtitle->set_burn (false);
@@ -71,7 +72,7 @@ BOOST_AUTO_TEST_CASE (dcp_subtitle_within_dcp_test)
        film->examine_and_add_content (content);
        wait_for_jobs ();
 
-       shared_ptr<DCPDecoder> decoder (new DCPDecoder (content, film->log(), false));
+       shared_ptr<DCPDecoder> decoder (new DCPDecoder (content, film->log()));
 
        list<ContentTimePeriod> ctp = decoder->text_subtitles_during (
                ContentTimePeriod (
@@ -82,10 +83,10 @@ BOOST_AUTO_TEST_CASE (dcp_subtitle_within_dcp_test)
                );
 
        BOOST_REQUIRE_EQUAL (ctp.size(), 2);
-       BOOST_CHECK_EQUAL (ctp.front().from, ContentTime::from_seconds (25 + 12 * 0.04));
-       BOOST_CHECK_EQUAL (ctp.front().to, ContentTime::from_seconds (26 + 4 * 0.04));
-       BOOST_CHECK_EQUAL (ctp.back().from, ContentTime::from_seconds (25 + 12 * 0.04));
-       BOOST_CHECK_EQUAL (ctp.back().to, ContentTime::from_seconds (26 + 4 * 0.04));
+       BOOST_CHECK_EQUAL (ctp.front().from.get(), ContentTime::from_seconds(25 + 12 * 0.04).get());
+       BOOST_CHECK_EQUAL (ctp.front().to.get(), ContentTime::from_seconds(26 + 4 * 0.04).get());
+       BOOST_CHECK_EQUAL (ctp.back().from.get(), ContentTime::from_seconds(25 + 12 * 0.04).get());
+       BOOST_CHECK_EQUAL (ctp.back().to.get(), ContentTime::from_seconds(26 + 4 * 0.04).get());
 
        list<ContentTextSubtitle> subs = decoder->subtitle->get_text (
                ContentTimePeriod (
@@ -101,3 +102,57 @@ BOOST_AUTO_TEST_CASE (dcp_subtitle_within_dcp_test)
        BOOST_CHECK_EQUAL (subs.front().subs.front().text(), "Noch mal.");
        BOOST_CHECK_EQUAL (subs.front().subs.back().text(), "Encore une fois.");
 }
+
+/** Test subtitles whose text includes things like &lt;b&gt; */
+BOOST_AUTO_TEST_CASE (dcp_subtitle_test2)
+{
+       shared_ptr<Film> film = new_test_film ("dcp_subtitle_test2");
+       film->set_container (Ratio::from_id ("185"));
+       film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR"));
+       film->set_name ("frobozz");
+       shared_ptr<DCPSubtitleContent> content (new DCPSubtitleContent (film, "test/data/dcp_sub2.xml"));
+       film->examine_and_add_content (content);
+       wait_for_jobs ();
+
+       shared_ptr<DCPSubtitleDecoder> decoder (new DCPSubtitleDecoder (content, film->log()));
+       list<ContentTextSubtitle> sub = decoder->subtitle->get_text (
+               ContentTimePeriod (ContentTime::from_seconds(0), ContentTime::from_seconds(2)), true, true
+               );
+       BOOST_REQUIRE_EQUAL (sub.size(), 1);
+       BOOST_REQUIRE_EQUAL (sub.front().subs.size(), 1);
+       BOOST_CHECK_EQUAL (sub.front().subs.front().text(), "&lt;b&gt;Hello world!&lt;/b&gt;");
+}
+
+/** Test a failure case */
+BOOST_AUTO_TEST_CASE (dcp_subtitle_test3)
+{
+       shared_ptr<Film> film = new_test_film ("dcp_subtitle_test3");
+       film->set_container (Ratio::from_id ("185"));
+       film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR"));
+       film->set_name ("frobozz");
+       film->set_interop (true);
+       shared_ptr<DCPSubtitleContent> content (new DCPSubtitleContent (film, "test/data/dcp_sub3.xml"));
+       film->examine_and_add_content (content);
+       wait_for_jobs ();
+
+       film->make_dcp ();
+       wait_for_jobs ();
+
+       shared_ptr<DCPSubtitleDecoder> decoder (new DCPSubtitleDecoder (content, film->log()));
+       list<ContentTextSubtitle> sub = decoder->subtitle->get_text (
+               ContentTimePeriod (ContentTime::from_seconds(0), ContentTime::from_seconds(2)), true, true
+               );
+       BOOST_REQUIRE_EQUAL (sub.size(), 1);
+       BOOST_REQUIRE_EQUAL (sub.front().subs.size(), 3);
+       list<dcp::SubtitleString> s = sub.front().subs;
+       list<dcp::SubtitleString>::const_iterator i = s.begin ();
+       BOOST_CHECK_EQUAL (i->text(), "This");
+       ++i;
+       BOOST_REQUIRE (i != s.end ());
+       BOOST_CHECK_EQUAL (i->text(), " is ");
+       ++i;
+       BOOST_REQUIRE (i != s.end ());
+       BOOST_CHECK_EQUAL (i->text(), "wrong.");
+       ++i;
+       BOOST_REQUIRE (i == s.end ());
+}