main.py 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009
  1. import modelverse_kernel.primitives as primitive_functions
  2. import modelverse_kernel.compiled as compiled_functions
  3. import modelverse_jit.jit as jit
  4. import sys
  5. import time
  6. if sys.version > '3': # pragma: no cover
  7. string_types = (str,)
  8. else:
  9. string_types = (str, unicode)
  10. class ModelverseKernel(object):
  11. def __init__(self, root):
  12. self.root = root
  13. self.returnvalue = None
  14. self.success = True
  15. self.generators = {}
  16. self.allow_compiled = True
  17. #self.allow_compiled = False
  18. # `self.jit` handles most JIT-related functionality.
  19. self.jit = jit.ModelverseJit()
  20. # To disable the JIT, uncomment the line below:
  21. # self.jit.set_jit_enabled(False)
  22. self.debug_info = "(no debug information found)"
  23. def execute_yields(self, username, operation, params, reply):
  24. try:
  25. self.success = True
  26. if username not in self.generators:
  27. self.generators[username] = {}
  28. if operation not in self.generators[username]:
  29. # Create the generator for the function to execute
  30. self.generators[username][operation] = getattr(self, operation)(username, *params)
  31. if reply is not None:
  32. return self.generators[username][operation].send(reply)
  33. else:
  34. return self.generators[username][operation].next()
  35. except StopIteration:
  36. # Done, so remove the generator
  37. del self.generators[username][operation]
  38. return None
  39. except:
  40. print("Unknown error @ %s" % self.debug_info)
  41. raise
  42. def execute_rule(self, username):
  43. user_root, = yield [("RD", [self.root, username])]
  44. user_frame, = yield [("RD", [user_root, "frame"])]
  45. inst, phase = yield [("RD", [user_frame, "IP"]),
  46. ("RD", [user_frame, "phase"]),
  47. ]
  48. self.new_debug, phase_v, inst_v = \
  49. yield [("RD", [inst, "__debug"]),
  50. ("RV", [phase]),
  51. ("RV", [inst]),
  52. ]
  53. if self.new_debug is not None:
  54. self.debug_info, = yield [("RV", [self.new_debug])]
  55. if phase_v == "finish":
  56. gen = self.helper_init(user_root)
  57. elif inst is None:
  58. raise Exception("Instruction pointer could not be found!")
  59. elif isinstance(phase_v, string_types):
  60. if phase_v == "init" and self.jit.is_jittable_entry_point(inst):
  61. #print("%-30s(%s)" % ("COMPILED " + str(self.jit.jitted_entry_points[inst]), phase_v))
  62. gen = self.execute_jit(user_root, inst, username)
  63. elif inst_v is None:
  64. raise Exception("%s: error understanding command (%s, %s)" % (self.debug_info, inst_v, phase_v))
  65. else:
  66. gen = self.get_inst_phase_generator(inst_v, phase_v, user_root)
  67. elif inst_v is None:
  68. raise Exception("%s: error understanding command (%s, %s)" % (self.debug_info, inst_v, phase_v))
  69. elif inst_v["value"] == "call":
  70. #print("%-30s(%s)" % ("call", "param"))
  71. gen = self.call_param(user_root)
  72. else:
  73. raise Exception("%s: error understanding command (%s, %s)" % (self.debug_info, inst_v, phase_v))
  74. try:
  75. inp = None
  76. while 1:
  77. inp = yield gen.send(inp)
  78. except StopIteration:
  79. pass
  80. except jit.JitCompilationFailedException as e:
  81. # Try again, but this time without the JIT.
  82. # print(e.message)
  83. gen = self.get_inst_phase_generator(inst_v, phase_v, user_root)
  84. try:
  85. inp = None
  86. while 1:
  87. inp = yield gen.send(inp)
  88. except StopIteration:
  89. pass
  90. def get_inst_phase_generator(self, inst_v, phase_v, user_root):
  91. """Gets a generator for the given instruction in the given phase,
  92. for the specified user root."""
  93. #print("%-30s(%s) -- %s" % (inst_v["value"], phase_v, username))
  94. return getattr(self, "%s_%s" % (inst_v["value"], phase_v))(user_root)
  95. ##########################
  96. ### Process primitives ###
  97. ##########################
  98. def load_primitives(self, username):
  99. hierarchy, = yield [("RD", [self.root, "__hierarchy"])]
  100. primitives, = yield [("RD", [hierarchy, "primitives"])]
  101. keys, = yield [("RDK", [primitives])]
  102. function_names = yield [("RV", [f]) for f in keys]
  103. signatures = yield [("RDN", [primitives, f]) for f in keys]
  104. bodies = yield [("RD", [f, "body"]) for f in signatures]
  105. for i in range(len(keys)):
  106. self.jit.register_compiled(
  107. bodies[i],
  108. getattr(primitive_functions, function_names[i]),
  109. function_names[i])
  110. def execute_jit(self, user_root, inst, username):
  111. # execute_jit
  112. user_frame, = yield [("RD", [user_root, "frame"])]
  113. symbols, = yield [("RD", [user_frame, "symbols"])]
  114. all_links, = yield [("RO", [symbols])]
  115. containers = yield [("RE", [v]) for v in all_links]
  116. outgoings = yield [("RO", [v]) for v in all_links]
  117. dict_values = yield [("RD", [v[1], "value"]) for v in containers]
  118. formals_1 = yield [("RE", [v[0]]) for v in outgoings]
  119. dict_keys_ref = yield [("RD", [v[1], "name"]) for v in formals_1]
  120. dict_keys = yield [("RV", [v]) for v in dict_keys_ref]
  121. parameters = dict(zip(dict_keys, dict_values))
  122. parameters["root"] = self.root
  123. parameters["user_root"] = user_root
  124. parameters["username"] = username
  125. parameters["mvk"] = self
  126. # Have the JIT compile the function.
  127. try:
  128. jit_gen = self.jit.jit_compile(user_root, inst)
  129. inp = None
  130. while 1:
  131. inp = yield jit_gen.send(inp)
  132. except primitive_functions.PrimitiveFinished as e:
  133. compiled_func = e.result
  134. # Run the compiled function.
  135. try:
  136. prim = compiled_func(**parameters)
  137. inp = None
  138. while 1:
  139. inp = yield prim.send(inp)
  140. except StopIteration:
  141. # Execution has ended without return value, so we have no idea what to do
  142. raise Exception("%s: primitive finished without returning a value!" % (self.debug_info))
  143. except primitive_functions.PrimitiveFinished as e:
  144. # Execution has ended with a returnvalue, so read it out from the exception being thrown
  145. result = e.result
  146. #if result is None:
  147. # raise Exception("Primitive raised exception: value of None for operation %s with parameters %s" % (compiled_func, str(parameters)))
  148. # Clean up the current stack, as if a return happened
  149. old_frame, = yield [("RD", [user_frame, "prev"])]
  150. lnk, = yield [("RDE", [old_frame, "returnvalue"])]
  151. _, _, _, _ = yield [("CD", [old_frame, "returnvalue", result]),
  152. ("CD", [user_root, "frame", old_frame]),
  153. ("DE", [lnk]),
  154. ("DN", [user_frame]),
  155. ]
  156. ########################################
  157. ### Execute input and output methods ###
  158. ########################################
  159. def get_output(self, username):
  160. user_root, = yield [("RD", [self.root, username])]
  161. first_output, = yield [("RD", [user_root, "output"])]
  162. next_output, rv = yield [("RD", [first_output, "next"]),
  163. ("RD", [first_output, "value"]),
  164. ]
  165. if next_output is None:
  166. self.success = False
  167. self.returnvalue = None
  168. else:
  169. rv_value, = yield [("RV", [rv])]
  170. _, _ = yield [("CD", [user_root, "output", next_output]),
  171. ("DN", [first_output]),
  172. ]
  173. self.returnvalue = rv_value
  174. def set_input(self, username, value):
  175. user_root, = yield [("RD", [self.root, username])]
  176. old_input, link = yield [("RD", [user_root, "last_input"]),
  177. ("RDE", [user_root, "last_input"]),
  178. ]
  179. new_input, = yield [("CN", [])]
  180. _, _ = yield [("CD", [user_root, "last_input", new_input]),
  181. ("CD", [old_input, "next", new_input]),
  182. ]
  183. new_value, = yield [("CNV", [value])]
  184. _, _ = yield [("CD", [old_input, "value", new_value]),
  185. ("DE", [link])
  186. ]
  187. self.returnvalue = {"id": 100, "value": "success"}
  188. #############################################
  189. ### Transformation rules for instructions ###
  190. #############################################
  191. def break_init(self, user_root):
  192. user_frame, = yield [("RD", [user_root, "frame"])]
  193. phase_link, ip_link = \
  194. yield [("RDE", [user_frame, "phase"]),
  195. ("RDE", [user_frame, "IP"])
  196. ]
  197. inst, = yield [("RD", [user_frame, "IP"])]
  198. while_inst, new_phase = \
  199. yield [("RD", [inst, "while"]),
  200. ("CNV", ["finish"]),
  201. ]
  202. _, _, _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  203. ("CD", [user_frame, "IP", while_inst]),
  204. ("DE", [phase_link]),
  205. ("DE", [ip_link]),
  206. ]
  207. def continue_init(self, user_root):
  208. user_frame, = yield [("RD", [user_root, "frame"])]
  209. ip_link, inst = yield [("RDE", [user_frame, "IP"]),
  210. ("RD", [user_frame, "IP"]),
  211. ]
  212. while_inst, = yield [("RD", [inst, "while"])]
  213. _, _ = yield [("CD", [user_frame, "IP", while_inst]),
  214. ("DE", [ip_link]),
  215. ]
  216. def if_init(self, user_root):
  217. user_frame, = yield [("RD", [user_root, "frame"])]
  218. evalstack, evalstack_link = \
  219. yield [("RD", [user_frame, "evalstack"]),
  220. ("RDE", [user_frame, "evalstack"]),
  221. ]
  222. inst, ip_link = yield [("RD", [user_frame, "IP"]),
  223. ("RDE", [user_frame, "IP"]),
  224. ]
  225. cond, = yield [("RD", [inst, "cond"])]
  226. new_evalstack, new_phase = \
  227. yield [("CN", []),
  228. ("CNV", ["cond"]),
  229. ]
  230. _, _, _, _, _, _, _ = \
  231. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  232. ("CD", [new_evalstack, "prev", evalstack]),
  233. ("CD", [user_frame, "IP", cond]),
  234. ("CD", [evalstack, "inst", inst]),
  235. ("CD", [evalstack, "phase", new_phase]),
  236. ("DE", [evalstack_link]),
  237. ("DE", [ip_link]),
  238. ]
  239. def if_cond(self, user_root):
  240. user_frame, = yield [("RD", [user_root, "frame"])]
  241. returnvalue, inst = yield [("RD", [user_frame, "returnvalue"]),
  242. ("RD", [user_frame, "IP"]),
  243. ]
  244. returnvalue_v, = yield [("RV", [returnvalue])]
  245. _else, = yield [("RD", [inst, "else"])]
  246. if returnvalue_v:
  247. phase_link, evalstack, evalstack_link, ip_link, _then, new_evalstack, evalstack_phase, new_phase = \
  248. yield [("RDE", [user_frame, "phase"]),
  249. ("RD", [user_frame, "evalstack"]),
  250. ("RDE", [user_frame, "evalstack"]),
  251. ("RDE", [user_frame, "IP"]),
  252. ("RD", [inst, "then"]),
  253. ("CN", []),
  254. ("CNV", ["finish"]),
  255. ("CNV", ["init"]),
  256. ]
  257. _, _, _, _, _, _, _, _, _ = \
  258. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  259. ("CD", [user_frame, "IP", _then]),
  260. ("CD", [new_evalstack, "prev", evalstack]),
  261. ("CD", [evalstack, "inst", inst]),
  262. ("CD", [evalstack, "phase", evalstack_phase]),
  263. ("CD", [user_frame, "phase", new_phase]),
  264. ("DE", [evalstack_link]),
  265. ("DE", [ip_link]),
  266. ("DE", [phase_link]),
  267. ]
  268. elif _else is None:
  269. phase_link, new_phase = \
  270. yield [("RDE", [user_frame, "phase"]),
  271. ("CNV", ["finish"]),
  272. ]
  273. _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  274. ("DE", [phase_link]),
  275. ]
  276. else:
  277. phase_link, evalstack, evalstack_link, ip_link = \
  278. yield [("RDE", [user_frame, "phase"]),
  279. ("RD", [user_frame, "evalstack"]),
  280. ("RDE", [user_frame, "evalstack"]),
  281. ("RDE", [user_frame, "IP"]),
  282. ]
  283. new_evalstack, new_phase, evalstack_phase = \
  284. yield [("CN", []),
  285. ("CNV", ["init"]),
  286. ("CNV", ["finish"]),
  287. ]
  288. _, _, _, _, _, _, _, _, _ = \
  289. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  290. ("CD", [user_frame, "IP", _else]),
  291. ("CD", [new_evalstack, "prev", evalstack]),
  292. ("CD", [evalstack, "inst", inst]),
  293. ("CD", [evalstack, "phase", evalstack_phase]),
  294. ("CD", [user_frame, "phase", new_phase]),
  295. ("DE", [evalstack_link]),
  296. ("DE", [ip_link]),
  297. ("DE", [phase_link]),
  298. ]
  299. def while_init(self, user_root):
  300. user_frame, = yield [("RD", [user_root, "frame"])]
  301. evalstack, evalstack_link, ip_link, inst = \
  302. yield [("RD", [user_frame, "evalstack"]),
  303. ("RDE", [user_frame, "evalstack"]),
  304. ("RDE", [user_frame, "IP"]),
  305. ("RD", [user_frame, "IP"]),
  306. ]
  307. cond, new_evalstack, new_phase = \
  308. yield [("RD", [inst, "cond"]),
  309. ("CN", []),
  310. ("CNV", ["cond"]),
  311. ]
  312. _, _, _, _, _, _, _ = \
  313. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  314. ("CD", [new_evalstack, "prev", evalstack]),
  315. ("CD", [user_frame, "IP", cond]),
  316. ("CD", [evalstack, "phase", new_phase]),
  317. ("CD", [evalstack, "inst", inst]),
  318. ("DE", [evalstack_link]),
  319. ("DE", [ip_link]),
  320. ]
  321. def while_cond(self, user_root):
  322. user_frame, = yield [("RD", [user_root, "frame"])]
  323. returnvalue, = yield [("RD", [user_frame, "returnvalue"])]
  324. returnvalue_v, = yield [("RV", [returnvalue])]
  325. if returnvalue_v:
  326. phase_link, evalstack, evalstack_link, ip_link, inst = \
  327. yield [("RDE", [user_frame, "phase"]),
  328. ("RD", [user_frame, "evalstack"]),
  329. ("RDE", [user_frame, "evalstack"]),
  330. ("RDE", [user_frame, "IP"]),
  331. ("RD", [user_frame, "IP"]),
  332. ]
  333. body, = yield [("RD", [inst, "body"])]
  334. new_evalstack, new_phase, evalstack_phase = \
  335. yield [("CN", []),
  336. ("CNV", ["init"]),
  337. ("CNV", ["init"]),
  338. ]
  339. _, _, _, _, _, _, _, _, _ = \
  340. yield [("CD", [user_frame, "IP", body]),
  341. ("CD", [user_frame, "phase", new_phase]),
  342. ("CD", [user_frame, "evalstack", new_evalstack]),
  343. ("CD", [new_evalstack, "prev", evalstack]),
  344. ("CD", [evalstack, "inst", inst]),
  345. ("CD", [evalstack, "phase", evalstack_phase]),
  346. ("DE", [evalstack_link]),
  347. ("DE", [ip_link]),
  348. ("DE", [phase_link]),
  349. ]
  350. else:
  351. phase_link, new_phase = \
  352. yield [("RDE", [user_frame, "phase"]),
  353. ("CNV", ["finish"]),
  354. ]
  355. _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  356. ("DE", [phase_link])
  357. ]
  358. def access_init(self, user_root):
  359. user_frame, = yield [("RD", [user_root, "frame"])]
  360. evalstack, evalstack_link, inst, ip_link = \
  361. yield [("RD", [user_frame, "evalstack"]),
  362. ("RDE", [user_frame, "evalstack"]),
  363. ("RD", [user_frame, "IP"]),
  364. ("RDE", [user_frame, "IP"]),
  365. ]
  366. var, new_evalstack, new_phase = \
  367. yield [("RD", [inst, "var"]),
  368. ("CN", []),
  369. ("CNV", ["eval"]),
  370. ]
  371. _, _, _, _, _, _, _ = \
  372. yield [("CD", [user_frame, "IP", var]),
  373. ("CD", [user_frame, "evalstack", new_evalstack]),
  374. ("CD", [new_evalstack, "prev", evalstack]),
  375. ("CD", [evalstack, "inst", inst]),
  376. ("CD", [evalstack, "phase", new_phase]),
  377. ("DE", [evalstack_link]),
  378. ("DE", [ip_link]),
  379. ]
  380. def access_eval(self, user_root):
  381. user_frame, = yield [("RD", [user_root, "frame"])]
  382. phase_link, returnvalue_link, returnvalue = \
  383. yield [("RDE", [user_frame, "phase"]),
  384. ("RDE", [user_frame, "returnvalue"]),
  385. ("RD", [user_frame, "returnvalue"]),
  386. ]
  387. value, new_phase = yield [("RD", [returnvalue, "value"]),
  388. ("CNV", ["finish"]),
  389. ]
  390. _, _, _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  391. ("CD", [user_frame, "returnvalue", value]),
  392. ("DE", [phase_link]),
  393. ("DE", [returnvalue_link]),
  394. ]
  395. def resolve_init(self, user_root):
  396. user_frame, = yield [("RD", [user_root, "frame"])]
  397. symbols, evalstack, evalstack_link, ip_link, inst = \
  398. yield [("RD", [user_frame, "symbols"]),
  399. ("RD", [user_frame, "evalstack"]),
  400. ("RDE", [user_frame, "evalstack"]),
  401. ("RDE", [user_frame, "IP"]),
  402. ("RD", [user_frame, "IP"]),
  403. ]
  404. var, = yield [("RD", [inst, "var"])]
  405. variable, = yield [("RDN", [symbols, var])]
  406. if variable is None:
  407. phase_link, returnvalue_link, _globals, var_name = \
  408. yield [("RDE", [user_frame, "phase"]),
  409. ("RDE", [user_frame, "returnvalue"]),
  410. ("RD", [user_root, "globals"]),
  411. ("RV", [var]),
  412. ]
  413. variable, new_phase = \
  414. yield [("RD", [_globals, var_name]),
  415. ("CNV", ["finish"]),
  416. ]
  417. if variable is None:
  418. raise Exception("%s: not found as global: %s" % (self.debug_info, var_name))
  419. # Resolved a global, so this is a string
  420. # Potentially, this might even be a function that we have precompiled already!
  421. # So check whether this is the case or not
  422. if self.allow_compiled:
  423. compiled_function = getattr(compiled_functions, var_name, None)
  424. if compiled_function is not None:
  425. # We have a compiled function ready!
  426. # Now we have to bind the ID to the compiled functions
  427. # For this, we read out the body of the resolved data
  428. compiler_val, = yield [("RD", [variable, "value"])]
  429. compiler_body, = yield [("RD", [compiler_val, "body"])]
  430. self.jit.register_compiled(compiler_body, compiled_function, var_name)
  431. else:
  432. phase_link, returnvalue_link, new_phase = \
  433. yield [("RDE", [user_frame, "phase"]),
  434. ("RDE", [user_frame, "returnvalue"]),
  435. ("CNV", ["finish"]),
  436. ]
  437. _, _, _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  438. ("CD", [user_frame, "returnvalue", variable]),
  439. ("DE", [phase_link]),
  440. ("DE", [returnvalue_link]),
  441. ]
  442. def assign_init(self, user_root):
  443. user_frame, = yield [("RD", [user_root, "frame"])]
  444. evalstack, evalstack_link, ip_link, inst = \
  445. yield [("RD", [user_frame, "evalstack"]),
  446. ("RDE", [user_frame, "evalstack"]),
  447. ("RDE", [user_frame, "IP"]),
  448. ("RD", [user_frame, "IP"]),
  449. ]
  450. var, new_evalstack, new_phase = \
  451. yield [("RD", [inst, "var"]),
  452. ("CN", []),
  453. ("CNV", ["value"]),
  454. ]
  455. _, _, _, _, _, _, _ = \
  456. yield [("CD", [user_frame, "IP", var]),
  457. ("CD", [user_frame, "evalstack", new_evalstack]),
  458. ("CD", [new_evalstack, "prev", evalstack]),
  459. ("CD", [evalstack, "inst", inst]),
  460. ("CD", [evalstack, "phase", new_phase]),
  461. ("DE", [evalstack_link]),
  462. ("DE", [ip_link]),
  463. ]
  464. def assign_value(self, user_root):
  465. user_frame, = yield [("RD", [user_root, "frame"])]
  466. phase_link, evalstack, returnvalue, evalstack_link, ip_link, inst = \
  467. yield [("RDE", [user_frame, "phase"]),
  468. ("RD", [user_frame, "evalstack"]),
  469. ("RD", [user_frame, "returnvalue"]),
  470. ("RDE", [user_frame, "evalstack"]),
  471. ("RDE", [user_frame, "IP"]),
  472. ("RD", [user_frame, "IP"]),
  473. ]
  474. value, new_evalstack, new_phase, evalstack_phase = \
  475. yield [("RD", [inst, "value"]),
  476. ("CN", []),
  477. ("CNV", ["init"]),
  478. ("CNV", ["assign"]),
  479. ]
  480. _, _, _, _, _, _, _, _, _, _ = \
  481. yield [("CD", [user_frame, "variable", returnvalue]),
  482. ("CD", [user_frame, "phase", new_phase]),
  483. ("CD", [user_frame, "evalstack", new_evalstack]),
  484. ("CD", [new_evalstack, "prev", evalstack]),
  485. ("CD", [evalstack, "inst", inst]),
  486. ("CD", [evalstack, "phase", evalstack_phase]),
  487. ("CD", [user_frame, "IP", value]),
  488. ("DE", [evalstack_link]),
  489. ("DE", [phase_link]),
  490. ("DE", [ip_link]),
  491. ]
  492. def assign_assign(self, user_root):
  493. user_frame, = yield [("RD", [user_root, "frame"])]
  494. phase_link, returnvalue, variable_link, variable = \
  495. yield [("RDE", [user_frame, "phase"]),
  496. ("RD", [user_frame, "returnvalue"]),
  497. ("RDE", [user_frame, "variable"]),
  498. ("RD", [user_frame, "variable"]),
  499. ]
  500. value_link, new_phase = \
  501. yield [("RDE", [variable, "value"]),
  502. ("CNV", ["finish"]),
  503. ]
  504. _, _, _, _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  505. ("CD", [variable, "value", returnvalue]),
  506. ("DE", [variable_link]),
  507. ("DE", [value_link]),
  508. ("DE", [phase_link]),
  509. ]
  510. def return_init(self, user_root):
  511. user_frame, = yield [("RD", [user_root, "frame"])]
  512. inst, = yield [("RD", [user_frame, "IP"])]
  513. value, = yield [("RD", [inst, "value"])]
  514. if value is None:
  515. prev_frame, = yield [("RD", [user_frame, "prev"])]
  516. _, _ = yield [("CD", [user_root, "frame", prev_frame]),
  517. ("DN", [user_frame]),
  518. ]
  519. else:
  520. evalstack, evalstack_link, ip_link, new_evalstack, evalstack_phase = \
  521. yield [("RD", [user_frame, "evalstack"]),
  522. ("RDE", [user_frame, "evalstack"]),
  523. ("RDE", [user_frame, "IP"]),
  524. ("CN", []),
  525. ("CNV", ["eval"]),
  526. ]
  527. _, _, _, _, _, _, _ = \
  528. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  529. ("CD", [new_evalstack, "prev", evalstack]),
  530. ("CD", [evalstack, "inst", inst]),
  531. ("CD", [evalstack, "phase", evalstack_phase]),
  532. ("CD", [user_frame, "IP", value]),
  533. ("DE", [evalstack_link]),
  534. ("DE", [ip_link]),
  535. ]
  536. def return_eval(self, user_root):
  537. user_frame, = yield [("RD", [user_root, "frame"])]
  538. prev_frame, = yield [("RD", [user_frame, "prev"])]
  539. returnvalue, old_returnvalue_link = \
  540. yield [("RD", [user_frame, "returnvalue"]),
  541. ("RDE", [prev_frame, "returnvalue"]),
  542. ]
  543. _, _, _, _ = yield [("CD", [user_root, "frame", prev_frame]),
  544. ("CD", [prev_frame, "returnvalue", returnvalue]),
  545. ("DE", [old_returnvalue_link]),
  546. ("DN", [user_frame]),
  547. ]
  548. def constant_init(self, user_root):
  549. user_frame, = yield [("RD", [user_root, "frame"])]
  550. phase_link, returnvalue_link, inst = \
  551. yield [("RDE", [user_frame, "phase"]),
  552. ("RDE", [user_frame, "returnvalue"]),
  553. ("RD", [user_frame, "IP"]),
  554. ]
  555. node, new_phase = yield [("RD", [inst, "node"]),
  556. ("CNV", ["finish"]),
  557. ]
  558. _, _, _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  559. ("CD", [user_frame, "returnvalue", node]),
  560. ("DE", [returnvalue_link]),
  561. ("DE", [phase_link]),
  562. ]
  563. def helper_init(self, user_root):
  564. user_frame, = yield [("RD", [user_root, "frame"])]
  565. inst, = yield [("RD", [user_frame, "IP"])]
  566. next, = yield [("RD", [inst, "next"])]
  567. if next is None:
  568. ip_link, phase_link, evalstack_top = \
  569. yield [("RDE", [user_frame, "IP"]),
  570. ("RDE", [user_frame, "phase"]),
  571. ("RD", [user_frame, "evalstack"]),
  572. ]
  573. evalstack, = yield [("RD", [evalstack_top, "prev"])]
  574. evalstack_inst, evalstack_phase, evalstack_inst_link, evalstack_phase_link = \
  575. yield [("RD", [evalstack, "inst"]),
  576. ("RD", [evalstack, "phase"]),
  577. ("RDE", [evalstack, "inst"]),
  578. ("RDE", [evalstack, "phase"]),
  579. ]
  580. _, _, _, _, _, _, _, _ = \
  581. yield [("CD", [user_frame, "evalstack", evalstack]),
  582. ("CD", [user_frame, "IP", evalstack_inst]),
  583. ("CD", [user_frame, "phase", evalstack_phase]),
  584. ("DE", [ip_link]),
  585. ("DE", [phase_link]),
  586. ("DE", [evalstack_inst_link]),
  587. ("DE", [evalstack_phase_link]),
  588. ("DN", [evalstack_top]),
  589. ]
  590. else:
  591. ip_link, phase_link, new_phase = \
  592. yield [("RDE", [user_frame, "IP"]),
  593. ("RDE", [user_frame, "phase"]),
  594. ("CNV", ["init"]),
  595. ]
  596. _, _, _, _ = yield [("CD", [user_frame, "IP", next]),
  597. ("CD", [user_frame, "phase", new_phase]),
  598. ("DE", [ip_link]),
  599. ("DE", [phase_link]),
  600. ]
  601. def call_init(self, user_root):
  602. user_frame, = yield [("RD", [user_root, "frame"])]
  603. symbols, evalstack, evalstack_link, ip_link, inst = \
  604. yield [("RD", [user_frame, "symbols"]),
  605. ("RD", [user_frame, "evalstack"]),
  606. ("RDE", [user_frame, "evalstack"]),
  607. ("RDE", [user_frame, "IP"]),
  608. ("RD", [user_frame, "IP"]),
  609. ]
  610. func, params = yield [("RD", [inst, "func"]),
  611. ("RD", [inst, "params"]),
  612. ]
  613. if params is None:
  614. new_evalstack, evalstack_phase = \
  615. yield [("CN", []),
  616. ("CNV", ["call"]),
  617. ]
  618. _, _, _, _, _, _, _ = \
  619. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  620. ("CD", [new_evalstack, "prev", evalstack]),
  621. ("CD", [evalstack, "inst", inst]),
  622. ("CD", [evalstack, "phase", evalstack_phase]),
  623. ("CD", [user_frame, "IP", func]),
  624. ("DE", [evalstack_link]),
  625. ("DE", [ip_link]),
  626. ]
  627. else:
  628. new_evalstack,= yield [("CN", [])]
  629. _, _, _, _, _, _, _ = \
  630. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  631. ("CD", [new_evalstack, "prev", evalstack]),
  632. ("CD", [evalstack, "inst", inst]),
  633. ("CD", [evalstack, "phase", params]),
  634. ("CD", [user_frame, "IP", func]),
  635. ("DE", [evalstack_link]),
  636. ("DE", [ip_link]),
  637. ]
  638. def call_call(self, user_root):
  639. user_frame, = yield [("RD", [user_root, "frame"])]
  640. inst, = yield [("RD", [user_frame, "IP"])]
  641. param, = yield [("RD", [inst, "last_param"])]
  642. if param is None:
  643. returnvalue, = yield [("RD", [user_frame, "returnvalue"])]
  644. body, = yield [("RD", [returnvalue, "body"])]
  645. self.jit.mark_entry_point(body)
  646. phase_link, frame_link, prev_phase, new_phase, new_frame, new_evalstack, new_symbols, new_returnvalue = \
  647. yield [("RDE", [user_frame, "phase"]),
  648. ("RDE", [user_root, "frame"]),
  649. ("CNV", ["finish"]),
  650. ("CNV", ["init"]),
  651. ("CN", []),
  652. ("CN", []),
  653. ("CN", []),
  654. ("CN", []),
  655. ]
  656. _, _, _, _, _, _, _, _, _, _, _ = \
  657. yield [("CD", [user_root, "frame", new_frame]),
  658. ("CD", [new_frame, "evalstack", new_evalstack]),
  659. ("CD", [new_frame, "symbols", new_symbols]),
  660. ("CD", [new_frame, "returnvalue", new_returnvalue]),
  661. ("CD", [new_frame, "caller", inst]),
  662. ("CD", [new_frame, "phase", new_phase]),
  663. ("CD", [new_frame, "IP", body]),
  664. ("CD", [new_frame, "prev", user_frame]),
  665. ("CD", [user_frame, "phase", prev_phase]),
  666. ("DE", [phase_link]),
  667. ("DE", [frame_link]),
  668. ]
  669. else:
  670. newer_frames, invoking_frames = \
  671. yield [("RRD", [user_frame, "prev"]),
  672. ("RRD", [inst, "caller"]),
  673. ]
  674. new_frame = self.find_overlapping(newer_frames, invoking_frames)
  675. phase_link, frame_link, new_symbols, new_IP = \
  676. yield [("RDE", [user_frame, "phase"]),
  677. ("RDE", [user_root, "frame"]),
  678. ("RD", [new_frame, "symbols"]),
  679. ("RD", [new_frame, "IP"]),
  680. ]
  681. signature, = yield [("RRD", [new_IP, "body"])]
  682. signature = signature[0]
  683. sig_params, last_param = \
  684. yield [("RD", [signature, "params"]),
  685. ("RD", [inst, "last_param"]),
  686. ]
  687. body, = yield [("RD", [new_IP, "body"])]
  688. self.jit.mark_entry_point(body)
  689. name, = yield [("RD", [last_param, "name"])]
  690. name_value, = yield [("RV", [name])]
  691. returnvalue, formal_parameter, new_phase, variable = \
  692. yield [("RD", [user_frame, "returnvalue"]),
  693. ("RD", [sig_params, name_value]),
  694. ("CNV", ["finish"]),
  695. ("CN", []),
  696. ]
  697. _, _, _, t1 = yield [("CD", [user_root, "frame", new_frame]),
  698. ("CD", [user_frame, "phase", new_phase]),
  699. ("CD", [variable, "value", returnvalue]),
  700. ("CE", [new_symbols, variable]),
  701. ]
  702. _, _, _ = yield [("CE", [t1, formal_parameter]),
  703. ("DE", [frame_link]),
  704. ("DE", [phase_link]),
  705. ]
  706. def find_overlapping(self, a, b):
  707. newer_frames = set(a)
  708. invoking_frames = set(b)
  709. matches = list(newer_frames.intersection(invoking_frames))
  710. if len(matches) == 1:
  711. return matches[0]
  712. elif len(matches) > 1:
  713. raise Exception("%s: error: multiple overlapping elements" % self.debug_info)
  714. else:
  715. raise Exception("%s: error: could not find any overlap" % self.debug_info)
  716. def call_param(self, user_root):
  717. user_frame, = yield [("RD", [user_root, "frame"])]
  718. inst, phase = yield [("RD", [user_frame, "IP"]),
  719. ("RD", [user_frame, "phase"]),
  720. ]
  721. params, last_param = \
  722. yield [("RD", [inst, "params"]),
  723. ("RD", [inst, "last_param"]),
  724. ]
  725. next_param, = yield [("RD", [params, "next_param"])]
  726. if params == phase:
  727. phase_link, ip_link, returnvalue, param_value, evalstack, evalstack_link = \
  728. yield [("RDE", [user_frame, "phase"]),
  729. ("RDE", [user_frame, "IP"]),
  730. ("RD", [user_frame, "returnvalue"]),
  731. ("RD", [params, "value"]),
  732. ("RD", [user_frame, "evalstack"]),
  733. ("RDE", [user_frame, "evalstack"]),
  734. ]
  735. body, = yield [("RD", [returnvalue, "body"])]
  736. new_frame, prev_evalstack, new_phase, prev_phase, new_evalstack, new_symbols, new_returnvalue = \
  737. yield [("CN", []),
  738. ("CN", []),
  739. ("CNV", ["init"]),
  740. ("CNV", ["init"]),
  741. ("CN", []),
  742. ("CN", []),
  743. ("CN", []),
  744. ]
  745. _, _, _, _, _, _, _, _, _, _, _, _, _, _, _ = \
  746. yield [("CD", [new_frame, "evalstack", new_evalstack]),
  747. ("CD", [new_frame, "symbols", new_symbols]),
  748. ("CD", [new_frame, "returnvalue", new_returnvalue]),
  749. ("CD", [new_frame, "caller", inst]),
  750. ("CD", [new_frame, "phase", new_phase]),
  751. ("CD", [new_frame, "IP", body]),
  752. ("CD", [new_frame, "prev", user_frame]),
  753. ("CD", [user_frame, "phase", prev_phase]),
  754. ("CD", [user_frame, "IP", param_value]),
  755. ("CD", [prev_evalstack, "prev", evalstack]),
  756. ("CD", [evalstack, "inst", inst]),
  757. ("CD", [user_frame, "evalstack", prev_evalstack]),
  758. ("DE", [evalstack_link]),
  759. ("DE", [ip_link]),
  760. ("DE", [phase_link]),
  761. ]
  762. if next_param is not None:
  763. _ = yield [("CD", [evalstack, "phase", next_param])]
  764. else:
  765. evalstack_phase, = \
  766. yield [("CNV", ["call"])]
  767. _ = yield [("CD", [evalstack, "phase", evalstack_phase])]
  768. else:
  769. frame_link, phase_link, newer_frames, invoking_frames = \
  770. yield [("RDE", [user_root, "frame"]),
  771. ("RDE", [user_frame, "phase"]),
  772. ("RRD", [user_frame, "prev"]),
  773. ("RRD", [inst, "caller"]),
  774. ]
  775. new_frame = self.find_overlapping(newer_frames, invoking_frames)
  776. ip_link, evalstack, evalstack_link, new_symbols, new_IP = \
  777. yield [("RDE", [user_frame, "IP"]),
  778. ("RD", [user_frame, "evalstack"]),
  779. ("RDE", [user_frame, "evalstack"]),
  780. ("RD", [new_frame, "symbols"]),
  781. ("RD", [new_frame, "IP"]),
  782. ]
  783. signature, = yield [("RRD", [new_IP, "body"])]
  784. signature = signature[0]
  785. sig_params, = yield [("RD", [signature, "params"])]
  786. if last_param == phase:
  787. prev_param, = \
  788. yield [("RRD", [last_param, "next_param"])]
  789. prev_param = prev_param[0]
  790. name, = yield [("RD", [prev_param, "name"])]
  791. name_value, = \
  792. yield [("RV", [name])]
  793. evalstack_phase, = \
  794. yield [("CNV", ["call"])]
  795. _ = yield [("CD", [evalstack, "phase", evalstack_phase])]
  796. formal_parameter, param_value = \
  797. yield [("RD", [sig_params, name_value]),
  798. ("RD", [last_param, "value"]),
  799. ]
  800. else:
  801. param_b, = yield [("RD", [user_frame, "phase"])]
  802. param_c, param_a = \
  803. yield [("RD", [param_b, "next_param"]),
  804. ("RRD", [param_b, "next_param"]),
  805. ]
  806. param_a = param_a[0]
  807. name, param_value = \
  808. yield [("RD", [param_a, "name"]),
  809. ("RD", [param_b, "value"]),
  810. ]
  811. name_value, = \
  812. yield [("RV", [name])]
  813. formal_parameter, _ = \
  814. yield [("RD", [sig_params, name_value]),
  815. ("CD", [evalstack, "phase", param_c]),
  816. ]
  817. new_phase, new_evalstack, variable, returnvalue = \
  818. yield [("CNV", ["init"]),
  819. ("CN", []),
  820. ("CN", []),
  821. ("RD", [user_frame, "returnvalue"]),
  822. ]
  823. _, _, _, _, _, _ = \
  824. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  825. ("CD", [new_evalstack, "prev", evalstack]),
  826. ("CD", [evalstack, "inst", inst]),
  827. ("CD", [user_frame, "phase", new_phase]),
  828. ("CD", [user_frame, "IP", param_value]),
  829. ("CD", [variable, "value", returnvalue]),
  830. ]
  831. t1, = yield [("CE", [new_symbols, variable])]
  832. _, _, _, _ = \
  833. yield [("CE", [t1, formal_parameter]),
  834. ("DE", [phase_link]),
  835. ("DE", [ip_link]),
  836. ("DE", [evalstack_link]),
  837. ]
  838. def input_init(self, user_root):
  839. user_frame, = yield [("RD", [user_root, "frame"])]
  840. returnvalue_link, _input = \
  841. yield [("RDE", [user_frame, "returnvalue"]),
  842. ("RD", [user_root, "input"]),
  843. ]
  844. value, next, phase_link = \
  845. yield [("RD", [_input, "value"]),
  846. ("RD", [_input, "next"]),
  847. ("RDE", [user_frame, "phase"]),
  848. ]
  849. if value is not None:
  850. v = yield [("RV", [value])]
  851. _, _, finish = \
  852. yield [("CD", [user_frame, "returnvalue", value]),
  853. ("CD", [user_root, "input", next]),
  854. ("CNV", ["finish"]),
  855. ]
  856. _, _, _, _ = \
  857. yield [("CD", [user_frame, "phase", finish]),
  858. ("DN", [_input]),
  859. ("DE", [returnvalue_link]),
  860. ("DE", [phase_link]),
  861. ]
  862. self.input_value = value
  863. else:
  864. # No input yet, so just wait and don't advance IP or phase
  865. self.input_value = None
  866. self.success = False
  867. def output_init(self, user_root):
  868. user_frame, = yield [("RD", [user_root, "frame"])]
  869. evalstack, evalstack_link, ip_link, inst = \
  870. yield [("RD", [user_frame, "evalstack"]),
  871. ("RDE", [user_frame, "evalstack"]),
  872. ("RDE", [user_frame, "IP"]),
  873. ("RD", [user_frame, "IP"]),
  874. ]
  875. value, new_evalstack, evalstack_phase = \
  876. yield [("RD", [inst, "value"]),
  877. ("CN", []),
  878. ("CNV", ["output"]),
  879. ]
  880. _, _, _, _, _, _, _ = \
  881. yield [("CD", [user_frame, "evalstack", new_evalstack]),
  882. ("CD", [new_evalstack, "prev", evalstack]),
  883. ("CD", [evalstack, "inst", inst]),
  884. ("CD", [evalstack, "phase", evalstack_phase]),
  885. ("CD", [user_frame, "IP", value]),
  886. ("DE", [evalstack_link]),
  887. ("DE", [ip_link]),
  888. ]
  889. def output_output(self, user_root):
  890. user_frame, = yield [("RD", [user_root, "frame"])]
  891. returnvalue_link, returnvalue, last_output, phase_link, last_output_link, new_last_output, finish = \
  892. yield [("RDE", [user_frame, "returnvalue"]),
  893. ("RD", [user_frame, "returnvalue"]),
  894. ("RD", [user_root, "last_output"]),
  895. ("RDE", [user_frame, "phase"]),
  896. ("RDE", [user_root, "last_output"]),
  897. ("CN", []),
  898. ("CNV", ["finish"]),
  899. ]
  900. _, _, _, _, _, _ = \
  901. yield [("CD", [last_output, "value", returnvalue]),
  902. ("CD", [last_output, "next", new_last_output]),
  903. ("CD", [user_root, "last_output", new_last_output]),
  904. ("CD", [user_frame, "phase", finish]),
  905. ("DE", [last_output_link]),
  906. ("DE", [phase_link]),
  907. ]
  908. def declare_init(self, user_root):
  909. user_frame, = yield [("RD", [user_root, "frame"])]
  910. inst, = yield [("RD", [user_frame, "IP"])]
  911. new_var, symbols, phase_link, empty_node, new_phase = \
  912. yield [("RD", [inst, "var"]),
  913. ("RD", [user_frame, "symbols"]),
  914. ("RDE", [user_frame, "phase"]),
  915. ("CN", []),
  916. ("CNV", ["finish"]),
  917. ]
  918. exists, = yield [("RDN", [symbols, new_var])]
  919. if exists is None:
  920. new_edge, = yield [("CE", [symbols, empty_node])]
  921. _ = yield [("CE", [new_edge, new_var])]
  922. _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  923. ("DE", [phase_link]),
  924. ]
  925. def global_init(self, user_root):
  926. user_frame, = yield [("RD", [user_root, "frame"])]
  927. inst, = yield [("RD", [user_frame, "IP"])]
  928. new_var, global_symbols, phase_link, empty_node, new_phase = \
  929. yield [("RD", [inst, "var"]),
  930. ("RD", [user_root, "globals"]),
  931. ("RDE", [user_frame, "phase"]),
  932. ("CN", []),
  933. ("CNV", ["finish"]),
  934. ]
  935. value, = yield [("RV", [new_var])]
  936. exists, = yield [("RD", [global_symbols, value])]
  937. if exists is None:
  938. yield [("CD", [global_symbols, value, empty_node])]
  939. _, _ = yield [("CD", [user_frame, "phase", new_phase]),
  940. ("DE", [phase_link])
  941. ]