1
1
mirror of https://github.com/kanaka/mal.git synced 2024-09-20 01:57:09 +03:00
mal/basic/step8_macros.in.bas
Joel Martin cc9dbd92e3 Basic: variable renaming. Save 2 kbytes.
Also, add variables.txt file with start of documenting meanings of
variables.

List of renamings/savings:

ZZ%   -> S%   : 131 bytes
ZL%   -> X    : 550 bytes
A%    -> A    : 192 bytes
E%    -> E    :  32 bytes
R%    -> R    : 381 bytes
AR%   -> AR   :  30 bytes
AY%   -> AY   :  71 bytes
AZ%   -> AZ   :  33 bytes
B%    -> B    :  47 bytes
AA%   -> AA   :  64 bytes
AB%   -> AB   :  25 bytes
F%    -> F    :  21 bytes
FF%   -> FF   :  14 bytes
ER%   -> ER   :  41 bytes
PR%   -> PR   :   7 bytes
T%    -> T    :  46 bytes
R0-9% -> R0-9 :  31 bytes
T0-9% -> T0-9 :  42 bytes
S1-4% -> S1-4 :  25 bytes
U0-9% -> U0-9 :  44 bytes
ZK%   -> ZK   :  10 bytes
ZI%   -> ZI   :  10 bytes
RC%   -> RC   :  16 bytes
K%/V% -> K/V  :  21 bytes
SD%   -> SD   :  16 bytes
ZS$   -> S$   :  40 bytes
HM%   -> H    :  10 bytes
SZ%   -> SZ   :  39 bytes
LV%   -> LV   :   9 bytes
EO%   -> O    :  18 bytes
C%    -> C    :   4 bytes
P%    -> P    :   4 bytes
2016-10-14 22:42:56 -05:00

614 lines
15 KiB
QBasic
Executable File

REM POKE 1, PEEK(1)AND248: REM enable all ROM areas as RAM
REM POKE 55,0: POKE 56,192: CLR: REM move BASIC end from $A000 to $C000
GOTO MAIN
REM $INCLUDE: 'readline.in.bas'
REM $INCLUDE: 'types.in.bas'
REM $INCLUDE: 'reader.in.bas'
REM $INCLUDE: 'printer.in.bas'
REM $INCLUDE: 'env.in.bas'
REM $INCLUDE: 'core.in.bas'
REM $INCLUDE: 'debug.in.bas'
REM READ(A$) -> R
MAL_READ:
GOSUB READ_STR
RETURN
REM PAIR_Q(B) -> R
PAIR_Q:
R=0
IF (Z%(B,0)AND15)<>6 AND (Z%(B,0)AND15)<>7 THEN RETURN
IF (Z%(B,1)=0) THEN RETURN
R=1
RETURN
REM QUASIQUOTE(A) -> R
QUASIQUOTE:
B=A:GOSUB PAIR_Q
IF R=1 THEN GOTO QQ_UNQUOTE
REM ['quote, ast]
AS$="quote":T=5:GOSUB STRING
B2%=R:B1%=A:GOSUB LIST2
RETURN
QQ_UNQUOTE:
R=A+1:GOSUB DEREF_R
IF (Z%(R,0)AND15)<>5 THEN GOTO QQ_SPLICE_UNQUOTE
IF S$(Z%(R,1))<>"unquote" THEN GOTO QQ_SPLICE_UNQUOTE
REM [ast[1]]
R=Z%(A,1)+1:GOSUB DEREF_R
Z%(R,0)=Z%(R,0)+16
RETURN
QQ_SPLICE_UNQUOTE:
REM push A on the stack
X=X+1:S%(X)=A
REM rest of cases call quasiquote on ast[1..]
A=Z%(A,1):GOSUB QUASIQUOTE:T6=R
REM pop A off the stack
A=S%(X):X=X-1
REM set A to ast[0] for last two cases
A=A+1:GOSUB DEREF_A
B=A:GOSUB PAIR_Q
IF R=0 THEN GOTO QQ_DEFAULT
B=A+1:GOSUB DEREF_B
IF (Z%(B,0)AND15)<>5 THEN GOTO QQ_DEFAULT
IF S$(Z%(B,1))<>"splice-unquote" THEN QQ_DEFAULT
REM ['concat, ast[0][1], quasiquote(ast[1..])]
B=Z%(A,1)+1:GOSUB DEREF_B:B2%=B
AS$="concat":T=5:GOSUB STRING:B3%=R
B1%=T6:GOSUB LIST3
REM release inner quasiquoted since outer list takes ownership
AY=B1%:GOSUB RELEASE
RETURN
QQ_DEFAULT:
REM ['cons, quasiquote(ast[0]), quasiquote(ast[1..])]
REM push T6 on the stack
X=X+1:S%(X)=T6
REM A set above to ast[0]
GOSUB QUASIQUOTE:B2%=R
REM pop T6 off the stack
T6=S%(X):X=X-1
AS$="cons":T=5:GOSUB STRING:B3%=R
B1%=T6:GOSUB LIST3
REM release inner quasiquoted since outer list takes ownership
AY=B1%:GOSUB RELEASE
AY=B2%:GOSUB RELEASE
RETURN
REM MACROEXPAND(A, E) -> A:
MACROEXPAND:
REM push original A
X=X+1:S%(X)=A
MACROEXPAND_LOOP:
REM list?
IF (Z%(A,0)AND15)<>6 THEN GOTO MACROEXPAND_DONE
REM non-empty?
IF Z%(A,1)=0 THEN GOTO MACROEXPAND_DONE
B=A+1:GOSUB DEREF_B
REM symbol? in first position
IF (Z%(B,0)AND15)<>5 THEN GOTO MACROEXPAND_DONE
REM defined in environment?
K=B:GOSUB ENV_FIND
IF R=-1 THEN GOTO MACROEXPAND_DONE
B=T4:GOSUB DEREF_B
REM macro?
IF (Z%(B,0)AND15)<>11 THEN GOTO MACROEXPAND_DONE
REM apply
F=B:AR=Z%(A,1):GOSUB APPLY
A=R
AY=S%(X)
REM if previous A was not the first A into macroexpand (i.e. an
REM intermediate form) then free it
IF A<>AY THEN ZM%=ZM%+1:ZR%(ZM%,0)=A:ZR%(ZM%,1)=LV
IF ER<>-2 THEN GOTO MACROEXPAND_DONE
GOTO MACROEXPAND_LOOP
MACROEXPAND_DONE:
X=X-1: REM pop original A
RETURN
REM EVAL_AST(A, E) -> R
REM called using GOTO to avoid basic return address stack usage
REM top of stack should have return label index
EVAL_AST:
REM push A and E on the stack
X=X+2:S%(X-1)=E:S%(X)=A
IF ER<>-2 THEN GOTO EVAL_AST_RETURN
GOSUB DEREF_A
T=Z%(A,0)AND15
IF T=5 THEN GOTO EVAL_AST_SYMBOL
IF T>=6 AND T<=8 THEN GOTO EVAL_AST_SEQ
REM scalar: deref to actual value and inc ref cnt
R=A:GOSUB DEREF_R
Z%(R,0)=Z%(R,0)+16
GOTO EVAL_AST_RETURN
EVAL_AST_SYMBOL:
K=A:GOSUB ENV_GET
GOTO EVAL_AST_RETURN
EVAL_AST_SEQ:
REM allocate the first entry
SZ=2:GOSUB ALLOC
REM make space on the stack
X=X+4
REM push type of sequence
S%(X-3)=T
REM push sequence index
S%(X-2)=-1
REM push future return value (new sequence)
S%(X-1)=R
REM push previous new sequence entry
S%(X)=R
EVAL_AST_SEQ_LOOP:
REM set new sequence entry type (with 1 ref cnt)
Z%(R,0)=S%(X-3)+16
Z%(R,1)=0
REM create value ptr placeholder
Z%(R+1,0)=14
Z%(R+1,1)=0
REM update index
S%(X-2)=S%(X-2)+1
REM check if we are done evaluating the source sequence
IF Z%(A,1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
REM if hashmap, skip eval of even entries (keys)
IF (S%(X-3)=8) AND ((S%(X-2) AND 1)=0) THEN GOTO EVAL_AST_DO_REF
GOTO EVAL_AST_DO_EVAL
EVAL_AST_DO_REF:
R=A+1:GOSUB DEREF_R: REM deref to target of referred entry
Z%(R,0)=Z%(R,0)+16: REM inc ref cnt of referred value
GOTO EVAL_AST_ADD_VALUE
EVAL_AST_DO_EVAL:
REM call EVAL for each entry
A=A+1:GOSUB EVAL
A=A-1
GOSUB DEREF_R: REM deref to target of evaluated entry
EVAL_AST_ADD_VALUE:
REM update previous value pointer to evaluated entry
Z%(S%(X)+1,1)=R
IF ER<>-2 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
REM allocate the next entry
SZ=2:GOSUB ALLOC
REM update previous sequence entry value to point to new entry
Z%(S%(X),1)=R
REM update previous ptr to current entry
S%(X)=R
REM process the next sequence entry from source list
A=Z%(A,1)
GOTO EVAL_AST_SEQ_LOOP
EVAL_AST_SEQ_LOOP_DONE:
REM if no error, get return value (new seq)
IF ER=-2 THEN R=S%(X-1)
REM otherwise, free the return value and return nil
IF ER<>-2 THEN R=0:AY=S%(X-1):GOSUB RELEASE
REM pop previous, return, index and type
X=X-4
GOTO EVAL_AST_RETURN
EVAL_AST_RETURN:
REM pop A and E off the stack
E=S%(X-1):A=S%(X):X=X-2
REM pop EVAL AST return label/address
RN%=S%(X):X=X-1
ON RN% GOTO EVAL_AST_RETURN_1,EVAL_AST_RETURN_2,EVAL_AST_RETURN_3
RETURN
REM EVAL(A, E)) -> R
EVAL:
LV=LV+1: REM track basic return stack level
REM push A and E on the stack
X=X+2:S%(X-1)=E:S%(X)=A
EVAL_TCO_RECUR:
REM AZ=A:PR=1:GOSUB PR_STR
REM PRINT "EVAL: "+R$+" [A:"+STR$(A)+", LV:"+STR$(LV)+"]"
GOSUB DEREF_A
GOSUB LIST_Q
IF R THEN GOTO APPLY_LIST
EVAL_NOT_LIST:
REM ELSE
REM push EVAL_AST return label/address
X=X+1:S%(X)=1
GOTO EVAL_AST
EVAL_AST_RETURN_1:
GOTO EVAL_RETURN
APPLY_LIST:
GOSUB MACROEXPAND
GOSUB LIST_Q
IF R<>1 THEN GOTO EVAL_NOT_LIST
GOSUB EMPTY_Q
IF R THEN R=A:Z%(R,0)=Z%(R,0)+16:GOTO EVAL_RETURN
A0%=A+1
R=A0%:GOSUB DEREF_R:A0%=R
REM get symbol in A$
IF (Z%(A0%,0)AND15)<>5 THEN A$=""
IF (Z%(A0%,0)AND15)=5 THEN A$=S$(Z%(A0%,1))
IF A$="def!" THEN GOTO EVAL_DEF
IF A$="let*" THEN GOTO EVAL_LET
IF A$="quote" THEN GOTO EVAL_QUOTE
IF A$="quasiquote" THEN GOTO EVAL_QUASIQUOTE
IF A$="defmacro!" THEN GOTO EVAL_DEFMACRO
IF A$="macroexpand" THEN GOTO EVAL_MACROEXPAND
IF A$="do" THEN GOTO EVAL_DO
IF A$="if" THEN GOTO EVAL_IF
IF A$="fn*" THEN GOTO EVAL_FN
GOTO EVAL_INVOKE
EVAL_GET_A3:
A3%=Z%(Z%(Z%(A,1),1),1)+1
R=A3%:GOSUB DEREF_R:A3%=R
EVAL_GET_A2:
A2%=Z%(Z%(A,1),1)+1
R=A2%:GOSUB DEREF_R:A2%=R
EVAL_GET_A1:
A1%=Z%(A,1)+1
R=A1%:GOSUB DEREF_R:A1%=R
RETURN
EVAL_DEF:
REM PRINT "def!"
GOSUB EVAL_GET_A2: REM set a1% and a2%
X=X+1:S%(X)=A1%: REM push A1%
A=A2%:GOSUB EVAL: REM eval a2
A1%=S%(X):X=X-1: REM pop A1%
IF ER<>-2 THEN GOTO EVAL_RETURN
REM set a1 in env to a2
K=A1%:V=R:GOSUB ENV_SET
GOTO EVAL_RETURN
EVAL_LET:
REM PRINT "let*"
GOSUB EVAL_GET_A2: REM set a1% and a2%
X=X+1:S%(X)=A2%: REM push/save A2%
X=X+1:S%(X)=E: REM push env for for later release
REM create new environment with outer as current environment
O=E:GOSUB ENV_NEW
E=R
EVAL_LET_LOOP:
IF Z%(A1%,1)=0 THEN GOTO EVAL_LET_LOOP_DONE
X=X+1:S%(X)=A1%: REM push A1%
REM eval current A1 odd element
A=Z%(A1%,1)+1:GOSUB EVAL
A1%=S%(X):X=X-1: REM pop A1%
REM set environment: even A1% key to odd A1% eval'd above
K=A1%+1:V=R:GOSUB ENV_SET
AY=R:GOSUB RELEASE: REM release our use, ENV_SET took ownership
REM skip to the next pair of A1% elements
A1%=Z%(Z%(A1%,1),1)
GOTO EVAL_LET_LOOP
EVAL_LET_LOOP_DONE:
E4%=S%(X):X=X-1: REM pop previous env
REM release previous environment if not the current EVAL env
IF E4%<>S%(X-2) THEN AY=E4%:GOSUB RELEASE
A2%=S%(X):X=X-1: REM pop A2%
A=A2%:GOTO EVAL_TCO_RECUR: REM TCO loop
EVAL_DO:
A=Z%(A,1): REM rest
REM TODO: TCO
REM push EVAL_AST return label/address
X=X+1:S%(X)=2
GOTO EVAL_AST
EVAL_AST_RETURN_2:
X=X+1:S%(X)=R: REM push eval'd list
A=R:GOSUB LAST: REM return the last element
AY=S%(X):X=X-1: REM pop eval'd list
GOSUB RELEASE: REM release the eval'd list
GOTO EVAL_RETURN
EVAL_QUOTE:
R=Z%(A,1)+1:GOSUB DEREF_R
Z%(R,0)=Z%(R,0)+16
GOTO EVAL_RETURN
EVAL_QUASIQUOTE:
R=Z%(A,1)+1:GOSUB DEREF_R
A=R:GOSUB QUASIQUOTE
REM add quasiquote result to pending release queue to free when
REM next lower EVAL level returns (LV)
ZM%=ZM%+1:ZR%(ZM%,0)=R:ZR%(ZM%,1)=LV
A=R:GOTO EVAL_TCO_RECUR: REM TCO loop
EVAL_DEFMACRO:
REM PRINT "defmacro!"
GOSUB EVAL_GET_A2: REM set a1% and a2%
X=X+1:S%(X)=A1%: REM push A1%
A=A2%:GOSUB EVAL: REM eval a2
A1%=S%(X):X=X-1: REM pop A1%
REM change function to macro
Z%(R,0)=Z%(R,0)+1
REM set a1 in env to a2
K=A1%:V=R:GOSUB ENV_SET
GOTO EVAL_RETURN
EVAL_MACROEXPAND:
REM PRINT "macroexpand"
R=Z%(A,1)+1:GOSUB DEREF_R
A=R:GOSUB MACROEXPAND:R=A
REM since we are returning it unevaluated, inc the ref cnt
Z%(R,0)=Z%(R,0)+16
GOTO EVAL_RETURN
EVAL_IF:
GOSUB EVAL_GET_A1: REM set a1%
REM push A
X=X+1:S%(X)=A
A=A1%:GOSUB EVAL
REM pop A
A=S%(X):X=X-1
IF (R=0) OR (R=1) THEN GOTO EVAL_IF_FALSE
EVAL_IF_TRUE:
AY=R:GOSUB RELEASE
GOSUB EVAL_GET_A2: REM set a1% and a2% after EVAL
A=A2%:GOTO EVAL_TCO_RECUR: REM TCO loop
EVAL_IF_FALSE:
AY=R:GOSUB RELEASE
REM if no false case (A3%), return nil
IF Z%(Z%(Z%(A,1),1),1)=0 THEN R=0:GOTO EVAL_RETURN
GOSUB EVAL_GET_A3: REM set a1% - a3% after EVAL
A=A3%:GOTO EVAL_TCO_RECUR: REM TCO loop
EVAL_FN:
GOSUB EVAL_GET_A2: REM set a1% and a2%
A=A2%:P=A1%:GOSUB MAL_FUNCTION
GOTO EVAL_RETURN
EVAL_INVOKE:
REM push EVAL_AST return label/address
X=X+1:S%(X)=3
GOTO EVAL_AST
EVAL_AST_RETURN_3:
REM if error, return f/args for release by caller
IF ER<>-2 THEN GOTO EVAL_RETURN
REM push f/args for release after call
X=X+1:S%(X)=R
F=R+1
AR=Z%(R,1): REM rest
R=F:GOSUB DEREF_R:F=R
IF (Z%(F,0)AND15)=9 THEN GOTO EVAL_DO_FUNCTION
IF (Z%(F,0)AND15)=10 THEN GOTO EVAL_DO_MAL_FUNCTION
REM if error, pop and return f/args for release by caller
R=S%(X):X=X-1
ER=-1:ER$="apply of non-function":GOTO EVAL_RETURN
EVAL_DO_FUNCTION:
GOSUB DO_FUNCTION
REM pop and release f/args
AY=S%(X):X=X-1:GOSUB RELEASE
GOTO EVAL_RETURN
EVAL_DO_MAL_FUNCTION:
E4%=E: REM save the current environment for release
REM create new environ using env stored with function
O=Z%(F+1,1):BI%=Z%(F+1,0):EX%=AR:GOSUB ENV_NEW_BINDS
REM release previous env if it is not the top one on the
REM stack (S%(X-2)) because our new env refers to it and
REM we no longer need to track it (since we are TCO recurring)
IF E4%<>S%(X-2) THEN AY=E4%:GOSUB RELEASE
REM claim the AST before releasing the list containing it
A=Z%(F,1):Z%(A,0)=Z%(A,0)+16
REM add AST to pending release queue to free as soon as EVAL
REM actually returns (LV+1)
ZM%=ZM%+1:ZR%(ZM%,0)=A:ZR%(ZM%,1)=LV+1
REM pop and release f/args
AY=S%(X):X=X-1:GOSUB RELEASE
REM A set above
E=R:GOTO EVAL_TCO_RECUR: REM TCO loop
EVAL_RETURN:
REM AZ=R: PR=1: GOSUB PR_STR
REM PRINT "EVAL_RETURN R: ["+R$+"] ("+STR$(R)+"), LV:"+STR$(LV)+",ER:"+STR$(ER)
REM release environment if not the top one on the stack
IF E<>S%(X-1) THEN AY=E:GOSUB RELEASE
LV=LV-1: REM track basic return stack level
REM release everything we couldn't release earlier
GOSUB RELEASE_PEND
REM trigger GC
TA%=FRE(0)
REM pop A and E off the stack
E=S%(X-1):A=S%(X):X=X-2
RETURN
REM PRINT(A) -> R$
MAL_PRINT:
AZ=A:PR=1:GOSUB PR_STR
RETURN
REM RE(A$) -> R
REM Assume RE% has repl_env
REM caller must release result
RE:
R1=0
GOSUB MAL_READ
R1=R
IF ER<>-2 THEN GOTO REP_DONE
A=R:E=RE%:GOSUB EVAL
REP_DONE:
REM Release memory from MAL_READ
IF R1<>0 THEN AY=R1:GOSUB RELEASE
RETURN: REM caller must release result of EVAL
REM REP(A$) -> R$
REM Assume RE% has repl_env
REP:
R1=0:R2=0
GOSUB MAL_READ
R1=R
IF ER<>-2 THEN GOTO REP_DONE
A=R:E=RE%:GOSUB EVAL
R2=R
IF ER<>-2 THEN GOTO REP_DONE
A=R:GOSUB MAL_PRINT
RT$=R$
REP_DONE:
REM Release memory from MAL_READ and EVAL
IF R2<>0 THEN AY=R2:GOSUB RELEASE
IF R1<>0 THEN AY=R1:GOSUB RELEASE
R$=RT$
RETURN
REM MAIN program
MAIN:
GOSUB INIT_MEMORY
LV=0
REM create repl_env
O=-1:GOSUB ENV_NEW:RE%=R
REM core.EXT: defined in Basic
E=RE%:GOSUB INIT_CORE_NS: REM set core functions in repl_env
ZT%=ZI: REM top of memory after base repl_env
REM core.mal: defined using the language itself
A$="(def! not (fn* (a) (if a false true)))"
GOSUB RE:AY=R:GOSUB RELEASE
A$="(def! load-file (fn* (f) (eval (read-string (str "
A$=A$+CHR$(34)+"(do "+CHR$(34)+" (slurp f) "+CHR$(34)+")"+CHR$(34)+")))))"
GOSUB RE:AY=R:GOSUB RELEASE
A$="(defmacro! cond (fn* (& xs) (if (> (count xs) 0) (list 'if (first xs)"
A$=A$+" (if (> (count xs) 1) (nth xs 1) (throw "+CHR$(34)+"odd number of"
A$=A$+" forms to cond"+CHR$(34)+")) (cons 'cond (rest (rest xs)))))))"
GOSUB RE:AY=R:GOSUB RELEASE
A$="(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs)"
A$=A$+" `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))"
GOSUB RE:AY=R:GOSUB RELEASE
REM load the args file
A$="(def! -*ARGS*- (load-file "+CHR$(34)+".args.mal"+CHR$(34)+"))"
GOSUB RE:AY=R:GOSUB RELEASE
REM set the argument list
A$="(def! *ARGV* (rest -*ARGS*-))"
GOSUB RE:AY=R:GOSUB RELEASE
REM get the first argument
A$="(first -*ARGS*-)"
GOSUB RE
REM if there is an argument, then run it as a program
IF R<>0 THEN AY=R:GOSUB RELEASE:GOTO RUN_PROG
REM no arguments, start REPL loop
IF R=0 THEN GOTO REPL_LOOP
RUN_PROG:
REM run a single mal program and exit
A$="(load-file (first -*ARGS*-))"
GOSUB RE
IF ER<>-2 THEN GOSUB PRINT_ERROR
END
REPL_LOOP:
A$="user> ":GOSUB READLINE: REM call input parser
IF EOF=1 THEN GOTO QUIT
A$=R$:GOSUB REP: REM call REP
IF ER<>-2 THEN GOSUB PRINT_ERROR:GOTO REPL_LOOP
PRINT R$
GOTO REPL_LOOP
QUIT:
REM P1%=ZT%: P2%=-1: GOSUB PR_MEMORY
GOSUB PR_MEMORY_SUMMARY
END
PRINT_ERROR:
PRINT "Error: "+ER$
ER=-2:ER$=""
RETURN