Adding eyes to snake.

nbeloglazov · Aug 11, 2010 · 5709847 · 5709847
1 parent 5b53b8a
commit 5709847
Show file tree

Hide file tree

Showing 2 changed files with 87 additions and 61 deletions.
diff --git a/src/snakejure/core.clj b/src/snakejure/core.clj
@@ -60,7 +60,8 @@
   {:body body
    :type :snake
    :dir :right
-   :color (Color/CYAN)})
+   :color (Color/CYAN)
+   :eyes-color (Color/BLACK)})
 
 (defn create-apple
   "Creates new apple in given point."

diff --git a/src/snakejure/game_gui.clj b/src/snakejure/game_gui.clj
@@ -1,49 +1,73 @@
 (ns snakejure.game-gui
-      (:use (clojure.contrib import-static
-			     logging)
-	    (snakejure core))
-      (:require (snakejure.levels basic))
-      (:import (java.awt Dimension)
-	       (java.awt.event ActionListener KeyAdapter)
-	       (javax.swing JPanel Timer)))
+  (:use (clojure.contrib import-static
+			 logging)
+	(snakejure core))
+  (:require (snakejure.levels basic))
+  (:import (java.awt Dimension)
+	   (java.awt.event ActionListener KeyAdapter)
+	   (javax.swing JPanel Timer)))
 
 (import-static java.awt.event.KeyEvent VK_SPACE VK_LEFT VK_RIGHT VK_UP VK_DOWN VK_ESCAPE)
 
-(def key-to-dir {VK_RIGHT :right
-                 VK_LEFT :left
-		 VK_UP :up
-		 VK_DOWN :down})
-(def size 20)
-(def speed 100)
-(def dir-keys (set (keys key-to-dir)))
+(def #^{:private true} key-to-dir {VK_RIGHT :right
+				   VK_LEFT :left
+				   VK_UP :up
+				   VK_DOWN :down})
+(def #^{:private true} size 20)
+(def #^{:private true} speed 100)
+(def #^{:private true} dir-keys (set (keys key-to-dir)))
 
 (defn- draw-point 
   "Draws point on given graphics g."
   [g [y x] [dy dx]]
   (.fillRect g (* size (+ x dx)) 
-	       (* size (+ y dy)) 
-	       size 
-	       size))
+	     (* size (+ y dy)) 
+	     size 
+	     size))
+
+(defn- draw-head
+  "Draws snake's head"
+  [g {[[y x :as head] & _] :body :keys [eyes-color color dir]} [dy dx :as d]]
+  (let [real-x (* size (+ x dx))
+	real-y (* size (+ y dy))
+	eye-size (/ size 5.0)
+	draw-eye (fn [y-in x-in]
+		   (.fillRect g (+ real-x (* x-in eye-size))
+			      (+ real-y (* y-in eye-size))
+			      eye-size
+			      eye-size))]
+    (.setColor g color)
+    (draw-point g head d)
+    (.setColor g eyes-color)
+    (when (#{:up :right} dir)
+      (draw-eye 1 3))
+    (when (#{:up :left} dir)
+      (draw-eye 1 1))
+    (when (#{:down :right} dir)
+      (draw-eye 3 3))
+    (when (#{:down :left} dir)
+      (draw-eye 3 1))))
 
 (defmulti paint (fn [g object & _] (:type object)))
 
-(defmethod paint :snake [g {:keys [body color]} d]
-  (.setColor g color)
-  (doseq [part body] (draw-point g part d)))
+(defmethod paint :snake [g {[_ & rst] :body color :color :as snake} d]
+	   (draw-head g snake d)
+	   (.setColor g color)
+	   (doseq [part rst] (draw-point g part d)))
 
 (defmethod paint :apple [g {:keys [location color]} d]
-  (.setColor g color)
-  (draw-point g location d))
+	   (.setColor g color)
+	   (draw-point g location d))
 
 (defmethod paint :wall [g {:keys [location color]} d]
-  (.setColor g color)
-  (draw-point g location d))
+	   (.setColor g color)
+	   (draw-point g location d))
 
 (defmethod paint :noiser [g {:keys [location color-noise color-silence]} snake d]
-  (if (overlaps-body? snake location)
-    (.setColor g color-noise)
-    (.setColor g color-silence))
-  (draw-point g location d))
+	   (if (overlaps-body? snake location)
+	     (.setColor g color-noise)
+	     (.setColor g color-silence))
+	   (draw-point g location d))
 
 (defn- switch-timer
   "Stop timer, if it's running.
@@ -58,43 +82,44 @@
   See source :)"
   [level]
   (proxy [JPanel ActionListener] []
-	 (paintComponent [g]
-			 (let-map [@level snake noisers walls apple d]
-				  (proxy-super paintComponent g)
-				  (paint g apple d)
-				  (paint g snake d)
-				  (doseq [noiser noisers] (paint g noiser snake d))
-				  (doseq [wall walls] (paint g wall d))))
-	 (getPreferredSize []
-			   (Dimension. (* (inc (dec width)) size) 
-				       (* (inc (dec height)) size)))))
+    (paintComponent [g]
+		    (let-map [@level snake noisers walls apple d]
+			     (proxy-super paintComponent g)
+			     (paint g apple d)
+			     (paint g snake d)
+			     (doseq [noiser noisers] (paint g noiser snake d))
+			     (doseq [wall walls] (paint g wall d))))
+    (getPreferredSize []
+		      (Dimension. (* (inc (dec width)) size) 
+				  (* (inc (dec height)) size)))))
 
 (defn- create-action-listener [level panel end-fn]
   (proxy [ActionListener] []
-	 (actionPerformed [e]
-			  (let-map [@level snake noisers walls] 
-			    (update-snake-position level)
-			    (trace (level-to-str @level :walls))
-			    (when (lose? snake walls) 
-			      (end-fn :lose)
-			      (switch-timer (.getSource e)))
-			    (when (win? snake noisers)
-			      (end-fn :win)
-			      (switch-timer (.getSource e)))
-			    (.repaint panel)))))
+    (actionPerformed [e]
+		     (let-map [@level snake noisers walls] 
+			      (update-snake-position level)
+			      (trace (level-to-str @level :walls))
+			      (when (lose? snake walls) 
+				(end-fn :lose)
+				(.. e getSource stop))
+			      (when (win? snake noisers)
+				(end-fn :win)
+				(.. e getSource stop))
+			      (.repaint panel)))))
 
 
-(defn- create-key-listener [level timer end-fn]
+(defn- create-key-listener [level timer end-fn panel]
   (proxy [KeyAdapter] []
-	 (keyPressed [e]
-		     (let [code (.getKeyCode e)]
-		       (when (= code VK_SPACE)
-			 (switch-timer timer))
-		       (when (= code VK_ESCAPE)
-			 (end-fn :exit)
-			 (switch-timer timer))
-		       (when (dir-keys code)
-			 (update-snake-direction level (key-to-dir (.getKeyCode e))))))))
+    (keyPressed [e]
+		(let [code (.getKeyCode e)]
+		  (when (= code VK_SPACE)
+		    (switch-timer timer))
+		  (when (= code VK_ESCAPE)
+		    (end-fn :exit)
+		    (.stop timer))
+		  (when (dir-keys code)
+		    (update-snake-direction level (key-to-dir (.getKeyCode e)))
+		    (.repaint panel))))))
 
 (defn create-game-panel [lvl end-fn]
   "Creates panel, which displays game level.
@@ -105,7 +130,7 @@
 	action-listener (create-action-listener level panel end-fn)
 	real-speed (if (:speed lvl) (:speed lvl) speed)
 	timer (Timer. real-speed action-listener)
-	key-listener (create-key-listener level timer end-fn)]
+	key-listener (create-key-listener level timer end-fn panel)]
     (doto panel
       (.setFocusable true)
       (.addKeyListener key-listener))))