class Window(QWidget):
keys = {Qt.Key_Left: "left", Qt.Key_Right: "right", Qt.Key_Down: "equal", Qt.Key_Space: "skip"}
def __init__(self):
super().__init__()
# Create window at center of the display.
display_geometry = QDesktopWidget().availableGeometry()
window_size = display_geometry.size() / 2
dx, dy = window_size.width(), window_size.height()
display_geometry = QDesktopWidget().availableGeometry()
display_center = display_geometry.center()
x = display_center.x() - dx / 2
y = display_center.y() - dy / 2
self.setGeometry(int(x), int(y), dx, dy)
self.setFixedSize(dx, dy)
# Initialize statistic and exercise result variable.
self.exercise = Exercise(num_shapes=2)
self.statistics = pd.DataFrame(columns=["time", "is_correct", "pressed", "solution"])
# Set window properties.
self.setWindowTitle('eligo_reactions')
self.show()
def paintEvent(self, e) -> None:
super(Window, self).paintEvent(e)
# If the time is running the previous task has not been solved, so do not update
# the window's content. Only if the timer is None
if self.exercise.task is None:
self.exercise.new_task()
logging.debug(self.exercise)
shapes_dict, shapes_task, task_description, operation = self.exercise.task
self.draw_legend(shapes_dict)
self.draw_exercise(task_description, task_count=self.exercise.counter)
self.draw_task(shapes_task, operation=operation)
def keyPressEvent(self, a0: QKeyEvent) -> None:
if a0.key() not in self.keys.keys():
return
time_needed = self.exercise.time_needed()
key_pressed = self.keys[a0.key()]
is_correct = (key_pressed == self.exercise.solution)
statistic = {"time": time_needed, "is_correct": is_correct,
"pressed": key_pressed, "solution": self.exercise.solution}
self.statistics = self.statistics.append(statistic, ignore_index=True)
self.exercise.reset_task()
self.update()
def closeEvent(self, a0: QCloseEvent) -> None:
logging.info("--- Statistics ---")
logging.info(self.statistics)
logging.info(self.statistics.mean())
################################################################################
# Drawing ######################################################################
################################################################################
def draw_exercise(self, task_description: str, task_count: int):
window_size = self.get_size()
text = f"Exercise {task_count}\nWelche Seite ist {task_description}?"
painter = QPainter()
painter.begin(self)
painter.setFont(QFont('Times', 35))
painter.drawText(QRect(0, 0, window_size[0], window_size[1] / 6), Qt.AlignCenter | Qt.AlignCenter, text)
painter.end()
def draw_task(self, task: typing.List[typing.List[Shape]], operation: str, size: int = 30):
window_size = self.get_size()
num_sides = len(task)
for s, shapes_s in enumerate(task):
x0 = window_size[0] / num_sides / 2 * (1 + 2 * s)
num_shapes = len(shapes_s)
for n, shape in enumerate(shapes_s):
x = int(x0 + size * 2 * (n - num_shapes + 1))
y = int(window_size[1] / 3)
painter = shape.draw(window=self, position=(x, y), size=size)
if n < num_shapes - 1:
painter.begin(self)
painter.setFont(QFont('Times', 35))
painter.drawText(QRect(x + size, y, size, size), Qt.AlignLeft | Qt.AlignCenter, operation)
painter.end()
def draw_legend(self, shape_dict: typing.List[Shape]):
window_size = self.get_size()
size = int(window_size[0] / (len(shape_dict) * 2 + 2))
for n, shape in enumerate(shape_dict):
x = int(size * (1 + n * 2))
y = int(window_size[1] / 4 * 3)
painter = shape.draw(window=self, position=(x, y), size=size)
painter.begin(self)
painter.drawText(QRect(x, y + size, size, size), Qt.AlignLeft | Qt.AlignCenter, f"{shape.data}")
painter.end()
################################################################################
# Window Geometry ##############################################################
################################################################################
def get_size(self) -> typing.Tuple[int, int]:
return self.geometry().width(), self.geometry().height()
def get_position(self) -> typing.Tuple[int, int]:
return self.geometry().x(), self.geometry().y()
