('print '(abs String Void))
(else (error #f "Couldn't find type for builtin" x))))
-(define (check-let env x)
+(define (check-let dls env x)
; acc is a pair of (env . annotated bindings)
(define (process-component acc comps)
(map
(lambda (c)
(let ([body (cadr (assoc c (let-bindings x)))])
- (check scc-env body)))
+ (check dls scc-env body)))
comps)]
; collect all the constraints in the scc
[cs
[new-env (car results)]
[annotated-bindings (cdr results)]
- [body-results (map (lambda (body) (check new-env body)) (let-body x))]
+ [body-results (map (lambda (body) (check dls new-env body)) (let-body x))]
[let-type (cadr (last body-results))]
[cs (fold-left (lambda (acc cs) (constraint-merge acc cs)) '() (map car body-results))]
[annotated `((let ,annotated-bindings ,@(map caddr body-results)) : ,let-type)])
(list cs let-type annotated)))
-(define (check-app env x)
+(define (check-app dls env x)
(if (eqv? (car x) (cadr x))
; recursive function (f f)
; TODO: what about ((f a) f)????
(list cs resolved-return-type annotated)))
; regular function
- (let* ([arg-type-res (check env (cadr x))]
+ (let* ([arg-type-res (check dls env (cadr x))]
[arg-type (cadr arg-type-res)]
- [func-type-res (check env (car x))]
+ [func-type-res (check dls env (car x))]
[func-type (cadr func-type-res)]
; f ~ a -> t0
(list cs return-type annotated))
(error #f "not a function"))))
+(define (check-case dls env x)
+
+ (define (check-match switch-type x)
+ (let ([pattern (car x)]
+ [expr (cadr x)])
+ (if (eqv? (ast-type pattern) 'app)
+ ; a pattern match with bindings
+ (let ([sum (assoc (car pattern) (cdr (assoc switch-type dls)))])
+ (unless sum (error #f "can't pattern match ~a with ~a" switch-type pattern))
+ (let* ([names (cdr pattern)]
+ [types (cdr sum)]
+ [new-env (fold-left env-insert env names types)])
+ (check dls new-env expr)))
+ ; a pattern match without bindings
+ (check dls env expr))))
+
+ (let* ([switch-type-res (check dls env (case-switch x))]
+ [switch-type (cadr switch-type-res)]
+
+ [case-expr-type-res (map (lambda (x) (check-match switch-type x)) (case-cases x))]
+ [case-expr-types (map cadr case-expr-type-res)]
+
+ [case-expr-equality-cs (fold-left constraint-merge '()
+ (map (lambda (t) (~ t (car case-expr-types)))
+ (cdr case-expr-types)))]
+
+ [resolved-type (substitute case-expr-equality-cs (car case-expr-types))]
+
+ [annotated `((case (,(case-expr x) : ,switch-type)
+ ,(map (lambda (c e et)
+ `(,c (,e : ,et)))
+ (map car (case-cases x))
+ (map cadr (case-cases x))
+ case-expr-types)) : ,resolved-type)]
+
+ [cs (fold-left constraint-merge '()
+ (cons (car switch-type-res) case-expr-equality-cs))])
+ (list cs resolved-type annotated)))
+
; returns a list (constraints type annotated)
-(define (check env x)
+(define (check dls env x)
(define (make-result cs type)
(list cs type `(,x : ,type)))
;; (display "check: ")
('builtin (make-result '() (builtin-type x)))
('if
- (let* ((cond-type-res (check env (cadr x)))
- (then-type-res (check env (caddr x)))
- (else-type-res (check env (cadddr x)))
+ (let* ((cond-type-res (check dls env (cadr x)))
+ (then-type-res (check dls env (caddr x)))
+ (else-type-res (check dls env (cadddr x)))
(then-eq-else-cs (~ (cadr then-type-res)
(cadr else-type-res)))
(cs (constraint-merge
(list cs return-type annotated)))
('var (make-result '() (env-lookup env x)))
- ('let (check-let env x))
+ ('let (check-let dls env x))
('lambda
(let* ([new-env (env-insert env (lambda-arg x) (fresh-tvar))]
- [body-type-res (check new-env (lambda-body x))]
+ [body-type-res (check dls new-env (lambda-body x))]
[cs (car body-type-res)]
[subd-env (substitute-env (car body-type-res) new-env)]
[arg-type (env-lookup subd-env (lambda-arg x))]
annotated)))
- ('app (check-app env x))
- ['case
- (let* ([expr-type-res (check env (case-expr x))]
- [expr-type (cadr expr-type-res)]
- [case-match-type-res (map (lambda (x) (check env x))
- (map car (case-cases x)))]
- [case-match-types (map cadr case-match-type-res)]
+ ('app (check-app dls env x))
+ ['case (check-case dls env x)])))
- [case-expr-type-res (map (lambda (x) (check env x))
- (map cadr (case-cases x)))]
- [case-expr-types (map cadr case-expr-type-res)]
-
- [case-match-equality-cs (fold-left constraint-merge '()
- (map (lambda (t) (~ t expr-type)) case-match-types))]
-
- [case-expr-equality-cs (fold-left constraint-merge '()
- (map (lambda (t) (~ t (car case-expr-types)))
- (cdr case-expr-types)))]
-
- [resolved-type (substitute case-expr-equality-cs (car case-expr-types))]
-
- [annotated `((case (,(case-expr x) : ,expr-type)
- ,(map (lambda (c e et)
- `(,c (,e : ,et)))
- (map car (case-cases x))
- (map cadr (case-cases x))
- case-expr-types)) : ,resolved-type)]
-
- [cs (fold-left constraint-merge '()
- (append case-match-equality-cs
- case-expr-equality-cs
- (car expr-type-res)))])
- (list cs resolved-type annotated))])))
;; (display "result of ")
;; (display x)
; we typecheck the lambda calculus only (only single arg lambdas)
(define (typecheck prog)
(let ([expanded (expand-pattern-matches prog)])
- (cadr (check (init-adts-env expanded) (normalize (program-body expanded))))))
+ (cadr (check (program-data-layouts prog)
+ (init-adts-env expanded)
+ (normalize (program-body expanded))))))
; before passing annotated types onto codegen
projects / scheme.git / blobdiff