PyCharm, better score ;), working ui generator

Author: bison--

.gitignore

@@ -33,3 +33,6 @@ nosetests.xml
 .mr.developer.cfg
 .project
 .pydevproject
+
+# PyCharm
+.idea

highscore.txt

@@ -1 +1 @@
-26
+29

snake.py

@@ -183,7 +183,7 @@ def run_game(self):
 		pygame.init()
 
 		# do fancy window stuff
-		pygame.display.set_caption("SNAKE")
+		pygame.display.set_caption("pySnake")
 		#pygame.display.set_icon(pygame.image.load('imgs/bandit.jpg'))
 		pygame.mouse.set_visible(False)
 

ui.py

@@ -13,17 +13,25 @@ def __init__(self, screen):
 		# pygame.display.set_mode(...)
 		self._screen = screen
 		self.keymap = {pygame.K_UP:1, pygame.K_RIGHT:2, pygame.K_DOWN:3, pygame.K_LEFT:4}
-		
+		self.selectedColor = (245, 101, 44)  # orange ;)
 		self.nextAction = None
 
 		self.menus = {}
+
+		self._selectedMenu = ''  # menu 'name'
+		self._selectedMenuItem = ''  # selected index
+		self._selectedMenuItemIndex = -1  # selected index
 
 	def interaction(self, eventKey):
 		canInteract = False
 
 		if eventKey in self.keymap:
 			canInteract = True
-			nextAction = self.keymap[eventKey]
+			self.nextAction = self.keymap[eventKey]
+			if self.nextAction == 1:  # UP
+				self.selectMenuItem(-1)
+			elif self.nextAction == 3:  # DOWN
+				self.selectMenuItem(1)
 		else:
 			# reset the action if multiple times pressed keys before
 			# actually did something
@@ -34,7 +42,7 @@ def interaction(self, eventKey):
 	def addMenu(self, menuKey, menuRows):
 		''' add an menu
 		menuKey: unique name of the menu
-		menueRows: [
+		menuRows: [
 					{"rowName":"title", "selectable":False, "font":"MS Comic Sans", "fontSize":30, "color":(0,0,255), "text":"UserInterfaceGenerator"},
 					{"rowName":"start", "selectable":True, "font":"MS Comic Sans", "fontSize":30, "color":(0,0,255), "text":"Start Game"},
 					{"rowName":"q", "selectable":True, "font":"MS Comic Sans", "fontSize":30, "color":(123,55,255), "text":"QUIT"}
@@ -53,11 +61,70 @@ def addSimpleMenu(self, menuKey, menuRows, font="MS Comic Sans", size=30, color=
 			menu.append({"rowName":"simple_" + str(rowCount), "selectable":False, "font":font, "fontSize":size, "color":color, "text":row})
 
 		self.menus[menuKey] = menu
-	
-	def draw(self, menuKey):
+
+	def __getAvailableIndexes(self):
+		indexList = []
+		if self._selectedMenu in self.menus:
+			for i in range(len(self.menus[self._selectedMenu])):
+				if self.menus[self._selectedMenu][i]['selectable']:
+					indexList.append(i)
+		return indexList
+
+	def selectMenuItem(self, direction=0):
+		''' select the menu item
+		direction: +1 (down) or -1 (up), 0 for the first selectable item in the current menu
+		'''
+		#self._selectedIndex
+
+		indexList = self.__getAvailableIndexes()
+		if len(indexList) == 0:
+			print('NO MENU ITEMS')
+			return -1
+
+		if self._selectedMenu in self.menus and direction == 0:
+			self._selectedMenuItemIndex = indexList[0]
+			self._selectedMenuItem = self.menus[self._selectedMenu][indexList[0]]
+		else:
+			currentIndex = 0
+			if self._selectedMenuItemIndex in indexList:
+				currentIndex = indexList.index(self._selectedMenuItemIndex)
+			nextIndex = currentIndex + direction
+
+			# allow top-down / down-top switching
+			if nextIndex >= len(indexList):  # last element ogf list!
+				self._selectedMenuItemIndex = indexList[0]
+				self._selectedMenuItem = self.menus[self._selectedMenu][self._selectedMenuItemIndex]
+			elif nextIndex < 0:
+				self._selectedMenuItemIndex = indexList[-1]  # last element ogf list!
+				self._selectedMenuItem = self.menus[self._selectedMenu][self._selectedMenuItemIndex]
+			else:
+				self._selectedMenuItemIndex = indexList[nextIndex]
+				self._selectedMenuItem = self.menus[self._selectedMenu][self._selectedMenuItemIndex]
+
+		return self._selectedMenuItem
+
+	def getSelectedItem(self):
+		if self._selectedMenu in self.menus and self._selectedMenuItemIndex >= 0:
+			return self.menus[self._selectedMenu][self._selectedMenuItemIndex]
+
+		print("WARNING! NO SELECTED MENU ITEM FOUND!")
+		return None
+
+	def draw(self, menuKey=''):
 		if menuKey in self.menus:
-			menuToDraw = self.menus[menuKey]
-			
+			self._selectedMenu = menuKey
+		elif menuKey == '':
+			if self._selectedMenu == '' and 'main' in self.menus:
+				self._selectedMenu = 'main'
+		else:
+			raise Exception("Error menu '"+ menuKey +"' does not exist and there is no 'main' menu")
+
+		menuToDraw = self.menus[self._selectedMenu]
+
+		if self._selectedMenuItemIndex == -1:
+			self.selectMenuItem(0)
+
+		if menuToDraw is not None:
 			#print fnt.size(resultText)
 			menuRowsCount = len(menuToDraw)
 			for i in range(menuRowsCount):
@@ -66,7 +133,12 @@ def draw(self, menuKey):
 				xPos = (self._screen.get_width() / 2)
 				yPos = (self._screen.get_height() / 30) * menuRowsCount
 				txt = menuToDraw[i]["text"]
-				self._screen.blit(fnt.render(txt, 1, menuToDraw[i]["color"]), ( xPos - (fnt.size(txt)[0] / 2), yPos +(i * fontSize)))
+
+				color = menuToDraw[i]["color"]
+				if i == self._selectedMenuItemIndex:
+					color = self.selectedColor
+
+				self._screen.blit(fnt.render(txt, 1, color), ( xPos - (fnt.size(txt)[0] / 2), yPos +(i * fontSize)))
 		else:
 			raise Exception("Error menu '"+ menuKey +"' does not exist")
 
@@ -81,7 +153,9 @@ def draw(self, menuKey):
 	iUi.addMenu("main", [
 						{"rowName":"title", "selectable":False, "font":"MS Comic Sans", "fontSize":30, "color":(0,0,255), "text":"UserInterfaceGenerator"},
 						{"rowName":"start", "selectable":True, "font":"MS Comic Sans", "fontSize":30, "color":(0,0,255), "text":"Start Game"},
-						{"rowName":"q", "selectable":True, "font":"MS Comic Sans", "fontSize":30, "color":(123,55,255), "text":"QUIT"}
+						{"rowName":"info",  "selectable":True, "font":"MS Comic Sans", "fontSize":30, "color":(123,55,255), "text":"1NF0$"},
+						{"rowName":"q",     "selectable":True, "font":"MS Comic Sans", "fontSize":30, "color":(123,55,255), "text":"QUIT"},
+	                    {"rowName":"notes",  "selectable":False, "font":"MS Comic Sans", "fontSize":30, "color":(123,55,255), "text":"sometext"},
 						]
 				)