0a664eba7c75f1e3a1d07ed23fd68f88fe4bbdae
[vuplus_dvbapp] / lib / python / Plugins / Extensions / DVDBurn / TitleProperties.py
1 from Screens.Screen import Screen
2 from Screens.ChoiceBox import ChoiceBox
3 from Screens.InputBox import InputBox
4 from Screens.MessageBox import MessageBox
5 from Screens.HelpMenu import HelpableScreen
6 from Components.ActionMap import HelpableActionMap, ActionMap
7 from Components.Sources.List import List
8 from Components.Sources.StaticText import StaticText
9 from Components.Sources.Progress import Progress
10 from Components.FileList import FileList
11 from Components.Pixmap import Pixmap
12 from enigma import ePicLoad
13 from Tools.Directories import fileExists, resolveFilename, SCOPE_PLUGINS, SCOPE_FONTS, SCOPE_HDD
14 from Components.config import config, getConfigListEntry, ConfigInteger, ConfigSubsection, ConfigSelection
15 from Components.ConfigList import ConfigListScreen
16 from Components.AVSwitch import AVSwitch
17 import DVDTitle
18
19 class TitleProperties(Screen,ConfigListScreen):
20         skin = """
21                 <screen name="TitleProperties" position="center,center" size="560,445" title="Properties of current title" >
22                         <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" alphatest="on" />
23                         <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" size="140,40" alphatest="on" />
24                         <ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" size="140,40" alphatest="on" />
25                         <widget source="key_red" render="Label" position="0,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#9f1313" transparent="1" />
26                         <widget source="key_green" render="Label" position="140,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#1f771f" transparent="1" />
27                         <widget source="key_blue" render="Label" position="420,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#18188b" transparent="1" />
28                         <widget source="serviceinfo" render="Label" position="10,46" size="350,144" font="Regular;18" />
29                         <widget name="thumbnail" position="370,46" size="180,144" alphatest="on" />
30                         <widget name="config" position="10,206" size="540,228" scrollbarMode="showOnDemand" />
31                 </screen>"""
32
33         def __init__(self, session, parent, project, title_idx):
34                 Screen.__init__(self, session)
35                 self.parent = parent
36                 self.project = project
37                 self.title_idx = title_idx
38
39                 self["key_red"] = StaticText(_("Cancel"))
40                 self["key_green"] = StaticText(_("OK"))
41                 self["key_blue"] = StaticText(_("Edit Title"))
42                 self["serviceinfo"] = StaticText()
43
44                 self["thumbnail"] = Pixmap()
45                 self.picload = ePicLoad()
46                 self.picload.PictureData.get().append(self.paintThumbPixmapCB)
47
48                 self.properties = project.titles[title_idx].properties
49                 ConfigListScreen.__init__(self, [])
50                 self.properties.crop = DVDTitle.ConfigFixedText("crop")
51                 self.properties.autochapter.addNotifier(self.initConfigList)
52                 self.properties.aspect.addNotifier(self.initConfigList)
53                 for audiotrack in self.properties.audiotracks:
54                         audiotrack.active.addNotifier(self.initConfigList)
55
56                 self["setupActions"] = ActionMap(["SetupActions", "ColorActions"],
57                 {
58                     "green": self.exit,
59                     "red": self.cancel,
60                     "blue": self.editTitle,
61                     "cancel": self.cancel,
62                     "ok": self.ok,
63                 }, -2)
64                 
65                 self.onShown.append(self.update)
66                 self.onLayoutFinish.append(self.layoutFinished)
67
68         def layoutFinished(self):
69                 self.setTitle(_("Properties of current title"))
70
71         def initConfigList(self, element=None):
72                 try:
73                         self.properties.position = ConfigInteger(default = self.title_idx+1, limits = (1, len(self.project.titles)))
74                         title = self.project.titles[self.title_idx]
75                         self.list = []
76                         self.list.append(getConfigListEntry("DVD " + _("Track"), self.properties.position))
77                         self.list.append(getConfigListEntry("DVD " + _("Title"), self.properties.menutitle))
78                         self.list.append(getConfigListEntry("DVD " + _("Description"), self.properties.menusubtitle))
79                         if config.usage.setup_level.index >= 2: # expert+
80                                 for audiotrack in self.properties.audiotracks:
81                                         DVB_aud = audiotrack.DVB_lang.getValue() or audiotrack.pid.getValue()
82                                         self.list.append(getConfigListEntry(_("burn audio track (%s)") % DVB_aud, audiotrack.active))
83                                         if audiotrack.active.getValue():
84                                                 self.list.append(getConfigListEntry(_("audio track (%s) format") % DVB_aud, audiotrack.format))
85                                                 self.list.append(getConfigListEntry(_("audio track (%s) language") % DVB_aud, audiotrack.language))
86                                                 
87                                 self.list.append(getConfigListEntry("DVD " + _("Aspect Ratio"), self.properties.aspect))
88                                 if self.properties.aspect.getValue() == "16:9":
89                                         self.list.append(getConfigListEntry("DVD " + "widescreen", self.properties.widescreen))
90                                 else:
91                                         self.list.append(getConfigListEntry("DVD " + "widescreen", self.properties.crop))
92                         if len(title.chaptermarks) == 0:
93                                 self.list.append(getConfigListEntry(_("Auto chapter split every ? minutes (0=never)"), self.properties.autochapter))
94                         infotext = "DVB " + _("Title") + ': ' + title.DVBname + "\n" + _("Description") + ': ' + title.DVBdescr + "\n" + _("Channel") + ': ' + title.DVBchannel + '\n' + _("Begin time") + title.formatDVDmenuText(": $D.$M.$Y, $T\n", self.title_idx+1)
95                         chaptermarks = title.getChapterMarks(template="$h:$m:$s")
96                         chapters_count = len(chaptermarks)
97                         if chapters_count >= 1:
98                                 infotext += str(chapters_count+1) + ' ' + _("chapters") + ': '
99                                 infotext += ' / '.join(chaptermarks)
100                         self["serviceinfo"].setText(infotext)
101                         self["config"].setList(self.list)
102                 except AttributeError:
103                         pass
104
105         def editTitle(self):
106                 self.parent.editTitle()
107
108         def update(self):
109                 print "[onShown]"
110                 self.initConfigList()
111                 self.loadThumb()
112
113         def loadThumb(self):
114                 thumbfile = self.project.titles[self.title_idx].inputfile.rsplit('.',1)[0] + ".png"
115                 sc = AVSwitch().getFramebufferScale()
116                 self.picload.setPara((self["thumbnail"].instance.size().width(), self["thumbnail"].instance.size().height(), sc[0], sc[1], False, 1, "#00000000"))
117                 self.picload.startDecode(thumbfile)
118
119         def paintThumbPixmapCB(self, picInfo=None):
120                 ptr = self.picload.getData()
121                 if ptr != None:
122                         self["thumbnail"].instance.setPixmap(ptr.__deref__())
123
124         def changedConfigList(self):
125                 self.initConfigList()
126
127         def exit(self):
128                 self.applySettings()
129                 self.close()
130
131         def applySettings(self):
132                 for x in self["config"].list:
133                         x[1].save()
134                 current_pos = self.title_idx+1
135                 new_pos = self.properties.position.getValue()
136                 if new_pos != current_pos:
137                         print "title got repositioned from ", current_pos, "to", new_pos
138                         swaptitle = self.project.titles.pop(current_pos-1)
139                         self.project.titles.insert(new_pos-1, swaptitle)
140
141         def ok(self):
142                 #key = self.keydict[self["config"].getCurrent()[1]]
143                 #if key in self.project.filekeys:
144                         #self.session.openWithCallback(self.FileBrowserClosed, FileBrowser, key, self.settings)
145                 pass
146
147         def cancel(self):
148                 self.close()
149
150 from Tools.ISO639 import LanguageCodes
151 class LanguageChoices():
152         def __init__(self):
153                 from Components.Language import language as syslanguage
154                 syslang = syslanguage.getLanguage()[:2]
155                 self.langdict = { }
156                 self.choices = []
157                 for key, val in LanguageCodes.iteritems():
158                         if len(key) == 2:
159                                 self.langdict[key] = val[0]
160                 for key, val in self.langdict.iteritems():
161                         if key not in (syslang, 'en'):
162                                 self.langdict[key] = val
163                                 self.choices.append((key, val))
164                 self.choices.sort()
165                 self.choices.insert(0,("nolang", ("unspecified")))
166                 self.choices.insert(1,(syslang, self.langdict[syslang]))
167                 if syslang != "en":
168                         self.choices.insert(2,("en", self.langdict["en"]))
169
170         def getLanguage(self, DVB_lang):
171                 DVB_lang = DVB_lang.lower()
172                 for word in ("stereo", "audio", "description", "2ch", "dolby digital"):
173                         DVB_lang = DVB_lang.replace(word,"").strip()
174                 for key, val in LanguageCodes.iteritems():
175                         if DVB_lang.find(key.lower()) == 0:
176                                 if len(key) == 2:
177                                         return key
178                                 else:
179                                         DVB_lang = (LanguageCodes[key])[0]
180                         elif DVB_lang.find(val[0].lower()) > -1:
181                                 if len(key) == 2:
182                                         return key
183                                 else:
184                                         DVB_lang = (LanguageCodes[key])[0]
185                 for key, val in self.langdict.iteritems():
186                         if val == DVB_lang:
187                                 return key
188                 return "nolang"
189
190 languageChoices = LanguageChoices()