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 ("<font color=\"#ff00ff\">simple color</font>", rs);
 312         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 313         BOOST_CHECK_EQUAL (r._subs.front().text, "simple color");
 314         BOOST_CHECK_EQUAL (r._subs.front().bold, false);
 315         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1, 0.1);
 316         BOOST_CHECK (fabs (r._subs.front().colour.g) < 0.01);
 317         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 1, 0.1);
 318         r._subs.clear ();
 319
 320         rs = sub::RawSubtitle();
 321         r.convert_line ("<font color=\"#FF00FF\">simple color in capitals</font>", rs);
 322         BOOST_CHECK_EQUAL (r._subs.size(), 1);
 323         BOOST_CHECK_EQUAL (r._subs.front().text, "simple color in capitals");
 324         BOOST_CHECK_EQUAL (r._subs.front().bold, false);
 325         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1, 0.1);
 326         BOOST_CHECK (fabs (r._subs.front().colour.g) < 0.01);
 327         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 1, 0.1);
 328         r._subs.clear ();
 329
 330         rs = sub::RawSubtitle();
 331         r.convert_line ("<font color=\"#ff0000\">some red text <b>in bold</b></font>", rs);
 332         BOOST_CHECK_EQUAL (r._subs.size(), 2);
 333         i = r._subs.begin ();
 334         BOOST_CHECK_EQUAL (i->text, "some red text ");
 335         BOOST_CHECK_EQUAL (i->bold, false);
 336         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 337         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 338         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 339         ++i;
 340         BOOST_CHECK_EQUAL (i->text, "in bold");
 341         BOOST_CHECK_EQUAL (i->bold, true);
 342         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 343         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 344         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 345         r._subs.clear ();
 346
 347         rs = sub::RawSubtitle();
 348         r.convert_line ("<Font color=\"#ff0000\">some red text <b>in bold</b></font>", rs);
 349         BOOST_CHECK_EQUAL (r._subs.size(), 2);
 350         i = r._subs.begin ();
 351         BOOST_CHECK_EQUAL (i->text, "some red text ");
 352         BOOST_CHECK_EQUAL (i->bold, false);
 353         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 354         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 355         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 356         ++i;
 357         BOOST_CHECK_EQUAL (i->text, "in bold");
 358         BOOST_CHECK_EQUAL (i->bold, true);
 359         BOOST_CHECK_CLOSE (i->colour.r, 1, 0.1);
 360         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 361         BOOST_CHECK (fabs (i->colour.b) < 0.01);
 362         r._subs.clear ();
 363
 364         rs = sub::RawSubtitle();
 365         r.convert_line ("<font color=\"#0000ff\">some blue text <b>in bold</b></font>", rs);
 366         BOOST_CHECK_EQUAL (r._subs.size(), 2);
 367         i = r._subs.begin ();
 368         BOOST_CHECK_EQUAL (i->text, "some blue text ");
 369         BOOST_CHECK_EQUAL (i->bold, false);
 370         BOOST_CHECK (fabs (i->colour.r) < 0.01);
 371         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 372         BOOST_CHECK_CLOSE (i->colour.b, 1, 0.1);
 373         ++i;
 374         BOOST_CHECK_EQUAL (i->text, "in bold");
 375         BOOST_CHECK_EQUAL (i->bold, true);
 376         BOOST_CHECK (fabs (i->colour.r) < 0.01);
 377         BOOST_CHECK (fabs (i->colour.g) < 0.01);
 378         BOOST_CHECK_CLOSE (i->colour.b, 1, 0.1);
 379         r._subs.clear ();
 380 }
 381
 382 /** Test SubripReader::convert_time */
 383 BOOST_AUTO_TEST_CASE (subrip_reader_convert_time_test)
 384 {
 385         sub::SubripReader reader;
 386         BOOST_CHECK_EQUAL (reader.convert_time ("00:03:10,500"), sub::Time::from_hms (0, 3, 10, 500));
 387         BOOST_CHECK_EQUAL (reader.convert_time ("04:19:51,782"), sub::Time::from_hms (4, 19, 51, 782));
 388 }
 389
 390 static void
 391 test (boost::filesystem::path p)
 392 {
 393         p = private_test / p;
 394         FILE* f = fopen (p.string().c_str(), "r");
 395         BOOST_CHECK (f);
 396         if (!f) {
 397                 cerr << p << " not found.\n";
 398                 return;
 399         }
 400         sub::SubripReader r (f);
 401         fclose (f);
 402 }
 403
 404 static void
 405 test_throw (boost::filesystem::path p)
 406 {
 407         p = private_test / p;
 408         FILE* f = fopen (p.string().c_str(), "r");
 409         BOOST_CHECK (f);
 410         if (!f) {
 411                 cerr << p << " not found.\n";
 412                 return;
 413         }
 414         BOOST_CHECK_THROW (sub::SubripReader r(f), sub::SubripError);
 415         fclose (f);
 416 }
 417
 418 /** Test of reading some typical .srt files */
 419 BOOST_AUTO_TEST_CASE (subrip_read_test)
 420 {
 421         test ("sintel_en.srt");
 422         test ("sintel_fr.srt");
 423         test ("FC.srt");
 424         test ("EU13.srt");
 425         test ("Subtitulos_H_eng.srt");
 426         test ("SWING.srt");
 427         test_throw ("subtitulo1.srt");
 428 }
 429
 430 #define SUB_START(f, t) \
 431         BOOST_REQUIRE (i != subs.end ()); \
 432         BOOST_CHECK_EQUAL (i->from, f); \
 433         BOOST_CHECK_EQUAL (i->to, t); \
 434         j = i->lines.begin ();
 435
 436 #define LINE(p)                                                 \
 437         BOOST_REQUIRE (j != i->lines.end ()); \
 438         BOOST_CHECK (j->vertical_position.line); \
 439         BOOST_CHECK_EQUAL (j->vertical_position.line.get(), p); \
 440         BOOST_CHECK (j->vertical_position.reference); \
 441         BOOST_CHECK_EQUAL (j->vertical_position.reference.get(), sub::TOP_OF_SUBTITLE); \
 442         k = j->blocks.begin (); \
 443         ++j;
 444
 445 #define BLOCK(t, f, s, b, i, u) \
 446         BOOST_REQUIRE (k != j->blocks.end ()); \
 447         BOOST_CHECK_EQUAL (k->text, t); \
 448         BOOST_CHECK_EQUAL (k->bold, b); \
 449         BOOST_CHECK_EQUAL (k->italic, i); \
 450         BOOST_CHECK_EQUAL (k->underline, u); \
 451         ++k;
 452
 453 #define SUB_END() \
 454         ++i;
 455
 456 /** Test reading of another .srt file */
 457 BOOST_AUTO_TEST_CASE (subrip_reader_test3)
 458 {
 459         boost::filesystem::path p = private_test / "DCP-o-matic_test_subs_1.srt";
 460         FILE* f = fopen (p.string().c_str(), "r");
 461         sub::SubripReader reader (f);
 462         fclose (f);
 463         auto subs = sub::collect<std::vector<sub::Subtitle>> (reader.subtitles());
 464
 465         auto i = subs.begin ();
 466         vector<sub::Line>::iterator j;
 467         vector<sub::Block>::iterator k;
 468
 469         BOOST_REQUIRE (i != subs.end ());
 470
 471         SUB_START (sub::Time::from_hms (0, 0, 0, 76), sub::Time::from_hms (0, 0, 1, 116));
 472         LINE (0);
 473         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 474         LINE (1);
 475         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 476         SUB_END ();
 477
 478         SUB_START (sub::Time::from_hms (0, 0, 1, 206), sub::Time::from_hms (0, 0, 2, 246));
 479         LINE (0);
 480         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 481         LINE (1);
 482         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 483         SUB_END ();
 484
 485         SUB_START (sub::Time::from_hms (0, 0, 2, 308), sub::Time::from_hms (0, 0, 3, 380));
 486         LINE (0);
 487         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 488         LINE (1);
 489         BLOCK ("This line is italic", "Arial", 30, false, true, false);
 490         SUB_END ();
 491
 492         SUB_START (sub::Time::from_hms (0, 0, 3, 404), sub::Time::from_hms (0, 0, 4, 484));
 493         LINE (0);
 494         BLOCK ("This line is italic", "Arial", 30, false, true, false);
 495         LINE (1);
 496         BLOCK ("This line is bold", "Arial", 30, true, false, false);
 497         SUB_END ();
 498
 499         SUB_START (sub::Time::from_hms (0, 0, 4, 519), sub::Time::from_hms (0, 0, 5, 604));
 500         LINE (0);
 501         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 502         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 503         LINE (1);
 504         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 505         BLOCK ("italic AND bold", "Arial", 30, true, true, false);
 506         SUB_END ();
 507
 508         SUB_START (sub::Time::from_hms (0, 0, 5, 628), sub::Time::from_hms (0, 0, 6, 712));
 509         LINE (0);
 510         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 511         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 512         LINE (1);
 513         BLOCK ("First three words", "Arial", 30, true, true, false);
 514         BLOCK (" are italic AND bold", "Arial", 30, false, false, false);
 515         SUB_END ();
 516
 517         SUB_START (sub::Time::from_hms (0, 0, 6, 736), sub::Time::from_hms (0, 0, 8, 31));
 518         LINE (0);
 519         BLOCK ("Last three words are ", "Arial", 30, false, false, false);
 520         BLOCK ("bold AND italic", "Arial", 30, true, true, false);
 521         LINE (1);
 522         BLOCK ("This line is normal", "Arial", 30, false, false, false);
 523         SUB_END ();
 524
 525         SUB_START (sub::Time::from_hms (0, 0, 8, 94), sub::Time::from_hms (0, 0, 9, 211));
 526         LINE (0);
 527         BLOCK ("Both lines are bold AND italic", "Arial", 30, true, true, false);
 528         LINE (1);
 529         BLOCK ("Both lines are bold AND italic", "Arial", 30, true, true, false);
 530         SUB_END ();
 531 }
 532
 533 /** Test reading of a .srt file with RTL text */
 534 BOOST_AUTO_TEST_CASE (subrip_reader_test4)
 535 {
 536         boost::filesystem::path p = private_test / "rtl.srt";
 537         FILE* f = fopen (p.string().c_str(), "r");
 538         sub::SubripReader reader (f);
 539         fclose (f);
 540         auto subs = sub::collect<std::vector<sub::Subtitle>>(reader.subtitles());
 541
 542         auto i = subs.begin();
 543         std::cout << i->lines.front().blocks.front().text << "\n";
 544
 545         std::string const t = i->lines.front().blocks.front().text;
 546         for (size_t i = 0; i < t.length() - 2; ++i) {
 547                 /* Check that unicode U+202B (right-to-left embedding) has been stripped */
 548                 unsigned char const a = t[i];
 549                 unsigned char const b = t[i+1];
 550                 unsigned char const c = t[i+2];
 551                 BOOST_CHECK ((a != 0xe2 || b != 0x80 || c != 0xab));
 552         }
 553
 554         BOOST_CHECK (t == "- \"(دريه فابينار)\"");
 555 }
 556
 557 /** Test <font color="rgba(255,255,255,255)"> */
 558 BOOST_AUTO_TEST_CASE (subrip_reader_test5)
 559 {
 560         sub::RawSubtitle rs;
 561         sub::SubripReader r;
 562         r.convert_line ("<font color=\"rgba(255,128,64,15)\">Foo bar</font>", rs);
 563         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 564         BOOST_CHECK_EQUAL (r._subs.front().text, "Foo bar");
 565         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 255.0 / 255, 0.1);
 566         BOOST_CHECK_CLOSE (r._subs.front().colour.g, 128.0 / 255, 0.1);
 567         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 64.0 / 255, 0.1);
 568         r._subs.clear ();
 569
 570         rs = sub::RawSubtitle ();
 571         r.convert_line ("<font color=\"rgba(1, 2 , 3, 4)\">Foo bar</font>", rs);
 572         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 573         BOOST_CHECK_EQUAL (r._subs.front().text, "Foo bar");
 574         BOOST_CHECK_CLOSE (r._subs.front().colour.r, 1.0 / 255, 0.1);
 575         BOOST_CHECK_CLOSE (r._subs.front().colour.g, 2.0 / 255, 0.1);
 576         BOOST_CHECK_CLOSE (r._subs.front().colour.b, 3.0 / 255, 0.1);
 577 }
 578
 579 /** Test alignment */
 580 BOOST_AUTO_TEST_CASE (subrip_reader_test6)
 581 {
 582         sub::RawSubtitle rs;
 583         rs.vertical_position.line = 0;
 584         rs.vertical_position.reference = sub::TOP_OF_SUBTITLE;
 585         sub::SubripReader r;
 586         r.convert_line ("Hello world", rs);
 587         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 588         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello world");
 589         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 590         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 591         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 592         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::TOP_OF_SUBTITLE);
 593         r._subs.clear ();
 594
 595         rs = sub::RawSubtitle ();
 596         rs.vertical_position.line = 0;
 597         r.convert_line ("{\\an1}Hello", rs);
 598         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 599         BOOST_CHECK_EQUAL (r._subs.front().text, "Hello");
 600         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 601         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 602         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 603         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 604         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 605         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::LEFT_OF_SCREEN);
 606         r._subs.clear ();
 607
 608         rs = sub::RawSubtitle ();
 609         rs.vertical_position.line = 0;
 610         r.convert_line ("{\\an2}to", rs);
 611         BOOST_REQUIRE_EQUAL (r._subs.size(), 1);
 612         BOOST_CHECK_EQUAL (r._subs.front().text, "to");
 613         BOOST_REQUIRE (r._subs.front().vertical_position.line);
 614         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.line.get(), 0);
 615         BOOST_REQUIRE (r._subs.front().vertical_position.reference);
 616         BOOST_CHECK_EQUAL (r._subs.front().vertical_position.reference.get(), sub::BOTTOM_OF_SCREEN);
 617         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.proportional, 0);
 618         BOOST_CHECK_EQUAL (r._subs.front().horizontal_position.reference, sub::HORIZONTAL_CENTRE_OF_SCREEN);
 619         r._subs.clear ();
 620
 621         rs = sub::RawSubtitle ();
 622         rs.vertical_position.line = 0;
 623         r.convert_line ("{\\an3}you", rs);
 624         BOOST_CHECK_EQUAL (r._subs.front().text, "you");
 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::RIGHT_OF_SCREEN);
 631         r._subs.clear ();
 632 }
 633