diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/client/players/RawSubs.hx | 151 |
1 files changed, 109 insertions, 42 deletions
diff --git a/src/client/players/RawSubs.hx b/src/client/players/RawSubs.hx index 4fcf3b6..cf37baa 100644 --- a/src/client/players/RawSubs.hx +++ b/src/client/players/RawSubs.hx @@ -25,6 +25,7 @@ class RawSubs { switch ext { case "ass": + parseAss(video, url); case "srt": parseSrt(video, url); case "vtt": @@ -61,57 +62,102 @@ class RawSubs { data += '${sub.time}\n'; data += '${sub.text}\n\n'; } - trace(data); final textBase64 = "data:text/plain;base64,"; final url = textBase64 + Base64.encode(Bytes.ofString(data)); onParsed(video, "SRT subtitles", url); }); } - // function saveSubs() { - // final options = [ - // for (i in 0...subsSelect.options.length) - // subsSelect.item(i) - // ]; - // var inputId = (cast el("#id-offset") : InputElement).value; - // var id = Std.parseInt(inputId); - // if (id == null) id = 1; - // final data:Array<String> = options.map(element -> { - // final value = element.textContent; - // final firstTime = Std.parseFloat(value.split("|")[0]); - // final secondTime = Std.parseFloat(value.split("|")[1]); - // // 00:00:00,498 --> 00:00:02,827 - // var time = '$id\n'; - // time += srvTimeFormat(stringDuration(firstTime)); - // time += " --> "; - // time += srvTimeFormat(stringDuration(secondTime)); - // time += '\ntext$id\n'; - // id++; - // return time; - // }); - // final data = data.join("\n"); - // Utils.saveFile("subs.srv", TextPlain, data); - // } + static function parseAss(video:VideoElement, url:String):Void { + window.fetch(url).then(response -> { + return response.text(); + }).then(text -> { + final subs:Array<{ + counter:Int, + start:String, + end:String, + text:String, + }> = []; + final lines = text.replace("\r\n", "\n").split("\n"); + final matchFormat = ~/^Format:/; + final matchDialogue = ~/^Dialogue:/; + final blockTags = ~/\{\\[^}]*\}/g; + final tags = ~/\\[^ ]+/g; + final drawingMode = ~/\\p[124]/; + var eventStart = false; + var formatFound = false; + final ids:Map<String, Int> = []; + var subsCounter = 1; + for (rawLine in lines) { + final line = rawLine.trim(); + if (!eventStart) { + eventStart = line.startsWith("[Events]"); + continue; + } - function stringDuration(seconds:Float):Duration { - final h = Std.int(seconds / 60 / 60); - final m = Std.int(seconds / 60) - h * 60; - final s = Std.int(seconds % 60); - final ms = Std.int((seconds - Std.int(seconds)) * 1000); - return { + if (!formatFound) { + formatFound = matchFormat.match(line); + if (!formatFound) continue; + final list = matchFormat.replace(line, "").split(","); + for (i in 0...list.length) { + ids[list[i].trim()] = i; + } + ids["_length"] = list.length; + } + + if (!matchDialogue.match(line)) continue; + var list = matchDialogue.replace(line, "").split(","); + while (list.length > ids["_length"]) { + final el = list.pop(); + list[list.length - 1] += el; + } + list = list.map((e) -> e.trim()); + var text = list[ids["Text"]]; + if (drawingMode.match(text)) text = ""; + text = blockTags.replace(text, ""); + text = tags.replace(text, ""); + subs.push({ + counter: subsCounter, + start: convertAssTime(list[ids["Start"]]), + end: convertAssTime(list[ids["End"]]), + text: text, + }); + subsCounter++; + } + + var data = "WEBVTT\n\n"; + for (sub in subs) { + data += '${sub.counter}\n'; + data += '${sub.start} --> ${sub.end}\n'; + data += '${sub.text}\n\n'; + } + final textBase64 = "data:text/plain;base64,"; + final url = textBase64 + Base64.encode(Bytes.ofString(data)); + onParsed(video, "ASS subtitles", url); + }); + } + + static final assTimeStamp = ~/([0-9]+):([0-9][0-9]):([0-9][0-9]).([0-9][0-9])/; + + static function convertAssTime(time:String):String { + if (!assTimeStamp.match(time)) { + return toVttTime({ + h: 0, + m: 0, + s: 0, + ms: 0, + }); + } + final h = Std.parseInt(assTimeStamp.matched(1)); + final m = Std.parseInt(assTimeStamp.matched(2)); + final s = Std.parseInt(assTimeStamp.matched(3)); + final ms = Std.parseInt(assTimeStamp.matched(4)); + return toVttTime({ h: h, m: m, s: s, - ms: ms - } - } - - function srvTimeFormat(time:Duration):String { - final h = '${time.h}'.lpad("0", 2); - final m = '${time.m}'.lpad("0", 2); - final s = '${time.s}'.lpad("0", 2); - final ms = '${time.ms}'.rpad("0", 3); - return '$h:$m:$s,$ms'; + ms: ms, + }); } static function parseVtt(video:VideoElement, url:String):Void { @@ -133,7 +179,28 @@ class RawSubs { video.appendChild(trackEl); } - static function encodeURI(data:String):String { + static inline function encodeURI(data:String):String { return js.Syntax.code("encodeURI({0})", data); } + + static inline function toVttTime(time:Duration):String { + final h = '${time.h}'.lpad("0", 2); + final m = '${time.m}'.lpad("0", 2); + final s = '${time.s}'.lpad("0", 2); + final ms = '${time.ms}'.rpad("0", 3).substr(0, 3); + return '$h:$m:$s.$ms'; + } + + static inline function secondsToDuration(seconds:Float):Duration { + final h = Std.int(seconds / 60 / 60); + final m = Std.int(seconds / 60) - h * 60; + final s = Std.int(seconds % 60); + final ms = Std.int((seconds - Std.int(seconds)) * 1000); + return { + h: h, + m: m, + s: s, + ms: ms + } + } } |