improve audio language detection
[vuplus_dvbapp] / lib / python / Plugins / Extensions / DVDBurn / TitleCutter.py
1 from Plugins.Extensions.CutListEditor.plugin import CutListEditor
2 from Components.ServiceEventTracker import ServiceEventTracker
3 from enigma import iPlayableService, iServiceInformation
4
5 class TitleCutter(CutListEditor):
6         def __init__(self, session, t):
7                 CutListEditor.__init__(self, session, t.source)
8                 self.skin = CutListEditor.skin
9                 self.session = session
10                 self.t = t
11                 self.__event_tracker = ServiceEventTracker(screen=self, eventmap=
12                         {
13                                 iPlayableService.evUpdatedInfo: self.getPMTInfo,
14                                 iPlayableService.evCuesheetChanged: self.refillList
15                         })
16                 self.onExecBegin.remove(self.showTutorial)
17
18         def getPMTInfo(self):
19                 service = self.session.nav.getCurrentService()
20                 audio = service and service.audioTracks()
21                 n = audio and audio.getNumberOfTracks() or 0
22                 if n > 0:
23                         from DVDTitle import ConfigFixedText
24                         from TitleProperties import languageChoices
25                         from Components.config import config, ConfigSubsection, ConfigSubList, ConfigSelection, ConfigYesNo
26                         self.t.properties.audiotracks = ConfigSubList()
27                         for x in range(n):
28                                 i = audio.getTrackInfo(x)
29                                 DVB_lang = i.getLanguage()
30                                 description = i.getDescription()
31                                 pid = str(i.getPID())
32                                 if description == "MPEG":
33                                         description = "MP2"
34                                 print "[audiotrack] pid:", pid, "description:", description, "language:", DVB_lang
35                                 self.t.properties.audiotracks.append(ConfigSubsection())
36                                 self.t.properties.audiotracks[-1].active = ConfigYesNo(default = True)
37                                 self.t.properties.audiotracks[-1].format = ConfigFixedText(description)
38                                 self.t.properties.audiotracks[-1].language = ConfigSelection(choices = languageChoices.choices, default=languageChoices.getLanguage(DVB_lang))
39                                 self.t.properties.audiotracks[-1].pid = ConfigFixedText(pid)
40                                 self.t.properties.audiotracks[-1].DVB_lang = ConfigFixedText(DVB_lang)
41                 sAspect = service.info().getInfo(iServiceInformation.sAspect)
42                 if sAspect in ( 1, 2, 5, 6, 9, 0xA, 0xD, 0xE ):
43                         aspect = "4:3"
44                 else:
45                         aspect = "16:9"
46                 self.t.properties.aspect.setValue(aspect)
47                 self.t.VideoType = service.info().getInfo(iServiceInformation.sVideoType)
48
49         def exit(self):
50                 self.session.nav.stopService()
51                 self.close(self.cut_list[:])
52
53 class CutlistReader(TitleCutter):
54         def __init__(self, session, t):
55                 TitleCutter.__init__(self, session, t)
56
57         def getPMTInfo(self):
58                 TitleCutter.getPMTInfo(self)
59                 self.close(self.cut_list[:])