Повышение эффективности в генераторе Clojure Mandelbrot

Question

Я хотел изучить Clojure и начал с генератора Мандельброта, используя quil, я заставил его работать - однако для генерации изображений требуется некоторое время, и это огромный ресурс. Любой совет, как сделать это быстрее или если вы обнаружите какие-либо незакрытые части моего кода.

Core.clj

(ns frac.core
  (:require [quil.core :as q])
  (:require [frac.complex :refer :all]))

(def scale (atom 0.01))
(def xoff (atom -200))
(def yoff (atom -200))
(def its 50)

(defn mandelbrot [r i]
 (count (take its (take-while #(<= (modu %) 2) (iterate #( add (mul % %) [r i]) [r i])))))

(defn gen []
  (let [p (q/pixels)
        w (q/width)
        h (q/height)]
    (doseq [x (range w) y (range h)]
      (let [m (mandelbrot (* @scale (+ x @xoff)) (* @scale (+ y @yoff)))
            c (if (= m its) 0 m)]
        (aset p (+ x (* y w)) (q/color (* 1.5 c) (* 4 c) (* 5.2 c))))))
  (q/update-pixels))

(defn setup []
  (gen))

(defn draw [])

(defn click []
  (swap! xoff #(+ (q/mouse-x) (- (/ (q/width) 2)) %))
  (swap! yoff #(+ (q/mouse-y) (- (/ (q/height) 2)) %))
  (gen))

(defn wheel [z] 
  (swap! scale #(if (pos? z) (* 1.1 %) (* 0.9 %)))
  (prn @scale)
  (gen))

(q/defsketch example
  :title "Mandel"
  :setup setup
  :draw draw
  :size [400 400]
  :mouse-clicked click
  :mouse-wheel wheel)

(defn -main [& args])

Complex.clj

(ns frac.complex)

(defn mul [z1 z2]
  (let [r1 (first z1)
        i1 (second z1)
        r2 (first z2)
        i2 (second z2)]
    [(- (* r1 r2) (* i1 i2)) (+ (* r1 i2) (* r2 i1))]))

(defn add [z1 z2]
  (let [r1 (first z1)
        i1 (second z1)
        r2 (first z2)
        i2 (second z2)]
    [(+ r1 r2) (+ i1 i2)]))

(defn modu [z]
  (let [r (first z)
        i (second z)]
    (Math/sqrt (+ (* r r) (* i i)))))

Answer 1

Попробуйте установить это:

(set! *unchecked-math* :warn-on-boxed)

и удалите все предупреждения. Используйте подсказки типа по мере необходимости.