*/
+/** @file test/subrip_test.cc
+ * @brief Various tests of the subrip code.
+ */
+
#include <boost/test/unit_test.hpp>
+#include <dcp/subtitle_content.h>
#include "lib/subrip.h"
#include "lib/subrip_content.h"
+#include "lib/subrip_decoder.h"
+#include "lib/render_subtitles.h"
+#include "test.h"
using std::list;
+using std::vector;
using std::string;
using boost::shared_ptr;
+using boost::dynamic_pointer_cast;
-/** Test SubRip::convert_time */
-BOOST_AUTO_TEST_CASE (subrip_time_test)
-{
- BOOST_CHECK_EQUAL (SubRip::convert_time ("00:03:10,500"), rint (((3 * 60) + 10 + 0.5) * TIME_HZ));
- BOOST_CHECK_EQUAL (SubRip::convert_time ("04:19:51,782"), rint (((4 * 3600) + (19 * 60) + 51 + 0.782) * TIME_HZ));
-}
-
-/** Test SubRip::convert_coordinate */
-BOOST_AUTO_TEST_CASE (subrip_coordinate_test)
-{
- BOOST_CHECK_EQUAL (SubRip::convert_coordinate ("foo:42"), 42);
- BOOST_CHECK_EQUAL (SubRip::convert_coordinate ("X1:999"), 999);
-}
-
-/** Test SubRip::convert_content */
-BOOST_AUTO_TEST_CASE (subrip_content_test)
-{
- list<string> c;
- list<SubRipSubtitlePiece> p;
-
- c.push_back ("Hello world");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- c.clear ();
-
- c.push_back ("<b>Hello world</b>");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- BOOST_CHECK_EQUAL (p.front().bold, true);
- c.clear ();
-
- c.push_back ("<i>Hello world</i>");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- BOOST_CHECK_EQUAL (p.front().italic, true);
- c.clear ();
-
- c.push_back ("<u>Hello world</u>");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- BOOST_CHECK_EQUAL (p.front().underline, true);
- c.clear ();
-
- c.push_back ("{b}Hello world{/b}");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- BOOST_CHECK_EQUAL (p.front().bold, true);
- c.clear ();
-
- c.push_back ("{i}Hello world{/i}");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- BOOST_CHECK_EQUAL (p.front().italic, true);
- c.clear ();
-
- c.push_back ("{u}Hello world{/u}");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 1);
- BOOST_CHECK_EQUAL (p.front().text, "Hello world");
- BOOST_CHECK_EQUAL (p.front().underline, true);
- c.clear ();
-
- c.push_back ("<b>This is <i>nesting</i> of subtitles</b>");
- p = SubRip::convert_content (c);
- BOOST_CHECK_EQUAL (p.size(), 3);
- list<SubRipSubtitlePiece>::iterator i = p.begin ();
- BOOST_CHECK_EQUAL (i->text, "This is ");
- BOOST_CHECK_EQUAL (i->bold, true);
- BOOST_CHECK_EQUAL (i->italic, false);
- ++i;
- BOOST_CHECK_EQUAL (i->text, "nesting");
- BOOST_CHECK_EQUAL (i->bold, true);
- BOOST_CHECK_EQUAL (i->italic, true);
- ++i;
- BOOST_CHECK_EQUAL (i->text, " of subtitles");
- BOOST_CHECK_EQUAL (i->bold, true);
- BOOST_CHECK_EQUAL (i->italic, false);
- ++i;
- c.clear ();
-}
-
-/** Test parsing of full SubRip file content */
-BOOST_AUTO_TEST_CASE (subrip_parse_test)
+/** Test rendering of a SubRip subtitle */
+BOOST_AUTO_TEST_CASE (subrip_render_test)
{
- shared_ptr<SubRipContent> content (new SubRipContent (shared_ptr<Film> (), "test/data/subrip.srt"));
- content->examine (shared_ptr<Job> ());
- BOOST_CHECK_EQUAL (content->full_length(), ((3 * 60) + 56.471) * TIME_HZ);
-
- SubRip s (content);
-
- list<SubRipSubtitle>::const_iterator i = s._subtitles.begin();
-
- BOOST_CHECK (i != s._subtitles.end ());
- BOOST_CHECK_EQUAL (i->from, ((1 * 60) + 49.200) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->to, ((1 * 60) + 52.351) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->pieces.size(), 1);
- BOOST_CHECK_EQUAL (i->pieces.front().text, "This is a subtitle, and it goes over two lines.");
-
- ++i;
- BOOST_CHECK (i != s._subtitles.end ());
- BOOST_CHECK_EQUAL (i->from, ((1 * 60) + 52.440) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->to, ((1 * 60) + 54.351) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->pieces.size(), 1);
- BOOST_CHECK_EQUAL (i->pieces.front().text, "We have emboldened this");
- BOOST_CHECK_EQUAL (i->pieces.front().bold, true);
-
- ++i;
- BOOST_CHECK (i != s._subtitles.end ());
- BOOST_CHECK_EQUAL (i->from, ((1 * 60) + 54.440) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->to, ((1 * 60) + 56.590) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->pieces.size(), 1);
- BOOST_CHECK_EQUAL (i->pieces.front().text, "And italicised this.");
- BOOST_CHECK_EQUAL (i->pieces.front().italic, true);
-
- ++i;
- BOOST_CHECK (i != s._subtitles.end ());
- BOOST_CHECK_EQUAL (i->from, ((1 * 60) + 56.680) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->to, ((1 * 60) + 58.955) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->pieces.size(), 1);
- BOOST_CHECK_EQUAL (i->pieces.front().text, "Shall I compare thee to a summers' day?");
-
- ++i;
- BOOST_CHECK (i != s._subtitles.end ());
- BOOST_CHECK_EQUAL (i->from, ((2 * 60) + 0.840) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->to, ((2 * 60) + 3.400) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->pieces.size(), 1);
- BOOST_CHECK_EQUAL (i->pieces.front().text, "Is this a dagger I see before me?");
-
- ++i;
- BOOST_CHECK (i != s._subtitles.end ());
- BOOST_CHECK_EQUAL (i->from, ((3 * 60) + 54.560) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->to, ((3 * 60) + 56.471) * TIME_HZ);
- BOOST_CHECK_EQUAL (i->pieces.size(), 1);
- BOOST_CHECK_EQUAL (i->pieces.front().text, "Hello world.");
-
- ++i;
- BOOST_CHECK (i == s._subtitles.end ());
+ shared_ptr<Film> film = new_test_film ("subrip_render_test");
+ shared_ptr<SubRipContent> content (new SubRipContent (film, "test/data/subrip.srt"));
+ content->examine (shared_ptr<Job> (), true);
+ BOOST_CHECK_EQUAL (content->full_length(), DCPTime::from_seconds ((3 * 60) + 56.471));
+
+ shared_ptr<SubRipDecoder> decoder (new SubRipDecoder (content));
+ list<ContentTextSubtitle> cts = decoder->get_text_subtitles (
+ ContentTimePeriod (
+ ContentTime::from_seconds (109), ContentTime::from_seconds (110)
+ ), false
+ );
+ BOOST_CHECK_EQUAL (cts.size(), 1);
+
+ PositionImage image = render_subtitles (cts.front().subs, dcp::Size (1998, 1080));
+ write_image (image.image, "build/test/subrip_render_test.png");
+ check_file ("build/test/subrip_render_test.png", "test/data/subrip_render_test.png");
}
projects / dcpomatic.git / blobdiff