""" Traffic light module SPDX - License - Identifier: LGPL - 3.0 - or -later Copyright © 2021 Project Author Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca Contributor Milad milad.aghamohamadnia@concordia.ca Contributor Guille guille.gutierrezmorote@concordia.ca """ import ast from typing import List, Union from city_model_structure.transport.phase import Phase from city_model_structure.transport.traffic_node import TrafficNode class TrafficLight(TrafficNode): """ Traffic light class """ def __init__(self, name, coordinates, offset, phases=None, edges=None, right_on_red=False): super().__init__(name, coordinates, edges=edges, node_type='TrafficLight') if phases is None: phases = [] self._right_on_red = right_on_red self._offset = offset self._phases = phases @property def right_on_red(self) -> Union[None, bool]: """ Get if is possible to turn right when the traffic light is red :return: None or Boolean """ return self._right_on_red @right_on_red.setter def right_on_red(self, value): """ Get if is possible to turn right when the traffic light is red :param value: Boolean """ if value is not None: self._right_on_red = ast.literal_eval(value) @property def offset(self) -> Union[None, int]: """ Get program initial time offset :return: None or int """ return self._offset @offset.setter def offset(self, value): """ Set program initial time offset :param value: int """ if value is not None: self._offset = int(value) @property def phases(self) -> List[Phase]: """ Get traffic light logic phases :return: [Phase] """ return self._phases @phases.setter def phases(self, value): """ Set traffic light logic phases :param value: [Phase] """ self._phases = value