k1x-factorytest/utils/speaker.py
2024-04-22 13:45:53 +08:00

135 lines
3.9 KiB
Python

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QGuiApplication
from PyQt5.QtWidgets import (
QApplication,
QMainWindow,
QFrame,
QVBoxLayout,
QGroupBox,
QLabel,
QStatusBar,
QPushButton,
QGridLayout
)
from cricket.lang import SimpleLang
result = False
class SpeakerTestWindow(SimpleLang):
def __init__(self, play_routine, stop_routine, languages) -> None:
super().__init__()
self.play_routine = play_routine
self.stop_routine = stop_routine
self.languages = languages
self.app = QApplication([])
self.window = QMainWindow()
self.window.setWindowTitle(self._get_text('title'))
# self.window.setWindowFlags(Qt.WindowStaysOnTopHint)
# Main content
content = QFrame(self.window)
content_layout = QVBoxLayout(content)
# Test step
box = QGroupBox(self.get_text('test_step'), content)
box_layout = QVBoxLayout(box)
label = QLabel(self._get_text('test_step'), box)
box_layout.addWidget(label)
content_layout.addWidget(box)
# Status
self.status = QStatusBar(content)
self.status.showMessage('Not running')
content_layout.addWidget(self.status)
# Toolbar
toolbar = QFrame(content)
toolbar_layout = QGridLayout(toolbar)
self.play_button = QPushButton(self.get_text('play_button'), toolbar)
self.play_button.clicked.connect(self.cmd_play)
self.play_button.setFocus()
toolbar_layout.addWidget(self.play_button, 0, 0)
self.stop_button = QPushButton(self.get_text('stop_button'), toolbar)
self.stop_button.setDisabled(True)
self.stop_button.clicked.connect(self.cmd_stop)
toolbar_layout.addWidget(self.stop_button, 0, 1)
self.pass_button = QPushButton(self.get_text('pass_button'), toolbar)
self.pass_button.setDisabled(True)
self.pass_button.clicked.connect(self.cmd_pass)
toolbar_layout.addWidget(self.pass_button, 0, 2)
self.fail_button = QPushButton(self.get_text('fail_button'), toolbar)
self.fail_button.setDisabled(True)
self.fail_button.clicked.connect(self.cmd_fail)
toolbar_layout.addWidget(self.fail_button, 0, 3)
content_layout.addWidget(toolbar)
self.window.setCentralWidget(content)
# Move to center
# screen = QGuiApplication.primaryScreen()
# if screen:
# screen_geometry = screen.geometry()
# window_geometry = self.window.geometry()
# x = (screen_geometry.width() - window_geometry.width()) // 2
# y = (screen_geometry.height() - window_geometry.height()) // 2
# self.window.move(x, y)
# self.window.show()
self.window.showFullScreen()
self.app.exec_()
def _get_text(self, key):
if self.languages:
lang = self.languages.get(self.current_lang)
if lang is not None:
text = lang.get(key)
if text is not None:
return text
return 'Undefined'
def cmd_play(self):
self.status.showMessage('Running...')
self.play_button.setDisabled(True)
self.play_routine()
self.stop_button.setDisabled(False)
self.pass_button.setDisabled(False)
self.fail_button.setDisabled(False)
def cmd_stop(self):
self.status.showMessage('Stopping...')
self.stop_button.setDisabled(True)
self.stop_routine()
self.play_button.setDisabled(False)
self.status.showMessage('Stopped')
def cmd_pass(self):
global result
self.status.showMessage('Stopping...')
self.stop_routine()
result = True
self.app.quit()
def cmd_fail(self):
global result
self.status.showMessage('Stopping...')
self.stop_routine()
result = False
self.app.quit()
def GetTestResult():
return result