wip: b30 ocr tab

ui/extends/tabs/tabOcr/tabOcr_B30.py

@@ -0,0 +1,51 @@
+import logging
+
+from arcaea_offline.database import Database
+from arcaea_offline.models import Chart, ScoreInsert
+from arcaea_offline_ocr.b30.chieri.v4.ocr import ChieriBotV4Ocr
+from arcaea_offline_ocr.b30.shared import B30OcrResultItem
+from PySide6.QtGui import QImage
+
+logger = logging.getLogger(__name__)
+
+from ui.extends.components.ocrQueue import OcrRunnable
+
+
+class ChieriV4OcrRunnable(OcrRunnable):
+    def __init__(self, ocr: ChieriBotV4Ocr, component):
+        super().__init__()
+        self.ocr = ocr
+        self.component = component.copy()
+
+    def run(self):
+        try:
+            result = self.ocr.ocr_component(self.component)
+            self.signals.resultReady.emit(result)
+        except Exception:
+            logger.exception("ChieriV4 ocr component error")
+        finally:
+            self.signals.finished.emit()
+
+
+def b30ResultToScoreInsert(_, qImage: QImage, result: B30OcrResultItem):
+    if not result.song_id and not result.title:
+        raise ValueError("no title or song_id")
+
+    db = Database()
+    if not result.song_id:
+        song_id = db.fuzzy_search_song_id(result.title)[0][0]
+    else:
+        song_id = result.song_id
+
+    chart = Chart.from_db_row(db.get_chart(song_id, result.rating_class))
+    score = ScoreInsert(
+        song_id=song_id,
+        rating_class=result.rating_class,
+        score=result.score,
+        time=1485014400,
+        pure=result.pure,
+        far=result.far,
+        lost=result.lost,
+    )
+
+    return (chart, score)