suzanne.soy/typed-racket
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixnums and integers can now be coerced for complex operations.

Browse Source 
original commit: 443d8b9f917055d051844bb1cd4078a75dcdd592
This commit is contained in:
Vincent St-Amour 2010-07-14 10:24:14 -04:00
parent b6597e15ed
commit 4da58f05c4
3 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
collects/tests/typed-scheme/optimizer/generic/inexact-complex-fixnum.rkt Normal file
View File

@ -0,0 +1,3 @@
#lang typed/scheme #:optimize
(require racket/unsafe/ops)
(+ 2 1.0+2.0i 3.0+6.0i)

2
collects/typed-scheme/optimizer/float.rkt
View File

@ -7,7 +7,7 @@
(types abbrev type-table utils subtype)
(optimizer utils fixnum))
(provide float-opt-expr)
(provide float-opt-expr float-expr int-expr)
(define (mk-float-tbl generic)

33
collects/typed-scheme/optimizer/inexact-complex.rkt
View File

@ -4,7 +4,7 @@
"../utils/utils.rkt"
(for-template scheme/base scheme/math scheme/flonum scheme/unsafe/ops)
(types abbrev type-table utils subtype)
(optimizer utils))
(optimizer utils float fixnum))
(provide inexact-complex-opt-expr)
@ -161,15 +161,6 @@
#`(#,@(append (syntax->list #'(c.bindings ...))
(list #'(imag-part (unsafe-fl- 0.0 c.imag-part)))))))
(pattern e:expr
;; special handling of inexact reals
#:when (subtypeof? #'e -Flonum)
#:with real-part (unboxed-gensym)
#:with imag-part #f
#:with (bindings ...)
#`((real-part #,((optimize) #'e))))
(pattern e:expr
;; can't work on inexact reals, which are a subtype of inexact
;; complexes, so this has to be equality
#:when (isoftype? #'e -InexactComplex)
#:with e* (unboxed-gensym)
#:with real-part (unboxed-gensym)
@ -177,7 +168,23 @@
#:with (bindings ...)
#`((e* #,((optimize) #'e))
(real-part (unsafe-flreal-part e*))
(imag-part (unsafe-flimag-part e*)))))
(imag-part (unsafe-flimag-part e*))))
;; special handling of reals
(pattern e:float-expr
#:with real-part (unboxed-gensym)
#:with imag-part #f
#:with (bindings ...)
#`((real-part #,((optimize) #'e))))
(pattern e:fixnum-expr
#:with real-part (unboxed-gensym)
#:with imag-part #f
#:with (bindings ...)
#`((real-part (unsafe-fx->fl #,((optimize) #'e)))))
(pattern e:int-expr
#:with real-part (unboxed-gensym)
#:with imag-part #f
#:with (bindings ...)
#`((real-part (->fl #,((optimize) #'e))))))
(define-syntax-class inexact-complex-unary-op
(pattern (~or (~literal real-part) (~literal flreal-part)) #:with unsafe #'unsafe-flreal-part)
@ -188,7 +195,7 @@
(define-syntax-class inexact-complex-expr
(pattern e:expr
#:when (subtypeof? #'e -InexactComplex)
#:when (isoftype? #'e -InexactComplex)
#:with opt ((optimize) #'e)))
(define-syntax-class inexact-complex-opt-expr
@ -196,7 +203,7 @@
#:with opt
(begin (log-optimization "unary inexact complex" #'op)
#'(op.unsafe n.opt)))
(pattern (~and exp (#%plain-app op:inexact-complex-binary-op e:inexact-complex-expr ...))
(pattern (~and exp (#%plain-app op:inexact-complex-binary-op e:expr ...))
#:when (isoftype? #'exp -InexactComplex)
#:with exp*:unboxed-inexact-complex-opt-expr #'exp
#:with opt