main.py 51 KB

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