fix selecting menu template and background image files, only allow start of burning...
[vuplus_dvbapp] / lib / python / Plugins / Extensions / DVDBurn / ProjectSettings.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 enigma import eListboxPythonMultiContent, gFont, RT_HALIGN_LEFT
12 from Tools.Directories import fileExists, resolveFilename, SCOPE_PLUGINS, SCOPE_FONTS, SCOPE_HDD
13 from Components.config import config, getConfigListEntry
14 from Components.ConfigList import ConfigListScreen
15
16 class FileBrowser(Screen, HelpableScreen):
17         skin = """
18         <screen name="FileBrowser" position="100,100" size="520,376" title="DVD File Browser" >
19                 <widget name="filelist" position="0,0" size="520,376" scrollbarMode="showOnDemand" />
20         </screen>"""
21         def __init__(self, session, scope, configRef):
22                 Screen.__init__(self, session)
23                 HelpableScreen.__init__(self)
24                 self.scope = scope
25                 pattern = ""
26                 self.configRef = configRef
27                 currDir = "/"
28                 if self.scope == "project":
29                         currDir = self.getDir()
30                         pattern = "(?i)^.*\.(ddvdp\.xml)"
31                 elif self.scope == "menutemplate":
32                         currDir = self.getDir()
33                         pattern = "(?i)^.*\.(ddvdm\.xml)"
34                 if self.scope == "menubg":
35                         currDir = self.getDir(configRef.getValue())
36                         pattern = "(?i)^.*\.(jpeg|jpg|jpe|png|bmp)"
37                 elif self.scope == "menuaudio":
38                         currDir = self.getDir(configRef.getValue())
39                         pattern = "(?i)^.*\.(mp2|m2a|ac3)"
40                 elif self.scope == "vmgm":
41                         currDir = self.getDir(configRef.getValue())
42                         pattern = "(?i)^.*\.(mpg|mpeg)"
43                 elif self.scope == "font_face":
44                         currDir = self.getDir(configRef.getValue(), resolveFilename(SCOPE_FONTS))
45                         pattern = "(?i)^.*\.(ttf)"
46                 elif self.scope == "isopath":
47                         currDir = configRef.getValue()
48                 elif self.scope == "image":
49                         currDir = resolveFilename(SCOPE_HDD)
50                         pattern = "(?i)^.*\.(iso)"
51
52                 self.filelist = FileList(currDir, matchingPattern=pattern)
53                 self["filelist"] = self.filelist
54
55                 self["FilelistActions"] = ActionMap(["OkCancelActions"],
56                         {
57                                 "ok": self.ok,
58                                 "cancel": self.exit
59                         })
60
61         def getDir(self, currentVal=None, defaultDir=None):
62                 if currentVal:
63                         return (currentVal.rstrip("/").rsplit("/",1))[0]
64                 return defaultDir or (resolveFilename(SCOPE_PLUGINS)+"Extensions/DVDBurn/")
65
66         def ok(self):
67                 if self.filelist.canDescent():
68                         self.filelist.descent()
69                         if self.scope == "image":
70                                 path = self["filelist"].getCurrentDirectory() or ""
71                                 if fileExists(path+"VIDEO_TS"):
72                                         self.close(path,self.scope,self.configRef)
73                 else:
74                         ret = self["filelist"].getCurrentDirectory() + '/' + self["filelist"].getFilename()
75                         self.close(ret,self.scope,self.configRef)
76
77         def exit(self):
78                 if self.scope == "isopath":
79                         self.close(self["filelist"].getCurrentDirectory(),self.scope,self.configRef)
80                 self.close(None,False,None)
81
82 class ProjectSettings(Screen,ConfigListScreen):
83         skin = """
84                 <screen position="90,83" size="560,445" title="Collection settings" >
85                     <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" alphatest="on" />
86                     <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" size="140,40" alphatest="on" />
87                     <ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" size="140,40" alphatest="on" />
88                     <ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" size="140,40" alphatest="on" />
89                     <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" />
90                     <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" />
91                     <widget source="key_yellow" render="Label" position="280,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#a08500" transparent="1" />
92                     <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" />
93                     <widget name="config" position="10,50" size="540,276" scrollbarMode="showOnDemand" />
94                     <widget source="info" render="Label" position="20,350" size="520,90" font="Regular;16" />
95                 </screen>"""
96
97         def __init__(self, session, project = None):
98                 Screen.__init__(self, session)
99                 self.project = project
100                 
101                 self["key_red"] = StaticText(_("Cancel"))
102                 self["key_green"] = StaticText(_("OK"))
103                 self["key_yellow"] = StaticText(_("Load"))
104                 if config.usage.setup_level.index >= 2: # expert+
105                         self["key_blue"] = StaticText(_("Save"))
106                 else:
107                         self["key_blue"] = StaticText()
108                 
109                 if config.usage.setup_level.index >= 2: # expert+
110                         infotext = _("Available format variables") + ":\n$i=" + _("Track") + ", $t=" + _("Title") + ", $d=" + _("Description") + ", $l=" + _("length") + ", $c=" + _("chapters") + ",\n" + _("Record") + " $T=" + _("Begin time") + ", $Y=" + _("Year") + ", $M=" + _("month") + ", $D=" + _("day") + ",\n$A=" + _("audio tracks") + ", $C=" + _("Channel") + ", $f=" + _("filename")
111                 else:
112                         infotext = ""
113                 self["info"] = StaticText(infotext)
114
115                 self.keydict = {}
116                 self.settings = project.settings
117                 ConfigListScreen.__init__(self, [])
118                 self.initConfigList()
119                 
120                 self["setupActions"] = ActionMap(["SetupActions", "ColorActions"],
121                 {
122                     "green": self.exit,
123                     "red": self.cancel,
124                     "blue": self.saveProject,
125                     "yellow": self.loadProject,
126                     "cancel": self.cancel,
127                     "ok": self.ok,
128                 }, -2)
129
130         def changedConfigList(self):
131                 key = self.keydict[self["config"].getCurrent()[1]]
132                 if key == "authormode" or key == "output":
133                         self.initConfigList()
134
135         def initConfigList(self):
136                 authormode = self.settings.authormode.getValue()
137                 output = self.settings.output.getValue()
138                 self.list = []
139                 self.list.append(getConfigListEntry(_("Collection name"), self.settings.name))
140                 self.list.append(getConfigListEntry(_("Authoring mode"), self.settings.authormode))
141                 self.list.append(getConfigListEntry(("Output"), self.settings.output))
142                 if output == "iso":
143                         self.list.append(getConfigListEntry(_("ISO path"), self.settings.isopath))
144                 if authormode.startswith("menu"):
145                         self.list.append(getConfigListEntry(_("Menu")+' '+_("template file"), self.settings.menutemplate))
146                         if config.usage.setup_level.index >= 2: # expert+
147                                 self.list.append(getConfigListEntry(_("Menu")+' '+_("Title"), self.project.menutemplate.settings.titleformat))
148                                 self.list.append(getConfigListEntry(_("Menu")+' '+_("Subtitles"), self.project.menutemplate.settings.subtitleformat))
149                                 self.list.append(getConfigListEntry(_("Menu")+' '+_("background image"), self.project.menutemplate.settings.menubg))
150                         #self.list.append(getConfigListEntry(_("Menu")+' '+_("headline")+' '+_("color"), self.settings.color_headline))
151                         #self.list.append(getConfigListEntry(_("Menu")+' '+_("text")+' '+_("color"), self.settings.color_button))
152                         #self.list.append(getConfigListEntry(_("Menu")+' '+_("highlighted button")+' '+_("color"), self.settings.color_highlight))
153                         #self.list.append(getConfigListEntry(_("Menu")+' '+_("font face"), self.settings.font_face))
154                         #self.list.append(getConfigListEntry(_("Font size")+' ('+_("headline")+', '+_("Title")+', '+_("Subtitles")+')', self.settings.font_size))
155                         #self.list.append(getConfigListEntry(_("Menu")+' '+_("spaces (top, between rows, left)"), self.settings.space))
156                         #self.list.append(getConfigListEntry(_("Menu")+' '+_("Audio"), self.settings.menuaudio))
157                 if config.usage.setup_level.index >= 2: # expert+
158                         if authormode != "data_ts":
159                                 self.list.append(getConfigListEntry(_("Titleset mode"), self.settings.titlesetmode))
160                                 if self.settings.titlesetmode.getValue() == "single" or authormode == "just_linked":
161                                         self.list.append(getConfigListEntry(_("VMGM (intro trailer)"), self.settings.vmgm))
162                         else:
163                                 self.list.append(getConfigListEntry(("DVD data format"), self.settings.dataformat))
164                 
165                 self["config"].setList(self.list)
166                 self.keydict = {}
167                 for key, val in self.settings.dict().iteritems():
168                         self.keydict[val] = key
169                 for key, val in self.project.menutemplate.settings.dict().iteritems():
170                         self.keydict[val] = key
171
172         def keyLeft(self):
173                 ConfigListScreen.keyLeft(self)
174                 key = self.keydict[self["config"].getCurrent()[1]]
175                 if key == "authormode" or key == "output" or key=="titlesetmode":
176                         self.initConfigList()
177
178         def keyRight(self):
179                 ConfigListScreen.keyRight(self)
180                 key = self.keydict[self["config"].getCurrent()[1]]
181                 if key == "authormode" or key == "output" or key=="titlesetmode":
182                         self.initConfigList()
183
184         def exit(self):
185                 self.applySettings()
186                 self.close(True)
187
188         def applySettings(self):
189                 for x in self["config"].list:
190                         x[1].save()
191                 
192         def ok(self):
193                 key = self.keydict[self["config"].getCurrent()[1]]
194                 from DVDProject import ConfigFilename
195                 if type(self["config"].getCurrent()[1]) == ConfigFilename:
196                         self.session.openWithCallback(self.FileBrowserClosed, FileBrowser, key, self["config"].getCurrent()[1])
197
198         def cancel(self):
199                 self.close(False)
200
201         def loadProject(self):
202                 self.session.openWithCallback(self.FileBrowserClosed, FileBrowser, "project", self.settings)
203
204         def saveProject(self):
205                 if config.usage.setup_level.index >= 2: # expert+
206                         self.applySettings()
207                         ret = self.project.saveProject(resolveFilename(SCOPE_PLUGINS)+"Extensions/DVDBurn/")
208                         if ret.startswith:
209                                 text = _("Save")+' '+_('OK')+':\n'+ret
210                                 self.session.open(MessageBox,text,type = MessageBox.TYPE_INFO)
211                         else:
212                                 text = _("Save")+' '+_('Error')
213                                 self.session.open(MessageBox,text,type = MessageBox.TYPE_ERROR)
214
215         def FileBrowserClosed(self, path, scope, configRef):
216                 if scope == "menutemplate":
217                         if self.project.menutemplate.loadTemplate(path):
218                                 print "[ProjectSettings] menu template loaded"
219                                 configRef.setValue(path)
220                                 self.initConfigList()
221                         else:
222                                 self.session.open(MessageBox,self.project.error,MessageBox.TYPE_ERROR)
223                 elif scope == "project":
224                         if self.project.loadProject(path):
225                                 configRef.setValue(path)
226                                 self.initConfigList()
227                         else:
228                                 self.session.open(MessageBox,self.project.error,MessageBox.TYPE_ERROR)
229                 elif scope:
230                         configRef.setValue(path)
231                         self.initConfigList()