Commit | Line | Data |
---|---|---|
23c5aa44 JM |
1 | function step5_tco(varargin), main(varargin), end |
2 | ||
3 | % read | |
4 | function ret = READ(str) | |
5 | ret = reader.read_str(str); | |
6 | end | |
7 | ||
8 | % eval | |
9 | function ret = eval_ast(ast, env) | |
10 | switch class(ast) | |
11 | case 'types.Symbol' | |
12 | ret = env.get(ast); | |
6a572dff JM |
13 | case 'types.List' |
14 | ret = types.List(); | |
23c5aa44 | 15 | for i=1:length(ast) |
6a572dff JM |
16 | ret.append(EVAL(ast.get(i), env)); |
17 | end | |
18 | case 'types.Vector' | |
19 | ret = types.Vector(); | |
20 | for i=1:length(ast) | |
21 | ret.append(EVAL(ast.get(i), env)); | |
22 | end | |
23 | case 'types.HashMap' | |
24 | ret = types.HashMap(); | |
25 | ks = ast.keys(); | |
26 | for i=1:length(ks) | |
27 | k = ks{i}; | |
28 | ret.set(EVAL(k, env), EVAL(ast.get(k), env)); | |
23c5aa44 JM |
29 | end |
30 | otherwise | |
31 | ret = ast; | |
32 | end | |
33 | end | |
34 | ||
35 | function ret = EVAL(ast, env) | |
36 | while true | |
0b234e13 | 37 | %fprintf('EVAL: %s\n', printer.pr_str(ast, true)); |
47699629 | 38 | if ~type_utils.list_Q(ast) |
23c5aa44 JM |
39 | ret = eval_ast(ast, env); |
40 | return; | |
41 | end | |
42 | ||
43 | % apply | |
9870acb6 JM |
44 | if length(ast) == 0 |
45 | ret = ast; | |
46 | return; | |
47 | end | |
6a572dff JM |
48 | if isa(ast.get(1),'types.Symbol') |
49 | a1sym = ast.get(1).name; | |
23c5aa44 JM |
50 | else |
51 | a1sym = '_@$fn$@_'; | |
52 | end | |
53 | switch (a1sym) | |
54 | case 'def!' | |
6a572dff | 55 | ret = env.set(ast.get(2), EVAL(ast.get(3), env)); |
23c5aa44 JM |
56 | return; |
57 | case 'let*' | |
47699629 | 58 | let_env = Env({env}); |
6a572dff JM |
59 | for i=1:2:length(ast.get(2)) |
60 | let_env.set(ast.get(2).get(i), EVAL(ast.get(2).get(i+1), let_env)); | |
23c5aa44 JM |
61 | end |
62 | env = let_env; | |
6a572dff | 63 | ast = ast.get(3); % TCO |
23c5aa44 | 64 | case 'do' |
6a572dff JM |
65 | el = eval_ast(ast.slice(2,length(ast)-1), env); |
66 | ast = ast.get(length(ast)); % TCO | |
23c5aa44 | 67 | case 'if' |
6a572dff | 68 | cond = EVAL(ast.get(2), env); |
23c5aa44 JM |
69 | if strcmp(class(cond), 'types.Nil') || ... |
70 | (islogical(cond) && cond == false) | |
71 | if length(ast) > 3 | |
6a572dff | 72 | ast = ast.get(4); % TCO |
23c5aa44 | 73 | else |
9d54117c | 74 | ret = type_utils.nil; |
23c5aa44 JM |
75 | return; |
76 | end | |
77 | else | |
6a572dff | 78 | ast = ast.get(3); % TCO |
23c5aa44 JM |
79 | end |
80 | case 'fn*' | |
9d54117c | 81 | fn = @(varargin) EVAL(ast.get(3), Env({env}, ast.get(2), ... |
6a572dff JM |
82 | types.List(varargin{:}))); |
83 | ret = types.Function(fn, ast.get(3), env, ast.get(2)); | |
23c5aa44 JM |
84 | return; |
85 | otherwise | |
86 | el = eval_ast(ast, env); | |
6a572dff JM |
87 | f = el.get(1); |
88 | args = el.slice(2); | |
b550d8b7 | 89 | if isa(f, 'types.Function') |
47699629 | 90 | env = Env({f.env}, f.params, args); |
23c5aa44 JM |
91 | ast = f.ast; % TCO |
92 | else | |
6a572dff | 93 | ret = f(args.data{:}); |
23c5aa44 JM |
94 | return |
95 | end | |
96 | end | |
97 | end | |
98 | end | |
99 | ||
100 | ||
101 | function ret = PRINT(ast) | |
102 | ret = printer.pr_str(ast, true); | |
103 | end | |
104 | ||
105 | % REPL | |
106 | function ret = rep(str, env) | |
107 | ret = PRINT(EVAL(READ(str), env)); | |
108 | end | |
109 | ||
110 | function main(args) | |
47699629 | 111 | repl_env = Env(); |
23c5aa44 JM |
112 | |
113 | % core.m: defined using matlab | |
114 | ns = core.ns(); ks = ns.keys(); | |
115 | for i=1:length(ks) | |
116 | k = ks{i}; | |
117 | repl_env.set(types.Symbol(k), ns(k)); | |
118 | end | |
119 | ||
120 | % core.mal: defined using the langauge itself | |
121 | rep('(def! not (fn* (a) (if a false true)))', repl_env); | |
122 | ||
123 | %cleanObj = onCleanup(@() disp('*** here1 ***')); | |
124 | while (true) | |
47699629 JM |
125 | try |
126 | line = input('user> ', 's'); | |
127 | catch err | |
128 | return | |
129 | end | |
23c5aa44 JM |
130 | if strcmp(strtrim(line),''), continue, end |
131 | try | |
132 | fprintf('%s\n', rep(line, repl_env)); | |
133 | catch err | |
134 | fprintf('Error: %s\n', err.message); | |
47699629 | 135 | type_utils.print_stack(err); |
23c5aa44 JM |
136 | end |
137 | end | |
138 | end |