1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
import sys
import pygame
import os
from .input import Input
from .command import Command
from .game import Game
import logging
import json
WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
IS_DEVELOPMENT_MODE = os.getenv("DEVELOPMENT", "").lower() != ""
OPTIONS = Game.all() + ["Quit"]
logger = logging.getLogger(__name__)
WIDTH = HEIGHT = FONT = screen = selected_index = None
ORIGINAL_WIDTH = ORIGINAL_HEIGHT = None
def init_display():
global WIDTH, HEIGHT, FONT, screen, selected_index, ORIGINAL_WIDTH, ORIGINAL_HEIGHT
ORIGINAL_WIDTH = pygame.display.Info().current_w // 2
ORIGINAL_HEIGHT = pygame.display.Info().current_h // 2
WIDTH, HEIGHT = ORIGINAL_WIDTH, ORIGINAL_HEIGHT
FONT = pygame.font.Font(None, 36)
screen = pygame.display.set_mode(
(WIDTH, HEIGHT) if IS_DEVELOPMENT_MODE else (0, 0),
pygame.RESIZABLE if IS_DEVELOPMENT_MODE else pygame.FULLSCREEN,
)
selected_index = 0
def draw_menu(is_joint_mode=False):
screen.fill(WHITE)
is_vertical = HEIGHT > WIDTH * 1.5
start_y = 100 if is_vertical else 150
spacing = 40 if is_vertical else 50
title_text = FONT.render("SplitScreen Duo", True, BLACK)
screen.blit(title_text, (WIDTH // 2 - title_text.get_width() // 2, start_y - 50))
for i, option in enumerate(OPTIONS):
text = FONT.render(option, True, BLACK if i != selected_index else (200, 0, 0))
screen.blit(text, (WIDTH // 2 - text.get_width() // 2, start_y + i * spacing))
pygame.display.flip()
def set_screen_orientation(is_joint_mode):
global WIDTH, HEIGHT, screen, ORIGINAL_WIDTH, ORIGINAL_HEIGHT
if is_joint_mode:
WIDTH = ORIGINAL_WIDTH
HEIGHT = ORIGINAL_HEIGHT * 2
else:
WIDTH = ORIGINAL_WIDTH
HEIGHT = ORIGINAL_HEIGHT
screen = pygame.display.set_mode(
(WIDTH, HEIGHT) if IS_DEVELOPMENT_MODE else (0, 0),
pygame.RESIZABLE if IS_DEVELOPMENT_MODE else pygame.FULLSCREEN,
)
logger.info(f"screen orientation changed to {'vertical' if is_joint_mode else 'horizontal'}")
def process_events(serial, instance, is_joint_mode):
global selected_index
input_handler = Input(debug=IS_DEVELOPMENT_MODE)
for event in pygame.event.get():
if event.type == pygame.QUIT:
return {"command": Command.QUIT.value, "action": None, "value": None}
action = input_handler.get_input(event)
if action == "UP":
selected_index = (selected_index - 1) % len(OPTIONS)
elif action == "DOWN":
selected_index = (selected_index + 1) % len(OPTIONS)
elif action == "SELECT":
if OPTIONS[selected_index] == "Quit":
if instance == "primary":
return {
"command": Command.QUIT.value,
"action": None,
"value": None,
}
else:
pygame.quit()
sys.exit()
else:
return {
"command": Command.SELECT_GAME.value,
"action": None,
"value": OPTIONS[selected_index],
}
elif action == "QUIT":
return {"command": Command.QUIT.value, "action": None, "value": None}
if (
event.type == pygame.KEYDOWN
and event.key == pygame.K_b
and instance == "primary"
and IS_DEVELOPMENT_MODE
):
is_joint_mode[0] = not is_joint_mode[0]
if instance == "primary":
set_screen_orientation(is_joint_mode[0])
logger.info(f"toggled joint mode to {is_joint_mode[0]}")
command = (
Command.ENTER_JOINT_MODE.value
if is_joint_mode[0]
else Command.EXIT_JOINT_MODE.value
)
serial.write(json.dumps({"command": command}).encode("utf-8"))
draw_menu(is_joint_mode[0] if isinstance(is_joint_mode, list) else is_joint_mode)
return None
|