div., game speed bar, current points

Author: bison--

joystick.py

@@ -22,7 +22,8 @@ def joystickAvailable(self):
 		else:
 			return False
 
-	def haveAction(self):
+	def getAction(self):
+		action = ""
 		if self.myJoystick != None:
 			try:
 				xAx = 0
@@ -44,27 +45,29 @@ def haveAction(self):
 					self.doMove = 1
 
 				if self.myJoystick.get_button(0) and self.joyButtonDown == False: # speed up
-					return "speedUp"
+					action = "speedUp"
 				elif self.myJoystick.get_button(1) and self.joyButtonDown == False: # speed down
-					return "speedDown"
+					action = "speedDown"
 				elif (self.myJoystick.get_button(9) and self.joyButtonDown == False)\
 					or (self.myJoystick.get_button(3) and self.joyButtonDown == False): # (re)start
-					return "restart"
-				else:
-					# make sure NO button is down for reset
-					someJoyButtonDown = False
-					for i in range(0, self.myJoystick.get_numbuttons()):
-						if (self.myJoystick.get_button(i)):
-							someJoyButtonDown = True
-					self.joyButtonDown = someJoyButtonDown
+					action = "restart"
+				elif (self.myJoystick.get_button(8) and self.joyButtonDown == False):
+					action = 'quit'
+				#else:
+				# make sure NO button is down for reset
+				someJoyButtonDown = False
+				for i in range(0, self.myJoystick.get_numbuttons()):
+					if (self.myJoystick.get_button(i)):
+						someJoyButtonDown = True
+				self.joyButtonDown = someJoyButtonDown
 
 				if self.doMove != -1:
-					return "move"
+					action = "move"
 			except Exception as ex:
 				print('JOYSTICK ERROR, DEACTIVATING:' + str(ex))
 				self.myJoystick = None
 
-		return ""
+		return action
 
 	def getMoveAction(self):
 		tmp = self.doMove

popUp.py

@@ -0,0 +1,42 @@
+import pygame
+import time
+class popUp(object):
+	def __init__(self, screen):
+		self._screen = screen
+		self.fnt = pygame.font.SysFont("MS Comic Sans" , 30)
+		self.color = (245, 101, 44)  # orange ;)
+		self.popUps = []
+		self.defaultDuration = 3
+
+
+	def singlePopUp(self, txt, duration=0):
+		if duration == 0:
+			duration = self.defaultDuration
+		self.popUps = []
+		self.popUps.append({"txt":txt, "start":time.time(), "duration":duration})
+
+	def addPopUp(self, txt, duration=0):
+		if duration == 0:
+			duration = self.defaultDuration
+		self.popUps.append({"txt":txt, "start":time.time(), "duration":duration})
+
+	def drawPopUps(self):
+		if len(self.popUps) == 0:
+			return False
+
+		fullText = ""
+		for i in range(len(self.popUps)):
+			if time.time() > self.popUps[i]['start'] + self.popUps[i]['duration']:
+				self.popUps.pop(i)
+				break
+
+		for popUp in self.popUps:
+			fullText += popUp['txt'] + " "
+
+		# TODO: multilines!
+		# http://stackoverflow.com/questions/2770886/pygames-message-multiple-lines
+		if fullText != "":
+			xPos = (self._screen.get_width() / 2)
+			self._screen.blit(self.fnt.render(fullText, 1, self.color), ( xPos - (self.fnt.size(fullText)[0] / 2), 5))
+
+		return True

snake.py

@@ -7,6 +7,7 @@
 import pygame
 from ui import ui
 from joystick import joystick
+from popUp import popUp
 from pygame.sprite import Sprite
 
 #sys.stdout = os.devnull
@@ -56,6 +57,7 @@ def __init__(self):
 		self.screen = None
 		self.playerBox = None
 		self.ui = None
+		self.popUp = None
 
 
 	def move(self, direction):
@@ -164,11 +166,20 @@ def gameSpeedUp(self):
 		self.gameSpeedFactor += 1
 		if self.gameSpeedFactor >= len(self.gameSpeedFactors):
 			self.gameSpeedFactor -= 1
+		self.speedChanged()
 
 	def gameSpeedDown(self):
 		self.gameSpeedFactor -= 1
 		if self.gameSpeedFactor < 0:
 			self.gameSpeedFactor = 0
+		self.speedChanged()
+
+	def speedChanged(self):
+		oneBased = self.gameSpeedFactor + 1
+		txt = "|" * oneBased
+		txt = "[" + txt.ljust(len(self.gameSpeedFactors)) + "]"
+
+		self.popUp.singlePopUp(txt)
 
 	def run_game(self):
 		# Game parameters
@@ -196,6 +207,7 @@ def run_game(self):
 
 		# init the menu, add stuff later
 		self.iUi = ui(self.screen)
+		self.popUp = popUp(self.screen)
 
 		pygame.joystick.init()
 		self.joystickInteract = joystick()
@@ -222,19 +234,22 @@ def run_game(self):
 			worldChanged = False
 
 			if self.joystickInteract.joystickAvailable():
-				if self.joystickInteract.haveAction() == "move":
+				joyAction = self.joystickInteract.getAction()
+				if joyAction == "move":
 					doMove = self.joystickInteract.getMoveAction()
-				elif  self.joystickInteract.haveAction() == "speedUp":
+				elif  joyAction == "speedUp":
 					self.gameSpeedUp()
-				elif self.joystickInteract.haveAction() == "speedDown":
+				elif joyAction == "speedDown":
 					self.gameSpeedDown()
-				elif self.joystickInteract.haveAction() == "restart":
+				elif joyAction == "restart":
 					self.resetGame()
+					gameOver = False
+				elif joyAction == 'quit':
+					self.exit_game()
 
 			for event in pygame.event.get():
 				if event.type == pygame.QUIT:
 					self.exit_game()
-
 				elif event.type == pygame.KEYDOWN:
 					if event.key in keymap:
 						doMove = keymap[event.key]
@@ -270,7 +285,7 @@ def run_game(self):
 				for elem in reversed(self.elements):
 					elem.update()
 
-				# add elements BEVORE blit is called and they change direction!
+				# add elements BEFORE blit is called and they change direction!
 				# WEIRD stuff would happen otherwise!!!1!!!!!!!!
 				if len(self.haveToAdd) > 0:
 					for i in range(len(self.haveToAdd)):
@@ -283,23 +298,27 @@ def run_game(self):
 							self.haveToAdd.pop(i)
 							break
 				else:
-					# if theres NOTHING to add to the Snake, add a new snak, if needed
+					# if there is NOTHING to add to the Snake, add a new snak, if needed
 					# preventing from spawning a snack inside the "new" tail of the snake and shit
 					self.addSnack(self.elements)
 
 				# update elements
 				for elem in reversed(self.elements):
 					elem.blit()
 
-				# if a snak has been eaten, add it to the to add list
+				# draw pop ups
+				self.popUp.drawPopUps()
+
+				# if a snack has been eaten, add it to the to add list
 				snackEaten = self.eatSnack(self.elements)
 				if snackEaten != None:
 					self.haveToAdd.append(snackEaten)
+					self.popUp.singlePopUp(str(self.getBodyLen()))
 
 				# collision!
 				if self.headDied(self.elements) == True:
 					gameOver = True
-				
+
 				pygame.display.flip()