(require '[clojure.string :as str])

(defn pad-input-map [in-map pad-val]
  (let [ly (count in-map) lx (count (first in-map))
        sy (+ 4 ly) sx (+ 4 lx)
        buffer (byte-array (repeat (* sy sx) pad-val))]
    (doseq [y (range 0 ly) x (range 0 lx)]
      (aset-byte buffer (+ (* sx (+ 2 y)) 2 x) (get-in in-map [y x])))
    (mapv vec (partition sx buffer))))

(defn get-number [in-map y x]
  (map (partial get-in in-map)
       [[(dec y) (dec x)] [(dec y) x] [(dec y) (inc x)]
        [y (dec x)]       [y x]       [y (inc x)]
        [(inc y) (dec x)] [(inc y) x] [(inc y) (inc x)]]))

(defn get-new-pixel [enhance-algo in-map y x]
  (->> (get-number in-map y x)
       (reduce #(cond-> (* 2 %1) (= 1 %2) inc) 0)
       (get enhance-algo)))

(defn build-output-number-map [enhance-algo in-map]
  (let [pad-val (if (pos? (first enhance-algo)) (first (first in-map)) 0)
        padded-in-map (pad-input-map in-map pad-val)]
    (->> (for [y (range 1 (dec (count padded-in-map)))]
           (future
             (mapv (partial get-new-pixel enhance-algo padded-in-map y)
                   (range 1 (dec (count (first padded-in-map)))))))
         (mapv deref))))

(defn count-lit-pixels [in-map] ((frequencies (flatten in-map)) 1))

(let [[enhance-algo in-map]
      (->> (slurp "./in")
           str/split-lines
           ((juxt (comp (partial mapv {\. 0 \# 1}) first)
                  (comp (partial mapv #(mapv {\. 0 \# 1} %)) (partial drop 2)))))
      image-output (iterate (partial build-output-number-map enhance-algo) in-map)]
  (println "Part 1:" (count-lit-pixels (nth image-output 2)))
  (println "Part 2:" (count-lit-pixels (nth image-output 50))))

(shutdown-agents)