main.py 52 KB

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