forked from tosca07/picochess
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dispatcher.py
192 lines (171 loc) · 8.92 KB
/
dispatcher.py
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
# Copyright (C) 2013-2018 Jean-Francois Romang (jromang@posteo.de)
# Shivkumar Shivaji ()
# Jürgen Précour (LocutusOfPenguin@posteo.de)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import logging
import queue
from threading import Timer, Thread, Lock
from copy import deepcopy
from typing import Dict, Set
from utilities import DisplayDgt, DispatchDgt, dispatch_queue
from dgt.api import Dgt, DgtApi
from dgt.menu import DgtMenu
logger = logging.getLogger(__name__)
class Dispatcher(DispatchDgt, Thread):
"""A dispatcher taking the dispatch_queue and fill dgt_queue with the commands in time."""
def __init__(self, dgtmenu: DgtMenu):
super(Dispatcher, self).__init__()
self.dgtmenu = dgtmenu
self.devices: Set[str] = set()
self.maxtimer: Dict[str, Timer] = {}
self.maxtimer_running: Dict[str, bool] = {}
self.clock_connected: Dict[str, bool] = {}
self.time_factor = 1 # This is for testing the duration - remove it lateron!
self.tasks: Dict[str, list] = {} # delayed task array
self.display_hash: Dict[str, int] = {} # Hash value of clock's display
self.process_lock: Dict[str, Lock] = {}
def register(self, device: str):
"""Register new device to send DgtApi messsages."""
logger.debug('device %s registered', device)
self.devices.add(device)
self.maxtimer_running[device] = False
self.clock_connected[device] = False
self.process_lock[device] = Lock()
self.tasks[device] = []
self.display_hash[device] = 0
def is_prio_device(self, dev, connect):
"""Return the most prio registered device."""
logger.debug('(%s) clock connected: %s', dev, connect)
if not connect:
return False
if 'i2c' in self.devices:
return 'i2c' == dev
if 'ser' in self.devices:
return 'ser' == dev
return 'web' == dev
def _stopped_maxtimer(self, dev: str):
self.maxtimer_running[dev] = False
self.dgtmenu.disable_picochess_displayed(dev)
if dev not in self.devices:
logger.debug('delete not registered (%s) tasks', dev)
self.tasks[dev] = []
return
if self.tasks[dev]:
logger.debug('processing delayed (%s) tasks: %s', dev, self.tasks[dev])
else:
logger.debug('(%s) max timer finished - returning to time display', dev)
DisplayDgt.show(Dgt.DISPLAY_TIME(force=False, wait=True, devs={dev}))
while self.tasks[dev]:
logger.debug('(%s) tasks has %i members', dev, len(self.tasks[dev]))
try:
message = self.tasks[dev].pop(0)
except IndexError:
break
with self.process_lock[dev]:
self._process_message(message, dev)
if self.maxtimer_running[dev]: # run over the task list until a maxtime command was processed
remaining = len(self.tasks[dev])
if remaining:
logger.debug('(%s) tasks stopped on %i remaining members', dev, remaining)
else:
logger.debug('(%s) tasks completed', dev)
break
def _process_message(self, message, dev: str):
do_handle = True
if repr(message) in (DgtApi.CLOCK_START, DgtApi.CLOCK_STOP, DgtApi.DISPLAY_TIME):
self.display_hash[dev] = 0 # Cant know the clock display if command changing the running status
else:
if repr(message) in (DgtApi.DISPLAY_MOVE, DgtApi.DISPLAY_TEXT):
if self.display_hash[dev] == hash(message) and not message.beep:
do_handle = False
else:
self.display_hash[dev] = hash(message)
if do_handle:
logger.debug('(%s) handle DgtApi: %s', dev, message)
if repr(message) == DgtApi.CLOCK_VERSION:
logger.debug('(%s) clock registered', dev)
self.clock_connected[dev] = True
clk = (DgtApi.DISPLAY_MOVE, DgtApi.DISPLAY_TEXT, DgtApi.DISPLAY_TIME,
DgtApi.CLOCK_SET, DgtApi.CLOCK_START, DgtApi.CLOCK_STOP)
if repr(message) in clk and not self.clock_connected[dev]:
logger.debug('(%s) clock still not registered => ignore %s', dev, message)
return
if hasattr(message, 'maxtime'):
if repr(message) == DgtApi.DISPLAY_TEXT:
if message.maxtime == 2.1: # 2.1=picochess message
self.dgtmenu.enable_picochess_displayed(dev)
if self.dgtmenu.inside_updt_menu():
if message.maxtime == 0.1: # 0.1=eBoard error
logger.debug('(%s) inside update menu => board errors not displayed', dev)
return
if message.maxtime == 1.1: # 1.1=eBoard connect
logger.debug('(%s) inside update menu => board connect not displayed', dev)
return
if message.maxtime > 0.1: # filter out "all the time" show and "eBoard error" messages
self.maxtimer[dev] = Timer(message.maxtime * self.time_factor, self._stopped_maxtimer, [dev])
self.maxtimer[dev].start()
logger.debug('(%s) showing %s for %.1f secs', dev, message, message.maxtime * self.time_factor)
self.maxtimer_running[dev] = True
if repr(message) == DgtApi.CLOCK_START and self.dgtmenu.inside_updt_menu():
logger.debug('(%s) inside update menu => clock not started', dev)
return
message.devs = {dev} # on new system, we only have ONE device each message - force this!
DisplayDgt.show(message)
else:
logger.debug('(%s) hash ignore DgtApi: %s', dev, message)
def stop_maxtimer(self, dev):
"""Stop the maxtimer."""
if self.maxtimer_running[dev]:
self.maxtimer[dev].cancel()
self.maxtimer[dev].join()
self.maxtimer_running[dev] = False
self.dgtmenu.disable_picochess_displayed(dev)
def run(self):
"""Call by threading.Thread start() function."""
logger.info('dispatch_queue ready')
while True:
# Check if we have something to display
try:
msg = dispatch_queue.get()
logger.debug('received command from dispatch_queue: %s devs: %s', msg, ','.join(msg.devs))
for dev in msg.devs & self.devices:
message = deepcopy(msg)
if self.maxtimer_running[dev]:
if hasattr(message, 'wait'):
if message.wait:
self.tasks[dev].append(message)
logger.debug('(%s) tasks delayed: %s', dev, self.tasks[dev])
continue
else:
logger.debug('ignore former maxtime - dev: %s', dev)
self.stop_maxtimer(dev)
if self.tasks[dev]:
logger.debug('delete following (%s) tasks: %s', dev, self.tasks[dev])
while self.tasks[dev]: # but do the last CLOCK_START()
command = self.tasks[dev].pop()
if repr(command) == DgtApi.CLOCK_START: # clock might be in set mode
logger.debug('processing (last) delayed clock start command')
with self.process_lock[dev]:
self._process_message(command, dev)
break
self.tasks[dev] = []
else:
logger.debug('command doesnt change the clock display => (%s) max timer ignored', dev)
else:
logger.debug('(%s) max timer not running => processing command: %s', dev, message)
with self.process_lock[dev]:
self._process_message(message, dev)
except queue.Empty:
pass