rlm@0: (ns laser.rasterize) rlm@0: rlm@0: (import '(java.io File)) rlm@0: (import '(org.apache.commons.io FileUtils)) rlm@0: (import '(javax.imageio ImageIO) ) rlm@0: (import '(javax.swing JFrame)) rlm@0: (import '(java.awt Color BorderLayout)) rlm@0: (import '(ij ImagePlus IJ)) rlm@0: (import '(java.lang Math)) rlm@2: (import '(java.awt Graphics2D Panel)) rlm@0: (import '(ij Macro)) rlm@0: rlm@0: (import '(java.io BufferedReader InputStreamReader)) rlm@0: (import '(java.awt.image BufferedImage)) rlm@0: rlm@0: (use 'clojure.contrib.str-utils) rlm@0: rlm@0: (use 'clojure.contrib.combinatorics) rlm@0: rlm@0: rlm@0: (use 'clojure.contrib.repl-utils) rlm@0: rlm@0: (set! *print-length* 20) rlm@0: rlm@0: rlm@0: (def img "/home/r/graster/test.png") rlm@0: rlm@1: rlm@1: (def feed 120) rlm@1: (def dpi [500, 500]) rlm@1: (def on_range [0.0, 0.5]) rlm@1: (def overshoot 0.5) rlm@1: (def offset [1.0, 1.0]) rlm@1: (def tiles [1, 1]) rlm@1: (def tile_size [false, false]) rlm@1: (def tile_spacing [0.125, 0.125]) rlm@1: (def feed 120) rlm@1: (def cut_feed 20) rlm@1: (def corner_radius 0) rlm@1: rlm@2: (defmulti display "Creates a JFrame and displays a buffered image" class) rlm@1: rlm@2: (defn- makePanel [image] (proxy [Panel] [] (paint [g] (.drawImage g image 0 0 nil)))) rlm@2: rlm@2: (defmethod display rlm@2: BufferedImage [image] rlm@2: (let [panel (makePanel image) rlm@2: frame (JFrame. "Oh Yeah!")] rlm@2: (.add frame panel) rlm@2: (.pack frame) rlm@2: (.setVisible frame true ) rlm@2: (.setSize frame(.getWidth image) (.getHeight image)))) rlm@2: rlm@2: (defmethod display rlm@2: ImagePlus [image] rlm@2: (display (.getBufferedImage image))) rlm@2: rlm@2: (defmethod display rlm@2: clojure.lang.PersistentHashMap [frame-hash] rlm@2: (display (frame-hash->bufferedImage frame-hash))) rlm@2: rlm@2: (defmethod display rlm@2: clojure.lang.PersistentArrayMap [frame-hash] rlm@2: (display (frame-hash->bufferedImage frame-hash))) rlm@1: rlm@1: rlm@0: (defn frame-hash rlm@0: "yields a convienent representation for the pixles in an image. rlm@0: Because of the size of the structvre generated, this must only be used rlm@0: in a transient way so that java can do it's garbage collection." rlm@0: [#^java.lang.String image-name] rlm@0: (let [image+ (ImagePlus. image-name)] rlm@0: (with-meta rlm@0: (let [buf (.. image+ getBufferedImage) rlm@0: color (.getColorModel buf)] rlm@0: (apply hash-map rlm@0: (interleave rlm@0: (doall (for [x (range (.getWidth image+)) y (range (.getHeight image+))] rlm@0: (vector x y))) rlm@0: (doall (for [x (range (.getWidth image+)) y (range (.getHeight image+))] rlm@0: (let [data (.getRGB buf x y)] rlm@0: (hash-map :r (bit-shift-right (bit-and 0xff0000 data) 16) rlm@0: :g (bit-shift-right (bit-and 0x00ff00 data) 8) rlm@0: :b (bit-and 0x0000ff data)))))))) rlm@0: {:width (.getWidth image+) :height (.getHeight image+)}))) rlm@0: rlm@0: rlm@1: (def white {:r 255, :g 255, :b 255}) rlm@1: (def black {:r 0, :g 0, :b 0}) rlm@1: rlm@1: (def expt #(Math/pow %1 %2)) rlm@1: rlm@1: (defn rgb-euclidian rlm@1: [{r1 :r g1 :g b1 :b} {r2 :r g2 :g b2 :b} ] rlm@1: (expt (+ (expt (- r1 r2) 2) rlm@1: (expt (- g1 g2) 2) rlm@1: (expt (- b1 b2) 2)) 0.5)) rlm@1: rlm@1: (defn b&w rlm@1: "turn everything strictly black or white" rlm@1: [window] rlm@1: (with-meta rlm@1: (zipmap rlm@1: (keys window) rlm@1: (map (fn [rgb] rlm@1: (if (> (rgb-euclidian rgb white) (rgb-euclidian rgb black)) rlm@1: black white)) rlm@1: (vals window))) (meta window))) rlm@1: rlm@1: rlm@1: rlm@3: (defn raster-preamble [] rlm@3: (str-join \newline rlm@3: ["M63 P0\nG61" rlm@3: (str \F feed) rlm@3: "M101" rlm@3: "M3 S1"])) rlm@3: rlm@3: rlm@1: rlm@1: rlm@2: rlm@2: rlm@2: rlm@2: rlm@2: rlm@2: rlm@2: rlm@2: rlm@1: rlm@1: rlm@1: rlm@1: rlm@1: rlm@1: rlm@1: rlm@1: rlm@1: rlm@1: rlm@0: (defn frame-hash->bufferedImage rlm@0: [frame-hash] rlm@0: (let [data (meta frame-hash) rlm@0: image (BufferedImage. (:width data) (:height data) BufferedImage/TYPE_INT_BGR)] rlm@0: rlm@0: (doall (for [element frame-hash] rlm@0: (let [coord (key element) rlm@0: rgb (val element) rlm@0: packed-RGB rlm@0: (+ (bit-shift-left (:r rgb) 16) rlm@0: (bit-shift-left (:g rgb) 8) rlm@0: (:b rgb))] rlm@0: (.setRGB image (first coord) (last coord) packed-RGB)))) rlm@0: image)) rlm@0: rlm@0: rlm@0: