init

ui/extends/shared/delegates/base.py

@@ -0,0 +1,152 @@
+from typing import Callable
+
+from PySide6.QtCore import QEvent, QModelIndex, QObject, QPoint, QSize, Qt
+from PySide6.QtGui import QBrush, QColor, QFont, QFontMetrics, QLinearGradient, QPainter
+from PySide6.QtWidgets import QApplication, QStyledItemDelegate, QStyleOptionViewItem
+
+
+class TextSegmentDelegate(QStyledItemDelegate):
+    VerticalPadding = 3
+    HorizontalPadding = 5
+
+    TextRole = 3375
+    ColorRole = TextRole + 1
+    BrushRole = TextRole + 2
+    GradientWrapperRole = TextRole + 3
+    FontRole = TextRole + 20
+
+    def getTextSegments(
+        self, index: QModelIndex, option
+    ) -> list[
+        list[
+            dict[
+                int,
+                str
+                | QColor
+                | QBrush
+                | Callable[[float, float, float, float], QLinearGradient]
+                | QFont,
+            ]
+        ]
+    ]:
+        return []
+
+    def sizeHint(self, option, index) -> QSize:
+        width = 0
+        height = self.VerticalPadding
+        fm: QFontMetrics = option.fontMetrics
+        for line in self.getTextSegments(index, option):
+            lineWidth = 4 * self.HorizontalPadding
+            lineHeight = 0
+            for textFrag in line:
+                font = textFrag.get(self.FontRole)
+                _fm = QFontMetrics(font) if font else fm
+                text = textFrag[self.TextRole]
+                textWidth = _fm.horizontalAdvance(text)
+                textHeight = _fm.height()
+                lineWidth += textWidth
+                lineHeight = max(lineHeight, textHeight)
+            width = max(lineWidth, width)
+            height += lineHeight + self.VerticalPadding
+        return QSize(width, height)
+
+    def paint(
+        self, painter: QPainter, option: QStyleOptionViewItem, index: QModelIndex
+    ):
+        self.initStyleOption(option, index)
+        # draw text only
+        baseX = option.rect.x() + self.HorizontalPadding
+        baseY = option.rect.y() + self.VerticalPadding
+        maxWidth = option.rect.width() - (2 * self.HorizontalPadding)
+        fm: QFontMetrics = option.fontMetrics
+        painter.save()
+        for line in self.getTextSegments(index, option):
+            lineBaseX = baseX
+            lineBaseY = baseY
+            lineHeight = 0
+            for textFrag in line:
+                painter.save()
+                # elide text, get font values
+                text = textFrag[self.TextRole]
+                fragMaxWidth = maxWidth - (lineBaseX - baseX)
+                font = textFrag.get(self.FontRole)
+                if font:
+                    painter.setFont(font)
+                    _fm = QFontMetrics(font)
+                else:
+                    _fm = fm
+                lineHeight = max(lineHeight, _fm.height())
+                elidedText = _fm.elidedText(
+                    text, Qt.TextElideMode.ElideRight, fragMaxWidth
+                )
+
+                # confirm proper color
+                brush = textFrag.get(self.BrushRole)
+                gradientWrapper = textFrag.get(self.GradientWrapperRole)
+                color = textFrag.get(self.ColorRole)
+                pen = painter.pen()
+                if brush:
+                    pen.setBrush(brush)
+                elif gradientWrapper:
+                    gradient = gradientWrapper(
+                        lineBaseX,
+                        lineBaseY + lineHeight - _fm.height(),
+                        fragMaxWidth,
+                        _fm.height(),
+                    )
+                    pen.setBrush(gradient)
+                elif color:
+                    pen.setColor(color)
+                painter.setPen(pen)
+
+                painter.drawText(
+                    QPoint(lineBaseX, lineBaseY + lineHeight - _fm.descent()),
+                    elidedText,
+                )
+                painter.restore()
+
+                # if text elided, skip to next line
+                # remember to add height before skipping
+                if _fm.boundingRect(text).width() >= fragMaxWidth:
+                    break
+                lineBaseX += _fm.horizontalAdvance(elidedText)
+
+            baseY += lineHeight + self.VerticalPadding
+        painter.restore()
+
+    def super_styledItemDelegate_paint(self, painter, option, index):
+        return super().paint(painter, option, index)
+
+
+class NoCommitWhenFocusOutEventFilter(QObject):
+    """
+    --DEPRECATED--
+
+    The default QAbstractItemDelegate implementation has a private function
+    `editorEventFilter()`, when editor sends focusOut/hide event, it emits the
+    `commitData(editor)` signal. We don't want this since we need to validate
+    the input, so we filter the event out and handle it by ourselves.
+
+    Reimplement `checkIsEditor(self, val) -> bool` to ensure this filter is
+    working. The default implementation always return `False`.
+    """
+
+    def checkIsEditor(self, val) -> bool:
+        return False
+
+    def eventFilter(self, object: QObject, event: QEvent) -> bool:
+        if self.checkIsEditor(object) and event.type() in [
+            QEvent.Type.FocusOut,
+            QEvent.Type.Hide,
+        ]:
+            widget = QApplication.focusWidget()
+            while widget:
+                # check if focus changed into editor's child
+                if self.checkIsEditor(widget):
+                    return False
+                widget = widget.parentWidget()
+
+            object.hide()
+            object.deleteLater()
+            return True
+        return False

ui/extends/shared/delegates/chartDelegate.py

@@ -0,0 +1,176 @@
+from typing import Union
+
+from arcaea_offline.models import Chart
+from arcaea_offline.utils import rating_class_to_short_text, rating_class_to_text
+from PySide6.QtCore import QDateTime, QModelIndex, Qt, Signal
+from PySide6.QtGui import QBrush, QColor
+from PySide6.QtWidgets import (
+    QFrame,
+    QHBoxLayout,
+    QLabel,
+    QMessageBox,
+    QPushButton,
+    QSizePolicy,
+    QStyleOptionViewItem,
+    QWidget,
+)
+
+from ui.implements.components.chartSelector import ChartSelector
+
+from .base import TextSegmentDelegate
+
+
+def chartToRichText(chart: Chart):
+    if isinstance(chart, Chart):
+        text = f"{chart.name_en} [{rating_class_to_short_text(chart.rating_class)}]"
+        text += "<br>"
+        text += f'<font color="gray">({chart.song_id}, {chart.package_id})</font>'
+    else:
+        text = "(unknown chart)"
+    return text
+
+
+class ChartSelectorDelegateWrapper(ChartSelector):
+    accepted = Signal()
+    rejected = Signal()
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+
+        self.delegateHLine = QFrame(self)
+        self.delegateHLine.setFrameShape(QFrame.Shape.HLine)
+        self.delegateHLine.setFrameShadow(QFrame.Shadow.Plain)
+        self.delegateHLine.setFixedHeight(5)
+        self.mainVerticalLayout.insertWidget(0, self.delegateHLine)
+
+        self.delegateHeader = QWidget(self)
+        self.delegateHeaderHBoxLayout = QHBoxLayout(self.delegateHeader)
+        self.delegateHeaderHBoxLayout.setContentsMargins(0, 0, 0, 0)
+        self.mainVerticalLayout.insertWidget(0, self.delegateHeader)
+
+        self.editorLabel = QLabel(self)
+        self.editorLabel.setSizePolicy(
+            QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Preferred
+        )
+        self.delegateHeaderHBoxLayout.addWidget(self.editorLabel)
+
+        self.editorCommitButton = QPushButton("Commit", self.delegateHeader)
+        self.editorCommitButton.clicked.connect(self.accepted)
+        self.delegateHeaderHBoxLayout.addWidget(self.editorCommitButton)
+
+        self.editorDiscardButton = QPushButton("Discard", self.delegateHeader)
+        self.editorDiscardButton.clicked.connect(self.rejected)
+        self.delegateHeaderHBoxLayout.addWidget(self.editorDiscardButton)
+
+    def setText(self, chart: Chart, _extra: str = None):
+        text = "Editing "
+        text += _extra or ""
+        text += "<br>"
+        text += (
+            chartToRichText(chart) if isinstance(chart, Chart) else "(unknown chart)"
+        )
+        self.editorLabel.setText(text)
+
+    def validate(self):
+        return isinstance(self.value(), Chart)
+
+
+class ChartDelegate(TextSegmentDelegate):
+    RatingClassColors = [
+        QColor("#399bb2"),
+        QColor("#809955"),
+        QColor("#702d60"),
+        QColor("#710f25"),
+    ]
+    ChartInvalidBackgroundColor = QColor("#e6a23c")
+
+    def getChart(self, index: QModelIndex) -> Chart | None:
+        return None
+
+    def getTextSegments(self, index: QModelIndex, option):
+        chart = self.getChart(index)
+        if not isinstance(chart, Chart):
+            return [
+                [{self.TextRole: "Chart Invalid", self.ColorRole: QColor("#ff0000")}]
+            ]
+
+        return [
+            [
+                {self.TextRole: f"{chart.name_en}"},
+            ],
+            [
+                {
+                    self.TextRole: f"{rating_class_to_text(chart.rating_class)} {chart.rating / 10:.1f}",
+                    self.ColorRole: self.RatingClassColors[chart.rating_class],
+                },
+            ],
+            [
+                {
+                    self.TextRole: f"({chart.song_id}, {chart.package_id})",
+                    self.ColorRole: option.widget.palette().placeholderText().color(),
+                },
+            ],
+        ]
+
+    def paintWarningBackground(self, index: QModelIndex) -> bool:
+        return True
+
+    def paint(self, painter, option, index):
+        # draw chartInvalid warning background
+        chart = self.getChart(index)
+        if not isinstance(chart, Chart) and self.paintWarningBackground(index):
+            painter.save()
+            painter.setPen(Qt.PenStyle.NoPen)
+            bgColor = QColor(self.ChartInvalidBackgroundColor)
+            bgColor.setAlpha(50)
+            painter.setBrush(bgColor)
+            painter.drawRect(option.rect)
+            painter.restore()
+        option.text = ""
+        super().paint(painter, option, index)
+
+    def checkIsEditor(self, val):
+        return isinstance(val, ChartSelectorDelegateWrapper)
+
+    def _closeEditor(self):
+        editor = self.sender()
+        self.closeEditor.emit(editor)
+
+    def _commitEditor(self):
+        editor = self.sender()
+        if editor.validate():
+            confirm = QMessageBox.question(
+                editor,
+                "Confirm",
+                f"Are you sure to change chart to<br><br>{chartToRichText(editor.value())}",
+            )
+            if confirm == QMessageBox.StandardButton.Yes:
+                self.commitData.emit(editor)
+                self.closeEditor.emit(editor)
+        else:
+            QMessageBox.critical(editor, "Invalid chart", "Cannot commit")
+
+    def createEditor(
+        self, parent: QWidget, option: QStyleOptionViewItem, index: QModelIndex
+    ) -> ChartSelectorDelegateWrapper:
+        if isinstance(self.getChart(index), Chart):
+            editor = ChartSelectorDelegateWrapper(parent)
+            editor.setWindowFlag(Qt.WindowType.Sheet, True)
+            editor.setWindowFlag(Qt.WindowType.FramelessWindowHint, True)
+            editor.setText(self.getChart(index))
+            editor.move(parent.mapToGlobal(parent.pos()))
+            editor.accepted.connect(self._commitEditor)
+            editor.rejected.connect(self._closeEditor)
+            return editor
+
+    def updateEditorGeometry(self, editor: QWidget, option, index: QModelIndex) -> None:
+        editor.move(editor.pos() + option.rect.topLeft())
+        editor.setMaximumWidth(option.rect.width())
+
+    def setEditorData(self, editor: ChartSelectorDelegateWrapper, index: QModelIndex):
+        if self.checkIsEditor(editor) and isinstance(self.getChart(index), Chart):
+            editor.selectChart(self.getChart(index))
+        return super().setEditorData(editor, index)
+
+    def setModelData(self, editor: ChartSelectorDelegateWrapper, model, index):
+        ...

ui/extends/shared/delegates/descriptionDelegate.py

@@ -0,0 +1,35 @@
+from PySide6.QtCore import QModelIndex, Qt
+from PySide6.QtWidgets import QStyle, QStyleOptionViewItem
+
+from .base import TextSegmentDelegate
+
+
+class DescriptionDelegate(TextSegmentDelegate):
+    MainTextRole = Qt.ItemDataRole.UserRole + 75
+    DescriptionTextRole = Qt.ItemDataRole.UserRole + 76
+
+    def getMainText(self, index: QModelIndex) -> str | None:
+        return index.data(self.MainTextRole)
+
+    def getDescriptionText(self, index: QModelIndex) -> str | None:
+        return index.data(self.DescriptionTextRole)
+
+    def getTextSegments(self, index: QModelIndex, option):
+        return [
+            [
+                {self.TextRole: self.getMainText(index) or ""},
+                {self.TextRole: " "},
+                {
+                    self.TextRole: self.getDescriptionText(index) or "",
+                    self.ColorRole: option.widget.palette().placeholderText().color(),
+                },
+            ]
+        ]
+
+    def paint(self, painter, option, index):
+        super().paint(painter, option, index)
+
+        optionNoText = QStyleOptionViewItem(option)
+        optionNoText.text = ""
+        style = option.widget.style()  # type: QStyle
+        style.drawControl(QStyle.ControlElement.CE_ItemViewItem, optionNoText, painter)

ui/extends/shared/delegates/scoreDelegate.py

@@ -0,0 +1,247 @@
+from typing import Union
+
+from arcaea_offline.calculate import calculate_score_range
+from arcaea_offline.models import Chart, Score, ScoreInsert
+from arcaea_offline.utils import (
+    rating_class_to_text,
+    score_to_grade_text,
+    zip_score_grade,
+)
+from PySide6.QtCore import QAbstractItemModel, QDateTime, QModelIndex, Qt, Signal
+from PySide6.QtGui import QColor, QFont, QLinearGradient
+from PySide6.QtWidgets import (
+    QAbstractItemDelegate,
+    QFrame,
+    QHBoxLayout,
+    QLabel,
+    QPushButton,
+    QSizePolicy,
+    QWidget,
+)
+
+from ui.implements.components.scoreEditor import ScoreEditor
+
+from ..utils import keepWidgetInScreen
+from .base import TextSegmentDelegate
+
+
+class ScoreEditorDelegateWrapper(ScoreEditor):
+    rejected = Signal()
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+
+        self.hLine = QFrame(self)
+        self.hLine.setFrameShape(QFrame.Shape.HLine)
+        self.hLine.setFrameShadow(QFrame.Shadow.Plain)
+        self.hLine.setFixedHeight(5)
+        self.formLayout.insertRow(0, self.hLine)
+
+        self.delegateHeader = QWidget(self)
+        self.delegateHeaderHBoxLayout = QHBoxLayout(self.delegateHeader)
+        self.delegateHeaderHBoxLayout.setContentsMargins(0, 0, 0, 0)
+
+        self.editorLabel = QLabel(self.delegateHeader)
+        self.editorLabel.setSizePolicy(
+            QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Preferred
+        )
+        self.delegateHeaderHBoxLayout.addWidget(self.editorLabel)
+
+        self.editorDiscardButton = QPushButton("Discard", self.delegateHeader)
+        self.editorDiscardButton.clicked.connect(self.rejected)
+        self.delegateHeaderHBoxLayout.addWidget(self.editorDiscardButton)
+
+        self.formLayout.insertRow(0, self.delegateHeader)
+
+    def setText(self, score: Score | ScoreInsert, _extra: str = None):
+        text = "Editing "
+        text += _extra or ""
+        text += f"score {score.score}"
+        text += f"<br>(P{score.pure} F{score.far} L{score.lost} | MR{score.max_recall})"
+        self.editorLabel.setText(text)
+
+
+class ScoreDelegate(TextSegmentDelegate):
+    @staticmethod
+    def createGradeGradientWrapper(topColor: QColor, bottomColor: QColor):
+        def wrapper(x, y, width, height):
+            gradient = QLinearGradient(x + (width / 2), y, x + (width / 2), y + height)
+            gradient.setColorAt(0.1, topColor)
+            gradient.setColorAt(0.9, bottomColor)
+            return gradient
+
+        return wrapper
+
+    ScoreMismatchBackgroundColor = QColor("#e6a23c")
+    PureFarLostColors = [
+        QColor("#f22ec6"),
+        QColor("#ff9028"),
+        QColor("#ff0c43"),
+    ]
+    GradeGradientsWrappers = [  # EX+, EX, AA, A. B, C, D
+        createGradeGradientWrapper(QColor("#83238c"), QColor("#2c72ae")),
+        createGradeGradientWrapper(QColor("#721b6b"), QColor("#295b8d")),
+        createGradeGradientWrapper(QColor("#5a3463"), QColor("#9b4b8d")),
+        createGradeGradientWrapper(QColor("#46324d"), QColor("#92588a")),
+        createGradeGradientWrapper(QColor("#43334a"), QColor("#755b7c")),
+        createGradeGradientWrapper(QColor("#3b2b27"), QColor("#80566b")),
+        createGradeGradientWrapper(QColor("#5d1d35"), QColor("#9f3c55")),
+    ]
+
+    def getScore(self, index: QModelIndex) -> Score | None:
+        return None
+
+    def getScoreInsert(self, index: QModelIndex) -> ScoreInsert | None:
+        return None
+
+    def _getScore(self, index: QModelIndex):
+        score = self.getScore(index)
+        scoreInsert = self.getScoreInsert(index)
+        return scoreInsert if score is None else score
+
+    def getChart(self, index: QModelIndex) -> Chart | None:
+        return None
+
+    def getScoreValidateOk(self, index: QModelIndex) -> bool | None:
+        score = self._getScore(index)
+        chart = self.getChart(index)
+
+        if isinstance(score, (Score, ScoreInsert)) and isinstance(chart, Chart):
+            scoreRange = calculate_score_range(chart, score.pure, score.far)
+            return scoreRange[0] <= score.score <= scoreRange[1]
+
+    def getScoreGradeGradientWrapper(self, score: int):
+        return zip_score_grade(score, self.GradeGradientsWrappers)
+
+    def getTextSegments(self, index, option):
+        score = self._getScore(index)
+        chart = self.getChart(index)
+        if not (isinstance(score, (Score, ScoreInsert)) and isinstance(chart, Chart)):
+            return [
+                [
+                    {
+                        self.TextRole: "Chart/Score Invalid",
+                        self.ColorRole: QColor("#ff0000"),
+                    }
+                ]
+            ]
+
+        score_font = QFont(option.font)
+        score_font.setPointSize(12)
+        score_grade_font = QFont(score_font)
+        score_grade_font.setBold(True)
+        return [
+            [
+                {
+                    self.TextRole: score_to_grade_text(score.score),
+                    self.GradientWrapperRole: self.getScoreGradeGradientWrapper(
+                        score.score
+                    ),
+                    self.FontRole: score_grade_font,
+                },
+                {self.TextRole: " | "},
+                {self.TextRole: str(score.score), self.FontRole: score_font},
+            ],
+            [
+                {
+                    self.TextRole: f"PURE {score.pure}",
+                    self.ColorRole: self.PureFarLostColors[0],
+                },
+                {self.TextRole: "  "},
+                {
+                    self.TextRole: f"FAR {score.far}",
+                    self.ColorRole: self.PureFarLostColors[1],
+                },
+                {self.TextRole: "  "},
+                {
+                    self.TextRole: f"LOST {score.lost}",
+                    self.ColorRole: self.PureFarLostColors[2],
+                },
+                {self.TextRole: " | "},
+                {self.TextRole: f"MAX RECALL {score.max_recall}"},
+            ],
+            [
+                {
+                    self.TextRole: QDateTime.fromSecsSinceEpoch(score.time).toString(
+                        "yyyy-MM-dd hh:mm:ss"
+                    )
+                }
+            ],
+        ]
+
+    def paintWarningBackground(self, index: QModelIndex) -> bool:
+        return True
+
+    def paint(self, painter, option, index):
+        # draw scoreMismatch warning background
+        score = self._getScore(index)
+        chart = self.getChart(index)
+        if (
+            isinstance(score, (Score, ScoreInsert))
+            and isinstance(chart, Chart)
+            and self.paintWarningBackground(index)
+        ):
+            scoreValidateOk = self.getScoreValidateOk(index)
+            if not scoreValidateOk:
+                painter.save()
+                painter.setPen(Qt.PenStyle.NoPen)
+                bgColor = QColor(self.ScoreMismatchBackgroundColor)
+                bgColor.setAlpha(50)
+                painter.setBrush(bgColor)
+                painter.drawRect(option.rect)
+                painter.restore()
+
+        option.text = ""
+        super().paint(painter, option, index)
+
+    def _closeEditor(self):
+        editor = self.sender()
+        self.closeEditor.emit(editor)
+
+    def _commitEditor(self):
+        editor = self.sender()
+        self.commitData.emit(editor)
+        self.closeEditor.emit(editor)
+
+    def createEditor(self, parent, option, index) -> ScoreEditorDelegateWrapper:
+        score = self._getScore(index)
+        chart = self.getChart(index)
+        if isinstance(score, (Score, ScoreInsert)) and isinstance(chart, Chart):
+            editor = ScoreEditorDelegateWrapper(parent)
+            editor.setWindowFlag(Qt.WindowType.Sheet, True)
+            editor.setWindowFlag(Qt.WindowType.FramelessWindowHint, True)
+            editor.setWindowTitle(
+                f"{chart.name_en}({chart.song_id}) | {rating_class_to_text(chart.rating_class)} | {chart.package_id}"
+            )
+            editor.setText(self._getScore(index))
+            editor.setValidateBeforeAccept(False)
+            editor.move(parent.mapToGlobal(parent.pos()))
+            editor.accepted.connect(self._commitEditor)
+            editor.rejected.connect(self._closeEditor)
+            editor.show()
+            return editor
+        return super().createEditor(parent, option, index)
+
+    def updateEditorGeometry(self, editor, option, index):
+        editor.setMaximumWidth(option.rect.width())
+        editor.move(editor.pos() + option.rect.topLeft())
+
+        keepWidgetInScreen(editor)
+
+    def setEditorData(self, editor: ScoreEditorDelegateWrapper, index) -> None:
+        score = self._getScore(index)
+        chart = self.getChart(index)
+        if isinstance(score, (Score, ScoreInsert)) and isinstance(chart, Chart):
+            editor.setChart(chart)
+            editor.setValue(score)
+
+    def confirmSetModelData(self, editor: ScoreEditorDelegateWrapper):
+        return editor.triggerValidateMessageBox()
+
+    def setModelData(
+        self,
+        editor: ScoreEditorDelegateWrapper,
+        model: QAbstractItemModel,
+        index: QModelIndex,
+    ):
+        ...