! Copyright (C) 2004, 2006 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
IN: interpreter
USING: arrays errors generic io kernel kernel-internals math
namespaces prettyprint sequences strings threads vectors words ;

! A Factor interpreter written in Factor. It can transfer the
! continuation to and from the primary interpreter. Used by
! compiler for partial evaluation, also by the walker.

! Meta-stacks;
SYMBOL: meta-d
: push-d meta-d get push ;
: pop-d meta-d get pop ;
: peek-d meta-d get peek ;
SYMBOL: meta-r
: push-r meta-r get push ;
: pop-r meta-r get pop ;
: peek-r meta-r get peek ;
SYMBOL: meta-c
: push-c meta-c get push ;
: pop-c meta-c get pop ;
: peek-c meta-c get peek ;
SYMBOL: meta-name
SYMBOL: meta-catch

! Call frame
SYMBOL: callframe
SYMBOL: callframe-scan
SYMBOL: callframe-end

: meta-callframe ( -- seq )
    { callframe callframe-scan callframe-end } [ get ] map ;

: (meta-call) ( quot -- )
    dup callframe set
    length callframe-end set
    0 callframe-scan set ;

! Callframe.
: up ( -- )
    pop-c callframe-end set
    pop-c callframe-scan set
    pop-c callframe set ;

: done-cf? ( -- ? ) callframe-scan get callframe-end get >= ;

: done? ( -- ? ) done-cf? meta-c get empty? and ;

: (next)
    callframe-scan get callframe get nth callframe-scan inc ;

: next ( quot -- )
    {
        { [ done? ] [ drop [ ] (meta-call) ] }
        { [ done-cf? ] [ drop up ] }
        { [ t ] [ >r (next) r> call ] }
    } cond ; inline

: meta-interp ( -- interp )
    meta-d get meta-r get meta-c get
    meta-name get meta-catch get <continuation> ;

: init-meta-interp ( -- )
    V{ } clone meta-catch set
    V{ } clone meta-name set
    V{ } clone meta-c set
    V{ } clone meta-r set
    V{ } clone meta-d set ;

: set-meta-interp ( interp -- )
    >continuation<
    meta-catch set
    meta-name set
    meta-c set
    meta-r set
    meta-d set ;

: save-callframe ( -- )
    done-cf? [
        callframe get push-c
        callframe-scan get push-c
        callframe-end get push-c
    ] unless ;

: meta-call ( quot -- )
    #! Note we do tail call optimization here.
    save-callframe (meta-call) ;

: <callframe> ( quot -- seq )
    0 over length 3array ;

: catch-harness ( continuation -- quot )
    [ [ c> 2array ] % , \ continue-with , ] [ ] make ;

: host-harness ( quot continuation -- )
    tuck [
        catch-harness , \ >c ,
        %
        [ c> drop continuation ] %
        ,
        \ continue-with ,
    ] [ ] make ;

: restore-harness ( obj -- )
    dup array? [
        init-meta-interp [ ] (meta-call)
        first2 schedule-thread-with
    ] [
        set-meta-interp
    ] if ;

: host-quot ( quot -- )
    [
        host-harness <callframe> meta-c get swap nappend
        meta-interp continue
    ] callcc1 restore-harness drop ;

: host-word ( word -- ) unit host-quot ;

GENERIC: do-1 ( object -- )

M: word do-1
    dup "meta-word" word-prop [ call ] [ host-word ] ?if ;

M: wrapper do-1 wrapped push-d ;

M: object do-1 push-d ;

GENERIC: do ( obj -- )

M: word do
    dup "meta-word" word-prop [
        call
    ] [
        dup compound? [ word-def meta-call ] [ host-word ] if
    ] ?if ;

M: object do do-1 ;

! The interpreter loses object identity of the name and catch
! stacks -- they are copied after each step -- so we execute
! these atomically and don't allow stepping into these words
\ >n [ \ >n host-word ] "meta-word" set-word-prop
\ n> [ \ n> host-word ] "meta-word" set-word-prop
\ >c [ \ >c host-word ] "meta-word" set-word-prop
\ c> [ \ c> host-word ] "meta-word" set-word-prop

\ call [ pop-d meta-call ] "meta-word" set-word-prop
\ execute [ pop-d do ] "meta-word" set-word-prop
\ if [ pop-d pop-d pop-d [ nip ] [ drop ] if meta-call ] "meta-word" set-word-prop
\ dispatch [ pop-d pop-d swap nth meta-call ] "meta-word" set-word-prop

: step ( -- ) [ do-1 ] next ;

: step-in ( -- ) [ do ] next ;

: step-out ( -- )
    callframe get callframe-scan get tail
    host-quot [ ] (meta-call) ;

: step-all ( -- )
    save-callframe
    meta-c [ V{ [ stop ] 0 1 } swap append ] change
    meta-interp schedule-thread yield
    V{ } clone meta-c set
    [ ] (meta-call) ;