main.py 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. import sys
  2. sys.path.append("wrappers")
  3. from modelverse import *
  4. import random, re, json, uuid
  5. from pprint import pprint
  6. from multiprocessing import Process, Pipe, freeze_support
  7. sys.path.append('services/DEVS/pypdevs/src')
  8. sys.path.append('services/DEVS/pypdevs/examples')
  9. from simulator import Controller
  10. from ps_model import Root
  11. import threading, time
  12. init(sys.argv[1])
  13. login("pypdevs_service", "my_password")
  14. def pypdevs_service(port):
  15. exec(service_get(port), globals(), {})
  16. controller = Controller(Root())
  17. def inputter():
  18. print("Waiting for input...")
  19. while 1:
  20. try:
  21. input = service_get(port)
  22. print('raw input = %s' % input)
  23. params = json.loads(input)
  24. print('json parsed = %s' % params)
  25. if not isinstance(params, list):
  26. params = [params]
  27. print("Sending input to simulator: %s" % params)
  28. if (len(params) > 1):
  29. controller.addInput(Event(params[0], "request", params[1:]))
  30. else:
  31. controller.addInput(Event(params[0], "request", []))
  32. except:
  33. import traceback
  34. traceback.print_exc()
  35. input_thread = threading.Thread(target=inputter)
  36. input_thread.daemon = True
  37. input_thread.start()
  38. output_listener = controller.addOutputListener(["reply"])
  39. def outputter():
  40. print("Waiting for output...")
  41. while 1:
  42. output_val = output_listener.fetch(-1)
  43. print("Got output from simulator: %s" % output_val)
  44. service_set(port, json.dumps({"name": output_val.getName(), "port": output_val.getPort(), "params": output_val.getParameters()}))
  45. output_thread = threading.Thread(target=outputter)
  46. output_thread.daemon = True
  47. output_thread.start()
  48. controller.start()
  49. service_register("pypdevs_simulator", pypdevs_service)
  50. try:
  51. while True:
  52. # Stay active, as we shouldn't exit while the service is running!
  53. time.sleep(1)
  54. finally:
  55. service_stop()