Commit | Line | Data |
---|---|---|
b550d8b7 JM |
1 | function step9_try(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 = is_pair(ast) | |
47699629 | 10 | ret = type_utils.sequential_Q(ast) && length(ast) > 0; |
b550d8b7 JM |
11 | end |
12 | ||
13 | function ret = quasiquote(ast) | |
14 | if ~is_pair(ast) | |
6a572dff JM |
15 | ret = types.List(types.Symbol('quote'), ast); |
16 | elseif isa(ast.get(1),'types.Symbol') && ... | |
17 | strcmp(ast.get(1).name, 'unquote') | |
18 | ret = ast.get(2); | |
19 | elseif is_pair(ast.get(1)) && ... | |
20 | isa(ast.get(1).get(1),'types.Symbol') && ... | |
21 | strcmp(ast.get(1).get(1).name, 'splice-unquote') | |
22 | ret = types.List(types.Symbol('concat'), ... | |
23 | ast.get(1).get(2), ... | |
24 | quasiquote(ast.slice(2))); | |
b550d8b7 | 25 | else |
6a572dff JM |
26 | ret = types.List(types.Symbol('cons'), ... |
27 | quasiquote(ast.get(1)), ... | |
28 | quasiquote(ast.slice(2))); | |
b550d8b7 JM |
29 | end |
30 | end | |
31 | ||
32 | function ret = is_macro_call(ast, env) | |
47699629 | 33 | if type_utils.list_Q(ast) && isa(ast.get(1), 'types.Symbol') && ... |
6a572dff JM |
34 | ~islogical(env.find(ast.get(1))) |
35 | f = env.get(ast.get(1)); | |
7f567f36 | 36 | ret = isa(f,'types.Function') && f.is_macro; |
b550d8b7 JM |
37 | else |
38 | ret = false; | |
39 | end | |
40 | end | |
41 | ||
42 | function ret = macroexpand(ast, env) | |
43 | while is_macro_call(ast, env) | |
6a572dff JM |
44 | mac = env.get(ast.get(1)); |
45 | args = ast.slice(2); | |
46 | ast = mac.fn(args.data{:}); | |
b550d8b7 JM |
47 | end |
48 | ret = ast; | |
49 | end | |
50 | ||
51 | function ret = eval_ast(ast, env) | |
52 | switch class(ast) | |
53 | case 'types.Symbol' | |
54 | ret = env.get(ast); | |
6a572dff JM |
55 | case 'types.List' |
56 | ret = types.List(); | |
b550d8b7 | 57 | for i=1:length(ast) |
6a572dff JM |
58 | ret.append(EVAL(ast.get(i), env)); |
59 | end | |
60 | case 'types.Vector' | |
61 | ret = types.Vector(); | |
62 | for i=1:length(ast) | |
63 | ret.append(EVAL(ast.get(i), env)); | |
64 | end | |
65 | case 'types.HashMap' | |
66 | ret = types.HashMap(); | |
67 | ks = ast.keys(); | |
68 | for i=1:length(ks) | |
69 | k = ks{i}; | |
70 | ret.set(EVAL(k, env), EVAL(ast.get(k), env)); | |
b550d8b7 JM |
71 | end |
72 | otherwise | |
73 | ret = ast; | |
74 | end | |
75 | end | |
76 | ||
77 | function ret = EVAL(ast, env) | |
78 | while true | |
0b234e13 | 79 | %fprintf('EVAL: %s\n', printer.pr_str(ast, true)); |
47699629 | 80 | if ~type_utils.list_Q(ast) |
b550d8b7 JM |
81 | ret = eval_ast(ast, env); |
82 | return; | |
83 | end | |
84 | ||
85 | % apply | |
9870acb6 JM |
86 | if length(ast) == 0 |
87 | ret = ast; | |
88 | return; | |
89 | end | |
b550d8b7 | 90 | ast = macroexpand(ast, env); |
47699629 | 91 | if ~type_utils.list_Q(ast) |
d5b81cc0 | 92 | ret = eval_ast(ast, env); |
b550d8b7 JM |
93 | return; |
94 | end | |
95 | ||
6a572dff JM |
96 | if isa(ast.get(1),'types.Symbol') |
97 | a1sym = ast.get(1).name; | |
b550d8b7 JM |
98 | else |
99 | a1sym = '_@$fn$@_'; | |
100 | end | |
101 | switch (a1sym) | |
102 | case 'def!' | |
6a572dff | 103 | ret = env.set(ast.get(2), EVAL(ast.get(3), env)); |
b550d8b7 JM |
104 | return; |
105 | case 'let*' | |
47699629 | 106 | let_env = Env({env}); |
6a572dff JM |
107 | for i=1:2:length(ast.get(2)) |
108 | let_env.set(ast.get(2).get(i), EVAL(ast.get(2).get(i+1), let_env)); | |
b550d8b7 JM |
109 | end |
110 | env = let_env; | |
6a572dff | 111 | ast = ast.get(3); % TCO |
b550d8b7 | 112 | case 'quote' |
6a572dff | 113 | ret = ast.get(2); |
b550d8b7 JM |
114 | return; |
115 | case 'quasiquote' | |
6a572dff | 116 | ast = quasiquote(ast.get(2)); % TCO |
b550d8b7 | 117 | case 'defmacro!' |
6a572dff | 118 | ret = env.set(ast.get(2), EVAL(ast.get(3), env)); |
b550d8b7 JM |
119 | ret.is_macro = true; |
120 | return; | |
121 | case 'macroexpand' | |
6a572dff | 122 | ret = macroexpand(ast.get(2), env); |
b550d8b7 JM |
123 | return; |
124 | case 'try*' | |
125 | try | |
6a572dff | 126 | ret = EVAL(ast.get(2), env); |
b550d8b7 JM |
127 | return; |
128 | catch e | |
6a572dff | 129 | if length(ast) > 2 && strcmp(ast.get(3).get(1).name, 'catch*') |
47699629 JM |
130 | if strcmp(e.identifier, 'MalException:object') |
131 | if exist('OCTAVE_VERSION', 'builtin') ~= 0 | |
132 | global error_object; | |
133 | exc = error_object; | |
134 | else | |
135 | exc = e.obj; | |
136 | end | |
b550d8b7 JM |
137 | else |
138 | exc = e.message; | |
139 | end | |
47699629 | 140 | catch_env = Env({env}, types.List(ast.get(3).get(2)), ... |
6a572dff JM |
141 | types.List(exc)); |
142 | ret = EVAL(ast.get(3).get(3), catch_env); | |
b550d8b7 JM |
143 | return; |
144 | else | |
dd7a4f55 | 145 | rethrow(e); |
b550d8b7 JM |
146 | end |
147 | end | |
148 | case 'do' | |
6a572dff JM |
149 | el = eval_ast(ast.slice(2,length(ast)-1), env); |
150 | ast = ast.get(length(ast)); % TCO | |
b550d8b7 | 151 | case 'if' |
6a572dff | 152 | cond = EVAL(ast.get(2), env); |
b550d8b7 JM |
153 | if strcmp(class(cond), 'types.Nil') || ... |
154 | (islogical(cond) && cond == false) | |
155 | if length(ast) > 3 | |
6a572dff | 156 | ast = ast.get(4); % TCO |
b550d8b7 | 157 | else |
47699629 | 158 | ret = type_utils.nil; |
b550d8b7 JM |
159 | return; |
160 | end | |
161 | else | |
6a572dff | 162 | ast = ast.get(3); % TCO |
b550d8b7 JM |
163 | end |
164 | case 'fn*' | |
47699629 | 165 | fn = @(varargin) EVAL(ast.get(3), Env({env}, ast.get(2), ... |
6a572dff JM |
166 | types.List(varargin{:}))); |
167 | ret = types.Function(fn, ast.get(3), env, ast.get(2)); | |
b550d8b7 JM |
168 | return; |
169 | otherwise | |
170 | el = eval_ast(ast, env); | |
6a572dff JM |
171 | f = el.get(1); |
172 | args = el.slice(2); | |
b550d8b7 | 173 | if isa(f, 'types.Function') |
47699629 | 174 | env = Env({f.env}, f.params, args); |
b550d8b7 JM |
175 | ast = f.ast; % TCO |
176 | else | |
6a572dff | 177 | ret = f(args.data{:}); |
b550d8b7 JM |
178 | return |
179 | end | |
180 | end | |
181 | end | |
182 | end | |
183 | ||
184 | ||
185 | function ret = PRINT(ast) | |
186 | ret = printer.pr_str(ast, true); | |
187 | end | |
188 | ||
189 | % REPL | |
190 | function ret = rep(str, env) | |
191 | ret = PRINT(EVAL(READ(str), env)); | |
192 | end | |
193 | ||
194 | function main(args) | |
47699629 | 195 | repl_env = Env(); |
b550d8b7 JM |
196 | |
197 | % core.m: defined using matlab | |
198 | ns = core.ns(); ks = ns.keys(); | |
199 | for i=1:length(ks) | |
200 | k = ks{i}; | |
201 | repl_env.set(types.Symbol(k), ns(k)); | |
202 | end | |
203 | repl_env.set(types.Symbol('eval'), @(a) EVAL(a, repl_env)); | |
6a572dff JM |
204 | rest_args = args(2:end); |
205 | repl_env.set(types.Symbol('*ARGV*'), types.List(rest_args{:})); | |
b550d8b7 JM |
206 | |
207 | % core.mal: defined using the langauge itself | |
208 | rep('(def! not (fn* (a) (if a false true)))', repl_env); | |
209 | rep('(def! load-file (fn* (f) (eval (read-string (str "(do " (slurp f) ")")))))"', repl_env); | |
210 | rep('(defmacro! cond (fn* (& xs) (if (> (count xs) 0) (list ''if (first xs) (if (> (count xs) 1) (nth xs 1) (throw "odd number of forms to cond")) (cons ''cond (rest (rest xs)))))))', repl_env); | |
211 | rep('(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))', repl_env); | |
212 | ||
213 | if ~isempty(args) | |
0b234e13 | 214 | rep(sprintf('(load-file "%s")', args{1}), repl_env); |
b550d8b7 JM |
215 | quit; |
216 | end | |
217 | ||
218 | %cleanObj = onCleanup(@() disp('*** here1 ***')); | |
219 | while (true) | |
47699629 JM |
220 | try |
221 | line = input('user> ', 's'); | |
222 | catch err | |
223 | return | |
224 | end | |
b550d8b7 JM |
225 | if strcmp(strtrim(line),''), continue, end |
226 | try | |
227 | fprintf('%s\n', rep(line, repl_env)); | |
228 | catch err | |
47699629 JM |
229 | if strcmp('MalException:object', err.identifier) |
230 | if exist('OCTAVE_VERSION', 'builtin') ~= 0 | |
231 | global error_object; | |
232 | fprintf('Error: %s\n', printer.pr_str(error_object, true)); | |
233 | else | |
234 | fprintf('Error: %s\n', printer.pr_str(err.obj, true)); | |
235 | end | |
b550d8b7 JM |
236 | else |
237 | fprintf('Error: %s\n', err.message); | |
238 | end | |
47699629 | 239 | type_utils.print_stack(err); |
b550d8b7 JM |
240 | end |
241 | end | |
242 | end |