from pypdevs.DEVS import CoupledDEVS
from src.DEVS.PolicedTrafficLight.Police import PoliceAgent
from src.DEVS.PolicedTrafficLight.TrafficLight import TrafficLight

class PolicedTraffic(CoupledDEVS):
    def __init__(self):
        CoupledDEVS.__init__(self, "Policed Traffic")
        self.agent = self.addSubModel(PoliceAgent())
        self.trafficLight = self.addSubModel(TrafficLight())

        self.connectPorts(self.agent.manualOverride, self.trafficLight.takeOver)