wip: Chieri V4 B30 ocr

src/arcaea_offline_ocr/b30/chieri/v4/rois.py

@@ -0,0 +1,132 @@
+from typing import List, Optional
+
+from ....crop import crop_xywh
+from ....types import Mat, XYWHRect
+from ....utils import apply_factor, construct_int_xywh_rect
+
+
+class ChieriBotV4ComponentRois:
+    def __init__(self, factor: Optional[float] = 1.0):
+        self.__factor = factor
+
+    @property
+    def factor(self):
+        return self.__factor
+
+    @factor.setter
+    def factor(self, factor: float):
+        self.__factor = factor
+
+    @property
+    def top_font_color_detect(self):
+        return apply_factor((35, 10, 120, 100), self.factor)
+
+    @property
+    def bottom_font_color_detect(self):
+        return apply_factor((30, 125, 175, 110), self.factor)
+
+    @property
+    def bg_point(self):
+        return apply_factor((75, 10), self.factor)
+
+    @property
+    def rating_class_rect(self):
+        return apply_factor((21, 40, 7, 20), self.factor)
+
+    @property
+    def title_rect(self):
+        return apply_factor((35, 10, 430, 50), self.factor)
+
+    @property
+    def score_rect(self):
+        return apply_factor((30, 60, 270, 55), self.factor)
+
+    @property
+    def pfl_rect(self):
+        return apply_factor((50, 125, 80, 100), self.factor)
+
+    @property
+    def date_rect(self):
+        return apply_factor((205, 200, 225, 25), self.factor)
+
+
+class ChieriBotV4Rois:
+    def __init__(self, factor: Optional[float] = 1.0):
+        self.__factor = factor
+        self.__component_rois = ChieriBotV4ComponentRois(factor)
+
+    @property
+    def component_rois(self):
+        return self.__component_rois
+
+    @property
+    def factor(self):
+        return self.__factor
+
+    @factor.setter
+    def factor(self, factor: float):
+        self.__factor = factor
+        self.__component_rois.factor = factor
+
+    @property
+    def top(self):
+        return apply_factor(823, self.factor)
+
+    @property
+    def left(self):
+        return apply_factor(107, self.factor)
+
+    @property
+    def width(self):
+        return apply_factor(502, self.factor)
+
+    @property
+    def height(self):
+        return apply_factor(240, self.factor)
+
+    @property
+    def vertical_gap(self):
+        return apply_factor(74, self.factor)
+
+    @property
+    def horizontal_gap(self):
+        return apply_factor(40, self.factor)
+
+    @property
+    def horizontal_items(self):
+        return 3
+
+    @property
+    def vertical_items(self):
+        return 10
+
+    @property
+    def b33_vertical_gap(self):
+        return apply_factor(121, self.factor)
+
+    def components(self, img_bgr: Mat) -> List[Mat]:
+        first_rect = XYWHRect(x=self.left, y=self.top, w=self.width, h=self.height)
+        results = []
+
+        for vi in range(self.vertical_items):
+            rect = XYWHRect(*first_rect)
+            rect += (0, (self.vertical_gap + self.height) * vi, 0, 0)
+            for hi in range(self.horizontal_items):
+                if hi > 0:
+                    rect += ((self.width + self.horizontal_gap), 0, 0, 0)
+                int_rect = construct_int_xywh_rect(rect)
+                results.append(crop_xywh(img_bgr, int_rect))
+
+        rect += (
+            -(self.width + self.horizontal_gap) * 2,
+            self.height + self.b33_vertical_gap,
+            0,
+            0,
+        )
+        for hi in range(self.horizontal_items):
+            if hi > 0:
+                rect += ((self.width + self.horizontal_gap), 0, 0, 0)
+            int_rect = construct_int_xywh_rect(rect)
+            results.append(crop_xywh(img_bgr, int_rect))
+
+        return results