Basic: enable Travis testing.
[jackhill/mal.git] / basic / stepA_mal.in.bas
CommitLineData
30a3d828
JM
1REM POKE 1, PEEK(1)AND248: REM enable all ROM areas as RAM
2REM POKE 55,0: POKE 56,192: CLR: REM move BASIC end from $A000 to $C000
3GOTO MAIN
4
5REM $INCLUDE: 'readline.in.bas'
6REM $INCLUDE: 'types.in.bas'
7REM $INCLUDE: 'reader.in.bas'
8REM $INCLUDE: 'printer.in.bas'
9REM $INCLUDE: 'env.in.bas'
10REM $INCLUDE: 'core.in.bas'
11
12REM $INCLUDE: 'debug.in.bas'
13
cc9dbd92 14REM READ(A$) -> R
30a3d828
JM
15MAL_READ:
16 GOSUB READ_STR
17 RETURN
18
cc9dbd92 19REM QUASIQUOTE(A) -> R
af621e3a 20SUB QUASIQUOTE
01e8850d
JM
21 REM pair?
22 IF (Z%(A,0)AND31)<6 OR (Z%(A,0)AND31)>7 THEN GOTO QQ_QUOTE
23 IF (Z%(A,1)=0) THEN GOTO QQ_QUOTE
24 GOTO QQ_UNQUOTE
25
26 QQ_QUOTE:
30a3d828 27 REM ['quote, ast]
cc9dbd92 28 AS$="quote":T=5:GOSUB STRING
bbab5c5d
JM
29 B2=R:B1=A:GOSUB LIST2
30 AY=B2:GOSUB RELEASE
30a3d828 31
af621e3a 32 GOTO QQ_DONE
30a3d828
JM
33
34 QQ_UNQUOTE:
cc9dbd92 35 R=A+1:GOSUB DEREF_R
bbab5c5d 36 IF (Z%(R,0)AND31)<>5 THEN GOTO QQ_SPLICE_UNQUOTE
cc9dbd92 37 IF S$(Z%(R,1))<>"unquote" THEN GOTO QQ_SPLICE_UNQUOTE
30a3d828 38 REM [ast[1]]
cc9dbd92 39 R=Z%(A,1)+1:GOSUB DEREF_R
bbab5c5d 40 Z%(R,0)=Z%(R,0)+32
30a3d828 41
af621e3a 42 GOTO QQ_DONE
30a3d828
JM
43
44 QQ_SPLICE_UNQUOTE:
cc9dbd92 45 REM push A on the stack
bbab5c5d 46 X=X+1:X%(X)=A
30a3d828 47 REM rest of cases call quasiquote on ast[1..]
af621e3a
JM
48 A=Z%(A,1):CALL QUASIQUOTE
49 T6=R
cc9dbd92 50 REM pop A off the stack
bbab5c5d 51 A=X%(X):X=X-1
cc9dbd92
JM
52
53 REM set A to ast[0] for last two cases
54 A=A+1:GOSUB DEREF_A
55
01e8850d
JM
56 REM pair?
57 IF (Z%(A,0)AND31)<6 OR (Z%(A,0)AND31)>7 THEN GOTO QQ_DEFAULT
58 IF (Z%(A,1)=0) THEN GOTO QQ_DEFAULT
59
cc9dbd92 60 B=A+1:GOSUB DEREF_B
bbab5c5d 61 IF (Z%(B,0)AND31)<>5 THEN GOTO QQ_DEFAULT
cc9dbd92 62 IF S$(Z%(B,1))<>"splice-unquote" THEN QQ_DEFAULT
30a3d828
JM
63 REM ['concat, ast[0][1], quasiquote(ast[1..])]
64
bbab5c5d
JM
65 B=Z%(A,1)+1:GOSUB DEREF_B:B2=B
66 AS$="concat":T=5:GOSUB STRING:B3=R
67 B1=T6:GOSUB LIST3
30a3d828 68 REM release inner quasiquoted since outer list takes ownership
bbab5c5d
JM
69 AY=B1:GOSUB RELEASE
70 AY=B3:GOSUB RELEASE
af621e3a 71 GOTO QQ_DONE
30a3d828
JM
72
73 QQ_DEFAULT:
74 REM ['cons, quasiquote(ast[0]), quasiquote(ast[1..])]
75
cc9dbd92 76 REM push T6 on the stack
bbab5c5d 77 X=X+1:X%(X)=T6
cc9dbd92 78 REM A set above to ast[0]
af621e3a
JM
79 CALL QUASIQUOTE
80 B2=R
cc9dbd92 81 REM pop T6 off the stack
bbab5c5d 82 T6=X%(X):X=X-1
30a3d828 83
bbab5c5d
JM
84 AS$="cons":T=5:GOSUB STRING:B3=R
85 B1=T6:GOSUB LIST3
30a3d828 86 REM release inner quasiquoted since outer list takes ownership
bbab5c5d
JM
87 AY=B1:GOSUB RELEASE
88 AY=B2:GOSUB RELEASE
89 AY=B3:GOSUB RELEASE
af621e3a
JM
90 QQ_DONE:
91END SUB
30a3d828 92
cc9dbd92 93REM MACROEXPAND(A, E) -> A:
af621e3a 94SUB MACROEXPAND
cc9dbd92 95 REM push original A
bbab5c5d 96 X=X+1:X%(X)=A
30a3d828
JM
97
98 MACROEXPAND_LOOP:
99 REM list?
bbab5c5d 100 IF (Z%(A,0)AND31)<>6 THEN GOTO MACROEXPAND_DONE
30a3d828 101 REM non-empty?
cc9dbd92
JM
102 IF Z%(A,1)=0 THEN GOTO MACROEXPAND_DONE
103 B=A+1:GOSUB DEREF_B
30a3d828 104 REM symbol? in first position
bbab5c5d 105 IF (Z%(B,0)AND31)<>5 THEN GOTO MACROEXPAND_DONE
30a3d828 106 REM defined in environment?
af621e3a 107 K=B:CALL ENV_FIND
cc9dbd92
JM
108 IF R=-1 THEN GOTO MACROEXPAND_DONE
109 B=T4:GOSUB DEREF_B
30a3d828 110 REM macro?
bbab5c5d 111 IF (Z%(B,0)AND31)<>11 THEN GOTO MACROEXPAND_DONE
30a3d828 112
af621e3a 113 F=B:AR=Z%(A,1):CALL APPLY
cc9dbd92 114 A=R
30a3d828 115
bbab5c5d 116 AY=X%(X)
cc9dbd92 117 REM if previous A was not the first A into macroexpand (i.e. an
30a3d828 118 REM intermediate form) then free it
bbab5c5d 119 IF A<>AY THEN Y=Y+1:Y%(Y,0)=A:Y%(Y,1)=LV
30a3d828 120
cc9dbd92 121 IF ER<>-2 THEN GOTO MACROEXPAND_DONE
30a3d828
JM
122 GOTO MACROEXPAND_LOOP
123
124 MACROEXPAND_DONE:
cc9dbd92 125 X=X-1: REM pop original A
af621e3a 126END SUB
30a3d828 127
cc9dbd92 128REM EVAL_AST(A, E) -> R
af621e3a 129SUB EVAL_AST
cc9dbd92 130 REM push A and E on the stack
bbab5c5d 131 X=X+2:X%(X-1)=E:X%(X)=A
30a3d828 132
cc9dbd92 133 IF ER<>-2 THEN GOTO EVAL_AST_RETURN
30a3d828
JM
134
135 GOSUB DEREF_A
136
bbab5c5d 137 T=Z%(A,0)AND31
cc9dbd92
JM
138 IF T=5 THEN GOTO EVAL_AST_SYMBOL
139 IF T>=6 AND T<=8 THEN GOTO EVAL_AST_SEQ
30a3d828
JM
140
141 REM scalar: deref to actual value and inc ref cnt
cc9dbd92 142 R=A:GOSUB DEREF_R
bbab5c5d 143 Z%(R,0)=Z%(R,0)+32
30a3d828
JM
144 GOTO EVAL_AST_RETURN
145
146 EVAL_AST_SYMBOL:
af621e3a
JM
147 K=A:GOTO ENV_GET
148 ENV_GET_RETURN:
30a3d828
JM
149 GOTO EVAL_AST_RETURN
150
151 EVAL_AST_SEQ:
a742287e
JM
152 REM allocate the first entry (T already set above)
153 L=0:N=0:GOSUB ALLOC
30a3d828
JM
154
155 REM make space on the stack
cc9dbd92 156 X=X+4
30a3d828 157 REM push type of sequence
bbab5c5d 158 X%(X-3)=T
30a3d828 159 REM push sequence index
bbab5c5d 160 X%(X-2)=-1
30a3d828 161 REM push future return value (new sequence)
bbab5c5d 162 X%(X-1)=R
30a3d828 163 REM push previous new sequence entry
bbab5c5d 164 X%(X)=R
30a3d828
JM
165
166 EVAL_AST_SEQ_LOOP:
30a3d828 167 REM update index
bbab5c5d 168 X%(X-2)=X%(X-2)+1
30a3d828
JM
169
170 REM check if we are done evaluating the source sequence
cc9dbd92 171 IF Z%(A,1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
30a3d828 172
01e8850d
JM
173 REM if we are returning to DO, then skip last element
174 IF X%(X-6)=2 AND Z%(Z%(A,1),1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
175
30a3d828 176 REM if hashmap, skip eval of even entries (keys)
af621e3a 177 IF (X%(X-3)=8) AND ((X%(X-2)AND1)=0) THEN GOTO EVAL_AST_DO_REF
30a3d828
JM
178 GOTO EVAL_AST_DO_EVAL
179
180 EVAL_AST_DO_REF:
cc9dbd92 181 R=A+1:GOSUB DEREF_R: REM deref to target of referred entry
bbab5c5d 182 Z%(R,0)=Z%(R,0)+32: REM inc ref cnt of referred value
30a3d828
JM
183 GOTO EVAL_AST_ADD_VALUE
184
185 EVAL_AST_DO_EVAL:
186 REM call EVAL for each entry
af621e3a 187 A=A+1:CALL EVAL
cc9dbd92 188 A=A-1
30a3d828
JM
189 GOSUB DEREF_R: REM deref to target of evaluated entry
190
191 EVAL_AST_ADD_VALUE:
192
193 REM update previous value pointer to evaluated entry
bbab5c5d 194 Z%(X%(X)+1,1)=R
30a3d828 195
cc9dbd92 196 IF ER<>-2 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
30a3d828
JM
197
198 REM allocate the next entry
a742287e 199 REM same new sequence entry type
bbab5c5d 200 T=X%(X-3):L=0:N=0:GOSUB ALLOC
30a3d828
JM
201
202 REM update previous sequence entry value to point to new entry
bbab5c5d 203 Z%(X%(X),1)=R
30a3d828 204 REM update previous ptr to current entry
bbab5c5d 205 X%(X)=R
30a3d828
JM
206
207 REM process the next sequence entry from source list
cc9dbd92 208 A=Z%(A,1)
30a3d828
JM
209
210 GOTO EVAL_AST_SEQ_LOOP
211 EVAL_AST_SEQ_LOOP_DONE:
212 REM if no error, get return value (new seq)
bbab5c5d 213 IF ER=-2 THEN R=X%(X-1)
30a3d828 214 REM otherwise, free the return value and return nil
bbab5c5d 215 IF ER<>-2 THEN R=0:AY=X%(X-1):GOSUB RELEASE
30a3d828
JM
216
217 REM pop previous, return, index and type
cc9dbd92 218 X=X-4
30a3d828
JM
219 GOTO EVAL_AST_RETURN
220
221 EVAL_AST_RETURN:
cc9dbd92 222 REM pop A and E off the stack
bbab5c5d 223 E=X%(X-1):A=X%(X):X=X-2
af621e3a 224END SUB
30a3d828 225
af621e3a
JM
226REM EVAL(A, E) -> R
227SUB EVAL
cc9dbd92 228 LV=LV+1: REM track basic return stack level
30a3d828 229
cc9dbd92 230 REM push A and E on the stack
bbab5c5d 231 X=X+2:X%(X-1)=E:X%(X)=A
30a3d828 232
01e8850d
JM
233 REM PRINT "EVAL A:"+STR$(A)+",X:"+STR$(X)+",LV:"+STR$(LV)+",FRE:"+STR$(FRE(0))
234
30a3d828
JM
235 EVAL_TCO_RECUR:
236
c7330b3d
JM
237 IF ER<>-2 THEN GOTO EVAL_RETURN
238
cc9dbd92
JM
239 REM AZ=A:PR=1:GOSUB PR_STR
240 REM PRINT "EVAL: "+R$+" [A:"+STR$(A)+", LV:"+STR$(LV)+"]"
30a3d828
JM
241
242 GOSUB DEREF_A
243
244 GOSUB LIST_Q
cc9dbd92 245 IF R THEN GOTO APPLY_LIST
30a3d828
JM
246 EVAL_NOT_LIST:
247 REM ELSE
af621e3a 248 CALL EVAL_AST
30a3d828
JM
249 GOTO EVAL_RETURN
250
251 APPLY_LIST:
af621e3a 252 CALL MACROEXPAND
30a3d828
JM
253
254 GOSUB LIST_Q
cc9dbd92 255 IF R<>1 THEN GOTO EVAL_NOT_LIST
30a3d828
JM
256
257 GOSUB EMPTY_Q
bbab5c5d 258 IF R THEN R=A:Z%(R,0)=Z%(R,0)+32:GOTO EVAL_RETURN
30a3d828 259
bbab5c5d
JM
260 A0=A+1
261 R=A0:GOSUB DEREF_R:A0=R
30a3d828
JM
262
263 REM get symbol in A$
bbab5c5d
JM
264 IF (Z%(A0,0)AND31)<>5 THEN A$=""
265 IF (Z%(A0,0)AND31)=5 THEN A$=S$(Z%(A0,1))
30a3d828
JM
266
267 IF A$="def!" THEN GOTO EVAL_DEF
268 IF A$="let*" THEN GOTO EVAL_LET
269 IF A$="quote" THEN GOTO EVAL_QUOTE
270 IF A$="quasiquote" THEN GOTO EVAL_QUASIQUOTE
271 IF A$="defmacro!" THEN GOTO EVAL_DEFMACRO
272 IF A$="macroexpand" THEN GOTO EVAL_MACROEXPAND
273 IF A$="try*" THEN GOTO EVAL_TRY
274 IF A$="do" THEN GOTO EVAL_DO
275 IF A$="if" THEN GOTO EVAL_IF
276 IF A$="fn*" THEN GOTO EVAL_FN
277 GOTO EVAL_INVOKE
278
279 EVAL_GET_A3:
bbab5c5d
JM
280 A3=Z%(Z%(Z%(A,1),1),1)+1
281 R=A3:GOSUB DEREF_R:A3=R
30a3d828 282 EVAL_GET_A2:
bbab5c5d
JM
283 A2=Z%(Z%(A,1),1)+1
284 R=A2:GOSUB DEREF_R:A2=R
30a3d828 285 EVAL_GET_A1:
bbab5c5d
JM
286 A1=Z%(A,1)+1
287 R=A1:GOSUB DEREF_R:A1=R
30a3d828
JM
288 RETURN
289
290 EVAL_DEF:
291 REM PRINT "def!"
bbab5c5d 292 GOSUB EVAL_GET_A2: REM set A1 and A2
30a3d828 293
bbab5c5d 294 X=X+1:X%(X)=A1: REM push A1
af621e3a 295 A=A2:CALL EVAL: REM eval a2
bbab5c5d 296 A1=X%(X):X=X-1: REM pop A1
30a3d828 297
cc9dbd92 298 IF ER<>-2 THEN GOTO EVAL_RETURN
30a3d828
JM
299
300 REM set a1 in env to a2
bbab5c5d 301 K=A1:V=R:GOSUB ENV_SET
30a3d828
JM
302 GOTO EVAL_RETURN
303
304 EVAL_LET:
305 REM PRINT "let*"
bbab5c5d 306 GOSUB EVAL_GET_A2: REM set A1 and A2
30a3d828 307
bbab5c5d
JM
308 X=X+1:X%(X)=A2: REM push/save A2
309 X=X+1:X%(X)=E: REM push env for for later release
30a3d828
JM
310
311 REM create new environment with outer as current environment
cc9dbd92
JM
312 O=E:GOSUB ENV_NEW
313 E=R
30a3d828 314 EVAL_LET_LOOP:
bbab5c5d 315 IF Z%(A1,1)=0 THEN GOTO EVAL_LET_LOOP_DONE
30a3d828 316
bbab5c5d 317 X=X+1:X%(X)=A1: REM push A1
30a3d828 318 REM eval current A1 odd element
af621e3a 319 A=Z%(A1,1)+1:CALL EVAL
bbab5c5d 320 A1=X%(X):X=X-1: REM pop A1
30a3d828 321
c7330b3d
JM
322 IF ER<>-2 THEN GOTO EVAL_LET_LOOP_DONE
323
bbab5c5d
JM
324 REM set environment: even A1 key to odd A1 eval'd above
325 K=A1+1:V=R:GOSUB ENV_SET
cc9dbd92 326 AY=R:GOSUB RELEASE: REM release our use, ENV_SET took ownership
30a3d828 327
bbab5c5d
JM
328 REM skip to the next pair of A1 elements
329 A1=Z%(Z%(A1,1),1)
30a3d828
JM
330 GOTO EVAL_LET_LOOP
331
332 EVAL_LET_LOOP_DONE:
bbab5c5d 333 E4=X%(X):X=X-1: REM pop previous env
30a3d828
JM
334
335 REM release previous environment if not the current EVAL env
bbab5c5d 336 IF E4<>X%(X-2) THEN AY=E4:GOSUB RELEASE
30a3d828 337
bbab5c5d
JM
338 A2=X%(X):X=X-1: REM pop A2
339 A=A2:GOTO EVAL_TCO_RECUR: REM TCO loop
30a3d828
JM
340
341 EVAL_DO:
cc9dbd92 342 A=Z%(A,1): REM rest
01e8850d 343 X=X+1:X%(X)=A: REM push/save A
30a3d828 344
af621e3a 345 CALL EVAL_AST
30a3d828 346
01e8850d
JM
347 REM cleanup
348 AY=R: REM get eval'd list for release
349
350 A=X%(X):X=X-1: REM pop/restore original A for LAST
351 GOSUB LAST: REM get last element for return
352 A=R: REM new recur AST
353
354 REM cleanup
355 GOSUB RELEASE: REM release eval'd list
356 AY=A:GOSUB RELEASE: REM release LAST value (not sure why)
357
358 GOTO EVAL_TCO_RECUR: REM TCO loop
30a3d828
JM
359
360 EVAL_QUOTE:
cc9dbd92 361 R=Z%(A,1)+1:GOSUB DEREF_R
bbab5c5d 362 Z%(R,0)=Z%(R,0)+32
30a3d828
JM
363 GOTO EVAL_RETURN
364
365 EVAL_QUASIQUOTE:
cc9dbd92 366 R=Z%(A,1)+1:GOSUB DEREF_R
af621e3a 367 A=R:CALL QUASIQUOTE
30a3d828 368 REM add quasiquote result to pending release queue to free when
cc9dbd92 369 REM next lower EVAL level returns (LV)
bbab5c5d 370 Y=Y+1:Y%(Y,0)=R:Y%(Y,1)=LV
30a3d828 371
cc9dbd92 372 A=R:GOTO EVAL_TCO_RECUR: REM TCO loop
30a3d828
JM
373
374 EVAL_DEFMACRO:
375 REM PRINT "defmacro!"
bbab5c5d 376 GOSUB EVAL_GET_A2: REM set A1 and A2
30a3d828 377
bbab5c5d 378 X=X+1:X%(X)=A1: REM push A1
af621e3a 379 A=A2:CALL EVAL: REM eval A2
bbab5c5d 380 A1=X%(X):X=X-1: REM pop A1
30a3d828
JM
381
382 REM change function to macro
cc9dbd92 383 Z%(R,0)=Z%(R,0)+1
30a3d828 384
bbab5c5d
JM
385 REM set A1 in env to A2
386 K=A1:V=R:GOSUB ENV_SET
30a3d828
JM
387 GOTO EVAL_RETURN
388
389 EVAL_MACROEXPAND:
390 REM PRINT "macroexpand"
cc9dbd92 391 R=Z%(A,1)+1:GOSUB DEREF_R
af621e3a
JM
392 A=R:CALL MACROEXPAND
393 R=A
30a3d828
JM
394
395 REM since we are returning it unevaluated, inc the ref cnt
bbab5c5d 396 Z%(R,0)=Z%(R,0)+32
30a3d828
JM
397 GOTO EVAL_RETURN
398
399 EVAL_TRY:
400 REM PRINT "try*"
bbab5c5d 401 GOSUB EVAL_GET_A1: REM set A1, A2, and A3
30a3d828 402
bbab5c5d 403 X=X+1:X%(X)=A: REM push/save A
af621e3a 404 A=A1:CALL EVAL: REM eval A1
bbab5c5d 405 A=X%(X):X=X-1: REM pop/restore A
30a3d828
JM
406
407 REM if there is not error or catch block then return
cc9dbd92 408 IF ER=-2 OR Z%(A,1)=0 THEN GOTO EVAL_RETURN
30a3d828
JM
409
410 REM create environment for the catch block eval
cc9dbd92 411 O=E:GOSUB ENV_NEW:E=R
30a3d828 412
bbab5c5d
JM
413 GOSUB EVAL_GET_A2: REM set A1 and A2
414 A=A2:GOSUB EVAL_GET_A2: REM set A1 and A2 from catch block
30a3d828 415
cc9dbd92 416 REM create object for ER=-1 type raw string errors
bbab5c5d 417 IF ER=-1 THEN AS$=ER$:T=4:GOSUB STRING:ER=R:Z%(R,0)=Z%(R,0)+32
30a3d828
JM
418
419 REM bind the catch symbol to the error object
bbab5c5d 420 K=A1:V=ER:GOSUB ENV_SET
c7330b3d 421 AY=R:GOSUB RELEASE: REM release our use, env took ownership
30a3d828
JM
422
423 REM unset error for catch eval
cc9dbd92 424 ER=-2:ER$=""
30a3d828 425
af621e3a 426 A=A2:CALL EVAL
30a3d828
JM
427
428 GOTO EVAL_RETURN
429
430 EVAL_IF:
bbab5c5d 431 GOSUB EVAL_GET_A1: REM set A1
cc9dbd92 432 REM push A
bbab5c5d 433 X=X+1:X%(X)=A
af621e3a 434 A=A1:CALL EVAL
cc9dbd92 435 REM pop A
bbab5c5d 436 A=X%(X):X=X-1
cc9dbd92 437 IF (R=0) OR (R=1) THEN GOTO EVAL_IF_FALSE
30a3d828
JM
438
439 EVAL_IF_TRUE:
cc9dbd92 440 AY=R:GOSUB RELEASE
bbab5c5d
JM
441 GOSUB EVAL_GET_A2: REM set A1 and A2 after EVAL
442 A=A2:GOTO EVAL_TCO_RECUR: REM TCO loop
30a3d828 443 EVAL_IF_FALSE:
cc9dbd92 444 AY=R:GOSUB RELEASE
bbab5c5d 445 REM if no false case (A3), return nil
c7330b3d
JM
446 B=A:GOSUB COUNT
447 IF R<4 THEN R=0:GOTO EVAL_RETURN
bbab5c5d
JM
448 GOSUB EVAL_GET_A3: REM set A1 - A3 after EVAL
449 A=A3:GOTO EVAL_TCO_RECUR: REM TCO loop
30a3d828
JM
450
451 EVAL_FN:
bbab5c5d
JM
452 GOSUB EVAL_GET_A2: REM set A1 and A2
453 A=A2:P=A1:GOSUB MAL_FUNCTION
30a3d828
JM
454 GOTO EVAL_RETURN
455
456 EVAL_INVOKE:
af621e3a 457 CALL EVAL_AST
30a3d828
JM
458
459 REM if error, return f/args for release by caller
cc9dbd92 460 IF ER<>-2 THEN GOTO EVAL_RETURN
30a3d828
JM
461
462 REM push f/args for release after call
bbab5c5d 463 X=X+1:X%(X)=R
30a3d828 464
cc9dbd92 465 F=R+1
30a3d828 466
cc9dbd92
JM
467 AR=Z%(R,1): REM rest
468 R=F:GOSUB DEREF_R:F=R
30a3d828 469
bbab5c5d
JM
470 REM if metadata, get the actual object
471 IF (Z%(F,0)AND31)>=16 THEN F=Z%(F,1)
472
473 IF (Z%(F,0)AND31)=9 THEN GOTO EVAL_DO_FUNCTION
474 IF (Z%(F,0)AND31)=10 THEN GOTO EVAL_DO_MAL_FUNCTION
30a3d828
JM
475
476 REM if error, pop and return f/args for release by caller
bbab5c5d 477 R=X%(X):X=X-1
cc9dbd92 478 ER=-1:ER$="apply of non-function":GOTO EVAL_RETURN
30a3d828
JM
479
480 EVAL_DO_FUNCTION:
01e8850d 481 REM regular function
af621e3a 482 IF Z%(F,1)<60 THEN GOSUB DO_FUNCTION:GOTO EVAL_DO_FUNCTION_SKIP
01e8850d 483 REM for recur functions (apply, map, swap!), use GOTO
af621e3a
JM
484 IF Z%(F,1)>60 THEN CALL DO_TCO_FUNCTION
485 EVAL_DO_FUNCTION_SKIP:
30a3d828
JM
486
487 REM pop and release f/args
bbab5c5d 488 AY=X%(X):X=X-1:GOSUB RELEASE
30a3d828
JM
489 GOTO EVAL_RETURN
490
491 EVAL_DO_MAL_FUNCTION:
bbab5c5d 492 E4=E: REM save the current environment for release
30a3d828
JM
493
494 REM create new environ using env stored with function
bbab5c5d 495 O=Z%(F+1,1):BI=Z%(F+1,0):EX=AR:GOSUB ENV_NEW_BINDS
30a3d828
JM
496
497 REM release previous env if it is not the top one on the
bbab5c5d 498 REM stack (X%(X-2)) because our new env refers to it and
30a3d828 499 REM we no longer need to track it (since we are TCO recurring)
bbab5c5d 500 IF E4<>X%(X-2) THEN AY=E4:GOSUB RELEASE
30a3d828
JM
501
502 REM claim the AST before releasing the list containing it
bbab5c5d 503 A=Z%(F,1):Z%(A,0)=Z%(A,0)+32
30a3d828 504 REM add AST to pending release queue to free as soon as EVAL
cc9dbd92 505 REM actually returns (LV+1)
bbab5c5d 506 Y=Y+1:Y%(Y,0)=A:Y%(Y,1)=LV+1
30a3d828
JM
507
508 REM pop and release f/args
bbab5c5d 509 AY=X%(X):X=X-1:GOSUB RELEASE
30a3d828 510
cc9dbd92
JM
511 REM A set above
512 E=R:GOTO EVAL_TCO_RECUR: REM TCO loop
30a3d828
JM
513
514 EVAL_RETURN:
cc9dbd92
JM
515 REM AZ=R: PR=1: GOSUB PR_STR
516 REM PRINT "EVAL_RETURN R: ["+R$+"] ("+STR$(R)+"), LV:"+STR$(LV)+",ER:"+STR$(ER)
30a3d828
JM
517
518 REM release environment if not the top one on the stack
bbab5c5d 519 IF E<>X%(X-1) THEN AY=E:GOSUB RELEASE
30a3d828 520
cc9dbd92 521 LV=LV-1: REM track basic return stack level
30a3d828
JM
522
523 REM release everything we couldn't release earlier
524 GOSUB RELEASE_PEND
525
526 REM trigger GC
bbab5c5d 527 TA=FRE(0)
30a3d828 528
cc9dbd92 529 REM pop A and E off the stack
bbab5c5d 530 E=X%(X-1):A=X%(X):X=X-2
30a3d828 531
af621e3a 532END SUB
30a3d828 533
cc9dbd92 534REM PRINT(A) -> R$
30a3d828 535MAL_PRINT:
cc9dbd92 536 AZ=A:PR=1:GOSUB PR_STR
30a3d828
JM
537 RETURN
538
cc9dbd92 539REM RE(A$) -> R
bbab5c5d 540REM Assume D has repl_env
30a3d828
JM
541REM caller must release result
542RE:
cc9dbd92 543 R1=0
30a3d828 544 GOSUB MAL_READ
cc9dbd92 545 R1=R
0e508fa5 546 IF ER<>-2 THEN GOTO RE_DONE
30a3d828 547
af621e3a 548 A=R:E=D:CALL EVAL
30a3d828 549
0e508fa5 550 RE_DONE:
30a3d828 551 REM Release memory from MAL_READ
cc9dbd92 552 IF R1<>0 THEN AY=R1:GOSUB RELEASE
30a3d828
JM
553 RETURN: REM caller must release result of EVAL
554
555REM REP(A$) -> R$
bbab5c5d 556REM Assume D has repl_env
af621e3a 557SUB REP
cc9dbd92 558 R1=0:R2=0
30a3d828 559 GOSUB MAL_READ
cc9dbd92
JM
560 R1=R
561 IF ER<>-2 THEN GOTO REP_DONE
30a3d828 562
af621e3a 563 A=R:E=D:CALL EVAL
cc9dbd92
JM
564 R2=R
565 IF ER<>-2 THEN GOTO REP_DONE
30a3d828 566
cc9dbd92 567 A=R:GOSUB MAL_PRINT
30a3d828
JM
568 RT$=R$
569
570 REP_DONE:
571 REM Release memory from MAL_READ and EVAL
cc9dbd92
JM
572 IF R2<>0 THEN AY=R2:GOSUB RELEASE
573 IF R1<>0 THEN AY=R1:GOSUB RELEASE
30a3d828 574 R$=RT$
af621e3a 575END SUB
30a3d828
JM
576
577REM MAIN program
578MAIN:
579 GOSUB INIT_MEMORY
580
cc9dbd92 581 LV=0
30a3d828
JM
582
583 REM create repl_env
bbab5c5d 584 O=-1:GOSUB ENV_NEW:D=R
30a3d828
JM
585
586 REM core.EXT: defined in Basic
bbab5c5d 587 E=D:GOSUB INIT_CORE_NS: REM set core functions in repl_env
30a3d828 588
bbab5c5d 589 ZT=ZI: REM top of memory after base repl_env
30a3d828
JM
590
591 REM core.mal: defined using the language itself
592 A$="(def! *host-language* "+CHR$(34)+"C64 Basic"+CHR$(34)+")"
cc9dbd92 593 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828
JM
594
595 A$="(def! not (fn* (a) (if a false true)))"
cc9dbd92 596 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828 597
0e508fa5 598 A$="(def! load-file (fn* (f) (eval (read-file f))))"
cc9dbd92 599 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828
JM
600
601 A$="(defmacro! cond (fn* (& xs) (if (> (count xs) 0) (list 'if (first xs)"
602 A$=A$+" (if (> (count xs) 1) (nth xs 1) (throw "+CHR$(34)+"odd number of"
603 A$=A$+" forms to cond"+CHR$(34)+")) (cons 'cond (rest (rest xs)))))))"
cc9dbd92 604 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828 605
4e7d8a1b
JM
606 A$="(def! *gensym-counter* (atom 0))"
607 GOSUB RE:AY=R:GOSUB RELEASE
608
609 A$="(def! gensym (fn* [] (symbol (str "+CHR$(34)+"G__"+CHR$(34)
610 A$=A$+" (swap! *gensym-counter* (fn* [x] (+ 1 x)))))))"
611 GOSUB RE:AY=R:GOSUB RELEASE
612
30a3d828 613 A$="(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs)"
4e7d8a1b
JM
614 A$=A$+" (let* (condvar (gensym)) `(let* (~condvar ~(first xs))"
615 A$=A$+" (if ~condvar ~condvar (or ~@(rest xs)))))))))"
cc9dbd92 616 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828
JM
617
618 REM load the args file
619 A$="(def! -*ARGS*- (load-file "+CHR$(34)+".args.mal"+CHR$(34)+"))"
cc9dbd92 620 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828
JM
621
622 REM set the argument list
623 A$="(def! *ARGV* (rest -*ARGS*-))"
cc9dbd92 624 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828
JM
625
626 REM get the first argument
627 A$="(first -*ARGS*-)"
628 GOSUB RE
629
630 REM if there is an argument, then run it as a program
cc9dbd92 631 IF R<>0 THEN AY=R:GOSUB RELEASE:GOTO RUN_PROG
30a3d828 632 REM no arguments, start REPL loop
cc9dbd92 633 IF R=0 THEN GOTO REPL
30a3d828
JM
634
635 RUN_PROG:
636 REM run a single mal program and exit
637 A$="(load-file (first -*ARGS*-))"
638 GOSUB RE
cc9dbd92 639 IF ER<>-2 THEN GOSUB PRINT_ERROR
01e8850d 640 GOTO QUIT
30a3d828
JM
641
642 REPL:
643 REM print the REPL startup header
644 A$="(println (str "+CHR$(34)+"Mal ["+CHR$(34)+" *host-language* "
645 A$=A$+CHR$(34)+"]"+CHR$(34)+"))"
cc9dbd92 646 GOSUB RE:AY=R:GOSUB RELEASE
30a3d828
JM
647
648 REPL_LOOP:
649 A$="user> ":GOSUB READLINE: REM call input parser
650 IF EOF=1 THEN GOTO QUIT
651
af621e3a 652 A$=R$:CALL REP: REM call REP
30a3d828 653
cc9dbd92 654 IF ER<>-2 THEN GOSUB PRINT_ERROR:GOTO REPL_LOOP
30a3d828
JM
655 PRINT R$
656 GOTO REPL_LOOP
657
658 QUIT:
4e7d8a1b 659 REM GOSUB PR_MEMORY_SUMMARY
30a3d828
JM
660 END
661
662 PRINT_ERROR:
663 REM if the error is an object, then print and free it
cc9dbd92 664 IF ER>=0 THEN AZ=ER:PR=0:GOSUB PR_STR:ER$=R$:AY=ER:GOSUB RELEASE
30a3d828 665 PRINT "Error: "+ER$
cc9dbd92 666 ER=-2:ER$=""
30a3d828
JM
667 RETURN
668