(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)