Spaces:
Sleeping
Sleeping
;;;; Copyright (C) 2009, 2010, 2011, 2012, 2013 Free Software Foundation, Inc. | |
;;;; | |
;;;; This library is free software; you can redistribute it and/or | |
;;;; modify it under the terms of the GNU Lesser General Public | |
;;;; License as published by the Free Software Foundation; either | |
;;;; version 3 of the License, or (at your option) any later version. | |
;;;; | |
;;;; This library 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 | |
;;;; Lesser General Public License for more details. | |
;;;; | |
;;;; You should have received a copy of the GNU Lesser General Public | |
;;;; License along with this library; if not, write to the Free Software | |
;;;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
;;;; | |
(define-module (language tree-il) | |
#:use-module (srfi srfi-1) | |
#:use-module (srfi srfi-11) | |
#:use-module (system base pmatch) | |
#:use-module (system base syntax) | |
#:export (tree-il-src | |
<void> void? make-void void-src | |
<const> const? make-const const-src const-exp | |
<primitive-ref> primitive-ref? make-primitive-ref primitive-ref-src primitive-ref-name | |
<lexical-ref> lexical-ref? make-lexical-ref lexical-ref-src lexical-ref-name lexical-ref-gensym | |
<lexical-set> lexical-set? make-lexical-set lexical-set-src lexical-set-name lexical-set-gensym lexical-set-exp | |
<module-ref> module-ref? make-module-ref module-ref-src module-ref-mod module-ref-name module-ref-public? | |
<module-set> module-set? make-module-set module-set-src module-set-mod module-set-name module-set-public? module-set-exp | |
<toplevel-ref> toplevel-ref? make-toplevel-ref toplevel-ref-src toplevel-ref-name | |
<toplevel-set> toplevel-set? make-toplevel-set toplevel-set-src toplevel-set-name toplevel-set-exp | |
<toplevel-define> toplevel-define? make-toplevel-define toplevel-define-src toplevel-define-name toplevel-define-exp | |
<conditional> conditional? make-conditional conditional-src conditional-test conditional-consequent conditional-alternate | |
<application> application? make-application application-src application-proc application-args | |
<sequence> sequence? make-sequence sequence-src sequence-exps | |
<lambda> lambda? make-lambda lambda-src lambda-meta lambda-body | |
<lambda-case> lambda-case? make-lambda-case lambda-case-src | |
lambda-case-req lambda-case-opt lambda-case-rest lambda-case-kw | |
lambda-case-inits lambda-case-gensyms | |
lambda-case-body lambda-case-alternate | |
<let> let? make-let let-src let-names let-gensyms let-vals let-body | |
<letrec> letrec? make-letrec letrec-src letrec-in-order? letrec-names letrec-gensyms letrec-vals letrec-body | |
<fix> fix? make-fix fix-src fix-names fix-gensyms fix-vals fix-body | |
<let-values> let-values? make-let-values let-values-src let-values-exp let-values-body | |
<dynwind> dynwind? make-dynwind dynwind-src dynwind-winder dynwind-body dynwind-unwinder | |
<dynlet> dynlet? make-dynlet dynlet-src dynlet-fluids dynlet-vals dynlet-body | |
<dynref> dynref? make-dynref dynref-src dynref-fluid | |
<dynset> dynset? make-dynset dynset-src dynset-fluid dynset-exp | |
<prompt> prompt? make-prompt prompt-src prompt-tag prompt-body prompt-handler | |
<abort> abort? make-abort abort-src abort-tag abort-args abort-tail | |
parse-tree-il | |
unparse-tree-il | |
tree-il->scheme | |
tree-il-fold | |
make-tree-il-folder | |
post-order! | |
pre-order! | |
tree-il=? | |
tree-il-hash)) | |
(define (print-tree-il exp port) | |
(format port "#<tree-il ~S>" (unparse-tree-il exp))) | |
(define-syntax borrow-core-vtables | |
(lambda (x) | |
(syntax-case x () | |
((_) | |
(let lp ((n 0) (out '())) | |
(if (< n (vector-length %expanded-vtables)) | |
(lp (1+ n) | |
(let* ((vtable (vector-ref %expanded-vtables n)) | |
(stem (struct-ref vtable (+ vtable-offset-user 0))) | |
(fields (struct-ref vtable (+ vtable-offset-user 2))) | |
(sfields (map | |
(lambda (f) (datum->syntax x f)) | |
fields)) | |
(type (datum->syntax x (symbol-append '< stem '>))) | |
(ctor (datum->syntax x (symbol-append 'make- stem))) | |
(pred (datum->syntax x (symbol-append stem '?)))) | |
(let lp ((n 0) (fields fields) | |
(out (cons* | |
#`(define (#,ctor #,@sfields) | |
(make-struct #,type 0 #,@sfields)) | |
#`(define (#,pred x) | |
(and (struct? x) | |
(eq? (struct-vtable x) #,type))) | |
#`(struct-set! #,type vtable-index-printer | |
print-tree-il) | |
#`(define #,type | |
(vector-ref %expanded-vtables #,n)) | |
out))) | |
(if (null? fields) | |
out | |
(lp (1+ n) | |
(cdr fields) | |
(let ((acc (datum->syntax | |
x (symbol-append stem '- (car fields))))) | |
(cons #`(define #,acc | |
(make-procedure-with-setter | |
(lambda (x) (struct-ref x #,n)) | |
(lambda (x v) (struct-set! x #,n v)))) | |
out))))))) | |
#`(begin #,@(reverse out)))))))) | |
(borrow-core-vtables) | |
;; (<void>) | |
;; (<const> exp) | |
;; (<primitive-ref> name) | |
;; (<lexical-ref> name gensym) | |
;; (<lexical-set> name gensym exp) | |
;; (<module-ref> mod name public?) | |
;; (<module-set> mod name public? exp) | |
;; (<toplevel-ref> name) | |
;; (<toplevel-set> name exp) | |
;; (<toplevel-define> name exp) | |
;; (<conditional> test consequent alternate) | |
;; (<application> proc args) | |
;; (<sequence> exps) | |
;; (<lambda> meta body) | |
;; (<lambda-case> req opt rest kw inits gensyms body alternate) | |
;; (<let> names gensyms vals body) | |
;; (<letrec> in-order? names gensyms vals body) | |
;; (<dynlet> fluids vals body) | |
(define-type (<tree-il> #:common-slots (src) #:printer print-tree-il) | |
(<fix> names gensyms vals body) | |
(<let-values> exp body) | |
(<dynwind> winder body unwinder) | |
(<dynref> fluid) | |
(<dynset> fluid exp) | |
(<prompt> tag body handler) | |
(<abort> tag args tail)) | |
(define (location x) | |
(and (pair? x) | |
(let ((props (source-properties x))) | |
(and (pair? props) props)))) | |
(define (parse-tree-il exp) | |
(let ((loc (location exp)) | |
(retrans (lambda (x) (parse-tree-il x)))) | |
(pmatch exp | |
((void) | |
(make-void loc)) | |
((apply ,proc . ,args) | |
(make-application loc (retrans proc) (map retrans args))) | |
((if ,test ,consequent ,alternate) | |
(make-conditional loc (retrans test) (retrans consequent) (retrans alternate))) | |
((primitive ,name) (guard (symbol? name)) | |
(make-primitive-ref loc name)) | |
((lexical ,name) (guard (symbol? name)) | |
(make-lexical-ref loc name name)) | |
((lexical ,name ,sym) (guard (symbol? name) (symbol? sym)) | |
(make-lexical-ref loc name sym)) | |
((set! (lexical ,name) ,exp) (guard (symbol? name)) | |
(make-lexical-set loc name name (retrans exp))) | |
((set! (lexical ,name ,sym) ,exp) (guard (symbol? name) (symbol? sym)) | |
(make-lexical-set loc name sym (retrans exp))) | |
((@ ,mod ,name) (guard (and-map symbol? mod) (symbol? name)) | |
(make-module-ref loc mod name #t)) | |
((set! (@ ,mod ,name) ,exp) (guard (and-map symbol? mod) (symbol? name)) | |
(make-module-set loc mod name #t (retrans exp))) | |
((@@ ,mod ,name) (guard (and-map symbol? mod) (symbol? name)) | |
(make-module-ref loc mod name #f)) | |
((set! (@@ ,mod ,name) ,exp) (guard (and-map symbol? mod) (symbol? name)) | |
(make-module-set loc mod name #f (retrans exp))) | |
((toplevel ,name) (guard (symbol? name)) | |
(make-toplevel-ref loc name)) | |
((set! (toplevel ,name) ,exp) (guard (symbol? name)) | |
(make-toplevel-set loc name (retrans exp))) | |
((define ,name ,exp) (guard (symbol? name)) | |
(make-toplevel-define loc name (retrans exp))) | |
((lambda ,meta ,body) | |
(make-lambda loc meta (retrans body))) | |
((lambda-case ((,req ,opt ,rest ,kw ,inits ,gensyms) ,body) ,alternate) | |
(make-lambda-case loc req opt rest kw | |
(map retrans inits) gensyms | |
(retrans body) | |
(and=> alternate retrans))) | |
((lambda-case ((,req ,opt ,rest ,kw ,inits ,gensyms) ,body)) | |
(make-lambda-case loc req opt rest kw | |
(map retrans inits) gensyms | |
(retrans body) | |
#f)) | |
((const ,exp) | |
(make-const loc exp)) | |
((begin . ,exps) | |
(make-sequence loc (map retrans exps))) | |
((let ,names ,gensyms ,vals ,body) | |
(make-let loc names gensyms (map retrans vals) (retrans body))) | |
((letrec ,names ,gensyms ,vals ,body) | |
(make-letrec loc #f names gensyms (map retrans vals) (retrans body))) | |
((letrec* ,names ,gensyms ,vals ,body) | |
(make-letrec loc #t names gensyms (map retrans vals) (retrans body))) | |
((fix ,names ,gensyms ,vals ,body) | |
(make-fix loc names gensyms (map retrans vals) (retrans body))) | |
((let-values ,exp ,body) | |
(make-let-values loc (retrans exp) (retrans body))) | |
((dynwind ,winder ,body ,unwinder) | |
(make-dynwind loc (retrans winder) (retrans body) (retrans unwinder))) | |
((dynlet ,fluids ,vals ,body) | |
(make-dynlet loc (map retrans fluids) (map retrans vals) (retrans body))) | |
((dynref ,fluid) | |
(make-dynref loc (retrans fluid))) | |
((dynset ,fluid ,exp) | |
(make-dynset loc (retrans fluid) (retrans exp))) | |
((prompt ,tag ,body ,handler) | |
(make-prompt loc (retrans tag) (retrans body) (retrans handler))) | |
((abort ,tag ,args ,tail) | |
(make-abort loc (retrans tag) (map retrans args) (retrans tail))) | |
(else | |
(error "unrecognized tree-il" exp))))) | |
(define (unparse-tree-il tree-il) | |
(record-case tree-il | |
((<void>) | |
'(void)) | |
((<application> proc args) | |
`(apply ,(unparse-tree-il proc) ,@(map unparse-tree-il args))) | |
((<conditional> test consequent alternate) | |
`(if ,(unparse-tree-il test) ,(unparse-tree-il consequent) ,(unparse-tree-il alternate))) | |
((<primitive-ref> name) | |
`(primitive ,name)) | |
((<lexical-ref> name gensym) | |
`(lexical ,name ,gensym)) | |
((<lexical-set> name gensym exp) | |
`(set! (lexical ,name ,gensym) ,(unparse-tree-il exp))) | |
((<module-ref> mod name public?) | |
`(,(if public? '@ '@@) ,mod ,name)) | |
((<module-set> mod name public? exp) | |
`(set! (,(if public? '@ '@@) ,mod ,name) ,(unparse-tree-il exp))) | |
((<toplevel-ref> name) | |
`(toplevel ,name)) | |
((<toplevel-set> name exp) | |
`(set! (toplevel ,name) ,(unparse-tree-il exp))) | |
((<toplevel-define> name exp) | |
`(define ,name ,(unparse-tree-il exp))) | |
((<lambda> meta body) | |
(if body | |
`(lambda ,meta ,(unparse-tree-il body)) | |
`(lambda ,meta (lambda-case)))) | |
((<lambda-case> req opt rest kw inits gensyms body alternate) | |
`(lambda-case ((,req ,opt ,rest ,kw ,(map unparse-tree-il inits) ,gensyms) | |
,(unparse-tree-il body)) | |
. ,(if alternate (list (unparse-tree-il alternate)) '()))) | |
((<const> exp) | |
`(const ,exp)) | |
((<sequence> exps) | |
`(begin ,@(map unparse-tree-il exps))) | |
((<let> names gensyms vals body) | |
`(let ,names ,gensyms ,(map unparse-tree-il vals) ,(unparse-tree-il body))) | |
((<letrec> in-order? names gensyms vals body) | |
`(,(if in-order? 'letrec* 'letrec) ,names ,gensyms | |
,(map unparse-tree-il vals) ,(unparse-tree-il body))) | |
((<fix> names gensyms vals body) | |
`(fix ,names ,gensyms ,(map unparse-tree-il vals) ,(unparse-tree-il body))) | |
((<let-values> exp body) | |
`(let-values ,(unparse-tree-il exp) ,(unparse-tree-il body))) | |
((<dynwind> winder body unwinder) | |
`(dynwind ,(unparse-tree-il winder) ,(unparse-tree-il body) | |
,(unparse-tree-il unwinder))) | |
((<dynlet> fluids vals body) | |
`(dynlet ,(map unparse-tree-il fluids) ,(map unparse-tree-il vals) | |
,(unparse-tree-il body))) | |
((<dynref> fluid) | |
`(dynref ,(unparse-tree-il fluid))) | |
((<dynset> fluid exp) | |
`(dynset ,(unparse-tree-il fluid) ,(unparse-tree-il exp))) | |
((<prompt> tag body handler) | |
`(prompt ,(unparse-tree-il tag) ,(unparse-tree-il body) ,(unparse-tree-il handler))) | |
((<abort> tag args tail) | |
`(abort ,(unparse-tree-il tag) ,(map unparse-tree-il args) | |
,(unparse-tree-il tail))))) | |
(define* (tree-il->scheme e #:optional (env #f) (opts '())) | |
(values ((@ (language scheme decompile-tree-il) | |
decompile-tree-il) | |
e env opts))) | |
(define (tree-il-fold leaf down up seed tree) | |
"Traverse TREE, calling LEAF on each leaf encountered, DOWN upon descent | |
into a sub-tree, and UP when leaving a sub-tree. Each of these procedures is | |
invoked as `(PROC TREE SEED)', where TREE is the sub-tree or leaf considered | |
and SEED is the current result, intially seeded with SEED. | |
This is an implementation of `foldts' as described by Andy Wingo in | |
``Applications of fold to XML transformation''." | |
(let loop ((tree tree) | |
(result seed)) | |
(if (or (null? tree) (pair? tree)) | |
(fold loop result tree) | |
(record-case tree | |
((<lexical-set> exp) | |
(up tree (loop exp (down tree result)))) | |
((<module-set> exp) | |
(up tree (loop exp (down tree result)))) | |
((<toplevel-set> exp) | |
(up tree (loop exp (down tree result)))) | |
((<toplevel-define> exp) | |
(up tree (loop exp (down tree result)))) | |
((<conditional> test consequent alternate) | |
(up tree (loop alternate | |
(loop consequent | |
(loop test (down tree result)))))) | |
((<application> proc args) | |
(up tree (loop (cons proc args) (down tree result)))) | |
((<sequence> exps) | |
(up tree (loop exps (down tree result)))) | |
((<lambda> body) | |
(let ((result (down tree result))) | |
(up tree | |
(if body | |
(loop body result) | |
result)))) | |
((<lambda-case> inits body alternate) | |
(up tree (if alternate | |
(loop alternate | |
(loop body (loop inits (down tree result)))) | |
(loop body (loop inits (down tree result)))))) | |
((<let> vals body) | |
(up tree (loop body | |
(loop vals | |
(down tree result))))) | |
((<letrec> vals body) | |
(up tree (loop body | |
(loop vals | |
(down tree result))))) | |
((<fix> vals body) | |
(up tree (loop body | |
(loop vals | |
(down tree result))))) | |
((<let-values> exp body) | |
(up tree (loop body (loop exp (down tree result))))) | |
((<dynwind> body winder unwinder) | |
(up tree (loop unwinder | |
(loop winder | |
(loop body (down tree result)))))) | |
((<dynlet> fluids vals body) | |
(up tree (loop body | |
(loop vals | |
(loop fluids (down tree result)))))) | |
((<dynref> fluid) | |
(up tree (loop fluid (down tree result)))) | |
((<dynset> fluid exp) | |
(up tree (loop exp (loop fluid (down tree result))))) | |
((<prompt> tag body handler) | |
(up tree | |
(loop tag (loop body (loop handler | |
(down tree result)))))) | |
((<abort> tag args tail) | |
(up tree (loop tail (loop args (loop tag (down tree result)))))) | |
(else | |
(leaf tree result)))))) | |
(define-syntax-rule (make-tree-il-folder seed ...) | |
(lambda (tree down up seed ...) | |
(define (fold-values proc exps seed ...) | |
(if (null? exps) | |
(values seed ...) | |
(let-values (((seed ...) (proc (car exps) seed ...))) | |
(fold-values proc (cdr exps) seed ...)))) | |
(let foldts ((tree tree) (seed seed) ...) | |
(let*-values | |
(((seed ...) (down tree seed ...)) | |
((seed ...) | |
(record-case tree | |
((<lexical-set> exp) | |
(foldts exp seed ...)) | |
((<module-set> exp) | |
(foldts exp seed ...)) | |
((<toplevel-set> exp) | |
(foldts exp seed ...)) | |
((<toplevel-define> exp) | |
(foldts exp seed ...)) | |
((<conditional> test consequent alternate) | |
(let*-values (((seed ...) (foldts test seed ...)) | |
((seed ...) (foldts consequent seed ...))) | |
(foldts alternate seed ...))) | |
((<application> proc args) | |
(let-values (((seed ...) (foldts proc seed ...))) | |
(fold-values foldts args seed ...))) | |
((<sequence> exps) | |
(fold-values foldts exps seed ...)) | |
((<lambda> body) | |
(if body | |
(foldts body seed ...) | |
(values seed ...))) | |
((<lambda-case> inits body alternate) | |
(let-values (((seed ...) (fold-values foldts inits seed ...))) | |
(if alternate | |
(let-values (((seed ...) (foldts body seed ...))) | |
(foldts alternate seed ...)) | |
(foldts body seed ...)))) | |
((<let> vals body) | |
(let*-values (((seed ...) (fold-values foldts vals seed ...))) | |
(foldts body seed ...))) | |
((<letrec> vals body) | |
(let*-values (((seed ...) (fold-values foldts vals seed ...))) | |
(foldts body seed ...))) | |
((<fix> vals body) | |
(let*-values (((seed ...) (fold-values foldts vals seed ...))) | |
(foldts body seed ...))) | |
((<let-values> exp body) | |
(let*-values (((seed ...) (foldts exp seed ...))) | |
(foldts body seed ...))) | |
((<dynwind> body winder unwinder) | |
(let*-values (((seed ...) (foldts body seed ...)) | |
((seed ...) (foldts winder seed ...))) | |
(foldts unwinder seed ...))) | |
((<dynlet> fluids vals body) | |
(let*-values (((seed ...) (fold-values foldts fluids seed ...)) | |
((seed ...) (fold-values foldts vals seed ...))) | |
(foldts body seed ...))) | |
((<dynref> fluid) | |
(foldts fluid seed ...)) | |
((<dynset> fluid exp) | |
(let*-values (((seed ...) (foldts fluid seed ...))) | |
(foldts exp seed ...))) | |
((<prompt> tag body handler) | |
(let*-values (((seed ...) (foldts tag seed ...)) | |
((seed ...) (foldts body seed ...))) | |
(foldts handler seed ...))) | |
((<abort> tag args tail) | |
(let*-values (((seed ...) (foldts tag seed ...)) | |
((seed ...) (fold-values foldts args seed ...))) | |
(foldts tail seed ...))) | |
(else | |
(values seed ...))))) | |
(up tree seed ...))))) | |
(define (post-order! f x) | |
(let lp ((x x)) | |
(record-case x | |
((<application> proc args) | |
(set! (application-proc x) (lp proc)) | |
(set! (application-args x) (map lp args))) | |
((<conditional> test consequent alternate) | |
(set! (conditional-test x) (lp test)) | |
(set! (conditional-consequent x) (lp consequent)) | |
(set! (conditional-alternate x) (lp alternate))) | |
((<lexical-set> name gensym exp) | |
(set! (lexical-set-exp x) (lp exp))) | |
((<module-set> mod name public? exp) | |
(set! (module-set-exp x) (lp exp))) | |
((<toplevel-set> name exp) | |
(set! (toplevel-set-exp x) (lp exp))) | |
((<toplevel-define> name exp) | |
(set! (toplevel-define-exp x) (lp exp))) | |
((<lambda> body) | |
(if body | |
(set! (lambda-body x) (lp body)))) | |
((<lambda-case> inits body alternate) | |
(set! inits (map lp inits)) | |
(set! (lambda-case-body x) (lp body)) | |
(if alternate | |
(set! (lambda-case-alternate x) (lp alternate)))) | |
((<sequence> exps) | |
(set! (sequence-exps x) (map lp exps))) | |
((<let> gensyms vals body) | |
(set! (let-vals x) (map lp vals)) | |
(set! (let-body x) (lp body))) | |
((<letrec> gensyms vals body) | |
(set! (letrec-vals x) (map lp vals)) | |
(set! (letrec-body x) (lp body))) | |
((<fix> gensyms vals body) | |
(set! (fix-vals x) (map lp vals)) | |
(set! (fix-body x) (lp body))) | |
((<let-values> exp body) | |
(set! (let-values-exp x) (lp exp)) | |
(set! (let-values-body x) (lp body))) | |
((<dynwind> body winder unwinder) | |
(set! (dynwind-body x) (lp body)) | |
(set! (dynwind-winder x) (lp winder)) | |
(set! (dynwind-unwinder x) (lp unwinder))) | |
((<dynlet> fluids vals body) | |
(set! (dynlet-fluids x) (map lp fluids)) | |
(set! (dynlet-vals x) (map lp vals)) | |
(set! (dynlet-body x) (lp body))) | |
((<dynref> fluid) | |
(set! (dynref-fluid x) (lp fluid))) | |
((<dynset> fluid exp) | |
(set! (dynset-fluid x) (lp fluid)) | |
(set! (dynset-exp x) (lp exp))) | |
((<prompt> tag body handler) | |
(set! (prompt-tag x) (lp tag)) | |
(set! (prompt-body x) (lp body)) | |
(set! (prompt-handler x) (lp handler))) | |
((<abort> tag args tail) | |
(set! (abort-tag x) (lp tag)) | |
(set! (abort-args x) (map lp args)) | |
(set! (abort-tail x) (lp tail))) | |
(else #f)) | |
(or (f x) x))) | |
(define (pre-order! f x) | |
(let lp ((x x)) | |
(let ((x (or (f x) x))) | |
(record-case x | |
((<application> proc args) | |
(set! (application-proc x) (lp proc)) | |
(set! (application-args x) (map lp args))) | |
((<conditional> test consequent alternate) | |
(set! (conditional-test x) (lp test)) | |
(set! (conditional-consequent x) (lp consequent)) | |
(set! (conditional-alternate x) (lp alternate))) | |
((<lexical-set> exp) | |
(set! (lexical-set-exp x) (lp exp))) | |
((<module-set> exp) | |
(set! (module-set-exp x) (lp exp))) | |
((<toplevel-set> exp) | |
(set! (toplevel-set-exp x) (lp exp))) | |
((<toplevel-define> exp) | |
(set! (toplevel-define-exp x) (lp exp))) | |
((<lambda> body) | |
(if body | |
(set! (lambda-body x) (lp body)))) | |
((<lambda-case> inits body alternate) | |
(set! inits (map lp inits)) | |
(set! (lambda-case-body x) (lp body)) | |
(if alternate (set! (lambda-case-alternate x) (lp alternate)))) | |
((<sequence> exps) | |
(set! (sequence-exps x) (map lp exps))) | |
((<let> vals body) | |
(set! (let-vals x) (map lp vals)) | |
(set! (let-body x) (lp body))) | |
((<letrec> vals body) | |
(set! (letrec-vals x) (map lp vals)) | |
(set! (letrec-body x) (lp body))) | |
((<fix> vals body) | |
(set! (fix-vals x) (map lp vals)) | |
(set! (fix-body x) (lp body))) | |
((<let-values> exp body) | |
(set! (let-values-exp x) (lp exp)) | |
(set! (let-values-body x) (lp body))) | |
((<dynwind> body winder unwinder) | |
(set! (dynwind-body x) (lp body)) | |
(set! (dynwind-winder x) (lp winder)) | |
(set! (dynwind-unwinder x) (lp unwinder))) | |
((<dynlet> fluids vals body) | |
(set! (dynlet-fluids x) (map lp fluids)) | |
(set! (dynlet-vals x) (map lp vals)) | |
(set! (dynlet-body x) (lp body))) | |
((<dynref> fluid) | |
(set! (dynref-fluid x) (lp fluid))) | |
((<dynset> fluid exp) | |
(set! (dynset-fluid x) (lp fluid)) | |
(set! (dynset-exp x) (lp exp))) | |
((<prompt> tag body handler) | |
(set! (prompt-tag x) (lp tag)) | |
(set! (prompt-body x) (lp body)) | |
(set! (prompt-handler x) (lp handler))) | |
((<abort> tag args tail) | |
(set! (abort-tag x) (lp tag)) | |
(set! (abort-args x) (map lp args)) | |
(set! (abort-tail x) (lp tail))) | |
(else #f)) | |
x))) | |
;; FIXME: We should have a better primitive than this. | |
(define (struct-nfields x) | |
(/ (string-length (symbol->string (struct-layout x))) 2)) | |
(define (tree-il=? a b) | |
(cond | |
((struct? a) | |
(and (struct? b) | |
(eq? (struct-vtable a) (struct-vtable b)) | |
;; Assume that all structs are tree-il, so we skip over the | |
;; src slot. | |
(let lp ((n (1- (struct-nfields a)))) | |
(or (zero? n) | |
(and (tree-il=? (struct-ref a n) (struct-ref b n)) | |
(lp (1- n))))))) | |
((pair? a) | |
(and (pair? b) | |
(tree-il=? (car a) (car b)) | |
(tree-il=? (cdr a) (cdr b)))) | |
(else | |
(equal? a b)))) | |
(define-syntax hash-bits | |
(make-variable-transformer | |
(lambda (x) | |
(syntax-case x () | |
(var | |
(identifier? #'var) | |
(logcount most-positive-fixnum)))))) | |
(define (tree-il-hash exp) | |
(let ((hash-depth 4) | |
(hash-width 3)) | |
(define (hash-exp exp depth) | |
(define (rotate x bits) | |
(logior (ash x (- bits)) | |
(ash (logand x (1- (ash 1 bits))) (- hash-bits bits)))) | |
(define (mix h1 h2) | |
(logxor h1 (rotate h2 8))) | |
(define (hash-struct s) | |
(let ((len (struct-nfields s)) | |
(h (hashq (struct-vtable s) most-positive-fixnum))) | |
(if (zero? depth) | |
h | |
(let lp ((i (max (- len hash-width) 1)) (h h)) | |
(if (< i len) | |
(lp (1+ i) (mix (hash-exp (struct-ref s i) (1+ depth)) h)) | |
h))))) | |
(define (hash-list l) | |
(let ((h (hashq 'list most-positive-fixnum))) | |
(if (zero? depth) | |
h | |
(let lp ((l l) (width 0) (h h)) | |
(if (< width hash-width) | |
(lp (cdr l) (1+ width) | |
(mix (hash-exp (car l) (1+ depth)) h)) | |
h))))) | |
(cond | |
((struct? exp) (hash-struct exp)) | |
((list? exp) (hash-list exp)) | |
(else (hash exp most-positive-fixnum)))) | |
(hash-exp exp 0))) | |