annotate src/laser/rasterize.clj @ 13:397ab24b4952

saving, to update with correct fix later
author Robert McIntyre <rlm@mit.edu>
date Sun, 29 Aug 2010 00:03:09 -0400
parents f952052e37b7
children db745c95aabd
rev   line source
rlm@0 1 (ns laser.rasterize)
rlm@0 2
rlm@0 3 (import '(java.io File))
rlm@0 4 (import '(org.apache.commons.io FileUtils))
rlm@0 5 (import '(javax.imageio ImageIO) )
rlm@0 6 (import '(javax.swing JFrame))
rlm@0 7 (import '(java.awt Color BorderLayout))
rlm@0 8 (import '(ij ImagePlus IJ))
rlm@0 9 (import '(java.lang Math))
rlm@2 10 (import '(java.awt Graphics2D Panel))
rlm@0 11 (import '(ij Macro))
rlm@0 12
rlm@0 13 (import '(java.io BufferedReader InputStreamReader))
rlm@0 14 (import '(java.awt.image BufferedImage))
rlm@0 15
rlm@11 16 ;(use 'clojure.contrib.str-utils)
rlm@11 17 ;(use 'clojure.contrib.seq-utils)
rlm@11 18 ;(use 'clojure.contrib.combinatorics)
rlm@11 19 ;(use 'clojure.contrib.duck-streams)
rlm@0 20
rlm@11 21 ;(use 'clojure.contrib.repl-utils)
rlm@0 22
rlm@11 23 ;(set! *print-length* 20)
rlm@0 24
rlm@0 25
rlm@0 26
rlm@1 27
rlm@1 28 (def feed 120)
rlm@1 29 (def dpi [500, 500])
rlm@8 30
rlm@8 31
rlm@1 32
rlm@1 33
rlm@5 34 (defn preserve-meta [f]
rlm@5 35 (fn [& x] (with-meta
rlm@5 36 (apply f x)
rlm@5 37 (meta (last x)))))
rlm@2 38
rlm@8 39 (defmulti frame-hash-multi class)
rlm@1 40
rlm@1 41
rlm@8 42 (defmethod frame-hash-multi ImagePlus
rlm@8 43 [image+]
rlm@0 44 (with-meta
rlm@0 45 (let [buf (.. image+ getBufferedImage)
rlm@0 46 color (.getColorModel buf)]
rlm@0 47 (apply hash-map
rlm@0 48 (interleave
rlm@0 49 (doall (for [x (range (.getWidth image+)) y (range (.getHeight image+))]
rlm@0 50 (vector x y)))
rlm@0 51 (doall (for [x (range (.getWidth image+)) y (range (.getHeight image+))]
rlm@0 52 (let [data (.getRGB buf x y)]
rlm@0 53 (hash-map :r (bit-shift-right (bit-and 0xff0000 data) 16)
rlm@0 54 :g (bit-shift-right (bit-and 0x00ff00 data) 8)
rlm@0 55 :b (bit-and 0x0000ff data))))))))
rlm@8 56 {:width (.getWidth image+) :height (.getHeight image+)}))
rlm@0 57
rlm@0 58
rlm@8 59 (defmethod frame-hash-multi String
rlm@8 60 [image-name]
rlm@8 61 (let [image+ (ImagePlus. image-name)]
rlm@8 62 (frame-hash-multi image+)))
rlm@5 63
rlm@5 64
rlm@8 65 (defn frame-hash
rlm@8 66 "yields a convienent representation for the pixles in an image.
rlm@8 67 Because of the size of the structvre generated, this must only be used
rlm@8 68 in a transient way so that java can do it's garbage collection."
rlm@8 69 [something]
rlm@8 70 (frame-hash-multi something))
rlm@8 71
rlm@8 72 ;(def frame-hash (preserve-meta frame-hash))
rlm@8 73
rlm@5 74
rlm@5 75
rlm@5 76
rlm@1 77 (def white {:r 255, :g 255, :b 255})
rlm@1 78 (def black {:r 0, :g 0, :b 0})
rlm@1 79
rlm@11 80
rlm@1 81
rlm@1 82 (defn rgb-euclidian
rlm@1 83 [{r1 :r g1 :g b1 :b} {r2 :r g2 :g b2 :b} ]
rlm@1 84 (expt (+ (expt (- r1 r2) 2)
rlm@1 85 (expt (- g1 g2) 2)
rlm@1 86 (expt (- b1 b2) 2)) 0.5))
rlm@1 87
rlm@1 88 (defn b&w
rlm@1 89 "turn everything strictly black or white"
rlm@1 90 [window]
rlm@1 91 (with-meta
rlm@1 92 (zipmap
rlm@1 93 (keys window)
rlm@1 94 (map (fn [rgb]
rlm@1 95 (if (> (rgb-euclidian rgb white) (rgb-euclidian rgb black))
rlm@1 96 black white))
rlm@1 97 (vals window))) (meta window)))
rlm@1 98
rlm@1 99
rlm@1 100
rlm@3 101 (defn raster-preamble []
rlm@3 102 (str-join \newline
rlm@3 103 ["M63 P0\nG61"
rlm@3 104 (str \F feed)
rlm@3 105 "M101"
rlm@8 106 "M3 S1\n"]))
rlm@3 107
rlm@4 108 (defn raster-epilogue []
rlm@4 109 (str-join \newline
rlm@8 110 ["M63 P0"
rlm@4 111 "M5"
rlm@8 112 "M2\n"]))
rlm@3 113
rlm@1 114
rlm@4 115 (defn raster-comment [string]
rlm@4 116 (str "(" (re-gsub #"[()]" "" string) ")"))
rlm@1 117
rlm@4 118 (defn filter-keys [fun m]
rlm@4 119 (select-keys m (filter fun (keys m))))
rlm@6 120
rlm@5 121 (def filter-keys (preserve-meta filter-keys))
rlm@2 122
rlm@4 123 (defn filter-vals [fun m]
rlm@5 124 (into {} (filter (comp fun val) m)))
rlm@6 125
rlm@5 126 (def filter-vals (preserve-meta filter-vals))
rlm@1 127
rlm@0 128 (defn frame-hash->bufferedImage
rlm@0 129 [frame-hash]
rlm@0 130 (let [data (meta frame-hash)
rlm@0 131 image (BufferedImage. (:width data) (:height data) BufferedImage/TYPE_INT_BGR)]
rlm@0 132
rlm@0 133 (doall (for [element frame-hash]
rlm@0 134 (let [coord (key element)
rlm@0 135 rgb (val element)
rlm@0 136 packed-RGB
rlm@0 137 (+ (bit-shift-left (:r rgb) 16)
rlm@0 138 (bit-shift-left (:g rgb) 8)
rlm@0 139 (:b rgb))]
rlm@0 140 (.setRGB image (first coord) (last coord) packed-RGB))))
rlm@0 141 image))
rlm@5 142
rlm@5 143 (defmulti display "Creates a JFrame and displays a buffered image" class)
rlm@0 144
rlm@5 145 (defn- makePanel [image] (proxy [Panel] [] (paint [g] (.drawImage g image 0 0 nil))))
rlm@0 146
rlm@6 147 (defn select-row [x window]
rlm@6 148 (filter-keys (comp (partial = x) first) window))
rlm@6 149
rlm@6 150
rlm@0 151
rlm@5 152 (defmethod display
rlm@5 153 BufferedImage [image]
rlm@5 154 (let [panel (makePanel image)
rlm@5 155 frame (JFrame. "Oh Yeah!")]
rlm@5 156 (.add frame panel)
rlm@5 157 (.pack frame)
rlm@5 158 (.setVisible frame true )
rlm@5 159 (.setSize frame(.getWidth image) (.getHeight image))))
rlm@5 160
rlm@5 161 (defmethod display
rlm@5 162 ImagePlus [image]
rlm@5 163 (display (.getBufferedImage image)))
rlm@5 164
rlm@5 165 (defmethod display
rlm@5 166 clojure.lang.PersistentHashMap [frame-hash]
rlm@5 167 (display (frame-hash->bufferedImage frame-hash)))
rlm@5 168
rlm@5 169 (defmethod display
rlm@5 170 clojure.lang.PersistentArrayMap [frame-hash]
rlm@5 171 (display (frame-hash->bufferedImage frame-hash)))
rlm@5 172
rlm@5 173
rlm@5 174
rlm@5 175
rlm@5 176
rlm@7 177 ;this is a sequence of rows
rlm@7 178
rlm@7 179 ;(defn span [row]
rlm@7 180 ; (let [sorted-row (sort #(< (first %1) (first %2)) row)]
rlm@7 181 ; (vector (first sorted-row) (last sorted-row))))
rlm@7 182
rlm@7 183
rlm@7 184 (defn row->gcode [[x-dpi y-dpi] row]
rlm@7 185 (let [[x1 y1] (first row)
rlm@13 186 [x2 y2] (last row)
rlm@13 187 x2 (+ x2 (* x-dpi 0.318))]
rlm@13 188
rlm@13 189
rlm@13 190 ; (println x2)
rlm@8 191 (str (format "G0 X%.3f Y%.3f\n"
rlm@7 192 (float (* x1 (/ x-dpi)))
rlm@7 193 (float (* y1 (/ y-dpi))))
rlm@7 194
rlm@7 195 (format "G1 X%.3f Y%.3f\n"
rlm@7 196 (float (* x2 (/ x-dpi)))
rlm@7 197 (float (* y2 (/ y-dpi)))))))
rlm@7 198
rlm@7 199 (defn gather-row [row]
rlm@7 200 (let [base [[(first (first row)) (first (first row))]]]
rlm@7 201 ; (println base)
rlm@7 202 (reduce
rlm@7 203 (fn colapse [collection new-n]
rlm@7 204
rlm@7 205 (let [collection (apply vector collection)
rlm@7 206 prevoius (last (last collection))
rlm@7 207 range-start (first (last collection))]
rlm@7 208 ; (println new-n)
rlm@7 209 ; (println prevoius)
rlm@7 210 ; (println range-start)
rlm@7 211 (if (<= new-n (+ prevoius 1))
rlm@11 212 (do ;(println "join")
rlm@7 213 ;(println (butlast collection))
rlm@11 214 (conj (apply vector (butlast collection))
rlm@11 215 (vector range-start new-n)))
rlm@11 216 (conj collection (vector new-n new-n)))))
rlm@7 217
rlm@7 218 base
rlm@11 219 (map first row))))
rlm@11 220
rlm@7 221
rlm@7 222
rlm@7 223
rlm@7 224 (defn row->gmask [[x-dpi y-dpi] forward? row]
rlm@11 225 ; (println forward?)
rlm@11 226 (let [start (float (* (/ x-dpi) (first (first
rlm@11 227 (if forward?
rlm@11 228 (reverse row) row)))))]
rlm@7 229
rlm@11 230 (let [preamble (if-not forward?
rlm@7 231 (format "0 0 0 %.3f\n" start)
rlm@11 232 (format "0 0 1 %.3f\n" start))
rlm@7 233 body
rlm@11 234 (for [[x y]
rlm@11 235 (if forward?
rlm@11 236 (reverse (gather-row row))
rlm@11 237 (gather-row row))]
rlm@7 238 (let [x (float (* x (/ x-dpi)))
rlm@13 239 y (float (* y (/ x-dpi)))
rlm@13 240 x (+ x 0.159)];; shift by a small margin.
rlm@11 241 (if-not forward?
rlm@7 242 (str (format "0 0 1 %.3f\n" x)
rlm@7 243 (format "0 1 1 %.3f\n" y))
rlm@7 244
rlm@11 245 (str (format "0 0 0 %.3f\n" y)
rlm@11 246 (format "0 1 0 %.3f\n" x)))))]
rlm@7 247
rlm@7 248 (str preamble (str-join "" body)))))
rlm@7 249
rlm@7 250
rlm@7 251
rlm@7 252 (defn make-rows [pic]
rlm@7 253 (map (partial sort #(< (first %1) (first %2)))
rlm@7 254 (partition-by last
rlm@7 255 (sort (fn [[x1 y1][x2 y2]] (> y2 y1))
rlm@7 256 (map first (filter-vals (partial = black) pic))))))
rlm@7 257
rlm@7 258
rlm@7 259
rlm@7 260 (defn generate-gmask [pic]
rlm@7 261
rlm@7 262 (str "1 0 0 0\n"
rlm@8 263 (str-join "" (map (fn [[index row]]
rlm@11 264 (row->gmask dpi (even? index) row))
rlm@8 265 (indexed (make-rows pic))))))
rlm@8 266
rlm@11 267
rlm@8 268 ;; 1 0 0 0
rlm@8 269 ;; 0 0 1 2.881
rlm@8 270 ;; 0 0 0 2.881
rlm@8 271 ;; 0 1 0 2.863
rlm@8 272 ;; 0 0 0 2.769
rlm@8 273 ;; 0 1 0 2.751
rlm@8 274 ;; 0 0 0 2.729
rlm@8 275 ;; 0 1 0 2.617
rlm@8 276 ;; 0 0 0 2.593
rlm@8 277 ;; 0 1 0 2.561
rlm@8 278 ;; 0 0 0 2.463
rlm@8 279 ;; 0 1 0 2.445
rlm@8 280 ;; 0 0 0 2.385
rlm@8 281 ;; 0 1 0 2.317
rlm@8 282 ;; 0 0 0 2.253
rlm@8 283 ;; 0 1 0 2.233
rlm@8 284 ;; 0 0 0 2.177
rlm@8 285
rlm@7 286
rlm@7 287
rlm@7 288 (defn generate-gcode [pic]
rlm@8 289 (str (raster-preamble)
rlm@8 290
rlm@8 291
rlm@8 292 (str-join "" (map (partial row->gcode dpi) (make-rows pic)))
rlm@8 293 (raster-epilogue)))
rlm@8 294
rlm@7 295
rlm@7 296
rlm@8 297 (defn rotate [degrees #^ImagePlus image]
rlm@8 298 (.rotate (.getChannelProcessor image) degrees)
rlm@8 299 image)
rlm@7 300
rlm@8 301 (defn map-keys [f m]
rlm@8 302 (into {} (map (fn [[key val]] [(f key) val]) m)))
rlm@8 303
rlm@8 304
rlm@8 305
rlm@8 306 (defn invert-frame-hash [pic]
rlm@8 307 (map-keys (fn [[x y]] [x (- (:height (meta pic)) y 1)]) pic ))
rlm@8 308
rlm@8 309
rlm@8 310 (defn generate-files [pic]
rlm@8 311 (let [image (invert-frame-hash (b&w (frame-hash (rotate 180 (ImagePlus. pic)))))]
rlm@8 312 (spit "/home/r/kevin/out.ngc" (generate-gcode image))
rlm@8 313 (spit "/home/r/kevin/out.gmask" (generate-gmask image))
rlm@8 314 image))
rlm@8 315
rlm@11 316
rlm@11 317
rlm@11 318 (defn update-state []
rlm@11 319 (def sing "/home/r/lasercutter/graster/signer4laser2x1.png")
rlm@11 320
rlm@11 321 (def pic (frame-hash (let [image (ImagePlus. sing)]
rlm@11 322 (.rotate (.getChannelProcessor image) 180)
rlm@11 323 image)))
rlm@11 324
rlm@11 325 (def pic (b&w pic)))
rlm@11 326
rlm@11 327
rlm@11 328
rlm@11 329
rlm@11 330
rlm@11 331
rlm@11 332
rlm@11 333
rlm@11 334
rlm@11 335
rlm@8 336
rlm@8 337 ;;;; testing
rlm@8 338
rlm@8 339 (defn init []
rlm@8 340 (let [stuff
rlm@8 341
rlm@8 342 (bound-fn []
rlm@8 343
rlm@8 344 (do
rlm@8 345 (println "hi everyone")
rlm@8 346 (def img "/home/r/kevin/sing.png")
rlm@8 347 (def pic (frame-hash (let [image (ImagePlus. img)]
rlm@8 348 (.rotate (.getChannelProcessor image) 180)
rlm@8 349 image)))
rlm@8 350
rlm@8 351
rlm@8 352 (def test-image
rlm@8 353 (invert-frame-hash (b&w (frame-hash (rotate 180 (ImagePlus. img))))))
rlm@8 354
rlm@8 355 (defn test-gmask []
rlm@8 356 (println (str-join "" (take 170 (generate-gmask test-image)))))
rlm@8 357
rlm@8 358 (println "ALL variables initialized!")
rlm@8 359
rlm@8 360 ))]
rlm@8 361 (.start
rlm@8 362 (Thread. stuff))))
rlm@8 363
rlm@8 364
rlm@8 365
rlm@8 366 (defn thread-test []
rlm@8 367
rlm@8 368 (let [temp *out*]
rlm@8 369 (.start
rlm@8 370 (Thread.
rlm@8 371 (fn []
rlm@11 372 (with-bindings {#'*out* temp}
rlm@8 373 (Thread/sleep 5000)
rlm@8 374 (println "hi")))))))
rlm@13 375
rlm@13 376
rlm@13 377 (comment
rlm@13 378 (do
rlm@13 379 (require 'rlm.quick)
rlm@13 380 (ns laser.rasterize)
rlm@13 381 (rlm.quick/dirty)
rlm@13 382 (use :reload-all 'laser.rasterize)
rlm@13 383 (undef map-keys)
rlm@13 384 (use :reload-all 'laser.rasterize)))