Implemented let*
[scheme.forth.jl.git] / src / scheme.4th
index 165d88f..d0bca19 100644 (file)
@@ -4,6 +4,7 @@ scheme definitions
 include term-colours.4th
 include defer-is.4th
 include catch-throw.4th
+include integer.4th
 include float.4th
 
 include debugging.4th
@@ -24,6 +25,8 @@ variable nexttype
     does> @ ;
 
 make-type fixnum-type
+make-type flonum-type
+make-type ratnum-type
 make-type boolean-type
 make-type character-type
 make-type string-type
@@ -33,7 +36,7 @@ make-type pair-type
 make-type symbol-type
 make-type primitive-proc-type
 make-type compound-proc-type
-make-type fileport-type
+make-type port-type
 : istype? ( obj type -- obj bool )
     over = ;
 
@@ -48,33 +51,21 @@ variable nextexception
     1 nextexception +!
     does> @ ;
 
-make-exception recoverable-exception
-make-exception unrecoverable-exception
-
-: display-exception-msg ( addr count -- )
+: except-message:
     bold fg red
     ." Exception: "
-    type
-    reset-term ;
-
-: throw" immediate 
-    [compile] s"
-
-    ['] rot , ['] dup ,
+;
 
-    [compile] if
-        ['] -rot ,
-        ['] display-exception-msg ,
-    [compile] then
+make-exception recoverable-exception
+make-exception unrecoverable-exception
 
-    ['] throw ,
-;
+: throw reset-term throw ;
 
 \ }}}
 
 \ ---- List-structured memory ---- {{{
 
-10000 constant scheme-memsize
+20000 constant scheme-memsize
 
 create car-cells scheme-memsize allot
 create car-type-cells scheme-memsize allot
@@ -100,7 +91,7 @@ variable nextfree
     then
 
     nextfree @ scheme-memsize >= if
-        unrecoverable-exception throw s" Out of memory!"
+        except-message: ." Out of memory!" unrecoverable-exception throw
     then
 ;
 
@@ -172,37 +163,6 @@ variable nextfree
 
 \ }}}
 
-\ ---- Port I/O ----  {{{
-
-: fileport>fid ( fileport -- fid )
-    drop ;
-
-: fid>fileport ( fid -- fileport )
-    fileport-type ;
-
-: open-input-file ( addr n -- fileport )
-    r/o open-file drop fid>fileport
-;
-
-: close-port ( fileport -- )
-    fileport>fid close-file drop
-;
-
-objvar console-i/o-port
-0 fileport-type console-i/o-port obj!
-
-objvar current-input-port
-console-i/o-port obj@ current-input-port obj!
-
-: read-port ( fileport -- obj )
-    current-input-port obj!
-    read ;
-
-: read-console ( -- obj )
-    console-i/o-port obj@ read-port ;
-
-\ }}}
-
 \ ---- Pre-defined symbols ---- {{{
 
 objvar symbol-table
@@ -302,6 +262,125 @@ create-symbol if                if-symbol
 create-symbol lambda            lambda-symbol
 create-symbol λ                 λ-symbol
 create-symbol begin             begin-symbol
+create-symbol eof               eof-symbol
+
+\ Symbol to be bound to welcome message procedure by library
+create-symbol welcome           welcome-symbol
+
+\ }}}
+
+\ ---- Port I/O ----  {{{
+
+( Ports are pairs with the fid in the car and the peek buffer in the cdr. )
+
+: fileport>fid ( fileport -- fid )
+    drop pair-type car drop ;
+
+: get-last-peek ( fileport -- char/nil )
+    drop pair-type cdr ;
+
+: set-last-peek ( char/nil fileport -- )
+    drop pair-type set-cdr!
+;
+
+: fid>fileport ( fid -- fileport )
+    fixnum-type nil cons drop port-type ;
+
+: open-input-file ( addr n -- fileport )
+    r/o open-file drop fid>fileport
+;
+
+: close-port ( fileport -- )
+    fileport>fid close-file drop
+;
+
+objvar console-i/o-port
+0 fixnum-type nil cons drop port-type console-i/o-port obj!
+
+objvar current-input-port
+console-i/o-port obj@ current-input-port obj!
+
+: read-char ( port -- char ) 
+    2dup get-last-peek nil? if
+        2drop
+        2dup console-i/o-port obj@ objeq? if
+            2drop
+            key character-type
+        else
+            fileport>fid pad 1 rot read-file 0= if
+                eof-symbol
+            else
+                pad @ character-type
+            then
+        then
+    else
+        nil 2rot set-cdr!
+    then
+;
+
+: peek-char ( port -- char )
+    2dup get-last-peek nil? if
+        2drop 2dup read-char
+        2dup 2rot set-last-peek
+    else
+        2swap 2drop
+    then
+;
+
+variable read-line-buffer-span
+variable read-line-buffer-offset
+
+( Hack to save original read-line while we transition to new one. )
+: orig-read-line immediate
+    ['] read-line , ;
+
+: read-line ( port -- string )
+
+    2dup get-last-peek
+    nil? if
+        2drop
+        0 read-line-buffer-offset !
+    else
+        2over nil 2swap set-last-peek
+        2dup drop '\n' = if
+            2drop nil nil cons exit
+        else
+            drop pad !
+            1 read-line-buffer-offset !
+        then
+    then
+
+    2dup console-i/o-port obj@ objeq? if
+        2drop
+        pad read-line-buffer-offset @ + 200 expect cr
+        span @ read-line-buffer-offset @ + read-line-buffer-span !
+    else
+        pad read-line-buffer-offset @ + 200 2over fileport>fid orig-read-line
+        drop swap read-line-buffer-offset @ + read-line-buffer-span !
+    then
+
+    nil
+    
+    begin
+        read-line-buffer-span @ 0>
+    while
+        pad read-line-buffer-span @ 1- + @ character-type 2swap cons
+        -1 read-line-buffer-span +!
+    repeat
+
+    nil? if
+        nil cons drop string-type
+    else
+        drop string-type
+    then
+;
+
+: read-port ( fileport -- obj )
+    current-input-port obj!
+    read ;
+
+: read-console ( -- obj )
+    console-i/o-port obj@ read-port ;
 
 \ }}}
 
@@ -380,24 +459,30 @@ objvar vals
 hide vars
 hide vals
 
+objvar var
+
 : lookup-var ( var env -- val )
+    2over var obj!
     get-vars-vals if
         2swap 2drop car
     else
-        recoverable-exception throw" Tried to read unbound variable."
+        except-message: ." tried to read unbound variable '" var obj@ print ." '." recoverable-exception  throw
     then
 ;
 
 : set-var ( var val env -- )
     >R >R 2swap R> R> ( val var env )
+    2over var obj!
     get-vars-vals if
         2swap 2drop ( val vals )
         set-car!
     else
-        recoverable-exception throw" Tried to set unbound variable."
+        except-message: ." tried to set unbound variable '" var obj@ print ." '." recoverable-exception throw
     then
 ;
 
+hide var
+
 objvar env
 
 : define-var ( var val env -- )
@@ -445,11 +530,11 @@ global-env obj!
 : ensure-arg-count ( args n -- )
     dup 0= if
         drop nil objeq? false = if
-            recoverable-exception throw" Too many arguments for primitive procedure."
+            except-message: ." Too many arguments for primitive procedure." recoverable-exception throw
         then
     else
         -rot nil? if
-            recoverable-exception throw" Too few arguments for primitive procedure."
+            except-message: ." Too few arguments for primitive procedure." recoverable-exception  throw
         then
         
         cdr rot 1- recurse
@@ -459,17 +544,17 @@ global-env obj!
 : ensure-arg-type-and-count ( tn tn-1 ... t2 t1 args n -- )
     dup 0= if
         drop nil objeq? false = if
-            recoverable-exception throw" Too many arguments for primitive procedure."
+            except-message: ." Too many arguments for primitive procedure." recoverable-exception throw
         then
     else
         -rot nil? if
-            recoverable-exception throw" Too few arguments for primitive procedure."
+            except-message: ." Too few arguments for primitive procedure." recoverable-exception throw
         then
 
         2dup cdr 2swap car ( ... t1 n args' arg1 )
         2rot 1- swap 2swap rot ( ... args' n-1 arg1 t1 )
         istype? false = if
-            recoverable-exception throw" Incorrect type for primitive procedure."
+            except-message: ." Incorrect type for primitive procedure." recoverable-exception throw
         then
 
         2drop recurse
@@ -539,7 +624,7 @@ global-env obj!
 
 : ensure-arg-type ( arg type -- arg )
     istype? false = if
-        recoverable-exception throw" Incorrect argument type for primitive procedure."
+        except-message: ." Incorrect argument type for primitive procedure." recoverable-exception throw
     then
 ;
 
@@ -553,6 +638,12 @@ objvar macro-table
 ( Look up macro in macro table. Returns nil if
   no macro is found. )
 : lookup-macro ( name_symbol -- proc )
+
+    symbol-type istype? invert if
+        \ Early exit if argument is not a symbol
+        2drop nil exit
+    then
+    
     macro-table obj@
 
     begin
@@ -642,7 +733,7 @@ parse-idx-stack parse-idx-sp !
         parse-str 160 expect cr
         span @ parse-str-span !
     else
-        parse-str 160 current-input-port obj@ fileport>fid read-line
+        parse-str 160 current-input-port obj@ fileport>fid orig-read-line
         drop swap parse-str-span !
 
         parse-str-span @ 0= and if append-eof then
@@ -785,6 +876,42 @@ parse-idx-stack parse-idx-sp !
     pop-parse-idx
 ;
 
+: ratnum? ( -- bool )
+    push-parse-idx
+
+    minus? plus? or if
+        inc-parse-idx
+    then
+
+    digit? invert if
+        pop-parse-idx false exit
+    else
+        inc-parse-idx
+    then
+
+    begin digit? while
+        inc-parse-idx
+    repeat
+
+    [char] / nextchar <> if
+        pop-parse-idx false exit
+    else
+        inc-parse-idx
+    then
+
+    digit? invert if
+        pop-parse-idx false exit
+    else
+        inc-parse-idx
+    then
+
+    begin digit? while
+        inc-parse-idx
+    repeat
+
+    delim? pop-parse-idx
+;
+
 : boolean? ( -- bool )
     nextchar [char] # <> if false exit then
 
@@ -852,7 +979,7 @@ parse-idx-stack parse-idx-sp !
 : string? ( -- bool )
     nextchar [char] " = ;
 
-: readfixnum ( -- num-atom )
+: readfixnum ( -- fixnum )
     plus? minus? or if
         minus?
         inc-parse-idx
@@ -872,6 +999,56 @@ parse-idx-stack parse-idx-sp !
     fixnum-type
 ;
 
+: readflonum ( -- flonum )
+    readfixnum drop
+    dup 0< swap abs i->f
+
+    [char] . nextchar = if
+        inc-parse-idx
+
+        10.0 ( f exp )
+
+        begin digit? while
+            nextchar [char] 0 - i->f ( f exp d )
+            over f/ rot f+ ( exp f' )
+            swap 10.0 f* ( f' exp' )
+            inc-parse-idx
+        repeat
+
+        drop
+    then
+
+    [char] e nextchar = [char] E nextchar = or if
+        inc-parse-idx
+        10.0
+        readfixnum drop i->f
+        f^ f*
+    then
+
+    swap if
+        -1.0 f*
+    then
+
+    flonum-type
+;
+
+: make-rational ( fixnum fixnum -- ratnum|fixnum )
+    drop swap drop
+    simplify
+
+    dup 1 = if
+        drop fixnum-type
+    else
+        fixnum-type swap fixnum-type
+        cons drop ratnum-type
+    then
+;
+
+: readratnum ( -- ratnum )
+    readfixnum inc-parse-idx readfixnum
+    make-rational
+;
+
 : readbool ( -- bool-obj )
     inc-parse-idx
     
@@ -900,35 +1077,51 @@ parse-idx-stack parse-idx-sp !
 ;
 
 : readstring ( -- charlist )
-    nextchar [char] " = if
-        inc-parse-idx
 
-        delim? false = if
-            bold fg red
-            ." No delimiter following right double quote. Aborting." cr
-            reset-term abort
+    nil nil
+
+    begin
+        nextchar [char] " <>
+    while
+        nextchar [char] \ = if
+            inc-parse-idx
+            nextchar case
+                [char] n of '\n' endof
+                [char] " of [char] " endof
+                [char] \
+            endcase
+        else
+            nextchar
         then
+        inc-parse-idx character-type
+        nil cons
 
-        dec-parse-idx
+        ( firstchar prevchar thischar )
 
-        0 nil-type exit
-    then
+        2swap nil? if
+            2drop 2swap 2drop 2dup  ( thischar thischar )
+        else
+            ( firstchar thischar prevchar )
+            2over 2swap  set-cdr! ( firstchar thischar )
+        then
+    repeat
 
-    nextchar [char] \ = if
-        inc-parse-idx
-        nextchar case
-            [char] n of '\n' endof
-            [char] " of [char] " endof
-            [char] \
-        endcase
-    else
-        nextchar
+    \ Discard previous character
+    2drop
+
+    inc-parse-idx
+    delim? false = if
+        bold fg red
+        ." No delimiter following right double quote. Aborting." cr
+        reset-term abort
     then
-    inc-parse-idx character-type
 
-    recurse
+    dec-parse-idx
 
-    cons
+    nil? if
+        nil cons
+    then
+    drop string-type
 ;
 
 : readsymbol ( -- charlist )
@@ -993,6 +1186,16 @@ parse-idx-stack parse-idx-sp !
         exit
     then
 
+    flonum? if
+        readflonum
+        exit
+    then
+
+    ratnum? if
+        readratnum
+        exit
+    then
+
     boolean? if
         readbool
         exit
@@ -1007,7 +1210,6 @@ parse-idx-stack parse-idx-sp !
         inc-parse-idx
 
         readstring
-        drop string-type
 
         nextchar [char] " <> if
             bold red ." Missing closing double-quote." reset-term cr
@@ -1082,6 +1284,8 @@ parse-idx-stack parse-idx-sp !
 : self-evaluating? ( obj -- obj bool )
     boolean-type istype? if true exit then
     fixnum-type istype? if true exit then
+    flonum-type istype? if true exit then
+    ratnum-type istype? if true exit then
     character-type istype? if true exit then
     string-type istype? if true exit then
     nil-type istype? if true exit then
@@ -1117,12 +1321,12 @@ parse-idx-stack parse-idx-sp !
     cdr ( env args )
 
     nil? if
-        recoverable-exception throw" no arguments to unquote."
+        except-message: ." no arguments to unquote." recoverable-exception throw
     then
 
     2dup cdr
     nil? false = if
-        recoverable-exception throw" too many arguments to unquote."
+        except-message: ." too many arguments to unquote." recoverable-exception throw
     then
 
     2drop car 2swap eval
@@ -1180,12 +1384,12 @@ defer eval-quasiquote-item
     2swap cdr ( env args )
 
     nil? if
-        recoverable-exception throw" no arguments to quasiquote."
+        except-message: ." no arguments to quasiquote." recoverable-exception throw
     then
 
     2dup cdr ( env args args-cdr )
     nil? false = if
-        recoverable-exception throw" too many arguments to quasiquote."
+        except-message: ." too many arguments to quasiquote." recoverable-exception throw
     then
 
     2drop car ( env arg )
@@ -1396,7 +1600,7 @@ hide env
 : flatten-proc-args ( argvals argnames -- argvals' argnames' )
     nil? if
         2over nil? false = if
-            recoverable-exception throw" Too many arguments for compound procedure."
+            except-message: ." Too many arguments for compound procedure." recoverable-exception throw
         else
             2drop
         then
@@ -1413,7 +1617,7 @@ hide env
 
     2over
     nil? if
-        recoverable-exception throw" Too few arguments for compound procedure."
+        except-message: ." Too few arguments for compound procedure." recoverable-exception throw
     else
         cdr
     then
@@ -1448,7 +1652,7 @@ hide env
                 R> drop ['] eval goto-deferred  \ Tail call optimization
             endof
 
-            recoverable-exception throw" Object not applicable."
+            except-message: ." object '" drop print ." ' not applicable." recoverable-exception throw
         endcase
 ;
 
@@ -1469,6 +1673,13 @@ hide env
 :noname ( obj env -- result )
     2swap
 
+    \ --- DEBUG ---
+    ( 
+      fg yellow ." Evaluating: " bold 2dup print reset-term
+      space fg green ." PS: " bold depth . reset-term
+      space fg blue  ." RS: " bold RSP@ RSP0 - . reset-term cr
+    )
+
     self-evaluating? if
         2swap 2drop
         exit
@@ -1564,16 +1775,23 @@ hide env
         then
     then
 
-    recoverable-exception throw" Tried to evaluate object with unknown type."
+    except-message: ." tried to evaluate object with unknown type." recoverable-exception throw
 ; is eval
 
 \ }}}
 
 \ ---- Print ---- {{{
 
-: printfixnum ( fixnumobj -- ) drop 0 .R ;
+: printfixnum ( fixnum -- ) drop 0 .R ;
+
+: printflonum ( flonum -- ) drop f. ;
 
-: printbool ( numobj -- )
+: printratnum ( ratnum -- )
+    drop pair-type 2dup
+    car print ." /" cdr print
+;
+
+: printbool ( bool -- )
     drop if
         ." #t"
     else
@@ -1643,6 +1861,8 @@ hide env
 
 :noname ( obj -- )
     fixnum-type istype? if printfixnum exit then
+    flonum-type istype? if printflonum exit then
+    ratnum-type istype? if printratnum exit then
     boolean-type istype? if printbool exit then
     character-type istype? if printchar exit then
     string-type istype? if printstring exit then
@@ -1652,8 +1872,9 @@ hide env
     primitive-proc-type istype? if printprim exit then
     compound-proc-type istype? if printcomp exit then
     none-type istype? if printnone exit then
+    port-type istype? if printport exit then
 
-    recoverable-exception throw" Tried to print object with unknown type."
+    except-message: ." tried to print object with unknown type." recoverable-exception throw
 ; is print
 
 \ }}}
@@ -1680,6 +1901,7 @@ variable gc-stack-depth
     string-type istype? if true exit then
     symbol-type istype? if true exit then
     compound-proc-type istype? if true exit then
+    port-type istype? if true exit then
 
     false
 ;
@@ -1746,6 +1968,7 @@ variable gc-stack-depth
 
     symbol-table obj@ gc-mark-obj
     macro-table obj@ gc-mark-obj
+    console-i/o-port obj@ gc-mark-obj
     global-env obj@ gc-mark-obj
 
     depth gc-stack-depth @ do
@@ -1764,24 +1987,6 @@ variable gc-stack-depth
 
 \ ---- Loading files ---- {{{
 
-: charlist>cstr ( charlist addr -- n )
-
-    dup 2swap ( origaddr addr charlist )
-
-    begin 
-        nil? false =
-    while
-        2dup cdr 2swap car 
-        drop ( origaddr addr charlist char )
-        -rot 2swap ( origaddr charlist addr char )
-        over !
-        1+ -rot ( origaddr nextaddr charlist )
-    repeat
-
-    2drop ( origaddr finaladdr ) 
-    swap -
-;
-
 : load ( addr n -- finalResult )
     open-input-file
 
@@ -1835,13 +2040,13 @@ variable gc-stack-depth
 ;
 
 : repl
-    cr ." Welcome to scheme.forth.jl!" cr
-       ." Use Ctrl-D to exit." cr
-
     empty-parse-str
 
     enable-gc
 
+    \ Display welcome message
+    welcome-symbol nil cons global-env obj@ eval 2drop
+
     begin
         ['] repl-body catch
         case