class Main:
def __init__(self, UPLOAD=True, USE_GUI=True, USE_FILESTREAM=False):
self._socket_address = 'src/client/data_stream.sock'
self._receiver = Receiver(self._socket_address)
self._geo_listen = Geometry()
self._frequency = 15
if USE_FILESTREAM:
self._stream = FileStream(
"gen_data/downtown-east2_only_turn_sigs_speed_lims.json",
self._socket_address)
self._send_frequency = 1
self._speed_limit = None
self._speed_time = None
self._gui = None
self._forgot_signals_event = threading.Event()
self._speeding_event = threading.Event()
self._aggressive_event = threading.Event()
self._green_event = threading.Event()
self._green_event.direction = None
self._speeding_event.speed_limit = None
self._last_turn_forget = None
self._turn_analyzer = TurnSignalAnalyzer(self._forgot_signals_event)
self._speeding_analyzer = SpeedingAnalyzer(self._speeding_event)
self._green = Green(1, self._green_event)
self._aggressive_analyze = AggressiveAnalyzer(self._aggressive_event)
self._evaluatebox_last_time = None
if USE_GUI:
self._gui = GUI(self._receiver.is_alive)
t = threading.Thread(target=self._mainloop, daemon=True)
t.daemon = True
t.start()
if UPLOAD:
self._distributor = Distributor(
'/src/client/upload_stream.sock', self._send_frequency)
s = threading.Thread(target=self._sender, daemon=True)
s.start()
if USE_GUI:
self._gui.mainloop()
else:
self._receiver.join() # keep main from exiting
def _mainloop(self):
while True:
sleep(1 / self._frequency)
if self._evaluatebox_last_time is None or self._evaluatebox_last_time - time.time() > 5:
self._evaluatebox_last_time = time.time()
self._gui._evaluate_box.set_value(EvaluateBox.GOOD)
if self._gui is None:
continue
if Geometry._pos is not None:
if Geometry._inter_pos:
self._gui.set_coords(*Geometry._inter_pos)
else:
self._gui.set_coords(*Geometry._pos)
if Geometry._marker is not None:
self._gui.set_marker(Geometry._marker)
if self._forgot_signals_event.is_set():
self._last_turn_forget = time.time()
self._gui._turn_signal_sym.set_vibrate(10)
self._forgot_signals_event.clear()
if self._last_turn_forget is not None and time.time() - self._last_turn_forget > 3:
self._gui._turn_signal_sym.set_vibrate(0)
if self._evaluatebox_last_time is None or time.time() - \
self._evaluatebox_last_time > 3:
if self._aggressive_event.is_set():
self._gui._evaluate_box.set_value(EvaluateBox.BAD)
self._evaluatebox_last_time = time.time()
self._aggressive_event.clear()
elif self._green_event.is_set():
if self._green_event.direction == 'up':
self._gui._evaluate_box.set_value(EvaluateBox.GEAR_UP)
elif self._green_event.direction == 'down':
self._gui._evaluate_box.set_value(
EvaluateBox.GEAR_DOWN)
self._evaluatebox_last_time = time.time()
self._green_event.clear()
else:
self._gui._evaluate_box.set_value(EvaluateBox.GOOD)
if self._speed_limit != self._speeding_event.speed_limit:
self._speed_limit = self._speeding_event.speed_limit
self._gui.set_speed_limit(self._speed_limit)
if self._speeding_event.is_set():
offset = 3
self._gui._speed_limit_sym.set_vibrate(
offset + (10 - offset) * self._speeding_event.speeding_percentage)
self._speeding_event.clear()
self._speed_time = time.time()
elif self._speed_time is not None and time.time() - self._speed_time > 2:
self._gui._speed_limit_sym.set_vibrate(0)
def _sender(self):
while True:
self._distributor.send()
sleep(1 / self._send_frequency)
