atomic_out.py 798 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. ### Model
  2. from pypdevs.DEVS import *
  3. class TrafficLightWithOutput(AtomicDEVS):
  4. def __init__(self):
  5. AtomicDEVS.__init__(self, "Light")
  6. self.state = "green"
  7. self.observe = self.addOutPort("observer")
  8. def intTransition(self):
  9. state = self.state
  10. return {"red": "green",
  11. "yellow": "red",
  12. "green": "yellow"}[state]
  13. def timeAdvance(self):
  14. state = self.state
  15. return {"red": 60,
  16. "yellow": 3,
  17. "green": 57}[state]
  18. def outputFnc(self):
  19. state = self.state
  20. if state == "red":
  21. v = "green"
  22. elif state == "yellow":
  23. v = "red"
  24. elif state == "green":
  25. v = "yellow"
  26. return {self.observe: [v]}
  27. ### Experiment
  28. from pypdevs.simulator import Simulator
  29. model = TrafficLightWithOutput()
  30. sim = Simulator(model)
  31. sim.setVerbose()
  32. sim.setTerminationTime(500)
  33. sim.simulate()