2009-03-28 17:39:04 -04:00
|
|
|
; -*- scheme -*-
|
|
|
|
|
2009-05-31 17:06:04 -04:00
|
|
|
(define Instructions
|
|
|
|
(let ((e (table))
|
|
|
|
(keys
|
2009-07-28 00:16:20 -04:00
|
|
|
[nop dup pop call tcall jmp brf brt jmp.l brf.l brt.l ret
|
2009-05-31 17:06:04 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
eq? eqv? equal? atom? not null? boolean? symbol?
|
|
|
|
number? bound? pair? builtin? vector? fixnum? function?
|
2009-05-31 17:06:04 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
cons list car cdr set-car! set-cdr!
|
|
|
|
apply
|
2009-05-31 17:06:04 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
+ - * / div0 = < compare
|
2009-05-31 17:06:04 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
vector aref aset!
|
2009-05-31 17:06:04 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
loadt loadf loadnil load0 load1 loadi8
|
|
|
|
loadv loadv.l
|
|
|
|
loadg loadg.l
|
|
|
|
loada loada.l loadc loadc.l
|
|
|
|
setg setg.l
|
|
|
|
seta seta.l setc setc.l
|
2009-05-31 17:06:04 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
closure argc vargc trycatch copyenv let for tapply
|
|
|
|
add2 sub2 neg largc lvargc
|
|
|
|
loada0 loada1 loadc00 loadc01 call.l tcall.l
|
|
|
|
brne brne.l cadr brnn brnn.l brn brn.l
|
|
|
|
optargs brbound
|
2009-05-31 17:06:04 -04:00
|
|
|
|
|
|
|
dummy_t dummy_f dummy_nil]))
|
2009-03-28 17:39:04 -04:00
|
|
|
(for 0 (1- (length keys))
|
|
|
|
(lambda (i)
|
2009-05-31 17:06:04 -04:00
|
|
|
(put! e (aref keys i) i)))))
|
2009-04-01 22:22:38 -04:00
|
|
|
|
|
|
|
(define arg-counts
|
2009-07-28 00:16:20 -04:00
|
|
|
(table eq? 2 eqv? 2
|
|
|
|
equal? 2 atom? 1
|
|
|
|
not 1 null? 1
|
|
|
|
boolean? 1 symbol? 1
|
|
|
|
number? 1 bound? 1
|
|
|
|
pair? 1 builtin? 1
|
|
|
|
vector? 1 fixnum? 1
|
|
|
|
cons 2 car 1
|
|
|
|
cdr 1 set-car! 2
|
|
|
|
set-cdr! 2 = 2
|
|
|
|
< 2 compare 2
|
|
|
|
aref 2 aset! 3
|
|
|
|
div0 2))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-07-16 21:30:26 -04:00
|
|
|
(define (make-code-emitter) (vector () (table) 0 +inf.0))
|
2009-06-06 17:15:54 -04:00
|
|
|
(define (bcode:code b) (aref b 0))
|
|
|
|
(define (bcode:ctable b) (aref b 1))
|
|
|
|
(define (bcode:nconst b) (aref b 2))
|
2009-07-16 21:30:26 -04:00
|
|
|
(define (bcode:cdepth b d) (aset! b 3 (min (aref b 3) d)))
|
2009-06-06 17:15:54 -04:00
|
|
|
; get an index for a referenced value in a bytecode object
|
|
|
|
(define (bcode:indexfor b v)
|
|
|
|
(let ((const-to-idx (bcode:ctable b))
|
|
|
|
(nconst (bcode:nconst b)))
|
|
|
|
(if (has? const-to-idx v)
|
|
|
|
(get const-to-idx v)
|
|
|
|
(begin (put! const-to-idx v nconst)
|
|
|
|
(prog1 nconst
|
|
|
|
(aset! b 2 (+ nconst 1)))))))
|
2009-03-28 17:39:04 -04:00
|
|
|
(define (emit e inst . args)
|
2009-06-10 20:34:50 -04:00
|
|
|
(if (null? args)
|
2009-07-28 00:16:20 -04:00
|
|
|
(if (and (eq? inst 'car) (pair? (aref e 0))
|
|
|
|
(eq? (car (aref e 0)) 'cdr))
|
|
|
|
(set-car! (aref e 0) 'cadr)
|
2009-07-24 00:20:09 -04:00
|
|
|
(aset! e 0 (cons inst (aref e 0))))
|
2009-06-10 20:34:50 -04:00
|
|
|
(begin
|
2009-07-28 00:16:20 -04:00
|
|
|
(if (memq inst '(loadv loadg setg))
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! args (list (bcode:indexfor e (car args)))))
|
|
|
|
(let ((longform
|
2009-07-28 00:16:20 -04:00
|
|
|
(assq inst '((loadv loadv.l) (loadg loadg.l) (setg setg.l)
|
|
|
|
(loada loada.l) (seta seta.l)))))
|
2009-06-10 20:34:50 -04:00
|
|
|
(if (and longform
|
|
|
|
(> (car args) 255))
|
|
|
|
(set! inst (cadr longform))))
|
|
|
|
(let ((longform
|
2009-07-28 00:16:20 -04:00
|
|
|
(assq inst '((loadc loadc.l) (setc setc.l)))))
|
2009-06-10 20:34:50 -04:00
|
|
|
(if (and longform
|
|
|
|
(or (> (car args) 255)
|
|
|
|
(> (cadr args) 255)))
|
|
|
|
(set! inst (cadr longform))))
|
2009-07-28 00:16:20 -04:00
|
|
|
(if (eq? inst 'loada)
|
2009-06-10 20:34:50 -04:00
|
|
|
(cond ((equal? args '(0))
|
2009-07-28 00:16:20 -04:00
|
|
|
(set! inst 'loada0)
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! args ()))
|
|
|
|
((equal? args '(1))
|
2009-07-28 00:16:20 -04:00
|
|
|
(set! inst 'loada1)
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! args ()))))
|
2009-07-28 00:16:20 -04:00
|
|
|
(if (eq? inst 'loadc)
|
2009-06-10 20:34:50 -04:00
|
|
|
(cond ((equal? args '(0 0))
|
2009-07-28 00:16:20 -04:00
|
|
|
(set! inst 'loadc00)
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! args ()))
|
|
|
|
((equal? args '(0 1))
|
2009-07-28 00:16:20 -04:00
|
|
|
(set! inst 'loadc01)
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! args ()))))
|
2009-07-24 00:20:09 -04:00
|
|
|
|
|
|
|
(let ((lasti (if (pair? (aref e 0))
|
|
|
|
(car (aref e 0)) ()))
|
|
|
|
(bc (aref e 0)))
|
2009-07-28 00:16:20 -04:00
|
|
|
(cond ((and (eq? inst 'brf) (eq? lasti 'not)
|
|
|
|
(eq? (cadr bc) 'null?))
|
|
|
|
(aset! e 0 (cons (car args) (cons 'brn (cddr bc)))))
|
|
|
|
((and (eq? inst 'brf) (eq? lasti 'not))
|
|
|
|
(aset! e 0 (cons (car args) (cons 'brt (cdr bc)))))
|
|
|
|
((and (eq? inst 'brf) (eq? lasti 'eq?))
|
|
|
|
(aset! e 0 (cons (car args) (cons 'brne (cdr bc)))))
|
|
|
|
((and (eq? inst 'brf) (eq? lasti 'null?))
|
|
|
|
(aset! e 0 (cons (car args) (cons 'brnn (cdr bc)))))
|
|
|
|
((and (eq? inst 'brt) (eq? lasti 'null?))
|
|
|
|
(aset! e 0 (cons (car args) (cons 'brn (cdr bc)))))
|
2009-07-24 00:20:09 -04:00
|
|
|
(else
|
|
|
|
(aset! e 0 (nreconc (cons inst args) bc)))))))
|
2009-03-28 17:39:04 -04:00
|
|
|
e)
|
|
|
|
|
|
|
|
(define (make-label e) (gensym))
|
2009-07-28 00:16:20 -04:00
|
|
|
(define (mark-label e l) (emit e 'label l))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
|
|
|
; convert symbolic bytecode representation to a byte array.
|
|
|
|
; labels are fixed-up.
|
|
|
|
(define (encode-byte-code e)
|
2009-05-31 17:06:04 -04:00
|
|
|
(let* ((cl (reverse! e))
|
2009-06-06 17:15:54 -04:00
|
|
|
(v (list->vector cl))
|
2009-06-10 20:34:50 -04:00
|
|
|
(long? (>= (+ (length v) ; 1 byte for each entry, plus...
|
|
|
|
; at most half the entries in this vector can be
|
|
|
|
; instructions accepting 32-bit arguments
|
2009-07-28 00:16:20 -04:00
|
|
|
(* 3 (div0 (length v) 2)))
|
2009-06-06 17:15:54 -04:00
|
|
|
65536)))
|
2009-03-28 17:39:04 -04:00
|
|
|
(let ((n (length v))
|
|
|
|
(i 0)
|
|
|
|
(label-to-loc (table))
|
|
|
|
(fixup-to-label (table))
|
|
|
|
(bcode (buffer))
|
2009-06-10 20:34:50 -04:00
|
|
|
(vi #f)
|
|
|
|
(nxt #f))
|
2009-06-27 19:07:22 -04:00
|
|
|
(io.write bcode #int32(0))
|
2009-03-28 17:39:04 -04:00
|
|
|
(while (< i n)
|
|
|
|
(begin
|
|
|
|
(set! vi (aref v i))
|
2009-07-28 00:16:20 -04:00
|
|
|
(if (eq? vi 'label)
|
2009-03-28 17:39:04 -04:00
|
|
|
(begin (put! label-to-loc (aref v (+ i 1)) (sizeof bcode))
|
|
|
|
(set! i (+ i 2)))
|
|
|
|
(begin
|
2009-03-28 19:46:02 -04:00
|
|
|
(io.write bcode
|
|
|
|
(byte
|
|
|
|
(get Instructions
|
2009-06-10 20:34:50 -04:00
|
|
|
(if long?
|
2009-03-28 19:46:02 -04:00
|
|
|
(case vi
|
2009-07-28 00:16:20 -04:00
|
|
|
(jmp 'jmp.l)
|
|
|
|
(brt 'brt.l)
|
|
|
|
(brf 'brf.l)
|
|
|
|
(brne 'brne.l)
|
|
|
|
(brnn 'brnn.l)
|
|
|
|
(brn 'brn.l)
|
2009-06-10 20:34:50 -04:00
|
|
|
(else vi))
|
2009-03-28 19:46:02 -04:00
|
|
|
vi))))
|
2009-03-28 17:39:04 -04:00
|
|
|
(set! i (+ i 1))
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! nxt (if (< i n) (aref v i) #f))
|
2009-07-28 00:16:20 -04:00
|
|
|
(cond ((memq vi '(jmp brf brt brne brnn brn))
|
2009-06-10 20:34:50 -04:00
|
|
|
(put! fixup-to-label (sizeof bcode) nxt)
|
2009-06-14 22:25:21 -04:00
|
|
|
(io.write bcode ((if long? int32 int16) 0))
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! i (+ i 1)))
|
2009-07-28 00:16:20 -04:00
|
|
|
((eq? vi 'brbound)
|
|
|
|
(io.write bcode (int32 nxt))
|
|
|
|
(set! i (+ i 1))
|
|
|
|
(put! fixup-to-label (sizeof bcode) (aref v i))
|
|
|
|
(io.write bcode (int32 0))
|
|
|
|
(set! i (+ i 1)))
|
2009-06-10 20:34:50 -04:00
|
|
|
((number? nxt)
|
|
|
|
(case vi
|
2009-07-28 00:16:20 -04:00
|
|
|
((loadv.l loadg.l setg.l loada.l seta.l
|
|
|
|
largc lvargc call.l tcall.l)
|
2009-06-14 22:25:21 -04:00
|
|
|
(io.write bcode (int32 nxt))
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! i (+ i 1)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((loadc setc) ; 2 uint8 args
|
2009-06-10 20:34:50 -04:00
|
|
|
(io.write bcode (uint8 nxt))
|
|
|
|
(set! i (+ i 1))
|
|
|
|
(io.write bcode (uint8 (aref v i)))
|
|
|
|
(set! i (+ i 1)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((loadc.l setc.l optargs) ; 2 int32 args
|
2009-06-14 22:25:21 -04:00
|
|
|
(io.write bcode (int32 nxt))
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! i (+ i 1))
|
2009-06-14 22:25:21 -04:00
|
|
|
(io.write bcode (int32 (aref v i)))
|
2009-06-10 20:34:50 -04:00
|
|
|
(set! i (+ i 1)))
|
|
|
|
|
|
|
|
(else
|
|
|
|
; other number arguments are always uint8
|
|
|
|
(io.write bcode (uint8 nxt))
|
|
|
|
(set! i (+ i 1)))))
|
|
|
|
(else #f))))))
|
2009-06-06 17:15:54 -04:00
|
|
|
|
2009-03-28 17:39:04 -04:00
|
|
|
(table.foreach
|
|
|
|
(lambda (addr labl)
|
|
|
|
(begin (io.seek bcode addr)
|
2009-06-14 22:25:21 -04:00
|
|
|
(io.write bcode ((if long? int32 int16)
|
|
|
|
(- (get label-to-loc labl)
|
|
|
|
addr)))))
|
2009-03-28 17:39:04 -04:00
|
|
|
fixup-to-label)
|
|
|
|
(io.tostring! bcode))))
|
|
|
|
|
|
|
|
(define (const-to-idx-vec e)
|
2009-06-06 17:15:54 -04:00
|
|
|
(let ((cvec (vector.alloc (bcode:nconst e))))
|
2009-04-26 18:19:32 -04:00
|
|
|
(table.foreach (lambda (val idx) (aset! cvec idx val))
|
2009-06-06 17:15:54 -04:00
|
|
|
(bcode:ctable e))
|
2009-04-26 18:19:32 -04:00
|
|
|
cvec))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
|
|
|
(define (index-of item lst start)
|
|
|
|
(cond ((null? lst) #f)
|
2009-04-26 18:19:32 -04:00
|
|
|
((eq? item (car lst)) start)
|
|
|
|
(else (index-of item (cdr lst) (+ start 1)))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-26 18:19:32 -04:00
|
|
|
(define (in-env? s env) (any (lambda (e) (memq s e)) env))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
|
|
|
(define (lookup-sym s env lev arg?)
|
|
|
|
(if (null? env)
|
|
|
|
'(global)
|
|
|
|
(let* ((curr (car env))
|
|
|
|
(i (index-of s curr 0)))
|
|
|
|
(if i
|
|
|
|
(if arg?
|
|
|
|
`(arg ,i)
|
|
|
|
`(closed ,lev ,i))
|
|
|
|
(lookup-sym s
|
|
|
|
(cdr env)
|
2009-04-14 20:12:01 -04:00
|
|
|
(if (or arg? (null? curr)) lev (+ lev 1))
|
2009-03-28 17:39:04 -04:00
|
|
|
#f)))))
|
|
|
|
|
2009-07-16 21:30:26 -04:00
|
|
|
; number of non-nulls
|
|
|
|
(define (nnn e) (count (lambda (x) (not (null? x))) e))
|
|
|
|
|
2009-07-20 00:57:17 -04:00
|
|
|
(define (printable? x) (not (iostream? x)))
|
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-sym g env s Is)
|
2009-04-14 20:12:01 -04:00
|
|
|
(let ((loc (lookup-sym s env 0 #t)))
|
2009-03-28 17:39:04 -04:00
|
|
|
(case (car loc)
|
|
|
|
(arg (emit g (aref Is 0) (cadr loc)))
|
2009-07-16 21:30:26 -04:00
|
|
|
(closed (emit g (aref Is 1) (cadr loc) (caddr loc))
|
|
|
|
; update index of most distant captured frame
|
|
|
|
(bcode:cdepth g (- (nnn (cdr env)) 1 (cadr loc))))
|
2009-07-20 00:57:17 -04:00
|
|
|
(else
|
|
|
|
(if (and (constant? s)
|
|
|
|
(printable? (top-level-value s)))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'loadv (top-level-value s))
|
2009-07-20 00:57:17 -04:00
|
|
|
(emit g (aref Is 2) s))))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-if g env tail? x)
|
2009-03-28 17:39:04 -04:00
|
|
|
(let ((elsel (make-label g))
|
2009-05-30 13:04:34 -04:00
|
|
|
(endl (make-label g))
|
|
|
|
(test (cadr x))
|
|
|
|
(then (caddr x))
|
|
|
|
(else (if (pair? (cdddr x))
|
|
|
|
(cadddr x)
|
|
|
|
#f)))
|
|
|
|
(cond ((eq? test #t)
|
|
|
|
(compile-in g env tail? then))
|
|
|
|
((eq? test #f)
|
|
|
|
(compile-in g env tail? else))
|
|
|
|
(else
|
|
|
|
(compile-in g env #f test)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'brf elsel)
|
2009-05-30 13:04:34 -04:00
|
|
|
(compile-in g env tail? then)
|
|
|
|
(if tail?
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'ret)
|
|
|
|
(emit g 'jmp endl))
|
2009-05-30 13:04:34 -04:00
|
|
|
(mark-label g elsel)
|
|
|
|
(compile-in g env tail? else)
|
|
|
|
(mark-label g endl)))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-begin g env tail? forms)
|
|
|
|
(cond ((atom? forms) (compile-in g env tail? #f))
|
2009-03-28 17:39:04 -04:00
|
|
|
((atom? (cdr forms))
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-in g env tail? (car forms)))
|
2009-03-28 17:39:04 -04:00
|
|
|
(else
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-in g env #f (car forms))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'pop)
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-begin g env tail? (cdr forms)))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-prog1 g env x)
|
|
|
|
(compile-in g env #f (cadr x))
|
2009-03-28 17:39:04 -04:00
|
|
|
(if (pair? (cddr x))
|
2009-04-01 22:22:38 -04:00
|
|
|
(begin (compile-begin g env #f (cddr x))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'pop))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-while g env cond body)
|
2009-03-28 17:39:04 -04:00
|
|
|
(let ((top (make-label g))
|
|
|
|
(end (make-label g)))
|
2009-04-14 20:12:01 -04:00
|
|
|
(compile-in g env #f #f)
|
2009-03-28 17:39:04 -04:00
|
|
|
(mark-label g top)
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-in g env #f cond)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'brf end)
|
|
|
|
(emit g 'pop)
|
2009-04-14 20:12:01 -04:00
|
|
|
(compile-in g env #f body)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'jmp top)
|
2009-03-28 17:39:04 -04:00
|
|
|
(mark-label g end)))
|
|
|
|
|
2009-04-15 23:05:38 -04:00
|
|
|
(define (1arg-lambda? func)
|
|
|
|
(and (pair? func)
|
|
|
|
(eq? (car func) 'lambda)
|
|
|
|
(pair? (cdr func))
|
|
|
|
(pair? (cadr func))
|
|
|
|
(length= (cadr func) 1)))
|
|
|
|
|
|
|
|
(define (compile-for g env lo hi func)
|
|
|
|
(if (1arg-lambda? func)
|
|
|
|
(begin (compile-in g env #f lo)
|
|
|
|
(compile-in g env #f hi)
|
|
|
|
(compile-in g env #f func)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'for))
|
2009-04-15 23:05:38 -04:00
|
|
|
(error "for: third form must be a 1-argument lambda")))
|
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-short-circuit g env tail? forms default branch)
|
|
|
|
(cond ((atom? forms) (compile-in g env tail? default))
|
|
|
|
((atom? (cdr forms)) (compile-in g env tail? (car forms)))
|
2009-03-28 17:39:04 -04:00
|
|
|
(else
|
|
|
|
(let ((end (make-label g)))
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-in g env #f (car forms))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'dup)
|
2009-04-01 22:22:38 -04:00
|
|
|
(emit g branch end)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'pop)
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-short-circuit g env tail? (cdr forms) default branch)
|
2009-03-28 17:39:04 -04:00
|
|
|
(mark-label g end)))))
|
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-and g env tail? forms)
|
2009-07-28 00:16:20 -04:00
|
|
|
(compile-short-circuit g env tail? forms #t 'brf))
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-or g env tail? forms)
|
2009-07-28 00:16:20 -04:00
|
|
|
(compile-short-circuit g env tail? forms #f 'brt))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-arglist g env lst)
|
2009-07-20 00:57:17 -04:00
|
|
|
(for-each (lambda (a)
|
|
|
|
(compile-in g env #f a))
|
|
|
|
lst)
|
|
|
|
(length lst))
|
2009-04-01 00:31:49 -04:00
|
|
|
|
2009-04-09 00:04:27 -04:00
|
|
|
(define (argc-error head count)
|
2009-07-28 00:16:20 -04:00
|
|
|
(error "compile error: " head " expects " count
|
|
|
|
(if (= count 1)
|
|
|
|
" argument."
|
|
|
|
" arguments.")))
|
2009-04-16 17:20:15 -04:00
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-app g env tail? x)
|
2009-04-16 17:20:15 -04:00
|
|
|
(let ((head (car x)))
|
|
|
|
(if (and (pair? head)
|
|
|
|
(eq? (car head) 'lambda)
|
2009-06-06 17:15:54 -04:00
|
|
|
(list? (cadr head))
|
2009-07-26 23:34:33 -04:00
|
|
|
(every symbol? (cadr head))
|
2009-07-20 00:57:17 -04:00
|
|
|
(not (length> (cadr head) 255)))
|
2009-04-16 17:20:15 -04:00
|
|
|
(compile-let g env tail? x)
|
|
|
|
(compile-call g env tail? x))))
|
|
|
|
|
|
|
|
(define (compile-let g env tail? x)
|
|
|
|
(let ((head (car x))
|
|
|
|
(args (cdr x)))
|
|
|
|
(unless (length= args (length (cadr head)))
|
2009-07-28 00:16:20 -04:00
|
|
|
(error "apply: incorrect number of arguments to " head))
|
2009-07-16 21:30:26 -04:00
|
|
|
(receive (the-f dept) (compile-f- env head #t)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'loadv the-f)
|
2009-07-16 21:30:26 -04:00
|
|
|
(bcode:cdepth g dept))
|
2009-04-16 17:20:15 -04:00
|
|
|
(let ((nargs (compile-arglist g env args)))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'copyenv)
|
|
|
|
(emit g (if tail? 'tcall 'call) (+ 1 nargs)))))
|
2009-04-16 17:20:15 -04:00
|
|
|
|
2009-06-14 22:25:21 -04:00
|
|
|
(define builtin->instruction
|
2009-07-28 00:16:20 -04:00
|
|
|
(let ((b2i (table number? 'number? cons 'cons
|
|
|
|
fixnum? 'fixnum? equal? 'equal?
|
|
|
|
eq? 'eq? symbol? 'symbol?
|
|
|
|
div0 'div0 builtin? 'builtin?
|
|
|
|
aset! 'aset! - '- boolean? 'boolean? not 'not
|
|
|
|
apply 'apply atom? 'atom?
|
|
|
|
set-cdr! 'set-cdr! / '/
|
|
|
|
function? 'function? vector 'vector
|
|
|
|
list 'list bound? 'bound?
|
|
|
|
< '< * '* cdr 'cdr null? 'null?
|
|
|
|
+ '+ eqv? 'eqv? compare 'compare aref 'aref
|
|
|
|
set-car! 'set-car! car 'car
|
|
|
|
pair? 'pair? = '= vector? 'vector?)))
|
2009-06-14 22:25:21 -04:00
|
|
|
(lambda (b)
|
|
|
|
(get b2i b #f))))
|
2009-04-26 18:19:32 -04:00
|
|
|
|
2009-07-20 00:57:17 -04:00
|
|
|
(define (compile-builtin-call g env tail? x head b nargs)
|
|
|
|
(let ((count (get arg-counts b #f)))
|
|
|
|
(if (and count
|
|
|
|
(not (length= (cdr x) count)))
|
|
|
|
(argc-error head count))
|
|
|
|
(case b ; handle special cases of vararg builtins
|
2009-07-28 00:16:20 -04:00
|
|
|
(list (if (= nargs 0) (emit g 'loadnil) (emit g b nargs)))
|
|
|
|
(+ (cond ((= nargs 0) (emit g 'load0))
|
|
|
|
((= nargs 2) (emit g 'add2))
|
|
|
|
(else (emit g b nargs))))
|
|
|
|
(- (cond ((= nargs 0) (argc-error head 1))
|
|
|
|
((= nargs 1) (emit g 'neg))
|
|
|
|
((= nargs 2) (emit g 'sub2))
|
|
|
|
(else (emit g b nargs))))
|
|
|
|
(* (if (= nargs 0) (emit g 'load1)
|
|
|
|
(emit g b nargs)))
|
|
|
|
(/ (if (= nargs 0)
|
|
|
|
(argc-error head 1)
|
|
|
|
(emit g b nargs)))
|
|
|
|
(vector (if (= nargs 0)
|
|
|
|
(emit g 'loadv [])
|
|
|
|
(emit g b nargs)))
|
|
|
|
(apply (if (< nargs 2)
|
|
|
|
(argc-error head 2)
|
|
|
|
(emit g (if tail? 'tapply 'apply) nargs)))
|
2009-07-20 00:57:17 -04:00
|
|
|
(else (emit g b)))))
|
|
|
|
|
2009-04-16 17:20:15 -04:00
|
|
|
(define (compile-call g env tail? x)
|
2009-04-01 00:31:49 -04:00
|
|
|
(let ((head (car x)))
|
2009-03-28 17:39:04 -04:00
|
|
|
(let ((head
|
|
|
|
(if (and (symbol? head)
|
|
|
|
(not (in-env? head env))
|
|
|
|
(bound? head)
|
|
|
|
(constant? head)
|
2009-04-17 10:41:15 -04:00
|
|
|
(builtin? (top-level-value head)))
|
|
|
|
(top-level-value head)
|
2009-03-28 17:39:04 -04:00
|
|
|
head)))
|
2009-07-20 00:57:17 -04:00
|
|
|
(if (length> (cdr x) 255)
|
2009-07-21 22:10:20 -04:00
|
|
|
; more than 255 arguments, need long versions of instructions
|
|
|
|
(begin (compile-in g env #f head)
|
|
|
|
(let ((nargs (compile-arglist g env (cdr x))))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g (if tail? 'tcall.l 'call.l) nargs)))
|
2009-07-20 00:57:17 -04:00
|
|
|
(let ((b (and (builtin? head)
|
|
|
|
(builtin->instruction head))))
|
2009-07-24 00:20:09 -04:00
|
|
|
(if (and (eq? head 'cadr)
|
|
|
|
(not (in-env? head env))
|
|
|
|
(equal? (top-level-value 'cadr) cadr)
|
|
|
|
(length= x 2))
|
|
|
|
(begin (compile-in g env #f (cadr x))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'cadr))
|
2009-07-24 00:20:09 -04:00
|
|
|
(begin
|
|
|
|
(if (not b)
|
|
|
|
(compile-in g env #f head))
|
|
|
|
(let ((nargs (compile-arglist g env (cdr x))))
|
|
|
|
(if b
|
|
|
|
(compile-builtin-call g env tail? x head b nargs)
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g (if tail? 'tcall 'call) nargs))))))))))
|
2009-04-01 22:22:38 -04:00
|
|
|
|
2009-07-17 22:16:18 -04:00
|
|
|
(define (expand-define form body)
|
|
|
|
(if (symbol? form)
|
|
|
|
`(set! ,form ,(car body))
|
|
|
|
`(set! ,(car form)
|
|
|
|
(lambda ,(cdr form) ,@body . ,(car form)))))
|
|
|
|
|
2009-04-26 23:21:53 -04:00
|
|
|
(define (fits-i8 x) (and (fixnum? x) (>= x -128) (<= x 127)))
|
|
|
|
|
2009-04-01 22:22:38 -04:00
|
|
|
(define (compile-in g env tail? x)
|
2009-07-28 00:16:20 -04:00
|
|
|
(cond ((symbol? x) (compile-sym g env x [loada loadc loadg]))
|
2009-03-28 17:39:04 -04:00
|
|
|
((atom? x)
|
2009-07-28 00:16:20 -04:00
|
|
|
(cond ((eq? x 0) (emit g 'load0))
|
|
|
|
((eq? x 1) (emit g 'load1))
|
|
|
|
((eq? x #t) (emit g 'loadt))
|
|
|
|
((eq? x #f) (emit g 'loadf))
|
|
|
|
((eq? x ()) (emit g 'loadnil))
|
|
|
|
((fits-i8 x) (emit g 'loadi8 x))
|
|
|
|
(else (emit g 'loadv x))))
|
2009-03-28 17:39:04 -04:00
|
|
|
(else
|
|
|
|
(case (car x)
|
2009-07-28 00:16:20 -04:00
|
|
|
(quote (emit g 'loadv (cadr x)))
|
2009-04-01 22:22:38 -04:00
|
|
|
(if (compile-if g env tail? x))
|
|
|
|
(begin (compile-begin g env tail? (cdr x)))
|
2009-04-14 20:12:01 -04:00
|
|
|
(prog1 (compile-prog1 g env x))
|
2009-07-16 21:30:26 -04:00
|
|
|
(lambda (receive (the-f dept) (compile-f- env x)
|
2009-07-28 00:16:20 -04:00
|
|
|
(begin (emit g 'loadv the-f)
|
2009-07-16 21:30:26 -04:00
|
|
|
(bcode:cdepth g dept)
|
|
|
|
(if (< dept (nnn env))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'closure)))))
|
2009-04-01 22:22:38 -04:00
|
|
|
(and (compile-and g env tail? (cdr x)))
|
|
|
|
(or (compile-or g env tail? (cdr x)))
|
2009-04-14 20:12:01 -04:00
|
|
|
(while (compile-while g env (cadr x) (cons 'begin (cddr x))))
|
2009-04-15 23:05:38 -04:00
|
|
|
(for (compile-for g env (cadr x) (caddr x) (cadddr x)))
|
2009-04-26 18:19:32 -04:00
|
|
|
(return (compile-in g env #t (cadr x))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'ret))
|
2009-04-01 22:22:38 -04:00
|
|
|
(set! (compile-in g env #f (caddr x))
|
2009-07-28 00:16:20 -04:00
|
|
|
(compile-sym g env (cadr x) [seta setc setg]))
|
2009-07-17 22:16:18 -04:00
|
|
|
(define (compile-in g env tail?
|
|
|
|
(expand-define (cadr x) (cddr x))))
|
2009-04-01 22:22:38 -04:00
|
|
|
(trycatch (compile-in g env #f `(lambda () ,(cadr x)))
|
2009-04-15 23:05:38 -04:00
|
|
|
(unless (1arg-lambda? (caddr x))
|
|
|
|
(error "trycatch: second form must be a 1-argument lambda"))
|
2009-04-01 22:22:38 -04:00
|
|
|
(compile-in g env #f (caddr x))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'trycatch))
|
2009-04-01 22:22:38 -04:00
|
|
|
(else (compile-app g env tail? x))))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-16 17:20:15 -04:00
|
|
|
(define (compile-f env f . let?)
|
2009-07-16 21:30:26 -04:00
|
|
|
(receive (ff ignore)
|
|
|
|
(apply compile-f- env f let?)
|
|
|
|
ff))
|
|
|
|
|
2009-07-17 22:16:18 -04:00
|
|
|
(define get-defined-vars
|
|
|
|
(letrec ((get-defined-vars-
|
|
|
|
(lambda (expr)
|
|
|
|
(cond ((atom? expr) ())
|
|
|
|
((and (eq? (car expr) 'define)
|
|
|
|
(pair? (cdr expr)))
|
|
|
|
(or (and (symbol? (cadr expr))
|
|
|
|
(list (cadr expr)))
|
|
|
|
(and (pair? (cadr expr))
|
|
|
|
(symbol? (caadr expr))
|
|
|
|
(list (caadr expr)))
|
|
|
|
()))
|
|
|
|
((eq? (car expr) 'begin)
|
|
|
|
(apply append (map get-defined-vars- (cdr expr))))
|
|
|
|
(else ())))))
|
|
|
|
(lambda (expr) (delete-duplicates (get-defined-vars- expr)))))
|
|
|
|
|
2009-07-26 23:34:33 -04:00
|
|
|
(define (lambda-vars l)
|
|
|
|
(define (check-formals l o)
|
|
|
|
(or
|
|
|
|
(null? l) (symbol? l)
|
|
|
|
(and
|
|
|
|
(pair? l)
|
|
|
|
(or (symbol? (car l))
|
|
|
|
(and (pair? (car l))
|
|
|
|
(or (every pair? (cdr l))
|
2009-07-28 00:16:20 -04:00
|
|
|
(error "compile error: invalid argument list "
|
|
|
|
o ". optional arguments must come last.")))
|
|
|
|
(error "compile error: invalid formal argument " (car l)
|
|
|
|
" in list " o))
|
2009-07-26 23:34:33 -04:00
|
|
|
(check-formals (cdr l) o))
|
|
|
|
(if (eq? l o)
|
2009-07-28 00:16:20 -04:00
|
|
|
(error "compile error: invalid argument list " o)
|
|
|
|
(error "compile error: invalid formal argument " l " in list " o))))
|
2009-07-26 23:34:33 -04:00
|
|
|
(check-formals l l)
|
|
|
|
(map (lambda (s) (if (pair? s) (car s) s))
|
|
|
|
(to-proper l)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
(define (emit-optional-arg-inits g env opta vars i)
|
|
|
|
; i is the lexical var index of the opt arg to process next
|
|
|
|
(if (pair? opta)
|
|
|
|
(let ((nxt (make-label g)))
|
|
|
|
(emit g 'brbound i nxt)
|
|
|
|
(compile-in g (cons (list-head vars i) env) #f (cadar opta))
|
|
|
|
(emit g 'seta i)
|
|
|
|
(emit g 'pop)
|
|
|
|
(mark-label g nxt)
|
|
|
|
(emit-optional-arg-inits g env (cdr opta) vars (+ i 1)))))
|
|
|
|
|
2009-07-17 22:16:18 -04:00
|
|
|
(define compile-f-
|
|
|
|
(let ((*defines-processed-token* (gensym)))
|
|
|
|
; to eval a top-level expression we need to avoid internal define
|
|
|
|
(set-top-level-value!
|
|
|
|
'compile-thunk
|
|
|
|
(lambda (expr)
|
|
|
|
(compile `(lambda () ,expr . ,*defines-processed-token*))))
|
|
|
|
|
|
|
|
(lambda (env f . let?)
|
|
|
|
; convert lambda to one body expression and process internal defines
|
|
|
|
(define (lambda-body e)
|
|
|
|
(let ((B (if (pair? (cddr e))
|
|
|
|
(if (pair? (cdddr e))
|
|
|
|
(cons 'begin (cddr e))
|
|
|
|
(caddr e))
|
|
|
|
#f)))
|
|
|
|
(let ((V (get-defined-vars B)))
|
|
|
|
(if (null? V)
|
|
|
|
B
|
|
|
|
(cons (list* 'lambda V B *defines-processed-token*)
|
|
|
|
(map (lambda (x) #f) V))))))
|
|
|
|
|
|
|
|
(let ((g (make-code-emitter))
|
|
|
|
(args (cadr f))
|
2009-07-28 00:16:20 -04:00
|
|
|
(atail (lastcdr (cadr f)))
|
2009-07-26 23:34:33 -04:00
|
|
|
(vars (lambda-vars (cadr f)))
|
|
|
|
(opta (filter pair? (cadr f)))
|
2009-07-17 22:16:18 -04:00
|
|
|
(name (if (eq? (lastcdr f) *defines-processed-token*)
|
|
|
|
'lambda
|
|
|
|
(lastcdr f))))
|
2009-07-28 00:16:20 -04:00
|
|
|
(let* ((nargs (if (atom? args) 0 (length args)))
|
|
|
|
(nreq (- nargs (length opta))))
|
2009-07-26 23:34:33 -04:00
|
|
|
|
|
|
|
; emit argument checking prologue
|
|
|
|
(if (not (null? opta))
|
2009-07-28 00:16:20 -04:00
|
|
|
(begin (emit g 'optargs (if (null? atail) nreq (- nreq)) nargs)
|
|
|
|
(emit-optional-arg-inits g env opta vars nreq)))
|
2009-07-26 23:34:33 -04:00
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
(cond ((not (null? let?)) (emit g 'let))
|
|
|
|
((> nargs 255) (emit g (if (null? atail)
|
|
|
|
'largc 'lvargc)
|
2009-07-26 23:34:33 -04:00
|
|
|
nargs))
|
2009-07-28 00:16:20 -04:00
|
|
|
((not (null? atail)) (emit g 'vargc nargs))
|
|
|
|
((null? opta) (emit g 'argc nargs)))
|
2009-07-26 23:34:33 -04:00
|
|
|
|
|
|
|
; compile body and return
|
|
|
|
(compile-in g (cons vars env) #t
|
|
|
|
(if (eq? (lastcdr f) *defines-processed-token*)
|
|
|
|
(caddr f)
|
|
|
|
(lambda-body f)))
|
2009-07-28 00:16:20 -04:00
|
|
|
(emit g 'ret)
|
2009-07-26 23:34:33 -04:00
|
|
|
(values (function (encode-byte-code (bcode:code g))
|
|
|
|
(const-to-idx-vec g) name)
|
|
|
|
(aref g 3)))))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-14 20:12:01 -04:00
|
|
|
(define (compile f) (compile-f () f))
|
|
|
|
|
2009-06-14 22:25:21 -04:00
|
|
|
(define (ref-int32-LE a i)
|
|
|
|
(int32 (+ (ash (aref a (+ i 0)) 0)
|
|
|
|
(ash (aref a (+ i 1)) 8)
|
|
|
|
(ash (aref a (+ i 2)) 16)
|
|
|
|
(ash (aref a (+ i 3)) 24))))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-06-14 22:25:21 -04:00
|
|
|
(define (ref-int16-LE a i)
|
|
|
|
(int16 (+ (ash (aref a (+ i 0)) 0)
|
|
|
|
(ash (aref a (+ i 1)) 8))))
|
2009-03-28 19:46:02 -04:00
|
|
|
|
2009-03-28 17:39:04 -04:00
|
|
|
(define (hex5 n)
|
2009-05-06 22:10:52 -04:00
|
|
|
(string.lpad (number->string n 16) 5 #\0))
|
2009-03-28 17:39:04 -04:00
|
|
|
|
2009-04-26 18:19:32 -04:00
|
|
|
(define (disassemble f . lev?)
|
|
|
|
(if (null? lev?)
|
|
|
|
(begin (disassemble f 0)
|
|
|
|
(newline)
|
|
|
|
(return #t)))
|
2009-05-12 21:13:40 -04:00
|
|
|
(let ((lev (car lev?))
|
|
|
|
(code (function:code f))
|
|
|
|
(vals (function:vals f)))
|
|
|
|
(define (print-val v)
|
|
|
|
(if (and (function? v) (not (builtin? v)))
|
|
|
|
(begin (princ "\n")
|
|
|
|
(disassemble v (+ lev 1)))
|
|
|
|
(print v)))
|
2009-06-27 19:07:22 -04:00
|
|
|
(dotimes (xx lev) (princ "\t"))
|
|
|
|
(princ "maxstack " (ref-int32-LE code 0) "\n")
|
|
|
|
(let ((i 4)
|
2009-05-12 21:13:40 -04:00
|
|
|
(N (length code)))
|
|
|
|
(while (< i N)
|
|
|
|
; find key whose value matches the current byte
|
|
|
|
(let ((inst (table.foldl (lambda (k v z)
|
|
|
|
(or z (and (eq? v (aref code i))
|
|
|
|
k)))
|
|
|
|
#f Instructions)))
|
2009-06-27 19:07:22 -04:00
|
|
|
(if (> i 4) (newline))
|
2009-05-12 21:13:40 -04:00
|
|
|
(dotimes (xx lev) (princ "\t"))
|
2009-06-27 19:07:22 -04:00
|
|
|
(princ (hex5 (- i 4)) ": "
|
2009-07-28 00:16:20 -04:00
|
|
|
(string inst) "\t")
|
2009-05-12 21:13:40 -04:00
|
|
|
(set! i (+ i 1))
|
|
|
|
(case inst
|
2009-07-28 00:16:20 -04:00
|
|
|
((loadv.l loadg.l setg.l)
|
2009-06-14 22:25:21 -04:00
|
|
|
(print-val (aref vals (ref-int32-LE code i)))
|
2009-05-12 21:13:40 -04:00
|
|
|
(set! i (+ i 4)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((loadv loadg setg)
|
2009-05-12 21:13:40 -04:00
|
|
|
(print-val (aref vals (aref code i)))
|
|
|
|
(set! i (+ i 1)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((loada seta call tcall list + - * / vector
|
|
|
|
argc vargc loadi8 apply tapply)
|
2009-05-12 21:13:40 -04:00
|
|
|
(princ (number->string (aref code i)))
|
|
|
|
(set! i (+ i 1)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((loada.l seta.l largc lvargc call.l tcall.l)
|
2009-06-14 22:25:21 -04:00
|
|
|
(princ (number->string (ref-int32-LE code i)))
|
2009-06-06 17:15:54 -04:00
|
|
|
(set! i (+ i 4)))
|
2009-07-28 00:16:20 -04:00
|
|
|
|
|
|
|
((loadc setc)
|
2009-05-12 21:13:40 -04:00
|
|
|
(princ (number->string (aref code i)) " ")
|
|
|
|
(set! i (+ i 1))
|
|
|
|
(princ (number->string (aref code i)))
|
|
|
|
(set! i (+ i 1)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((loadc.l setc.l optargs)
|
2009-06-14 22:25:21 -04:00
|
|
|
(princ (number->string (ref-int32-LE code i)) " ")
|
2009-06-06 17:15:54 -04:00
|
|
|
(set! i (+ i 4))
|
2009-06-14 22:25:21 -04:00
|
|
|
(princ (number->string (ref-int32-LE code i)))
|
2009-06-06 17:15:54 -04:00
|
|
|
(set! i (+ i 4)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((brbound)
|
|
|
|
(princ (number->string (ref-int32-LE code i)) " ")
|
|
|
|
(set! i (+ i 4))
|
|
|
|
(princ "@" (hex5 (+ i -4 (ref-int32-LE code i))))
|
|
|
|
(set! i (+ i 4)))
|
|
|
|
|
|
|
|
((jmp brf brt brne brnn brn)
|
2009-07-08 01:53:29 -04:00
|
|
|
(princ "@" (hex5 (+ i -4 (ref-int16-LE code i))))
|
2009-05-12 21:13:40 -04:00
|
|
|
(set! i (+ i 2)))
|
|
|
|
|
2009-07-28 00:16:20 -04:00
|
|
|
((jmp.l brf.l brt.l brne.l brnn.l brn.l)
|
2009-07-08 01:53:29 -04:00
|
|
|
(princ "@" (hex5 (+ i -4 (ref-int32-LE code i))))
|
2009-05-12 21:13:40 -04:00
|
|
|
(set! i (+ i 4)))
|
|
|
|
|
|
|
|
(else #f)))))))
|
2009-04-19 18:22:17 -04:00
|
|
|
|
2009-03-28 17:39:04 -04:00
|
|
|
#t
|