1 from Screen import Screen
2 from Components.ServiceEventTracker import ServiceEventTracker
3 from Components.ActionMap import ActionMap
4 from Components.ConfigList import ConfigListScreen
5 from Components.ChoiceList import ChoiceList, ChoiceEntryComponent
6 from Components.config import config, ConfigSubsection, getConfigListEntry, ConfigNothing, ConfigSelection, ConfigOnOff
7 from Components.MultiContent import MultiContentEntryText
8 from Components.Sources.List import List
9 from Components.Sources.Boolean import Boolean
10 from Components.SystemInfo import SystemInfo
12 from enigma import iPlayableService
14 from Tools.ISO639 import LanguageCodes
15 from Tools.BoundFunction import boundFunction
16 FOCUS_CONFIG, FOCUS_STREAMS = range(2)
18 class AudioSelection(Screen, ConfigListScreen):
19 def __init__(self, session, infobar=None):
20 Screen.__init__(self, session)
22 self["streams"] = List([])
23 self["key_red"] = Boolean(False)
24 self["key_green"] = Boolean(False)
25 self["key_yellow"] = Boolean(True)
26 self["key_blue"] = Boolean(False)
28 ConfigListScreen.__init__(self, [])
29 self.infobar = infobar or self.session.infobar
31 self.__event_tracker = ServiceEventTracker(screen=self, eventmap=
33 iPlayableService.evUpdatedInfo: self.__updatedInfo
35 self.cached_subtitle_checked = False
36 self.__selected_subtitle = None
38 self["actions"] = ActionMap(["ColorActions", "SetupActions", "DirectionActions"],
41 "green": self.keyGreen,
42 "yellow": self.keyYellow,
45 "cancel": self.cancel,
50 self.settings = ConfigSubsection()
51 choicelist = [("audio",_("audio tracks")), ("subtitles",_("Subtitles"))]
52 self.settings.menupage = ConfigSelection(choices = choicelist)
53 self.settings.menupage.addNotifier(self.fillList)
54 self.onLayoutFinish.append(self.__layoutFinished)
56 def __layoutFinished(self):
57 self["config"].instance.setSelectionEnable(False)
58 self.focus = FOCUS_STREAMS
60 def fillList(self, arg=None):
65 service = self.session.nav.getCurrentService()
66 self.audioTracks = audio = service and service.audioTracks()
67 n = audio and audio.getNumberOfTracks() or 0
69 if self.settings.menupage.getValue() == "audio":
70 self.setTitle(_("Select audio track"))
71 if SystemInfo["CanDownmixAC3"]:
72 print "config.av.downmix_ac3.value=", config.av.downmix_ac3.value
73 self.settings.downmix = ConfigOnOff(default=config.av.downmix_ac3.value)
74 self.settings.downmix.addNotifier(self.changeAC3Downmix, initial_call = False)
75 conflist.append(getConfigListEntry(_("AC3 downmix"), self.settings.downmix))
76 self["key_red"].setBoolean(True)
79 self.audioChannel = service.audioChannel()
80 print "self.audioChannel.getCurrentChannel()", self.audioChannel.getCurrentChannel()
81 choicelist = [("0",_("left")), ("1",_("stereo")), ("2", _("right"))]
82 self.settings.channelmode = ConfigSelection(choices = choicelist, default = str(self.audioChannel.getCurrentChannel()))
83 self.settings.channelmode.addNotifier(self.changeMode, initial_call = False)
84 conflist.append(getConfigListEntry(_("Channel"), self.settings.channelmode))
85 self["key_green"].setBoolean(True)
87 selectedAudio = self.audioTracks.getCurrentTrack()
88 print "selectedAudio:", selectedAudio
92 i = audio.getTrackInfo(x)
93 languages = i.getLanguage().split('/')
94 description = i.getDescription() or _("<unknown>")
98 if selectedAudio == x:
99 selected = _("Running")
103 for lang in languages:
106 if LanguageCodes.has_key(lang):
107 language += LanguageCodes[lang][0]
114 streams.append((x, "", number, description, language, selected))
119 elif self.settings.menupage.getValue() == "subtitles":
120 self.setTitle(_("Subtitle selection"))
122 self.settings.dummy = ConfigNothing()
123 conflist.append(getConfigListEntry("", self.settings.dummy))
124 self["key_red"].setBoolean(False)
125 conflist.append(getConfigListEntry("", self.settings.dummy))
126 self["key_green"].setBoolean(False)
128 if self.subtitlesEnabled():
129 sel = self.infobar.selected_subtitle
135 subtitlelist = self.getSubtitleList()
137 if len(subtitlelist):
138 for x in subtitlelist:
141 language = _("<unknown>")
144 if sel and x[:4] == sel[:4]:
145 selected = _("Running")
149 if LanguageCodes.has_key(x[4]):
150 language = LanguageCodes[x[4]][0]
156 number = "%x" % (x[1])
160 number = "%x%02x" % (x[3],x[2])
163 types = (" UTF-8 text "," SSA / AAS "," .SRT file ")
164 description = types[x[2]]
166 streams.append((x, "", number, description, language, selected))
172 conflist.append(getConfigListEntry(_("Menu"), self.settings.menupage))
174 from Components.PluginComponent import plugins
175 from Plugins.Plugin import PluginDescriptor
177 if hasattr(self.infobar, "runPlugin"):
179 def __init__(self, fnc, *args):
182 def __call__(self, *args, **kwargs):
185 Plugins = [ (p.name, PluginCaller(self.infobar.runPlugin, p)) for p in plugins.getPlugins(where = PluginDescriptor.WHERE_AUDIOMENU) ]
187 print "Plugins", Plugins
189 self.settings.dummy = ConfigNothing()
191 self["key_blue"].setBoolean(True)
192 conflist.append(getConfigListEntry(Plugins[0][0], self.settings.dummy))
193 self.plugincallfunc = Plugins[0][1]
195 self["config"].list = conflist
196 self["config"].l.setList(conflist)
198 print "setting self[streams].list:", streams
199 self["streams"].list = streams
200 print "setting self[streams].setIndex:", selectedidx
201 self["streams"].setIndex(selectedidx)
204 def __updatedInfo(self):
207 def getSubtitleList(self):
208 s = self.infobar and self.infobar.getCurrentServiceSubtitle()
209 l = s and s.getSubtitleList() or [ ]
212 def subtitlesEnabled(self):
213 return self.infobar.subtitles_enabled
215 def enableSubtitle(self, subtitles):
216 print "[enableSubtitle]", subtitles
217 if self.infobar.selected_subtitle != subtitles:
218 self.infobar.subtitles_enabled = False
219 self.infobar.selected_subtitle = subtitles
221 self.infobar.subtitles_enabled = True
223 def changeAC3Downmix(self, downmix):
224 print "changeAC3Downmix config.av.downmix_ac3.value=", config.av.downmix_ac3.value, downmix.getValue()
225 if downmix.getValue() == True:
226 config.av.downmix_ac3.value = True
228 config.av.downmix_ac3.value = False
229 config.av.downmix_ac3.save()
231 def changeMode(self, mode):
232 print "changeMode", mode, mode.getValue()
234 self.audioChannel.selectChannel(int(mode.getValue()))
236 def changeAudio(self, audio):
237 print "changeAudio", audio, "self.session.nav.getCurrentService().audioTracks().getNumberOfTracks():", self.session.nav.getCurrentService().audioTracks().getNumberOfTracks()
239 if isinstance(track, int):
240 if self.session.nav.getCurrentService().audioTracks().getNumberOfTracks() > track:
241 self.audioTracks.selectTrack(track)
244 if self.focus == FOCUS_CONFIG:
245 ConfigListScreen.keyLeft(self)
246 elif self.focus == FOCUS_STREAMS:
247 self["streams"].setIndex(0)
249 def keyRight(self, config = False):
250 if config or self.focus == FOCUS_CONFIG:
251 if self["config"].getCurrentIndex() < 3:
252 ConfigListScreen.keyRight(self)
253 elif hasattr(self, "plugincallfunc"):
254 self.plugincallfunc()
255 if self.focus == FOCUS_STREAMS and self["streams"].count():
256 self["streams"].setIndex(self["streams"].count()-1)
259 if self["key_red"].getBoolean():
263 if self["key_green"].getBoolean():
267 if self["key_yellow"].getBoolean():
271 if self["key_blue"].getBoolean():
275 def colorkey(self, idx):
276 self["config"].setCurrentIndex(idx)
280 print "[keyUp]", self["streams"].getIndex()
281 if self.focus == FOCUS_CONFIG:
282 self["config"].instance.moveSelection(self["config"].instance.moveUp)
283 elif self.focus == FOCUS_STREAMS:
284 if self["streams"].getIndex() == 0:
285 self["config"].instance.setSelectionEnable(True)
286 self["streams"].style = "notselected"
287 self["config"].setCurrentIndex(len(self["config"].getList())-1)
288 self.focus = FOCUS_CONFIG
290 self["streams"].selectPrevious()
293 print "[keyDown]", self["config"].getCurrentIndex(), len(self["config"].getList())-1
294 if self.focus == FOCUS_CONFIG:
295 if self["config"].getCurrentIndex() < len(self["config"].getList())-1:
296 self["config"].instance.moveSelection(self["config"].instance.moveDown)
298 self["config"].instance.setSelectionEnable(False)
299 self["streams"].style = "default"
300 self.focus = FOCUS_STREAMS
301 elif self.focus == FOCUS_STREAMS:
302 self["streams"].selectNext()
305 print "[keyok]", self["streams"].list, self["streams"].getCurrent()
306 if self.focus == FOCUS_STREAMS and self["streams"].list:
307 cur = self["streams"].getCurrent()
308 if self.settings.menupage.getValue() == "audio" and cur[0] is not None:
309 self.changeAudio(cur[2])
311 if self.settings.menupage.getValue() == "subtitles" and cur[0] is not None:
312 if self.infobar.selected_subtitle == cur[0]:
313 self.enableSubtitle(None)
314 selectedidx = self["streams"].getIndex()
316 self["streams"].setIndex(selectedidx)
318 self.enableSubtitle(cur[0])
321 elif self.focus == FOCUS_CONFIG: