Move float->standard str used in PD2V & PLS to helpers

Author: rlaphoenix

pvsfunc/helpers.py

@@ -4,6 +4,19 @@
 from typing import Iterable, List
 
 
+def get_standard(aspect: float) -> str:
+    """Convert an aspect float to a standard string."""
+    return {
+        0: "?",
+        24 / 1: "FILM",
+        25 / 1: "PAL",
+        50 / 1: "PALi",
+        30000 / 1001: "NTSC",
+        60000 / 1001: "NTSCi",
+        24000 / 1001: "NTSC (FILM)"
+    }[aspect]
+
+
 def calculate_aspect_ratio(width: int, height: int) -> str:
     """Calculate the aspect-ratio gcd string from resolution."""
     r = math.gcd(width, height)

pvsfunc/pd2v.py

@@ -11,7 +11,7 @@
 from pyd2v import D2V
 from vapoursynth import core
 
-from pvsfunc.helpers import group_by_int, list_select_every, calculate_aspect_ratio, calculate_par
+from pvsfunc.helpers import group_by_int, list_select_every, calculate_aspect_ratio, calculate_par, get_standard
 
 
 class PD2V:
@@ -47,15 +47,7 @@ def __init__(self, file: str, verbose=False):
             coded_f = len(self.flags)
             progressive_f = sum(f["progressive_frame"] for f in self.flags)
             progressive_p = (progressive_f / coded_f) * 100
-            standard = {
-                0: "?",
-                24 / 1: "FILM",
-                25 / 1: "PAL",
-                50 / 1: "PALi",
-                30000 / 1001: "NTSC",
-                60000 / 1001: "NTSCi",
-                24000 / 1001: "NTSC (FILM)"
-            }[self.clip.fps.numerator / self.clip.fps.denominator]
+            standard = get_standard(self.clip.fps.numerator / self.clip.fps.denominator)
             dar = self.d2v.settings["Aspect_Ratio"]
             if isinstance(dar, list):
                 dar = dar[0]

pvsfunc/pls.py

@@ -6,7 +6,7 @@
 from pymediainfo import MediaInfo
 from vapoursynth import core
 
-from pvsfunc.helpers import calculate_aspect_ratio
+from pvsfunc.helpers import calculate_aspect_ratio, get_standard
 
 
 class PLS:
@@ -32,15 +32,7 @@ def __init__(self, file: str, verbose=False):
         )
 
         if verbose:
-            standard = {
-                0: "?",
-                24 / 1: "FILM",
-                25 / 1: "PAL",
-                50 / 1: "PALi",
-                30000 / 1001: "NTSC",
-                60000 / 1001: "NTSCi",
-                24000 / 1001: "NTSC (FILM)"
-            }[self.clip.fps.numerator / self.clip.fps.denominator]
+            standard = get_standard(self.clip.fps.numerator / self.clip.fps.denominator)
             sar = calculate_aspect_ratio(self.clip.width, self.clip.height)
             self.clip = core.text.Text(
                 self.clip,