memoize.c 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917
  1. /* Copyright 1995-2015,2018
  2. Free Software Foundation, Inc.
  3. This file is part of Guile.
  4. Guile is free software: you can redistribute it and/or modify it
  5. under the terms of the GNU Lesser General Public License as published
  6. by the Free Software Foundation, either version 3 of the License, or
  7. (at your option) any later version.
  8. Guile is distributed in the hope that it will be useful, but WITHOUT
  9. ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10. FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
  11. License for more details.
  12. You should have received a copy of the GNU Lesser General Public
  13. License along with Guile. If not, see
  14. <https://www.gnu.org/licenses/>. */
  15. #ifdef HAVE_CONFIG_H
  16. # include <config.h>
  17. #endif
  18. #include <string.h>
  19. #include <stdio.h>
  20. #include "alist.h"
  21. #include "boolean.h"
  22. #include "continuations.h"
  23. #include "dynstack.h"
  24. #include "eq.h"
  25. #include "expand.h"
  26. #include "gsubr.h"
  27. #include "list.h"
  28. #include "macros.h"
  29. #include "modules.h"
  30. #include "numbers.h"
  31. #include "pairs.h"
  32. #include "ports.h"
  33. #include "print.h"
  34. #include "srcprop.h"
  35. #include "strings.h"
  36. #include "symbols.h"
  37. #include "threads.h"
  38. #include "throw.h"
  39. #include "variable.h"
  40. #include "vectors.h"
  41. #include "memoize.h"
  42. #define CAR(x) SCM_CAR(x)
  43. #define CDR(x) SCM_CDR(x)
  44. #define CAAR(x) SCM_CAAR(x)
  45. #define CADR(x) SCM_CADR(x)
  46. #define CDAR(x) SCM_CDAR(x)
  47. #define CDDR(x) SCM_CDDR(x)
  48. #define CADDR(x) SCM_CADDR(x)
  49. #define CDDDR(x) SCM_CDDDR(x)
  50. #define CADDDR(x) SCM_CADDDR(x)
  51. #define VECTOR_REF(v, i) (SCM_VECTOR_REF (v, i))
  52. #define VECTOR_SET(v, i, x) (SCM_VECTOR_SET (v, i, x))
  53. #define VECTOR_LENGTH(v) (SCM_VECTOR_LENGTH (v))
  54. SCM_SYMBOL (sym_case_lambda_star, "case-lambda*");
  55. /* Primitives not exposed to general Scheme. */
  56. static SCM wind;
  57. static SCM unwind;
  58. static SCM push_fluid;
  59. static SCM pop_fluid;
  60. static SCM push_dynamic_state;
  61. static SCM pop_dynamic_state;
  62. static SCM
  63. do_wind (SCM in, SCM out)
  64. {
  65. scm_dynstack_push_dynwind (&SCM_I_CURRENT_THREAD->dynstack, in, out);
  66. return SCM_UNSPECIFIED;
  67. }
  68. static SCM
  69. do_unwind (void)
  70. {
  71. scm_dynstack_pop (&SCM_I_CURRENT_THREAD->dynstack);
  72. return SCM_UNSPECIFIED;
  73. }
  74. static SCM
  75. do_push_fluid (SCM fluid, SCM val)
  76. {
  77. scm_thread *thread = SCM_I_CURRENT_THREAD;
  78. scm_dynstack_push_fluid (&thread->dynstack, fluid, val,
  79. thread->dynamic_state);
  80. return SCM_UNSPECIFIED;
  81. }
  82. static SCM
  83. do_pop_fluid (void)
  84. {
  85. scm_thread *thread = SCM_I_CURRENT_THREAD;
  86. scm_dynstack_unwind_fluid (&thread->dynstack, thread->dynamic_state);
  87. return SCM_UNSPECIFIED;
  88. }
  89. static SCM
  90. do_push_dynamic_state (SCM state)
  91. {
  92. scm_thread *thread = SCM_I_CURRENT_THREAD;
  93. scm_dynstack_push_dynamic_state (&thread->dynstack, state,
  94. thread->dynamic_state);
  95. return SCM_UNSPECIFIED;
  96. }
  97. static SCM
  98. do_pop_dynamic_state (void)
  99. {
  100. scm_thread *thread = SCM_I_CURRENT_THREAD;
  101. scm_dynstack_unwind_dynamic_state (&thread->dynstack,
  102. thread->dynamic_state);
  103. return SCM_UNSPECIFIED;
  104. }
  105. /* {Evaluator memoized expressions}
  106. */
  107. scm_t_bits scm_tc16_memoized;
  108. #define MAKMEMO(n, args) \
  109. (scm_cons (SCM_I_MAKINUM (n), args))
  110. #define MAKMEMO_SEQ(head,tail) \
  111. MAKMEMO (SCM_M_SEQ, scm_cons (head, tail))
  112. #define MAKMEMO_IF(test, then, else_) \
  113. MAKMEMO (SCM_M_IF, scm_cons (test, scm_cons (then, else_)))
  114. #define FIXED_ARITY(nreq) \
  115. scm_list_1 (SCM_I_MAKINUM (nreq))
  116. #define REST_ARITY(nreq, rest) \
  117. scm_list_2 (SCM_I_MAKINUM (nreq), rest)
  118. #define FULL_ARITY(nreq, rest, nopt, kw, ninits, unbound, alt) \
  119. scm_list_n (SCM_I_MAKINUM (nreq), rest, SCM_I_MAKINUM (nopt), kw, \
  120. SCM_I_MAKINUM (ninits), unbound, alt, SCM_UNDEFINED)
  121. #define MAKMEMO_LAMBDA(body, arity, meta) \
  122. MAKMEMO (SCM_M_LAMBDA, \
  123. scm_cons (body, scm_cons (meta, arity)))
  124. #define MAKMEMO_CAPTURE_ENV(vars, body) \
  125. MAKMEMO (SCM_M_CAPTURE_ENV, scm_cons (vars, body))
  126. #define MAKMEMO_LET(inits, body) \
  127. MAKMEMO (SCM_M_LET, scm_cons (inits, body))
  128. #define MAKMEMO_QUOTE(exp) \
  129. MAKMEMO (SCM_M_QUOTE, exp)
  130. #define MAKMEMO_CAPTURE_MODULE(exp) \
  131. MAKMEMO (SCM_M_CAPTURE_MODULE, exp)
  132. #define MAKMEMO_APPLY(proc, args)\
  133. MAKMEMO (SCM_M_APPLY, scm_list_2 (proc, args))
  134. #define MAKMEMO_CONT(proc) \
  135. MAKMEMO (SCM_M_CONT, proc)
  136. #define MAKMEMO_CALL_WITH_VALUES(prod, cons) \
  137. MAKMEMO (SCM_M_CALL_WITH_VALUES, scm_cons (prod, cons))
  138. #define MAKMEMO_CALL(proc, args) \
  139. MAKMEMO (SCM_M_CALL, scm_cons (proc, args))
  140. #define MAKMEMO_LEX_REF(pos) \
  141. MAKMEMO (SCM_M_LEXICAL_REF, pos)
  142. #define MAKMEMO_LEX_SET(pos, val) \
  143. MAKMEMO (SCM_M_LEXICAL_SET, scm_cons (pos, val))
  144. #define MAKMEMO_BOX_REF(box) \
  145. MAKMEMO (SCM_M_BOX_REF, box)
  146. #define MAKMEMO_BOX_SET(box, val) \
  147. MAKMEMO (SCM_M_BOX_SET, scm_cons (box, val))
  148. #define MAKMEMO_TOP_BOX(mode, var) \
  149. MAKMEMO (SCM_M_RESOLVE, scm_cons (SCM_I_MAKINUM (mode), var))
  150. #define MAKMEMO_MOD_BOX(mode, mod, var, public) \
  151. MAKMEMO (SCM_M_RESOLVE, \
  152. scm_cons (SCM_I_MAKINUM (mode), \
  153. scm_cons (mod, scm_cons (var, public))))
  154. #define MAKMEMO_CALL_WITH_PROMPT(tag, thunk, handler) \
  155. MAKMEMO (SCM_M_CALL_WITH_PROMPT, scm_cons (tag, scm_cons (thunk, handler)))
  156. /* This table must agree with the list of M_ constants in memoize.h */
  157. static const char *const memoized_tags[] =
  158. {
  159. "seq",
  160. "if",
  161. "lambda",
  162. "capture-env",
  163. "let",
  164. "quote",
  165. "capture-module",
  166. "apply",
  167. "call/cc",
  168. "call-with-values",
  169. "call",
  170. "lexical-ref",
  171. "lexical-set!",
  172. "box-ref",
  173. "box-set!",
  174. "resolve",
  175. "call-with-prompt",
  176. };
  177. /* Memoization-time environments mirror the structure of eval-time
  178. environments. Each link in the chain at memoization-time corresponds
  179. to a link at eval-time.
  180. env := module | (link, env)
  181. module := #f | #t
  182. link := flat-link . nested-link
  183. flat-link := (#t . ((var . pos) ...))
  184. nested-link := (#f . #(var ...))
  185. A module of #f indicates that the current module has not yet been
  186. captured. Memoizing a capture-module expression will capture the
  187. module.
  188. Flat environments copy the values for a set of free variables into a
  189. flat environment, via the capture-env expression. During memoization
  190. a flat link collects the values of free variables, along with their
  191. resolved outer locations. We are able to copy values because the
  192. incoming expression has already been assignment-converted. Flat
  193. environments prevent closures from hanging on to too much memory.
  194. Nested environments have a rib of "let" bindings, and link to an
  195. outer environment.
  196. */
  197. static int
  198. try_lookup_rib (SCM x, SCM rib)
  199. {
  200. int idx = 0;
  201. for (; idx < VECTOR_LENGTH (rib); idx++)
  202. if (scm_is_eq (x, VECTOR_REF (rib, idx)))
  203. return idx; /* bound */
  204. return -1;
  205. }
  206. static int
  207. lookup_rib (SCM x, SCM rib)
  208. {
  209. int idx = try_lookup_rib (x, rib);
  210. if (idx < 0)
  211. abort ();
  212. return idx;
  213. }
  214. static SCM
  215. make_pos (int depth, int width)
  216. {
  217. return scm_cons (SCM_I_MAKINUM (depth), SCM_I_MAKINUM (width));
  218. }
  219. static SCM
  220. push_nested_link (SCM vars, SCM env)
  221. {
  222. return scm_acons (SCM_BOOL_F, vars, env);
  223. }
  224. static SCM
  225. push_flat_link (SCM env)
  226. {
  227. return scm_acons (SCM_BOOL_T, SCM_EOL, env);
  228. }
  229. static int
  230. env_link_is_flat (SCM env_link)
  231. {
  232. return scm_is_true (CAR (env_link));
  233. }
  234. static SCM
  235. env_link_vars (SCM env_link)
  236. {
  237. return CDR (env_link);
  238. }
  239. static void
  240. env_link_add_flat_var (SCM env_link, SCM var, SCM pos)
  241. {
  242. SCM vars = env_link_vars (env_link);
  243. if (scm_is_false (scm_assq (var, vars)))
  244. scm_set_cdr_x (env_link, scm_acons (var, pos, vars));
  245. }
  246. static SCM
  247. lookup (SCM x, SCM env)
  248. {
  249. int d = 0;
  250. for (; scm_is_pair (env); env = CDR (env), d++)
  251. {
  252. SCM link = CAR (env);
  253. if (env_link_is_flat (link))
  254. {
  255. int w;
  256. SCM vars;
  257. for (vars = env_link_vars (link), w = scm_ilength (vars) - 1;
  258. scm_is_pair (vars);
  259. vars = CDR (vars), w--)
  260. if (scm_is_eq (x, (CAAR (vars))))
  261. return make_pos (d, w);
  262. env_link_add_flat_var (link, x, lookup (x, CDR (env)));
  263. return make_pos (d, scm_ilength (env_link_vars (link)) - 1);
  264. }
  265. else
  266. {
  267. int w = try_lookup_rib (x, env_link_vars (link));
  268. if (w < 0)
  269. continue;
  270. return make_pos (d, w);
  271. }
  272. }
  273. abort ();
  274. }
  275. static SCM
  276. capture_flat_env (SCM lambda, SCM env)
  277. {
  278. int nenv;
  279. SCM vars, link, locs;
  280. link = CAR (env);
  281. vars = env_link_vars (link);
  282. nenv = scm_ilength (vars);
  283. locs = scm_c_make_vector (nenv, SCM_BOOL_F);
  284. for (; scm_is_pair (vars); vars = CDR (vars))
  285. scm_c_vector_set_x (locs, --nenv, CDAR (vars));
  286. return MAKMEMO_CAPTURE_ENV (locs, lambda);
  287. }
  288. /* Abbreviate SCM_EXPANDED_REF. Copied because I'm not sure about symbol pasting */
  289. #define REF(x,type,field) \
  290. (scm_struct_ref (x, SCM_I_MAKINUM (SCM_EXPANDED_##type##_##field)))
  291. static SCM list_of_guile = SCM_BOOL_F;
  292. static SCM memoize (SCM exp, SCM env);
  293. static SCM
  294. memoize_exps (SCM exps, SCM env)
  295. {
  296. SCM ret;
  297. for (ret = SCM_EOL; scm_is_pair (exps); exps = CDR (exps))
  298. ret = scm_cons (memoize (CAR (exps), env), ret);
  299. return scm_reverse_x (ret, SCM_UNDEFINED);
  300. }
  301. static SCM
  302. capture_env (SCM env)
  303. {
  304. if (scm_is_false (env))
  305. return SCM_BOOL_T;
  306. return env;
  307. }
  308. static SCM
  309. maybe_makmemo_capture_module (SCM exp, SCM env)
  310. {
  311. if (scm_is_false (env))
  312. return MAKMEMO_CAPTURE_MODULE (exp);
  313. return exp;
  314. }
  315. static SCM
  316. memoize (SCM exp, SCM env)
  317. {
  318. if (!SCM_EXPANDED_P (exp))
  319. abort ();
  320. switch (SCM_EXPANDED_TYPE (exp))
  321. {
  322. case SCM_EXPANDED_VOID:
  323. return MAKMEMO_QUOTE (SCM_UNSPECIFIED);
  324. case SCM_EXPANDED_CONST:
  325. return MAKMEMO_QUOTE (REF (exp, CONST, EXP));
  326. case SCM_EXPANDED_PRIMITIVE_REF:
  327. if (scm_is_eq (scm_current_module (), scm_the_root_module ()))
  328. return maybe_makmemo_capture_module
  329. (MAKMEMO_BOX_REF (MAKMEMO_TOP_BOX (SCM_EXPANDED_TOPLEVEL_REF,
  330. REF (exp, PRIMITIVE_REF, NAME))),
  331. env);
  332. else
  333. return MAKMEMO_BOX_REF (MAKMEMO_MOD_BOX (SCM_EXPANDED_MODULE_REF,
  334. list_of_guile,
  335. REF (exp, PRIMITIVE_REF, NAME),
  336. SCM_BOOL_F));
  337. case SCM_EXPANDED_LEXICAL_REF:
  338. return MAKMEMO_LEX_REF (lookup (REF (exp, LEXICAL_REF, GENSYM), env));
  339. case SCM_EXPANDED_LEXICAL_SET:
  340. return MAKMEMO_LEX_SET (lookup (REF (exp, LEXICAL_SET, GENSYM), env),
  341. memoize (REF (exp, LEXICAL_SET, EXP), env));
  342. case SCM_EXPANDED_MODULE_REF:
  343. return MAKMEMO_BOX_REF (MAKMEMO_MOD_BOX
  344. (SCM_EXPANDED_MODULE_REF,
  345. REF (exp, MODULE_REF, MOD),
  346. REF (exp, MODULE_REF, NAME),
  347. REF (exp, MODULE_REF, PUBLIC)));
  348. case SCM_EXPANDED_MODULE_SET:
  349. return MAKMEMO_BOX_SET (MAKMEMO_MOD_BOX
  350. (SCM_EXPANDED_MODULE_SET,
  351. REF (exp, MODULE_SET, MOD),
  352. REF (exp, MODULE_SET, NAME),
  353. REF (exp, MODULE_SET, PUBLIC)),
  354. memoize (REF (exp, MODULE_SET, EXP), env));
  355. case SCM_EXPANDED_TOPLEVEL_REF:
  356. return maybe_makmemo_capture_module
  357. (MAKMEMO_BOX_REF (MAKMEMO_TOP_BOX (SCM_EXPANDED_TOPLEVEL_REF,
  358. REF (exp, TOPLEVEL_REF, NAME))),
  359. env);
  360. case SCM_EXPANDED_TOPLEVEL_SET:
  361. return maybe_makmemo_capture_module
  362. (MAKMEMO_BOX_SET (MAKMEMO_TOP_BOX (SCM_EXPANDED_TOPLEVEL_SET,
  363. REF (exp, TOPLEVEL_SET, NAME)),
  364. memoize (REF (exp, TOPLEVEL_SET, EXP),
  365. capture_env (env))),
  366. env);
  367. case SCM_EXPANDED_TOPLEVEL_DEFINE:
  368. return maybe_makmemo_capture_module
  369. (MAKMEMO_BOX_SET (MAKMEMO_TOP_BOX (SCM_EXPANDED_TOPLEVEL_DEFINE,
  370. REF (exp, TOPLEVEL_DEFINE, NAME)),
  371. memoize (REF (exp, TOPLEVEL_DEFINE, EXP),
  372. capture_env (env))),
  373. env);
  374. case SCM_EXPANDED_CONDITIONAL:
  375. return MAKMEMO_IF (memoize (REF (exp, CONDITIONAL, TEST), env),
  376. memoize (REF (exp, CONDITIONAL, CONSEQUENT), env),
  377. memoize (REF (exp, CONDITIONAL, ALTERNATE), env));
  378. case SCM_EXPANDED_CALL:
  379. {
  380. SCM proc, args;
  381. proc = REF (exp, CALL, PROC);
  382. args = memoize_exps (REF (exp, CALL, ARGS), env);
  383. return MAKMEMO_CALL (memoize (proc, env), args);
  384. }
  385. case SCM_EXPANDED_PRIMCALL:
  386. {
  387. SCM name, args;
  388. int nargs;
  389. name = REF (exp, PRIMCALL, NAME);
  390. args = memoize_exps (REF (exp, PRIMCALL, ARGS), env);
  391. nargs = scm_ilength (args);
  392. if (nargs == 3
  393. && scm_is_eq (name, scm_from_latin1_symbol ("call-with-prompt")))
  394. return MAKMEMO_CALL_WITH_PROMPT (CAR (args),
  395. CADR (args),
  396. CADDR (args));
  397. else if (nargs == 2
  398. && scm_is_eq (name, scm_from_latin1_symbol ("apply")))
  399. return MAKMEMO_APPLY (CAR (args), CADR (args));
  400. else if (nargs == 1
  401. && scm_is_eq (name,
  402. scm_from_latin1_symbol
  403. ("call-with-current-continuation")))
  404. return MAKMEMO_CONT (CAR (args));
  405. else if (nargs == 2
  406. && scm_is_eq (name,
  407. scm_from_latin1_symbol ("call-with-values")))
  408. return MAKMEMO_CALL_WITH_VALUES (CAR (args), CADR (args));
  409. else if (nargs == 1
  410. && scm_is_eq (name,
  411. scm_from_latin1_symbol ("variable-ref")))
  412. return MAKMEMO_BOX_REF (CAR (args));
  413. else if (nargs == 2
  414. && scm_is_eq (name,
  415. scm_from_latin1_symbol ("variable-set!")))
  416. return MAKMEMO_BOX_SET (CAR (args), CADR (args));
  417. else if (nargs == 2
  418. && scm_is_eq (name, scm_from_latin1_symbol ("wind")))
  419. return MAKMEMO_CALL (MAKMEMO_QUOTE (wind), args);
  420. else if (nargs == 0
  421. && scm_is_eq (name, scm_from_latin1_symbol ("unwind")))
  422. return MAKMEMO_CALL (MAKMEMO_QUOTE (unwind), SCM_EOL);
  423. else if (nargs == 2
  424. && scm_is_eq (name, scm_from_latin1_symbol ("push-fluid")))
  425. return MAKMEMO_CALL (MAKMEMO_QUOTE (push_fluid), args);
  426. else if (nargs == 0
  427. && scm_is_eq (name, scm_from_latin1_symbol ("pop-fluid")))
  428. return MAKMEMO_CALL (MAKMEMO_QUOTE (pop_fluid), SCM_EOL);
  429. else if (nargs == 1
  430. && scm_is_eq (name,
  431. scm_from_latin1_symbol ("push-dynamic-state")))
  432. return MAKMEMO_CALL (MAKMEMO_QUOTE (push_dynamic_state), args);
  433. else if (nargs == 0
  434. && scm_is_eq (name,
  435. scm_from_latin1_symbol ("pop-dynamic-state")))
  436. return MAKMEMO_CALL (MAKMEMO_QUOTE (pop_dynamic_state), SCM_EOL);
  437. else if (scm_is_eq (scm_current_module (), scm_the_root_module ()))
  438. return MAKMEMO_CALL (maybe_makmemo_capture_module
  439. (MAKMEMO_BOX_REF
  440. (MAKMEMO_TOP_BOX (SCM_EXPANDED_TOPLEVEL_REF,
  441. name)),
  442. env),
  443. args);
  444. else
  445. return MAKMEMO_CALL (MAKMEMO_BOX_REF
  446. (MAKMEMO_MOD_BOX (SCM_EXPANDED_MODULE_REF,
  447. list_of_guile,
  448. name,
  449. SCM_BOOL_F)),
  450. args);
  451. }
  452. case SCM_EXPANDED_SEQ:
  453. return MAKMEMO_SEQ (memoize (REF (exp, SEQ, HEAD), env),
  454. memoize (REF (exp, SEQ, TAIL), env));
  455. case SCM_EXPANDED_LAMBDA:
  456. /* The body will be a lambda-case. */
  457. {
  458. SCM meta, body, proc, new_env;
  459. meta = REF (exp, LAMBDA, META);
  460. body = REF (exp, LAMBDA, BODY);
  461. new_env = push_flat_link (capture_env (env));
  462. proc = memoize (body, new_env);
  463. SCM_SETCAR (SCM_CDR (SCM_MEMOIZED_ARGS (proc)), meta);
  464. return maybe_makmemo_capture_module (capture_flat_env (proc, new_env),
  465. env);
  466. }
  467. case SCM_EXPANDED_LAMBDA_CASE:
  468. {
  469. SCM req, rest, opt, kw, inits, vars, body, alt;
  470. SCM unbound, arity, rib, new_env;
  471. int nreq, nopt, ninits;
  472. req = REF (exp, LAMBDA_CASE, REQ);
  473. rest = scm_not (scm_not (REF (exp, LAMBDA_CASE, REST)));
  474. opt = REF (exp, LAMBDA_CASE, OPT);
  475. kw = REF (exp, LAMBDA_CASE, KW);
  476. inits = REF (exp, LAMBDA_CASE, INITS);
  477. vars = REF (exp, LAMBDA_CASE, GENSYMS);
  478. body = REF (exp, LAMBDA_CASE, BODY);
  479. alt = REF (exp, LAMBDA_CASE, ALTERNATE);
  480. nreq = scm_ilength (req);
  481. nopt = scm_is_pair (opt) ? scm_ilength (opt) : 0;
  482. ninits = scm_ilength (inits);
  483. /* This relies on assignment conversion turning inits into a
  484. sequence of CONST expressions whose values are a unique
  485. "unbound" token. */
  486. unbound = ninits ? REF (CAR (inits), CONST, EXP) : SCM_BOOL_F;
  487. rib = scm_vector (vars);
  488. new_env = push_nested_link (rib, env);
  489. if (scm_is_true (kw))
  490. {
  491. /* (aok? (kw name sym) ...) -> (aok? (kw . index) ...) */
  492. SCM aok = CAR (kw), indices = SCM_EOL;
  493. for (kw = CDR (kw); scm_is_pair (kw); kw = CDR (kw))
  494. {
  495. SCM k;
  496. int idx;
  497. k = CAR (CAR (kw));
  498. idx = lookup_rib (CADDR (CAR (kw)), rib);
  499. indices = scm_acons (k, SCM_I_MAKINUM (idx), indices);
  500. }
  501. kw = scm_cons (aok, scm_reverse_x (indices, SCM_UNDEFINED));
  502. }
  503. if (scm_is_false (alt) && scm_is_false (kw) && scm_is_false (opt))
  504. {
  505. if (scm_is_false (rest))
  506. arity = FIXED_ARITY (nreq);
  507. else
  508. arity = REST_ARITY (nreq, SCM_BOOL_T);
  509. }
  510. else if (scm_is_true (alt))
  511. arity = FULL_ARITY (nreq, rest, nopt, kw, ninits, unbound,
  512. SCM_MEMOIZED_ARGS (memoize (alt, env)));
  513. else
  514. arity = FULL_ARITY (nreq, rest, nopt, kw, ninits, unbound,
  515. SCM_BOOL_F);
  516. return MAKMEMO_LAMBDA (memoize (body, new_env), arity,
  517. SCM_EOL /* meta, filled in later */);
  518. }
  519. case SCM_EXPANDED_LET:
  520. {
  521. SCM vars, exps, body, varsv, inits, new_env;
  522. int i;
  523. vars = REF (exp, LET, GENSYMS);
  524. exps = REF (exp, LET, VALS);
  525. body = REF (exp, LET, BODY);
  526. varsv = scm_vector (vars);
  527. inits = scm_c_make_vector (VECTOR_LENGTH (varsv),
  528. SCM_BOOL_F);
  529. new_env = push_nested_link (varsv, capture_env (env));
  530. for (i = 0; scm_is_pair (exps); exps = CDR (exps), i++)
  531. VECTOR_SET (inits, i, memoize (CAR (exps), env));
  532. return maybe_makmemo_capture_module
  533. (MAKMEMO_LET (inits, memoize (body, new_env)), env);
  534. }
  535. default:
  536. abort ();
  537. }
  538. }
  539. SCM_DEFINE (scm_memoize_expression, "memoize-expression", 1, 0, 0,
  540. (SCM exp),
  541. "Memoize the expression @var{exp}.")
  542. #define FUNC_NAME s_scm_memoize_expression
  543. {
  544. SCM_ASSERT_TYPE (SCM_EXPANDED_P (exp), exp, 1, FUNC_NAME, "expanded");
  545. return memoize (scm_convert_assignment (exp), SCM_BOOL_F);
  546. }
  547. #undef FUNC_NAME
  548. SCM_SYMBOL (sym_placeholder, "_");
  549. static SCM unmemoize (SCM expr);
  550. static SCM
  551. unmemoize_exprs (SCM exprs)
  552. {
  553. SCM ret, tail;
  554. if (scm_is_null (exprs))
  555. return SCM_EOL;
  556. ret = scm_list_1 (unmemoize (CAR (exprs)));
  557. tail = ret;
  558. for (exprs = CDR (exprs); !scm_is_null (exprs); exprs = CDR (exprs))
  559. {
  560. SCM_SETCDR (tail, scm_list_1 (unmemoize (CAR (exprs))));
  561. tail = CDR (tail);
  562. }
  563. return ret;
  564. }
  565. static SCM
  566. unmemoize_bindings (SCM inits)
  567. {
  568. SCM ret = SCM_EOL;
  569. int n = scm_c_vector_length (inits);
  570. while (n--)
  571. ret = scm_cons (unmemoize (scm_c_vector_ref (inits, n)), ret);
  572. return ret;
  573. }
  574. static SCM
  575. unmemoize_lexical (SCM n)
  576. {
  577. char buf[32];
  578. buf[31] = 0;
  579. snprintf (buf, 31, "<%u,%u>", scm_to_uint32 (CAR (n)),
  580. scm_to_uint32 (CDR (n)));
  581. return scm_from_utf8_symbol (buf);
  582. }
  583. static SCM
  584. unmemoize (const SCM expr)
  585. {
  586. SCM args;
  587. args = SCM_MEMOIZED_ARGS (expr);
  588. switch (SCM_MEMOIZED_TAG (expr))
  589. {
  590. case SCM_M_APPLY:
  591. return scm_cons (scm_from_latin1_symbol ("apply"),
  592. unmemoize_exprs (args));
  593. case SCM_M_SEQ:
  594. return scm_list_3 (scm_sym_begin, unmemoize (CAR (args)),
  595. unmemoize (CDR (args)));
  596. case SCM_M_CALL:
  597. return unmemoize_exprs (args);
  598. case SCM_M_CONT:
  599. return scm_list_2 (scm_from_latin1_symbol
  600. ("call-with-current_continuation"),
  601. unmemoize (args));
  602. case SCM_M_CALL_WITH_VALUES:
  603. return scm_list_3 (scm_from_latin1_symbol ("call-with-values"),
  604. unmemoize (CAR (args)), unmemoize (CDR (args)));
  605. case SCM_M_CAPTURE_MODULE:
  606. return scm_list_2 (scm_from_latin1_symbol ("capture-module"),
  607. unmemoize (args));
  608. case SCM_M_IF:
  609. return scm_list_4 (scm_sym_if, unmemoize (scm_car (args)),
  610. unmemoize (scm_cadr (args)), unmemoize (scm_cddr (args)));
  611. case SCM_M_LAMBDA:
  612. {
  613. SCM body = CAR (args), spec = CDDR (args);
  614. if (scm_is_null (CDR (spec)))
  615. return scm_list_3 (scm_sym_lambda,
  616. scm_make_list (CAR (spec), sym_placeholder),
  617. unmemoize (CAR (args)));
  618. else if (scm_is_null (SCM_CDDR (spec)))
  619. {
  620. SCM formals = scm_make_list (CAR (spec), sym_placeholder);
  621. return scm_list_3 (scm_sym_lambda,
  622. scm_is_true (CADR (spec))
  623. ? scm_cons_star (sym_placeholder, formals)
  624. : formals,
  625. unmemoize (CAR (args)));
  626. }
  627. else
  628. {
  629. SCM alt, tail;
  630. alt = CADDDR (CDDDR (spec));
  631. if (scm_is_true (alt))
  632. tail = CDR (unmemoize (alt));
  633. else
  634. tail = SCM_EOL;
  635. return scm_cons
  636. (sym_case_lambda_star,
  637. scm_cons (scm_list_2 (scm_list_5 (CAR (spec),
  638. CADR (spec),
  639. CADDR (spec),
  640. CADDDR (spec),
  641. CADR (CDDDR (spec))),
  642. unmemoize (body)),
  643. tail));
  644. }
  645. }
  646. case SCM_M_CAPTURE_ENV:
  647. return scm_list_3 (scm_from_latin1_symbol ("capture-env"),
  648. CAR (args),
  649. unmemoize (CDR (args)));
  650. case SCM_M_LET:
  651. return scm_list_3 (scm_sym_let,
  652. unmemoize_bindings (CAR (args)),
  653. unmemoize (CDR (args)));
  654. case SCM_M_QUOTE:
  655. return scm_list_2 (scm_sym_quote, args);
  656. case SCM_M_LEXICAL_REF:
  657. return unmemoize_lexical (args);
  658. case SCM_M_LEXICAL_SET:
  659. return scm_list_3 (scm_sym_set_x, unmemoize_lexical (CAR (args)),
  660. unmemoize (CDR (args)));
  661. case SCM_M_BOX_REF:
  662. return scm_list_2 (scm_from_latin1_symbol ("variable-ref"),
  663. unmemoize (args));
  664. case SCM_M_BOX_SET:
  665. return scm_list_3 (scm_from_latin1_symbol ("variable-set!"),
  666. unmemoize (CAR (args)),
  667. unmemoize (CDR (args)));
  668. case SCM_M_RESOLVE:
  669. if (SCM_VARIABLEP (args))
  670. return args;
  671. else if (scm_is_symbol (CDR (args)))
  672. return CDR (args);
  673. else
  674. return scm_list_3
  675. (scm_is_true (CDDDR (args)) ? scm_sym_at : scm_sym_atat,
  676. scm_i_finite_list_copy (CADR (args)),
  677. CADDR (args));
  678. case SCM_M_CALL_WITH_PROMPT:
  679. return scm_list_4 (scm_from_latin1_symbol ("call-with-prompt"),
  680. unmemoize (CAR (args)),
  681. unmemoize (CADR (args)),
  682. unmemoize (CDDR (args)));
  683. default:
  684. abort ();
  685. }
  686. }
  687. SCM_DEFINE (scm_unmemoize_expression, "unmemoize-expression", 1, 0, 0,
  688. (SCM m),
  689. "Unmemoize the memoized expression @var{m}.")
  690. #define FUNC_NAME s_scm_unmemoize_expression
  691. {
  692. return unmemoize (m);
  693. }
  694. #undef FUNC_NAME
  695. SCM_DEFINE (scm_memoized_typecode, "memoized-typecode", 1, 0, 0,
  696. (SCM sym),
  697. "Return the memoized typecode corresponding to the symbol @var{sym}.")
  698. #define FUNC_NAME s_scm_memoized_typecode
  699. {
  700. int i;
  701. SCM_VALIDATE_SYMBOL (1, sym);
  702. for (i = 0; i < sizeof(memoized_tags)/sizeof(const char*); i++)
  703. if (strcmp (scm_i_symbol_chars (sym), memoized_tags[i]) == 0)
  704. return scm_from_int32 (i);
  705. return SCM_BOOL_F;
  706. }
  707. #undef FUNC_NAME
  708. SCM_SYMBOL (scm_unbound_variable_key, "unbound-variable");
  709. static void error_unbound_variable (SCM symbol) SCM_NORETURN;
  710. static void error_unbound_variable (SCM symbol)
  711. {
  712. scm_error (scm_unbound_variable_key, NULL, "Unbound variable: ~S",
  713. scm_list_1 (symbol), SCM_BOOL_F);
  714. }
  715. SCM_DEFINE (scm_sys_resolve_variable, "%resolve-variable", 2, 0, 0,
  716. (SCM loc, SCM mod),
  717. "Look up and return the variable for @var{loc}.")
  718. #define FUNC_NAME s_scm_sys_resolve_variable
  719. {
  720. int mode;
  721. if (scm_is_false (mod))
  722. mod = scm_the_root_module ();
  723. mode = scm_to_int (scm_car (loc));
  724. loc = scm_cdr (loc);
  725. switch (mode)
  726. {
  727. case SCM_EXPANDED_TOPLEVEL_REF:
  728. case SCM_EXPANDED_TOPLEVEL_SET:
  729. {
  730. SCM var = scm_module_variable (mod, loc);
  731. if (scm_is_false (var)
  732. || (mode == SCM_EXPANDED_TOPLEVEL_REF
  733. && scm_is_false (scm_variable_bound_p (var))))
  734. error_unbound_variable (loc);
  735. return var;
  736. }
  737. case SCM_EXPANDED_TOPLEVEL_DEFINE:
  738. {
  739. return scm_module_ensure_local_variable (mod, loc);
  740. }
  741. case SCM_EXPANDED_MODULE_REF:
  742. case SCM_EXPANDED_MODULE_SET:
  743. {
  744. SCM var;
  745. mod = scm_resolve_module (scm_car (loc));
  746. if (scm_is_true (scm_cddr (loc)))
  747. mod = scm_module_public_interface (mod);
  748. var = scm_module_lookup (mod, scm_cadr (loc));
  749. if (mode == SCM_EXPANDED_MODULE_SET
  750. && scm_is_false (scm_variable_bound_p (var)))
  751. error_unbound_variable (scm_cadr (loc));
  752. return var;
  753. }
  754. default:
  755. scm_wrong_type_arg (FUNC_NAME, 1, loc);
  756. return SCM_BOOL_F;
  757. }
  758. }
  759. #undef FUNC_NAME
  760. void
  761. scm_init_memoize ()
  762. {
  763. #include "memoize.x"
  764. wind = scm_c_make_gsubr ("wind", 2, 0, 0, do_wind);
  765. unwind = scm_c_make_gsubr ("unwind", 0, 0, 0, do_unwind);
  766. push_fluid = scm_c_make_gsubr ("push-fluid", 2, 0, 0, do_push_fluid);
  767. pop_fluid = scm_c_make_gsubr ("pop-fluid", 0, 0, 0, do_pop_fluid);
  768. push_dynamic_state = scm_c_make_gsubr ("push-dynamic_state", 1, 0, 0,
  769. do_push_dynamic_state);
  770. pop_dynamic_state = scm_c_make_gsubr ("pop-dynamic_state", 0, 0, 0,
  771. do_pop_dynamic_state);
  772. list_of_guile = scm_list_1 (scm_from_latin1_symbol ("guile"));
  773. }