2428 lines
84 KiB
Scheme
2428 lines
84 KiB
Scheme
;;; Ikarus Scheme -- A compiler for R6RS Scheme.
|
|
;;; Copyright (C) 2006,2007,2008 Abdulaziz Ghuloum
|
|
;;;
|
|
;;; This program is free software: you can redistribute it and/or modify
|
|
;;; it under the terms of the GNU General Public License version 3 as
|
|
;;; published by the Free Software Foundation.
|
|
;;;
|
|
;;; This program is distributed in the hope that it will be useful, but
|
|
;;; WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
;;; General Public License for more details.
|
|
;;;
|
|
;;; You should have received a copy of the GNU General Public License
|
|
;;; along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
(library (ikarus.compiler)
|
|
(export compile-core-expr-to-port
|
|
assembler-output optimize-cp
|
|
current-primitive-locations eval-core
|
|
compile-core-expr expand/optimize optimizer-output
|
|
cp0-effort-limit cp0-size-limit optimize-level
|
|
perform-tag-analysis tag-analysis-output)
|
|
(import
|
|
(rnrs hashtables)
|
|
(ikarus system $fx)
|
|
(ikarus system $pairs)
|
|
(only (ikarus system $codes) $code->closure)
|
|
(only (ikarus system $structs) $struct-ref $struct/rtd?)
|
|
(except (ikarus)
|
|
optimize-level debug-optimizer
|
|
fasl-write scc-letrec optimize-cp
|
|
compile-core-expr-to-port assembler-output
|
|
current-primitive-locations eval-core
|
|
cp0-size-limit cp0-effort-limit
|
|
expand/optimize optimizer-output
|
|
tag-analysis-output perform-tag-analysis)
|
|
(ikarus.fasl.write)
|
|
(ikarus.intel-assembler))
|
|
|
|
|
|
(define-syntax struct-case
|
|
(lambda (x)
|
|
(define (enumerate fld* i)
|
|
(syntax-case fld* ()
|
|
[() #'()]
|
|
[(x . x*)
|
|
(with-syntax ([i i] [i* (enumerate #'x* (fx+ i 1))])
|
|
#'(i . i*))]))
|
|
(define (generate-body ctxt cls*)
|
|
(syntax-case cls* (else)
|
|
[() (with-syntax ([x x]) #'(error #f "unmatched " v 'x))]
|
|
[([else b b* ...]) #'(begin b b* ...)]
|
|
[([(rec-name rec-field* ...) b b* ...] . rest) (identifier? #'rec-name)
|
|
(with-syntax ([altern (generate-body ctxt #'rest)]
|
|
[(id* ...) (enumerate #'(rec-field* ...) 0)]
|
|
[rtd #'(type-descriptor rec-name)])
|
|
#'(if ($struct/rtd? v rtd)
|
|
(let ([rec-field* ($struct-ref v id*)] ...)
|
|
b b* ...)
|
|
altern))]))
|
|
(syntax-case x ()
|
|
[(_ expr cls* ...)
|
|
(with-syntax ([body (generate-body #'_ #'(cls* ...))])
|
|
#'(let ([v expr]) body))])))
|
|
|
|
|
|
|
|
|
|
(define (remq1 x ls)
|
|
(cond
|
|
[(null? ls) '()]
|
|
[(eq? x (car ls)) (cdr ls)]
|
|
[else
|
|
(let ([t (remq1 x (cdr ls))])
|
|
(cond
|
|
[(eq? t (cdr ls)) ls]
|
|
[else (cons (car ls) t)]))]))
|
|
|
|
(define (singleton x) (list x))
|
|
|
|
(define (union s1 s2)
|
|
(define (add* s1 s2)
|
|
(cond
|
|
[(null? s1) s2]
|
|
[else (add (car s1) (add* (cdr s1) s2))]))
|
|
(define (add x s)
|
|
(cond
|
|
[(memq x s) s]
|
|
[else (cons x s)]))
|
|
(cond
|
|
[(null? s1) s2]
|
|
[(null? s2) s1]
|
|
[else (add* s1 s2)]))
|
|
|
|
(define (difference s1 s2)
|
|
(define (rem* s1 s2)
|
|
(cond
|
|
[(null? s1) s2]
|
|
[else (remq1 (car s1) (rem* (cdr s1) s2))]))
|
|
(cond
|
|
[(null? s1) '()]
|
|
[(null? s2) s1]
|
|
[else (rem* s2 s1)]))
|
|
|
|
|
|
|
|
(define-struct constant (value))
|
|
(define-struct code-loc (label))
|
|
(define-struct foreign-label (label))
|
|
(define-struct var
|
|
(name reg-conf frm-conf var-conf reg-move frm-move var-move
|
|
loc index referenced global-loc))
|
|
(define-struct cp-var (idx))
|
|
(define-struct frame-var (idx))
|
|
(define-struct new-frame (base-idx size body))
|
|
(define-struct save-cp (loc))
|
|
(define-struct eval-cp (check body))
|
|
(define-struct return (value))
|
|
(define-struct call-cp
|
|
(call-convention label save-cp? rp-convention base-idx arg-count live-mask))
|
|
(define-struct tailcall-cp (convention label arg-count))
|
|
(define-struct primcall (op arg*))
|
|
(define-struct primref (name))
|
|
(define-struct conditional (test conseq altern))
|
|
(define-struct interrupt-call (test handler))
|
|
(define-struct bind (lhs* rhs* body))
|
|
(define-struct recbind (lhs* rhs* body))
|
|
(define-struct rec*bind (lhs* rhs* body))
|
|
(define-struct fix (lhs* rhs* body))
|
|
|
|
(define-struct seq (e0 e1))
|
|
(define-struct case-info (label args proper))
|
|
(define-struct clambda-case (info body))
|
|
(define-struct clambda (label cases cp free name))
|
|
(define-struct closure (code free* well-known?))
|
|
(define-struct funcall (op rand*))
|
|
(define-struct jmpcall (label op rand*))
|
|
(define-struct forcall (op rand*))
|
|
(define-struct codes (list body))
|
|
(define-struct assign (lhs rhs))
|
|
(define-struct mvcall (producer consumer))
|
|
|
|
(define-struct known (expr type))
|
|
|
|
(define-struct shortcut (body handler))
|
|
|
|
(define-struct fvar (idx))
|
|
(define-struct object (val))
|
|
(define-struct locals (vars body))
|
|
(define-struct nframe (vars live body))
|
|
(define-struct nfv (conf loc var-conf frm-conf nfv-conf))
|
|
(define-struct ntcall (target value args mask size))
|
|
(define-struct asm-instr (op dst src))
|
|
(define-struct disp (s0 s1))
|
|
|
|
;;; this define-structure definition for compatibility with the
|
|
;;; notation used in Oscar's thesis.
|
|
(define-syntax define-structure
|
|
(lambda (stx)
|
|
(define (fmt ctxt)
|
|
(lambda (str . args)
|
|
(datum->syntax ctxt
|
|
(string->symbol
|
|
(apply format str (map syntax->datum args))))))
|
|
(syntax-case stx ()
|
|
[(_ (name fields ...))
|
|
#'(define-struct name (fields ...))]
|
|
[(_ (name fields ...) ([others defaults] ...))
|
|
(with-syntax ([(pred maker (getters ...) (setters ...))
|
|
(let ([fmt (fmt #'name)])
|
|
(list (fmt "~s?" #'name)
|
|
(fmt "make-~s" #'name)
|
|
(map (lambda (x) (fmt "~s-~s" #'name x))
|
|
#'(fields ... others ...))
|
|
(map (lambda (x) (fmt "set-~s-~s!" #'name x))
|
|
#'(fields ... others ...))))])
|
|
#'(module (name pred getters ... setters ... maker)
|
|
(module P (name pred getters ... setters ... maker)
|
|
(define-struct name (fields ... others ...)))
|
|
(module (maker)
|
|
(define (maker fields ...)
|
|
(import P)
|
|
(maker fields ... defaults ...)))
|
|
(module (name pred getters ... setters ...)
|
|
(import P))))])))
|
|
;;;
|
|
(define-structure (prelex name operand)
|
|
([source-referenced? #f]
|
|
[source-assigned? #f]
|
|
[residual-referenced? #f]
|
|
[residual-assigned? #f]
|
|
[global-location #f]))
|
|
|
|
(define mkfvar
|
|
(let ([cache '()])
|
|
(lambda (i)
|
|
(cond
|
|
[(fixnum? i)
|
|
(cond
|
|
[(assv i cache) => cdr]
|
|
[else
|
|
(let ([fv (make-fvar i)])
|
|
(set! cache (cons (cons i fv) cache))
|
|
fv)])]
|
|
[else (error 'mkfvar "not a fixnum" i)]))))
|
|
|
|
(define (unique-var name)
|
|
(make-var name #f #f #f #f #f #f #f #f #f #f))
|
|
|
|
(define (recordize x)
|
|
(define *cookie* (gensym))
|
|
(define (gen-fml* fml*)
|
|
(cond
|
|
[(pair? fml*)
|
|
(let ([v (make-prelex (car fml*) #f)])
|
|
(putprop (car fml*) *cookie* v)
|
|
(cons v (gen-fml* (cdr fml*))))]
|
|
[(symbol? fml*)
|
|
(let ([v (make-prelex fml* #f)])
|
|
(putprop fml* *cookie* v)
|
|
v)]
|
|
[else '()]))
|
|
(define (ungen-fml* fml*)
|
|
(cond
|
|
[(pair? fml*)
|
|
(remprop (car fml*) *cookie*)
|
|
(ungen-fml* (cdr fml*))]
|
|
[(symbol? fml*)
|
|
(remprop fml* *cookie*)]))
|
|
(define (properize fml*)
|
|
(cond
|
|
[(pair? fml*)
|
|
(cons (car fml*) (properize (cdr fml*)))]
|
|
[(null? fml*) '()]
|
|
[else (list fml*)]))
|
|
(define (quoted-sym x)
|
|
(if (and (list? x)
|
|
(fx= (length x) 2)
|
|
(eq? 'quote (car x))
|
|
(symbol? (cadr x)))
|
|
(cadr x)
|
|
(error 'quoted-sym "not a quoted symbol" x)))
|
|
(define (quoted-string x)
|
|
(if (and (list? x)
|
|
(fx= (length x) 2)
|
|
(eq? 'quote (car x))
|
|
(string? (cadr x)))
|
|
(cadr x)
|
|
(error 'quoted-string "not a quoted string" x)))
|
|
(define (lexical x)
|
|
(getprop x *cookie*))
|
|
(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]))
|
|
(cond
|
|
[(and (pair? x) (eq? (car x) 'case-lambda))
|
|
(let f ([cls* (cdr x)])
|
|
(cond
|
|
[(null? cls*) '()]
|
|
[(matching? (caar cls*) args)
|
|
(caar cls*)]
|
|
[else (f (cdr cls*))]))]
|
|
[else '()]))
|
|
(define (make-global-set! lhs rhs)
|
|
(make-funcall (make-primref '$init-symbol-value!)
|
|
(list (make-constant lhs) rhs)))
|
|
(define-syntax equal-case
|
|
(lambda (x)
|
|
(syntax-case x ()
|
|
[(_ val clause* ...)
|
|
(with-syntax ([body
|
|
(let f ([clause* #'(clause* ...)])
|
|
(syntax-case clause* (else)
|
|
[([else e e* ...])
|
|
#'(begin e e* ...)]
|
|
[([(datum* ...) e e* ...] . rest)
|
|
(with-syntax ([rest (f #'rest)])
|
|
#'(if (member t '(datum* ...))
|
|
(begin e e* ...)
|
|
rest))]))])
|
|
#'(let ([t val]) body))])))
|
|
(define (E x ctxt)
|
|
(cond
|
|
[(pair? x)
|
|
(equal-case (car x)
|
|
[(quote) (make-constant (cadr x))]
|
|
[(if)
|
|
(make-conditional
|
|
(E (cadr x) #f)
|
|
(E (caddr x) ctxt)
|
|
(E (cadddr x) ctxt))]
|
|
[(set!)
|
|
(let ([lhs (cadr x)] [rhs (caddr x)])
|
|
(cond
|
|
[(lexical lhs) =>
|
|
(lambda (var)
|
|
(set-prelex-source-assigned?! var #t)
|
|
(make-assign var (E rhs lhs)))]
|
|
[else (make-global-set! lhs (E rhs lhs))]))]
|
|
[(begin)
|
|
(let f ([a (cadr x)] [d (cddr x)])
|
|
(cond
|
|
[(null? d) (E a ctxt)]
|
|
[else
|
|
(make-seq (E a #f) (f (car d) (cdr d)))]))]
|
|
[(letrec)
|
|
(let ([bind* (cadr x)] [body (caddr x)])
|
|
(let ([lhs* (map car bind*)]
|
|
[rhs* (map cadr bind*)])
|
|
(let ([nlhs* (gen-fml* lhs*)])
|
|
(let ([expr (make-recbind nlhs* (map E rhs* lhs*) (E body ctxt))])
|
|
(ungen-fml* lhs*)
|
|
expr))))]
|
|
[(letrec*)
|
|
(let ([bind* (cadr x)] [body (caddr x)])
|
|
(let ([lhs* (map car bind*)]
|
|
[rhs* (map cadr bind*)])
|
|
(let ([nlhs* (gen-fml* lhs*)])
|
|
(let ([expr (make-rec*bind nlhs* (map E rhs* lhs*) (E body ctxt))])
|
|
(ungen-fml* lhs*)
|
|
expr))))]
|
|
[(library-letrec*)
|
|
(let ([bind* (cadr x)] [body (caddr x)])
|
|
(let ([lhs* (map car bind*)]
|
|
[loc* (map cadr bind*)]
|
|
[rhs* (map caddr bind*)])
|
|
(let ([nlhs* (gen-fml* lhs*)])
|
|
(for-each
|
|
(lambda (lhs loc)
|
|
(set-prelex-global-location! lhs loc))
|
|
nlhs* loc*)
|
|
(let ([expr (make-rec*bind nlhs* (map E rhs* lhs*)
|
|
(let f ([lhs* nlhs*] [loc* loc*])
|
|
(cond
|
|
[(null? lhs*) (E body ctxt)]
|
|
[(not (car loc*)) (f (cdr lhs*) (cdr loc*))]
|
|
[else (f (cdr lhs*) (cdr loc*))])))])
|
|
(ungen-fml* lhs*)
|
|
expr))))]
|
|
[(case-lambda)
|
|
(let ([cls*
|
|
(map
|
|
(let ([ctxt (if (pair? ctxt) (car ctxt) #f)])
|
|
(lambda (cls)
|
|
(let ([fml* (car cls)] [body (cadr cls)])
|
|
(let ([nfml* (gen-fml* fml*)])
|
|
(let ([body (E body ctxt)])
|
|
(ungen-fml* fml*)
|
|
(make-clambda-case
|
|
(make-case-info
|
|
(gensym)
|
|
(properize nfml*)
|
|
(list? fml*))
|
|
body))))))
|
|
(cdr x))])
|
|
(make-clambda (gensym) cls* #f #f
|
|
(and (symbol? ctxt) ctxt)))]
|
|
[(lambda)
|
|
(E `(case-lambda ,(cdr x)) ctxt)]
|
|
[(foreign-call)
|
|
(let ([name (quoted-string (cadr x))] [arg* (cddr x)])
|
|
(make-forcall name (map (lambda (x) (E x #f)) arg*)))]
|
|
[(primitive)
|
|
(let ([var (cadr x)])
|
|
(make-primref var))]
|
|
[((primitive make-parameter))
|
|
(case (length x)
|
|
[(2)
|
|
(let ([val-expr (cadr x)]
|
|
[t (gensym 't)]
|
|
[x (gensym 'x)])
|
|
(E `((lambda (,t)
|
|
(case-lambda
|
|
[() ,t]
|
|
[(,x) (set! ,t ,x)]))
|
|
,val-expr)
|
|
ctxt))]
|
|
[(3)
|
|
(let ([val-expr (cadr x)]
|
|
[guard-expr (caddr x)]
|
|
[f (gensym 'f)]
|
|
[t (gensym 't)]
|
|
[t0 (gensym 't)]
|
|
[x (gensym 'x)])
|
|
(E `((case-lambda
|
|
[(,t ,f)
|
|
(if ((primitive procedure?) ,f)
|
|
((case-lambda
|
|
[(,t0)
|
|
(case-lambda
|
|
[() ,t0]
|
|
[(,x) (set! ,t0 (,f ,x))])])
|
|
(,f ,t))
|
|
((primitive die)
|
|
'make-parameter
|
|
'"not a procedure"
|
|
,f))])
|
|
,val-expr
|
|
,guard-expr)
|
|
ctxt))]
|
|
[else
|
|
(make-funcall
|
|
(make-primref 'make-parameter)
|
|
(map (lambda (x) (E x #f)) (cdr x)))])]
|
|
[else
|
|
(let ([names (get-fmls (car x) (cdr x))])
|
|
(make-funcall
|
|
(E (car x) (list ctxt))
|
|
(let f ([arg* (cdr x)] [names names])
|
|
(cond
|
|
[(pair? names)
|
|
(cons
|
|
(E (car arg*) (car names))
|
|
(f (cdr arg*) (cdr names)))]
|
|
[else
|
|
(map (lambda (x) (E x #f)) arg*)]))))])]
|
|
[(symbol? x)
|
|
(cond
|
|
[(lexical x) =>
|
|
(lambda (var)
|
|
(set-prelex-source-referenced?! var #t)
|
|
var)]
|
|
[else
|
|
(make-funcall
|
|
(make-primref 'top-level-value)
|
|
(list (make-constant x)))])]
|
|
[else (error 'recordize "invalid expression" x)]))
|
|
(E x #f))
|
|
|
|
(define (unparse x)
|
|
(define (E-args proper x)
|
|
(if proper
|
|
(map E x)
|
|
(let f ([a (car x)] [d (cdr x)])
|
|
(cond
|
|
[(null? d) (E a)]
|
|
[else (cons (E a) (f (car d) (cdr d)))]))))
|
|
(define (E x)
|
|
(struct-case x
|
|
[(constant c) `(quote ,c)]
|
|
[(known x t) `(known ,(E x) ,(T:description t))]
|
|
[(code-loc x) `(code-loc ,x)]
|
|
[(var x) (string->symbol (format ":~a" x))]
|
|
[(prelex name) (string->symbol (format ":~a" name))]
|
|
[(primref x) x]
|
|
[(conditional test conseq altern)
|
|
`(if ,(E test) ,(E conseq) ,(E altern))]
|
|
[(interrupt-call e0 e1)
|
|
`(interrupt-call ,(E e0) ,(E e1))]
|
|
[(primcall op arg*) `(,op . ,(map E arg*))]
|
|
[(bind lhs* rhs* body)
|
|
`(let ,(map (lambda (lhs rhs) (list (E lhs) (E rhs))) lhs* rhs*)
|
|
,(E body))]
|
|
[(recbind lhs* rhs* body)
|
|
`(letrec ,(map (lambda (lhs rhs) (list (E lhs) (E rhs))) lhs* rhs*)
|
|
,(E body))]
|
|
[(rec*bind lhs* rhs* body)
|
|
`(letrec* ,(map (lambda (lhs rhs) (list (E lhs) (E rhs))) lhs* rhs*)
|
|
,(E body))]
|
|
;[(library-recbind lhs* loc* rhs* body)
|
|
; `(letrec ,(map (lambda (lhs loc rhs) (list (E lhs) loc (E rhs)))
|
|
; lhs* loc* rhs*)
|
|
; ,(E body))]
|
|
[(fix lhs* rhs* body)
|
|
`(fix ,(map (lambda (lhs rhs) (list (E lhs) (E rhs))) lhs* rhs*)
|
|
,(E body))]
|
|
[(seq e0 e1)
|
|
(let ()
|
|
(define (f x ac)
|
|
(struct-case x
|
|
[(seq e0 e1) (f e0 (f e1 ac))]
|
|
[else (cons (E x) ac)]))
|
|
(cons 'begin (f e0 (f e1 '()))))]
|
|
[(clambda-case info body)
|
|
`( ; label: ,(case-info-label info)
|
|
,(E-args (case-info-proper info) (case-info-args info))
|
|
,(E body))]
|
|
[(clambda g cls* cp free)
|
|
`(clambda (label: ,g) ; cp: ,(E cp) ) ;free: ,(map E free))
|
|
,@(map E cls*))]
|
|
[(clambda label clauses free)
|
|
`(code ,label . ,(map E clauses))]
|
|
[(closure code free* wk?)
|
|
`(closure ,@(if wk? '(wk) '()) ,(E code) ,(map E free*))]
|
|
[(codes list body)
|
|
`(codes ,(map E list)
|
|
,(E body))]
|
|
[(funcall rator rand*) `(funcall ,(E rator) . ,(map E rand*))]
|
|
[(jmpcall label rator rand*)
|
|
`(jmpcall ,label ,(E rator) . ,(map E rand*))]
|
|
[(forcall rator rand*) `(foreign-call ,rator . ,(map E rand*))]
|
|
[(assign lhs rhs) `(set! ,(E lhs) ,(E rhs))]
|
|
[(return x) `(return ,(E x))]
|
|
[(new-frame base-idx size body)
|
|
`(new-frame [base: ,base-idx]
|
|
[size: ,size]
|
|
,(E body))]
|
|
[(frame-var idx)
|
|
(string->symbol (format "fv.~a" idx))]
|
|
[(cp-var idx)
|
|
(string->symbol (format "cp.~a" idx))]
|
|
[(save-cp expr)
|
|
`(save-cp ,(E expr))]
|
|
[(eval-cp check body)
|
|
`(eval-cp ,check ,(E body))]
|
|
[(call-cp call-convention label save-cp? rp-convention base-idx arg-count live-mask)
|
|
`(call-cp [conv: ,call-convention]
|
|
[label: ,label]
|
|
[rpconv: ,(if (symbol? rp-convention)
|
|
rp-convention
|
|
(E rp-convention))]
|
|
[base-idx: ,base-idx]
|
|
[arg-count: ,arg-count]
|
|
[live-mask: ,live-mask])]
|
|
[(tailcall-cp convention label arg-count)
|
|
`(tailcall-cp ,convention ,label ,arg-count)]
|
|
[(foreign-label x) `(foreign-label ,x)]
|
|
[(mvcall prod cons) `(mvcall ,(E prod) ,(E cons))]
|
|
[(fvar idx) (string->symbol (format "fv.~a" idx))]
|
|
[(nfv idx) 'nfv]
|
|
[(locals vars body) `(locals ,(map E vars) ,(E body))]
|
|
[(asm-instr op d s)
|
|
`(asm ,op ,(E d) ,(E s))]
|
|
[(disp s0 s1)
|
|
`(disp ,(E s0) ,(E s1))]
|
|
[(nframe vars live body) `(nframe ;[vars: ,(map E vars)]
|
|
;[live: ,(map E live)]
|
|
,(E body))]
|
|
[(shortcut body handler)
|
|
`(shortcut ,(E body) ,(E handler))]
|
|
[(ntcall target valuw args mask size)
|
|
`(ntcall ,target ,size)]
|
|
[else
|
|
(if (symbol? x)
|
|
x
|
|
"#<unknown>")]))
|
|
(E x))
|
|
|
|
|
|
(define (unparse-pretty x)
|
|
(define n 0)
|
|
(define h (make-eq-hashtable))
|
|
(define (Var x)
|
|
(or (hashtable-ref h x #f)
|
|
(let ([v (string->symbol (format "~a_~a" (prelex-name x) n))])
|
|
(hashtable-set! h x v)
|
|
(set! n (+ n 1))
|
|
v)))
|
|
(define (map f ls)
|
|
(cond
|
|
[(null? ls) '()]
|
|
[else
|
|
(let ([a (f (car ls))])
|
|
(cons a (map f (cdr ls))))]))
|
|
(define (E-args proper x)
|
|
(if proper
|
|
(map Var x)
|
|
(let f ([a (car x)] [d (cdr x)])
|
|
(cond
|
|
[(null? d) (Var a)]
|
|
[else
|
|
(let ([a (Var a)])
|
|
(cons a (f (car d) (cdr d))))]))))
|
|
(define (clambda-clause x)
|
|
(struct-case x
|
|
[(clambda-case info body)
|
|
(let ([args (E-args (case-info-proper info) (case-info-args info)) ])
|
|
(list args (E body)))]))
|
|
(define (build-let b* body)
|
|
(cond
|
|
[(and (= (length b*) 1)
|
|
(pair? body)
|
|
(or (eq? (car body) 'let*)
|
|
(and (eq? (car body) 'let)
|
|
(= (length (cadr body)) 1))))
|
|
(list 'let* (append b* (cadr body)) (caddr body))]
|
|
[else
|
|
(list 'let b* body)]))
|
|
(define (E x)
|
|
(struct-case x
|
|
[(constant c) `(quote ,c)]
|
|
[(prelex) (Var x)]
|
|
[(primref x) x]
|
|
[(known x t) `(known ,(E x) ,(T:description t))]
|
|
[(conditional test conseq altern)
|
|
(cons 'if (map E (list test conseq altern)))]
|
|
[(primcall op arg*) (cons op (map E arg*))]
|
|
[(bind lhs* rhs* body)
|
|
(let* ([lhs* (map Var lhs*)]
|
|
[rhs* (map E rhs*)]
|
|
[body (E body)])
|
|
(import (only (ikarus) map))
|
|
(build-let (map list lhs* rhs*) body))]
|
|
[(fix lhs* rhs* body)
|
|
(let* ([lhs* (map Var lhs*)]
|
|
[rhs* (map E rhs*)]
|
|
[body (E body)])
|
|
(import (only (ikarus) map))
|
|
(list 'letrec (map list lhs* rhs*) body))]
|
|
[(seq e0 e1)
|
|
(cons 'begin
|
|
(let f ([e0 e0] [e* (list e1)])
|
|
(struct-case e0
|
|
[(seq e00 e01)
|
|
(f e00 (cons e01 e*))]
|
|
[else
|
|
(let ([x (E e0)])
|
|
(if (null? e*)
|
|
(list x)
|
|
(cons x (f (car e*) (cdr e*)))))])))]
|
|
[(clambda g cls* cp free)
|
|
(let ([cls* (map clambda-clause cls*)])
|
|
(cond
|
|
[(= (length cls*) 1) (cons 'lambda (car cls*))]
|
|
[else (cons 'case-lambda cls*)]))]
|
|
[(funcall rator rand*)
|
|
(let ([rator (E rator)])
|
|
(cons rator (map E rand*)))]
|
|
[(forcall rator rand*) `(foreign-call ,rator . ,(map E rand*))]
|
|
[(assign lhs rhs) `(set! ,(E lhs) ,(E rhs))]
|
|
[(foreign-label x) `(foreign-label ,x)]
|
|
[else x]))
|
|
(E x))
|
|
|
|
(define open-mvcalls (make-parameter #t))
|
|
|
|
(define (optimize-direct-calls x)
|
|
(define who 'optimize-direct-calls)
|
|
(define (make-conses ls)
|
|
(cond
|
|
[(null? ls) (make-constant '())]
|
|
[else
|
|
(make-funcall (make-primref 'cons)
|
|
(list (car ls) (make-conses (cdr ls))))]))
|
|
(define (properize lhs* rhs*)
|
|
(cond
|
|
[(null? lhs*) (error who "improper improper")]
|
|
[(null? (cdr lhs*))
|
|
(list (make-conses rhs*))]
|
|
[else (cons (car rhs*) (properize (cdr lhs*) (cdr rhs*)))]))
|
|
(define (inline-case cls rand*)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(struct-case info
|
|
[(case-info label fml* proper)
|
|
(if proper
|
|
(and (fx= (length fml*) (length rand*))
|
|
(make-bind fml* rand* body))
|
|
(and (fx<= (length fml*) (length rand*))
|
|
(make-bind fml* (properize fml* rand*) body)))])]))
|
|
(define (try-inline cls* rand* default)
|
|
(cond
|
|
[(null? cls*) default]
|
|
[(inline-case (car cls*) rand*)]
|
|
[else (try-inline (cdr cls*) rand* default)]))
|
|
(define (inline rator rand*)
|
|
(define (valid-mv-consumer? x)
|
|
(struct-case x
|
|
[(clambda L cases F)
|
|
(and (fx= (length cases) 1)
|
|
(struct-case (car cases)
|
|
[(clambda-case info body)
|
|
(struct-case info
|
|
[(case-info L args proper) proper])]))]
|
|
[else #f]))
|
|
(define (single-value-consumer? x)
|
|
(struct-case x
|
|
[(clambda L cases F)
|
|
(and (fx= (length cases) 1)
|
|
(struct-case (car cases)
|
|
[(clambda-case info body)
|
|
(struct-case info
|
|
[(case-info L args proper)
|
|
(and proper (fx= (length args) 1))])]))]
|
|
[else #f]))
|
|
(define (valid-mv-producer? x)
|
|
(struct-case x
|
|
[(funcall) #t]
|
|
[(conditional) #f]
|
|
[(bind lhs* rhs* body) (valid-mv-producer? body)]
|
|
[else #f] ;; FIXME BUG
|
|
))
|
|
(struct-case rator
|
|
[(clambda g cls*)
|
|
(try-inline cls* rand*
|
|
(make-funcall rator rand*))]
|
|
[(primref op)
|
|
(case op
|
|
;;; FIXME HERE
|
|
[(call-with-values)
|
|
(cond
|
|
[(and (open-mvcalls) (fx= (length rand*) 2))
|
|
(let ([producer (inline (car rand*) '())]
|
|
[consumer (cadr rand*)])
|
|
(cond
|
|
[(single-value-consumer? consumer)
|
|
(inline consumer (list producer))]
|
|
[(and (valid-mv-consumer? consumer)
|
|
(valid-mv-producer? producer))
|
|
(make-mvcall producer consumer)]
|
|
[else
|
|
(make-funcall rator rand*)]))]
|
|
[else
|
|
(make-funcall rator rand*)])]
|
|
[else
|
|
(make-funcall rator rand*)])]
|
|
[(bind lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(inline body rand*)
|
|
(make-bind lhs* rhs*
|
|
(call-expr body rand*)))]
|
|
[(recbind lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(inline body rand*)
|
|
(make-recbind lhs* rhs*
|
|
(call-expr body rand*)))]
|
|
[(rec*bind lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(inline body rand*)
|
|
(make-rec*bind lhs* rhs*
|
|
(call-expr body rand*)))]
|
|
[else (make-funcall rator rand*)]))
|
|
(define (call-expr x rand*)
|
|
(cond
|
|
[(clambda? x) (inline x rand*)]
|
|
[(and (prelex? x) (not (prelex-source-assigned? x)))
|
|
;;; FIXME: did we do the analysis yet?
|
|
(make-funcall x rand*)]
|
|
[else
|
|
(let ([t (make-prelex 'tmp #f)])
|
|
(set-prelex-source-referenced?! t #t)
|
|
(make-bind (list t) (list x)
|
|
(make-funcall t rand*)))]))
|
|
(define (Expr x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(prelex) (assert (prelex-source-referenced? x)) x]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(make-bind lhs* (map Expr rhs*) (Expr body))]
|
|
[(recbind lhs* rhs* body)
|
|
(make-recbind lhs* (map Expr rhs*) (Expr body))]
|
|
[(rec*bind lhs* rhs* body)
|
|
(make-rec*bind lhs* (map Expr rhs*) (Expr body))]
|
|
[(conditional test conseq altern)
|
|
(make-conditional
|
|
(Expr test)
|
|
(Expr conseq)
|
|
(Expr altern))]
|
|
[(seq e0 e1)
|
|
(make-seq (Expr e0) (Expr e1))]
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map (lambda (x)
|
|
(struct-case x
|
|
[(clambda-case info body)
|
|
(make-clambda-case info (Expr body))]))
|
|
cls*)
|
|
cp free name)]
|
|
[(funcall rator rand*)
|
|
(inline (Expr rator) (map Expr rand*))]
|
|
[(forcall rator rand*)
|
|
(make-forcall rator (map Expr rand*))]
|
|
[(assign lhs rhs)
|
|
(assert (prelex-source-assigned? lhs))
|
|
(make-assign lhs (Expr rhs))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(Expr x))
|
|
|
|
|
|
#|
|
|
(letrec* (bi ...
|
|
[x (let ([lhs* rhs*] ...) body)]
|
|
bj ...)
|
|
body)
|
|
===?
|
|
(letrec* (bi ...
|
|
[tmp* rhs*] ...
|
|
[lhs* tmp*] ...
|
|
[x body]
|
|
bj ...)
|
|
body)
|
|
|#
|
|
|
|
|
|
(define (optimize-letrec/scc x)
|
|
(define who 'optimize-letrec/scc)
|
|
(module (get-sccs-in-order)
|
|
(define-struct node (data link* lowlink root done collection))
|
|
(define (create-graph v* e** data*)
|
|
(define h (make-eq-hashtable))
|
|
(let ([v*
|
|
(let f ([v* v*] [data* data*])
|
|
(cond
|
|
[(null? v*) '()]
|
|
[else
|
|
(let ([node (make-node (car data*) '() #f #f #f #f)])
|
|
(hashtable-set! h (car v*) node)
|
|
(cons node (f (cdr v*) (cdr data*))))]))])
|
|
(for-each
|
|
(lambda (v e*)
|
|
(set-node-link*! v
|
|
(map (lambda (f)
|
|
(or (hashtable-ref h f #f)
|
|
(error who "invalid node" f)))
|
|
e*)))
|
|
v* e**)
|
|
v*))
|
|
(define (compute-sccs v*) ; Tarjan's algorithm
|
|
(define scc* '())
|
|
(define (compute-sccs v)
|
|
(define index 0)
|
|
(define stack '())
|
|
(define (tarjan v)
|
|
(let ([v-index index])
|
|
(set-node-root! v v-index)
|
|
(set! stack (cons v stack))
|
|
(set! index (fx+ index 1))
|
|
(for-each
|
|
(lambda (v^)
|
|
(unless (node-done v^)
|
|
(unless (node-root v^) (tarjan v^))
|
|
(set-node-root! v (fxmin (node-root v) (node-root v^)))))
|
|
(node-link* v))
|
|
(when (fx= (node-root v) v-index)
|
|
(set! scc*
|
|
(cons
|
|
(let f ([ls stack])
|
|
(let ([v^ (car ls)])
|
|
(set-node-done! v^ #t)
|
|
(cons v^ (if (eq? v^ v)
|
|
(begin (set! stack (cdr ls)) '())
|
|
(f (cdr ls))))))
|
|
scc*)))))
|
|
(tarjan v))
|
|
(for-each (lambda (v) (unless (node-done v) (compute-sccs v))) v*)
|
|
(reverse scc*))
|
|
(define (get-sccs-in-order n* e** data*)
|
|
(let ([G (create-graph n* e** data*)])
|
|
(let ([sccs (compute-sccs G)])
|
|
(map (lambda (scc) (map node-data scc)) sccs)))))
|
|
(define (gen-letrecs scc* ordered? body)
|
|
(define (mkfix b* body)
|
|
(if (null? b*)
|
|
body
|
|
(make-fix (map binding-lhs b*)
|
|
(map binding-rhs b*)
|
|
body)))
|
|
(define (gen-letrec scc fix* body)
|
|
(define (mklet lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
body
|
|
(make-bind lhs* rhs* body)))
|
|
(define (lambda-binding? x)
|
|
(and (not (prelex-source-assigned? (binding-lhs x)))
|
|
(clambda? (binding-rhs x))))
|
|
(define (mkset!s b* body)
|
|
(cond
|
|
[(null? b*) body]
|
|
[else
|
|
(let* ([b (car b*)]
|
|
[lhs (binding-lhs b)])
|
|
(unless (prelex-source-assigned? lhs)
|
|
;(printf "MADE COMPLEX ~s\n" (unparse lhs))
|
|
(set-prelex-source-assigned?! lhs
|
|
(or (prelex-global-location lhs) #t)))
|
|
(make-seq
|
|
(make-assign lhs (binding-rhs b))
|
|
(mkset!s (cdr b*) body)))]))
|
|
(cond
|
|
[(null? (cdr scc))
|
|
(let ([b (car scc)])
|
|
(cond
|
|
[(lambda-binding? b)
|
|
(values (cons b fix*) body)]
|
|
[(not (memq b (binding-free* b)))
|
|
(values '()
|
|
(mklet (list (binding-lhs b))
|
|
(list (binding-rhs b))
|
|
(mkfix fix* body)))]
|
|
[else
|
|
(values '()
|
|
(mklet (list (binding-lhs b))
|
|
(list (make-funcall (make-primref 'void) '()))
|
|
(mkset!s scc
|
|
(mkfix fix* body))))]))]
|
|
[else
|
|
(let-values ([(lambda* complex*)
|
|
(partition lambda-binding? scc)])
|
|
(cond
|
|
[(null? complex*)
|
|
(values (append lambda* fix*) body)]
|
|
[else
|
|
(let ([complex*
|
|
(if ordered? (sort-bindings complex*) complex*)])
|
|
(values '()
|
|
(mklet (map binding-lhs complex*)
|
|
(map (lambda (x)
|
|
(make-funcall (make-primref 'void) '()))
|
|
complex*)
|
|
(mkfix (append lambda* fix*)
|
|
(mkset!s complex* body)))))]))]))
|
|
(let-values ([(fix* body)
|
|
(let f ([scc* scc*])
|
|
(cond
|
|
[(null? scc*) (values '() body)]
|
|
[else
|
|
(let-values ([(fix* body) (f (cdr scc*))])
|
|
(gen-letrec (car scc*) fix* body))]))])
|
|
(mkfix fix* body)))
|
|
(define (do-recbind lhs* rhs* body bc ordered?)
|
|
(define (make-bindings lhs* rhs* bc i)
|
|
(cond
|
|
[(null? lhs*) '()]
|
|
[else
|
|
(let ([b (make-binding i (car lhs*) (car rhs*) #f bc '())])
|
|
(set-prelex-operand! (car lhs*) b)
|
|
(cons b (make-bindings (cdr lhs*) (cdr rhs*) bc (+ i 1))))]))
|
|
(define (complex? x)
|
|
(or (binding-complex x)
|
|
(prelex-source-assigned? (binding-lhs x))))
|
|
(define (insert-order-edges b*)
|
|
(define (mark pb b*)
|
|
(unless (null? b*)
|
|
(let ([b (car b*)])
|
|
(if (complex? b)
|
|
(let ([free* (binding-free* b)])
|
|
(unless (memq pb free*)
|
|
(set-binding-free*! b (cons pb free*)))
|
|
(mark b (cdr b*)))
|
|
(mark pb (cdr b*))))))
|
|
(unless (null? b*)
|
|
(let ([b (car b*)])
|
|
(if (complex? b)
|
|
(mark b (cdr b*))
|
|
(insert-order-edges (cdr b*))))))
|
|
(let ([b* (make-bindings lhs* rhs* bc 0)])
|
|
(for-each (lambda (b) (set-binding-rhs! b (E (binding-rhs b) b))) b*)
|
|
(for-each (lambda (x) (set-prelex-operand! x #f)) lhs*)
|
|
(let ([body (E body bc)])
|
|
(when ordered? (insert-order-edges b*))
|
|
(let ([scc* (get-sccs-in-order b* (map binding-free* b*) b*)])
|
|
;(printf "SCCS:\n")
|
|
;(for-each
|
|
; (lambda (scc)
|
|
; (printf " ~s\n"
|
|
; (map unparse (map binding-lhs scc))))
|
|
; scc*)
|
|
(gen-letrecs scc* ordered? body)))))
|
|
(define (sort-bindings ls)
|
|
(list-sort
|
|
(lambda (x y) (< (binding-serial x) (binding-serial y)))
|
|
ls))
|
|
(define-struct binding (serial lhs rhs complex prev free*))
|
|
(define (mark-complex bc)
|
|
(unless (binding-complex bc)
|
|
(set-binding-complex! bc #t)
|
|
(mark-complex (binding-prev bc))))
|
|
(define (mark-free var bc)
|
|
(let ([rb (prelex-operand var)])
|
|
(when rb
|
|
(let ([lb
|
|
(let ([pr (binding-prev rb)])
|
|
(let f ([bc bc])
|
|
(let ([bcp (binding-prev bc)])
|
|
(cond
|
|
[(eq? bcp pr) bc]
|
|
[else (f bcp)]))))])
|
|
(let ([free* (binding-free* lb)])
|
|
(unless (memq rb free*)
|
|
;(printf "MARK FREE ~s in ~s\n"
|
|
; (unparse (binding-lhs rb))
|
|
; (unparse (binding-lhs lb)))
|
|
(set-binding-free*! lb (cons rb free*))))))))
|
|
(define (E* x* bc)
|
|
(map (lambda (x) (E x bc)) x*))
|
|
(define (L x bc)
|
|
(struct-case x
|
|
[(clambda g cls* cp free name)
|
|
(let ([bc (make-binding #f #f #f #t bc '())])
|
|
(make-clambda g
|
|
(map (lambda (x)
|
|
(struct-case x
|
|
[(clambda-case info body)
|
|
(make-clambda-case info (E body bc))]))
|
|
cls*)
|
|
cp free name))]))
|
|
(define (E x bc)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(prelex)
|
|
(assert (prelex-source-referenced? x))
|
|
(mark-free x bc)
|
|
(when (prelex-source-assigned? x)
|
|
(mark-complex bc))
|
|
x]
|
|
[(assign lhs rhs)
|
|
(assert (prelex-source-assigned? lhs))
|
|
;(set-prelex-source-assigned?! lhs #t)
|
|
(mark-free lhs bc)
|
|
(mark-complex bc)
|
|
(make-assign lhs (E rhs bc))]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(E body bc)
|
|
(make-bind lhs* (E* rhs* bc) (E body bc)))]
|
|
[(recbind lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(E body bc)
|
|
(do-recbind lhs* rhs* body bc #f))]
|
|
[(rec*bind lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(E body bc)
|
|
(do-recbind lhs* rhs* body bc #t))]
|
|
[(conditional e0 e1 e2)
|
|
(make-conditional (E e0 bc) (E e1 bc) (E e2 bc))]
|
|
[(seq e0 e1) (make-seq (E e0 bc) (E e1 bc))]
|
|
[(clambda g cls* cp free name)
|
|
(L x bc)]
|
|
[(funcall rator rand*)
|
|
(mark-complex bc)
|
|
(make-funcall (E rator bc) (E* rand* bc))]
|
|
[(mvcall p c)
|
|
(mark-complex bc)
|
|
(make-mvcall (E p bc) (E c bc))]
|
|
[(forcall rator rand*)
|
|
(mark-complex bc)
|
|
(make-forcall rator (E* rand* bc))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
;(printf "===========================================\n")
|
|
(let ([x (E x (make-binding #f #f #f #t #t '()))])
|
|
;(pretty-print (unparse x))
|
|
x))
|
|
|
|
(include "ikarus.compiler.source-optimizer.ss")
|
|
|
|
(define (rewrite-assignments x)
|
|
(define who 'rewrite-assignments)
|
|
(define (fix-lhs* lhs*)
|
|
(cond
|
|
[(null? lhs*) (values '() '() '())]
|
|
[else
|
|
(let ([x (car lhs*)])
|
|
(let-values ([(lhs* a-lhs* a-rhs*) (fix-lhs* (cdr lhs*))])
|
|
(cond
|
|
[(and (prelex-source-assigned? x) (not (prelex-global-location x)))
|
|
(let ([t (make-prelex 'assignment-tmp #f)])
|
|
(set-prelex-source-referenced?! t #t)
|
|
(values (cons t lhs*) (cons x a-lhs*) (cons t a-rhs*)))]
|
|
[else
|
|
(values (cons x lhs*) a-lhs* a-rhs*)])))]))
|
|
(define (bind-assigned lhs* rhs* body)
|
|
(cond
|
|
[(null? lhs*) body]
|
|
[else
|
|
(make-bind lhs*
|
|
(map (lambda (rhs) (make-funcall (make-primref 'vector) (list rhs))) rhs*)
|
|
body)]))
|
|
(define (Expr x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(prelex)
|
|
(cond
|
|
[(prelex-source-assigned? x)
|
|
(cond
|
|
[(prelex-global-location x) =>
|
|
(lambda (loc)
|
|
(make-funcall
|
|
(make-primref '$symbol-value)
|
|
(list (make-constant loc))))]
|
|
[else
|
|
(make-funcall (make-primref '$vector-ref)
|
|
(list x (make-constant 0)))])]
|
|
[else x])]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(let-values ([(lhs* a-lhs* a-rhs*) (fix-lhs* lhs*)])
|
|
(make-bind lhs* (map Expr rhs*)
|
|
(bind-assigned a-lhs* a-rhs* (Expr body))))]
|
|
[(fix lhs* rhs* body)
|
|
(make-fix lhs* (map Expr rhs*) (Expr body))]
|
|
[(conditional test conseq altern)
|
|
(make-conditional (Expr test) (Expr conseq) (Expr altern))]
|
|
[(seq e0 e1) (make-seq (Expr e0) (Expr e1))]
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map (lambda (cls)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(struct-case info
|
|
[(case-info label fml* proper)
|
|
(let-values ([(fml* a-lhs* a-rhs*) (fix-lhs* fml*)])
|
|
(make-clambda-case
|
|
(make-case-info label fml* proper)
|
|
(bind-assigned a-lhs* a-rhs* (Expr body))))])]))
|
|
cls*)
|
|
cp free name)]
|
|
[(forcall op rand*)
|
|
(make-forcall op (map Expr rand*))]
|
|
[(funcall rator rand*)
|
|
(make-funcall (Expr rator) (map Expr rand*))]
|
|
[(assign lhs rhs)
|
|
(cond
|
|
[(prelex-source-assigned? lhs) =>
|
|
(lambda (where)
|
|
(cond
|
|
[(symbol? where)
|
|
(make-funcall (make-primref '$init-symbol-value!)
|
|
(list (make-constant where) (Expr rhs)))]
|
|
[(prelex-global-location lhs) =>
|
|
(lambda (loc)
|
|
(make-funcall (make-primref '$set-symbol-value!)
|
|
(list (make-constant loc) (Expr rhs))))]
|
|
[else
|
|
(make-funcall (make-primref '$vector-set!)
|
|
(list lhs (make-constant 0) (Expr rhs)))]))]
|
|
[else
|
|
(error 'rewrite-assignments "not assigned" lhs x)])]
|
|
[(mvcall p c) (make-mvcall (Expr p) (Expr c))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(Expr x))
|
|
|
|
(include "ikarus.compiler.tag-annotation-analysis.ss")
|
|
|
|
(define (introduce-vars x)
|
|
(define who 'introduce-vars)
|
|
(define (lookup x)
|
|
(let ([v (prelex-operand x)])
|
|
(assert (var? v))
|
|
v))
|
|
(define (convert-prelex x)
|
|
(assert (not (var? (prelex-operand x))))
|
|
(let ([v (unique-var (prelex-name x))])
|
|
(set-var-referenced! v (prelex-source-referenced? x))
|
|
(set-var-global-loc! v (prelex-global-location x))
|
|
(set-prelex-operand! x v)
|
|
v))
|
|
(define (A x)
|
|
(struct-case x
|
|
[(known x t) (make-known (E x) t)]
|
|
[else (E x)]))
|
|
(define (E x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(prelex) (lookup x)]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(let ([lhs* (map convert-prelex lhs*)])
|
|
(make-bind lhs* (map E rhs*) (E body)))]
|
|
[(fix lhs* rhs* body)
|
|
(let ([lhs* (map convert-prelex lhs*)])
|
|
(make-fix lhs* (map E rhs*) (E body)))]
|
|
[(conditional e0 e1 e2)
|
|
(make-conditional (E e0) (E e1) (E e2))]
|
|
[(seq e0 e1) (make-seq (E e0) (E e1))]
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map
|
|
(lambda (cls)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(struct-case info
|
|
[(case-info label args proper)
|
|
(let ([args (map convert-prelex args)])
|
|
(make-clambda-case
|
|
(make-case-info label args proper)
|
|
(E body)))])]))
|
|
cls*)
|
|
cp free name)]
|
|
[(primcall rator rand*)
|
|
(make-primcall rator (map A rand*))]
|
|
[(funcall rator rand*)
|
|
(make-funcall (A rator) (map A rand*))]
|
|
[(forcall rator rand*) (make-forcall rator (map E rand*))]
|
|
[(assign lhs rhs)
|
|
(make-assign (lookup lhs) (E rhs))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(E x))
|
|
|
|
(define (sanitize-bindings x)
|
|
(define who 'sanitize-bindings)
|
|
(define (CLambda x)
|
|
(struct-case x
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map (lambda (cls)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(struct-case info
|
|
[(case-info label fml* proper)
|
|
(make-clambda-case
|
|
(make-case-info label fml* proper)
|
|
(Expr body))])]))
|
|
cls*)
|
|
cp free name)]))
|
|
(define (do-fix lhs* rhs* body)
|
|
(if (null? lhs*)
|
|
(Expr body)
|
|
(make-fix lhs* (map CLambda rhs*) (Expr body))))
|
|
(define (A x)
|
|
(struct-case x
|
|
[(known x t) (make-known (Expr x) t)]
|
|
[else (Expr x)]))
|
|
(define (Expr x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(var) x]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(let-values ([(lambda* other*)
|
|
(partition
|
|
(lambda (x) (clambda? (cdr x)))
|
|
(map cons lhs* rhs*))])
|
|
(make-bind (map car other*)
|
|
(map Expr (map cdr other*))
|
|
(do-fix (map car lambda*) (map cdr lambda*)
|
|
body)))]
|
|
[(fix lhs* rhs* body)
|
|
(do-fix lhs* rhs* body)]
|
|
[(conditional test conseq altern)
|
|
(make-conditional (Expr test) (Expr conseq) (Expr altern))]
|
|
[(seq e0 e1) (make-seq (Expr e0) (Expr e1))]
|
|
[(clambda g cls* cp free name)
|
|
(let ([t (unique-var 'anon)])
|
|
(make-fix (list t) (list (CLambda x)) t))]
|
|
[(forcall op rand*)
|
|
(make-forcall op (map Expr rand*))]
|
|
[(funcall rator rand*)
|
|
(make-funcall (A rator) (map A rand*))]
|
|
[(mvcall p c) (make-mvcall (Expr p) (Expr c))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(Expr x))
|
|
|
|
|
|
(define (untag x)
|
|
(struct-case x
|
|
[(known x t) (values x t)]
|
|
[else (values x #f)]))
|
|
|
|
(define (tag x t)
|
|
(if t
|
|
(make-known x t)
|
|
x))
|
|
|
|
(define (optimize-for-direct-jumps x)
|
|
(define who 'optimize-for-direct-jumps)
|
|
(define (init-var x)
|
|
(set-var-referenced! x #f))
|
|
(define (set-var x v)
|
|
(struct-case v
|
|
[(clambda) (set-var-referenced! x v)]
|
|
[(var)
|
|
(cond
|
|
[(bound-var v) => (lambda (v) (set-var-referenced! x v))]
|
|
[else (void)])]
|
|
[else (void)]))
|
|
(define (bound-var x)
|
|
(var-referenced x))
|
|
(define (optimize c rator rand*)
|
|
(let ([n (length rand*)])
|
|
(struct-case c
|
|
[(clambda main-label cls*)
|
|
(let f ([cls* cls*])
|
|
(cond
|
|
[(null? cls*)
|
|
;;; none matching?
|
|
(make-funcall rator rand*)]
|
|
[else
|
|
(struct-case (clambda-case-info (car cls*))
|
|
[(case-info label fml* proper)
|
|
(cond
|
|
[proper
|
|
(if (fx= n (length fml*))
|
|
(make-jmpcall label (strip rator) (map strip rand*))
|
|
(f (cdr cls*)))]
|
|
[else
|
|
(if (fx<= (length (cdr fml*)) n)
|
|
(make-jmpcall label (strip rator)
|
|
(let f ([fml* (cdr fml*)] [rand* rand*])
|
|
(cond
|
|
[(null? fml*)
|
|
;;; FIXME: construct list afterwards
|
|
(list (make-funcall (make-primref 'list) rand*))]
|
|
[else
|
|
(cons (strip (car rand*))
|
|
(f (cdr fml*) (cdr rand*)))])))
|
|
(f (cdr cls*)))])])]))])))
|
|
(define (strip x)
|
|
(struct-case x
|
|
[(known x t) x]
|
|
[else x]))
|
|
(define (CLambda x)
|
|
(struct-case x
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map (lambda (cls)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(for-each init-var (case-info-args info))
|
|
(make-clambda-case info (Expr body))]))
|
|
cls*)
|
|
cp free name)]))
|
|
(define (A x)
|
|
(struct-case x
|
|
[(known x t) (make-known (Expr x) t)]
|
|
[else (Expr x)]))
|
|
(define (A- x)
|
|
(struct-case x
|
|
[(known x t) (Expr x)]
|
|
[else (Expr x)]))
|
|
(define (Expr x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(var) x]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(for-each init-var lhs*)
|
|
(let ([rhs* (map Expr rhs*)])
|
|
(for-each set-var lhs* rhs*)
|
|
(make-bind lhs* rhs* (Expr body)))]
|
|
[(fix lhs* rhs* body)
|
|
(for-each set-var lhs* rhs*)
|
|
(make-fix lhs* (map CLambda rhs*) (Expr body))]
|
|
[(conditional test conseq altern)
|
|
(make-conditional (Expr test) (Expr conseq) (Expr altern))]
|
|
[(seq e0 e1) (make-seq (Expr e0) (Expr e1))]
|
|
[(forcall op rand*)
|
|
(make-forcall op (map Expr rand*))]
|
|
[(funcall rator rand*)
|
|
(let-values ([(rator t) (untag (A rator))])
|
|
(cond
|
|
[(and (var? rator) (bound-var rator)) =>
|
|
(lambda (c)
|
|
(optimize c rator (map A rand*)))]
|
|
[(and (primref? rator)
|
|
(eq? (primref-name rator) '$$apply))
|
|
(make-jmpcall (sl-apply-label)
|
|
(A- (car rand*))
|
|
(map A- (cdr rand*)))]
|
|
[else
|
|
(make-funcall (tag rator t) (map A rand*))]))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(Expr x))
|
|
|
|
|
|
(define (insert-global-assignments x)
|
|
(define who 'insert-global-assignments)
|
|
(define (global-assign lhs* body)
|
|
(cond
|
|
[(null? lhs*) body]
|
|
[(var-global-loc (car lhs*)) =>
|
|
(lambda (loc)
|
|
(make-seq
|
|
(make-funcall (make-primref '$init-symbol-value!)
|
|
(list (make-constant loc) (car lhs*)))
|
|
(global-assign (cdr lhs*) body)))]
|
|
[else (global-assign (cdr lhs*) body)]))
|
|
(define (global-fix lhs* body)
|
|
(cond
|
|
[(null? lhs*) body]
|
|
[(var-global-loc (car lhs*)) =>
|
|
(lambda (loc)
|
|
(make-seq
|
|
(make-funcall (make-primref '$set-symbol-value/proc!)
|
|
(list (make-constant loc) (car lhs*)))
|
|
(global-assign (cdr lhs*) body)))]
|
|
[else (global-assign (cdr lhs*) body)]))
|
|
(define (A x)
|
|
(struct-case x
|
|
[(known x t) (make-known (Expr x) t)]
|
|
[else (Expr x)]))
|
|
(define (Expr x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(var)
|
|
(cond
|
|
[(var-global-loc x) =>
|
|
(lambda (loc)
|
|
(make-funcall
|
|
(make-primref '$symbol-value)
|
|
(list (make-constant loc))))]
|
|
[else x])]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(make-bind lhs* (map Expr rhs*)
|
|
(global-assign lhs* (Expr body)))]
|
|
[(fix lhs* rhs* body)
|
|
(make-fix lhs* (map Expr rhs*)
|
|
(global-fix lhs* (Expr body)))]
|
|
[(conditional test conseq altern)
|
|
(make-conditional (Expr test) (Expr conseq) (Expr altern))]
|
|
[(seq e0 e1) (make-seq (Expr e0) (Expr e1))]
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map (lambda (cls)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(make-clambda-case info (Expr body))]))
|
|
cls*)
|
|
cp free name)]
|
|
[(forcall op rand*)
|
|
(make-forcall op (map Expr rand*))]
|
|
[(funcall rator rand*)
|
|
(make-funcall (A rator) (map A rand*))]
|
|
[(jmpcall label rator rand*)
|
|
(make-jmpcall label (Expr rator) (map Expr rand*))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(define (AM x)
|
|
(struct-case x
|
|
[(known x t) (make-known (Main x) t)]
|
|
[else (Main x)]))
|
|
(define (Main x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(var) x]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body)
|
|
(make-bind lhs* (map Main rhs*)
|
|
(global-assign lhs* (Main body)))]
|
|
[(fix lhs* rhs* body)
|
|
(make-fix lhs* (map Main rhs*)
|
|
(global-fix lhs* (Main body)))]
|
|
[(conditional test conseq altern)
|
|
(make-conditional (Main test) (Main conseq) (Main altern))]
|
|
[(seq e0 e1) (make-seq (Main e0) (Main e1))]
|
|
[(clambda g cls* cp free name)
|
|
(make-clambda g
|
|
(map (lambda (cls)
|
|
(struct-case cls
|
|
[(clambda-case info body)
|
|
(make-clambda-case info (Expr body))]))
|
|
cls*)
|
|
cp free name)]
|
|
[(forcall op rand*)
|
|
(make-forcall op (map Main rand*))]
|
|
[(funcall rator rand*)
|
|
(make-funcall (AM rator) (map AM rand*))]
|
|
[(jmpcall label rator rand*)
|
|
(make-jmpcall label (Main rator) (map Main rand*))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
(let ([x (Main x)])
|
|
;(pretty-print x)
|
|
x))
|
|
|
|
|
|
|
|
(define optimize-cp (make-parameter #t))
|
|
|
|
(define (convert-closures prog)
|
|
(define who 'convert-closures)
|
|
(define (Expr* x*)
|
|
(cond
|
|
[(null? x*) (values '() '())]
|
|
[else
|
|
(let-values ([(a a-free) (Expr (car x*))]
|
|
[(d d-free) (Expr* (cdr x*))])
|
|
(values (cons a d) (union a-free d-free)))]))
|
|
(define (do-clambda* lhs* x*)
|
|
(cond
|
|
[(null? x*) (values '() '())]
|
|
[else
|
|
(let-values ([(a a-free) (do-clambda (car lhs*) (car x*))]
|
|
[(d d-free) (do-clambda* (cdr lhs*) (cdr x*))])
|
|
(values (cons a d) (union a-free d-free)))]))
|
|
(define (do-clambda lhs x)
|
|
(struct-case x
|
|
[(clambda g cls* _cp _free name)
|
|
(let-values ([(cls* free)
|
|
(let f ([cls* cls*])
|
|
(cond
|
|
[(null? cls*) (values '() '())]
|
|
[else
|
|
(struct-case (car cls*)
|
|
[(clambda-case info body)
|
|
(let-values ([(body body-free) (Expr body)]
|
|
[(cls* cls*-free) (f (cdr cls*))])
|
|
(values
|
|
(cons (make-clambda-case info body) cls*)
|
|
(union (difference body-free (case-info-args info))
|
|
cls*-free)))])]))])
|
|
(values
|
|
(make-closure
|
|
(make-clambda g cls* lhs free name)
|
|
free
|
|
#f)
|
|
free))]))
|
|
(define (A x)
|
|
(struct-case x
|
|
[(known x t)
|
|
(let-values ([(x free) (Expr x)])
|
|
(values (make-known x t) free))]
|
|
[else (Expr x)]))
|
|
(define (A* x*)
|
|
(cond
|
|
[(null? x*) (values '() '())]
|
|
[else
|
|
(let-values ([(a a-free) (A (car x*))]
|
|
[(d d-free) (A* (cdr x*))])
|
|
(values (cons a d) (union a-free d-free)))]))
|
|
(define (Expr ex)
|
|
(struct-case ex
|
|
[(constant) (values ex '())]
|
|
[(var)
|
|
(set-var-index! ex #f)
|
|
(values ex (singleton ex))]
|
|
[(primref) (values ex '())]
|
|
[(bind lhs* rhs* body)
|
|
(let-values ([(rhs* rhs-free) (Expr* rhs*)]
|
|
[(body body-free) (Expr body)])
|
|
(values (make-bind lhs* rhs* body)
|
|
(union rhs-free (difference body-free lhs*))))]
|
|
[(fix lhs* rhs* body)
|
|
(for-each (lambda (x) (set-var-index! x #t)) lhs*)
|
|
(let-values ([(rhs* rfree) (do-clambda* lhs* rhs*)]
|
|
[(body bfree) (Expr body)])
|
|
(for-each
|
|
(lambda (lhs rhs)
|
|
(when (var-index lhs)
|
|
(set-closure-well-known?! rhs #t)
|
|
(set-var-index! lhs #f)))
|
|
lhs* rhs*)
|
|
(values (make-fix lhs* rhs* body)
|
|
(difference (union bfree rfree) lhs*)))]
|
|
[(conditional test conseq altern)
|
|
(let-values ([(test test-free) (Expr test)]
|
|
[(conseq conseq-free) (Expr conseq)]
|
|
[(altern altern-free) (Expr altern)])
|
|
(values (make-conditional test conseq altern)
|
|
(union test-free (union conseq-free altern-free))))]
|
|
[(seq e0 e1)
|
|
(let-values ([(e0 e0-free) (Expr e0)]
|
|
[(e1 e1-free) (Expr e1)])
|
|
(values (make-seq e0 e1) (union e0-free e1-free)))]
|
|
[(forcall op rand*)
|
|
(let-values ([(rand* rand*-free) (Expr* rand*)])
|
|
(values (make-forcall op rand*) rand*-free))]
|
|
[(funcall rator rand*)
|
|
(let-values ([(rator rat-free) (A rator)]
|
|
[(rand* rand*-free) (A* rand*)])
|
|
(values (make-funcall rator rand*)
|
|
(union rat-free rand*-free)))]
|
|
[(jmpcall label rator rand*)
|
|
(let-values ([(rator rat-free)
|
|
(if (optimize-cp) (Rator rator) (Expr rator))]
|
|
[(rand* rand*-free)
|
|
(A* rand*)])
|
|
(values (make-jmpcall label rator rand*)
|
|
(union rat-free rand*-free)))]
|
|
[else (error who "invalid expression" ex)]))
|
|
(define (Rator x)
|
|
(struct-case x
|
|
[(var) (values x (singleton x))]
|
|
;[(known x t)
|
|
; (let-values ([(x free) (Rator x)])
|
|
; (values (make-known x t) free))]
|
|
[else (Expr x)]))
|
|
(let-values ([(prog free) (Expr prog)])
|
|
(unless (null? free)
|
|
(error 'convert-closures "free vars encountered in program"
|
|
(map unparse free)))
|
|
prog))
|
|
|
|
|
|
|
|
(define (optimize-closures/lift-codes x)
|
|
(define who 'optimize-closures/lift-codes)
|
|
(define all-codes '())
|
|
(module (unset! set-subst! get-subst copy-subst!)
|
|
(define-struct prop (val))
|
|
(define (unset! x)
|
|
(unless (var? x) (error 'unset! "not a var" x))
|
|
(set-var-index! x #f))
|
|
(define (set-subst! x v)
|
|
(unless (var? x) (error 'set-subst! "not a var" x))
|
|
(set-var-index! x (make-prop v)))
|
|
(define (copy-subst! lhs rhs)
|
|
(unless (var? lhs) (error 'copy-subst! "not a var" lhs))
|
|
(cond
|
|
[(and (var? rhs) (var-index rhs)) =>
|
|
(lambda (v)
|
|
(cond
|
|
[(prop? v) (set-var-index! lhs v)]
|
|
[else (set-var-index! lhs #f)]))]
|
|
[else (set-var-index! lhs #f)]))
|
|
(define (get-subst x)
|
|
(unless (var? x) (error 'get-subst "not a var" x))
|
|
(struct-case (var-index x)
|
|
[(prop v) v]
|
|
[else #f])))
|
|
(define (combinator? x)
|
|
(struct-case x
|
|
[(closure code free*)
|
|
(null? free*)]
|
|
[else #f]))
|
|
(define (lift-code cp code free*)
|
|
(struct-case code
|
|
[(clambda label cls* cp/dropped free*/dropped name)
|
|
(let ([cls* (map
|
|
(lambda (x)
|
|
(struct-case x
|
|
[(clambda-case info body)
|
|
(for-each unset! (case-info-args info))
|
|
(make-clambda-case info (E body))]))
|
|
cls*)])
|
|
(let ([g (make-code-loc label)])
|
|
(set! all-codes
|
|
(cons (make-clambda label cls* cp free* name)
|
|
all-codes))
|
|
g))]))
|
|
(define (trim p? ls)
|
|
(cond
|
|
[(null? ls) '()]
|
|
[(p? (car ls)) (trim p? (cdr ls))]
|
|
[else
|
|
(cons (car ls) (trim p? (cdr ls)))]))
|
|
(define (do-bind lhs* rhs* body)
|
|
(for-each unset! lhs*)
|
|
(let ([rhs* (map E rhs*)])
|
|
(for-each copy-subst! lhs* rhs*)
|
|
(let ([body (E body)])
|
|
(for-each unset! lhs*)
|
|
(make-bind lhs* rhs* body))))
|
|
(define (trim-free ls)
|
|
(cond
|
|
[(null? ls) '()]
|
|
[(get-forward! (car ls)) =>
|
|
(lambda (what)
|
|
(let ([rest (trim-free (cdr ls))])
|
|
(struct-case what
|
|
[(closure) rest]
|
|
[(var) (if (memq what rest) rest (cons what rest))]
|
|
[else (error who "invalid value in trim-free" what)])))]
|
|
[else (cons (car ls) (trim-free (cdr ls)))]))
|
|
(define (do-fix lhs* rhs* body)
|
|
(for-each unset! lhs*)
|
|
(let ([free** ;;; trim the free lists first; after init.
|
|
(map (lambda (lhs rhs)
|
|
;;; remove self also
|
|
(remq lhs (trim-free (closure-free* rhs))))
|
|
lhs* rhs*)])
|
|
(define-struct node (name code deps whacked free wk?))
|
|
(let ([node*
|
|
(map (lambda (lhs rhs)
|
|
(let ([n (make-node lhs (closure-code rhs) '() #f '()
|
|
(closure-well-known? rhs))])
|
|
(set-subst! lhs n)
|
|
n))
|
|
lhs* rhs*)])
|
|
;;; if x is free in y, then whenever x becomes a non-combinator,
|
|
;;; y also becomes a non-combinator. Here, we mark these
|
|
;;; dependencies.
|
|
(for-each
|
|
(lambda (my-node free*)
|
|
(for-each (lambda (fvar)
|
|
(cond
|
|
[(get-subst fvar) => ;;; one of ours
|
|
(lambda (her-node)
|
|
(set-node-deps! her-node
|
|
(cons my-node (node-deps her-node))))]
|
|
[else ;;; not one of ours
|
|
(set-node-free! my-node
|
|
(cons fvar (node-free my-node)))]))
|
|
free*))
|
|
node* free**)
|
|
;;; Next, we go over the list of nodes, and if we find one
|
|
;;; that has any free variables, we know it's a non-combinator,
|
|
;;; so we whack it and add it to all of its dependents.
|
|
(let ()
|
|
(define (process-node x)
|
|
(when (cond
|
|
[(null? (node-free x)) #f]
|
|
;[(and (node-wk? x) (null? (cdr (node-free x)))) #f]
|
|
[else #t])
|
|
(unless (node-whacked x)
|
|
(set-node-whacked! x #t)
|
|
(for-each
|
|
(lambda (y)
|
|
(set-node-free! y
|
|
(cons (node-name x) (node-free y)))
|
|
(process-node y))
|
|
(node-deps x)))))
|
|
(for-each process-node node*))
|
|
;;; Now those that have free variables are actual closures.
|
|
;;; Those with no free variables are actual combinators.
|
|
(let ([rhs*
|
|
(map
|
|
(lambda (node)
|
|
(let ([wk? (node-wk? node)]
|
|
[name (node-name node)]
|
|
[free (node-free node)])
|
|
(let ([closure
|
|
(make-closure (node-code node) free wk?)])
|
|
(cond
|
|
[(null? free)
|
|
(set-subst! name closure)]
|
|
[(and (null? (cdr free)) wk?)
|
|
(set-subst! name closure)]
|
|
[else
|
|
(unset! name)])
|
|
closure)))
|
|
node*)])
|
|
(for-each
|
|
(lambda (lhs^ closure)
|
|
(let* ([lhs (get-forward! lhs^)]
|
|
[free
|
|
(filter var?
|
|
(remq lhs (trim-free (closure-free* closure))))])
|
|
(set-closure-free*! closure free)
|
|
(set-closure-code! closure
|
|
(lift-code
|
|
lhs
|
|
(closure-code closure)
|
|
(closure-free* closure)))))
|
|
lhs*
|
|
rhs*)
|
|
(let ([body (E body)])
|
|
(let f ([lhs* lhs*] [rhs* rhs*] [l* '()] [r* '()])
|
|
(cond
|
|
[(null? lhs*)
|
|
(if (null? l*)
|
|
body
|
|
(make-fix l* r* body))]
|
|
[else
|
|
(let ([lhs (car lhs*)] [rhs (car rhs*)])
|
|
(cond
|
|
[(get-subst lhs)
|
|
(unset! lhs)
|
|
(f (cdr lhs*) (cdr rhs*) l* r*)]
|
|
[else
|
|
(f (cdr lhs*) (cdr rhs*)
|
|
(cons lhs l*) (cons rhs r*))]))])))))))
|
|
(define (get-forward! x)
|
|
(when (eq? x 'q)
|
|
(error who "BUG: circular dep"))
|
|
(let ([y (get-subst x)])
|
|
(cond
|
|
[(not y) x]
|
|
[(var? y)
|
|
(set-subst! x 'q)
|
|
(let ([y (get-forward! y)])
|
|
(set-subst! x y)
|
|
y)]
|
|
[(closure? y)
|
|
(let ([free (closure-free* y)])
|
|
(cond
|
|
[(null? free) y]
|
|
[(null? (cdr free))
|
|
(set-subst! x 'q)
|
|
(let ([y (get-forward! (car free))])
|
|
(set-subst! x y)
|
|
y)]
|
|
[else y]))]
|
|
[else x])))
|
|
(define (A x)
|
|
(struct-case x
|
|
[(known x t) (make-known (E x) t)]
|
|
[else (E x)]))
|
|
(define (E x)
|
|
(struct-case x
|
|
[(constant) x]
|
|
[(var) (get-forward! x)]
|
|
[(primref) x]
|
|
[(bind lhs* rhs* body) (do-bind lhs* rhs* body)]
|
|
[(fix lhs* rhs* body) (do-fix lhs* rhs* body)]
|
|
[(conditional test conseq altern)
|
|
(make-conditional (E test) (E conseq) (E altern))]
|
|
[(seq e0 e1) (make-seq (E e0) (E e1))]
|
|
[(forcall op rand*) (make-forcall op (map E rand*))]
|
|
[(funcall rator rand*) (make-funcall (A rator) (map A rand*))]
|
|
[(jmpcall label rator rand*)
|
|
(make-jmpcall label (E rator) (map E rand*))]
|
|
[else (error who "invalid expression" (unparse x))]))
|
|
;(when (optimize-cp)
|
|
; (printf "BEFORE\n")
|
|
; (parameterize ([pretty-width 200])
|
|
; (pretty-print (unparse x))))
|
|
(let ([x (E x)])
|
|
(let ([v (make-codes all-codes x)])
|
|
;(when (optimize-cp)
|
|
; (printf "AFTER\n")
|
|
; (parameterize ([pretty-width 200])
|
|
; (pretty-print (unparse v))))
|
|
v)))
|
|
|
|
|
|
|
|
(begin ;;; DEFINITIONS
|
|
(module (wordsize)
|
|
(include "ikarus.config.ss"))
|
|
(define wordshift
|
|
(case wordsize
|
|
[(4) 2]
|
|
[(8) 3]
|
|
[else
|
|
(error 'ikarus "wordsize is neither 4 nor 8" wordsize)]))
|
|
(define fx-scale wordsize)
|
|
(define object-alignment (* 2 wordsize))
|
|
(define align-shift (+ wordshift 1))
|
|
(define fx-shift wordshift)
|
|
(define fx-mask (- wordsize 1))
|
|
(define fx-tag 0)
|
|
(define bool-f #x2F)
|
|
(define bool-t #x3F)
|
|
(define bool-mask #xEF)
|
|
(define bool-tag #x2F)
|
|
(define bool-shift 4)
|
|
(define nil #x4F)
|
|
(define eof #x5F) ; double check
|
|
(define unbound #x6F) ; double check
|
|
(define void-object #x7F) ; double check
|
|
(define bwp-object #x8F) ; double check
|
|
(define char-size 4)
|
|
(define char-shift 8)
|
|
(define char-tag #x0F)
|
|
(define char-mask #xFF)
|
|
(define pair-mask 7)
|
|
(define pair-tag 1)
|
|
(define disp-car 0)
|
|
(define disp-cdr wordsize)
|
|
(define pair-size (* 2 wordsize))
|
|
|
|
(define flonum-tag #x17)
|
|
(define flonum-size 16)
|
|
(define disp-flonum-data 8)
|
|
|
|
(define ratnum-tag #x27)
|
|
(define disp-ratnum-num (* 1 wordsize))
|
|
(define disp-ratnum-den (* 2 wordsize))
|
|
(define ratnum-size (* 4 wordsize))
|
|
|
|
(define compnum-tag #x37)
|
|
(define disp-compnum-real (* 1 wordsize))
|
|
(define disp-compnum-imag (* 2 wordsize))
|
|
(define compnum-size (* 4 wordsize))
|
|
|
|
(define cflonum-tag #x47)
|
|
(define disp-cflonum-real (* 1 wordsize))
|
|
(define disp-cflonum-imag (* 2 wordsize))
|
|
(define cflonum-size (* 4 wordsize))
|
|
|
|
(define bignum-mask #b111)
|
|
(define bignum-tag #b011)
|
|
(define bignum-sign-mask #b1000)
|
|
(define bignum-sign-shift 3)
|
|
(define bignum-length-shift 4)
|
|
(define disp-bignum-data wordsize)
|
|
|
|
(define pagesize 4096)
|
|
(define pageshift 12)
|
|
|
|
(define bytevector-mask 7)
|
|
(define bytevector-tag 2)
|
|
(define disp-bytevector-length 0)
|
|
(define disp-bytevector-data 8)
|
|
|
|
(define ptag-mask 7)
|
|
(define symbol-ptag 5)
|
|
(define symbol-record-tag #x5F)
|
|
(define disp-symbol-record-string (* 1 wordsize))
|
|
(define disp-symbol-record-ustring (* 2 wordsize))
|
|
(define disp-symbol-record-value (* 3 wordsize))
|
|
(define disp-symbol-record-proc (* 4 wordsize))
|
|
(define disp-symbol-record-plist (* 5 wordsize))
|
|
(define symbol-record-size (* 6 wordsize))
|
|
|
|
(define record-tag 5)
|
|
(define record-mask 7)
|
|
|
|
(define vector-tag 5)
|
|
(define vector-mask 7)
|
|
(define disp-vector-length 0)
|
|
(define disp-vector-data wordsize)
|
|
(define string-mask 7)
|
|
(define string-tag 6)
|
|
(define disp-string-length 0)
|
|
(define disp-string-data wordsize)
|
|
(define closure-mask 7)
|
|
(define closure-tag 3)
|
|
(define disp-closure-code 0)
|
|
(define disp-closure-data wordsize)
|
|
(define continuation-tag #x1F)
|
|
(define disp-continuation-top (* 1 wordsize))
|
|
(define disp-continuation-size (* 2 wordsize))
|
|
(define disp-continuation-next (* 3 wordsize))
|
|
(define continuation-size (* 4 wordsize))
|
|
(define code-tag #x2F)
|
|
(define disp-code-instrsize (* 1 wordsize))
|
|
(define disp-code-relocsize (* 2 wordsize))
|
|
(define disp-code-freevars (* 3 wordsize))
|
|
(define disp-code-annotation (* 4 wordsize))
|
|
(define disp-code-unused (* 5 wordsize))
|
|
(define disp-code-data (* 6 wordsize))
|
|
|
|
(define transcoder-mask #xFF) ;;; 0011
|
|
(define transcoder-tag #x7F) ;;; 0011
|
|
(define transcoder-payload-shift 10)
|
|
|
|
(define transcoder-write-utf8-mask #x1000)
|
|
(define transcoder-write-byte-mask #x2000)
|
|
(define transcoder-read-utf8-mask #x4000)
|
|
(define transcoder-read-byte-mask #x8000)
|
|
(define transcoder-handling-mode-shift 16)
|
|
(define transcoder-handling-mode-bits 2)
|
|
(define transcoder-eol-style-shift 18)
|
|
(define transcoder-eol-style-bits 3)
|
|
(define transcoder-codec-shift 21)
|
|
(define transcoder-codec-bits 3)
|
|
|
|
(define transcoder-handling-mode:none #b00)
|
|
(define transcoder-handling-mode:ignore #b01)
|
|
(define transcoder-handling-mode:raise #b10)
|
|
(define transcoder-handling-mode:replace #b11)
|
|
|
|
(define transcoder-eol-style:none #b000)
|
|
(define transcoder-eol-style:lf #b001)
|
|
(define transcoder-eol-style:cr #b010)
|
|
(define transcoder-eol-style:crlf #b011)
|
|
(define transcoder-eol-style:nel #b100)
|
|
(define transcoder-eol-style:crnel #b101)
|
|
(define transcoder-eol-style:ls #b110)
|
|
|
|
(define transcoder-codec:none #b000)
|
|
(define transcoder-codec:latin-1 #b001)
|
|
(define transcoder-codec:utf-8 #b010)
|
|
(define transcoder-codec:utf-16 #b011)
|
|
|
|
(define port-tag #x3F)
|
|
(define port-mask #x3F)
|
|
(define disp-port-attrs 0)
|
|
(define disp-port-index (* 1 wordsize))
|
|
(define disp-port-size (* 2 wordsize))
|
|
(define disp-port-buffer (* 3 wordsize))
|
|
(define disp-port-transcoder (* 4 wordsize))
|
|
(define disp-port-id (* 5 wordsize))
|
|
(define disp-port-read! (* 6 wordsize))
|
|
(define disp-port-write! (* 7 wordsize))
|
|
(define disp-port-get-position (* 8 wordsize))
|
|
(define disp-port-set-position! (* 9 wordsize))
|
|
(define disp-port-close (* 10 wordsize))
|
|
(define disp-port-cookie (* 11 wordsize))
|
|
(define disp-port-position (* 12 wordsize))
|
|
(define disp-port-unused (* 13 wordsize))
|
|
(define port-size (* 14 wordsize))
|
|
|
|
(define disp-tcbucket-tconc 0)
|
|
(define disp-tcbucket-key (* 1 wordsize))
|
|
(define disp-tcbucket-val (* 2 wordsize))
|
|
(define disp-tcbucket-next (* 3 wordsize))
|
|
(define tcbucket-size (* 4 wordsize))
|
|
(define record-ptag 5)
|
|
(define record-pmask 7)
|
|
(define disp-struct-rtd 0)
|
|
(define disp-struct-data wordsize)
|
|
|
|
;;; refer to the picture in src/ikarus-collect.c for details
|
|
;;; on how call-frames are laid out. (search for livemask)
|
|
(define call-instruction-size
|
|
(case wordsize
|
|
[(4) 5]
|
|
[(8) 10]
|
|
[else (die 'call-instruction-size "invalid" wordsize)]))
|
|
(define disp-frame-size (- (+ call-instruction-size (* 3 wordsize))))
|
|
(define disp-frame-offset (- (+ call-instruction-size (* 2 wordsize))))
|
|
(define disp-multivalue-rp (- (+ call-instruction-size (* 1 wordsize))))
|
|
|
|
(define dirty-word -1))
|
|
|
|
;(define pcb-allocation-pointer (* 0 wordsize)) NOT USED
|
|
(define pcb-allocation-redline (* 1 wordsize))
|
|
;(define pcb-frame-pointer (* 2 wordsize)) NOT USED
|
|
(define pcb-frame-base (* 3 wordsize))
|
|
(define pcb-frame-redline (* 4 wordsize))
|
|
(define pcb-next-continuation (* 5 wordsize))
|
|
;(define pcb-system-stack (* 6 wordsize)) NOT USED
|
|
(define pcb-dirty-vector (* 7 wordsize))
|
|
(define pcb-arg-list (* 8 wordsize))
|
|
(define pcb-engine-counter (* 9 wordsize))
|
|
(define pcb-interrupted (* 10 wordsize))
|
|
(define pcb-base-rtd (* 11 wordsize))
|
|
(define pcb-collect-key (* 12 wordsize))
|
|
|
|
|
|
(define (fx? x)
|
|
(let* ([intbits (* wordsize 8)]
|
|
[fxbits (- intbits fx-shift)])
|
|
(and (or (fixnum? x) (bignum? x))
|
|
(<= (- (expt 2 (- fxbits 1)))
|
|
x
|
|
(- (expt 2 (- fxbits 1)) 1)))))
|
|
|
|
|
|
(module ()
|
|
;;; initialize the cogen
|
|
(code-entry-adjustment (- disp-code-data vector-tag)))
|
|
|
|
(begin ;;; COGEN HELERS
|
|
(define (align n)
|
|
(fxsll (fxsra (fx+ n (fxsub1 object-alignment)) align-shift) align-shift))
|
|
(define (mem off val)
|
|
(cond
|
|
[(fixnum? off) (list 'disp (int off) val)]
|
|
[(register? off) (list 'disp off val)]
|
|
[else (error 'mem "invalid disp" off)]))
|
|
(define-syntax int
|
|
(syntax-rules ()
|
|
[(_ x) x]))
|
|
(define (obj x) (list 'obj x))
|
|
(define (byte x) (list 'byte x))
|
|
(define (byte-vector x) (list 'byte-vector x))
|
|
(define (movzbl src targ) (list 'movzbl src targ))
|
|
(define (sall src targ) (list 'sall src targ))
|
|
(define (sarl src targ) (list 'sarl src targ))
|
|
(define (shrl src targ) (list 'shrl src targ))
|
|
(define (notl src) (list 'notl src))
|
|
(define (pushl src) (list 'pushl src))
|
|
(define (popl src) (list 'popl src))
|
|
(define (orl src targ) (list 'orl src targ))
|
|
(define (xorl src targ) (list 'xorl src targ))
|
|
(define (andl src targ) (list 'andl src targ))
|
|
(define (movl src targ) (list 'movl src targ))
|
|
(define (leal src targ) (list 'leal src targ))
|
|
(define (movb src targ) (list 'movb src targ))
|
|
(define (addl src targ) (list 'addl src targ))
|
|
(define (imull src targ) (list 'imull src targ))
|
|
(define (idivl src) (list 'idivl src))
|
|
(define (subl src targ) (list 'subl src targ))
|
|
(define (push src) (list 'push src))
|
|
(define (pop targ) (list 'pop targ))
|
|
(define (sete targ) (list 'sete targ))
|
|
(define (call targ) (list 'call targ))
|
|
(define (tail-indirect-cpr-call)
|
|
(jmp (mem (fx- disp-closure-code closure-tag) cpr)))
|
|
(define (indirect-cpr-call)
|
|
(call (mem (fx- disp-closure-code closure-tag) cpr)))
|
|
(define (negl targ) (list 'negl targ))
|
|
(define (label x) (list 'label x))
|
|
(define (label-address x) (list 'label-address x))
|
|
(define (ret) '(ret))
|
|
(define (cltd) '(cltd))
|
|
(define (cmpl arg1 arg2) (list 'cmpl arg1 arg2))
|
|
(define (je label) (list 'je label))
|
|
(define (jne label) (list 'jne label))
|
|
(define (jle label) (list 'jle label))
|
|
(define (jge label) (list 'jge label))
|
|
(define (jg label) (list 'jg label))
|
|
(define (jl label) (list 'jl label))
|
|
(define (jb label) (list 'jb label))
|
|
(define (ja label) (list 'ja label))
|
|
(define (jo label) (list 'jo label))
|
|
(define (jmp label) (list 'jmp label))
|
|
(define esp '%esp) ; stack base pointer
|
|
(define al '%al)
|
|
(define ah '%ah)
|
|
(define bh '%bh)
|
|
(define cl '%cl)
|
|
(define eax '%eax)
|
|
(define ebx '%ebx)
|
|
(define ecx '%ecx)
|
|
(define edx '%edx)
|
|
(define apr '%ebp) ; allocation pointer
|
|
(define fpr '%esp) ; frame pointer
|
|
(define cpr '%edi) ; closure pointer
|
|
(define pcr '%esi) ; pcb pointer
|
|
(define register? symbol?)
|
|
(define (argc-convention n)
|
|
(fx- 0 (fxsll n fx-shift))))
|
|
|
|
|
|
(define (primref->symbol op)
|
|
(unless (symbol? op) (error 'primref->symbol "not a symbol" op))
|
|
(cond
|
|
[((current-primitive-locations) op) =>
|
|
(lambda (x)
|
|
(unless (symbol? x)
|
|
(error 'primitive-location
|
|
"not a valid location for ~s" x op))
|
|
x)]
|
|
[else
|
|
(let ()
|
|
(define-condition-type &url &condition
|
|
make-url-condition
|
|
url-condition?
|
|
(url condition-url))
|
|
(raise
|
|
(condition
|
|
(make-error)
|
|
(make-who-condition 'ikarus)
|
|
(make-message-condition "primitive not supported yet")
|
|
(make-message-condition
|
|
"please file a bug report to help us prioritize our goals")
|
|
(make-url-condition
|
|
"https://bugs.launchpad.net/ikarus/+filebug")
|
|
(make-irritants-condition (list op)))))]))
|
|
|
|
;(define (primref-loc op)
|
|
; (mem (fx- disp-symbol-record-proc record-tag)
|
|
; (obj (primref->symbol op))))
|
|
|
|
|
|
|
|
(module ;assembly-labels
|
|
(refresh-cached-labels!
|
|
sl-apply-label
|
|
sl-continuation-code-label
|
|
sl-invalid-args-label
|
|
sl-mv-ignore-rp-label
|
|
sl-mv-error-rp-label
|
|
sl-values-label
|
|
sl-cwv-label)
|
|
(define-syntax define-cached
|
|
(lambda (x)
|
|
(syntax-case x ()
|
|
[(_ refresh [(name*) b* b** ...] ...)
|
|
(with-syntax ([(v* ...) (generate-temporaries #'(name* ...))])
|
|
#'(begin
|
|
(define v* #f) ...
|
|
(define (name*)
|
|
(or v* (error 'name* "uninitialized label"))) ...
|
|
(define (refresh)
|
|
(define-syntax name*
|
|
(lambda (stx)
|
|
(syntax-error stx
|
|
"cannot use label before it is defined")))
|
|
...
|
|
(let* ([name* (let ([label (let () b* b** ...)])
|
|
(set! v* label)
|
|
(lambda () label))] ...)
|
|
(void)))))])))
|
|
(define-cached refresh-cached-labels!
|
|
[(sl-apply-label)
|
|
(let ([SL_apply (gensym "SL_apply")]
|
|
[L_apply_done (gensym)]
|
|
[L_apply_loop (gensym)])
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(list 0
|
|
(label SL_apply)
|
|
(movl (mem fpr eax) ebx)
|
|
(cmpl (int nil) ebx)
|
|
(je (label L_apply_done))
|
|
(label L_apply_loop)
|
|
(movl (mem (fx- disp-car pair-tag) ebx) ecx)
|
|
(movl (mem (fx- disp-cdr pair-tag) ebx) ebx)
|
|
(movl ecx (mem fpr eax))
|
|
(subl (int wordsize) eax)
|
|
(cmpl (int nil) ebx)
|
|
(jne (label L_apply_loop))
|
|
(label L_apply_done)
|
|
(addl (int wordsize) eax)
|
|
(tail-indirect-cpr-call))))
|
|
SL_apply)]
|
|
[(sl-continuation-code-label)
|
|
(define SL_continuation_code (gensym "SL_continuation_code"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(let ([L_cont_zero_args (gensym)]
|
|
[L_cont_mult_args (gensym)]
|
|
[L_cont_one_arg (gensym)]
|
|
[L_cont_mult_move_args (gensym)]
|
|
[L_cont_mult_copy_loop (gensym)])
|
|
(list 1 ; freevars
|
|
(label SL_continuation_code)
|
|
(movl (mem (fx- disp-closure-data closure-tag) cpr) ebx) ; captured-k
|
|
(movl ebx (mem pcb-next-continuation pcr)) ; set
|
|
(movl (mem pcb-frame-base pcr) ebx)
|
|
(cmpl (int (argc-convention 1)) eax)
|
|
(jg (label L_cont_zero_args))
|
|
(jl (label L_cont_mult_args))
|
|
(label L_cont_one_arg)
|
|
(movl (mem (fx- 0 wordsize) fpr) eax)
|
|
(movl ebx fpr)
|
|
(subl (int wordsize) fpr)
|
|
(ret)
|
|
(label L_cont_zero_args)
|
|
(subl (int wordsize) ebx)
|
|
(movl ebx fpr)
|
|
(movl (mem 0 ebx) ebx) ; return point
|
|
(jmp (mem disp-multivalue-rp ebx)) ; go
|
|
(label L_cont_mult_args)
|
|
(subl (int wordsize) ebx)
|
|
(cmpl ebx fpr)
|
|
(jne (label L_cont_mult_move_args))
|
|
(movl (mem 0 ebx) ebx)
|
|
(jmp (mem disp-multivalue-rp ebx))
|
|
(label L_cont_mult_move_args)
|
|
; move args from fpr to ebx
|
|
(movl (int 0) ecx)
|
|
(label L_cont_mult_copy_loop)
|
|
(subl (int wordsize) ecx)
|
|
(movl (mem fpr ecx) edx)
|
|
(movl edx (mem ebx ecx))
|
|
(cmpl ecx eax)
|
|
(jne (label L_cont_mult_copy_loop))
|
|
(movl ebx fpr)
|
|
(movl (mem 0 ebx) ebx)
|
|
(jmp (mem disp-multivalue-rp ebx))))))
|
|
SL_continuation_code]
|
|
[(sl-invalid-args-label)
|
|
(define SL_invalid_args (gensym "SL_invalid_args"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(list 0
|
|
(label SL_invalid_args)
|
|
;;;
|
|
(movl cpr (mem (fx- 0 wordsize) fpr)) ; first arg
|
|
(negl eax)
|
|
(movl eax (mem (fx- 0 (fx* 2 wordsize)) fpr))
|
|
(movl (obj (primref->symbol '$incorrect-args-error-handler)) cpr)
|
|
(movl (mem (- disp-symbol-record-proc record-tag) cpr) cpr)
|
|
;(movl (primref-loc '$incorrect-args-error-handler) cpr)
|
|
(movl (int (argc-convention 2)) eax)
|
|
(tail-indirect-cpr-call))))
|
|
SL_invalid_args]
|
|
[(sl-mv-ignore-rp-label)
|
|
(define SL_multiple_values_ignore_rp (gensym "SL_multiple_ignore_error_rp"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(list 0
|
|
(label SL_multiple_values_ignore_rp)
|
|
(ret))))
|
|
SL_multiple_values_ignore_rp]
|
|
[(sl-mv-error-rp-label)
|
|
(define SL_multiple_values_error_rp (gensym "SL_multiple_values_error_rp"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(list 0
|
|
(label SL_multiple_values_error_rp)
|
|
(movl (obj (primref->symbol '$multiple-values-error)) cpr)
|
|
(movl (mem (- disp-symbol-record-proc record-tag) cpr) cpr)
|
|
;(movl (primref-loc '$multiple-values-error) cpr)
|
|
(tail-indirect-cpr-call))))
|
|
SL_multiple_values_error_rp]
|
|
[(sl-values-label)
|
|
(define SL_values (gensym "SL_values"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(let ([L_values_one_value (gensym)]
|
|
[L_values_many_values (gensym)])
|
|
(list 0 ; no freevars
|
|
'(name values)
|
|
(label SL_values)
|
|
(cmpl (int (argc-convention 1)) eax)
|
|
(je (label L_values_one_value))
|
|
(label L_values_many_values)
|
|
(movl (mem 0 fpr) ebx) ; return point
|
|
(jmp (mem disp-multivalue-rp ebx)) ; go
|
|
(label L_values_one_value)
|
|
(movl (mem (fx- 0 wordsize) fpr) eax)
|
|
(ret)))))
|
|
SL_values]
|
|
[(sl-nonprocedure-error-label)
|
|
(define SL_nonprocedure (gensym "SL_nonprocedure"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(list 0
|
|
(label SL_nonprocedure)
|
|
(movl cpr (mem (fx- 0 wordsize) fpr)) ; first arg
|
|
(movl (obj (primref->symbol '$apply-nonprocedure-error-handler)) cpr)
|
|
(movl (mem (- disp-symbol-record-proc record-tag) cpr) cpr)
|
|
;(movl (primref-loc '$apply-nonprocedure-error-handler) cpr)
|
|
(movl (int (argc-convention 1)) eax)
|
|
(tail-indirect-cpr-call))))
|
|
SL_nonprocedure]
|
|
[(sl-cwv-label)
|
|
(define SL_call_with_values (gensym "SL_call_with_values"))
|
|
(assemble-sources (lambda (x) #f)
|
|
(list
|
|
(let ([L_cwv_done (gensym)]
|
|
[L_cwv_loop (gensym)]
|
|
[L_cwv_multi_rp (gensym)]
|
|
[L_cwv_call (gensym)])
|
|
(list
|
|
0 ; no free vars
|
|
'(name call-with-values)
|
|
(label SL_call_with_values)
|
|
(cmpl (int (argc-convention 2)) eax)
|
|
(jne (label (sl-invalid-args-label)))
|
|
(movl (mem (fx- 0 wordsize) fpr) ebx) ; producer
|
|
(movl ebx cpr)
|
|
(andl (int closure-mask) ebx)
|
|
(cmpl (int closure-tag) ebx)
|
|
(jne (label (sl-nonprocedure-error-label)))
|
|
(movl (int (argc-convention 0)) eax)
|
|
;(subl (int (fx* wordsize 2)) fpr)
|
|
(compile-call-frame
|
|
3
|
|
'#(#b110)
|
|
(label-address L_cwv_multi_rp)
|
|
(indirect-cpr-call))
|
|
;;; one value returned
|
|
;(addl (int (fx* wordsize 2)) fpr)
|
|
(movl (mem (fx* -2 wordsize) fpr) ebx) ; consumer
|
|
(movl ebx cpr)
|
|
(movl eax (mem (fx- 0 wordsize) fpr))
|
|
(movl (int (argc-convention 1)) eax)
|
|
(andl (int closure-mask) ebx)
|
|
(cmpl (int closure-tag) ebx)
|
|
(jne (label (sl-nonprocedure-error-label)))
|
|
(tail-indirect-cpr-call)
|
|
;;; multiple values returned
|
|
(label L_cwv_multi_rp)
|
|
; because values does not pop the return point
|
|
; we have to adjust fp one more word here
|
|
(addl (int (fx* wordsize 3)) fpr)
|
|
(movl (mem (fx* -2 wordsize) fpr) cpr) ; consumer
|
|
(cmpl (int (argc-convention 0)) eax)
|
|
(je (label L_cwv_done))
|
|
(movl (int (fx* -4 wordsize)) ebx)
|
|
(addl fpr ebx) ; ebx points to first value
|
|
(movl ebx ecx)
|
|
(addl eax ecx) ; ecx points to the last value
|
|
(label L_cwv_loop)
|
|
(movl (mem 0 ebx) edx)
|
|
(movl edx (mem (fx* 3 wordsize) ebx))
|
|
(subl (int wordsize) ebx)
|
|
(cmpl ecx ebx)
|
|
(jge (label L_cwv_loop))
|
|
(label L_cwv_done)
|
|
(movl cpr ebx)
|
|
(andl (int closure-mask) ebx)
|
|
(cmpl (int closure-tag) ebx)
|
|
(jne (label (sl-nonprocedure-error-label)))
|
|
(tail-indirect-cpr-call)))))
|
|
SL_call_with_values]
|
|
))
|
|
|
|
(define (print-instr x)
|
|
(cond
|
|
[(and (pair? x) (eq? (car x) 'seq))
|
|
(for-each print-instr (cdr x))]
|
|
[else
|
|
(printf " ~s\n" x)]))
|
|
|
|
(define optimizer-output (make-parameter #f))
|
|
(define perform-tag-analysis (make-parameter #f))
|
|
|
|
(define (compile-core-expr->code p)
|
|
(let* ([p (recordize p)]
|
|
[p (parameterize ([open-mvcalls #f])
|
|
(optimize-direct-calls p))]
|
|
[p (optimize-letrec/scc p)]
|
|
[p (source-optimize p)]
|
|
[dummy
|
|
(begin
|
|
(when (optimizer-output)
|
|
(pretty-print (unparse-pretty p)))
|
|
#f)]
|
|
[p (rewrite-assignments p)]
|
|
[p (if (perform-tag-analysis)
|
|
(introduce-tags p)
|
|
p)]
|
|
[p (introduce-vars p)]
|
|
[p (sanitize-bindings p)]
|
|
[p (optimize-for-direct-jumps p)]
|
|
[p (insert-global-assignments p)]
|
|
[p (convert-closures p)]
|
|
[p (optimize-closures/lift-codes p)])
|
|
(let ([ls* (alt-cogen p)])
|
|
(when (assembler-output)
|
|
(parameterize ([gensym-prefix "L"]
|
|
[print-gensym #f])
|
|
(for-each
|
|
(lambda (ls)
|
|
(newline)
|
|
(for-each print-instr ls))
|
|
ls*)))
|
|
(let ([code*
|
|
(assemble-sources
|
|
(lambda (x)
|
|
(if (closure? x)
|
|
(if (null? (closure-free* x))
|
|
(code-loc-label (closure-code x))
|
|
(error 'compile "BUG: non-thunk escaped" x))
|
|
#f))
|
|
ls*)])
|
|
(car code*)))))
|
|
|
|
(define compile-core-expr-to-port
|
|
(lambda (expr port)
|
|
(fasl-write (compile-core-expr->code expr) port)))
|
|
|
|
|
|
(define (compile-core-expr x)
|
|
(let ([code (compile-core-expr->code x)])
|
|
($code->closure code)))
|
|
|
|
(define assembler-output (make-parameter #f))
|
|
|
|
|
|
(define eval-core
|
|
(lambda (x) ((compile-core-expr x))))
|
|
|
|
(include "ikarus.compiler.altcogen.ss")
|
|
|
|
(define current-primitive-locations
|
|
(let ([plocs (lambda (x) #f)])
|
|
(case-lambda
|
|
[() plocs]
|
|
[(p)
|
|
(if (procedure? p)
|
|
(begin
|
|
(set! plocs p)
|
|
(refresh-cached-labels!))
|
|
(error 'current-primitive-locations "not a procedure" p))])))
|
|
|
|
(define expand/optimize
|
|
(case-lambda
|
|
[(p) (expand/optimize p (interaction-environment))]
|
|
[(p env)
|
|
(unless (environment? env)
|
|
(env 'expand/optimize "not an environment" env))
|
|
(let-values ([(p lib*) (expand p env)])
|
|
(let* ([p (recordize p)]
|
|
[p (parameterize ([open-mvcalls #f])
|
|
(optimize-direct-calls p))]
|
|
[p (optimize-letrec/scc p)]
|
|
[p (source-optimize p)])
|
|
(unparse-pretty p)))]))
|
|
|
|
)
|
|
|
|
|