main.py 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513
  1. import matplotlib
  2. matplotlib.use("TkAgg")
  3. from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
  4. from matplotlib.figure import Figure
  5. from matplotlib import rcParams
  6. rcParams.update({'figure.autolayout': True})
  7. from Tkinter import *
  8. from PIL import Image, ImageTk
  9. import tkSimpleDialog
  10. import urllib
  11. import urllib2
  12. import json
  13. import time
  14. JUMP = 50
  15. MAX_WIDTH = 10 * JUMP
  16. MAX_HEIGHT = 10 * JUMP
  17. CLICK_TOLERANCE = 5
  18. address = "http://127.0.0.1:8001"
  19. username = "test"
  20. root = Tk()
  21. names = {}
  22. event_entry = StringVar()
  23. setting_initial = False
  24. request_new_state = False
  25. canvas = Canvas(root, width=MAX_WIDTH, height=MAX_HEIGHT, bg="white")
  26. name = 0
  27. class FakeLayer():
  28. def __init__(self, address):
  29. self.types = {}
  30. self.sources = {}
  31. self.targets = {}
  32. self.attrs = {}
  33. def send_event(self, event):
  34. pass
  35. def read_available_attributes(self, name):
  36. return [("name", "String")]
  37. def read_attribute(self, name, attr):
  38. return self.attrs.get(name, {}).get(attr, None)
  39. def set_attribute(self, name, attr, value):
  40. self.attrs.setdefault(name, {})[attr] = value
  41. def instantiate_element(self, name, block_type):
  42. self.types[name] = block_type
  43. def instantiate_link(self, name, link_type, source, target):
  44. self.types[name] = link_type
  45. self.sources[name] = source
  46. self.targets[name] = target
  47. self.attrs[name] = {}
  48. def simulate(self):
  49. pass
  50. def pause(self):
  51. pass
  52. def delete(self):
  53. pass
  54. def switch_initial(self, name):
  55. pass
  56. def auto_sanitize(self, auto):
  57. pass
  58. def set_current(self, name):
  59. pass
  60. attribute = []
  61. available_attrs = []
  62. inp_evts = []
  63. state = []
  64. outp_evts = []
  65. #inp_evts = [(2, "arm"), (6, "detected")]
  66. #state = [(0, "idle"), (2, "armed"), (6, "detected"), (8, "idle")]
  67. #outp_evts = [(6, "soundAlarm"), ]
  68. do_color = "grey"
  69. def poll(address):
  70. simulation_time = None
  71. working_available_attrs = []
  72. while 1:
  73. returnvalue = json.loads(urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "get_output", "username": username}))).read())
  74. print("Process " + str(returnvalue))
  75. if (returnvalue.startswith("AVAILABLE_ATTR_VALUE")):
  76. working_available_attrs.append([json.loads(returnvalue.split(" ", 1)[1]), None])
  77. elif (returnvalue.startswith("AVAILABLE_ATTR_TYPE")):
  78. working_available_attrs[-1][1] = json.loads(returnvalue.split(" ", 1)[1])
  79. elif (returnvalue.startswith("AVAILABLE_ATTR_END")):
  80. available_attrs.append(working_available_attrs)
  81. working_available_attrs = []
  82. elif (returnvalue.startswith("ATTR_VALUE")):
  83. v = returnvalue.split(" ", 1)[1]
  84. if v == "None":
  85. v = None
  86. else:
  87. v = json.loads(v)
  88. attribute.append(v)
  89. elif (returnvalue.startswith("SIM_TIME")):
  90. simulation_time = json.loads(returnvalue.split(" ", 1)[1])
  91. elif (returnvalue.startswith("SIM_STATE")):
  92. state.append((simulation_time, returnvalue.split(" ", 1)[1]))
  93. elif (returnvalue.startswith("SIM_EVENT")):
  94. inp_evts.append((simulation_time, returnvalue.split(" ", 1)[1]))
  95. elif (returnvalue.startswith("SIM_RAISE")):
  96. outp_evts.append((simulation_time, returnvalue.split(" ", 1)[1]))
  97. elif (returnvalue.startswith("CONFORMANCE_OK")):
  98. global do_color
  99. do_color = "grey"
  100. elif (returnvalue.startswith("CONFORMANCE_FAIL")):
  101. global do_color
  102. do_color = "red"
  103. elif (returnvalue.startswith("REQUEST_CURRENT_STATE")):
  104. global do_color
  105. global request_new_state
  106. do_color = "blue"
  107. request_new_state = True
  108. else:
  109. print("Error: got unknown result: " + returnvalue)
  110. class MvLayer():
  111. def __init__(self, address):
  112. import threading
  113. self.address = address
  114. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % username, "username": "user_manager"}))).read()
  115. thrd = threading.Thread(target=poll, args=[address])
  116. thrd.daemon = True
  117. thrd.start()
  118. def color():
  119. while 1:
  120. global do_color
  121. root.configure(bg=do_color)
  122. time.sleep(0.1)
  123. thrd = threading.Thread(target=color)
  124. thrd.daemon = True
  125. thrd.start()
  126. def read_available_attributes(self, name):
  127. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"read_available_attributes"', "username": username}))).read()
  128. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % name, "username": username}))).read()
  129. while 1:
  130. try:
  131. print("Attribute: " + str(available_attrs))
  132. return available_attrs.pop(0)
  133. except IndexError:
  134. time.sleep(0.1)
  135. def read_attribute(self, name, attr):
  136. print("Sending read_attribute")
  137. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"read_attribute"', "username": username}))).read()
  138. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % name, "username": username}))).read()
  139. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % attr, "username": username}))).read()
  140. print("Waiting for attribute")
  141. while 1:
  142. try:
  143. print("Attribute: " + str(attribute))
  144. return attribute.pop(0)
  145. except IndexError:
  146. time.sleep(0.1)
  147. def set_attribute(self, name, attr, value):
  148. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"set_attribute"', "username": username}))).read()
  149. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % name, "username": username}))).read()
  150. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % attr, "username": username}))).read()
  151. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": json.dumps(value), "username": username}))).read()
  152. def instantiate_element(self, name, block_type):
  153. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"instantiate_node"', "username": username}))).read()
  154. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (block_type), "username": username}))).read()
  155. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (name), "username": username}))).read()
  156. def instantiate_link(self, name, link_type, source, target):
  157. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"instantiate_association"', "username": username}))).read()
  158. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (link_type), "username": username}))).read()
  159. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (name), "username": username}))).read()
  160. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (source), "username": username}))).read()
  161. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (target), "username": username}))).read()
  162. def simulate(self):
  163. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"simulate"', "username": username}))).read()
  164. def pause(self):
  165. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"pause"', "username": username}))).read()
  166. def send_event(self, event):
  167. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"event"', "username": username}))).read()
  168. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (event), "username": username}))).read()
  169. def delete(self, block):
  170. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"delete_element"', "username": username}))).read()
  171. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (block), "username": username}))).read()
  172. def switch_initial(self, name):
  173. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"switch_initial"', "username": username}))).read()
  174. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (name), "username": username}))).read()
  175. def auto_sanitize(self, auto):
  176. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"auto_sanitize"', "username": username}))).read()
  177. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": "true" if auto else "false", "username": username}))).read()
  178. def set_current(self, name):
  179. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (name), "username": username}))).read()
  180. def lower(value):
  181. return value / JUMP * JUMP
  182. def upper(value):
  183. return (value / JUMP + 1) * JUMP
  184. def avg(a, b):
  185. return float(a + b) / 2
  186. class InterfaceCore():
  187. drawn = set()
  188. refs = dict()
  189. mv = MvLayer(address)
  190. #mv = FakeLayer(address)
  191. def delete(self, x, y):
  192. lname = self.find((x, y))
  193. self.mv.delete(lname)
  194. for i in self.refs[lname]:
  195. canvas.delete(i)
  196. del self.refs[lname]
  197. if lname in names:
  198. self.canvas.delete(names[lname])
  199. del names[lname]
  200. self.drawn = set([e for e in self.drawn if e[4] != lname])
  201. def clicked(self, event):
  202. if self.find((event.x, event.y)):
  203. # Something already there, so don't add, but modify
  204. lname = self.find((event.x, event.y))
  205. global request_new_state
  206. if request_new_state:
  207. request_new_state = True
  208. self.mv.set_current(lname)
  209. else:
  210. attrs = self.mv.read_available_attributes(lname)
  211. print("Processing attributes " + str(attrs))
  212. for attr, t in attrs:
  213. print("Reading attribute " + str(attr))
  214. old_value = self.mv.read_attribute(lname, attr)
  215. print("Got value " + str(old_value))
  216. if old_value == "None":
  217. old_value = None
  218. new_value = tkSimpleDialog.askstring("Attribute modification", attr, initialvalue=old_value)
  219. canvas.focus_set()
  220. if t == "Float":
  221. new_value = float(new_value)
  222. elif t == "String":
  223. new_value = str(new_value)
  224. elif t == "Natural":
  225. new_value = int(new_value)
  226. else:
  227. print("Got unknown type: " + str(t))
  228. self.mv.set_attribute(lname, attr, new_value)
  229. print("Set value")
  230. if attr in ["name", "event"]:
  231. if lname in names:
  232. self.canvas.delete(names[lname])
  233. del names[lname]
  234. entry = [x for x in self.drawn if str(x[4]) == str(lname)][0]
  235. xc, xy = avg(entry[0], entry[2]), avg(entry[1], entry[3])
  236. names[lname] = self.canvas.create_text(xc, xy, text=new_value)
  237. else:
  238. global name
  239. x = event.x
  240. y = event.y
  241. self.mv.instantiate_element(str(name), "State")
  242. r = canvas.create_oval(lower(x), lower(y), upper(x), upper(y), fill="white")
  243. b = (lower(x), lower(y), upper(x), upper(y), str(name), "NODE")
  244. self.drawn.add(b)
  245. self.refs[str(name)] = [r]
  246. name += 1
  247. def switch_initial(self, evt):
  248. if self.find((evt.x, evt.y)):
  249. lname = self.find((evt.x, evt.y))
  250. self.mv.switch_initial(lname)
  251. # Update visual representation
  252. for f in self.refs.values():
  253. self.canvas.itemconfigure(f, width=1)
  254. self.canvas.itemconfigure(self.refs[lname], width=4)
  255. def find(self, location):
  256. def sqrt(a):
  257. return a**0.5
  258. x, y = location
  259. matches = []
  260. for e in self.drawn:
  261. x1, y1, x2, y2, x0, y0 = e[0], e[1], e[2], e[3], x, y
  262. if e[5] == "LINK":
  263. distance = abs((y2 - y1) * x0 - (x2 - x1) * y0 + x2 * y1 - y2 * x1) / sqrt((y2 - y1) ** 2 + (x2 - x1) ** 2) - CLICK_TOLERANCE
  264. elif e[5] == "NODE":
  265. if x0 <= x2 and x0 > x1 and y0 <= y2 and y0 > y1:
  266. distance = 0.0
  267. else:
  268. distance = float("inf")
  269. matches.append((distance, e[4]))
  270. if matches:
  271. minimal_distance, name = sorted(matches)[0]
  272. if minimal_distance <= 0:
  273. return name
  274. return None
  275. def draw(self, start, end):
  276. source = self.find(start)
  277. target = self.find(end)
  278. if source and target:
  279. global name
  280. self.mv.instantiate_link(str(name), "Transition", source, target)
  281. self.refs[str(name)] = [canvas.create_line(start[0], start[1], end[0], end[1], fill="black", arrow=LAST)]
  282. self.drawn.add((start[0], start[1], end[0], end[1], str(name), "LINK"))
  283. name += 1
  284. def add_event(self, event):
  285. self.mv.send_event(event)
  286. core = InterfaceCore()
  287. def clicked(event):
  288. if setting_initial:
  289. core.switch_initial(event)
  290. else:
  291. core.clicked(event)
  292. def draw(event):
  293. global start_location
  294. start_location = (event.x, event.y)
  295. def release(event):
  296. core.draw(start_location, (event.x, event.y))
  297. def simulate():
  298. core.mv.simulate()
  299. def pause():
  300. core.mv.pause()
  301. def delete(event):
  302. core.delete(event.x, event.y)
  303. def add_event():
  304. core.add_event(event_entry.get())
  305. def control_released(evt):
  306. global setting_initial
  307. setting_initial = False
  308. def control_pressed(evt):
  309. global setting_initial
  310. setting_initial = True
  311. def auto_sanitize():
  312. core.mv.auto_sanitize(True)
  313. def manual_sanitize():
  314. core.mv.auto_sanitize(False)
  315. buttons = [
  316. Button(root, text="START", command=simulate),
  317. Button(root, text="PAUSE", command=pause),
  318. Button(root, text="AUTO", command=auto_sanitize),
  319. Button(root, text="MANUAL", command=manual_sanitize),
  320. Entry(root, textvariable=event_entry),
  321. Button(root, text="EVENT", command=add_event),
  322. ]
  323. for i, b in enumerate(buttons):
  324. b.grid(row=0, column=i)
  325. canvas.grid(row=1,column=0,columnspan=len(buttons))
  326. core.canvas = canvas
  327. for i in range(JUMP, MAX_HEIGHT, JUMP):
  328. canvas.create_line(0, i, MAX_HEIGHT, i, fill="grey")
  329. for i in range(JUMP, MAX_WIDTH, JUMP):
  330. canvas.create_line(i, 0, i, MAX_WIDTH, fill="grey")
  331. canvas.focus_set()
  332. canvas.bind("<KeyPress-Control_L>", control_pressed)
  333. canvas.bind("<KeyRelease-Control_L>", control_released)
  334. canvas.bind("<Button-1>", clicked)
  335. canvas.bind("<Button-2>", delete)
  336. canvas.bind("<Button-3>", draw)
  337. canvas.bind("<ButtonRelease-3>", release)
  338. visual = Toplevel(root)
  339. # Example:
  340. # simulation = [(1, "A"), (3, "B"), (4, "A")]
  341. #inp_evts = [(2, "arm"), (6, "detected")]
  342. #state = [(0, "idle"), (2, "armed"), (6, "detected"), (8, "idle")]
  343. #outp_evts = [(6, "soundAlarm"), ]
  344. inp_evts = []
  345. state = []
  346. outp_evts = []
  347. frame = Frame(visual)
  348. frame.pack(fill=BOTH,expand=True)
  349. figsize = (6, 3)
  350. f_inp = Figure(figsize=figsize)
  351. a_inp = f_inp.add_subplot(111)
  352. a_inp.plot([], [])
  353. f_inp.suptitle("Inputs")
  354. fcanvas_inp = FigureCanvasTkAgg(f_inp, frame)
  355. fcanvas_inp.show()
  356. fcanvas_inp.get_tk_widget().pack()
  357. f_state = Figure(figsize=figsize)
  358. a_state = f_state.add_subplot(111)
  359. a_state.plot([], [])
  360. f_state.suptitle("State")
  361. fcanvas_state = FigureCanvasTkAgg(f_state, frame)
  362. fcanvas_state.show()
  363. fcanvas_state.get_tk_widget().pack()
  364. f_outp = Figure(figsize=figsize)
  365. a_outp = f_outp.add_subplot(111)
  366. a_outp.plot([], [])
  367. f_outp.suptitle("Outputs")
  368. fcanvas_outp = FigureCanvasTkAgg(f_outp, frame)
  369. fcanvas_outp.show()
  370. fcanvas_outp.get_tk_widget().pack()
  371. reverse_lookup_inp = {}
  372. reverse_lookup_state = {}
  373. reverse_lookup_outp = {}
  374. def update_graphs():
  375. # Input events
  376. times = [x[0] for x in inp_evts]
  377. events = [reverse_lookup_inp.setdefault(x[1], len(reverse_lookup_inp)) for x in inp_evts]
  378. lookup = [None] * len(reverse_lookup_inp)
  379. for k, v in reverse_lookup_inp.items():
  380. lookup[v] = k
  381. a_inp.clear()
  382. a_inp.plot(times, events, linestyle="none", marker="o")
  383. #f_inp.get_axes()[0].set_xbound(lower=0.0, upper=10.0)
  384. f_inp.get_axes()[0].set_yticks(range(-2, len(lookup) + 2))
  385. f_inp.get_axes()[0].set_yticklabels(["", ""] + lookup + ["", ""])
  386. fcanvas_inp.draw()
  387. # States
  388. times = [x[0] for x in state]
  389. events = [reverse_lookup_state.setdefault(x[1], len(reverse_lookup_state)) for x in state]
  390. lookup = [None] * len(reverse_lookup_state)
  391. for k, v in reverse_lookup_state.items():
  392. lookup[v] = k
  393. a_state.clear()
  394. a_state.plot(times, events, linestyle="solid", marker="o", drawstyle="steps-post")
  395. #f_state.get_axes()[0].set_xbound(lower=0.0, upper=10.0)
  396. f_state.get_axes()[0].set_yticks(range(-2, len(lookup) + 2))
  397. f_state.get_axes()[0].set_yticklabels(["", ""] + lookup + ["", ""])
  398. fcanvas_state.draw()
  399. # Input events
  400. times = [x[0] for x in outp_evts]
  401. events = [reverse_lookup_outp.setdefault(x[1], len(reverse_lookup_outp)) for x in outp_evts]
  402. lookup = [None] * len(reverse_lookup_outp)
  403. for k, v in reverse_lookup_outp.items():
  404. lookup[v] = k
  405. a_outp.clear()
  406. a_outp.plot(times, events, linestyle="none", marker="o")
  407. #f_outp.get_axes()[0].set_xbound(lower=0.0, upper=10.0)
  408. f_outp.get_axes()[0].set_yticks(range(-2, len(lookup) + 2))
  409. f_outp.get_axes()[0].set_yticklabels(["", ""] + lookup + ["", ""])
  410. fcanvas_outp.draw()
  411. root.after(500, update_graphs)
  412. root.after(500, update_graphs)
  413. root.mainloop()