(let* ((heap-offsets (map (lambda (i) (+ 8 (* 8 i)))
(range 0 (length captured))))) ; 4, 12, 20, etc.
+ (emit "## creating closure")
+
(emit "movq heap_start@GOTPCREL(%rip), %rbx")
(emit "movq (%rbx), %rax") ; %rax = heap addr of closure
; point heap_start to next space
(emit "addq $~a, (%rbx)" (+ 8 (* 8 (length captured))))
+ (emit "## storing address to lambda")
; store the address to the lambda code
(emit "movq ~a@GOTPCREL(%rip), %rbx" label)
(emit "movq %rbx, 0(%rax)")
+ (emit "## storing captives")
; store the captured vars
(for-each
(lambda (var-name new-offset)
- (emit "movq ~a(%rbp), ~a(rax)"
- (cdr (assoc var-name env))
- new-offset))
+ (begin
+ (emit "movq ~a(%rbp), %rbx" (cdr (assoc var-name env)))
+ (emit "movq %rbx, ~a(%rax)" new-offset)))
captured
heap-offsets)))
(stack-offsets (map (lambda (i)
(* (- wordsize) i))
- (range 1 (length params))))
+ (range 0 (length params))))
(env (map cons params stack-offsets)))
(emit "~a:" label)
(newline)
(emit "push %rbp") ; preserve caller's base pointer
+
(emit "movq %rsp, %rbp") ; set up our own base pointer
+ (emit "subq $8, %rbp")
; load the captured vars onto the stack
(for-each
(lambda (i)
- (emit "movq ~a(~a), ~a(%rbp)"
- i (param-register 0) (* (- wordsize) i)))
+ (begin
+ (emit "movq ~a(~a), %rbx" i (param-register 0))
+ (emit "movq %rbx, ~a(%rbp)" (* (- wordsize) i))))
(range 0 (length captured)))
; load the args onto the stack
(for-each
(lambda (i)
- (emit "movq ~a, ~a(%rbp)"
- (param-register i) (* (- wordsize) i)))
- (range 1 (length args)))
+ (begin
+ (emit "movq ~a, %rbx" (param-register (+ 1 i)))
+ (emit "movq %rbx, ~a(%rbp)"
+ (* (- wordsize)
+ (+ (length captured) i)))))
+ (range 0 (length args)))
(codegen-expr body (* (- wordsize) (+ 1 (length params))) env)
('+ "_add")
('- "_sub")
('* "_mul")
+ ('! "_not")
+ ('= "_eq")
('bool->int "_bool2int")
- (else (error #f "fill this out"))))
+ ('print "_print")
+ (else (error #f "don't know this builtin"))))
(define (builtin-args e)
(case e
('+ '(x y))
('- '(x y))
('* '(x y))
+ ('! '(x))
+ ('= '(x y))
('bool->int '(x))
- (else (error #f "fill this out"))))
+ ('print '(x))
+ (else (error #f "don't know this builtin"))))
(define (add-builtin-lambda e)
(let* [(label (builtin-name e))
('app (append
; if a builtin is used as a function, don't generate lambda
(if (eqv? 'builtin (ast-type (car e)))
- '()
+ (list (car e))
(list (extract (car e))))
(map extract (cdr e))))