(ns advent.pathfind
  (:require [play-clj.core :refer :all]
            [play-clj.ui :refer :all]
            [play-clj.utils :refer :all]
            [play-clj.g2d :refer :all]
            [clojure.pprint]
            [clojure.data.priority-map :refer [priority-map]])
  (:import [com.badlogic.gdx.files FileHandle]
           [com.badlogic.gdx Files]
           [java.lang Math]
           [com.badlogic.gdx.graphics Camera Color GL20 OrthographicCamera
            PerspectiveCamera Pixmap Pixmap$Format PixmapIO Texture
            VertexAttributes$Usage]))

(defn printmap [my-map & [skip]]
  (let [skip (or skip 1)]
    (doseq [row (take-nth skip my-map)]
      (println (take-nth skip (map {1 \space 0 "W" "X" "X" "." "."} row))))))

(defn random-map [] (-> (vec (take 640 (repeatedly (fn [] (vec (take 480 (repeatedly  (fn [] (rand-nth [1  1  1  5 5  0])))))))))
                        (update-in [1 1] (constantly 1))
                        (update-in [639 479] (constantly 1))))


(defn neighbors [[^long x ^long y] my-map]
  (let [candidates [[^long (dec x) ^long (dec y)] [x ^long (dec y)] [^long (inc x) ^long (dec y)]
                    [^long (dec x) y]                   [^long (inc x) y]
                    [^long (dec x) ^long (inc y)] [x ^long (inc y)] [^long (inc x) ^long (inc y)]]
        height (count (first my-map))
        width (count my-map)]
    (remove #(= 0 (get-in my-map %))
            (filter (fn [[x y]] (and (< -1 x width)
                                    (< -1 y height))) candidates))))


(defn resolve-path [came-from play-loc target-loc]
  (let [came-from (into {} came-from)]


    (if (nil? (came-from target-loc))
      nil
      (loop [path []
             current-node (vec target-loc)]
        (if (or (= current-node play-loc)
                (nil? current-node))
          (reverse (conj path current-node))
          (recur
           (conj path current-node)
           (came-from (vec current-node))))))))

(def d2 ^double (- (Math/sqrt 2) 1))
(defn heuristic [^long goal-x ^long goal-y ^long current-x ^long current-y]
  (let [dist-x ^long (if (< goal-x current-x)
                       ^long (- current-x goal-x)
                       ^long (- goal-x current-x ))
        dist-y ^long (if (< goal-y current-y)
                       ^long (- current-y goal-y)
                       ^long (- goal-y current-y))]
    ^long (+ dist-x dist-y (* ^double d2
                        ^long (min dist-x dist-y)))))

(def my-count (atom 0))

(def cost-comparator (comparator (fn [{^long a :cost} {^long b :cost}] (do (swap! my-count inc) (< a b)))))

(defn visit-all [my-map play-loc target-loc]
  (reset! my-count 0)
  (let [play-loc (vec play-loc)
        target-log (vec target-loc)]
    (if (= 0 (get-in my-map target-loc))
      nil
      (let [cost-so-far ^java.util.HashMap (java.util.HashMap. {play-loc 0}) 
            came-from ^java.util.HashMap (java.util.HashMap.)
            fronteir ^java.util.PriorityQueue (java.util.PriorityQueue. 100 cost-comparator)]
        (.offer fronteir {:cost 0 :loc play-loc})
        (loop [current-loc (.poll fronteir)]
          (if (or (nil? current-loc)
                  (= (:loc current-loc) target-loc))
            
            (do (println @my-count)
                (resolve-path came-from play-loc target-loc))
            (do (doseq [neighbor (neighbors (:loc current-loc) my-map)]
                  (let [cost-for-neighbor (.get cost-so-far neighbor)
                        new-cost (+ (.get cost-so-far (:loc current-loc)) (get-in my-map neighbor))]
                    (when (or (nil? cost-for-neighbor)
                              (< new-cost cost-for-neighbor))
                      (.put came-from (vec neighbor) (vec (:loc current-loc)))
                      (.put cost-so-far (vec neighbor) new-cost)
                      (.offer fronteir {:cost (+ new-cost (heuristic (first target-loc) (second target-loc) (first neighbor) (second neighbor)))
                                        :loc neighbor}))))
                (recur (.poll fronteir)))))))))


(defn print-resolved [path my-map]
  (doseq [row (reduce (fn [acc path]
                        (if path
                          (update-in acc (reverse path) (constantly "X"))
                          acc))
                my-map
                path)]
    
    (println (map {1 \space 5 "." 0 "W" "X" "X"} row)))
  nil)

(defn test-pathfind []
  (let [my-map (random-map)
        path (visit-all my-map [1 1] [639 479 ])]
    (println "Test")
    #_(print-resolved path my-map)
    (println path)))

(defn map-from-resource [filename]
  (let [pm (pixmap filename)
        black (color 0 0 0 255)
        painful (color 255 0 0 255)]
    (vec (for [x (range (pixmap! pm :get-width))]
           (vec (for [y (reverse (range (pixmap! pm :get-height)))
                      :let [current-color (color (pixmap! pm :get-pixel x y))]]
                  (cond
                   (color! current-color :equals black) 0
                   (color! current-color :equals painful) 1 
                   :else 1)))))))