Better handling of booleans.
[scheme.forth.jl.git] / scheme.4th
index 5fe8b3e..0bbfbf5 100644 (file)
 vocabulary scheme
 scheme definitions
 
-\ Cons cell memory
-1000 constant memsize
-create car memsize allot
-create cdr memsize allot
-create types memsize allot
+include term-colours.4th
 
-0 constant symbol-type
-1 constant int-type
-2 constant list-type
-3 constant bool-type
+0 constant number-type
+1 constant boolean-type
+2 constant character-type
+: istype? ( obj -- obj b )
+    over = ;
 
-variable nextfree
-0 nextfree !
+\ ---- Read ----
+
+variable parse-idx
+variable stored-parse-idx
+create parse-str 161 allot
+variable parse-str-span
 
-: stack
-    create here 1+ , allot ;
 
+create parse-idx-stack 10 allot 
+variable parse-idx-sp
+parse-idx-stack parse-idx-sp !
 
-: push ( st v -- )
-    over @ !
-    1 swap +!
+: push-parse-idx
+    parse-idx @ parse-idx-sp @ !
+    1 parse-idx-sp +!
 ;
 
-: pop ( st -- v )
-    dup @       ( s0 sp )
-    1-          ( so sp' )
+: pop-parse-idx
+    parse-idx-sp @ parse-idx-stack <= abort" Parse index stack underflow."
 
-    2dup = abort" Stack underflow."
-    
-    dup @       ( s0 sp' v )
-    -rot swap   ( v sp' s0 )
-    !
-;
+    1 parse-idx-sp -!
 
-100 stack parse-stack 
-variable parse-idx
-variable parse-str
+    parse-idx-sp @ @ parse-idx ! ;
 
 
-: inc-parse-idx parse-idx +! ;
-: dec-parse-idx parse-idx -! ;
+: append-newline
+    '\n' parse-str parse-str-span @ + !
+    1 parse-str-span +! ;
 
-: ?charavailable ( -- bool )
-    parse-str @ @ parse-idx @ >
-;
+: empty-parse-str
+    0 parse-str-span !
+    0 parse-idx ! ;
+
+: getline
+    parse-str 160 expect cr
+    span @ parse-str-span !
+    append-newline
+    0 parse-idx ! ;
+
+: inc-parse-idx
+    1 parse-idx +! ;
+
+: dec-parse-idx
+    1 parse-idx -! ;
+
+: charavailable? ( -- bool )
+    parse-str-span @ parse-idx @ > ;
 
 : nextchar ( -- char )
-    ?charavailable if
-        parse-str @ 1+ parse-idx @ + @
-    else
-        0
-    then
-;
+    charavailable? false = if getline then
+    parse-str parse-idx @ + @ ;
 
-: ?whitespace ( -- bool )
+: whitespace? ( -- bool )
     nextchar BL = 
-    nextchar '\n' = or
-;
+    nextchar '\n' = or ;
 
-: ?delim ( -- bool )
-    ?whitespace
+: eof? ( -- bool )
+    nextchar 4 = ;
+
+: delim? ( -- bool )
+    whitespace?
     nextchar [char] ( = or
     nextchar [char] ) = or
 ;
 
 : eatspaces
     begin
-        ?whitespace
+        whitespace?
     while
-            1 parse-idx +!
+        inc-parse-idx
     repeat
 ;
 
-: parsebool
+: digit? ( -- bool )
+    nextchar [char] 0 >=
+    nextchar [char] 9 <=
+    and ;
 
-    nextchar emit cr
-    trace
+: minus? ( -- bool )
+    nextchar [char] - = ;
 
-    false
-    nextchar [char] # <> if exit then
+: number? ( -- bool )
+    digit? minus? or false = if
+        false
+        exit
+    then
 
-    1 inc-parse-idx
+    push-parse-idx
+    inc-parse-idx
 
-    nextchar dup [char] t = swap [char] f = or
-    not if
-        1 dec-parse-idx
-        exit
+    begin digit? while
+        inc-parse-idx
+    repeat
+
+    delim? if
+        pop-parse-idx
+        true
+    else
+        pop-parse-idx
+        false
     then
+;
 
-    1 inc-parse-idx
+: boolean? ( -- bool )
+    nextchar [char] # <> if false exit then
 
-    ?delim not if
-        2 dec-parse-idx
-        exit
+    push-parse-idx
+    inc-parse-idx
+
+    nextchar [char] t <>
+    nextchar [char] f <>
+    and if pop-parse-idx false exit then
+
+    inc-parse-idx
+    delim? if
+        pop-parse-idx
+        true
+    else
+        pop-parse-idx
+        false
     then
 ;
 
-: parsetoken
+: str-equiv? ( str -- bool )
+    push-parse-idx
 
-    eatspaces
+    true
 
-    \ Parens
+    swap dup rot + swap
+    do
+        i @ nextchar <> if
+            drop false
+            leave
+        then
 
-    nextchar [char] ( = if
-        \ todo
-        exit
+        inc-parse-idx
+    loop
+
+    delim? <> if drop false then
+
+    pop-parse-idx
+;
+
+: character? ( -- bool )
+    nextchar [char] # <> if false exit then
+
+    push-parse-idx
+    inc-parse-idx
+
+    nextchar [char] \ <> if pop-parse-idx false exit then
+
+    inc-parse-idx
+
+    S" newline" str-equiv? if true exit then
+    S" space" str-equiv? if true exit then
+    S" tab" str-equiv? if true exit then
+
+    charavailable? false = if pop-parse-idx false exit then
+
+    pop-parse-idx true
+;
+
+: readnum ( -- num-atom )
+    minus? dup if
+        inc-parse-idx
+    then
+
+    0
+
+    begin digit? while
+        10 * nextchar [char] 0 - +
+        inc-parse-idx
+    repeat
+
+    swap if negate then
+
+    number-type
+;
+
+: readbool ( -- bool-atom )
+    inc-parse-idx
+    
+    nextchar [char] f = if
+        false
+    else
+        true
     then
 
-    nextchar [char] ) = if
-        \ todo
+    inc-parse-idx
+
+    boolean-type
+;
+
+\ Parse a scheme expression
+: read ( -- obj )
+
+    eatspaces
+
+    number? if
+        readnum
         exit
     then
 
-    parsebool if
+    boolean? if
+        readbool
         exit
-    exit
-;
+    then
 
-\ Parse a counted string into a scheme expression
-: parseexp ( straddr n -- exp )
-    0 parse-idx !
+    eof? if
+        bold fg blue ." Moriturus te saluto." reset-term ."  ok" cr
+        quit
+    then
 
-    begin
-        parsetoken
-    nextchar 0 =
-    until
+    bold fg red ." Error parsing string starting at character '"
+    nextchar emit
+    ." '. Aborting." reset-term cr
+    abort
 ;
 
-\ ---- REPL ----
+\ ---- Eval ----
 
-: escape 27 emit [char] [ emit ;
-: escape-end [char] m emit ;
+: self-evaluating? ( obj -- obj bool )
+    number-type istype? if true exit then
+    boolean-type istype? if true exit then
+    false ;
 
-: set-term-colour
-    escape [char] 3 emit [char] 0 + emit escape-end
-;
+: eval
+    self-evaluating? if
+        exit
+    then
 
-: reset-term
-    escape [char] 0 escape-end
+    bold fg red ." Error evaluating expression - unrecognized type. Aborting." reset-term cr
+    abort
 ;
 
-: colour
-    create ,
-does>
-    @ set-term-colour
+\ ---- Print ----
+
+: printnum ( numobj -- ) drop . ;
+
+: printbool ( numobj -- )
+    drop if
+        ." #t"
+    else
+        ." #f"
+    then
 ;
 
-1 colour red
-2 colour green
-3 colour yellow
-4 colour blue
-5 colour magenta
-6 colour cyan
-7 colour white
+: print ( obj -- )
+    ." ; "
+    number-type istype? if printnum exit then
+    boolean-type istype? if printbool exit then
+;
 
-create repl-buffer 161 allot
+\ ---- REPL ----
 
 : repl
-    repl-buffer parse-str !
+    cr ." Welcome to scheme.forth.jl!" cr
+       ." Use Ctrl-D to exit." cr
 
-    cr
+    empty-parse-str
 
     begin
-        green ." => " white
-
-        repl-buffer 1+ 160 expect cr
-        span @ repl-buffer !
-
-        parseexp
-        \ eval
+        cr bold fg green ." > " reset-term
+        read
+        eval
+        fg cyan print reset-term
     again
 ;