Commit | Line | Data |
---|---|---|
c3023f26 JM |
1 | function step7_quote(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) | |
6a572dff | 10 | ret = types.sequential_Q(ast) && length(ast) > 0; |
c3023f26 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))); | |
c3023f26 | 25 | else |
6a572dff JM |
26 | ret = types.List(types.Symbol('cons'), ... |
27 | quasiquote(ast.get(1)), ... | |
28 | quasiquote(ast.slice(2))); | |
c3023f26 JM |
29 | end |
30 | end | |
31 | ||
32 | function ret = eval_ast(ast, env) | |
33 | switch class(ast) | |
34 | case 'types.Symbol' | |
35 | ret = env.get(ast); | |
6a572dff JM |
36 | case 'types.List' |
37 | ret = types.List(); | |
c3023f26 | 38 | for i=1:length(ast) |
6a572dff JM |
39 | ret.append(EVAL(ast.get(i), env)); |
40 | end | |
41 | case 'types.Vector' | |
42 | ret = types.Vector(); | |
43 | for i=1:length(ast) | |
44 | ret.append(EVAL(ast.get(i), env)); | |
45 | end | |
46 | case 'types.HashMap' | |
47 | ret = types.HashMap(); | |
48 | ks = ast.keys(); | |
49 | for i=1:length(ks) | |
50 | k = ks{i}; | |
51 | ret.set(EVAL(k, env), EVAL(ast.get(k), env)); | |
c3023f26 JM |
52 | end |
53 | otherwise | |
54 | ret = ast; | |
55 | end | |
56 | end | |
57 | ||
58 | function ret = EVAL(ast, env) | |
59 | while true | |
0b234e13 | 60 | %fprintf('EVAL: %s\n', printer.pr_str(ast, true)); |
6a572dff | 61 | if ~types.list_Q(ast) |
c3023f26 JM |
62 | ret = eval_ast(ast, env); |
63 | return; | |
64 | end | |
65 | ||
66 | % apply | |
6a572dff JM |
67 | if isa(ast.get(1),'types.Symbol') |
68 | a1sym = ast.get(1).name; | |
c3023f26 JM |
69 | else |
70 | a1sym = '_@$fn$@_'; | |
71 | end | |
72 | switch (a1sym) | |
73 | case 'def!' | |
6a572dff | 74 | ret = env.set(ast.get(2), EVAL(ast.get(3), env)); |
c3023f26 JM |
75 | return; |
76 | case 'let*' | |
77 | let_env = Env(env); | |
6a572dff JM |
78 | for i=1:2:length(ast.get(2)) |
79 | let_env.set(ast.get(2).get(i), EVAL(ast.get(2).get(i+1), let_env)); | |
c3023f26 JM |
80 | end |
81 | env = let_env; | |
6a572dff | 82 | ast = ast.get(3); % TCO |
c3023f26 | 83 | case 'quote' |
6a572dff | 84 | ret = ast.get(2); |
c3023f26 JM |
85 | return; |
86 | case 'quasiquote' | |
6a572dff | 87 | ast = quasiquote(ast.get(2)); % TCO |
c3023f26 | 88 | case 'do' |
6a572dff JM |
89 | el = eval_ast(ast.slice(2,length(ast)-1), env); |
90 | ast = ast.get(length(ast)); % TCO | |
c3023f26 | 91 | case 'if' |
6a572dff | 92 | cond = EVAL(ast.get(2), env); |
c3023f26 JM |
93 | if strcmp(class(cond), 'types.Nil') || ... |
94 | (islogical(cond) && cond == false) | |
95 | if length(ast) > 3 | |
6a572dff | 96 | ast = ast.get(4); % TCO |
c3023f26 JM |
97 | else |
98 | ret = types.nil; | |
99 | return; | |
100 | end | |
101 | else | |
6a572dff | 102 | ast = ast.get(3); % TCO |
c3023f26 JM |
103 | end |
104 | case 'fn*' | |
6a572dff JM |
105 | fn = @(varargin) EVAL(ast.get(3), Env(env, ast.get(2), ... |
106 | types.List(varargin{:}))); | |
107 | ret = types.Function(fn, ast.get(3), env, ast.get(2)); | |
c3023f26 JM |
108 | return; |
109 | otherwise | |
110 | el = eval_ast(ast, env); | |
6a572dff JM |
111 | f = el.get(1); |
112 | args = el.slice(2); | |
b550d8b7 | 113 | if isa(f, 'types.Function') |
c3023f26 JM |
114 | env = Env(f.env, f.params, args); |
115 | ast = f.ast; % TCO | |
116 | else | |
6a572dff | 117 | ret = f(args.data{:}); |
c3023f26 JM |
118 | return |
119 | end | |
120 | end | |
121 | end | |
122 | end | |
123 | ||
124 | ||
125 | function ret = PRINT(ast) | |
126 | ret = printer.pr_str(ast, true); | |
127 | end | |
128 | ||
129 | % REPL | |
130 | function ret = rep(str, env) | |
131 | ret = PRINT(EVAL(READ(str), env)); | |
132 | end | |
133 | ||
134 | function main(args) | |
135 | repl_env = Env(false); | |
136 | ||
137 | % core.m: defined using matlab | |
138 | ns = core.ns(); ks = ns.keys(); | |
139 | for i=1:length(ks) | |
140 | k = ks{i}; | |
141 | repl_env.set(types.Symbol(k), ns(k)); | |
142 | end | |
143 | repl_env.set(types.Symbol('eval'), @(a) EVAL(a, repl_env)); | |
6a572dff JM |
144 | rest_args = args(2:end); |
145 | repl_env.set(types.Symbol('*ARGV*'), types.List(rest_args{:})); | |
c3023f26 JM |
146 | |
147 | % core.mal: defined using the langauge itself | |
148 | rep('(def! not (fn* (a) (if a false true)))', repl_env); | |
149 | rep('(def! load-file (fn* (f) (eval (read-string (str "(do " (slurp f) ")")))))"', repl_env); | |
150 | ||
151 | if ~isempty(args) | |
0b234e13 | 152 | rep(sprintf('(load-file "%s")', args{1}), repl_env); |
c3023f26 JM |
153 | quit; |
154 | end | |
155 | ||
156 | %cleanObj = onCleanup(@() disp('*** here1 ***')); | |
157 | while (true) | |
158 | line = input('user> ', 's'); | |
159 | if strcmp(strtrim(line),''), continue, end | |
160 | try | |
161 | fprintf('%s\n', rep(line, repl_env)); | |
162 | catch err | |
163 | fprintf('Error: %s\n', err.message); | |
164 | fprintf('%s\n', getReport(err, 'extended')); | |
165 | end | |
166 | end | |
167 | end |