mkv-episode-matcher 0.1.4__py3-none-any.whl → 0.1.9__py3-none-any.whl

mkv_episode_matcher/libraries/pgs2srt/pgsreader.py

@@ -1,221 +1,221 @@
-#!/usr/bin/env python3
-
-from collections import namedtuple
-from os.path import split as pathsplit
-
-# Constants for Segments
-PDS = int('0x14', 16)
-ODS = int('0x15', 16)
-PCS = int('0x16', 16)
-WDS = int('0x17', 16)
-END = int('0x80', 16)
-
-# Named tuple access for static PDS palettes
-Palette = namedtuple('Palette', "Y Cr Cb Alpha")
-
-class InvalidSegmentError(Exception):
-    '''Raised when a segment does not match PGS specification'''
-
-
-class PGSReader:
-
-    def __init__(self, filepath):
-        self.filedir, self.file = pathsplit(filepath)
-        with open(filepath, 'rb') as f:
-            self.bytes = f.read()
-
-
-    def make_segment(self, bytes_):
-        cls = SEGMENT_TYPE[bytes_[10]]
-        return cls(bytes_)
-
-    def iter_segments(self):
-        bytes_ = self.bytes[:]
-        while bytes_:
-            size = 13 + int(bytes_[11:13].hex(), 16)
-            yield self.make_segment(bytes_[:size])
-            bytes_ = bytes_[size:]
-
-    def iter_displaysets(self):
-        ds = []
-        for s in self.iter_segments():
-            ds.append(s)
-            if s.type == 'END':
-                yield DisplaySet(ds)
-                ds = []
-
-    @property
-    def segments(self):
-        if not hasattr(self, '_segments'):
-            self._segments = list(self.iter_segments())
-        return self._segments
-
-    @property
-    def displaysets(self):
-        if not hasattr(self, '_displaysets'):
-            self._displaysets = list(self.iter_displaysets())
-        return self._displaysets
-
-class BaseSegment:
-
-    SEGMENT = {
-        PDS: 'PDS',
-        ODS: 'ODS',
-        PCS: 'PCS',
-        WDS: 'WDS',
-        END: 'END'
-    }
-
-    def __init__(self, bytes_):
-        self.bytes = bytes_
-        if bytes_[:2] != b'PG':
-            raise InvalidSegmentError
-        self.pts = int(bytes_[2:6].hex(), base=16) / 90
-        self.dts = int(bytes_[6:10].hex(), base=16) / 90
-        self.type = self.SEGMENT[bytes_[10]]
-        self.size = int(bytes_[11:13].hex(), base=16)
-        self.data = bytes_[13:]
-
-    def __len__(self):
-        return self.size
-
-    @property
-    def presentation_timestamp(self): return self.pts
-
-    @property
-    def decoding_timestamp(self): return self.dts
-
-    @property
-    def segment_type(self): return self.type
-
-class PresentationCompositionSegment(BaseSegment):
-
-    class CompositionObject:
-
-        def __init__(self, bytes_):
-            self.bytes = bytes_
-            self.object_id = int(bytes_[0:2].hex(), base=16)
-            self.window_id = bytes_[2]
-            self.cropped = bool(bytes_[3])
-            self.x_offset = int(bytes_[4:6].hex(), base=16)
-            self.y_offset = int(bytes_[6:8].hex(), base=16)
-            if self.cropped:
-                self.crop_x_offset = int(bytes_[8:10].hex(), base=16)
-                self.crop_y_offset = int(bytes_[10:12].hex(), base=16)
-                self.crop_width = int(bytes_[12:14].hex(), base=16)
-                self.crop_height = int(bytes_[14:16].hex(), base=16)
-
-    STATE = {
-        int('0x00', base=16): 'Normal',
-        int('0x40', base=16): 'Acquisition Point',
-        int('0x80', base=16): 'Epoch Start'
-    }
-
-    def __init__(self, bytes_):
-        BaseSegment.__init__(self, bytes_)
-        self.width = int(self.data[0:2].hex(), base=16)
-        self.height = int(self.data[2:4].hex(), base=16)
-        self.frame_rate = self.data[4]
-        self._num = int(self.data[5:7].hex(), base=16)
-        self._state = self.STATE[self.data[7]]
-        self.palette_update = bool(self.data[8])
-        self.palette_id = self.data[9]
-        self._num_comps = self.data[10]
-
-    @property
-    def composition_number(self): return self._num
-
-    @property
-    def composition_state(self): return self._state
-
-    @property
-    def composition_objects(self):
-        if not hasattr(self, '_composition_objects'):
-            self._composition_objects = self.get_composition_objects()
-            if len(self._composition_objects) != self._num_comps:
-                print('Warning: Number of composition objects asserted '
-                      'does not match the amount found.')
-        return self._composition_objects
-
-    def get_composition_objects(self):
-        bytes_ = self.data[11:]
-        comps = []
-        while bytes_:
-            length = 8 * (1 + bool(bytes_[3]))
-            comps.append(self.CompositionObject(bytes_[:length]))
-            bytes_ = bytes_[length:]
-        return comps
-
-class WindowDefinitionSegment(BaseSegment):
-
-    def __init__(self, bytes_):
-        BaseSegment.__init__(self, bytes_)
-        self.num_windows = self.data[0]
-        self.window_id = self.data[1]
-        self.x_offset = int(self.data[2:4].hex(), base=16)
-        self.y_offset = int(self.data[4:6].hex(), base=16)
-        self.width = int(self.data[6:8].hex(), base=16)
-        self.height = int(self.data[8:10].hex(), base=16)
-
-class PaletteDefinitionSegment(BaseSegment):
-
-    def __init__(self, bytes_):
-        BaseSegment.__init__(self, bytes_)
-        self.palette_id = self.data[0]
-        self.version = self.data[1]
-        self.palette = [Palette(0, 0, 0, 0)] * 256
-        # Slice from byte 2 til end of segment. Divide by 5 to determine number of palette entries
-        # Iterate entries. Explode the 5 bytes into namedtuple Palette. Must be exploded
-        for entry in range(len(self.data[2:]) // 5):
-            i = 2 + entry * 5
-            self.palette[self.data[i]] = Palette(*self.data[i + 1:i + 5])
-
-class ObjectDefinitionSegment(BaseSegment):
-
-    SEQUENCE = {
-        int('0x40', base=16): 'Last',
-        int('0x80', base=16): 'First',
-        int('0xc0', base=16): 'First and last'
-    }
-
-    def __init__(self, bytes_):
-        BaseSegment.__init__(self, bytes_)
-        self.id = int(self.data[0:2].hex(), base=16)
-        self.version = self.data[2]
-        self.in_sequence = self.SEQUENCE[self.data[3]]
-        self.data_len = int(self.data[4:7].hex(), base=16)
-        self.width = int(self.data[7:9].hex(), base=16)
-        self.height = int(self.data[9:11].hex(), base=16)
-        self.img_data = self.data[11:]
-        if len(self.img_data) != self.data_len - 4:
-            print('Warning: Image data length asserted does not match the '
-                  'length found.')
-
-class EndSegment(BaseSegment):
-
-    @property
-    def is_end(self): return True
-
-
-SEGMENT_TYPE = {
-    PDS: PaletteDefinitionSegment,
-    ODS: ObjectDefinitionSegment,
-    PCS: PresentationCompositionSegment,
-    WDS: WindowDefinitionSegment,
-    END: EndSegment
-}
-
-class DisplaySet:
-
-    def __init__(self, segments):
-        self.segments = segments
-        self.segment_types = [s.type for s in segments]
-        self.has_image = 'ODS' in self.segment_types
-
-def segment_by_type_getter(type_):
-    def f(self):
-        return [s for s in self.segments if s.type == type_]
-    return f
-
-for type_ in BaseSegment.SEGMENT.values():
-    setattr(DisplaySet, type_.lower(), property(segment_by_type_getter(type_)))
+#!/usr/bin/env python3
+
+from collections import namedtuple
+from os.path import split as pathsplit
+
+# Constants for Segments
+PDS = int('0x14', 16)
+ODS = int('0x15', 16)
+PCS = int('0x16', 16)
+WDS = int('0x17', 16)
+END = int('0x80', 16)
+
+# Named tuple access for static PDS palettes
+Palette = namedtuple('Palette', "Y Cr Cb Alpha")
+
+class InvalidSegmentError(Exception):
+    '''Raised when a segment does not match PGS specification'''
+
+
+class PGSReader:
+
+    def __init__(self, filepath):
+        self.filedir, self.file = pathsplit(filepath)
+        with open(filepath, 'rb') as f:
+            self.bytes = f.read()
+
+
+    def make_segment(self, bytes_):
+        cls = SEGMENT_TYPE[bytes_[10]]
+        return cls(bytes_)
+
+    def iter_segments(self):
+        bytes_ = self.bytes[:]
+        while bytes_:
+            size = 13 + int(bytes_[11:13].hex(), 16)
+            yield self.make_segment(bytes_[:size])
+            bytes_ = bytes_[size:]
+
+    def iter_displaysets(self):
+        ds = []
+        for s in self.iter_segments():
+            ds.append(s)
+            if s.type == 'END':
+                yield DisplaySet(ds)
+                ds = []
+
+    @property
+    def segments(self):
+        if not hasattr(self, '_segments'):
+            self._segments = list(self.iter_segments())
+        return self._segments
+
+    @property
+    def displaysets(self):
+        if not hasattr(self, '_displaysets'):
+            self._displaysets = list(self.iter_displaysets())
+        return self._displaysets
+
+class BaseSegment:
+
+    SEGMENT = {
+        PDS: 'PDS',
+        ODS: 'ODS',
+        PCS: 'PCS',
+        WDS: 'WDS',
+        END: 'END'
+    }
+
+    def __init__(self, bytes_):
+        self.bytes = bytes_
+        if bytes_[:2] != b'PG':
+            raise InvalidSegmentError
+        self.pts = int(bytes_[2:6].hex(), base=16) / 90
+        self.dts = int(bytes_[6:10].hex(), base=16) / 90
+        self.type = self.SEGMENT[bytes_[10]]
+        self.size = int(bytes_[11:13].hex(), base=16)
+        self.data = bytes_[13:]
+
+    def __len__(self):
+        return self.size
+
+    @property
+    def presentation_timestamp(self): return self.pts
+
+    @property
+    def decoding_timestamp(self): return self.dts
+
+    @property
+    def segment_type(self): return self.type
+
+class PresentationCompositionSegment(BaseSegment):
+
+    class CompositionObject:
+
+        def __init__(self, bytes_):
+            self.bytes = bytes_
+            self.object_id = int(bytes_[0:2].hex(), base=16)
+            self.window_id = bytes_[2]
+            self.cropped = bool(bytes_[3])
+            self.x_offset = int(bytes_[4:6].hex(), base=16)
+            self.y_offset = int(bytes_[6:8].hex(), base=16)
+            if self.cropped:
+                self.crop_x_offset = int(bytes_[8:10].hex(), base=16)
+                self.crop_y_offset = int(bytes_[10:12].hex(), base=16)
+                self.crop_width = int(bytes_[12:14].hex(), base=16)
+                self.crop_height = int(bytes_[14:16].hex(), base=16)
+
+    STATE = {
+        int('0x00', base=16): 'Normal',
+        int('0x40', base=16): 'Acquisition Point',
+        int('0x80', base=16): 'Epoch Start'
+    }
+
+    def __init__(self, bytes_):
+        BaseSegment.__init__(self, bytes_)
+        self.width = int(self.data[0:2].hex(), base=16)
+        self.height = int(self.data[2:4].hex(), base=16)
+        self.frame_rate = self.data[4]
+        self._num = int(self.data[5:7].hex(), base=16)
+        self._state = self.STATE[self.data[7]]
+        self.palette_update = bool(self.data[8])
+        self.palette_id = self.data[9]
+        self._num_comps = self.data[10]
+
+    @property
+    def composition_number(self): return self._num
+
+    @property
+    def composition_state(self): return self._state
+
+    @property
+    def composition_objects(self):
+        if not hasattr(self, '_composition_objects'):
+            self._composition_objects = self.get_composition_objects()
+            if len(self._composition_objects) != self._num_comps:
+                print('Warning: Number of composition objects asserted '
+                      'does not match the amount found.')
+        return self._composition_objects
+
+    def get_composition_objects(self):
+        bytes_ = self.data[11:]
+        comps = []
+        while bytes_:
+            length = 8 * (1 + bool(bytes_[3]))
+            comps.append(self.CompositionObject(bytes_[:length]))
+            bytes_ = bytes_[length:]
+        return comps
+
+class WindowDefinitionSegment(BaseSegment):
+
+    def __init__(self, bytes_):
+        BaseSegment.__init__(self, bytes_)
+        self.num_windows = self.data[0]
+        self.window_id = self.data[1]
+        self.x_offset = int(self.data[2:4].hex(), base=16)
+        self.y_offset = int(self.data[4:6].hex(), base=16)
+        self.width = int(self.data[6:8].hex(), base=16)
+        self.height = int(self.data[8:10].hex(), base=16)
+
+class PaletteDefinitionSegment(BaseSegment):
+
+    def __init__(self, bytes_):
+        BaseSegment.__init__(self, bytes_)
+        self.palette_id = self.data[0]
+        self.version = self.data[1]
+        self.palette = [Palette(0, 0, 0, 0)] * 256
+        # Slice from byte 2 til end of segment. Divide by 5 to determine number of palette entries
+        # Iterate entries. Explode the 5 bytes into namedtuple Palette. Must be exploded
+        for entry in range(len(self.data[2:]) // 5):
+            i = 2 + entry * 5
+            self.palette[self.data[i]] = Palette(*self.data[i + 1:i + 5])
+
+class ObjectDefinitionSegment(BaseSegment):
+
+    SEQUENCE = {
+        int('0x40', base=16): 'Last',
+        int('0x80', base=16): 'First',
+        int('0xc0', base=16): 'First and last'
+    }
+
+    def __init__(self, bytes_):
+        BaseSegment.__init__(self, bytes_)
+        self.id = int(self.data[0:2].hex(), base=16)
+        self.version = self.data[2]
+        self.in_sequence = self.SEQUENCE[self.data[3]]
+        self.data_len = int(self.data[4:7].hex(), base=16)
+        self.width = int(self.data[7:9].hex(), base=16)
+        self.height = int(self.data[9:11].hex(), base=16)
+        self.img_data = self.data[11:]
+        if len(self.img_data) != self.data_len - 4:
+            print('Warning: Image data length asserted does not match the '
+                  'length found.')
+
+class EndSegment(BaseSegment):
+
+    @property
+    def is_end(self): return True
+
+
+SEGMENT_TYPE = {
+    PDS: PaletteDefinitionSegment,
+    ODS: ObjectDefinitionSegment,
+    PCS: PresentationCompositionSegment,
+    WDS: WindowDefinitionSegment,
+    END: EndSegment
+}
+
+class DisplaySet:
+
+    def __init__(self, segments):
+        self.segments = segments
+        self.segment_types = [s.type for s in segments]
+        self.has_image = 'ODS' in self.segment_types
+
+def segment_by_type_getter(type_):
+    def f(self):
+        return [s for s in self.segments if s.type == type_]
+    return f
+
+for type_ in BaseSegment.SEGMENT.values():
+    setattr(DisplaySet, type_.lower(), property(segment_by_type_getter(type_)))

mkv_episode_matcher/libraries/pgs2srt/requirements.txt

@@ -1,4 +1,4 @@
-pytesseract==0.3.7
-numpy==1.19.4
-Pillow==8.2.0
-tld~=0.12.3
+pytesseract==0.3.7
+numpy==1.19.4
+Pillow==8.2.0
+tld~=0.12.3