wip: ScoreEditor

This commit is contained in:
2023-09-01 06:45:12 +08:00
parent af178744b8
commit 565260d6ce
5 changed files with 741 additions and 274 deletions

View File

@ -1,10 +1,18 @@
from enum import IntEnum
from typing import Optional
from typing import Any, Optional
from arcaea_offline.calculate import calculate_score_range
from arcaea_offline.models import Chart, Score
from PySide6.QtCore import QCoreApplication, QDateTime, Signal, Slot
from PySide6.QtWidgets import QMessageBox, QWidget
from PySide6.QtWidgets import (
QCheckBox,
QComboBox,
QDateTimeEdit,
QLineEdit,
QMessageBox,
QSpinBox,
QWidget,
)
from ui.designer.components.scoreEditor_ui import Ui_ScoreEditor
@ -14,6 +22,7 @@ class ScoreValidateResult(IntEnum):
ScoreMismatch = 1
ScoreEmpty = 2
ChartInvalid = 50
ScoreIncomplete = 100
class ScoreEditor(Ui_ScoreEditor, QWidget):
@ -26,6 +35,7 @@ class ScoreEditor(Ui_ScoreEditor, QWidget):
self.__validateBeforeAccept = True
self.__chart = None
self.__score_id = None
self.scoreLineEdit.textChanged.connect(self.valueChanged)
self.pureSpinBox.valueChanged.connect(self.valueChanged)
@ -33,22 +43,41 @@ class ScoreEditor(Ui_ScoreEditor, QWidget):
self.lostSpinBox.valueChanged.connect(self.valueChanged)
self.dateTimeEdit.dateTimeChanged.connect(self.valueChanged)
self.maxRecallSpinBox.valueChanged.connect(self.valueChanged)
self.modifierComboBox.currentIndexChanged.connect(self.valueChanged)
self.clearTypeComboBox.currentIndexChanged.connect(self.valueChanged)
self.commentLineEdit.textChanged.connect(self.valueChanged)
self.pureNoneCheckBox.toggled.connect(self.valueChanged)
self.farNoneCheckBox.toggled.connect(self.valueChanged)
self.lostNoneCheckBox.toggled.connect(self.valueChanged)
self.dateNoneCheckBox.toggled.connect(self.valueChanged)
self.maxRecallNoneCheckBox.toggled.connect(self.valueChanged)
self.modifierNoneCheckBox.toggled.connect(self.valueChanged)
self.clearTypeNoneCheckBox.toggled.connect(self.valueChanged)
self.commentNoneCheckBox.toggled.connect(self.valueChanged)
self.valueChanged.connect(self.validateScore)
self.valueChanged.connect(self.updateValidateLabel)
self.modifierComboBox.addItem("NORMAL", 0)
self.modifierComboBox.addItem("EASY", 1)
self.modifierComboBox.addItem("HARD", 2)
self.modifierComboBox.setCurrentIndex(-1)
self.clearTypeComboBox.addItem("TRACK LOST", 0)
self.clearTypeComboBox.addItem("TRACK COMPLETE", 1)
self.clearTypeComboBox.addItem("HARD LOST", 2)
self.clearTypeComboBox.addItem("NORMAL CLEAR", 1)
self.clearTypeComboBox.addItem("FULL RECALL", 2)
self.clearTypeComboBox.addItem("PURE MEMORY", 3)
self.clearTypeComboBox.addItem("EASY CLEAR", 4)
self.clearTypeComboBox.addItem("HARD CLEAR", 5)
self.clearTypeComboBox.setCurrentIndex(-1)
self.dateTimeEdit.setDateTime(QDateTime.currentDateTime())
def setValidateBeforeAccept(self, __bool: bool):
self.__validateBeforeAccept = __bool
def triggerValidateMessageBox(self):
validate = self.validateScore()
if validate == ScoreValidateResult.Ok:
if validate in [ScoreValidateResult.Ok, ScoreValidateResult.ScoreIncomplete]:
return True
if validate == ScoreValidateResult.ChartInvalid:
QMessageBox.critical(
@ -132,9 +161,12 @@ class ScoreEditor(Ui_ScoreEditor, QWidget):
score = self.value()
if score.pure is None or score.far is None:
return ScoreValidateResult.ScoreIncomplete
score_range = calculate_score_range(self.__chart.notes, score.pure, score.far)
score_in_range = score_range[0] <= score.score <= score_range[1]
note_in_range = score.pure + score.far + score.lost <= self.__chart.notes
score_in_range = score_range[0] <= score.score <= score_range[1]
if not score_in_range or not note_in_range:
return ScoreValidateResult.ScoreMismatch
if score.score == 0:
@ -152,37 +184,114 @@ class ScoreEditor(Ui_ScoreEditor, QWidget):
text = QCoreApplication.translate("ScoreEditor", "validate.scoreMismatch")
elif validate == ScoreValidateResult.ScoreEmpty:
text = QCoreApplication.translate("ScoreEditor", "validate.scoreEmpty")
elif validate == ScoreValidateResult.ScoreIncomplete:
text = QCoreApplication.translate("ScoreEditor", "validate.scoreIncomplete")
else:
text = QCoreApplication.translate("ScoreEditor", "validate.unknownState")
self.validateLabel.setText(text)
def value(self):
if isinstance(self.__chart, Chart):
return Score(
song_id=self.__chart.song_id,
rating_class=self.__chart.rating_class,
score=self.score(),
pure=self.pureSpinBox.value(),
far=self.farSpinBox.value(),
lost=self.lostSpinBox.value(),
date=self.dateTimeEdit.dateTime().toSecsSinceEpoch(),
max_recall=self.maxRecallSpinBox.value()
if self.maxRecallSpinBox.value() > -1
else None,
def __getItemBaseName(self, item: QLineEdit | QSpinBox | QDateTimeEdit | QComboBox):
if isinstance(item, QSpinBox):
return item.objectName().replace("SpinBox", "")
elif isinstance(item, QLineEdit):
if item.objectName() == "scoreLineEdit":
return "score"
return item.objectName().replace("LineEdit", "")
elif isinstance(item, QComboBox):
return item.objectName().replace("ComboBox", "")
elif isinstance(item, QDateTimeEdit):
return "date"
def __getItemNoneCheckBox(self, itemBaseName: str) -> QCheckBox | None:
return self.findChild(QCheckBox, f"{itemBaseName}NoneCheckBox")
def __getItemEnabled(self, itemBaseName: str):
return not self.__getItemNoneCheckBox(itemBaseName).isChecked()
def getItemValue(self, item: QLineEdit | QSpinBox | QDateTimeEdit | QComboBox):
if isinstance(item, QDateTimeEdit) and item.objectName() == "dateTimeEdit":
return (
None
if self.dateNoneCheckBox.isChecked()
else self.dateTimeEdit.dateTime().toSecsSinceEpoch()
)
itemBaseName = self.__getItemBaseName(item)
itemEnabled = self.__getItemEnabled(itemBaseName)
if isinstance(item, QSpinBox):
return item.value() if itemEnabled else None
elif isinstance(item, QLineEdit):
return item.text() if itemEnabled else None
elif isinstance(item, QComboBox):
return item.currentData() if itemEnabled else None
def value(self):
if not isinstance(self.__chart, Chart):
return
score = Score(
song_id=self.__chart.song_id, rating_class=self.__chart.rating_class
)
if self.__score_id is not None:
score.id = self.__score_id
score.score = self.score()
score.pure = self.getItemValue(self.pureSpinBox)
score.far = self.getItemValue(self.farSpinBox)
score.lost = self.getItemValue(self.lostSpinBox)
score.date = self.getItemValue(self.dateTimeEdit)
score.max_recall = self.getItemValue(self.maxRecallSpinBox)
score.modifier = self.getItemValue(self.modifierComboBox)
score.clear_type = self.getItemValue(self.clearTypeComboBox)
score.comment = self.getItemValue(self.commentLineEdit)
return score
def setItemValue(
self, item: QLineEdit | QSpinBox | QDateTimeEdit | QComboBox, value: Any
):
if isinstance(item, QDateTimeEdit) and item.objectName() == "dateTimeEdit":
if value is None:
self.dateNoneCheckBox.setChecked(True)
else:
self.dateNoneCheckBox.setChecked(False)
self.dateTimeEdit.setDateTime(QDateTime.fromSecsSinceEpoch(value))
itemBaseName = self.__getItemBaseName(item)
itemNoneCheckBox = self.__getItemNoneCheckBox(itemBaseName)
if value is None:
itemNoneCheckBox.setChecked(True)
return
else:
itemNoneCheckBox.setChecked(False)
if isinstance(item, QSpinBox):
item.setValue(value)
elif isinstance(item, QLineEdit):
item.setText(value)
elif isinstance(item, QComboBox):
item.setCurrentIndex(value)
def setValue(self, score: Score):
if isinstance(score, Score):
scoreText = str(score.score)
scoreText = scoreText.rjust(8, "0")
self.scoreLineEdit.setText(scoreText)
self.pureSpinBox.setValue(score.pure)
self.farSpinBox.setValue(score.far)
self.lostSpinBox.setValue(score.lost)
self.dateTimeEdit.setDateTime(QDateTime.fromSecsSinceEpoch(score.date))
if score.max_recall is not None:
self.maxRecallSpinBox.setValue(score.max_recall)
if not isinstance(score, Score):
return
if score.id is not None:
self.__score_id = score.id
self.idLabel.setText(str(self.__score_id))
scoreText = str(score.score)
scoreText = scoreText.rjust(8, "0")
self.scoreLineEdit.setText(scoreText)
self.setItemValue(self.pureSpinBox, score.pure)
self.setItemValue(self.farSpinBox, score.far)
self.setItemValue(self.lostSpinBox, score.lost)
self.setItemValue(self.dateTimeEdit, score.date)
self.setItemValue(self.maxRecallSpinBox, score.max_recall)
self.setItemValue(self.modifierComboBox, score.modifier)
self.setItemValue(self.clearTypeComboBox, score.clear_type)
self.setItemValue(self.commentLineEdit, score.comment)
def reset(self):
self.setChart(None)
@ -190,5 +299,7 @@ class ScoreEditor(Ui_ScoreEditor, QWidget):
self.pureSpinBox.setValue(0)
self.farSpinBox.setValue(0)
self.lostSpinBox.setValue(0)
self.maxRecallSpinBox.setValue(-1)
self.maxRecallSpinBox.setValue(0)
self.modifierComboBox.setCurrentIndex(-1)
self.clearTypeComboBox.setCurrentIndex(-1)
self.commentLineEdit.setText("")