make, swift3: fix parsing empty literal sequences.
[jackhill/mal.git] / basic / stepA_mal.in.bas
index cdbeeab..9e6297d 100755 (executable)
@@ -1,5 +1,6 @@
 GOTO MAIN
 
+REM $INCLUDE: 'mem.in.bas'
 REM $INCLUDE: 'types.in.bas'
 REM $INCLUDE: 'readline.in.bas'
 REM $INCLUDE: 'reader.in.bas'
@@ -9,16 +10,14 @@ REM $INCLUDE: 'core.in.bas'
 
 REM $INCLUDE: 'debug.in.bas'
 
-REM READ(A$) -> R
-MAL_READ:
-  GOSUB READ_STR
-  RETURN
+REM READ is inlined in RE
 
 REM QUASIQUOTE(A) -> R
 SUB QUASIQUOTE
   REM pair?
-  IF (Z%(A,0)AND 31)<6 OR (Z%(A,0)AND 31)>7 THEN GOTO QQ_QUOTE
-  IF (Z%(A,1)=0) THEN GOTO QQ_QUOTE
+  GOSUB TYPE_A
+  IF T<6 OR T>7 THEN GOTO QQ_QUOTE
+  IF (Z%(A+1)=0) THEN GOTO QQ_QUOTE
   GOTO QQ_UNQUOTE
 
   QQ_QUOTE:
@@ -30,35 +29,36 @@ SUB QUASIQUOTE
     GOTO QQ_DONE
 
   QQ_UNQUOTE:
-    R=A:GOSUB VAL_R
-    IF (Z%(R,0)AND 31)<>5 THEN GOTO QQ_SPLICE_UNQUOTE
-    IF S$(Z%(R,1))<>"unquote" THEN GOTO QQ_SPLICE_UNQUOTE
+    R=Z%(A+2)
+    IF (Z%(R)AND 31)<>5 THEN GOTO QQ_SPLICE_UNQUOTE
+    IF S$(Z%(R+1))<>"unquote" THEN GOTO QQ_SPLICE_UNQUOTE
       REM [ast[1]]
-      R=Z%(A,1):GOSUB VAL_R
-      Z%(R,0)=Z%(R,0)+32
+      R=Z%(Z%(A+1)+2)
+      GOSUB INC_REF_R
 
       GOTO QQ_DONE
 
   QQ_SPLICE_UNQUOTE:
     GOSUB PUSH_A
     REM rest of cases call quasiquote on ast[1..]
-    A=Z%(A,1):CALL QUASIQUOTE
+    A=Z%(A+1):CALL QUASIQUOTE
     W=R
     GOSUB POP_A
 
     REM set A to ast[0] for last two cases
-    GOSUB VAL_A
+    A=Z%(A+2)
 
     REM pair?
-    IF (Z%(A,0)AND 31)<6 OR (Z%(A,0)AND 31)>7 THEN GOTO QQ_DEFAULT
-    IF (Z%(A,1)=0) THEN GOTO QQ_DEFAULT
+    GOSUB TYPE_A
+    IF T<6 OR T>7 THEN GOTO QQ_DEFAULT
+    IF (Z%(A+1)=0) THEN GOTO QQ_DEFAULT
 
-    B=A:GOSUB VAL_B
-    IF (Z%(B,0)AND 31)<>5 THEN GOTO QQ_DEFAULT
-    IF S$(Z%(B,1))<>"splice-unquote" THEN QQ_DEFAULT
+    B=Z%(A+2)
+    IF (Z%(B)AND 31)<>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):GOSUB VAL_B
+      B=Z%(Z%(A+1)+2)
       B$="concat":T=5:GOSUB STRING:C=R
       A=W:GOSUB LIST3
       REM release inner quasiquoted since outer list takes ownership
@@ -90,20 +90,21 @@ SUB MACROEXPAND
 
   MACROEXPAND_LOOP:
     REM list?
-    IF (Z%(A,0)AND 31)<>6 THEN GOTO MACROEXPAND_DONE
+    GOSUB TYPE_A
+    IF T<>6 THEN GOTO MACROEXPAND_DONE
     REM non-empty?
-    IF Z%(A,1)=0 THEN GOTO MACROEXPAND_DONE
-    B=A:GOSUB VAL_B
+    IF Z%(A+1)=0 THEN GOTO MACROEXPAND_DONE
+    B=Z%(A+2)
     REM symbol? in first position
-    IF (Z%(B,0)AND 31)<>5 THEN GOTO MACROEXPAND_DONE
+    IF (Z%(B)AND 31)<>5 THEN GOTO MACROEXPAND_DONE
     REM defined in environment?
     K=B:CALL ENV_FIND
     IF R=-1 THEN GOTO MACROEXPAND_DONE
     B=R4
     REM macro?
-    IF (Z%(B,0)AND 31)<>11 THEN GOTO MACROEXPAND_DONE
+    IF (Z%(B)AND 31)<>11 THEN GOTO MACROEXPAND_DONE
 
-    F=B:AR=Z%(A,1):CALL APPLY
+    F=B:AR=Z%(A+1):CALL APPLY
     A=R
 
     GOSUB PEEK_Q:AY=Q
@@ -126,13 +127,13 @@ SUB EVAL_AST
 
   IF ER<>-2 THEN GOTO EVAL_AST_RETURN
 
-  T=Z%(A,0)AND 31
+  GOSUB TYPE_A
   IF T=5 THEN GOTO EVAL_AST_SYMBOL
-  IF T>=6 AND T<=8 THEN GOTO EVAL_AST_SEQ
+  IF T>5 AND T<9 THEN GOTO EVAL_AST_SEQ
 
   REM scalar: deref to actual value and inc ref cnt
   R=A
-  Z%(R,0)=Z%(R,0)+32
+  GOSUB INC_REF_R
   GOTO EVAL_AST_RETURN
 
   EVAL_AST_SYMBOL:
@@ -146,22 +147,23 @@ SUB EVAL_AST
 
     EVAL_AST_SEQ_LOOP:
       REM check if we are done evaluating the source sequence
-      IF Z%(A,1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
+      IF Z%(A+1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
 
       REM if we are returning to DO, then skip last element
       REM The EVAL_DO call to EVAL_AST must be call #2 for EVAL_AST to
       REM return early and for TCO to work
       Q=5:GOSUB PEEK_Q_Q
-      IF Q=2 AND Z%(Z%(A,1),1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
+      IF Q=2 AND Z%(Z%(A+1)+1)=0 THEN GOTO EVAL_AST_SEQ_LOOP_DONE
 
       REM call EVAL for each entry
       GOSUB PUSH_A
-      IF T<>8 THEN GOSUB VAL_A
-      IF T=8 THEN A=Z%(A+1,1)
+      IF T<>8 THEN A=Z%(A+2)
+      IF T=8 THEN A=Z%(A+3)
       Q=T:GOSUB PUSH_Q: REM push/save type
       CALL EVAL
       GOSUB POP_Q:T=Q: REM pop/restore type
       GOSUB POP_A
+      M=R
 
       REM if error, release the unattached element
       REM TODO: is R=0 correct?
@@ -169,17 +171,15 @@ SUB EVAL_AST
 
       REM for hash-maps, copy the key (inc ref since we are going to
       REM release it below)
-      IF T=8 THEN M=Z%(A+1,0):Z%(M,0)=Z%(M,0)+32
+      IF T=8 THEN N=M:M=Z%(A+2):Z%(M)=Z%(M)+32
 
-      REM value evaluated above
-      N=R
 
       REM update the return sequence structure
       REM release N (and M if T=8) since seq takes full ownership
       C=1:GOSUB MAP_LOOP_UPDATE
 
       REM process the next sequence entry from source list
-      A=Z%(A,1)
+      A=Z%(A+1)
 
       GOTO EVAL_AST_SEQ_LOOP
     EVAL_AST_SEQ_LOOP_DONE:
@@ -224,13 +224,13 @@ SUB EVAL
     IF R<>1 THEN GOTO EVAL_NOT_LIST
 
     GOSUB EMPTY_Q
-    IF R THEN R=A:Z%(R,0)=Z%(R,0)+32:GOTO EVAL_RETURN
+    IF R THEN R=A:GOSUB INC_REF_R:GOTO EVAL_RETURN
 
-    A0=Z%(A+1,1)
+    A0=Z%(A+2)
 
     REM get symbol in A$
-    IF (Z%(A0,0)AND 31)<>5 THEN A$=""
-    IF (Z%(A0,0)AND 31)=5 THEN A$=S$(Z%(A0,1))
+    IF (Z%(A0)AND 31)<>5 THEN A$=""
+    IF (Z%(A0)AND 31)=5 THEN A$=S$(Z%(A0+1))
 
     IF A$="def!" THEN GOTO EVAL_DEF
     IF A$="let*" THEN GOTO EVAL_LET
@@ -245,14 +245,11 @@ SUB EVAL
     GOTO EVAL_INVOKE
 
     EVAL_GET_A3:
-      R=Z%(Z%(Z%(A,1),1),1)
-      GOSUB VAL_R:A3=R
+      A3=Z%(Z%(Z%(Z%(A+1)+1)+1)+2)
     EVAL_GET_A2:
-      R=Z%(Z%(A,1),1)
-      GOSUB VAL_R:A2=R
+      A2=Z%(Z%(Z%(A+1)+1)+2)
     EVAL_GET_A1:
-      R=Z%(A,1)
-      GOSUB VAL_R:A1=R
+      A1=Z%(Z%(A+1)+2)
       RETURN
 
     EVAL_DEF:
@@ -280,21 +277,21 @@ SUB EVAL
       C=E:GOSUB ENV_NEW
       E=R
       EVAL_LET_LOOP:
-        IF Z%(A1,1)=0 THEN GOTO EVAL_LET_LOOP_DONE
+        IF Z%(A1+1)=0 THEN GOTO EVAL_LET_LOOP_DONE
 
         Q=A1:GOSUB PUSH_Q: REM push A1
         REM eval current A1 odd element
-        A=Z%(A1,1):GOSUB VAL_A:CALL EVAL
+        A=Z%(Z%(A1+1)+2):CALL EVAL
         GOSUB POP_Q:A1=Q: REM pop A1
 
         IF ER<>-2 THEN GOTO EVAL_LET_LOOP_DONE
 
-        REM set environment: even A1 key to odd A1 eval'd above
-        K=Z%(A1+1,1):C=R:GOSUB ENV_SET
+        REM set key/value in the environment
+        K=Z%(A1+2):C=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)
+        A1=Z%(Z%(A1+1)+1)
         GOTO EVAL_LET_LOOP
 
       EVAL_LET_LOOP_DONE:
@@ -308,7 +305,7 @@ SUB EVAL
         A=A2:GOTO EVAL_TCO_RECUR: REM TCO loop
 
     EVAL_DO:
-      A=Z%(A,1): REM rest
+      A=Z%(A+1): REM rest
       GOSUB PUSH_A: REM push/save A
 
       REM this must be EVAL_AST call #2 for EVAL_AST to return early
@@ -329,12 +326,12 @@ SUB EVAL
       GOTO EVAL_TCO_RECUR: REM TCO loop
 
     EVAL_QUOTE:
-      R=Z%(A,1):GOSUB VAL_R
-      Z%(R,0)=Z%(R,0)+32
+      R=Z%(Z%(A+1)+2)
+      GOSUB INC_REF_R
       GOTO EVAL_RETURN
 
     EVAL_QUASIQUOTE:
-      R=Z%(A,1):GOSUB VAL_R
+      R=Z%(Z%(A+1)+2)
       A=R:CALL QUASIQUOTE
       A=R
       REM add quasiquote result to pending release queue to free when
@@ -352,7 +349,7 @@ SUB EVAL
       GOSUB POP_Q:A1=Q: REM pop A1
 
       REM change function to macro
-      Z%(R,0)=Z%(R,0)+1
+      Z%(R)=Z%(R)+1
 
       REM set A1 in env to A2
       K=A1:C=R:GOSUB ENV_SET
@@ -360,33 +357,34 @@ SUB EVAL
 
     EVAL_MACROEXPAND:
       REM PRINT "macroexpand"
-      R=Z%(A,1):GOSUB VAL_R
+      R=Z%(Z%(A+1)+2)
       A=R:CALL MACROEXPAND
       R=A
 
       REM since we are returning it unevaluated, inc the ref cnt
-      Z%(R,0)=Z%(R,0)+32
+      GOSUB INC_REF_R
       GOTO EVAL_RETURN
 
     EVAL_TRY:
       REM PRINT "try*"
-      GOSUB EVAL_GET_A1: REM set A1, A2, and A3
+      GOSUB EVAL_GET_A1: REM set A1
 
       GOSUB PUSH_A: REM push/save A
       A=A1:CALL EVAL: REM eval A1
       GOSUB POP_A: REM pop/restore A
 
-      REM if there is not error or catch block then return
-      IF ER=-2 OR Z%(A,1)=0 THEN GOTO EVAL_RETURN
+      GOSUB EVAL_GET_A2: REM set A1 and A2
+
+      REM if there is no error or catch block then return
+      IF ER=-2 OR A2=0 THEN GOTO EVAL_RETURN
 
       REM create environment for the catch block eval
       C=E:GOSUB ENV_NEW:E=R
 
-      GOSUB EVAL_GET_A2: REM set A1 and A2
       A=A2:GOSUB EVAL_GET_A2: REM set A1 and A2 from catch block
 
       REM create object for ER=-1 type raw string errors
-      IF ER=-1 THEN B$=E$:T=4:GOSUB STRING:ER=R:Z%(R,0)=Z%(R,0)+32
+      IF ER=-1 THEN B$=E$:T=4:GOSUB STRING:ER=R:GOSUB INC_REF_R
 
       REM bind the catch symbol to the error object
       K=A1:C=ER:GOSUB ENV_SET
@@ -404,7 +402,7 @@ SUB EVAL
       GOSUB PUSH_A: REM push/save A
       A=A1:CALL EVAL
       GOSUB POP_A: REM pop/restore A
-      IF (R=0) OR (R=1) THEN GOTO EVAL_IF_FALSE
+      IF (R=0) OR (R=2) THEN GOTO EVAL_IF_FALSE
 
       EVAL_IF_TRUE:
         AY=R:GOSUB RELEASE
@@ -414,13 +412,13 @@ SUB EVAL
         AY=R:GOSUB RELEASE
         REM if no false case (A3), return nil
         GOSUB COUNT
-        IF R<4 THEN R=0:Z%(R,0)=Z%(R,0)+32:GOTO EVAL_RETURN
+        IF R<4 THEN R=0:GOSUB INC_REF_R: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:B=A1:GOSUB MAL_FUNCTION
+      T=10:L=A2:M=A1:N=E:GOSUB ALLOC: REM mal function
       GOTO EVAL_RETURN
 
     EVAL_INVOKE:
@@ -432,14 +430,14 @@ SUB EVAL
       REM push f/args for release after call
       GOSUB PUSH_R
 
-      AR=Z%(R,1): REM rest
-      GOSUB VAL_R:F=R
+      AR=Z%(R+1): REM rest
+      F=Z%(R+2)
 
       REM if metadata, get the actual object
-      IF (Z%(F,0)AND 31)>=16 THEN F=Z%(F,1)
+      GOSUB TYPE_F
+      IF T=14 THEN F=Z%(F+1):GOSUB TYPE_F
 
-      IF (Z%(F,0)AND 31)=9 THEN GOTO EVAL_DO_FUNCTION
-      IF (Z%(F,0)AND 31)=10 THEN GOTO EVAL_DO_MAL_FUNCTION
+      ON T-8 GOTO EVAL_DO_FUNCTION,EVAL_DO_MAL_FUNCTION,EVAL_DO_MAL_FUNCTION
 
       REM if error, pop and return f/args for release by caller
       GOSUB POP_R
@@ -447,9 +445,9 @@ SUB EVAL
 
       EVAL_DO_FUNCTION:
         REM regular function
-        IF Z%(F,1)<60 THEN GOSUB DO_FUNCTION:GOTO EVAL_DO_FUNCTION_SKIP
+        IF Z%(F+1)<60 THEN GOSUB DO_FUNCTION:GOTO EVAL_DO_FUNCTION_SKIP
         REM for recur functions (apply, map, swap!), use GOTO
-        IF Z%(F,1)>60 THEN CALL DO_TCO_FUNCTION
+        IF Z%(F+1)>60 THEN CALL DO_TCO_FUNCTION
         EVAL_DO_FUNCTION_SKIP:
 
         REM pop and release f/args
@@ -460,8 +458,8 @@ SUB EVAL
       EVAL_DO_MAL_FUNCTION:
         Q=E:GOSUB PUSH_Q: REM save the current environment for release
 
-        REM create new environ using env stored with function
-        C=Z%(F+1,1):A=Z%(F+1,0):B=AR:GOSUB ENV_NEW_BINDS
+        REM create new environ using env and params stored in function
+        C=Z%(F+3):A=Z%(F+2):B=AR:GOSUB ENV_NEW_BINDS
 
         REM release previous env if it is not the top one on the
         REM stack (X%(X-2)) because our new env refers to it and
@@ -471,7 +469,7 @@ SUB EVAL
         IF AY<>Q THEN GOSUB RELEASE
 
         REM claim the AST before releasing the list containing it
-        A=Z%(F,1):Z%(A,0)=Z%(A,0)+32
+        A=Z%(F+1):Z%(A)=Z%(A)+32
         REM add AST to pending release queue to free as soon as EVAL
         REM actually returns (LV+1)
         LV=LV+1:GOSUB PEND_A_LV:LV=LV-1
@@ -506,17 +504,15 @@ SUB EVAL
 
 END SUB
 
-REM PRINT(A) -> R$
-MAL_PRINT:
-  AZ=A:B=1:GOSUB PR_STR
-  RETURN
+REM PRINT is inlined in REP
+
 
 REM RE(A$) -> R
 REM Assume D has repl_env
 REM caller must release result
 RE:
   R1=-1
-  GOSUB MAL_READ
+  GOSUB READ_STR: REM inlined MAL_READ
   R1=R
   IF ER<>-2 THEN GOTO RE_DONE
 
@@ -536,7 +532,7 @@ SUB REP
   R2=R
   IF ER<>-2 THEN GOTO REP_DONE
 
-  A=R:GOSUB MAL_PRINT
+  AZ=R:B=1:GOSUB PR_STR: REM MAL_PRINT
 
   REP_DONE:
     REM Release memory from MAL_READ and EVAL
@@ -573,11 +569,11 @@ MAIN:
   A$=A$+" forms to cond"+CHR$(34)+")) (cons 'cond (rest (rest xs)))))))"
   GOSUB RE:AY=R:GOSUB RELEASE
 
-  A$="(def! *gensym-counter* (atom 0))"
+  A$="(def! inc (fn* [x] (+ x 1)))"
   GOSUB RE:AY=R:GOSUB RELEASE
 
-  A$="(def! gensym (fn* [] (symbol (str "+CHR$(34)+"G__"+CHR$(34)
-  A$=A$+" (swap! *gensym-counter* (fn* [x] (+ 1 x)))))))"
+  A$="(def! gensym (let* [counter (atom 0)] (fn* [] (symbol (str "+CHR$(34)
+  A$=A$+"G__"+CHR$(34)+" (swap! counter inc))))))"
   GOSUB RE:AY=R:GOSUB RELEASE
 
   A$="(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs)"
@@ -589,6 +585,8 @@ MAIN:
   A$="(def! -*ARGS*- (load-file "+CHR$(34)+".args.mal"+CHR$(34)+"))"
   GOSUB RE:AY=R:GOSUB RELEASE
 
+  IF ER>-2 THEN GOSUB PRINT_ERROR:END
+
   REM set the argument list
   A$="(def! *ARGV* (rest -*ARGS*-))"
   GOSUB RE:AY=R:GOSUB RELEASE
@@ -597,12 +595,14 @@ MAIN:
   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
+  IF R<16 THEN GOTO REPL
+
+  REM if there is an argument, then run it as a program
 
   RUN_PROG:
+    REM free up first arg because we get it again
+    AY=R:GOSUB RELEASE
     REM run a single mal program and exit
     A$="(load-file (first -*ARGS*-))"
     GOSUB RE
@@ -613,7 +613,7 @@ MAIN:
     REM print the REPL startup header
     REM save memory by printing this directly
     #cbm PRINT "Mal [C64 BASIC]"
-    #qbasic PRINT "Mal [C64 QBasic]"
+    #qbasic PRINT "Mal [QBasic]"
 
   REPL_LOOP:
     A$="user> ":GOSUB READLINE: REM call input parser
@@ -627,8 +627,13 @@ MAIN:
     GOTO REPL_LOOP
 
   QUIT:
-    GOSUB PR_MEMORY_SUMMARY_SMALL
-    END
+    REM GOSUB PR_MEMORY_SUMMARY_SMALL
+    REM GOSUB PR_MEMORY_MAP
+    REM P1=0:P2=ZI:GOSUB PR_MEMORY
+    REM P1=D:GOSUB PR_OBJECT
+    REM P1=ZK:GOSUB PR_OBJECT
+    #cbm END
+    #qbasic SYSTEM
 
   PRINT_ERROR:
     REM if the error is an object, then print and free it