main.py 52 KB

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