atomic_ext.py 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. ### Model
  2. from pypdevs.DEVS import *
  3. from pypdevs.infinity import INFINITY
  4. class TrafficLight(AtomicDEVS):
  5. def __init__(self):
  6. AtomicDEVS.__init__(self, "Light")
  7. self.state = "green"
  8. self.observe = self.addOutPort("observer")
  9. self.interrupt = self.addInPort("interrupt")
  10. def intTransition(self):
  11. state = self.state
  12. return {"red": "green",
  13. "yellow": "red",
  14. "green": "yellow"}[state]
  15. def timeAdvance(self):
  16. state = self.state
  17. return {"red": 60,
  18. "yellow": 3,
  19. "green": 57,
  20. "manual": INFINITY}[state]
  21. def outputFnc(self):
  22. state = self.state
  23. if state == "red":
  24. v = "green"
  25. elif state == "yellow":
  26. v = "red"
  27. elif state == "green":
  28. v = "yellow"
  29. return {self.observe: [v]}
  30. def extTransition(self, inputs):
  31. inp = inputs[self.interrupt][0]
  32. if inp == "manual":
  33. return "manual"
  34. elif inp == "auto":
  35. if self.state == "manual":
  36. return "red"
  37. ### Experiment
  38. from pypdevs.simulator import Simulator
  39. model = TrafficLight()
  40. sim = Simulator(model)
  41. sim.setVerbose()
  42. sim.setTerminationTime(500)
  43. sim.simulate()