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         auto t = reader.convert_time("00:03:10,500", ",");
 412         BOOST_REQUIRE(t);
 413         BOOST_CHECK_EQUAL(*t, sub::Time::from_hms(0, 3, 10, 500));
 414         t = reader.convert_time("04:19:51,782", ",");
 415         BOOST_REQUIRE(t);
 416         BOOST_CHECK_EQUAL(*t, sub::Time::from_hms(4, 19, 51, 782));
 417 }
 418
 419 static void
 420 test (boost::filesystem::path p)
 421 {
 422         p = private_test / p;
 423         FILE* f = fopen (p.string().c_str(), "r");
 424         BOOST_CHECK (f);
 425         if (!f) {
 426                 cerr << p << " not found.\n";
 427                 return;
 428         }
 429         sub::SubripReader r (f);
 430         fclose (f);
 431 }
 432
 433 static void
 434 test_throw (boost::filesystem::path p)
 435 {
 436         p = private_test / p;
 437         FILE* f = fopen (p.string().c_str(), "r");
 438         BOOST_CHECK (f);
 439         if (!f) {
 440                 cerr << p << " not found.\n";
 441                 return;
 442         }
 443         BOOST_CHECK_THROW (sub::SubripReader r(f), sub::SubripError);
 444         fclose (f);
 445 }
 446
 447 /** Test of reading some typical .srt files */
 448 BOOST_AUTO_TEST_CASE (subrip_read_test)
 449 {
 450         test ("sintel_en.srt");
 451         test ("sintel_fr.srt");
 452         test ("FC.srt");
 453         test ("EU13.srt");
 454         test ("Subtitulos_H_eng.srt");
 455         test ("SWING.srt");
 456         test_throw ("subtitulo1.srt");
 457 }
 458
 459 #define SUB_START(f, t) \
 460         BOOST_REQUIRE (i != subs.end ()); \
 461         BOOST_CHECK_EQUAL (i->from, f); \
 462         BOOST_CHECK_EQUAL (i->to, t); \
 463         j = i->lines.begin ();
 464
 465 #define LINE(p)                                                 \
 466         BOOST_REQUIRE (j != i->lines.end ()); \
 467         BOOST_CHECK (j->vertical_position.line); \
 468         BOOST_CHECK_EQUAL (j->vertical_position.line.get(), p); \
 469         BOOST_CHECK (j->vertical_position.reference); \
 470         BOOST_CHECK_EQUAL (j->vertical_position.reference.get(), sub::TOP_OF_SUBTITLE); \
 471         k = j->blocks.begin (); \
 472         ++j;
 473
 474 #define BLOCK(t, f, s, b, i, u) \
 475         BOOST_REQUIRE (k != j->blocks.end ()); \
 476         BOOST_CHECK_EQUAL (k->text, t); \
 477         BOOST_CHECK_EQUAL (k->bold, b); \
 478         BOOST_CHECK_EQUAL (k->italic, i); \
 479         BOOST_CHECK_EQUAL (k->underline, u); \
 480         ++k;
 481
 482 #define SUB_END() \
 483         ++i;
 484
 485 /** Test reading of another .srt file */
 486 BOOST_AUTO_TEST_CASE (subrip_reader_test3)
 487 {
 488         boost::filesystem::path p = private_test / "DCP-o-matic_test_subs_1.srt";
 489         FILE* f = fopen (p.string().c_str(), "r");
 490         sub::SubripReader reader (f);
 491         fclose (f);
 492         auto subs = sub::collect<std::vector<sub::Subtitle>> (reader.subtitles());
 493
 494         auto i = subs.begin ();
 495         vector<sub::Line>::iterator j;
 496         vector<sub::Block>::iterator k;
 497
 498         BOOST_REQUIRE (i != subs.end ());
 499
 500         SUB_START (sub::Time::from_hms (0, 0, 0, 76), sub::Time::from_hms (0, 0, 1, 116));
 501         LINE (0);
 502         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 503         LINE (1);
 504         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 505         SUB_END ();
 506
 507         SUB_START (sub::Time::from_hms (0, 0, 1, 206), sub::Time::from_hms (0, 0, 2, 246));
 508         LINE (0);
 509         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 510         LINE (1);
 511         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 512         SUB_END ();
 513
 514         SUB_START (sub::Time::from_hms (0, 0, 2, 308), sub::Time::from_hms (0, 0, 3, 380));
 515         LINE (0);
 516         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 517         LINE (1);
 518         BLOCK ("This line is italic", "Arial", 30, false, true, false);
 519         SUB_END ();
 520
 521         SUB_START (sub::Time::from_hms (0, 0, 3, 404), sub::Time::from_hms (0, 0, 4, 484));
 522         LINE (0);
 523         BLOCK ("This line is italic", "Arial", 30, false, true, false);
 524         LINE (1);
 525         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 526         SUB_END ();
 527
 528         SUB_START (sub::Time::from_hms (0, 0, 4, 519), sub::Time::from_hms (0, 0, 5, 604));
 529         LINE (0);
 530         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 531         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 532         LINE (1);
 533         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 534         BLOCK ("italic AND bold", "Arial", 30, true, true, false);
 535         SUB_END ();
 536
 537         SUB_START (sub::Time::from_hms (0, 0, 5, 628), sub::Time::from_hms (0, 0, 6, 712));
 538         LINE (0);
 539         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 540         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 541         LINE (1);
 542         BLOCK ("First three words", "Arial", 30, true, true, false);
 543         BLOCK (" are italic AND bold", "Arial", 30, false, false, false);
 544         SUB_END ();
 545
 546         SUB_START (sub::Time::from_hms (0, 0, 6, 736), sub::Time::from_hms (0, 0, 8, 31));
 547         LINE (0);
 548         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 549         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 550         LINE (1);
 551         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 552         SUB_END ();
 553
 554         SUB_START (sub::Time::from_hms (0, 0, 8, 94), sub::Time::from_hms (0, 0, 9, 211));
 555         LINE (0);
 556         BLOCK ("Both lines are bold AND italic", "Arial", 30, true, true, false);
 557         LINE (1);
 558         BLOCK ("Both lines are bold AND italic", "Arial", 30, true, true, false);
 559         SUB_END ();
 560 }
 561
 562 /** Test reading of a .srt file with RTL text */
 563 BOOST_AUTO_TEST_CASE (subrip_reader_test4)
 564 {
 565         boost::filesystem::path p = private_test / "rtl.srt";
 566         FILE* f = fopen (p.string().c_str(), "r");
 567         sub::SubripReader reader (f);
 568         fclose (f);
 569         auto subs = sub::collect<std::vector<sub::Subtitle>>(reader.subtitles());
 570
 571         auto i = subs.begin();
 572         std::cout << i->lines.front().blocks.front().text << "\n";
 573
 574         std::string const t = i->lines.front().blocks.front().text;
 575         for (size_t i = 0; i < t.length() - 2; ++i) {
 576                 /* Check that unicode U+202B (right-to-left embedding) has been stripped */
 577                 unsigned char const a = t[i];
 578                 unsigned char const b = t[i+1];
 579                 unsigned char const c = t[i+2];
 580                 BOOST_CHECK ((a != 0xe2 || b != 0x80 || c != 0xab));
 581         }
 582
 583         BOOST_CHECK (t == "- \"(دريه فابينار)\"");
 584 }
 585
 586 /** Test <font color="rgba(255,255,255,255)"> */
 587 BOOST_AUTO_TEST_CASE (subrip_reader_test5)
 588 {
 589         sub::RawSubtitle rs;
 590         sub::SubripReader r;
 591         r.convert_line ("<font color=\"rgba(255,128,64,15)\">Foo bar</font>", rs);
 592         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 593         BOOST_CHECK_EQUAL (r._subs.front().text, "Foo bar");
 594         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 255.0 / 255, 0.1);
 595         BOOST_CHECK_CLOSE (r._subs.front().colour.g, 128.0 / 255, 0.1);
 596         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 64.0 / 255, 0.1);
 597         r._subs.clear ();
 598
 599         rs = sub::RawSubtitle ();
 600         r.convert_line ("<font color=\"rgba(1, 2 , 3, 4)\">Foo bar</font>", rs);
 601         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 602         BOOST_CHECK_EQUAL (r._subs.front().text, "Foo bar");
 603         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1.0 / 255, 0.1);
 604         BOOST_CHECK_CLOSE (r._subs.front().colour.g, 2.0 / 255, 0.1);
 605         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 3.0 / 255, 0.1);
 606 }
 607
 608 /** Test alignment */
 609 BOOST_AUTO_TEST_CASE (subrip_reader_test6)
 610 {
 611         sub::RawSubtitle rs;
 612         rs.vertical_position.line = 0;
 613         rs.vertical_position.reference = sub::TOP_OF_SUBTITLE;
 614         sub::SubripReader r;
 615         r.convert_line ("Hello world", rs);
 616         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 617         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 618         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 619         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 620         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 621         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
 622         r._subs.clear ();
 623
 624         rs = sub::RawSubtitle ();
 625         rs.vertical_position.line = 0;
 626         r.convert_line ("{\\an1}Hello", rs);
 627         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 628         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello");
 629         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 630         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 631         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 632         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 633         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 634         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::LEFT_OF_SCREEN);
 635         r._subs.clear ();
 636
 637         rs = sub::RawSubtitle ();
 638         rs.vertical_position.line = 0;
 639         r.convert_line ("{\\an2}to", rs);
 640         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 641         BOOST_CHECK_EQUAL (r._subs.front().text, "to");
 642         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 643         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 644         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 645         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 646         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 647         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::HORIZONTAL_CENTRE_OF_SCREEN);
 648         r._subs.clear ();
 649
 650         rs = sub::RawSubtitle ();
 651         rs.vertical_position.line = 0;
 652         r.convert_line ("{\\an3}you", rs);
 653         BOOST_CHECK_EQUAL (r._subs.front().text, "you");
 654         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 655         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 656         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 657         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 658         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 659         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::RIGHT_OF_SCREEN);
 660         r._subs.clear ();
 661 }
 662
 663
 664 BOOST_AUTO_TEST_CASE(subrip_with_unicode_line_separator_test)
 665 {
 666         auto f = fopen ("test/data/newline.srt", "r");
 667         BOOST_REQUIRE(f);
 668         sub::SubripReader reader(f);
 669         fclose(f);
 670         auto subs = sub::collect<std::vector<sub::Subtitle>>(reader.subtitles());
 671
 672         BOOST_REQUIRE_EQUAL(subs.size(), 2U);
 673
 674         BOOST_REQUIRE_EQUAL(subs[0].lines.size(), 2U);
 675         BOOST_REQUIRE_EQUAL(subs[0].lines[0].blocks.size(), 1U);
 676         BOOST_CHECK_EQUAL(subs[0].lines[0].blocks[0].text, "Du fühlst dich danach besser.");
 677         BOOST_REQUIRE_EQUAL(subs[0].lines[1].blocks.size(), 1U);
 678         BOOST_CHECK_EQUAL(subs[0].lines[1].blocks[0].text, "Okay, Kleiner?");
 679
 680         BOOST_REQUIRE_EQUAL(subs[1].lines.size(), 2U);
 681         BOOST_REQUIRE_EQUAL(subs[1].lines[0].blocks.size(), 1U);
 682         BOOST_CHECK_EQUAL(subs[1].lines[0].blocks[0].text, "Sie kann es nicht machen");
 683         BOOST_REQUIRE_EQUAL(subs[1].lines[1].blocks.size(), 1U);
 684         BOOST_CHECK_EQUAL(subs[1].lines[1].blocks[0].text, "wenn du dich bewegst.");
 685 }
 686