fix: wrongly grouping display sets, fixes #14

pgsrip/media.py

@@ -41,7 +41,7 @@ def create_items(media_path: MediaPath, display_sets: typing.Iterable[DisplaySet
         index = 0
         candidates: typing.List[PgsSubtitleItem] = []
         for ds in display_sets:
-            if current_sets and ds.is_epoch_start():
+            if current_sets and ds.is_start():
                 candidates.append(PgsSubtitleItem(index, media_path, current_sets))
                 current_sets = []
                 index += 1
@@ -61,7 +61,7 @@ def create_items(media_path: MediaPath, display_sets: typing.Iterable[DisplaySet
     @staticmethod
     def generate_image(display_sets: typing.Iterable[DisplaySet]):
         for ds in display_sets:
-            if not ds.pcs.is_epoch_start():
+            if not ds.pcs.is_start():
                 continue
 
             palettes: typing.List[Palette] = []

pgsrip/pgs.py

@@ -259,8 +259,8 @@ def attributes(self):
             'num_objects': 'number_composition_objects'
         }
 
-    def is_epoch_start(self):
-        return self.composition_state == CompositionState.EPOCH_START
+    def is_start(self):
+        return self.composition_state in (CompositionState.EPOCH_START, CompositionState.ACQUISITION_POINT)
 
 
 class WindowDefinitionSegment(BaseSegment):
@@ -414,8 +414,8 @@ def ods_segments(self):
     def end(self):
         return [s for s in self.segments if isinstance(s, EndSegment)][0]
 
-    def is_epoch_start(self):
-        return self.pcs.is_epoch_start()
+    def is_start(self):
+        return self.pcs.is_start()
 
     def is_valid(self):
         valid = True

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pgsrip"
-version = "0.1.8"
+version = "0.1.9"
 description = "Rip your PGS subtitles"
 authors = [
     "Rato",