80 lines
3.1 KiB
C++
80 lines
3.1 KiB
C++
#include "subtitleparser.h"
|
||
|
||
#include <QTest>
|
||
|
||
using namespace Qt::Literals::StringLiterals;
|
||
|
||
class SubtitleParserTest : public QObject
|
||
{
|
||
Q_OBJECT
|
||
|
||
private Q_SLOTS:
|
||
void parsesSrt()
|
||
{
|
||
const QString srt = u"1\r\n00:00:01,000 --> 00:00:02,500\r\n<i>Hello</i>\r\nthere\r\n\r\n"
|
||
"2\r\n00:01:00,000 --> 00:01:01,000 X1:10\r\n{\\an8}Top line\r\n"_s;
|
||
const auto cues = Subtitles::parseSrt(srt);
|
||
QCOMPARE(cues.size(), 2);
|
||
QCOMPARE(cues.at(0), (SubtitleCue{1000, 2500, u"Hello\nthere"_s}));
|
||
QCOMPARE(cues.at(1), (SubtitleCue{60000, 61000, u"Top line"_s}));
|
||
}
|
||
|
||
void parsesWebVtt()
|
||
{
|
||
const QString vtt = u"WEBVTT\n\nintro\n00:05.250 --> 00:07.000 align:start\n<v Anna>Hi!</v>\n\n"
|
||
"01:00:00.000 --> 01:00:01.000\nLater\n"_s;
|
||
const auto cues = Subtitles::parseSrt(vtt);
|
||
QCOMPARE(cues.size(), 2);
|
||
QCOMPARE(cues.at(0), (SubtitleCue{5250, 7000, u"Hi!"_s}));
|
||
QCOMPARE(cues.at(1), (SubtitleCue{3600000, 3601000, u"Later"_s}));
|
||
}
|
||
|
||
void parsesAss()
|
||
{
|
||
const QString ass = u"[Script Info]\nTitle: test\n\n[Events]\n"
|
||
"Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\n"
|
||
"Dialogue: 0,0:00:01.50,0:00:03.00,Default,,0,0,0,,{\\i1}Wait,{\\i0} what?\\NNo way.\n"
|
||
"Comment: 0,0:00:04.00,0:00:05.00,Default,,0,0,0,,ignored\n"
|
||
"Dialogue: 0,0:00:04.00,0:00:05.00,Sign,,0,0,0,,{\\p1}m 0 0 l 10 0 10 10{\\p0}\n"_s;
|
||
const auto cues = Subtitles::parseAss(ass);
|
||
QCOMPARE(cues.size(), 2);
|
||
QCOMPARE(cues.at(0), (SubtitleCue{1500, 3000, u"Wait, what?\nNo way."_s}));
|
||
// Vector drawings carry no words.
|
||
QVERIFY(cues.at(1).text.isEmpty());
|
||
}
|
||
|
||
void mergesRepeatedFrames()
|
||
{
|
||
QList<SubtitleCue> cues{
|
||
{0, 1000, u"Hello"_s},
|
||
{1000, 1500, u""_s},
|
||
{5000, 6000, u"Sign"_s}, // too far from the earlier pieces: a new cue
|
||
{7000, 7040, u"Xkcd Qwer"_s}, // a lone 40 ms animation frame
|
||
};
|
||
// One sign drawn frame by frame, 40 ms each.
|
||
for (qint64 t = 2000; t < 2400; t += 40) {
|
||
cues.append({t, t + 40, u"Sign"_s});
|
||
}
|
||
const auto merged = Subtitles::mergeRepeated(cues);
|
||
QCOMPARE(merged.size(), 3);
|
||
QCOMPARE(merged.at(0), (SubtitleCue{0, 1000, u"Hello"_s}));
|
||
QCOMPARE(merged.at(1), (SubtitleCue{2000, 2400, u"Sign"_s}));
|
||
QCOMPARE(merged.at(2), (SubtitleCue{5000, 6000, u"Sign"_s}));
|
||
}
|
||
|
||
void writesSrtThatParsesBack()
|
||
{
|
||
const QList<SubtitleCue> cues{
|
||
{1000, 2500, u"Xin chào\nbạn"_s},
|
||
{3723004, 3724000, u"Một giờ sau"_s},
|
||
};
|
||
const QString srt = Subtitles::toSrt(cues);
|
||
QVERIFY(srt.startsWith(u"1\n00:00:01,000 --> 00:00:02,500\nXin chào\nbạn\n\n"_s));
|
||
QCOMPARE(Subtitles::parseSrt(srt), cues);
|
||
}
|
||
};
|
||
|
||
QTEST_GUILESS_MAIN(SubtitleParserTest)
|
||
|
||
#include "subtitleparsertest.moc"
|