3670916a11
Replaced pointwise operators with macros that expand to applications of `array-map'; allows more precise return types and reduces compilation time Changed literal array syntax to use #() to delimit rows instead of [] (still suggest using square parens, though) Minor refactoring Fixed a macro so that the only problem with "array-tests.rkt" now is that typed/rackunit is b0rked
61 lines
2.5 KiB
Racket
61 lines
2.5 KiB
Racket
#lang typed/racket
|
|
|
|
(require typed/racket/base
|
|
math/array
|
|
typed/rackunit)
|
|
|
|
(: mandelbrot (Float Integer -> (Array Integer)))
|
|
(define (mandelbrot step max-iters)
|
|
(define xs (sequence->list (in-range -2.0 0.800001 step)))
|
|
(define ys (sequence->list (in-range -1.4 1.400001 step)))
|
|
(define x ((inst list*->array Float) xs flonum?))
|
|
(define y (array-slice-ref ((inst list*->array Float) ys flonum?) (list ::... (::new 1))))
|
|
(define c (array->fcarray (array+ x (array-scale y +1.0i))))
|
|
(define-values (z divtime)
|
|
(for/fold: ([z : FCArray c]
|
|
[divtime : (Array Integer) (make-array (array-shape c) max-iters)]
|
|
) ([i (in-range max-iters)])
|
|
(let ([z (fcarray+ (fcarray-sqr z) c)])
|
|
(values z (array-strict (array-if (array-and (flarray> (fcarray-magnitude z) (flarray 4.0))
|
|
(array= divtime (array max-iters)))
|
|
(array i)
|
|
divtime))))))
|
|
divtime)
|
|
|
|
(check-equal? (mandelbrot 0.2 20)
|
|
(array #[#[0 1 1 1 1 1 1 1 1 1 1 1 1 1 1]
|
|
#[1 1 1 1 1 1 2 2 2 3 2 2 1 1 1]
|
|
#[1 1 1 1 1 2 2 2 3 6 20 3 2 1 1]
|
|
#[1 1 1 2 2 2 3 4 5 20 17 4 3 2 1]
|
|
#[1 1 2 2 3 3 4 11 20 20 20 10 14 2 2]
|
|
#[1 2 3 4 6 6 6 20 20 20 20 20 9 3 2]
|
|
#[2 3 4 6 18 20 14 20 20 20 20 20 20 3 2]
|
|
#[20 20 20 20 20 20 20 20 20 20 20 20 5 3 2]
|
|
#[2 3 4 6 18 20 14 20 20 20 20 20 20 3 2]
|
|
#[1 2 3 4 6 6 6 20 20 20 20 20 9 3 2]
|
|
#[1 1 2 2 3 3 4 11 20 20 20 10 14 2 2]
|
|
#[1 1 1 2 2 2 3 4 5 20 17 4 3 2 1]
|
|
#[1 1 1 1 1 2 2 2 3 6 20 3 2 1 1]
|
|
#[1 1 1 1 1 1 2 2 2 3 2 2 1 1 1]
|
|
#[0 1 1 1 1 1 1 1 1 1 1 1 1 1 1]]))
|
|
|
|
(begin
|
|
(require images/flomap)
|
|
|
|
(: array->flomap ((Array Real) -> flomap))
|
|
(define (array->flomap arr)
|
|
(let ([arr (array->flarray arr)])
|
|
(define ds (array-shape arr))
|
|
(match ds
|
|
[(vector h w) (flomap (flarray-data arr) 1 w h)]
|
|
[(vector h w c) (flomap (flarray-data arr) c w h)]
|
|
[_ (error 'array->flomap "expected array with 2 or 3 dimensions; given shape ~e" ds)])))
|
|
|
|
(define fm
|
|
(array->flomap
|
|
(time (mandelbrot 0.007 20))))
|
|
|
|
(flomap-size fm)
|
|
|
|
(flomap->bitmap (flomap-normalize fm)))
|