main.py 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374
  1. import matplotlib
  2. matplotlib.use("TkAgg")
  3. from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
  4. from matplotlib.figure import Figure
  5. from Tkinter import *
  6. import tkSimpleDialog
  7. import urllib
  8. import urllib2
  9. import json
  10. import time
  11. JUMP = 40
  12. MAX_WIDTH = 20 * JUMP
  13. MAX_HEIGHT = 20 * JUMP
  14. address = "http://127.0.0.1:8001"
  15. username = "test"
  16. root = Tk()
  17. canvas = Canvas(root, width=MAX_WIDTH, height=MAX_HEIGHT, bg="white")
  18. canvas.pack()
  19. name = 0
  20. class FakeLayer():
  21. def __init__(self, address):
  22. self.types = {}
  23. self.sources = {}
  24. self.targets = {}
  25. self.attrs = {}
  26. def read_available_attributes(self, name):
  27. if self.types[name] == "const":
  28. return ["value"]
  29. else:
  30. return []
  31. def read_attribute(self, name, attr):
  32. return self.attr.get(name, {}).get(attr, None)
  33. def set_attribute(self, name, attr, value):
  34. self.attrs[name][attr] = value
  35. def instantiate_block(self, name, block_type):
  36. self.types[name] = block_type
  37. def instantiate_link(self, name, link_type, source, target):
  38. self.types[name] = link_type
  39. self.sources[name] = source
  40. self.targets[name] = target
  41. self.attrs[name] = {}
  42. def simulate(self):
  43. pass
  44. def step(self):
  45. pass
  46. def pause(self):
  47. pass
  48. attribute = []
  49. available_attrs = []
  50. simulation = []
  51. def poll(address):
  52. working_available_attrs = []
  53. working_simulation = None
  54. while 1:
  55. returnvalue = json.loads(urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "get_output", "username": username}))).read())
  56. print("Process " + str(returnvalue))
  57. if (returnvalue.startswith("AVAILABLE_ATTR_VALUE")):
  58. working_available_attrs.append([json.loads(returnvalue.split(" ", 1)[1]), None])
  59. elif (returnvalue.startswith("AVAILABLE_ATTR_TYPE")):
  60. working_available_attrs[-1][1] = json.loads(returnvalue.split(" ", 1)[1])
  61. elif (returnvalue.startswith("AVAILABLE_ATTR_END")):
  62. available_attrs.append(working_available_attrs)
  63. working_available_attrs = []
  64. elif (returnvalue.startswith("ATTR_VALUE")):
  65. v = returnvalue.split(" ", 1)[1]
  66. if v == "None":
  67. v = None
  68. else:
  69. v = json.loads(v)
  70. attribute.append(v)
  71. elif (returnvalue.startswith("SIM_TIME")):
  72. working_simulation = (json.loads(returnvalue.split(" ", 1)[1]), {})
  73. elif (returnvalue.startswith("SIM_PROBE")):
  74. blockname, blockvalue = returnvalue.split(" ", 1)[1].rsplit(" ", 1)
  75. working_simulation[1][json.loads(blockname)] = json.loads(blockvalue)
  76. elif (returnvalue.startswith("SIM_END")):
  77. simulation.append(working_simulation)
  78. working_simulation = None
  79. elif (returnvalue.startswith("CONFORMANCE_OK")):
  80. root.configure(background="grey")
  81. elif (returnvalue.startswith("CONFORMANCE_FAIL")):
  82. root.configure(background="red")
  83. else:
  84. print("Error: got unknown result: " + returnvalue)
  85. class MvLayer():
  86. def __init__(self, address):
  87. import threading
  88. self.address = address
  89. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % username, "username": "user_manager"}))).read()
  90. thrd = threading.Thread(target=poll, args=[address])
  91. thrd.daemon = True
  92. thrd.start()
  93. def read_available_attributes(self, name):
  94. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"read_available_attributes"', "username": username}))).read()
  95. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % name, "username": username}))).read()
  96. while not available_attrs:
  97. time.sleep(0.1)
  98. return available_attrs.pop(0)
  99. def read_attribute(self, name, attr):
  100. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"read_attribute"', "username": username}))).read()
  101. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % name, "username": username}))).read()
  102. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % attr, "username": username}))).read()
  103. while not attribute:
  104. time.sleep(0.1)
  105. return attribute.pop(0)
  106. def set_attribute(self, name, attr, value):
  107. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"set_attribute"', "username": username}))).read()
  108. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % name, "username": username}))).read()
  109. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % attr, "username": username}))).read()
  110. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": json.dumps(value), "username": username}))).read()
  111. def instantiate_block(self, name, block_type):
  112. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"instantiate_node"', "username": username}))).read()
  113. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (block_type), "username": username}))).read()
  114. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (name), "username": username}))).read()
  115. def instantiate_link(self, name, link_type, source, target):
  116. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"instantiate_association"', "username": username}))).read()
  117. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (link_type), "username": username}))).read()
  118. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (name), "username": username}))).read()
  119. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (source), "username": username}))).read()
  120. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (target), "username": username}))).read()
  121. def simulate(self):
  122. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"simulate"', "username": username}))).read()
  123. def step(self):
  124. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"step"', "username": username}))).read()
  125. def pause(self):
  126. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"pause"', "username": username}))).read()
  127. def delete(self, block):
  128. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"delete_element"', "username": username}))).read()
  129. urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "value": '"%s"' % (block), "username": username}))).read()
  130. def lower(value):
  131. return value / JUMP * JUMP
  132. def upper(value):
  133. return (value / JUMP + 1) * JUMP
  134. def avg(a, b):
  135. return float(a + b) / 2
  136. class InterfaceCore():
  137. mode = ""
  138. drawn = set()
  139. refs = dict()
  140. mv = MvLayer(address)
  141. #mv = FakeLayer(address)
  142. def set_mode(self, mode):
  143. self.mode = mode
  144. def delete(self, x, y):
  145. lname = self.find((x, y))
  146. self.mv.delete(lname)
  147. for i in self.refs[lname]:
  148. canvas.delete(i)
  149. del self.refs[lname]
  150. self.drawn = set([e for e in self.drawn if e[4] != lname])
  151. def clicked(self, event):
  152. if self.find((event.x, event.y)):
  153. # Something already there, so don't add, but modify
  154. lname = self.find((event.x, event.y))
  155. attrs = self.mv.read_available_attributes(lname)
  156. print("Managing " + str(attrs))
  157. if not attrs:
  158. print("No attrs to manage!")
  159. for attr, t in attrs:
  160. print("Reading data from " + str(attr))
  161. old_value = self.mv.read_attribute(lname, attr)
  162. if old_value == "None":
  163. old_value = None
  164. new_value = tkSimpleDialog.askstring("Attribute modification", attr, initialvalue=old_value)
  165. if t == "Float":
  166. new_value = float(new_value)
  167. else:
  168. print("Got unknown type: " + str(t))
  169. self.mv.set_attribute(lname, attr, new_value)
  170. elif self.mode not in ["AdditionBlock", "NegatorBlock", "ConstantBlock", "MultiplyBlock", "ConstantBlock", "InverseBlock", "DelayBlock", "IntegratorBlock", "DerivatorBlock"]:
  171. print("Cannot create something not guaranteed to be block type!")
  172. else:
  173. global name
  174. x = event.x
  175. y = event.y
  176. self.mv.instantiate_block(str(name), self.mode)
  177. r = canvas.create_rectangle(lower(x), lower(y), upper(x), upper(y), fill="white")
  178. t = canvas.create_text(avg(lower(x), upper(x)), avg(lower(y), upper(y)), text=self.mode, fill="black")
  179. b = (lower(x), lower(y), upper(x), upper(y), str(name))
  180. self.drawn.add(b)
  181. self.refs[str(name)] = [r, t]
  182. name += 1
  183. def find(self, location):
  184. x, y = location
  185. for e in self.drawn:
  186. if (e[0] <= x and
  187. e[1] <= y and
  188. e[2] >= x and
  189. e[3] >= y):
  190. return e[4]
  191. print("Found nothing at that location!")
  192. return []
  193. def draw(self, start, end):
  194. source = self.find(start)
  195. target = self.find(end)
  196. print("Connect from %s to %s" % (source, target))
  197. if source and target:
  198. if self.mode not in ["Link", "InitialCondition"]:
  199. print("Cannot create something not guaranteed to be link type!")
  200. elif source == target:
  201. print("Cannot create link to self")
  202. else:
  203. global name
  204. self.mv.instantiate_link(str(name), self.mode, source, target)
  205. self.refs[str(name)] = [canvas.create_line(start[0], start[1], end[0], end[1], fill="black", arrow=LAST)]
  206. name += 1
  207. core = InterfaceCore()
  208. def clicked(event):
  209. core.clicked(event)
  210. def draw(event):
  211. global start_location
  212. start_location = (event.x, event.y)
  213. def release(event):
  214. core.draw(start_location, (event.x, event.y))
  215. def simulate():
  216. core.mv.simulate()
  217. def step():
  218. core.mv.step()
  219. def pause():
  220. core.mv.pause()
  221. def addition():
  222. core.set_mode("AdditionBlock")
  223. def negation():
  224. core.set_mode("NegatorBlock")
  225. def link():
  226. core.set_mode("Link")
  227. def multiply():
  228. core.set_mode("MultiplyBlock")
  229. def constant():
  230. core.set_mode("ConstantBlock")
  231. def inverse():
  232. core.set_mode("InverseBlock")
  233. def ic():
  234. core.set_mode("InitialCondition")
  235. def delay():
  236. core.set_mode("DelayBlock")
  237. def derivator():
  238. core.set_mode("DerivatorBlock")
  239. def integrator():
  240. core.set_mode("IntegratorBlock")
  241. def delete(event):
  242. core.delete(event.x, event.y)
  243. Button(root, text="+", command=addition).pack()
  244. Button(root, text="-x", command=negation).pack()
  245. Button(root, text="*", command=multiply).pack()
  246. Button(root, text="1/x", command=inverse).pack()
  247. Button(root, text="Constant", command=constant).pack()
  248. Button(root, text="Delay", command=delay).pack()
  249. Button(root, text="d/dx", command=derivator).pack()
  250. Button(root, text="integrator", command=integrator).pack()
  251. Button(root, text="Link", command=link).pack()
  252. Button(root, text="InitialCondition", command=ic).pack()
  253. Button(root, text="SIM", command=simulate).pack()
  254. Button(root, text="STEP", command=step).pack()
  255. Button(root, text="PAUSE", command=pause).pack()
  256. core.canvas = canvas
  257. for i in range(JUMP, MAX_HEIGHT, JUMP):
  258. canvas.create_line(0, i, MAX_HEIGHT, i, fill="grey")
  259. for i in range(JUMP, MAX_WIDTH, JUMP):
  260. canvas.create_line(i, 0, i, MAX_WIDTH, fill="grey")
  261. canvas.bind("<Button-1>", clicked)
  262. canvas.bind("<Button-2>", delete)
  263. canvas.bind("<Button-3>", draw)
  264. canvas.bind("<ButtonRelease-3>", release)
  265. visual = Toplevel(root)
  266. probes = {}
  267. values = {}
  268. #simulation = [(1, {"a": 1, "b": 2}), (2, {"a": 3}), (3, {"a": 4, "b": 6})]
  269. def update_graphs():
  270. while simulation:
  271. t, results = simulation.pop(0)
  272. for k, v in results.items():
  273. if k in probes:
  274. fcanvas, a = probes[k]
  275. else:
  276. f = Figure(figsize=(5,5), dpi=100)
  277. a = f.add_subplot(111)
  278. a.plot([], [])
  279. fcanvas = FigureCanvasTkAgg(f, visual)
  280. fcanvas.show()
  281. fcanvas.get_tk_widget().pack()
  282. probes[k] = (fcanvas, a)
  283. values[k] = ([], [])
  284. values[k][0].append(t)
  285. values[k][1].append(v)
  286. a.clear()
  287. a.plot(values[k][0], values[k][1], linestyle="none", marker="o")
  288. fcanvas.draw()
  289. root.after(50, update_graphs)
  290. root.after(50, update_graphs)
  291. root.mainloop()