X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/0a3c667bfb6a1552975c1fd2325e7435363c2bc9..2c3fe3f9a4eabde0440dbf86fb39ac26062cdc65:/lib/python/Screens/ChoiceBox.py

diff --git a/lib/python/Screens/ChoiceBox.py b/lib/python/Screens/ChoiceBox.py
index 7192a229..4739b7e1 100644
--- a/lib/python/Screens/ChoiceBox.py
+++ b/lib/python/Screens/ChoiceBox.py
@@ -1,22 +1,35 @@
-from enigma import *
 from Screens.Screen import Screen
-from Screens.MessageBox import MessageBox
 from Components.ActionMap import NumberActionMap
 from Components.Label import Label
-from Components.MenuList import MenuList
-from Components.GUIComponent import *
-
-import os
+from Components.ChoiceList import ChoiceEntryComponent, ChoiceList
+from Components.Sources.StaticText import StaticText
 
 class ChoiceBox(Screen):
-	def __init__(self, session, title = "", list = []):
+	def __init__(self, session, title = "", list = [], keys = None, selection = 0):
 		Screen.__init__(self, session)
 
 		self["text"] = Label(title)
-		self.list = list
-		self["list"] = MenuList(list)
+		self.list = []
+		self.summarylist = []
+		if keys is None:
+			self.__keys = [ "1", "2", "3", "4", "5", "6", "7", "8", "9", "0", "red", "green", "yellow", "blue" ] + (len(list) - 10) * [""]
+		else:
+			self.__keys = keys + (len(list) - len(keys)) * [""]
+			
+		self.keymap = {}
+		pos = 0
+		for x in list:
+			strpos = str(self.__keys[pos])
+			self.list.append(ChoiceEntryComponent(key = strpos, text = x))
+			if self.__keys[pos] != "":
+				self.keymap[self.__keys[pos]] = list[pos]
+			self.summarylist.append((self.__keys[pos],x[0]))
+			pos += 1
+		self["list"] = ChoiceList(list = self.list, selection = selection)
+		self["summary_list"] = StaticText()
+		self.updateSummary()
 				
-		self["actions"] = NumberActionMap(["WizardActions", "InputActions"], 
+		self["actions"] = NumberActionMap(["WizardActions", "InputActions", "ColorActions", "DirectionActions"], 
 		{
 			"ok": self.go,
 			"back": self.cancel,
@@ -29,7 +42,13 @@ class ChoiceBox(Screen):
 			"7": self.keyNumberGlobal,
 			"8": self.keyNumberGlobal,
 			"9": self.keyNumberGlobal,
-			"0": self.keyNumberGlobal
+			"0": self.keyNumberGlobal,
+			"red": self.keyRed,
+			"green": self.keyGreen,
+			"yellow": self.keyYellow,
+			"blue": self.keyBlue,
+			"up": self.up,
+			"down": self.down
 		}, -1)
 		
 	def keyLeft(self):
@@ -38,13 +57,74 @@ class ChoiceBox(Screen):
 	def keyRight(self):
 		pass
 	
+	def up(self):
+		if len(self["list"].list) > 0:
+			while 1:
+				self["list"].instance.moveSelection(self["list"].instance.moveUp)
+				self.updateSummary(self["list"].l.getCurrentSelectionIndex())
+				if self["list"].l.getCurrentSelection()[0][0] != "--" or self["list"].l.getCurrentSelectionIndex() == 0:
+					break
+
+	def down(self):
+		if len(self["list"].list) > 0:
+			while 1:
+				self["list"].instance.moveSelection(self["list"].instance.moveDown)
+				self.updateSummary(self["list"].l.getCurrentSelectionIndex())
+				if self["list"].l.getCurrentSelection()[0][0] != "--" or self["list"].l.getCurrentSelectionIndex() == len(self["list"].list) - 1:
+					break
+
+	# runs a number shortcut
 	def keyNumberGlobal(self, number):
-		print "pressed", number
-		#self["input"].number(number)
-		
+		self.goKey(str(number))
+
+	# runs the current selected entry
 	def go(self):
-		self.close(self["list"].l.getCurrentSelection())
-		#self.close(self["input"].getText())
-		
+		cursel = self["list"].l.getCurrentSelection()
+		if cursel:
+			self.goEntry(cursel[0])
+		else:
+			self.cancel()
+
+	# runs a specific entry
+	def goEntry(self, entry):
+		if len(entry) > 2 and isinstance(entry[1], str) and entry[1] == "CALLFUNC":
+			# CALLFUNC wants to have the current selection as argument
+			arg = self["list"].l.getCurrentSelection()[0]
+			entry[2](arg)
+		else:
+			self.close(entry)
+
+	# lookups a key in the keymap, then runs it
+	def goKey(self, key):
+		if self.keymap.has_key(key):
+			entry = self.keymap[key]
+			self.goEntry(entry)
+
+	# runs a color shortcut
+	def keyRed(self):
+		self.goKey("red")
+
+	def keyGreen(self):
+		self.goKey("green")
+
+	def keyYellow(self):
+		self.goKey("yellow")
+
+	def keyBlue(self):
+		self.goKey("blue")
+
+	def updateSummary(self, curpos=0):
+		pos = 0
+		summarytext = ""
+		for entry in self.summarylist:
+			if pos > curpos-2 and pos < curpos+5:
+				if pos == curpos:
+					summarytext += ">"
+				else:
+					summarytext += entry[0]
+				summarytext += ' ' + entry[1] + '\n'
+			pos += 1
+		self["summary_list"].setText(summarytext)
+
 	def cancel(self):
-		self.close(None)
\ No newline at end of file
+		self.close(None)