main.py 54 KB

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