Fix day11pt1 and add correct day11pt2

[adventofcode2019.git] / src / adventofcode2019 / day11.clj

diff --git a/src/adventofcode2019/day11.clj b/src/adventofcode2019/day11.clj
index 12213b904841759918a782210ffcf85493ba2253..026512088204895cf56aae021c55f87037249fa5 100644 (file)
--- a/src/adventofcode2019/day11.clj
+++ b/src/adventofcode2019/day11.clj
@@ -4,9 +4,10 @@
               [clojure.string :as str]
               [clojure.core.match :refer [match]]])
 
-(defn painterbot-9000 [program]
+(defn painterbot-9000 [program input]
   (let [stop-on-output #(= (count (:output %)) 2)
         get-next-ic-output (partial i/intcode-until stop-on-output)
+        start-state (i/build-state program {:input [input]})
         rotate (fn [heading turning]
                  (match [heading turning]
                         [:up    0] :left
@@ -15,28 +16,56 @@
                         [:right 0] :up
                         [:up    1] :right
                         [:down  1] :left
-                        [:left  1] :down
-                        [:right 1] :up))
+                        [:left  1] :up
+                        [:right 1] :down))
         travel (fn [[x y] heading]
                  (case heading
-                   :up    [x (inc y)]
-                   :down  [x (dec y)]
+                   :up    [x (dec y)]
+                   :down  [x (inc y)]
                    :left  [(dec x) y]
                    :right [(inc x) y]))]
-    (loop [program-state (i/build-state program {:input [0]})
+    (loop [program-state (get-next-ic-output start-state)
            tileset {}
            position [0 0]
            heading :up]
       (if (:exit program-state)
         tileset
-        (let [next-state (get-next-ic-output program-state)
-              [color direction] (:output next-state)
-              new-heading (rotate heading direction)]
-          (recur (assoc next-state :output []) (assoc tileset position color)
-                 (travel position new-heading) new-heading))))))
+        (let [[color direction] (:output program-state)
+              new-heading (rotate heading direction)
+              new-position (travel position new-heading)
+              prep-state (assoc program-state 
+                                :output [] 
+                                :input [(get tileset new-position 0)])]
+          (recur (get-next-ic-output prep-state)
+                 (assoc tileset position color)
+                 new-position new-heading))))))
+
+(defn find-bounds [tiles]
+  (let [x-list (map first (keys tiles))
+        y-list (map second (keys tiles))
+        min-x (reduce min x-list)
+        max-x (reduce max x-list)
+        min-y (reduce min y-list)
+        max-y (reduce max y-list)]
+    [[min-x max-x] [min-y max-y]]))
+
+(defn make-field [[dx dy]]
+  (let [line (vec (repeat dx \space))]
+    (mapv (constantly line) (range dy))))
+
+(defn draw-tiles [tiles]
+  (let [[[min-x max-x] [min-y max-y]] (find-bounds tiles)
+        to-text #(match % 0 \space 
+                          1 \u2588) ; full-block
+        field (make-field [(- max-x min-x -1) (- max-y min-y -1)])
+        paint-tile (fn [field [x y] color]
+                       (assoc-in field [(- y min-y) (- x min-x)] (to-text color)))]
+    (mapv str/join (reduce-kv paint-tile field tiles))))
 
 (defn day11 []
   (let [input (mapv parse-int (get-list-from-file (input-file) #","))
-        painted-tiles (painterbot-9000 input)] 
+        painted-tiles (painterbot-9000 input 0)
+        reg-id-tiles (painterbot-9000 input 1)] 
     (part1 (count painted-tiles))
-    #_(part2)))
+    (part2 "see below")
+    (run! println (draw-tiles reg-id-tiles))))