59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
import pickle
|
|
import sys
|
|
from random import randrange
|
|
|
|
from shared import daf
|
|
from shared.TrafficLightColor import Color
|
|
from shared.message_broker_wrapper import MBWrapper
|
|
|
|
sys.modules['daf'] = daf
|
|
|
|
|
|
class Orchestrator:
|
|
vins = []
|
|
tls = {}
|
|
|
|
_daf_mbs = {}
|
|
_velocity_mbs = {}
|
|
|
|
def __init__(self):
|
|
# TODO fetch available car VINs from Entity Ident
|
|
self.vins.append('SB164ABN1PE082000')
|
|
self.vins.append('SB999ABN1PE082111')
|
|
self.vins.append('SB555ABN1PE082555')
|
|
print('vins', self.vins)
|
|
|
|
# TODO fetch available TLs from Entity Ident
|
|
self.tls['traffic-light-1'] = {'color': Color.RED, 'switching_time': 120}
|
|
self.tls['traffic-light-2'] = {'color': Color.GREEN, 'switching_time': 240}
|
|
self.tls['traffic-light-3'] = {'color': Color.RED, 'switching_time': 360}
|
|
print('tls', self.tls)
|
|
|
|
def setup_msg_queues(self):
|
|
for vin in self.vins:
|
|
daf_channel = MBWrapper(exchange_name='vehicle_daf_{}'.format(vin), callback=self.handle_daf_receive)
|
|
daf_channel.setup_receiver()
|
|
self._daf_mbs[vin] = daf_channel
|
|
|
|
velocity_channel = MBWrapper(exchange_name='vehicle_velocity_{}'.format(vin))
|
|
velocity_channel.setup_sender()
|
|
self._velocity_mbs[vin] = velocity_channel
|
|
|
|
for tl in self.tls:
|
|
tl_channel = MBWrapper(exchange_name='traffic_light_state_{}'.format(tl), callback=self.handle_tl_state_receive)
|
|
tl_channel.setup_receiver()
|
|
|
|
def handle_daf_receive(self, msg):
|
|
received_daf_object = pickle.loads(msg)
|
|
print(received_daf_object)
|
|
|
|
# TODO use the data from the traffic lights (self.tls)
|
|
# to transmit a new target velocity for this vehicle to achieve a green wave
|
|
response_channel = self._velocity_mbs[received_daf_object.vehicle_identification_number]
|
|
response_channel.send(randrange(0, 130))
|
|
|
|
def handle_tl_state_receive(self, msg):
|
|
tl_state = pickle.loads(msg)
|
|
self.tls[tl_state['tlid']]['color'] = tl_state['color']
|
|
print(tl_state)
|