X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/3c066dfc061d51c59caf201e302cfb2301b5f0f7..5d441513d7c81c7e52399801dd34c97480f4e136:/lib/python/Plugins/Extensions/DVDBurn/TitleList.py diff --git a/lib/python/Plugins/Extensions/DVDBurn/TitleList.py b/lib/python/Plugins/Extensions/DVDBurn/TitleList.py index d67f1123..aeb163f3 100644 --- a/lib/python/Plugins/Extensions/DVDBurn/TitleList.py +++ b/lib/python/Plugins/Extensions/DVDBurn/TitleList.py @@ -1,70 +1,20 @@ -import DVDProject, TitleList, TitleCutter - +import DVDProject, TitleList, TitleCutter, TitleProperties, ProjectSettings, DVDToolbox, Process from Screens.Screen import Screen from Screens.ChoiceBox import ChoiceBox from Screens.InputBox import InputBox from Screens.MessageBox import MessageBox from Screens.HelpMenu import HelpableScreen +from Screens.TaskView import JobView +from Components.Task import job_manager from Components.ActionMap import HelpableActionMap, ActionMap from Components.Sources.List import List from Components.Sources.StaticText import StaticText from Components.Sources.Progress import Progress -from Components.FileList import FileList +from Components.Label import Label from enigma import eListboxPythonMultiContent, gFont, RT_HALIGN_LEFT -from Tools.Directories import resolveFilename, SCOPE_PLAYLIST - -class WaitBox(MessageBox): - def __init__(self, session, callback): - MessageBox.__init__(self, session, text=_("please wait, loading picture..."), type = MessageBox.TYPE_INFO) - self.skinName = "MessageBox" - self.CB = callback - self.onShown.append(self.runCB) - - def ok(self): - pass - - def runCB(self): - from enigma import eTimer - self.delayTimer = eTimer() - self.delayTimer.callback.append(self.CB) - self.delayTimer.start(10,1) - -class FileBrowser(Screen, HelpableScreen): - skin = """ - - - """ - def __init__(self, session, currDir = None, projectBrowser = False): - Screen.__init__(self, session) - HelpableScreen.__init__(self) - self.projectBrowser = projectBrowser - if not currDir: - currDir = "/" - - if projectBrowser: - self.filelist = FileList(currDir, matchingPattern = "(?i)^.*\.(ddvdp\.xml)") - else: - self.filelist = FileList(currDir, matchingPattern = "(?i)^.*\.(jpeg|jpg|jpe|png|bmp)") - self["filelist"] = self.filelist - - self["FilelistActions"] = ActionMap(["OkCancelActions"], - { - "ok": self.ok, - "cancel": self.exit - }) - - def ok(self): - if self.filelist.canDescent(): - self.filelist.descent() - else: - ret = self["filelist"].getCurrentDirectory() + self["filelist"].getFilename() - self.close(ret,self.projectBrowser) - - def exit(self): - self.close(None) - +from Tools.Directories import resolveFilename, SCOPE_PLUGINS -class TitleList(Screen): +class TitleList(Screen, HelpableScreen): skin = """ @@ -75,7 +25,8 @@ class TitleList(Screen): - + + @@ -85,14 +36,15 @@ class TitleList(Screen): def __init__(self, session, project = None): Screen.__init__(self, session) + HelpableScreen.__init__(self) self["titleactions"] = HelpableActionMap(self, "DVDTitleList", { "addTitle": (self.addTitle, _("Add a new title"), _("Add title")), - "editTitle": (self.editTitle, _("Edit chapters of current title"), _("Edit title")), + "titleProperties": (self.titleProperties, _("Properties of current title"), _("Title properties")), "removeCurrentTitle": (self.removeCurrentTitle, _("Remove currently selected title"), _("Remove title")), - "saveProject": (self.saveProject, _("Save current collection to disk"), _("Save")), - "burnProject": (self.burnProject, _("Burn DVD"), _("Burn DVD")), + "settings": (self.settings, _("Collection settings"), _("Settings")), + "burnProject": (self.askBurnProject, _("Burn DVD"), _("Burn DVD")), }) self["MovieSelectionActions"] = HelpableActionMap(self, "MovieSelectionActions", @@ -100,22 +52,21 @@ class TitleList(Screen): "contextMenu": (self.showMenu, _("menu")), }) - self["key_red"] = StaticText(_("Add title")) - self["key_green"] = StaticText(_("Edit title")) - self["key_yellow"] = StaticText(_("Remove title")) - self["key_blue"] = StaticText(_("Save")) - - self["title_label"] = StaticText() - self["space_label"] = StaticText() - self["space_bar"] = Progress() - self["actions"] = ActionMap(["OkCancelActions"], { "cancel": self.leave }) - #Action("addTitle", self.addTitle) - + self["key_red"] = StaticText() + self["key_green"] = StaticText(_("Add title")) + self["key_yellow"] = StaticText() + self["key_blue"] = StaticText(_("Settings")) + + self["title_label"] = StaticText() + self["error_label"] = Label("") + self["space_label"] = StaticText() + self["space_bar"] = Progress() + if project is not None: self.project = project else: @@ -123,184 +74,233 @@ class TitleList(Screen): self["titles"] = List(list = [ ], enableWrapAround = True, item_height=30, fonts = [gFont("Regular", 20)]) self.updateTitleList() - self.updateCollectionName() - - #self["addTitle"] = ActionButton("titleactions", "addTitle") - #self["editTitle"] = ActionButton("titleactions", "editTitle") - #self["removeCurrentTitle"] = ActionButton("titleactions", "removeCurrentTitle") - #self["saveProject"] = ActionButton("titleactions", "saveProject") - #self["burnProject"] = ActionButton("titleactions", "burnProject") - + self.previous_size = 0 + + def checkBackgroundJobs(self): + for job in job_manager.getPendingJobs(): + print "type(job):", type(job) + print "Process.DVDJob:", Process.DVDJob + if type(job) == Process.DVDJob: + self.backgroundJob = job + return + self.backgroundJob = None + def showMenu(self): menu = [] - menu.append((_("Add a new title"), "addtitle")); - menu.append((_("Remove title"), "removetitle")); - menu.append((_("Edit chapters of current title"), "edittitle")); - menu.append((_("Set collection name"), "setname")); - menu.append((_("Set menu background"), "setbackground")); - menu.append((_("Save current collection to disk"), "save")); - menu.append((_("Load saved collection from disk"), "load")); - menu.append((_("Preview menu"), "previewMenu")); - menu.append((_("Burn DVD"), "burn")); + self.checkBackgroundJobs() + if self.backgroundJob: + j = self.backgroundJob + menu.append(("%s: %s (%d%%)" % (j.getStatustext(), j.name, int(100*j.progress/float(j.end))), self.showBackgroundJob)) + menu.append((_("DVD media toolbox"), self.toolbox)) + menu.append((_("Preview menu"), self.previewMenu)) + if self.project.settings.output.getValue() == "dvd": + if len(self["titles"].list): + menu.append((_("Burn DVD"), self.burnProject)) + elif self.project.settings.output.getValue() == "iso": + menu.append((_("Create DVD-ISO"), self.burnProject)) + menu.append((_("Burn existing image to DVD"), self.selectImage)) + menu.append((_("Edit chapters of current title"), self.editTitle)) + menu.append((_("Reset and renumerate title names"), self.resetTitles)) + menu.append((_("Exit"), self.leave)) self.session.openWithCallback(self.menuCallback, ChoiceBox, title="", list=menu) def menuCallback(self, choice): - if choice is None: - return + if choice: + choice[1]() - if choice[1] == "removetitle": - self.removeCurrentTitle() - elif choice[1] == "addtitle": - self.addTitle() - elif choice[1] == "edittitle": - self.editTitle() - elif choice[1] == "setname": - self.setName() - elif choice[1] == "setbackground": - self.showFileBrowser(False) - elif choice[1] == "save": - self.saveProject() - elif choice[1] == "load": - self.showFileBrowser(True) - elif choice[1] == "previewMenu": - self.previewMenu() - elif choice[1] == "burn": - self.burnProject() - - def setName(self): - self.session.openWithCallback(self.setNameCallback, InputBox, title=_("Set collection name"), text=self.project.name, maxSize=False, visible_width = 56) - - def setNameCallback(self, name): - if name is not None: - self.project.name = name - self.updateCollectionName() + def showBackgroundJob(self): + job_manager.in_background = False + self.session.openWithCallback(self.JobViewCB, JobView, self.backgroundJob) + self.backgroundJob = None + + def titleProperties(self): + if self.getCurrentTitle(): + self.session.openWithCallback(self.updateTitleList, TitleProperties.TitleProperties, self, self.project, self["titles"].getIndex()) + + def selectImage(self): + self.session.openWithCallback(self.burnISO, ProjectSettings.FileBrowser, "image", self.project.settings) def newProject(self): self.project = DVDProject.DVDProject() - if self.loadProject(): + if self.loadTemplate(): self.project.session = self.session - self.updateCollectionName() - - def updateCollectionName(self): - self["title_label"].text = _("Table of content for collection") + " \"" + self.project.name + "\":" + self.settingsCB() def addTitle(self): from Screens.MovieSelection import MovieSelection + from Components.Button import Button + from Components.ActionMap import HelpableActionMap class MovieSelectionNoMenu(MovieSelection): def __init__(self, session): MovieSelection.__init__(self, session) self.skinName = "MovieSelection" - + self["key_red"] = Button(_("Edit title")) + self["key_green"] = Button(_("Add")) + self["ColorActions"] = HelpableActionMap(self, "ColorActions", + { + "red": (self.movieSelected, _("Add a new title")), + "green": (self.insertWithoutEdit, ("insert without cutlist editor")) + }) + def updateTags(self): + pass def doContext(self): print "context menu forbidden inside DVDBurn to prevent calling multiple instances" - + def insertWithoutEdit(self): + current = self.getCurrent() + if current is not None: + current.edit = False + self.close(current) + def movieSelected(self): + current = self.getCurrent() + if current is not None: + current.edit = True + self.close(current) self.session.openWithCallback(self.selectedSource, MovieSelectionNoMenu) def selectedSource(self, source): if source is None: return None t = self.project.addService(source) - self.editTitle(t, readOnly=True) + try: + editor = source.edit + except AttributeError: + editor = True + self.editTitle(t, editor) def removeCurrentTitle(self): title = self.getCurrentTitle() + self.removeTitle(title) + + def removeTitle(self, title): if title is not None: self.project.titles.remove(title) self.updateTitleList() - def saveProject(self): - self.project.saveProject(resolveFilename(SCOPE_PLAYLIST)) + def toolbox(self): + self.session.open(DVDToolbox.DVDToolbox) + + def settings(self): + self.session.openWithCallback(self.settingsCB, ProjectSettings.ProjectSettings, self.project) + + def settingsCB(self, update=True): + if not update: + return + self["title_label"].text = _("Table of content for collection") + " \"" + self.project.settings.name.getValue() + "\":" - def loadProject(self, filename=None): - if not filename: - filename = resolveFilename(SCOPE_PLAYLIST)+"DreamboxDVDtemplate.ddvdp.xml" - if self.project.loadProject(filename): + def loadTemplate(self): + filename = resolveFilename(SCOPE_PLUGINS)+"Extensions/DVDBurn/DreamboxDVD.ddvdp.xml" + if self.project.load(filename): + self["error_label"].hide() return True else: - try: - self.session.open(MessageBox,self.project.error,MessageBox.TYPE_ERROR) - except: - self["title_label"].text = self.project.error - print self.project.error + self["error_label"].text = self.project.error + self["error_label"].show() return False - def burnProject(self): - self.project.waitboxref = self.project.session.open(WaitBox,self.burnProjectCB) + def askBurnProject(self): + if len(self["titles"].list): + self.session.openWithCallback(self.burnProject,MessageBox,text = _("Do you want to burn this collection to DVD medium?"), type = MessageBox.TYPE_YESNO) - def burnProjectCB(self): - import Process - job = Process.Burn(self.session, self.project) - from Screens.TaskView import JobView - self.session.open(JobView, job) + def burnProject(self, answer=True): + if not answer: + return + if self.project.settings.authormode.getValue() == "data_ts": + job = Process.DVDdataJob(self.project) + job_manager.AddJob(job) + job_manager.in_background = False + self.session.openWithCallback(self.JobViewCB, JobView, job) + else: + job = Process.DVDJob(self.project) + job_manager.AddJob(job) + job_manager.in_background = False + self.session.openWithCallback(self.JobViewCB, JobView, job) + + def burnISO(self, path, scope, configRef): + if path: + job = Process.DVDisoJob(self.project, path) + job_manager.AddJob(job) + job_manager.in_background = False + self.session.openWithCallback(self.JobViewCB, JobView, job) + + def JobViewCB(self, in_background): + job_manager.in_background = in_background def previewMenu(self): - self.project.waitboxref = self.project.session.open(WaitBox,self.previewMenuCB) + job = Process.DVDJob(self.project, menupreview=True) + job_manager.in_background = False + job_manager.AddJob(job) - def previewMenuCB(self): - import Process - job = Process.PreviewMenu(self.session, self.project) - def updateTitleList(self): res = [ ] - totalsize = 0 for title in self.project.titles: - a = [ title, (eListboxPythonMultiContent.TYPE_TEXT, 0, 10, 500, 50, 0, RT_HALIGN_LEFT, title.name) ] + a = [ title, (eListboxPythonMultiContent.TYPE_TEXT, 0, 5, 500, 25, 0, RT_HALIGN_LEFT, title.properties.menutitle.getValue()) ] res.append(a) - totalsize += title.estimatedDiskspace self["titles"].list = res - self.updateSize(totalsize) - - def updateSize(self, totalsize): - size = int((totalsize/1024)/1024) - max_SL = 4370 - max_DL = 7950 - if size > max_DL: - percent = 100 * size / float(max_DL) + self.updateSize() + if len(res): + self["key_red"].text = _("Remove title") + self["key_yellow"].text = _("Title properties") + else: + self["key_red"].text = "" + self["key_yellow"].text = "" + + def updateSize(self): + size = self.project.size + if size > self.project.MAX_DL: + percent = 100 * size / float(self.project.MAX_DL) self["space_label"].text = "%d MB - " % size + _("exceeds dual layer medium!") + " (%.2f%% " % (100-percent) + _("free") + ")" self["space_bar"].value = int(percent) - elif size > max_SL: - percent = 100 * size / float(max_DL) + if self.previous_size < self.project.MAX_DL: + self.session.open(MessageBox,text = _("exceeds dual layer medium!"), type = MessageBox.TYPE_ERROR) + elif size > self.project.MAX_SL: + percent = 100 * size / float(self.project.MAX_DL) self["space_label"].text = "%d MB " % size + _("of a DUAL layer medium used.") + " (%.2f%% " % (100-percent) + _("free") + ")" self["space_bar"].value = int(percent) - elif size < max_SL: - percent = 100 * size / float(max_SL) + if self.previous_size < self.project.MAX_SL: + self.session.open(MessageBox,text = _("Your collection exceeds the size of a single layer medium, you will need a blank dual layer DVD!"), type = MessageBox.TYPE_INFO) + elif size < self.project.MAX_SL: + percent = 100 * size / float(self.project.MAX_SL) self["space_label"].text = "%d MB " % size + _("of a SINGLE layer medium used.") + " (%.2f%% " % (100-percent) + _("free") + ")" self["space_bar"].value = int(percent) + self.previous_size = size def getCurrentTitle(self): t = self["titles"].getCurrent() return t and t[0] - def editTitle(self, title = None, readOnly = False): + def editTitle(self, title = None, editor = True): t = title or self.getCurrentTitle() if t is not None: self.current_edit_title = t - if readOnly: - self.session.openWithCallback(self.titleEditDone, TitleCutter.CutlistReader, t) - else: + if editor: self.session.openWithCallback(self.titleEditDone, TitleCutter.TitleCutter, t) + else: + self.session.openWithCallback(self.titleEditDone, TitleCutter.CutlistReader, t) def titleEditDone(self, cutlist): t = self.current_edit_title + t.initDVDmenuText(self.project,len(self.project.titles)) t.cuesheet = cutlist t.produceFinalCuesheet() - print "title edit of %s done, resulting cutlist:" % (t.source.toString()), t.cutlist, "chaptermarks:", t.chaptermarks + if t.VideoType != 0: + self.session.openWithCallback(self.DVDformatCB,MessageBox,text = _("The DVD standard doesn't support H.264 (HDTV) video streams. Do you want to create a Dreambox format data DVD (which will not play in stand-alone DVD players) instead?"), type = MessageBox.TYPE_YESNO) + else: + self.updateTitleList() + + def resetTitles(self): + count = 0 + for title in self.project.titles: + count += 1 + title.initDVDmenuText(self.project,count) self.updateTitleList() + def DVDformatCB(self, answer): + t = self.current_edit_title + if answer == True: + self.project.settings.authormode.setValue("data_ts") + self.updateTitleList() + else: + self.removeTitle(t) + def leave(self): self.close() - - def showFileBrowser(self, projectBrowser=False): - if projectBrowser: - currDir = resolveFilename(SCOPE_PLAYLIST) - else: - currDir = self.project.menubg - if len(currDir) > 1: - currDir = (currDir.rstrip("/").rsplit("/",1))[0] - self.session.openWithCallback(self.FileBrowserClosed, FileBrowser, currDir, projectBrowser) - - def FileBrowserClosed(self, path, projectBrowser=False): - if projectBrowser: - self.loadProject(path) - else: - self.project.menubg = path