init

ui/implements/tabs/tabAbout.py

@@ -0,0 +1,23 @@
+from PySide6.QtCore import Qt, Slot
+from PySide6.QtGui import QPixmap
+from PySide6.QtWidgets import QMessageBox, QWidget
+
+from ui.designer.tabs.tabAbout_ui import Ui_TabAbout
+
+
+class TabAbout(Ui_TabAbout, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+        logoPixmap = QPixmap(":/images/logo.png").scaled(
+            300,
+            300,
+            Qt.AspectRatioMode.KeepAspectRatio,
+            Qt.TransformationMode.SmoothTransformation,
+        )
+        self.logoLabel.setPixmap(logoPixmap)
+
+    @Slot()
+    def on_aboutQtButton_clicked(self):
+        QMessageBox.aboutQt(self)

ui/implements/tabs/tabDb/tabDb_Manage.py

@@ -0,0 +1,30 @@
+import logging
+import traceback
+
+from arcaea_offline.database import Database
+from PySide6.QtCore import Slot
+from PySide6.QtWidgets import QFileDialog, QMessageBox, QWidget
+
+from ui.designer.tabs.tabDb.tabDb_Manage_ui import Ui_TabDb_Manage
+
+logger = logging.getLogger(__name__)
+
+
+class TabDb_Manage(Ui_TabDb_Manage, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+    @Slot()
+    def on_syncArcSongDbButton_clicked(self):
+        dbFile, filter = QFileDialog.getOpenFileName(
+            self, None, "", "DB File (*.db);;*"
+        )
+        try:
+            Database().update_arcsong_db(dbFile)
+            QMessageBox.information(self, "OK", "OK")
+        except Exception as e:
+            logging.exception("Sync arcsong.db error")
+            QMessageBox.critical(
+                self, "Sync Error", "\n".join(traceback.format_exception(e))
+            )

ui/implements/tabs/tabDb/tabDb_ScoreTableViewer.py

@@ -0,0 +1,114 @@
+from arcaea_offline.models import ScoreInsert
+from PySide6.QtCore import QModelIndex, Qt, Slot
+from PySide6.QtGui import QColor, QPalette
+from PySide6.QtWidgets import QMessageBox
+
+from ui.extends.shared.delegates.chartDelegate import ChartDelegate
+from ui.extends.shared.delegates.scoreDelegate import ScoreDelegate
+from ui.extends.shared.models.tables.score import (
+    DbScoreTableModel,
+    DbScoreTableSortFilterProxyModel,
+)
+from ui.implements.components.dbTableViewer import DbTableViewer
+
+
+class TableChartDelegate(ChartDelegate):
+    def getChart(self, index):
+        return index.data(DbScoreTableModel.ChartRole)
+
+
+class TableScoreDelegate(ScoreDelegate):
+    def getChart(self, index):
+        return index.data(DbScoreTableModel.ChartRole)
+
+    def getScoreInsert(self, index: QModelIndex) -> ScoreInsert | None:
+        return super().getScoreInsert(index)
+
+    def getScore(self, index):
+        return index.data(DbScoreTableModel.ScoreRole)
+
+    def setModelData(self, editor, model, index):
+        if super().confirmSetModelData(editor):
+            model.setData(index, editor.value(), DbScoreTableModel.ScoreRole)
+
+
+class DbScoreTableViewer(DbTableViewer):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+
+        self.tableModel = DbScoreTableModel(self)
+        self.tableProxyModel = DbScoreTableSortFilterProxyModel(self)
+        self.tableProxyModel.setSourceModel(self.tableModel)
+        self.tableView.setModel(self.tableProxyModel)
+        self.tableView.setItemDelegateForColumn(1, TableChartDelegate(self.tableView))
+        self.tableView.setItemDelegateForColumn(2, TableScoreDelegate(self.tableView))
+
+        tableViewPalette = QPalette(self.tableView.palette())
+        highlightColor = QColor(tableViewPalette.color(QPalette.ColorRole.Highlight))
+        highlightColor.setAlpha(25)
+        tableViewPalette.setColor(QPalette.ColorRole.Highlight, highlightColor)
+        self.tableView.setPalette(tableViewPalette)
+        self.tableModel.dataChanged.connect(self.resizeTableView)
+
+        self.fillSortComboBox()
+
+    def fillSortComboBox(self):
+        self.sort_comboBox.addItem("ID", [0, 1])
+        self.sort_comboBox.addItem(
+            "Score", [2, DbScoreTableSortFilterProxyModel.Sort_C2_ScoreRole]
+        )
+        self.sort_comboBox.addItem(
+            "Time", [2, DbScoreTableSortFilterProxyModel.Sort_C2_TimeRole]
+        )
+        self.sort_comboBox.addItem("Potential", [3, 1])
+        self.sort_comboBox.setCurrentIndex(0)
+        self.on_sort_comboBox_activated()
+
+    @Slot()
+    def resizeTableView(self):
+        self.tableView.resizeRowsToContents()
+        self.tableView.resizeColumnsToContents()
+
+    @Slot()
+    def on_sort_comboBox_activated(self):
+        self.sortProxyModel()
+
+    @Slot()
+    def on_sort_descendingCheckBox_toggled(self):
+        self.sortProxyModel()
+
+    @Slot()
+    def sortProxyModel(self):
+        if self.sort_comboBox.currentIndex() > -1:
+            column, role = self.sort_comboBox.currentData()
+            self.tableProxyModel.setSortRole(role)
+            self.tableProxyModel.sort(
+                column,
+                Qt.SortOrder.DescendingOrder
+                if self.sort_descendingCheckBox.isChecked()
+                else Qt.SortOrder.AscendingOrder,
+            )
+
+    @Slot()
+    def on_action_removeSelectedButton_clicked(self):
+        rows = [
+            srcIndex.row()
+            for srcIndex in [
+                self.tableProxyModel.mapToSource(proxyIndex)
+                for proxyIndex in self.tableView.selectionModel().selectedRows()
+            ]
+        ]
+        result = QMessageBox.warning(
+            self,
+            "Warning",
+            f"Removing {len(rows)} row(s). Are you sure?",
+            QMessageBox.StandardButton.Yes,
+            QMessageBox.StandardButton.No,
+        )
+        if result == QMessageBox.StandardButton.Yes:
+            self.tableModel.removeRowList(rows)
+
+    @Slot()
+    def on_refreshButton_clicked(self):
+        self.tableModel.syncDb()
+        self.resizeTableView()

ui/implements/tabs/tabDbEntry.py

@@ -0,0 +1,16 @@
+from PySide6.QtCore import QCoreApplication
+from PySide6.QtWidgets import QWidget
+
+from ui.designer.tabs.tabDbEntry_ui import Ui_TabDbEntry
+from ui.implements.tabs.tabDb.tabDb_ScoreTableViewer import DbScoreTableViewer
+
+
+class TabDbEntry(Ui_TabDbEntry, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+        self.tabWidget.addTab(
+            DbScoreTableViewer(self),
+            QCoreApplication.translate("TabDbEntry", "tab.scoreTableViewer"),
+        )

ui/implements/tabs/tabInputScore.py

@@ -0,0 +1,33 @@
+import traceback
+
+from arcaea_offline.database import Database
+from PySide6.QtCore import QCoreApplication, QModelIndex
+from PySide6.QtWidgets import QMessageBox, QWidget
+
+from ui.designer.tabs.tabInputScore_ui import Ui_TabInputScore
+
+
+class TabInputScore(Ui_TabInputScore, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+        self.chartSelector.valueChanged.connect(self.updateScoreEditorChart)
+        self.scoreEditor.accepted.connect(self.commit)
+
+    def updateScoreEditorChart(self):
+        chart = self.chartSelector.value()
+        self.scoreEditor.setChart(chart)
+
+    def commit(self):
+        try:
+            Database().insert_score(self.scoreEditor.value())
+            self.scoreEditor.reset()
+        except Exception as e:
+            QMessageBox.critical(
+                self,
+                # fmt: off
+                QCoreApplication.translate("General", "tracebackFormatExceptionOnly.title"),
+                QCoreApplication.translate("General", "tracebackFormatExceptionOnly.content").format(traceback.format_exception_only(e))
+                # fmt: on
+            )

ui/implements/tabs/tabOcr.py

@@ -0,0 +1,133 @@
+import pytesseract
+from arcaea_offline_ocr_device_creation_wizard.implements.wizard import Wizard
+from PySide6.QtCore import QModelIndex, Qt, Slot
+from PySide6.QtGui import QColor, QPalette
+from PySide6.QtWidgets import QFileDialog, QWidget
+
+from ui.designer.tabs.tabOcr_ui import Ui_TabOcr
+from ui.extends.settings import Settings
+from ui.extends.tabs.tabOcr import (
+    ImageDelegate,
+    OcrQueueModel,
+    OcrQueueTableProxyModel,
+    TableChartDelegate,
+    TableScoreDelegate,
+)
+
+
+class TabOcr(Ui_TabOcr, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+        self.deviceFileSelector.filesSelected.connect(self.deviceFileSelected)
+        self.tesseractFileSelector.filesSelected.connect(
+            self.tesseractFileSelectorFilesSelected
+        )
+
+        settings = Settings()
+        self.deviceFileSelector.selectFile(settings.devicesJsonFile())
+        self.tesseractFileSelector.selectFile(settings.tesseractPath())
+        self.deviceComboBox.selectDevice(settings.deviceUuid())
+
+        self.ocrQueueModel = OcrQueueModel(self)
+        self.ocrQueueModel.dataChanged.connect(self.resizeViewWhenScoreChanged)
+        self.ocrQueueModel.started.connect(self.ocrStarted)
+        self.ocrQueueModel.finished.connect(self.ocrFinished)
+        self.ocrQueueProxyModel = OcrQueueTableProxyModel(self)
+        self.ocrQueueProxyModel.setSourceModel(self.ocrQueueModel)
+
+        self.tableView.setModel(self.ocrQueueProxyModel)
+        self.tableView.setItemDelegateForColumn(1, ImageDelegate(self.tableView))
+        self.tableView.setItemDelegateForColumn(2, TableChartDelegate(self.tableView))
+        self.tableView.setItemDelegateForColumn(3, TableScoreDelegate(self.tableView))
+
+        tableViewPalette = QPalette(self.tableView.palette())
+        highlightColor = QColor(tableViewPalette.color(QPalette.ColorRole.Highlight))
+        highlightColor.setAlpha(25)
+        tableViewPalette.setColor(QPalette.ColorRole.Highlight, highlightColor)
+        self.tableView.setPalette(tableViewPalette)
+
+    @Slot(QModelIndex, QModelIndex, list)
+    def resizeViewWhenScoreChanged(
+        self, topleft: QModelIndex, bottomRight: QModelIndex, roles: list[int]
+    ):
+        if OcrQueueModel.ScoreInsertRole in roles:
+            rows = [*range(topleft.row(), bottomRight.row() + 1)]
+            [self.tableView.resizeRowToContents(row) for row in rows]
+            self.tableView.resizeColumnsToContents()
+
+    @Slot()
+    def on_openWizardButton_clicked(self):
+        wizard = Wizard(self)
+        wizard.open()
+
+    def deviceFileSelected(self):
+        selectedFiles = self.deviceFileSelector.selectedFiles()
+        if selectedFiles:
+            file = selectedFiles[0]
+            self.deviceComboBox.loadDevicesJson(file)
+
+    def tesseractFileSelectorFilesSelected(self):
+        selectedFiles = self.tesseractFileSelector.selectedFiles()
+        if selectedFiles:
+            pytesseract.pytesseract.tesseract_cmd = selectedFiles[0]
+
+    def setOcrButtonsEnabled(self, __bool: bool):
+        self.ocr_addImageButton.setEnabled(__bool)
+        self.ocr_removeSelectedButton.setEnabled(__bool)
+        self.ocr_removeAllButton.setEnabled(__bool)
+        self.ocr_startButton.setEnabled(__bool)
+        self.ocr_acceptSelectedButton.setEnabled(__bool)
+        self.ocr_acceptAllButton.setEnabled(__bool)
+        self.ocr_ignoreValidateCheckBox.setEnabled(__bool)
+
+    @Slot()
+    def on_ocr_addImageButton_clicked(self):
+        files, _filter = QFileDialog.getOpenFileNames(
+            self, None, "", "Image Files (*.png *.jpg *.jpeg *.bmp *.webp);;*"
+        )
+        for file in files:
+            self.ocrQueueModel.addItem(file)
+        self.tableView.resizeRowsToContents()
+        self.tableView.resizeColumnsToContents()
+
+    @Slot()
+    def on_ocr_startButton_clicked(self):
+        self.ocrQueueModel.startQueue(self.deviceComboBox.currentData())
+
+    def ocrStarted(self):
+        self.setOcrButtonsEnabled(False)
+
+    def ocrFinished(self):
+        self.setOcrButtonsEnabled(True)
+
+    @Slot()
+    def on_ocr_removeSelectedButton_clicked(self):
+        rows = [
+            modelIndex.row()
+            for modelIndex in self.tableView.selectionModel().selectedRows(0)
+        ]
+        self.ocrQueueModel.removeItems(rows)
+
+    @Slot()
+    def on_ocr_removeAllButton_clicked(self):
+        self.ocrQueueModel.clear()
+
+    @Slot()
+    def on_ocr_acceptSelectedButton_clicked(self):
+        ignoreValidate = (
+            self.ocr_ignoreValidateCheckBox.checkState() == Qt.CheckState.Checked
+        )
+        rows = [
+            modelIndex.row()
+            for modelIndex in self.tableView.selectionModel().selectedRows(0)
+        ]
+        self.ocrQueueModel.acceptItems(rows, ignoreValidate)
+
+    @Slot()
+    def on_ocr_acceptAllButton_clicked(self):
+        ignoreValidate = (
+            self.ocr_ignoreValidateCheckBox.checkState() == Qt.CheckState.Checked
+        )
+        self.ocrQueueModel.acceptAllItems(ignoreValidate)

ui/implements/tabs/tabOcrDisabled.py

@@ -0,0 +1,9 @@
+from PySide6.QtWidgets import QWidget
+
+from ui.designer.tabs.tabOcrDisabled_ui import Ui_TabOcrDisabled
+
+
+class TabOcrDisabled(Ui_TabOcrDisabled, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)

ui/implements/tabs/tabOverview.py

@@ -0,0 +1,18 @@
+from arcaea_offline.database import Database
+from PySide6.QtWidgets import QWidget
+
+from ui.designer.tabs.tabOverview_ui import Ui_TabOverview
+
+
+class TabOverview(Ui_TabOverview, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+        self.db = Database()
+        self.db.register_update_hook(self.updateOverview)
+        self.updateOverview()
+
+    def updateOverview(self):
+        b30 = self.db.get_b30() or 0.00
+        self.b30Label.setText(str(f"{b30:.3f}"))

ui/implements/tabs/tabSettings.py

@@ -0,0 +1,23 @@
+from PySide6.QtCore import QModelIndex, Slot
+from PySide6.QtWidgets import QListWidgetItem, QWidget
+
+from ui.designer.tabs.tabSettings_ui import Ui_TabSettings
+
+
+class SettingsEntryItem(QListWidgetItem):
+    pass
+
+
+class TabSettings(Ui_TabSettings, QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+
+        self.listWidget.addItem("Default")
+        self.listWidget.activated.connect(self.switchPage)
+
+        self.listWidget.setCurrentRow(self.stackedWidget.currentIndex())
+
+    @Slot(QModelIndex)
+    def switchPage(self, index: QModelIndex):
+        self.stackedWidget.setCurrentIndex(index.row())