511 lines
18 KiB
Scheme
511 lines
18 KiB
Scheme
|
|
(import (ikarus))
|
|
|
|
(define (make-annotated-procedure ann proc)
|
|
(import (ikarus system $codes))
|
|
($make-annotated-procedure ann proc))
|
|
|
|
(define-struct trace (src expr rator rands))
|
|
|
|
(module (get-traces debug-call)
|
|
|
|
(define outer-ring-size 30)
|
|
(define inner-ring-size 10)
|
|
|
|
(define end-marker -1)
|
|
|
|
(define-struct icell (prev next num content))
|
|
(define-struct ocell (prev next num cf icell))
|
|
|
|
(define (make-ring n cell-prev cell-next cell-prev-set! cell-next-set! make-cell)
|
|
(let ([ring (make-cell)])
|
|
(cell-prev-set! ring ring)
|
|
(cell-next-set! ring ring)
|
|
(do ((n n (- n 1)))
|
|
((<= n 1))
|
|
(let ([cell (make-cell)]
|
|
[next (cell-next ring)])
|
|
(cell-prev-set! cell ring)
|
|
(cell-next-set! cell next)
|
|
(cell-prev-set! next cell)
|
|
(cell-next-set! ring cell)))
|
|
ring))
|
|
|
|
(define (make-double-ring n m)
|
|
(make-ring n
|
|
ocell-prev ocell-next set-ocell-prev! set-ocell-next!
|
|
(lambda ()
|
|
(make-ocell #f #f end-marker #f
|
|
(make-ring m
|
|
icell-prev icell-next set-icell-prev! set-icell-next!
|
|
(lambda () (make-icell #f #f end-marker (lambda () #f))))))))
|
|
|
|
(define (ring->list x cell-num cell-prev cell-content)
|
|
(let f ([x x] [orig #f])
|
|
(if (or (eq? x orig) (eqv? (cell-num x) end-marker))
|
|
'()
|
|
(cons (cons (cell-num x) (cell-content x))
|
|
(f (cell-prev x) (or orig x))))))
|
|
|
|
(define (get-traces)
|
|
(ring->list step-ring ocell-num ocell-prev
|
|
(lambda (x)
|
|
(ring->list (ocell-icell x) icell-num icell-prev icell-content))))
|
|
|
|
(define step-ring
|
|
(make-double-ring outer-ring-size inner-ring-size))
|
|
|
|
(define (debug-call src expr rator rands)
|
|
(call/cf
|
|
(lambda (cf)
|
|
(if (eq? cf (ocell-cf step-ring))
|
|
(reduce src expr rator rands)
|
|
(let ([cf #f] [pcf #f])
|
|
(dynamic-wind
|
|
(lambda ()
|
|
(let ([prev step-ring])
|
|
(let ([next (ocell-next prev)])
|
|
(set! pcf (ocell-cf prev))
|
|
(set-ocell-num! next (+ (ocell-num prev) 1))
|
|
(set-icell-num! (ocell-icell next) end-marker)
|
|
(set! step-ring next)
|
|
(set-ocell-cf! step-ring cf))))
|
|
(lambda ()
|
|
(call/cf
|
|
(lambda (cf2)
|
|
(set! cf cf2)
|
|
(set-ocell-cf! step-ring cf)
|
|
(reduce src expr rator rands))))
|
|
(lambda ()
|
|
(let ([next step-ring])
|
|
(let ([prev (ocell-prev next)])
|
|
(set-ocell-num! prev (- (ocell-num next) 1))
|
|
(set-ocell-num! next end-marker)
|
|
(set-icell-num! (ocell-icell next) end-marker)
|
|
(set-ocell-cf! prev pcf)
|
|
(set! step-ring prev))))))))))
|
|
|
|
(define (reduce src expr rator rands)
|
|
(define (mark-reduction! x)
|
|
(let ([prev (ocell-icell step-ring)])
|
|
(let ([next (icell-next prev)])
|
|
(set-icell-content! next x)
|
|
(set-icell-num! next (+ (icell-num prev) 1))
|
|
(set-ocell-icell! step-ring next))))
|
|
(mark-reduction! (make-trace src expr rator rands))
|
|
(apply rator rands))
|
|
|
|
)
|
|
|
|
(define (primitive-value x)
|
|
(import (ikarus system $codes))
|
|
(import (ikarus system $structs))
|
|
(import (ikarus system $flonums))
|
|
(import (ikarus system $bignums))
|
|
(case x
|
|
[($code-reloc-vector) (lambda (x) ($code-reloc-vector x))]
|
|
[($code-freevars) (lambda (x) ($code-freevars x))]
|
|
[($code-size) (lambda (x) ($code-size x))]
|
|
[($code-annotation) (lambda (x) ($code-annotation x))]
|
|
[($code-ref) (lambda (x i) ($code-ref x i))]
|
|
[($code-set!) (lambda (x i v) ($code-set! x i v))]
|
|
[($code->closure) (lambda (x) ($code->closure x))]
|
|
[($closure-code) (lambda (x) ($closure-code x))]
|
|
[($struct)
|
|
(case-lambda
|
|
[(x0)
|
|
($struct x0)]
|
|
[(x0 x1)
|
|
($struct x0 x1)]
|
|
[(x0 x1 x2)
|
|
($struct x0 x1 x2)]
|
|
[(x0 x1 x2 x3)
|
|
($struct x0 x1 x2 x3)]
|
|
[(x0 x1 x2 x3 x4)
|
|
($struct x0 x1 x2 x3 x4)]
|
|
[(x0 x1 x2 x3 x4 x5)
|
|
($struct x0 x1 x2 x3 x4 x5)]
|
|
[(x0 x1 x2 x3 x4 x5 x6)
|
|
($struct x0 x1 x2 x3 x4 x5 x6)]
|
|
[(x0 x1 x2 x3 x4 x5 x6 x7)
|
|
($struct x0 x1 x2 x3 x4 x5 x6 x7)]
|
|
[(x0 x1 x2 x3 x4 x5 x6 x7 x8)
|
|
($struct x0 x1 x2 x3 x4 x5 x6 x7 x8)]
|
|
[(x0 x1 x2 x3 x4 x5 x6 x7 x8 x9)
|
|
($struct x0 x1 x2 x3 x4 x5 x6 x7 x8 x9)]
|
|
[(x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 x10)
|
|
($struct x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 x10)]
|
|
[(x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 x10 x11)
|
|
($struct x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 x10 x11)]
|
|
)]
|
|
[($struct-set!) struct-set!]
|
|
[($struct-ref) struct-ref]
|
|
[($annotated-procedure-annotation)
|
|
(lambda (x) ($annotated-procedure-annotation x))]
|
|
[($char<=) char<=?]
|
|
[($string-ref) string-ref]
|
|
[($flonum-u8-ref)
|
|
(lambda (x i)
|
|
(case i
|
|
[(0) ($flonum-u8-ref x 0)]
|
|
[(1) ($flonum-u8-ref x 1)]
|
|
[(2) ($flonum-u8-ref x 2)]
|
|
[(3) ($flonum-u8-ref x 3)]
|
|
[(4) ($flonum-u8-ref x 4)]
|
|
[(5) ($flonum-u8-ref x 5)]
|
|
[(6) ($flonum-u8-ref x 6)]
|
|
[(7) ($flonum-u8-ref x 7)]
|
|
[else (error 'flonum-u8-ref "invalid index" i)]))]
|
|
[($bignum-positive?) positive?]
|
|
[($bignum-byte-ref) (lambda (x i) ($bignum-byte-ref x i))]
|
|
[($bignum-size) (lambda (x) ($bignum-size x))]
|
|
[else (system-value x)]))
|
|
|
|
|
|
(define core-interpret
|
|
(let ()
|
|
(define who 'interpret)
|
|
(define (lexical-set! env i j v)
|
|
(vector-set! (list-ref env i) j v))
|
|
(define (lexical-ref env i j)
|
|
(vector-ref (list-ref env i) j))
|
|
(define (global-set! loc v)
|
|
(set-symbol-value! loc v))
|
|
(define (global-ref loc)
|
|
(top-level-value loc))
|
|
(define (extend-env ls env)
|
|
(define (properize x)
|
|
(cond
|
|
[(null? x) '()]
|
|
[(pair? x) (cons (car x) (properize (cdr x)))]
|
|
[else (list x)]))
|
|
(let ([v (list->vector (properize ls))])
|
|
(values (cons v env) (vector-length v))))
|
|
(define (lookup x env)
|
|
(let f ([i 0] [env env])
|
|
(cond
|
|
[(null? env) `(global ,x)]
|
|
[else
|
|
(let g ([j 0] [rib (car env)])
|
|
(cond
|
|
[(= j (vector-length rib))
|
|
(f (+ i 1) (cdr env))]
|
|
[(eq? x (vector-ref rib j)) `(lexical ,i ,j)]
|
|
[else (g (+ j 1) rib)]))])))
|
|
(define (get-fmls x args)
|
|
(define (matching? fmls args)
|
|
(cond
|
|
[(null? fmls) (null? args)]
|
|
[(pair? fmls) (and (pair? args) (matching? (cdr fmls) (cdr args)))]
|
|
[else #t]))
|
|
(define (get-cls* x)
|
|
(if (pair? x)
|
|
(case (car x)
|
|
[(case-lambda) (cdr x)]
|
|
[(annotated-case-lambda) (cddr x)]
|
|
[else '()])
|
|
'()))
|
|
(let f ([cls* (get-cls* x)])
|
|
(cond
|
|
[(null? cls*) #f]
|
|
[(matching? (caar cls*) args)
|
|
(caar cls*)]
|
|
[else (f (cdr cls*))])))
|
|
(define (compile-letrec* binding* body env ctxt)
|
|
(let ([lhs* (map car binding*)]
|
|
[rhs* (map cadr binding*)])
|
|
(let-values ([(env n) (extend-env lhs* env)])
|
|
(let ([rhs* (map (lambda (lhs rhs)
|
|
(compile-expr rhs env lhs))
|
|
lhs* rhs*)]
|
|
[body (compile-expr body env ctxt)])
|
|
(lambda (env)
|
|
(let ([vec (make-vector n)])
|
|
(let ([env (cons vec env)])
|
|
(let f ([i 0] [rhs* rhs*])
|
|
(if (null? rhs*)
|
|
(body env)
|
|
(begin
|
|
(vector-set! vec i ((car rhs*) env))
|
|
(f (+ i 1) (cdr rhs*))))))))))))
|
|
(define (compile-library-letrec* binding* body env ctxt)
|
|
(let ([lhs* (map car binding*)]
|
|
[loc* (map cadr binding*)]
|
|
[rhs* (map caddr binding*)])
|
|
(let-values ([(env n) (extend-env lhs* env)])
|
|
(let ([rhs* (map (lambda (lhs rhs)
|
|
(compile-expr rhs env lhs))
|
|
lhs* rhs*)]
|
|
[body (compile-expr body env ctxt)])
|
|
(lambda (env)
|
|
(let ([vec (make-vector n)])
|
|
(let ([env (cons vec env)])
|
|
(let f ([i 0] [rhs* rhs*] [loc* loc*])
|
|
(if (null? rhs*)
|
|
(body env)
|
|
(let ([v ((car rhs*) env)])
|
|
(vector-set! vec i v)
|
|
(global-set! (car loc*) v)
|
|
(f (+ i 1) (cdr rhs*) (cdr loc*))))))))))))
|
|
(define (compile-case-lambda ae binding* env ctxt)
|
|
(define (compile-clause fmls expr k)
|
|
(let-values ([(env n) (extend-env fmls env)])
|
|
(let ([expr (compile-expr expr env
|
|
(if (pair? ctxt) (car ctxt) #f))])
|
|
(cond
|
|
[(list? fmls)
|
|
(lambda (env args argcount)
|
|
(if (= n argcount)
|
|
(expr (cons (list->vector args) env))
|
|
(k env args argcount)))]
|
|
[else
|
|
(lambda (env args argcount)
|
|
(let ([n1 (- n 1)])
|
|
(if (>= argcount n1)
|
|
(let ([vec (make-vector n)])
|
|
(let f ([ls args] [i 0])
|
|
(cond
|
|
[(= i n1)
|
|
(vector-set! vec i ls)]
|
|
[else
|
|
(vector-set! vec i (car ls))
|
|
(f (cdr ls) (+ i 1))]))
|
|
(expr (cons vec env)))
|
|
(k env args argcount))))]))))
|
|
(let ([proc
|
|
(let f ([binding* binding*])
|
|
(cond
|
|
[(null? binding*)
|
|
(lambda (env args argcount)
|
|
(assertion-violation 'apply
|
|
"incorrect number of arguments"
|
|
args))]
|
|
[else
|
|
(let ([b (car binding*)])
|
|
(compile-clause (car b) (cadr b)
|
|
(f (cdr binding*))))]))])
|
|
(lambda (env)
|
|
(make-annotated-procedure
|
|
(cons
|
|
(and (symbol? ctxt) ctxt)
|
|
(and (annotation? ae) (annotation-source ae)))
|
|
(lambda args
|
|
(proc env args (length args)))))))
|
|
(define (compile-var expr env)
|
|
(let ([x (lookup expr env)])
|
|
(case (car x)
|
|
[(lexical)
|
|
(let ([i (cadr x)] [j (caddr x)])
|
|
(lambda (env)
|
|
(lexical-ref env i j)))]
|
|
[(global)
|
|
(let ([loc (cadr x)])
|
|
(lambda (env)
|
|
(global-ref loc)))]
|
|
[else (die who "invalid value" x)])))
|
|
(define (compile-set! lhs rhs env)
|
|
(let ([rhs (compile-expr rhs env lhs)]
|
|
[x (lookup lhs env)])
|
|
(case (car x)
|
|
[(lexical)
|
|
(let ([i (cadr x)] [j (caddr x)])
|
|
(lambda (env)
|
|
(lexical-set! env i j (rhs env))))]
|
|
[(global)
|
|
(let ([loc (cadr x)])
|
|
(lambda (env)
|
|
(global-set! loc (rhs env))))]
|
|
[else (die who "invalid set! target" lhs)])))
|
|
(define (compile-if e0 e1 e2 env ctxt)
|
|
(let ([e0 (compile-expr e0 env #f)]
|
|
[e1 (compile-expr e1 env ctxt)]
|
|
[e2 (compile-expr e2 env ctxt)])
|
|
(lambda (env)
|
|
(if (e0 env) (e1 env) (e2 env)))))
|
|
(define (compile-begin e e* env ctxt)
|
|
(cond
|
|
[(null? e*) (compile-expr e env ctxt)]
|
|
[else
|
|
(let ([e0 (compile-expr e env #f)]
|
|
[e* (compile-begin (car e*) (cdr e*) env ctxt)])
|
|
(lambda (env) (e0 env) (e* env)))]))
|
|
(define (get-src/expr ae)
|
|
(if (annotation? ae)
|
|
(values (annotation-source ae) (annotation-stripped ae))
|
|
(values #f (syntax->datum ae))))
|
|
(define (compile-let e e* env ctxt)
|
|
(define (build-let fmls e* body)
|
|
(let ([e*
|
|
(list->vector
|
|
(map (lambda (x e) (compile-expr e env x))
|
|
fmls e*))])
|
|
(let-values ([(env _) (extend-env fmls env)])
|
|
(let ([body (compile-expr body env ctxt)])
|
|
(lambda (env)
|
|
(body (cons (vector-map (lambda (x) (x env)) e*) env)))))))
|
|
(define (dispatch cls*)
|
|
(define (try cls)
|
|
(let ([fmls (car cls)] [body (cadr cls)])
|
|
(and (list? fmls)
|
|
(= (length fmls) (length e*))
|
|
(build-let fmls e* body))))
|
|
(and (pair? cls*) (or (try (car cls*)) (dispatch (cdr cls*)))))
|
|
(and (pair? e)
|
|
(case (car e)
|
|
[(lambda) (dispatch (list (cdr e)))]
|
|
[(case-lambda) (dispatch (cdr e))]
|
|
[(annotated-case-lambda) (dispatch (cddr e))]
|
|
[else #f])))
|
|
(define (compile-call ae e e* env ctxt)
|
|
(or (compile-let e e* env ctxt)
|
|
(let ([e (compile-expr e env (list ctxt))]
|
|
[e* (map (lambda (x) (compile-expr x env #f)) e*)])
|
|
(if ae
|
|
(let-values ([(src expr) (get-src/expr ae)])
|
|
(lambda (env)
|
|
(let ([e (e env)] [e* (map (lambda (e) (e env)) e*)])
|
|
(debug-call src expr e e*))))
|
|
(lambda (env)
|
|
(let ([e (e env)] [e* (map (lambda (e) (e env)) e*)])
|
|
(debug-call #f #f e e*)))))))
|
|
(define (compile-expr expr env ctxt)
|
|
(cond
|
|
[(symbol? expr) (compile-var expr env)]
|
|
[(and (pair? expr) (list? expr))
|
|
(let ([a (car expr)] [d (cdr expr)])
|
|
(case a
|
|
[(quote) (let ([v (car d)]) (lambda (env) v))]
|
|
[(set!) (compile-set! (car d) (cadr d) env)]
|
|
[(if)
|
|
(compile-if (car d) (cadr d) (caddr d) env ctxt)]
|
|
[(begin)
|
|
(compile-begin (car d) (cdr d) env ctxt)]
|
|
[(lambda)
|
|
(compile-case-lambda #f (list d) env ctxt)]
|
|
[(case-lambda)
|
|
(compile-case-lambda #f d env ctxt)]
|
|
[(annotated-case-lambda)
|
|
(compile-case-lambda (car d) (cdr d) env ctxt)]
|
|
[(letrec letrec*)
|
|
(compile-letrec* (car d) (cadr d) env ctxt)]
|
|
[(library-letrec*)
|
|
(compile-library-letrec* (car d) (cadr d) env ctxt)]
|
|
[(primitive)
|
|
(let ([v (primitive-value (car d))])
|
|
(lambda (env) v))]
|
|
[(annotated-call)
|
|
(compile-call (car d) (cadr d) (cddr d) env ctxt)]
|
|
[(foreign-call)
|
|
(let ([name (car d)] [args (cdr d)])
|
|
(let ([ls (map (lambda (x) (gensym)) args)])
|
|
(let ([code
|
|
(real-eval
|
|
`(lambda ,ls (foreign-call ,name . ,ls)))])
|
|
(compile-expr `(',code . ,args) env ctxt))))]
|
|
[else
|
|
(compile-call #f a d env ctxt)]))]
|
|
[else
|
|
(die who "invalid expression" expr)]))
|
|
(lambda (expr)
|
|
((compile-expr expr '() #f) '()))))
|
|
|
|
|
|
(define (print-trace x)
|
|
(define (chop x)
|
|
(if (> (string-length x) 60)
|
|
(format "~a#..." (substring x 0 56))
|
|
x))
|
|
(let ([n (car x)] [x (cdr x)])
|
|
(printf " [~a] ~s\n" n (trace-expr x))
|
|
(let ([src (trace-src x)])
|
|
(when (pair? src)
|
|
(printf " source: char ~a of ~a\n" (cdr src) (car src))))
|
|
(printf " operator: ~s\n" (trace-rator x))
|
|
(printf " operands: ")
|
|
(let ([ls (map (lambda (x) (format "~s" x)) (trace-rands x))])
|
|
(if (< (apply + 1 (length ls) (map string-length ls)) 60)
|
|
(write (trace-rands x))
|
|
(begin
|
|
(display "(")
|
|
(let f ([a (car ls)] [ls (cdr ls)])
|
|
(display (chop a))
|
|
(if (null? ls)
|
|
(display ")")
|
|
(begin
|
|
(display "\n ")
|
|
(f (car ls) (cdr ls))))))))
|
|
(newline)))
|
|
|
|
(define (print-step x)
|
|
(let ([n (car x)] [ls (cdr x)])
|
|
(unless (null? ls)
|
|
(printf "FRAME ~s:\n" n)
|
|
(for-each print-trace (reverse ls)))))
|
|
|
|
(define (print-all-traces)
|
|
(newline)
|
|
(for-each print-step (reverse (get-traces))))
|
|
|
|
(define (guarded-start proc)
|
|
(with-exception-handler
|
|
(lambda (con)
|
|
; (print-all-traces)
|
|
((call/cc
|
|
(lambda (k)
|
|
(printf "Condition trapped by debugger.\n")
|
|
(print-condition con)
|
|
(printf "[t] Trace. [r] Reraise condition. [e] Exit. [^D] Return.\n")
|
|
(new-cafe
|
|
(lambda (x)
|
|
(case x
|
|
[(R r) (k (lambda () (raise-continuable con)))]
|
|
[(E e) (exit 0)]
|
|
[(T t) (print-all-traces)]
|
|
[else (printf "invalid option\n")])))
|
|
void))))
|
|
proc))
|
|
|
|
(define (start-repl)
|
|
(display "Ikarus Interpreter\n\n")
|
|
(new-cafe
|
|
(lambda (x)
|
|
(guarded-start
|
|
(lambda ()
|
|
(eval x (interaction-environment)))))))
|
|
|
|
(define (start-script script-name args)
|
|
(command-line-arguments (cons script-name args))
|
|
(guarded-start
|
|
(lambda ()
|
|
(load-r6rs-script script-name #f #t))))
|
|
|
|
(define original-eval (current-core-eval))
|
|
|
|
(define (real-eval x)
|
|
(parameterize ([current-core-eval original-eval])
|
|
(eval x (environment '(ikarus)))))
|
|
|
|
(current-core-eval core-interpret)
|
|
(apply
|
|
(case-lambda
|
|
[(interpreter flag script-name . rest)
|
|
(if (string=? flag "--r6rs-script")
|
|
(start-script script-name rest)
|
|
(error interpreter "invalid args" (cons* flag script-name rest)))]
|
|
[(interpreter) (start-repl)]
|
|
[(interpreter . rest)
|
|
(error interpreter "invalid args" rest)])
|
|
(command-line-arguments))
|
|
|
|
|
|
|
|
|
|
|
|
#!eof
|
|
|
|
(print-graph #t)
|
|
|
|
;(write (make-double-rib 5 5))
|
|
(write (make-ring 10 (lambda () #f)))
|
|
(newline)
|