main.py 62 KB

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