test/subrip_reader_test.cc

   1 /*
   2     Copyright (C) 2014-2020 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include "subrip_reader.h"
  21 #include "subtitle.h"
  22 #include "test.h"
  23 #include "exceptions.h"
  24 #include "collect.h"
  25 #include <boost/test/unit_test.hpp>
  26 #include <boost/filesystem.hpp>
  27 #include <cmath>
  28 #include <iostream>
  29 #include <cstdio>
  30
  31 using std::cerr;
  32 using std::vector;
  33 using std::fabs;
  34
  35 /* Test reading of a Subrip file */
  36 BOOST_AUTO_TEST_CASE (subrip_reader_test)
  37 {
  38         FILE* f = fopen ("test/data/test.srt", "r");
  39         sub::SubripReader reader (f);
  40         fclose (f);
  41         auto subs = sub::collect<std::vector<sub::Subtitle>> (reader.subtitles());
  42
  43         auto i = subs.begin ();
  44
  45
  46         /* First subtitle */
  47
  48         BOOST_CHECK (i != subs.end ());
  49         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 0, 41, 90));
  50         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 0, 42, 210));
  51
  52         auto j = i->lines.begin();
  53         BOOST_CHECK (j != i->lines.end ());
  54         BOOST_CHECK_EQUAL (j->blocks.size(), 1);
  55         auto b = j->blocks.front();
  56         BOOST_CHECK_EQUAL (b.text, "This is a subtitle");
  57         /* No font is specified by subrip, so none should be seen here */
  58         BOOST_CHECK (!b.font);
  59         BOOST_CHECK (!b.font_size.specified());
  60         BOOST_CHECK_EQUAL (b.bold, false);
  61         BOOST_CHECK_EQUAL (b.italic, false);
  62         BOOST_CHECK_EQUAL (b.underline, false);
  63         BOOST_REQUIRE (j->vertical_position.line);
  64         BOOST_CHECK_EQUAL (j->vertical_position.line.get(), 0);
  65         BOOST_CHECK_EQUAL (j->vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
  66         ++j;
  67
  68         BOOST_CHECK (j != i->lines.end ());
  69         BOOST_CHECK_EQUAL (j->blocks.size(), 1);
  70         b = j->blocks.front ();
  71         BOOST_CHECK_EQUAL (b.text, "and that's a line break");
  72         /* No font is specified by subrip, so none should be seen here */
  73         BOOST_CHECK (!b.font);
  74         BOOST_CHECK (!b.font_size.specified());
  75         BOOST_CHECK_EQUAL (b.bold, false);
  76         BOOST_CHECK_EQUAL (b.italic, false);
  77         BOOST_CHECK_EQUAL (b.underline, false);
  78         BOOST_REQUIRE (j->vertical_position.line);
  79         BOOST_CHECK_EQUAL (j->vertical_position.line.get(), 1);
  80         BOOST_CHECK_EQUAL (j->vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
  81         ++i;
  82
  83
  84         /* Second subtitle */
  85
  86         BOOST_CHECK (i != subs.end ());
  87         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 1, 10));
  88         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 2, 100));
  89
  90         BOOST_CHECK_EQUAL (i->lines.size(), 1);
  91         sub::Line l = i->lines.front ();
  92         BOOST_CHECK_EQUAL (l.blocks.size(), 7);
  93         BOOST_CHECK_EQUAL (l.vertical_position.line.get(), 0);
  94         BOOST_CHECK_EQUAL (l.vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
  95
  96         auto k = l.blocks.begin();
  97
  98         BOOST_CHECK (k != l.blocks.end ());
  99         BOOST_CHECK_EQUAL (k->text, "This is some ");
 100         /* No font is specified by subrip, so none should be seen here */
 101         BOOST_CHECK (!b.font);
 102         BOOST_CHECK (!b.font_size.specified());
 103         BOOST_CHECK_EQUAL (k->bold, false);
 104         BOOST_CHECK_EQUAL (k->italic, false);
 105         BOOST_CHECK_EQUAL (k->underline, false);
 106         ++k;
 107
 108         BOOST_CHECK (k != l.blocks.end ());
 109         BOOST_CHECK_EQUAL (k->text, "bold");
 110         /* No font is specified by subrip, so none should be seen here */
 111         BOOST_CHECK (!b.font);
 112         BOOST_CHECK (!b.font_size.specified());
 113         BOOST_CHECK_EQUAL (k->bold, true);
 114         BOOST_CHECK_EQUAL (k->italic, false);
 115         BOOST_CHECK_EQUAL (k->underline, false);
 116         ++k;
 117
 118         BOOST_CHECK (k != l.blocks.end ());
 119         BOOST_CHECK_EQUAL (k->text, " and some ");
 120         /* No font is specified by subrip, so none should be seen here */
 121         BOOST_CHECK (!b.font);
 122         BOOST_CHECK (!b.font_size.specified());
 123         BOOST_CHECK_EQUAL (k->bold, false);
 124         BOOST_CHECK_EQUAL (k->italic, false);
 125         BOOST_CHECK_EQUAL (k->underline, false);
 126         ++k;
 127
 128         BOOST_CHECK (k != l.blocks.end ());
 129         BOOST_CHECK_EQUAL (k->text, "bold italic");
 130         /* No font is specified by subrip, so none should be seen here */
 131         BOOST_CHECK (!b.font);
 132         BOOST_CHECK (!b.font_size.specified());
 133         BOOST_CHECK_EQUAL (k->bold, true);
 134         BOOST_CHECK_EQUAL (k->italic, true);
 135         BOOST_CHECK_EQUAL (k->underline, false);
 136         ++k;
 137
 138         BOOST_CHECK (k != l.blocks.end ());
 139         BOOST_CHECK_EQUAL (k->text, " and some ");
 140         /* No font is specified by subrip, so none should be seen here */
 141         BOOST_CHECK (!b.font);
 142         BOOST_CHECK (!b.font_size.specified());
 143         BOOST_CHECK_EQUAL (k->bold, false);
 144         BOOST_CHECK_EQUAL (k->italic, false);
 145         BOOST_CHECK_EQUAL (k->underline, false);
 146         ++k;
 147
 148         BOOST_CHECK (k != l.blocks.end ());
 149         BOOST_CHECK_EQUAL (k->text, "underlined");
 150         /* No font is specified by subrip, so none should be seen here */
 151         BOOST_CHECK (!b.font);
 152         BOOST_CHECK (!b.font_size.specified());
 153         BOOST_CHECK_EQUAL (k->bold, false);
 154         BOOST_CHECK_EQUAL (k->italic, false);
 155         BOOST_CHECK_EQUAL (k->underline, true);
 156         ++k;
 157
 158         BOOST_CHECK (k != l.blocks.end ());
 159         BOOST_CHECK_EQUAL (k->text, ".");
 160         /* No font is specified by subrip, so none should be seen here */
 161         BOOST_CHECK (!b.font);
 162         BOOST_CHECK (!b.font_size.specified());
 163         BOOST_CHECK_EQUAL (k->bold, false);
 164         BOOST_CHECK_EQUAL (k->italic, false);
 165         BOOST_CHECK_EQUAL (k->underline, false);
 166         ++k;
 167
 168         BOOST_CHECK (k == l.blocks.end ());
 169 }
 170
 171 /* Test reading of another Subrip file */
 172 BOOST_AUTO_TEST_CASE (subrip_reader_test2)
 173 {
 174         FILE* f = fopen ("test/data/test2.srt", "r");
 175         sub::SubripReader reader (f);
 176         fclose (f);
 177         auto subs = sub::collect<vector<sub::Subtitle>> (reader.subtitles());
 178
 179         auto i = subs.begin();
 180
 181         BOOST_CHECK (i != subs.end ());
 182         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 49, 200));
 183         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 52, 351));
 184         BOOST_CHECK_EQUAL (i->lines.size(), 2);
 185         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "This is a subtitle, and it goes ");
 186         BOOST_CHECK_EQUAL (i->lines.back().blocks.front().text, "over two lines.");
 187
 188         ++i;
 189         BOOST_CHECK (i != subs.end ());
 190         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 52, 440));
 191         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 54, 351));
 192         BOOST_CHECK_EQUAL (i->lines.size(), 1);
 193         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "We have emboldened this");
 194         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().bold, true);
 195
 196         ++i;
 197         BOOST_CHECK (i != subs.end ());
 198         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 54, 440));
 199         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 56, 590));
 200         BOOST_CHECK_EQUAL (i->lines.size(), 1);
 201         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "And italicised this.");
 202         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().italic, true);
 203
 204         ++i;
 205         BOOST_CHECK (i != subs.end ());
 206         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 1, 56, 680));
 207         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 1, 58, 955));
 208         BOOST_CHECK_EQUAL (i->lines.size(), 1);
 209         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Shall I compare thee to a summers' day?");
 210
 211         ++i;
 212         BOOST_CHECK (i != subs.end ());
 213         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 2, 0, 840));
 214         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 2, 3, 400));
 215         BOOST_CHECK_EQUAL (i->lines.size(), 1);
 216         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Is this a dagger I see before me?");
 217
 218         ++i;
 219         BOOST_CHECK (i != subs.end ());
 220         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 3, 54, 560));
 221         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 3, 56, 471));
 222         BOOST_CHECK_EQUAL (i->lines.size(), 1);
 223         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Hello world.");
 224
 225         ++i;
 226         BOOST_CHECK (i != subs.end ());
 227         BOOST_CHECK_EQUAL (i->from, sub::Time::from_hms (0, 4, 50, 123));
 228         BOOST_CHECK_EQUAL (i->to, sub::Time::from_hms (0, 4, 55, 23));
 229         BOOST_CHECK_EQUAL (i->lines.size(), 2);
 230         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().text, "Some italics over");
 231         BOOST_CHECK_EQUAL (i->lines.front().blocks.front().italic, true);
 232         BOOST_CHECK_EQUAL (i->lines.back().blocks.front().text, "multiple lines");
 233         BOOST_CHECK_EQUAL (i->lines.back().blocks.front().italic, true);
 234
 235         ++i;
 236         BOOST_CHECK (i == subs.end ());
 237 }
 238
 239 /** Test SubripReader::convert_line */
 240 BOOST_AUTO_TEST_CASE (subrip_reader_convert_line_test)
 241 {
 242         sub::SubripReader r;
 243
 244         sub::RawSubtitle rs;
 245         r.convert_line ("Hello world", rs);
 246         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 247         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 248         r._subs.clear ();
 249
 250         rs = sub::RawSubtitle();
 251         r.convert_line ("<b>Hello world</b>", rs);
 252         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 253         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 254         BOOST_CHECK_EQUAL (r._subs.front().bold, true);
 255         r._subs.clear ();
 256
 257         rs = sub::RawSubtitle();
 258         r.convert_line ("<i>Hello world</i>", rs);
 259         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 260         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 261         BOOST_CHECK_EQUAL (r._subs.front().italic, true);
 262         r._subs.clear ();
 263
 264         rs = sub::RawSubtitle();
 265         r.convert_line ("<u>Hello world</u>", rs);
 266         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 267         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 268         BOOST_CHECK_EQUAL (r._subs.front().underline, true);
 269         r._subs.clear ();
 270
 271         rs = sub::RawSubtitle();
 272         r.convert_line ("{b}Hello world{/b}", rs);
 273         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 274         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 275         BOOST_CHECK_EQUAL (r._subs.front().bold, true);
 276         r._subs.clear ();
 277
 278         rs = sub::RawSubtitle();
 279         r.convert_line ("{i}Hello world{/i}", rs);
 280         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 281         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 282         BOOST_CHECK_EQUAL (r._subs.front().italic, true);
 283         r._subs.clear ();
 284
 285         rs = sub::RawSubtitle();
 286         r.convert_line ("{u}Hello world{/u}", rs);
 287         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 288         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 289         BOOST_CHECK_EQUAL (r._subs.front().underline, true);
 290         r._subs.clear ();
 291
 292         rs = sub::RawSubtitle();
 293         r.convert_line ("<b>This is <i>nesting</i> of subtitles</b>", rs);
 294         BOOST_CHECK_EQUAL (r._subs.size(), 3);
 295         auto i = r._subs.begin();
 296         BOOST_CHECK_EQUAL (i->text, "This is ");
 297         BOOST_CHECK_EQUAL (i->bold, true);
 298         BOOST_CHECK_EQUAL (i->italic, false);
 299         ++i;
 300         BOOST_CHECK_EQUAL (i->text, "nesting");
 301         BOOST_CHECK_EQUAL (i->bold, true);
 302         BOOST_CHECK_EQUAL (i->italic, true);
 303         ++i;
 304         BOOST_CHECK_EQUAL (i->text, " of subtitles");
 305         BOOST_CHECK_EQUAL (i->bold, true);
 306         BOOST_CHECK_EQUAL (i->italic, false);
 307         ++i;
 308         r._subs.clear ();
 309
 310         rs = sub::RawSubtitle();
 311         r.convert_line ("<B>This is <I>nesting</I> of subtitles</B>", rs);
 312         BOOST_CHECK_EQUAL (r._subs.size(), 3);
 313         i = r._subs.begin();
 314         BOOST_CHECK_EQUAL (i->text, "This is ");
 315         BOOST_CHECK_EQUAL (i->bold, true);
 316         BOOST_CHECK_EQUAL (i->italic, false);
 317         ++i;
 318         BOOST_CHECK_EQUAL (i->text, "nesting");
 319         BOOST_CHECK_EQUAL (i->bold, true);
 320         BOOST_CHECK_EQUAL (i->italic, true);
 321         ++i;
 322         BOOST_CHECK_EQUAL (i->text, " of subtitles");
 323         BOOST_CHECK_EQUAL (i->bold, true);
 324         BOOST_CHECK_EQUAL (i->italic, false);
 325         ++i;
 326         r._subs.clear ();
 327
 328         rs = sub::RawSubtitle();
 329         r.convert_line ("<font color=\"#ff00ff\">simple color</font>", rs);
 330         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 331         BOOST_CHECK_EQUAL (r._subs.front().text, "simple color");
 332         BOOST_CHECK_EQUAL (r._subs.front().bold, false);
 333         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1, 0.1);
 334         BOOST_CHECK (fabs (r._subs.front().colour.g) < 0.01);
 335         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 1, 0.1);
 336         r._subs.clear ();
 337
 338         rs = sub::RawSubtitle();
 339         r.convert_line ("<font color=\"#FF00FF\">simple color in capitals</font>", rs);
 340         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 341         BOOST_CHECK_EQUAL (r._subs.front().text, "simple color in capitals");
 342         BOOST_CHECK_EQUAL (r._subs.front().bold, false);
 343         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1, 0.1);
 344         BOOST_CHECK (fabs (r._subs.front().colour.g) < 0.01);
 345         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 1, 0.1);
 346         r._subs.clear ();
 347
 348         rs = sub::RawSubtitle();
 349         r.convert_line ("<font color=\"#ff0000\">some red text <b>in bold</b></font>", rs);
 350         BOOST_CHECK_EQUAL (r._subs.size(), 2);
 351         i = r._subs.begin ();
 352         BOOST_CHECK_EQUAL (i->text, "some red text ");
 353         BOOST_CHECK_EQUAL (i->bold, false);
 354         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 355         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 356         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 357         ++i;
 358         BOOST_CHECK_EQUAL (i->text, "in bold");
 359         BOOST_CHECK_EQUAL (i->bold, true);
 360         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 361         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 362         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 363         r._subs.clear ();
 364
 365         rs = sub::RawSubtitle();
 366         r.convert_line ("<Font color=\"#ff0000\">some red text <b>in bold</b></font>", rs);
 367         BOOST_CHECK_EQUAL (r._subs.size(), 2);
 368         i = r._subs.begin ();
 369         BOOST_CHECK_EQUAL (i->text, "some red text ");
 370         BOOST_CHECK_EQUAL (i->bold, false);
 371         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 372         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 373         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 374         ++i;
 375         BOOST_CHECK_EQUAL (i->text, "in bold");
 376         BOOST_CHECK_EQUAL (i->bold, true);
 377         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 378         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 379         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 380         r._subs.clear ();
 381
 382         rs = sub::RawSubtitle();
 383         r.convert_line ("<font color=\"#0000ff\">some blue text <b>in bold</b></font>", rs);
 384         BOOST_CHECK_EQUAL (r._subs.size(), 2);
 385         i = r._subs.begin ();
 386         BOOST_CHECK_EQUAL (i->text, "some blue text ");
 387         BOOST_CHECK_EQUAL (i->bold, false);
 388         BOOST_CHECK (fabs (i->colour.r) < 0.01);
 389         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 390         BOOST_CHECK_CLOSE (i->colour.b, 1, 0.1);
 391         ++i;
 392         BOOST_CHECK_EQUAL (i->text, "in bold");
 393         BOOST_CHECK_EQUAL (i->bold, true);
 394         BOOST_CHECK (fabs (i->colour.r) < 0.01);
 395         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 396         BOOST_CHECK_CLOSE (i->colour.b, 1, 0.1);
 397         r._subs.clear ();
 398
 399         rs = sub::RawSubtitle();
 400         r.convert_line ("<< angle brackets but no HTML >>", rs);
 401         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 402         i = r._subs.begin ();
 403         BOOST_CHECK_EQUAL (i->text, "<< angle brackets but no HTML >>");
 404         r._subs.clear();
 405 }
 406
 407 /** Test SubripReader::convert_time */
 408 BOOST_AUTO_TEST_CASE (subrip_reader_convert_time_test)
 409 {
 410         sub::SubripReader reader;
 411         BOOST_CHECK_EQUAL (reader.convert_time ("00:03:10,500"), sub::Time::from_hms (0, 3, 10, 500));
 412         BOOST_CHECK_EQUAL (reader.convert_time ("04:19:51,782"), sub::Time::from_hms (4, 19, 51, 782));
 413 }
 414
 415 static void
 416 test (boost::filesystem::path p)
 417 {
 418         p = private_test / p;
 419         FILE* f = fopen (p.string().c_str(), "r");
 420         BOOST_CHECK (f);
 421         if (!f) {
 422                 cerr << p << " not found.\n";
 423                 return;
 424         }
 425         sub::SubripReader r (f);
 426         fclose (f);
 427 }
 428
 429 static void
 430 test_throw (boost::filesystem::path p)
 431 {
 432         p = private_test / p;
 433         FILE* f = fopen (p.string().c_str(), "r");
 434         BOOST_CHECK (f);
 435         if (!f) {
 436                 cerr << p << " not found.\n";
 437                 return;
 438         }
 439         BOOST_CHECK_THROW (sub::SubripReader r(f), sub::SubripError);
 440         fclose (f);
 441 }
 442
 443 /** Test of reading some typical .srt files */
 444 BOOST_AUTO_TEST_CASE (subrip_read_test)
 445 {
 446         test ("sintel_en.srt");
 447         test ("sintel_fr.srt");
 448         test ("FC.srt");
 449         test ("EU13.srt");
 450         test ("Subtitulos_H_eng.srt");
 451         test ("SWING.srt");
 452         test_throw ("subtitulo1.srt");
 453 }
 454
 455 #define SUB_START(f, t) \
 456         BOOST_REQUIRE (i != subs.end ()); \
 457         BOOST_CHECK_EQUAL (i->from, f); \
 458         BOOST_CHECK_EQUAL (i->to, t); \
 459         j = i->lines.begin ();
 460
 461 #define LINE(p)                                                 \
 462         BOOST_REQUIRE (j != i->lines.end ()); \
 463         BOOST_CHECK (j->vertical_position.line); \
 464         BOOST_CHECK_EQUAL (j->vertical_position.line.get(), p); \
 465         BOOST_CHECK (j->vertical_position.reference); \
 466         BOOST_CHECK_EQUAL (j->vertical_position.reference.get(), sub::TOP_OF_SUBTITLE); \
 467         k = j->blocks.begin (); \
 468         ++j;
 469
 470 #define BLOCK(t, f, s, b, i, u) \
 471         BOOST_REQUIRE (k != j->blocks.end ()); \
 472         BOOST_CHECK_EQUAL (k->text, t); \
 473         BOOST_CHECK_EQUAL (k->bold, b); \
 474         BOOST_CHECK_EQUAL (k->italic, i); \
 475         BOOST_CHECK_EQUAL (k->underline, u); \
 476         ++k;
 477
 478 #define SUB_END() \
 479         ++i;
 480
 481 /** Test reading of another .srt file */
 482 BOOST_AUTO_TEST_CASE (subrip_reader_test3)
 483 {
 484         boost::filesystem::path p = private_test / "DCP-o-matic_test_subs_1.srt";
 485         FILE* f = fopen (p.string().c_str(), "r");
 486         sub::SubripReader reader (f);
 487         fclose (f);
 488         auto subs = sub::collect<std::vector<sub::Subtitle>> (reader.subtitles());
 489
 490         auto i = subs.begin ();
 491         vector<sub::Line>::iterator j;
 492         vector<sub::Block>::iterator k;
 493
 494         BOOST_REQUIRE (i != subs.end ());
 495
 496         SUB_START (sub::Time::from_hms (0, 0, 0, 76), sub::Time::from_hms (0, 0, 1, 116));
 497         LINE (0);
 498         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 499         LINE (1);
 500         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 501         SUB_END ();
 502
 503         SUB_START (sub::Time::from_hms (0, 0, 1, 206), sub::Time::from_hms (0, 0, 2, 246));
 504         LINE (0);
 505         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 506         LINE (1);
 507         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 508         SUB_END ();
 509
 510         SUB_START (sub::Time::from_hms (0, 0, 2, 308), sub::Time::from_hms (0, 0, 3, 380));
 511         LINE (0);
 512         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 513         LINE (1);
 514         BLOCK ("This line is italic", "Arial", 30, false, true, false);
 515         SUB_END ();
 516
 517         SUB_START (sub::Time::from_hms (0, 0, 3, 404), sub::Time::from_hms (0, 0, 4, 484));
 518         LINE (0);
 519         BLOCK ("This line is italic", "Arial", 30, false, true, false);
 520         LINE (1);
 521         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 522         SUB_END ();
 523
 524         SUB_START (sub::Time::from_hms (0, 0, 4, 519), sub::Time::from_hms (0, 0, 5, 604));
 525         LINE (0);
 526         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 527         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 528         LINE (1);
 529         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 530         BLOCK ("italic AND bold", "Arial", 30, true, true, false);
 531         SUB_END ();
 532
 533         SUB_START (sub::Time::from_hms (0, 0, 5, 628), sub::Time::from_hms (0, 0, 6, 712));
 534         LINE (0);
 535         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 536         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 537         LINE (1);
 538         BLOCK ("First three words", "Arial", 30, true, true, false);
 539         BLOCK (" are italic AND bold", "Arial", 30, false, false, false);
 540         SUB_END ();
 541
 542         SUB_START (sub::Time::from_hms (0, 0, 6, 736), sub::Time::from_hms (0, 0, 8, 31));
 543         LINE (0);
 544         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 545         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 546         LINE (1);
 547         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 548         SUB_END ();
 549
 550         SUB_START (sub::Time::from_hms (0, 0, 8, 94), sub::Time::from_hms (0, 0, 9, 211));
 551         LINE (0);
 552         BLOCK ("Both lines are bold AND italic", "Arial", 30, true, true, false);
 553         LINE (1);
 554         BLOCK ("Both lines are bold AND italic", "Arial", 30, true, true, false);
 555         SUB_END ();
 556 }
 557
 558 /** Test reading of a .srt file with RTL text */
 559 BOOST_AUTO_TEST_CASE (subrip_reader_test4)
 560 {
 561         boost::filesystem::path p = private_test / "rtl.srt";
 562         FILE* f = fopen (p.string().c_str(), "r");
 563         sub::SubripReader reader (f);
 564         fclose (f);
 565         auto subs = sub::collect<std::vector<sub::Subtitle>>(reader.subtitles());
 566
 567         auto i = subs.begin();
 568         std::cout << i->lines.front().blocks.front().text << "\n";
 569
 570         std::string const t = i->lines.front().blocks.front().text;
 571         for (size_t i = 0; i < t.length() - 2; ++i) {
 572                 /* Check that unicode U+202B (right-to-left embedding) has been stripped */
 573                 unsigned char const a = t[i];
 574                 unsigned char const b = t[i+1];
 575                 unsigned char const c = t[i+2];
 576                 BOOST_CHECK ((a != 0xe2 || b != 0x80 || c != 0xab));
 577         }
 578
 579         BOOST_CHECK (t == "- \"(دريه فابينار)\"");
 580 }
 581
 582 /** Test <font color="rgba(255,255,255,255)"> */
 583 BOOST_AUTO_TEST_CASE (subrip_reader_test5)
 584 {
 585         sub::RawSubtitle rs;
 586         sub::SubripReader r;
 587         r.convert_line ("<font color=\"rgba(255,128,64,15)\">Foo bar</font>", rs);
 588         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 589         BOOST_CHECK_EQUAL (r._subs.front().text, "Foo bar");
 590         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 255.0 / 255, 0.1);
 591         BOOST_CHECK_CLOSE (r._subs.front().colour.g, 128.0 / 255, 0.1);
 592         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 64.0 / 255, 0.1);
 593         r._subs.clear ();
 594
 595         rs = sub::RawSubtitle ();
 596         r.convert_line ("<font color=\"rgba(1, 2 , 3, 4)\">Foo bar</font>", rs);
 597         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 598         BOOST_CHECK_EQUAL (r._subs.front().text, "Foo bar");
 599         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1.0 / 255, 0.1);
 600         BOOST_CHECK_CLOSE (r._subs.front().colour.g, 2.0 / 255, 0.1);
 601         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 3.0 / 255, 0.1);
 602 }
 603
 604 /** Test alignment */
 605 BOOST_AUTO_TEST_CASE (subrip_reader_test6)
 606 {
 607         sub::RawSubtitle rs;
 608         rs.vertical_position.line = 0;
 609         rs.vertical_position.reference = sub::TOP_OF_SUBTITLE;
 610         sub::SubripReader r;
 611         r.convert_line ("Hello world", rs);
 612         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 613         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 614         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 615         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 616         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 617         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
 618         r._subs.clear ();
 619
 620         rs = sub::RawSubtitle ();
 621         rs.vertical_position.line = 0;
 622         r.convert_line ("{\\an1}Hello", rs);
 623         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 624         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello");
 625         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 626         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 627         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 628         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 629         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 630         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::LEFT_OF_SCREEN);
 631         r._subs.clear ();
 632
 633         rs = sub::RawSubtitle ();
 634         rs.vertical_position.line = 0;
 635         r.convert_line ("{\\an2}to", rs);
 636         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 637         BOOST_CHECK_EQUAL (r._subs.front().text, "to");
 638         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 639         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 640         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 641         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 642         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 643         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::HORIZONTAL_CENTRE_OF_SCREEN);
 644         r._subs.clear ();
 645
 646         rs = sub::RawSubtitle ();
 647         rs.vertical_position.line = 0;
 648         r.convert_line ("{\\an3}you", rs);
 649         BOOST_CHECK_EQUAL (r._subs.front().text, "you");
 650         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 651         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 652         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 653         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 654         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 655         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::RIGHT_OF_SCREEN);
 656         r._subs.clear ();
 657 }
 658
 659
 660 BOOST_AUTO_TEST_CASE(subrip_with_unicode_line_separator_test)
 661 {
 662         auto f = fopen ("test/data/newline.srt", "r");
 663         BOOST_REQUIRE(f);
 664         sub::SubripReader reader(f);
 665         fclose(f);
 666         auto subs = sub::collect<std::vector<sub::Subtitle>>(reader.subtitles());
 667
 668         BOOST_REQUIRE_EQUAL(subs.size(), 2U);
 669
 670         BOOST_REQUIRE_EQUAL(subs[0].lines.size(), 2U);
 671         BOOST_REQUIRE_EQUAL(subs[0].lines[0].blocks.size(), 1U);
 672         BOOST_CHECK_EQUAL(subs[0].lines[0].blocks[0].text, "Du fühlst dich danach besser.");
 673         BOOST_REQUIRE_EQUAL(subs[0].lines[1].blocks.size(), 1U);
 674         BOOST_CHECK_EQUAL(subs[0].lines[1].blocks[0].text, "Okay, Kleiner?");
 675
 676         BOOST_REQUIRE_EQUAL(subs[1].lines.size(), 2U);
 677         BOOST_REQUIRE_EQUAL(subs[1].lines[0].blocks.size(), 1U);
 678         BOOST_CHECK_EQUAL(subs[1].lines[0].blocks[0].text, "Sie kann es nicht machen");
 679         BOOST_REQUIRE_EQUAL(subs[1].lines[1].blocks.size(), 1U);
 680         BOOST_CHECK_EQUAL(subs[1].lines[1].blocks[0].text, "wenn du dich bewegst.");
 681 }
 682