summaryrefslogtreecommitdiffhomepage
path: root/libs/pysubs2/mpl2.py
blob: 5c90bb4f827dd910a3a8185526f045e472116056 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# coding=utf-8

from __future__ import print_function, division, unicode_literals
import re

from .time import times_to_ms
from .formatbase import FormatBase
from .ssaevent import SSAEvent


# thanks to http://otsaloma.io/gaupol/doc/api/aeidon.files.mpl2_source.html
MPL2_FORMAT = re.compile(r"^(?um)\[(-?\d+)\]\[(-?\d+)\](.*)")


class MPL2Format(FormatBase):
    @classmethod
    def guess_format(cls, text):
        if MPL2_FORMAT.search(text):
            return "mpl2"

    @classmethod
    def from_file(cls, subs, fp, format_, **kwargs):
        def prepare_text(lines):
            out = []
            for s in lines.split("|"):
                s = s.strip()

                if s.startswith("/"):
                    # line beginning with '/' is in italics
                    s = r"{\i1}%s{\i0}" % s[1:].strip()

                out.append(s)
            return "\\N".join(out)

        subs.events = [SSAEvent(start=times_to_ms(s=float(start) / 10), end=times_to_ms(s=float(end) / 10),
                       text=prepare_text(text)) for start, end, text in MPL2_FORMAT.findall(fp.getvalue())]

    @classmethod
    def to_file(cls, subs, fp, format_, **kwargs):

        # TODO handle italics
        for line in subs:
            if line.is_comment:
                continue

            print("[{start}][{end}] {text}".format(start=int(line.start // 100),
                                                   end=int(line.end // 100),
                                                   text=line.plaintext.replace("\n", "|")),
                  file=fp)