/*
- Copyright (C) 2014 Carl Hetherington <cth@carlh.net>
+ Copyright (C) 2014-2015 Carl Hetherington <cth@carlh.net>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
#include <boost/test/unit_test.hpp>
#include <boost/filesystem.hpp>
#include <fstream>
+#include <cmath>
+#include <iostream>
using std::list;
+using std::cerr;
using std::ifstream;
using std::vector;
+using std::fabs;
/* Test reading of a Subrip file */
BOOST_AUTO_TEST_CASE (subrip_reader_test)
list<sub::Subtitle>::iterator i = subs.begin ();
-
+
/* First subtitle */
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 0, 41, 90));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 0, 42, 210));
-
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 0, 41, 90));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 0, 42, 210));
+
list<sub::Line>::iterator j = i->lines.begin ();
BOOST_CHECK (j != i->lines.end ());
BOOST_CHECK_EQUAL (j->blocks.size(), 1);
sub::Block b = j->blocks.front ();
BOOST_CHECK_EQUAL (b.text, "This is a subtitle");
- BOOST_CHECK_EQUAL (b.font, "Arial");
- BOOST_CHECK_EQUAL (b.font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (b.bold, false);
BOOST_CHECK_EQUAL (b.italic, false);
BOOST_CHECK_EQUAL (b.underline, false);
BOOST_CHECK_EQUAL (j->blocks.size(), 1);
b = j->blocks.front ();
BOOST_CHECK_EQUAL (b.text, "and that's a line break");
- BOOST_CHECK_EQUAL (b.font, "Arial");
- BOOST_CHECK_EQUAL (b.font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (b.bold, false);
BOOST_CHECK_EQUAL (b.italic, false);
BOOST_CHECK_EQUAL (b.underline, false);
BOOST_CHECK_EQUAL (j->vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
++i;
-
+
/* Second subtitle */
-
+
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 1, 1, 10));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 1, 2, 100));
-
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 1, 10));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 2, 100));
+
BOOST_CHECK_EQUAL (i->lines.size(), 1);
sub::Line l = i->lines.front ();
BOOST_CHECK_EQUAL (l.blocks.size(), 7);
BOOST_CHECK_EQUAL (l.vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
list<sub::Block>::iterator k = l.blocks.begin ();
-
+
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, "This is some ");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, false);
BOOST_CHECK_EQUAL (k->italic, false);
BOOST_CHECK_EQUAL (k->underline, false);
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, "bold");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, true);
BOOST_CHECK_EQUAL (k->italic, false);
BOOST_CHECK_EQUAL (k->underline, false);
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, " and some ");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, false);
BOOST_CHECK_EQUAL (k->italic, false);
BOOST_CHECK_EQUAL (k->underline, false);
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, "bold italic");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, true);
BOOST_CHECK_EQUAL (k->italic, true);
BOOST_CHECK_EQUAL (k->underline, false);
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, " and some ");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, false);
BOOST_CHECK_EQUAL (k->italic, false);
BOOST_CHECK_EQUAL (k->underline, false);
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, "underlined");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, false);
BOOST_CHECK_EQUAL (k->italic, false);
BOOST_CHECK_EQUAL (k->underline, true);
BOOST_CHECK (k != l.blocks.end ());
BOOST_CHECK_EQUAL (k->text, ".");
- BOOST_CHECK_EQUAL (k->font, "Arial");
- BOOST_CHECK_EQUAL (k->font_size.points().get(), 48);
+ /* No font is specified by subrip, so none should be seen here */
+ BOOST_CHECK (!b.font);
+ BOOST_CHECK (!b.font_size.specified());
BOOST_CHECK_EQUAL (k->bold, false);
BOOST_CHECK_EQUAL (k->italic, false);
BOOST_CHECK_EQUAL (k->underline, false);
++k;
-
+
BOOST_CHECK (k == l.blocks.end ());
}
list<sub::Subtitle>::const_iterator i = subs.begin();
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 1, 49, 200));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 1, 52, 351));
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 49, 200));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 52, 351));
BOOST_CHECK_EQUAL (i->lines.size(), 2);
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "This is a subtitle, and it goes ");
BOOST_CHECK_EQUAL (i->lines.back().blocks.front().text, "over two lines.");
++i;
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 1, 52, 440));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 1, 54, 351));
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 52, 440));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 54, 351));
BOOST_CHECK_EQUAL (i->lines.size(), 1);
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "We have emboldened this");
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().bold, true);
++i;
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 1, 54, 440));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 1, 56, 590));
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 54, 440));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 56, 590));
BOOST_CHECK_EQUAL (i->lines.size(), 1);
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "And italicised this.");
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().italic, true);
++i;
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 1, 56, 680));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 1, 58, 955));
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 56, 680));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 58, 955));
BOOST_CHECK_EQUAL (i->lines.size(), 1);
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Shall I compare thee to a summers' day?");
++i;
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 2, 0, 840));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 2, 3, 400));
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 2, 0, 840));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 2, 3, 400));
BOOST_CHECK_EQUAL (i->lines.size(), 1);
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Is this a dagger I see before me?");
++i;
BOOST_CHECK (i != subs.end ());
- BOOST_CHECK_EQUAL (i->from.metric().get(), sub::MetricTime (0, 3, 54, 560));
- BOOST_CHECK_EQUAL (i->to.metric().get(), sub::MetricTime (0, 3, 56, 471));
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 3, 54, 560));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 3, 56, 471));
BOOST_CHECK_EQUAL (i->lines.size(), 1);
BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Hello world.");
+ ++i;
+ BOOST_CHECK (i != subs.end ());
+ BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 4, 50, 123));
+ BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 4, 55, 23));
+ BOOST_CHECK_EQUAL (i->lines.size(), 2);
+ BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Some italics over");
+ BOOST_CHECK_EQUAL (i->lines.front().blocks.front().italic, true);
+ BOOST_CHECK_EQUAL (i->lines.back().blocks.front().text, "multiple lines");
+ BOOST_CHECK_EQUAL (i->lines.back().blocks.front().italic, true);
+
++i;
BOOST_CHECK (i == subs.end ());
}
BOOST_AUTO_TEST_CASE (subrip_reader_convert_line_test)
{
sub::SubripReader r;
-
- r.convert_line ("Hello world", 0, sub::TimePair (), sub::TimePair ());
+
+ sub::RawSubtitle rs;
+ r.convert_line ("Hello world", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
r._subs.clear ();
- r.convert_line ("<b>Hello world</b>", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("<b>Hello world</b>", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
BOOST_CHECK_EQUAL (r._subs.front().bold, true);
r._subs.clear ();
- r.convert_line ("<i>Hello world</i>", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("<i>Hello world</i>", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
BOOST_CHECK_EQUAL (r._subs.front().italic, true);
r._subs.clear ();
- r.convert_line ("<u>Hello world</u>", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("<u>Hello world</u>", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
BOOST_CHECK_EQUAL (r._subs.front().underline, true);
r._subs.clear ();
- r.convert_line ("{b}Hello world{/b}", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("{b}Hello world{/b}", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
BOOST_CHECK_EQUAL (r._subs.front().bold, true);
r._subs.clear ();
- r.convert_line ("{i}Hello world{/i}", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("{i}Hello world{/i}", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
BOOST_CHECK_EQUAL (r._subs.front().italic, true);
r._subs.clear ();
- r.convert_line ("{u}Hello world{/u}", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("{u}Hello world{/u}", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 1);
BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
BOOST_CHECK_EQUAL (r._subs.front().underline, true);
r._subs.clear ();
- r.convert_line ("<b>This is <i>nesting</i> of subtitles</b>", 0, sub::TimePair (), sub::TimePair ());
+ rs = sub::RawSubtitle();
+ r.convert_line ("<b>This is <i>nesting</i> of subtitles</b>", rs);
BOOST_CHECK_EQUAL (r._subs.size(), 3);
- list<sub::RawSubtitle>::iterator i = r._subs.begin ();
+ list<sub::RawSubtitle>::iterator i = r._subs.begin ();
BOOST_CHECK_EQUAL (i->text, "This is ");
BOOST_CHECK_EQUAL (i->bold, true);
BOOST_CHECK_EQUAL (i->italic, false);
BOOST_CHECK_EQUAL (i->italic, false);
++i;
r._subs.clear ();
-}
+ rs = sub::RawSubtitle();
+ r.convert_line ("<font color=\"#ff00ff\">simple color</font>", rs);
+ BOOST_CHECK_EQUAL (r._subs.size(), 1);
+ BOOST_CHECK_EQUAL (r._subs.front().text, "simple color");
+ BOOST_CHECK_EQUAL (r._subs.front().bold, false);
+ BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1, 0.1);
+ BOOST_CHECK (fabs (r._subs.front().colour.g) < 0.01);
+ BOOST_CHECK_CLOSE (r._subs.front().colour.b, 1, 0.1);
+ r._subs.clear ();
+
+ rs = sub::RawSubtitle();
+ r.convert_line ("<font color=\"#ff0000\">some red text <b>in bold</b></font>", rs);
+ BOOST_CHECK_EQUAL (r._subs.size(), 2);
+ i = r._subs.begin ();
+ BOOST_CHECK_EQUAL (i->text, "some red text ");
+ BOOST_CHECK_EQUAL (i->bold, false);
+ BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
+ BOOST_CHECK (fabs (i->colour.g) < 0.01);
+ BOOST_CHECK (fabs (i->colour.b) < 0.01);
+ ++i;
+ BOOST_CHECK_EQUAL (i->text, "in bold");
+ BOOST_CHECK_EQUAL (i->bold, true);
+ BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
+ BOOST_CHECK (fabs (i->colour.g) < 0.01);
+ BOOST_CHECK (fabs (i->colour.b) < 0.01);
+ r._subs.clear ();
+
+ rs = sub::RawSubtitle();
+ r.convert_line ("<font color=\"#0000ff\">some blue text <b>in bold</b></font>", rs);
+ BOOST_CHECK_EQUAL (r._subs.size(), 2);
+ i = r._subs.begin ();
+ BOOST_CHECK_EQUAL (i->text, "some blue text ");
+ BOOST_CHECK_EQUAL (i->bold, false);
+ BOOST_CHECK (fabs (i->colour.r) < 0.01);
+ BOOST_CHECK (fabs (i->colour.g) < 0.01);
+ BOOST_CHECK_CLOSE (i->colour.b, 1, 0.1);
+ ++i;
+ BOOST_CHECK_EQUAL (i->text, "in bold");
+ BOOST_CHECK_EQUAL (i->bold, true);
+ BOOST_CHECK (fabs (i->colour.r) < 0.01);
+ BOOST_CHECK (fabs (i->colour.g) < 0.01);
+ BOOST_CHECK_CLOSE (i->colour.b, 1, 0.1);
+ r._subs.clear ();
+}
/** Test SubripReader::convert_time */
BOOST_AUTO_TEST_CASE (subrip_reader_convert_time_test)
{
- BOOST_CHECK_EQUAL (sub::SubripReader::convert_time ("00:03:10,500"), sub::TimePair (sub::MetricTime (0, 3, 10, 500)));
- BOOST_CHECK_EQUAL (sub::SubripReader::convert_time ("04:19:51,782"), sub::TimePair (sub::MetricTime (4, 19, 51, 782)));
+ BOOST_CHECK_EQUAL (sub::SubripReader::convert_time ("00:03:10,500"), sub::Time::from_hms (0, 3, 10, 500));
+ BOOST_CHECK_EQUAL (sub::SubripReader::convert_time ("04:19:51,782"), sub::Time::from_hms (4, 19, 51, 782));
}
-
static void
test (boost::filesystem::path p)
{
p = private_test / p;
FILE* f = fopen (p.string().c_str(), "r");
+ BOOST_CHECK (f);
+ if (!f) {
+ cerr << p << " not found.\n";
+ return;
+ }
sub::SubripReader r (f);
fclose (f);
}
BOOST_AUTO_TEST_CASE (subrip_read_test)
{
test ("sintel_en.srt");
+ test ("sintel_fr.srt");
test ("Fight.Club.1999.720p.BRRip.x264-x0r.srt");
+ test ("EU13.srt");
}