streamdeck-gui-ng 4.1.3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- streamdeck_gui_ng-4.1.3.dist-info/METADATA +141 -0
- streamdeck_gui_ng-4.1.3.dist-info/RECORD +62 -0
- streamdeck_gui_ng-4.1.3.dist-info/WHEEL +4 -0
- streamdeck_gui_ng-4.1.3.dist-info/entry_points.txt +4 -0
- streamdeck_gui_ng-4.1.3.dist-info/licenses/LICENSE +21 -0
- streamdeck_ui/__init__.py +6 -0
- streamdeck_ui/api.py +712 -0
- streamdeck_ui/button.ui +1214 -0
- streamdeck_ui/cli/__init__.py +0 -0
- streamdeck_ui/cli/commands.py +191 -0
- streamdeck_ui/cli/server.py +292 -0
- streamdeck_ui/config.py +244 -0
- streamdeck_ui/dimmer.py +93 -0
- streamdeck_ui/display/__init__.py +0 -0
- streamdeck_ui/display/background_color_filter.py +41 -0
- streamdeck_ui/display/display_grid.py +265 -0
- streamdeck_ui/display/empty_filter.py +43 -0
- streamdeck_ui/display/filter.py +65 -0
- streamdeck_ui/display/image_filter.py +144 -0
- streamdeck_ui/display/keypress_filter.py +63 -0
- streamdeck_ui/display/pipeline.py +74 -0
- streamdeck_ui/display/pulse_filter.py +54 -0
- streamdeck_ui/display/text_filter.py +142 -0
- streamdeck_ui/fonts/roboto/LICENSE.txt +202 -0
- streamdeck_ui/fonts/roboto/Roboto-Black.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-BlackItalic.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-Bold.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-BoldItalic.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-Italic.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-Light.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-LightItalic.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-Medium.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-MediumItalic.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-Regular.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-Thin.ttf +0 -0
- streamdeck_ui/fonts/roboto/Roboto-ThinItalic.ttf +0 -0
- streamdeck_ui/gui.py +1423 -0
- streamdeck_ui/icons/add_page.png +0 -0
- streamdeck_ui/icons/cross.png +0 -0
- streamdeck_ui/icons/gear.png +0 -0
- streamdeck_ui/icons/horizontal-align.png +0 -0
- streamdeck_ui/icons/remove_page.png +0 -0
- streamdeck_ui/icons/vertical-align.png +0 -0
- streamdeck_ui/icons/warning_icon_button.png +0 -0
- streamdeck_ui/logger.py +11 -0
- streamdeck_ui/logo.png +0 -0
- streamdeck_ui/main.ui +407 -0
- streamdeck_ui/mock_streamdeck.py +204 -0
- streamdeck_ui/model.py +78 -0
- streamdeck_ui/modules/__init__.py +0 -0
- streamdeck_ui/modules/fonts.py +150 -0
- streamdeck_ui/modules/keyboard.py +447 -0
- streamdeck_ui/modules/utils/__init__.py +0 -0
- streamdeck_ui/modules/utils/timers.py +35 -0
- streamdeck_ui/resources.qrc +10 -0
- streamdeck_ui/resources_rc.py +324 -0
- streamdeck_ui/semaphore.py +38 -0
- streamdeck_ui/settings.ui +155 -0
- streamdeck_ui/stream_deck_monitor.py +157 -0
- streamdeck_ui/ui_button.py +421 -0
- streamdeck_ui/ui_main.py +267 -0
- streamdeck_ui/ui_settings.py +119 -0
streamdeck_ui/api.py
ADDED
|
@@ -0,0 +1,712 @@
|
|
|
1
|
+
"""Defines the Python API for interacting with the StreamDeck Configuration UI"""
|
|
2
|
+
|
|
3
|
+
import os
|
|
4
|
+
import threading
|
|
5
|
+
from copy import deepcopy
|
|
6
|
+
from functools import partial
|
|
7
|
+
from typing import Dict, List, Optional, Tuple, Union
|
|
8
|
+
|
|
9
|
+
from PIL.ImageQt import ImageQt
|
|
10
|
+
from PySide6.QtCore import QObject, Signal
|
|
11
|
+
from PySide6.QtGui import QImage, QPixmap
|
|
12
|
+
from StreamDeck.Devices import StreamDeck
|
|
13
|
+
from StreamDeck.Transport.Transport import TransportError
|
|
14
|
+
|
|
15
|
+
from streamdeck_ui.config import (
|
|
16
|
+
DEFAULT_BACKGROUND_COLOR,
|
|
17
|
+
DEFAULT_FONT,
|
|
18
|
+
DEFAULT_FONT_COLOR,
|
|
19
|
+
DEFAULT_FONT_SIZE,
|
|
20
|
+
FONTS_PATH,
|
|
21
|
+
STATE_FILE,
|
|
22
|
+
read_state_from_config,
|
|
23
|
+
write_state_to_config,
|
|
24
|
+
)
|
|
25
|
+
from streamdeck_ui.dimmer import Dimmer
|
|
26
|
+
from streamdeck_ui.display.background_color_filter import BackgroundColorFilter
|
|
27
|
+
from streamdeck_ui.display.display_grid import DisplayGrid
|
|
28
|
+
from streamdeck_ui.display.filter import Filter
|
|
29
|
+
from streamdeck_ui.display.image_filter import ImageFilter
|
|
30
|
+
from streamdeck_ui.display.text_filter import TextFilter
|
|
31
|
+
from streamdeck_ui.logger import logger
|
|
32
|
+
from streamdeck_ui.model import ButtonMultiState, ButtonState, DeckState
|
|
33
|
+
from streamdeck_ui.stream_deck_monitor import StreamDeckMonitor
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
class KeySignalEmitter(QObject):
|
|
37
|
+
key_pressed = Signal(str, int, bool)
|
|
38
|
+
|
|
39
|
+
|
|
40
|
+
class StreamDeckSignalEmitter(QObject):
|
|
41
|
+
attached = Signal(dict)
|
|
42
|
+
"A signal that is raised whenever a new StreamDeck is attached."
|
|
43
|
+
detached = Signal(str)
|
|
44
|
+
"A signal that is raised whenever a StreamDeck is detached. "
|
|
45
|
+
cpu_changed = Signal(str, int)
|
|
46
|
+
|
|
47
|
+
|
|
48
|
+
class StreamDeckServer:
|
|
49
|
+
"""A StreamDeckServer represents the core server logic for interacting and
|
|
50
|
+
managing multiple Stream Decks.
|
|
51
|
+
"""
|
|
52
|
+
|
|
53
|
+
decks_by_serial: Dict[str, StreamDeck.StreamDeck] = {}
|
|
54
|
+
"Lookup with serial number -> StreamDeck"
|
|
55
|
+
|
|
56
|
+
decks_map_id_to_serial: Dict[str, str] = {}
|
|
57
|
+
"Lookup with device.id -> serial number"
|
|
58
|
+
|
|
59
|
+
state: Dict[str, DeckState] = {}
|
|
60
|
+
"The data structure holding configuration for all Stream Decks by serial number"
|
|
61
|
+
|
|
62
|
+
key_event_lock: threading.Lock
|
|
63
|
+
"Lock to serialize key press events"
|
|
64
|
+
|
|
65
|
+
lock: threading.Lock = threading.Lock()
|
|
66
|
+
"Lock to coordinate polling, updates etc to Stream Decks"
|
|
67
|
+
|
|
68
|
+
display_handlers: Dict[str, DisplayGrid] = {}
|
|
69
|
+
"Lookup with serial number for each Stream Deck display handler"
|
|
70
|
+
|
|
71
|
+
dimmers: Dict[str, Dimmer] = {}
|
|
72
|
+
"Lookup with serial number for each Stream Deck dimmer"
|
|
73
|
+
|
|
74
|
+
monitor: Optional[StreamDeckMonitor] = None
|
|
75
|
+
"Monitors for Stream Deck(s) attached to the computer"
|
|
76
|
+
|
|
77
|
+
plugevents = StreamDeckSignalEmitter()
|
|
78
|
+
"Use the connect method on the attached and detached methods to subscribe"
|
|
79
|
+
|
|
80
|
+
streamdeck_keys = KeySignalEmitter()
|
|
81
|
+
"Use the connect method on the key_pressed signal to subscribe"
|
|
82
|
+
|
|
83
|
+
def __init__(self) -> None:
|
|
84
|
+
self.decks_by_serial: Dict[str, StreamDeck.StreamDeck] = {}
|
|
85
|
+
|
|
86
|
+
# REVIEW: Should we use the same lock as the display? What exactly
|
|
87
|
+
# are we protecting? The UI is signaled via message passing.
|
|
88
|
+
self.key_event_lock = threading.Lock()
|
|
89
|
+
self.display_handlers: Dict[str, DisplayGrid] = {}
|
|
90
|
+
|
|
91
|
+
self.lock: threading.Lock = threading.Lock()
|
|
92
|
+
self.dimmers: Dict[str, Dimmer] = {}
|
|
93
|
+
|
|
94
|
+
# REVIEW: Should we just create one signal emitter for
|
|
95
|
+
# plug events and key signals?
|
|
96
|
+
self.streamdeck_keys = KeySignalEmitter()
|
|
97
|
+
self.plugevents = StreamDeckSignalEmitter()
|
|
98
|
+
|
|
99
|
+
def stop_dimmer(self, serial_number: str) -> None:
|
|
100
|
+
"""Stops the dimmer for the given Stream Deck
|
|
101
|
+
|
|
102
|
+
:param serial_number: The Stream Deck serial number.
|
|
103
|
+
:type serial_number: str
|
|
104
|
+
"""
|
|
105
|
+
self.dimmers[serial_number].stop()
|
|
106
|
+
|
|
107
|
+
def reset_dimmer(self, serial_number: str) -> bool:
|
|
108
|
+
"""Resets the dimmer for the given Stream Deck. This means the display
|
|
109
|
+
will not be dimmed and the timer starts. Reloads configuration.
|
|
110
|
+
|
|
111
|
+
Args:
|
|
112
|
+
serial_number (str): The Stream Deck serial number
|
|
113
|
+
Returns:
|
|
114
|
+
bool: Returns True if the dimmer had to be reset (i.e. woken up), False otherwise.
|
|
115
|
+
"""
|
|
116
|
+
self.dimmers[serial_number].brightness = self.get_brightness(serial_number)
|
|
117
|
+
self.dimmers[serial_number].brightness_dimmed = self.get_brightness_dimmed(serial_number)
|
|
118
|
+
return self.dimmers[serial_number].reset()
|
|
119
|
+
|
|
120
|
+
def toggle_dimmers(self):
|
|
121
|
+
"""If at least one Deck is still "on", all will be dimmed off. Otherwise,
|
|
122
|
+
toggles displays on.
|
|
123
|
+
"""
|
|
124
|
+
at_least_one = False
|
|
125
|
+
for _serial_number, dimmer in self.dimmers.items():
|
|
126
|
+
if not dimmer.dimmed:
|
|
127
|
+
at_least_one = True
|
|
128
|
+
break
|
|
129
|
+
|
|
130
|
+
for _serial_number, dimmer in self.dimmers.items():
|
|
131
|
+
if at_least_one:
|
|
132
|
+
dimmer.dim()
|
|
133
|
+
else:
|
|
134
|
+
dimmer.dim(True)
|
|
135
|
+
|
|
136
|
+
def _cpu_usage_callback(self, serial_number: str, cpu_usage: int):
|
|
137
|
+
"""An internal method that takes emits a signal on a QObject.
|
|
138
|
+
|
|
139
|
+
:param serial_number: The Stream Deck serial number
|
|
140
|
+
:type serial_number: str
|
|
141
|
+
:param cpu_usage: The current CPU usage
|
|
142
|
+
:type cpu_usage: int
|
|
143
|
+
"""
|
|
144
|
+
self.plugevents.cpu_changed.emit(serial_number, cpu_usage)
|
|
145
|
+
|
|
146
|
+
def _key_change_callback(self, serial_number: str, _deck: StreamDeck.StreamDeck, key: int, state: bool) -> None:
|
|
147
|
+
"""Callback whenever a key is pressed.
|
|
148
|
+
|
|
149
|
+
Stream Deck key events fire on a background thread. Emit a signal
|
|
150
|
+
to bring it back to UI thread, so we can use Qt objects for timers etc.
|
|
151
|
+
Since multiple keys could fire simultaneously, we need to protect
|
|
152
|
+
shared state with a lock
|
|
153
|
+
"""
|
|
154
|
+
with self.key_event_lock:
|
|
155
|
+
self.display_handlers[serial_number].set_keypress(key, state)
|
|
156
|
+
self.streamdeck_keys.key_pressed.emit(serial_number, key, state)
|
|
157
|
+
|
|
158
|
+
def get_display_timeout(self, serial_number: str) -> int:
|
|
159
|
+
"""Returns the amount of time in seconds before the display gets dimmed."""
|
|
160
|
+
if serial_number not in self.state:
|
|
161
|
+
return 0
|
|
162
|
+
return self.state[serial_number].display_timeout
|
|
163
|
+
|
|
164
|
+
def set_display_timeout(self, serial_number: str, timeout: int) -> None:
|
|
165
|
+
"""Sets the amount of time in seconds before the display gets dimmed."""
|
|
166
|
+
if serial_number not in self.state:
|
|
167
|
+
return
|
|
168
|
+
|
|
169
|
+
if self.state[serial_number].display_timeout == timeout:
|
|
170
|
+
return
|
|
171
|
+
|
|
172
|
+
self.state[serial_number].display_timeout = timeout
|
|
173
|
+
self.dimmers[serial_number].timeout = timeout
|
|
174
|
+
|
|
175
|
+
self._save_state()
|
|
176
|
+
|
|
177
|
+
def _save_state(self):
|
|
178
|
+
self.export_config(STATE_FILE)
|
|
179
|
+
|
|
180
|
+
def open_config(self, config_file: str):
|
|
181
|
+
self.state = read_state_from_config(config_file)
|
|
182
|
+
|
|
183
|
+
def import_config(self, config_file: str) -> None:
|
|
184
|
+
self.stop()
|
|
185
|
+
self.open_config(config_file)
|
|
186
|
+
self._save_state()
|
|
187
|
+
self.start()
|
|
188
|
+
|
|
189
|
+
def export_config(self, output_file: str) -> None:
|
|
190
|
+
write_state_to_config(output_file, self.state)
|
|
191
|
+
|
|
192
|
+
def _on_steam_deck_attached(self, streamdeck_id: str, streamdeck: StreamDeck):
|
|
193
|
+
streamdeck.open()
|
|
194
|
+
streamdeck.reset()
|
|
195
|
+
serial_number = streamdeck.get_serial_number()
|
|
196
|
+
|
|
197
|
+
self.decks_map_id_to_serial[streamdeck_id] = serial_number
|
|
198
|
+
self.decks_by_serial[serial_number] = streamdeck
|
|
199
|
+
|
|
200
|
+
self.set_default_state(serial_number, streamdeck.deck_type())
|
|
201
|
+
self._initialize_stream_deck_page_state(serial_number, 0, streamdeck.key_count())
|
|
202
|
+
|
|
203
|
+
streamdeck.set_key_callback(partial(self._key_change_callback, serial_number))
|
|
204
|
+
self._update_streamdeck_filters(serial_number)
|
|
205
|
+
|
|
206
|
+
self.dimmers[serial_number] = Dimmer(
|
|
207
|
+
self.get_display_timeout(serial_number),
|
|
208
|
+
self.get_brightness(serial_number),
|
|
209
|
+
self.get_brightness_dimmed(serial_number),
|
|
210
|
+
lambda brightness: self.decks_by_serial[serial_number].set_brightness(brightness),
|
|
211
|
+
)
|
|
212
|
+
self.dimmers[serial_number].reset()
|
|
213
|
+
|
|
214
|
+
self.plugevents.attached.emit(
|
|
215
|
+
{
|
|
216
|
+
"id": streamdeck_id,
|
|
217
|
+
"serial_number": serial_number,
|
|
218
|
+
"type": streamdeck.deck_type(),
|
|
219
|
+
"layout": streamdeck.key_layout(),
|
|
220
|
+
}
|
|
221
|
+
)
|
|
222
|
+
|
|
223
|
+
def set_default_state(self, serial_number: str, deck_type: str):
|
|
224
|
+
if serial_number in self.state:
|
|
225
|
+
return
|
|
226
|
+
elif deck_type in self.state:
|
|
227
|
+
logger.info(f"no configuration found for {serial_number}, use generic configuration for type: {deck_type}.")
|
|
228
|
+
self.state[serial_number] = deepcopy(self.state[deck_type])
|
|
229
|
+
|
|
230
|
+
def _initialize_stream_deck_page_state(self, serial_number: str, page: int, key_count: int):
|
|
231
|
+
"""Initializes the state for the given serial number. This allocates
|
|
232
|
+
buttons and pages based on the layout.
|
|
233
|
+
|
|
234
|
+
:param serial_number: The Stream Deck serial number
|
|
235
|
+
:type serial_number: str
|
|
236
|
+
:param page: The page of the Stream Deck
|
|
237
|
+
:type page: int
|
|
238
|
+
:param key_count: The total number of buttons on the Stream Deck
|
|
239
|
+
:type key_count: int
|
|
240
|
+
"""
|
|
241
|
+
self.state[serial_number] = self.state.setdefault(serial_number, DeckState())
|
|
242
|
+
for button in range(key_count):
|
|
243
|
+
self._button_state(serial_number, page, button)
|
|
244
|
+
|
|
245
|
+
def add_new_page(self, serial_number: str):
|
|
246
|
+
"""Adds a new page to the Stream Deck
|
|
247
|
+
|
|
248
|
+
:param serial_number: The Stream Deck serial number
|
|
249
|
+
:type serial_number: str
|
|
250
|
+
:return: The new page index
|
|
251
|
+
:rtype: int
|
|
252
|
+
"""
|
|
253
|
+
pages = self.get_pages(serial_number)
|
|
254
|
+
new_page_index = self._calculate_new_index(pages)
|
|
255
|
+
self._initialize_stream_deck_page_state(
|
|
256
|
+
serial_number, new_page_index, self.decks_by_serial[serial_number].key_count()
|
|
257
|
+
)
|
|
258
|
+
self.display_handlers[serial_number].initialize_page(new_page_index)
|
|
259
|
+
self.display_handlers[serial_number].synchronize()
|
|
260
|
+
|
|
261
|
+
return new_page_index
|
|
262
|
+
|
|
263
|
+
@staticmethod
|
|
264
|
+
def _calculate_new_index(items: List[int]) -> int:
|
|
265
|
+
"""Calculates the next free index for a list of items"""
|
|
266
|
+
items_set = set(items)
|
|
267
|
+
max_item = max(items) if items else 0
|
|
268
|
+
|
|
269
|
+
for item_index in range(1, max_item + 2):
|
|
270
|
+
if item_index not in items_set:
|
|
271
|
+
return item_index
|
|
272
|
+
return max_item + 2
|
|
273
|
+
|
|
274
|
+
def remove_page(self, serial_number: str, page: int):
|
|
275
|
+
"""Removes a page from the Stream Deck
|
|
276
|
+
|
|
277
|
+
:param serial_number: The Stream Deck serial number
|
|
278
|
+
:type serial_number: str
|
|
279
|
+
:param page: The page index
|
|
280
|
+
:type page: int
|
|
281
|
+
"""
|
|
282
|
+
if len(self.get_pages(serial_number)) == 1:
|
|
283
|
+
return
|
|
284
|
+
|
|
285
|
+
del self.state[serial_number].buttons[page]
|
|
286
|
+
self.display_handlers[serial_number].remove_page(page)
|
|
287
|
+
|
|
288
|
+
def _on_steam_deck_detached(self, deck_id: str):
|
|
289
|
+
serial_number = self.decks_map_id_to_serial.get(deck_id, None)
|
|
290
|
+
if serial_number:
|
|
291
|
+
self._cleanup(deck_id, serial_number)
|
|
292
|
+
self.plugevents.detached.emit(serial_number)
|
|
293
|
+
|
|
294
|
+
def _cleanup(self, deck_id: str, serial_number: str):
|
|
295
|
+
display_grid = self.display_handlers[serial_number]
|
|
296
|
+
display_grid.stop()
|
|
297
|
+
del self.display_handlers[serial_number]
|
|
298
|
+
|
|
299
|
+
dimmer = self.dimmers[serial_number]
|
|
300
|
+
dimmer.stop()
|
|
301
|
+
del self.dimmers[serial_number]
|
|
302
|
+
|
|
303
|
+
streamdeck = self.decks_by_serial[serial_number]
|
|
304
|
+
try:
|
|
305
|
+
if streamdeck.connected():
|
|
306
|
+
streamdeck.set_brightness(50)
|
|
307
|
+
streamdeck.reset()
|
|
308
|
+
streamdeck.close()
|
|
309
|
+
except TransportError:
|
|
310
|
+
pass
|
|
311
|
+
|
|
312
|
+
del self.decks_by_serial[serial_number]
|
|
313
|
+
del self.decks_map_id_to_serial[deck_id]
|
|
314
|
+
|
|
315
|
+
def start(self):
|
|
316
|
+
if not self.monitor:
|
|
317
|
+
self.monitor = StreamDeckMonitor(self.lock, self._on_steam_deck_attached, self._on_steam_deck_detached)
|
|
318
|
+
self.monitor.start()
|
|
319
|
+
|
|
320
|
+
def stop(self):
|
|
321
|
+
self.monitor.stop()
|
|
322
|
+
|
|
323
|
+
def get_deck_layout(self, serial_number: str) -> Tuple[int, int]:
|
|
324
|
+
"""Returns a tuple containing the number of rows and columns for the specified Stream Deck"""
|
|
325
|
+
return self.decks_by_serial[serial_number].key_layout()
|
|
326
|
+
|
|
327
|
+
def _button_state(self, serial_number: str, page: int, button: int, state: Optional[int] = None) -> ButtonState:
|
|
328
|
+
multi_state = self._button_multi_state(serial_number, page, button)
|
|
329
|
+
# if no state is specified, use the current state
|
|
330
|
+
choose_state = state or multi_state.state
|
|
331
|
+
# if the choose state is not in the states dict, add it
|
|
332
|
+
multi_state.states[choose_state] = multi_state.states.setdefault(choose_state, ButtonState())
|
|
333
|
+
return multi_state.states[choose_state]
|
|
334
|
+
|
|
335
|
+
def get_button_state_object(self, serial_number: str, page: int, button: int, state: int) -> ButtonState:
|
|
336
|
+
"""Returns the ButtonState object for the given button"""
|
|
337
|
+
return self._button_state(serial_number, page, button, state)
|
|
338
|
+
|
|
339
|
+
def _button_multi_state(self, serial_number: str, page: int, button: int) -> ButtonMultiState:
|
|
340
|
+
"""Returns the ButtonMultiState for the given button"""
|
|
341
|
+
# if the page is not in the pages dict, add it
|
|
342
|
+
self.state[serial_number].buttons[page] = self.state[serial_number].buttons.setdefault(page, {})
|
|
343
|
+
# if the button is not in the buttons dict, add it with a default state
|
|
344
|
+
self.state[serial_number].buttons[page][button] = (
|
|
345
|
+
self.state[serial_number]
|
|
346
|
+
.buttons[page]
|
|
347
|
+
.setdefault(button, ButtonMultiState(state=0, states={0: ButtonState()}))
|
|
348
|
+
)
|
|
349
|
+
return self.state[serial_number].buttons[page][button]
|
|
350
|
+
|
|
351
|
+
def get_button_state(self, serial_number: str, page: int, button: int) -> int:
|
|
352
|
+
"""Returns the state of a button"""
|
|
353
|
+
return self._button_multi_state(serial_number, page, button).state
|
|
354
|
+
|
|
355
|
+
def get_button_states(self, serial_number: str, page: int, button: int) -> List[int]:
|
|
356
|
+
"""Returns the states of a button"""
|
|
357
|
+
return sorted(list(self._button_multi_state(serial_number, page, button).states.keys()))
|
|
358
|
+
|
|
359
|
+
def add_new_button_state(self, serial_number: str, page: int, button: int) -> int:
|
|
360
|
+
"""Adds a new button state"""
|
|
361
|
+
states = self.get_button_states(serial_number, page, button)
|
|
362
|
+
new_button_state_index = self._calculate_new_index(states)
|
|
363
|
+
self._button_multi_state(serial_number, page, button).states[new_button_state_index] = ButtonState()
|
|
364
|
+
return new_button_state_index
|
|
365
|
+
|
|
366
|
+
def remove_button_state(self, serial_number: str, page: int, button: int, state: int) -> None:
|
|
367
|
+
"""Removes a button state"""
|
|
368
|
+
if len(self.get_button_states(serial_number, page, button)) == 1:
|
|
369
|
+
return
|
|
370
|
+
del self._button_multi_state(serial_number, page, button).states[state]
|
|
371
|
+
|
|
372
|
+
def set_button_state(self, serial_number: str, page: int, button: int, state: int) -> None:
|
|
373
|
+
"""Sets the state of a button"""
|
|
374
|
+
if self.get_button_state(serial_number, page, button) != state:
|
|
375
|
+
states = self.get_button_states(serial_number, page, button)
|
|
376
|
+
if state in states:
|
|
377
|
+
self._button_multi_state(serial_number, page, button).state = state
|
|
378
|
+
self._save_state()
|
|
379
|
+
self._update_button_filters(serial_number, page, button)
|
|
380
|
+
display_handler = self.display_handlers[serial_number]
|
|
381
|
+
display_handler.synchronize()
|
|
382
|
+
|
|
383
|
+
def get_button_switch_state(self, serial_number: str, page: int, button: int) -> int:
|
|
384
|
+
"""Returns the state switch set for the specified button. 0 implies no state switch."""
|
|
385
|
+
return self._button_state(serial_number, page, button).switch_state
|
|
386
|
+
|
|
387
|
+
def set_button_switch_state(self, serial_number: str, page: int, button: int, switch_state: int) -> None:
|
|
388
|
+
"""Sets the state switch associated with the button"""
|
|
389
|
+
if self.get_button_switch_state(serial_number, page, button) != switch_state:
|
|
390
|
+
self._button_state(serial_number, page, button).switch_state = switch_state
|
|
391
|
+
self._save_state()
|
|
392
|
+
|
|
393
|
+
def swap_buttons(self, serial_number: str, page: int, source_button: int, target_button: int) -> None:
|
|
394
|
+
"""Swaps the properties of the source and target buttons"""
|
|
395
|
+
temp = self.state[serial_number].buttons[page][source_button]
|
|
396
|
+
self.state[serial_number].buttons[page][source_button] = self.state[serial_number].buttons[page][target_button]
|
|
397
|
+
self.state[serial_number].buttons[page][target_button] = temp
|
|
398
|
+
self._save_state()
|
|
399
|
+
|
|
400
|
+
# Update rendering for these two images
|
|
401
|
+
self._update_button_filters(serial_number, page, source_button)
|
|
402
|
+
self._update_button_filters(serial_number, page, target_button)
|
|
403
|
+
display_handler = self.display_handlers[serial_number]
|
|
404
|
+
display_handler.synchronize()
|
|
405
|
+
|
|
406
|
+
def set_button_text(self, deck_id: str, page: int, button: int, text: str) -> None:
|
|
407
|
+
"""Set the text associated with a button"""
|
|
408
|
+
if self.get_button_text(deck_id, page, button) != text:
|
|
409
|
+
self._button_state(deck_id, page, button).text = text
|
|
410
|
+
self._save_state()
|
|
411
|
+
self._update_button_filters(deck_id, page, button)
|
|
412
|
+
display_handler = self.display_handlers[deck_id]
|
|
413
|
+
display_handler.synchronize()
|
|
414
|
+
|
|
415
|
+
def get_button_text(self, deck_id: str, page: int, button: int) -> str:
|
|
416
|
+
"""Returns the text set for the specified button"""
|
|
417
|
+
return self._button_state(deck_id, page, button).text
|
|
418
|
+
|
|
419
|
+
def set_button_icon(self, deck_id: str, page: int, button: int, icon: str) -> None:
|
|
420
|
+
"""Sets the icon associated with a button"""
|
|
421
|
+
if self.get_button_icon(deck_id, page, button) != icon:
|
|
422
|
+
self._button_state(deck_id, page, button).icon = icon
|
|
423
|
+
self._save_state()
|
|
424
|
+
|
|
425
|
+
self._update_button_filters(deck_id, page, button)
|
|
426
|
+
display_handler = self.display_handlers[deck_id]
|
|
427
|
+
display_handler.synchronize()
|
|
428
|
+
|
|
429
|
+
def get_button_text_vertical_align(self, serial_number: str, page: int, button: int) -> str:
|
|
430
|
+
"""Gets the vertical text alignment. Values are bottom, middle-bottom, middle, middle-top, top"""
|
|
431
|
+
return self._button_state(serial_number, page, button).text_vertical_align
|
|
432
|
+
|
|
433
|
+
def get_button_text_horizontal_align(self, serial_number: str, page: int, button: int) -> str:
|
|
434
|
+
"""Gets the horizontal text alignment. Values are left, center, right"""
|
|
435
|
+
return self._button_state(serial_number, page, button).text_horizontal_align
|
|
436
|
+
|
|
437
|
+
def set_button_text_horizontal_align(self, serial_number: str, page: int, button: int, alignment: str) -> None:
|
|
438
|
+
"""Gets the horizontal text alignment. Values are left, center, right"""
|
|
439
|
+
if self.get_button_text_horizontal_align(serial_number, page, button) != alignment:
|
|
440
|
+
self._button_state(serial_number, page, button).text_horizontal_align = alignment
|
|
441
|
+
self._save_state()
|
|
442
|
+
self._update_button_filters(serial_number, page, button)
|
|
443
|
+
display_handler = self.display_handlers[serial_number]
|
|
444
|
+
display_handler.synchronize()
|
|
445
|
+
|
|
446
|
+
def set_button_text_vertical_align(self, serial_number: str, page: int, button: int, alignment: str) -> None:
|
|
447
|
+
"""Gets the vertical text alignment. Values are bottom, middle-bottom, middle, middle-top, top"""
|
|
448
|
+
if self.get_button_text_vertical_align(serial_number, page, button) != alignment:
|
|
449
|
+
self._button_state(serial_number, page, button).text_vertical_align = alignment
|
|
450
|
+
self._save_state()
|
|
451
|
+
self._update_button_filters(serial_number, page, button)
|
|
452
|
+
display_handler = self.display_handlers[serial_number]
|
|
453
|
+
display_handler.synchronize()
|
|
454
|
+
|
|
455
|
+
def set_button_font_color(self, serial_number: str, page: int, button: int, color: str) -> None:
|
|
456
|
+
"""Sets the text color associated with a button"""
|
|
457
|
+
if self.get_button_font_color(serial_number, page, button) != color:
|
|
458
|
+
# Don't pollute .streamdeck_ui.json with entries of the default value
|
|
459
|
+
if color == DEFAULT_FONT_COLOR:
|
|
460
|
+
color = ""
|
|
461
|
+
self._button_state(serial_number, page, button).font_color = color
|
|
462
|
+
self._save_state()
|
|
463
|
+
self._update_button_filters(serial_number, page, button)
|
|
464
|
+
|
|
465
|
+
try:
|
|
466
|
+
display_handler = self.display_handlers[serial_number]
|
|
467
|
+
display_handler.synchronize()
|
|
468
|
+
except KeyError:
|
|
469
|
+
raise ValueError(f"Invalid serial number: {serial_number}")
|
|
470
|
+
|
|
471
|
+
def get_button_font_color(self, serial_number: str, page: int, button: int) -> str:
|
|
472
|
+
"""Returns the text color set for the specified button"""
|
|
473
|
+
return self._button_state(serial_number, page, button).font_color
|
|
474
|
+
|
|
475
|
+
def set_button_background_color(self, serial_number: str, page: int, button: int, color: str) -> None:
|
|
476
|
+
"""Sets the background color associated with a button"""
|
|
477
|
+
if self.get_button_background_color(serial_number, page, button) != color:
|
|
478
|
+
# Don't pollute .streamdeck_ui.json with entries of the default value
|
|
479
|
+
if color == DEFAULT_BACKGROUND_COLOR:
|
|
480
|
+
color = ""
|
|
481
|
+
self._button_state(serial_number, page, button).background_color = color
|
|
482
|
+
self._save_state()
|
|
483
|
+
self._update_button_filters(serial_number, page, button)
|
|
484
|
+
|
|
485
|
+
try:
|
|
486
|
+
display_handler = self.display_handlers[serial_number]
|
|
487
|
+
display_handler.synchronize()
|
|
488
|
+
except KeyError:
|
|
489
|
+
raise ValueError(f"Invalid serial number: {serial_number}")
|
|
490
|
+
|
|
491
|
+
def get_button_background_color(self, serial_number: str, page: int, button: int) -> str:
|
|
492
|
+
"""Returns the background color set for the specified button"""
|
|
493
|
+
return self._button_state(serial_number, page, button).background_color
|
|
494
|
+
|
|
495
|
+
def get_button_icon_pixmap(self, serial_number: str, page: int, button: int) -> Optional[QPixmap]:
|
|
496
|
+
"""Returns the QPixmap value for the given button (streamdeck, page, button)"""
|
|
497
|
+
pil_image = self.display_handlers[serial_number].get_image(page, button)
|
|
498
|
+
if pil_image:
|
|
499
|
+
qt_image = ImageQt(pil_image)
|
|
500
|
+
qt_image = qt_image.convertToFormat(QImage.Format.Format_ARGB32)
|
|
501
|
+
return QPixmap(qt_image)
|
|
502
|
+
return None
|
|
503
|
+
|
|
504
|
+
def get_button_icon(self, serial_number: str, page: int, button: int) -> str:
|
|
505
|
+
"""Returns the icon path for the specified button"""
|
|
506
|
+
return self._button_state(serial_number, page, button).icon
|
|
507
|
+
|
|
508
|
+
def set_button_change_brightness(self, serial_number: str, page: int, button: int, amount: int) -> None:
|
|
509
|
+
"""Sets the brightness changing associated with a button"""
|
|
510
|
+
if self.get_button_change_brightness(serial_number, page, button) != amount:
|
|
511
|
+
self._button_state(serial_number, page, button).brightness_change = amount
|
|
512
|
+
self._save_state()
|
|
513
|
+
|
|
514
|
+
def get_button_change_brightness(self, serial_number: str, page: int, button: int) -> int:
|
|
515
|
+
"""Returns the brightness change set for a particular button"""
|
|
516
|
+
return self._button_state(serial_number, page, button).brightness_change
|
|
517
|
+
|
|
518
|
+
def set_button_command(self, serial_number: str, page: int, button: int, command: str) -> None:
|
|
519
|
+
"""Sets the command associated with the button"""
|
|
520
|
+
if self.get_button_command(serial_number, page, button) != command:
|
|
521
|
+
self._button_state(serial_number, page, button).command = command
|
|
522
|
+
self._save_state()
|
|
523
|
+
|
|
524
|
+
def get_button_command(self, serial_number: str, page: int, button: int) -> str:
|
|
525
|
+
"""Returns the command set for the specified button"""
|
|
526
|
+
return self._button_state(serial_number, page, button).command
|
|
527
|
+
|
|
528
|
+
def set_button_switch_page(self, serial_number: str, page: int, button: int, switch_page: int) -> None:
|
|
529
|
+
"""Sets the page switch associated with the button"""
|
|
530
|
+
if self.get_button_switch_page(serial_number, page, button) != switch_page:
|
|
531
|
+
self._button_state(serial_number, page, button).switch_page = switch_page
|
|
532
|
+
self._save_state()
|
|
533
|
+
|
|
534
|
+
def get_button_switch_page(self, serial_number: str, page: int, button: int) -> int:
|
|
535
|
+
"""Returns the page switch set for the specified button. 0 implies no page switch."""
|
|
536
|
+
return self._button_state(serial_number, page, button).switch_page
|
|
537
|
+
|
|
538
|
+
def set_button_keys(self, serial_number: str, page: int, button: int, keys: str) -> None:
|
|
539
|
+
"""Sets the keys associated with the button"""
|
|
540
|
+
if self.get_button_keys(serial_number, page, button) != keys:
|
|
541
|
+
self._button_state(serial_number, page, button).keys = keys
|
|
542
|
+
self._save_state()
|
|
543
|
+
|
|
544
|
+
def set_button_font(self, serial_number: str, page: int, button: int, font: str) -> None:
|
|
545
|
+
if self.get_button_font(serial_number, page, button) != font:
|
|
546
|
+
# Don't pollute .streamdeck_ui.json with entries of the default value
|
|
547
|
+
if font.endswith(DEFAULT_FONT):
|
|
548
|
+
font = ""
|
|
549
|
+
self._button_state(serial_number, page, button).font = font
|
|
550
|
+
self._save_state()
|
|
551
|
+
self._update_button_filters(serial_number, page, button)
|
|
552
|
+
display_handler = self.display_handlers[serial_number]
|
|
553
|
+
display_handler.synchronize()
|
|
554
|
+
|
|
555
|
+
def get_button_font_size(self, serial_number: str, page: int, button: int) -> int:
|
|
556
|
+
"""Returns the font size set for the specified button"""
|
|
557
|
+
return self._button_state(serial_number, page, button).font_size
|
|
558
|
+
|
|
559
|
+
def set_button_font_size(self, serial_number: str, page: int, button: int, font_size: int) -> None:
|
|
560
|
+
if self.get_button_font_size(serial_number, page, button) != font_size:
|
|
561
|
+
# Don't pollute .streamdeck_ui.json with entries of the default value
|
|
562
|
+
if font_size == DEFAULT_FONT_SIZE:
|
|
563
|
+
font_size = 0
|
|
564
|
+
self._button_state(serial_number, page, button).font_size = font_size
|
|
565
|
+
self._save_state()
|
|
566
|
+
self._update_button_filters(serial_number, page, button)
|
|
567
|
+
display_handler = self.display_handlers[serial_number]
|
|
568
|
+
display_handler.synchronize()
|
|
569
|
+
|
|
570
|
+
def get_button_keys(self, serial_number: str, page: int, button: int) -> str:
|
|
571
|
+
"""Returns the keys set for the specified button"""
|
|
572
|
+
return self._button_state(serial_number, page, button).keys
|
|
573
|
+
|
|
574
|
+
def get_button_font(self, serial_number: str, page: int, button: int) -> str:
|
|
575
|
+
"""Returns the font set for the specified button"""
|
|
576
|
+
return self._button_state(serial_number, page, button).font
|
|
577
|
+
|
|
578
|
+
def set_button_write(self, serial_number: str, page: int, button: int, write: str) -> None:
|
|
579
|
+
"""Sets the text meant to be written when button is pressed"""
|
|
580
|
+
if self.get_button_write(serial_number, page, button) != write:
|
|
581
|
+
self._button_state(serial_number, page, button).write = write
|
|
582
|
+
self._save_state()
|
|
583
|
+
|
|
584
|
+
def get_button_write(self, serial_number: str, page: int, button: int) -> str:
|
|
585
|
+
"""Returns the text to be produced when the specified button is pressed"""
|
|
586
|
+
return self._button_state(serial_number, page, button).write
|
|
587
|
+
|
|
588
|
+
def set_button_force_refresh(self, serial_number: str, page: int, button: int, force_refresh: bool) -> None:
|
|
589
|
+
"""Sets whether to force icon refresh after command execution"""
|
|
590
|
+
if self.get_button_force_refresh(serial_number, page, button) != force_refresh:
|
|
591
|
+
self._button_state(serial_number, page, button).force_refresh = force_refresh
|
|
592
|
+
self._save_state()
|
|
593
|
+
|
|
594
|
+
def get_button_force_refresh(self, serial_number: str, page: int, button: int) -> bool:
|
|
595
|
+
"""Returns whether icon refresh is forced after command execution"""
|
|
596
|
+
return self._button_state(serial_number, page, button).force_refresh
|
|
597
|
+
|
|
598
|
+
def set_brightness(self, serial_number: str, brightness: int) -> None:
|
|
599
|
+
"""Sets the brightness for every button on the deck"""
|
|
600
|
+
if self.get_brightness(serial_number) != brightness:
|
|
601
|
+
self.decks_by_serial[serial_number].set_brightness(brightness)
|
|
602
|
+
self.state[serial_number].brightness = brightness
|
|
603
|
+
self._save_state()
|
|
604
|
+
|
|
605
|
+
def get_brightness(self, serial_number: str) -> int:
|
|
606
|
+
"""Gets the brightness that is set for the specified stream deck"""
|
|
607
|
+
return self.state[serial_number].brightness
|
|
608
|
+
|
|
609
|
+
def get_brightness_dimmed(self, serial_number: str) -> int:
|
|
610
|
+
"""Gets the percentage value of the full brightness that is used when dimming the specified
|
|
611
|
+
stream deck"""
|
|
612
|
+
return self.state[serial_number].brightness_dimmed
|
|
613
|
+
|
|
614
|
+
def set_brightness_dimmed(self, serial_number: str, brightness_dimmed: int) -> None:
|
|
615
|
+
"""Sets the percentage value that will be used for dimming the full brightness"""
|
|
616
|
+
self.state[serial_number].brightness_dimmed = brightness_dimmed
|
|
617
|
+
self._save_state()
|
|
618
|
+
|
|
619
|
+
def change_brightness(self, deck_id: str, amount: int = 1) -> None:
|
|
620
|
+
"""Change the brightness of the deck by the specified amount"""
|
|
621
|
+
brightness = max(min(self.get_brightness(deck_id) + amount, 100), 0)
|
|
622
|
+
self.set_brightness(deck_id, brightness)
|
|
623
|
+
self.dimmers[deck_id].brightness = brightness
|
|
624
|
+
self.dimmers[deck_id].reset()
|
|
625
|
+
|
|
626
|
+
def get_pages(self, serial_number: str) -> List[int]:
|
|
627
|
+
"""Returns pages for the specified stream deck"""
|
|
628
|
+
return sorted(list(self.state[serial_number].buttons.keys()))
|
|
629
|
+
|
|
630
|
+
def get_page(self, serial_number: str) -> int:
|
|
631
|
+
"""Gets the current page shown on the stream deck"""
|
|
632
|
+
return self.state[serial_number].page
|
|
633
|
+
|
|
634
|
+
def set_page(self, serial_number: str, page: int) -> None:
|
|
635
|
+
"""Sets the current page shown on the stream deck"""
|
|
636
|
+
if self.get_page(serial_number) != page:
|
|
637
|
+
if page not in self.get_pages(serial_number):
|
|
638
|
+
return
|
|
639
|
+
self.state[serial_number].page = page
|
|
640
|
+
self._save_state()
|
|
641
|
+
|
|
642
|
+
display_handler = self.display_handlers[serial_number]
|
|
643
|
+
|
|
644
|
+
# Let the display know to process new set of pipelines
|
|
645
|
+
display_handler.set_page(page)
|
|
646
|
+
# Wait for at least one cycle
|
|
647
|
+
display_handler.synchronize()
|
|
648
|
+
|
|
649
|
+
def _update_streamdeck_filters(self, serial_number: str):
|
|
650
|
+
"""Updates the filters for all the StreamDeck buttons.
|
|
651
|
+
|
|
652
|
+
:param serial_number: The StreamDeck serial number.
|
|
653
|
+
:type serial_number: str
|
|
654
|
+
"""
|
|
655
|
+
|
|
656
|
+
# if deck is not attached then do nothing
|
|
657
|
+
if serial_number not in self.decks_by_serial:
|
|
658
|
+
return
|
|
659
|
+
|
|
660
|
+
pages = self.get_pages(serial_number)
|
|
661
|
+
display_handler = self.display_handlers.get(
|
|
662
|
+
serial_number, DisplayGrid(self.lock, self.decks_by_serial[serial_number], pages, self._cpu_usage_callback)
|
|
663
|
+
)
|
|
664
|
+
display_handler.set_page(self.get_page(serial_number))
|
|
665
|
+
self.display_handlers[serial_number] = display_handler
|
|
666
|
+
|
|
667
|
+
for page, buttons in self.state[serial_number].buttons.items():
|
|
668
|
+
for button in buttons:
|
|
669
|
+
self._update_button_filters(serial_number, page, button)
|
|
670
|
+
|
|
671
|
+
display_handler.start()
|
|
672
|
+
|
|
673
|
+
def _update_button_filters(self, serial_number: str, page: int, button: int):
|
|
674
|
+
"""Sets the filters for a given button. Any previous filters are replaced.
|
|
675
|
+
|
|
676
|
+
:param serial_number: The StreamDeck serial number
|
|
677
|
+
:type serial_number: str
|
|
678
|
+
:param page: The page number
|
|
679
|
+
:type page: int
|
|
680
|
+
:param button: The button to update
|
|
681
|
+
:type button: int
|
|
682
|
+
"""
|
|
683
|
+
display_handler = self.display_handlers[serial_number]
|
|
684
|
+
button_settings = self._button_state(serial_number, page, button)
|
|
685
|
+
filters: List[Filter] = []
|
|
686
|
+
|
|
687
|
+
background_color = button_settings.background_color or DEFAULT_BACKGROUND_COLOR
|
|
688
|
+
filters.append(BackgroundColorFilter(background_color))
|
|
689
|
+
|
|
690
|
+
if button_settings.icon:
|
|
691
|
+
filters.append(ImageFilter(button_settings.icon))
|
|
692
|
+
|
|
693
|
+
if button_settings.text:
|
|
694
|
+
font_size = button_settings.font_size or DEFAULT_FONT_SIZE
|
|
695
|
+
font_color = button_settings.font_color or DEFAULT_FONT_COLOR
|
|
696
|
+
font = button_settings.font or DEFAULT_FONT
|
|
697
|
+
# if font is not absolute means a default font, prefix it
|
|
698
|
+
if not font.startswith("/"):
|
|
699
|
+
font = os.path.join(FONTS_PATH, font)
|
|
700
|
+
# add fallback font logic
|
|
701
|
+
filters.append(
|
|
702
|
+
TextFilter(
|
|
703
|
+
button_settings.text,
|
|
704
|
+
font,
|
|
705
|
+
font_size,
|
|
706
|
+
font_color,
|
|
707
|
+
button_settings.text_vertical_align,
|
|
708
|
+
button_settings.text_horizontal_align,
|
|
709
|
+
)
|
|
710
|
+
)
|
|
711
|
+
|
|
712
|
+
display_handler.replace(page, button, filters)
|