X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/ec970c4f4534a4ed6f1f4d930bfb8d56bf5531a9..1b586ab3e622c38f22280ad21d2cd5a08c4424b9:/lib/python/Plugins/Extensions/DVDBurn/Process.py diff --git a/lib/python/Plugins/Extensions/DVDBurn/Process.py b/lib/python/Plugins/Extensions/DVDBurn/Process.py index c4221154..1cc11e2d 100644 --- a/lib/python/Plugins/Extensions/DVDBurn/Process.py +++ b/lib/python/Plugins/Extensions/DVDBurn/Process.py @@ -1,4 +1,5 @@ -from Components.Task import Task, Job, job_manager, DiskspacePrecondition, Condition +from Components.Task import Task, Job, DiskspacePrecondition, Condition, ToolExistsPrecondition +from Components.Harddisk import harddiskmanager from Screens.MessageBox import MessageBox class png2yuvTask(Task): @@ -7,17 +8,15 @@ class png2yuvTask(Task): self.setTool("/usr/bin/png2yuv") self.args += ["-n1", "-Ip", "-f25", "-j", inputfile] self.dumpFile = outputfile - self.weighting = 10 + self.weighting = 15 def run(self, callback, task_progress_changed): Task.run(self, callback, task_progress_changed) + self.container.stdoutAvail.remove(self.processStdout) self.container.dumpToFile(self.dumpFile) def processStderr(self, data): - print "[png2yuvTask]", data - - def processStdout(self, data): - pass + print "[png2yuvTask]", data[:-1] class mpeg2encTask(Task): def __init__(self, job, inputfile, outputfile): @@ -25,14 +24,14 @@ class mpeg2encTask(Task): self.setTool("/usr/bin/mpeg2enc") self.args += ["-f8", "-np", "-a2", "-o", outputfile] self.inputFile = inputfile - self.weighting = 10 + self.weighting = 25 def run(self, callback, task_progress_changed): Task.run(self, callback, task_progress_changed) self.container.readFromFile(self.inputFile) def processOutputLine(self, line): - print "[mpeg2encTask]", line + print "[mpeg2encTask]", line[:-1] class spumuxTask(Task): def __init__(self, job, xmlfile, inputfile, outputfile): @@ -41,18 +40,16 @@ class spumuxTask(Task): self.args += [xmlfile] self.inputFile = inputfile self.dumpFile = outputfile - self.weighting = 10 + self.weighting = 15 def run(self, callback, task_progress_changed): Task.run(self, callback, task_progress_changed) + self.container.stdoutAvail.remove(self.processStdout) self.container.dumpToFile(self.dumpFile) self.container.readFromFile(self.inputFile) def processStderr(self, data): - print "[spumuxTask]", data - - def processStdout(self, data): - pass + print "[spumuxTask]", data[:-1] class MakeFifoNode(Task): def __init__(self, job, number): @@ -69,6 +66,19 @@ class LinkTS(Task): self.args += ["-s", sourcefile, link_name] self.weighting = 10 +class CopyMeta(Task): + def __init__(self, job, sourcefile): + Task.__init__(self, job, "Copy title meta files") + self.setTool("/bin/cp") + from os import listdir + path, filename = sourcefile.rstrip("/").rsplit("/",1) + tsfiles = listdir(path) + for file in tsfiles: + if file.startswith(filename+"."): + self.args += [path+'/'+file] + self.args += [self.job.workspace] + self.weighting = 15 + class DemuxTask(Task): def __init__(self, job, inputfile): Task.__init__(self, job, "Demux video into ES") @@ -107,7 +117,7 @@ class DemuxTask(Task): self.generated_files.append(file) def haveProgress(self, progress): - print "PROGRESS [%s]" % progress + #print "PROGRESS [%s]" % progress MSG_CHECK = "check & synchronize audio file" MSG_DONE = "done..." if progress == "preparing collection(s)...": @@ -144,28 +154,55 @@ class DemuxTask(Task): for f in self.generated_files: os.remove(f) +class MplexTaskPostcondition(Condition): + def check(self, task): + if task.error == task.ERROR_UNDERRUN: + return True + return task.error is None + + def getErrorMessage(self, task): + return { + task.ERROR_UNDERRUN: ("Can't multiplex source video!"), + task.ERROR_UNKNOWN: ("An unknown error occured!") + }[task.error] + class MplexTask(Task): - def __init__(self, job, outputfile, inputfiles=None, demux_task=None): + ERROR_UNDERRUN, ERROR_UNKNOWN = range(2) + def __init__(self, job, outputfile, inputfiles=None, demux_task=None, weighting = 500): Task.__init__(self, job, "Mux ES into PS") - self.weighting = 500 + self.weighting = weighting self.demux_task = demux_task + self.postconditions.append(MplexTaskPostcondition()) self.setTool("/usr/bin/mplex") self.args += ["-f8", "-o", outputfile, "-v1"] if inputfiles: self.args += inputfiles + def setTool(self, tool): + self.cmd = tool + self.args = [tool] + self.global_preconditions.append(ToolExistsPrecondition()) + # we don't want the ReturncodePostcondition in this case because for right now we're just gonna ignore the fact that mplex fails with a buffer underrun error on some streams (this always at the very end) + def prepare(self): + self.error = None if self.demux_task: self.args += self.demux_task.generated_files def processOutputLine(self, line): - print "[MplexTask] processOutputLine=", line + print "[MplexTask] ", line[:-1] + if line.startswith("**ERROR:"): + if line.find("Frame data under-runs detected") != -1: + self.error = self.ERROR_UNDERRUN + else: + self.error = self.ERROR_UNKNOWN class RemoveESFiles(Task): def __init__(self, job, demux_task): Task.__init__(self, job, "Remove temp. files") self.demux_task = demux_task self.setTool("/bin/rm") + self.weighting = 10 def prepare(self): self.args += ["-f"] @@ -173,10 +210,8 @@ class RemoveESFiles(Task): self.args += [self.demux_task.cutfile] class DVDAuthorTask(Task): - def __init__(self, job, diskspaceNeeded): + def __init__(self, job): Task.__init__(self, job, "Authoring DVD") - - self.global_preconditions.append(DiskspacePrecondition(diskspaceNeeded)) self.weighting = 300 self.setTool("/usr/bin/dvdauthor") self.CWD = self.job.workspace @@ -184,7 +219,7 @@ class DVDAuthorTask(Task): self.menupreview = job.menupreview def processOutputLine(self, line): - print "[DVDAuthorTask] processOutputLine=", line + print "[DVDAuthorTask] ", line[:-1] if not self.menupreview and line.startswith("STAT: Processing"): self.callback(self, [], stay_resident=True) @@ -204,12 +239,14 @@ class WaitForResidentTasks(Task): callback(self, []) class BurnTaskPostcondition(Condition): + RECOVERABLE = True def check(self, task): return task.error is None def getErrorMessage(self, task): return { - task.ERROR_MEDIA: _("Medium is not a writeable DVD!"), + task.ERROR_NOTWRITEABLE: _("Medium is not a writeable DVD!"), + task.ERROR_LOAD: _("Could not load Medium! No disc inserted?"), task.ERROR_SIZE: _("Content does not fit on DVD!"), task.ERROR_WRITE_FAILED: _("Write failed!"), task.ERROR_DVDROM: _("No (supported) DVDROM found!"), @@ -218,26 +255,15 @@ class BurnTaskPostcondition(Condition): }[task.error] class BurnTask(Task): - ERROR_MEDIA, ERROR_SIZE, ERROR_WRITE_FAILED, ERROR_DVDROM, ERROR_ISOFS, ERROR_UNKNOWN = range(6) - def __init__(self, job): - Task.__init__(self, job, "burn") - + ERROR_NOTWRITEABLE, ERROR_LOAD, ERROR_SIZE, ERROR_WRITE_FAILED, ERROR_DVDROM, ERROR_ISOFS, ERROR_UNKNOWN = range(7) + def __init__(self, job, extra_args=[], tool="/bin/growisofs"): + Task.__init__(self, job, job.name) self.weighting = 500 self.end = 120 # 100 for writing, 10 for buffer flush, 10 for closing disc self.postconditions.append(BurnTaskPostcondition()) - self.setTool("/bin/growisofs") - volName = self.getASCIIname(job.project.settings.name.getValue()) - self.args += ["-dvd-video", "-dvd-compat", "-Z", "/dev/cdroms/cdrom0", "-V", volName, "-P", "Dreambox", "-use-the-force-luke=dummy", self.job.workspace + "/dvd"] - - def getASCIIname(self, name): - ASCIIname = "" - for char in name.decode("utf-8").encode("ascii","replace"): - if ord(char) <= 0x20 or ( ord(char) >= 0x3a and ord(char) <= 0x40 ): - ASCIIname += '_' - else: - ASCIIname += char - return ASCIIname - + self.setTool(tool) + self.args += extra_args + def prepare(self): self.error = None @@ -253,7 +279,9 @@ class BurnTask(Task): self.progress = 110 elif line.startswith(":-["): if line.find("ASC=30h") != -1: - self.error = self.ERROR_MEDIA + self.error = self.ERROR_NOTWRITEABLE + if line.find("ASC=24h") != -1: + self.error = self.ERROR_LOAD else: self.error = self.ERROR_UNKNOWN print "BurnTask: unknown error %s" % line @@ -262,10 +290,10 @@ class BurnTask(Task): self.error = self.ERROR_SIZE elif line.find("write failed") != -1: self.error = self.ERROR_WRITE_FAILED - elif line.find("unable to open64(\"/dev/cdroms/cdrom0\",O_RDONLY): No such file or directory") != -1: # fixme + elif line.find("unable to open64(") != -1 and line.find(",O_RDONLY): No such file or directory") != -1: self.error = self.ERROR_DVDROM elif line.find("media is not recognized as recordable DVD") != -1: - self.error = self.ERROR_MEDIA + self.error = self.ERROR_NOTWRITEABLE else: self.error = self.ERROR_UNKNOWN print "BurnTask: unknown error %s" % line @@ -281,26 +309,53 @@ class RemoveDVDFolder(Task): Task.__init__(self, job, "Remove temp. files") self.setTool("/bin/rm") self.args += ["-rf", self.job.workspace] + self.weighting = 10 -class PreviewTask(Task): +class CheckDiskspaceTask(Task): def __init__(self, job): + Task.__init__(self, job, "Checking free space") + totalsize = 50*1024*1024 # require an extra safety 50 MB + maxsize = 0 + for title in job.project.titles: + titlesize = title.estimatedDiskspace + if titlesize > maxsize: maxsize = titlesize + totalsize += titlesize + diskSpaceNeeded = totalsize + maxsize + self.global_preconditions.append(DiskspacePrecondition(diskSpaceNeeded)) + self.weighting = 5 + + def run(self, callback, task_progress_changed): + failed_preconditions = self.checkPreconditions(True) + self.checkPreconditions(False) + if len(failed_preconditions): + callback(self, failed_preconditions) + return + self.callback = callback + self.task_progress_changed = task_progress_changed + Task.processFinished(self, 0) + +class PreviewTask(Task): + def __init__(self, job, path): Task.__init__(self, job, "Preview") self.postconditions.append(PreviewTaskPostcondition()) self.job = job + self.path = path + self.weighting = 10 def run(self, callback, task_progress_changed): self.callback = callback self.task_progress_changed = task_progress_changed - if self.job.project.waitboxref: - self.job.project.waitboxref.close() if self.job.menupreview: - self.waitAndOpenPlayer() + self.previewProject() else: - self.job.project.session.openWithCallback(self.previewCB, MessageBox, _("Do you want to preview this DVD before burning?"), timeout = 60, default = False) + from Tools import Notifications + Notifications.AddNotificationWithCallback(self.previewCB, MessageBox, _("Do you want to preview this DVD before burning?"), timeout = 60, default = False) + + def abort(self): + self.finish(aborted = True) def previewCB(self, answer): if answer == True: - self.waitAndOpenPlayer() + self.previewProject() else: self.closedCB(True) @@ -308,7 +363,8 @@ class PreviewTask(Task): if self.job.menupreview: self.closedCB(True) else: - self.job.project.session.openWithCallback(self.closedCB, MessageBox, _("Do you want to burn this collection to DVD medium?") ) + from Tools import Notifications + Notifications.AddNotificationWithCallback(self.closedCB, MessageBox, _("Do you want to burn this collection to DVD medium?") ) def closedCB(self, answer): if answer == True: @@ -316,15 +372,9 @@ class PreviewTask(Task): else: Task.processFinished(self, 1) - def waitAndOpenPlayer(self): - from enigma import eTimer - self.delayTimer = eTimer() - self.delayTimer.callback.append(self.previewProject) - self.delayTimer.start(10,1) - def previewProject(self): from Plugins.Extensions.DVDPlayer.plugin import DVDPlayer - self.job.project.session.openWithCallback(self.playerClosed, DVDPlayer, dvd_filelist= [ self.job.project.workspace + "/dvd/VIDEO_TS/" ]) + self.job.project.session.openWithCallback(self.playerClosed, DVDPlayer, dvd_filelist= [ self.path ]) class PreviewTaskPostcondition(Condition): def check(self, task): @@ -333,31 +383,17 @@ class PreviewTaskPostcondition(Condition): def getErrorMessage(self, task): return "Cancel" -def getTitlesPerMenu(nr_titles): - if nr_titles < 6: - titles_per_menu = 5 - else: - titles_per_menu = 4 - return titles_per_menu - def formatTitle(template, title, track): - print template template = template.replace("$i", str(track)) - print template template = template.replace("$t", title.name) - print template template = template.replace("$d", title.descr) - print template template = template.replace("$c", str(len(title.chaptermarks)+1)) - print template + template = template.replace("$A", str(title.audiotracks)) template = template.replace("$f", title.inputfile) - print template template = template.replace("$C", title.channel) - print template l = title.length lengthstring = "%d:%02d:%02d" % (l/3600, l%3600/60, l%60) template = template.replace("$l", lengthstring) - print template if title.timeCreate: template = template.replace("$Y", str(title.timeCreate[0])) template = template.replace("$M", str(title.timeCreate[1])) @@ -368,129 +404,185 @@ def formatTitle(template, title, track): template = template.replace("$Y", "").replace("$M", "").replace("$D", "").replace("$T", "") return template.decode("utf-8") -def CreateMenus(job): - import os, Image, ImageDraw, ImageFont - imgwidth = 720 - imgheight = 576 - s = job.project.settings - im_bg_orig = Image.open(s.menubg.getValue()) - if im_bg_orig.size != (imgwidth, imgheight): - im_bg_orig = im_bg_orig.resize((720, 576)) - - fontsizes = s.font_size.getValue() - fontface = s.font_face.getValue() - - font0 = ImageFont.truetype(fontface, fontsizes[0]) - font1 = ImageFont.truetype(fontface, fontsizes[1]) - font2 = ImageFont.truetype(fontface, fontsizes[2]) +class ImagingPostcondition(Condition): + def check(self, task): + return task.returncode == 0 - color_headline = tuple(s.color_headline.getValue()) - color_button = tuple(s.color_button.getValue()) - color_highlight = tuple(s.color_highlight.getValue()) - spu_palette = [ 0x60, 0x60, 0x60 ] + s.color_highlight.getValue() + def getErrorMessage(self, task): + return _("Failed") + ": python-imaging" - nr_titles = len(job.project.titles) - titles_per_menu = getTitlesPerMenu(nr_titles) - job.nr_menus = ((nr_titles+titles_per_menu-1)/titles_per_menu) - - #a new menu_count every 5 titles (1,2,3,4,5->1 ; 6,7,8,9,10->2 etc.) - for menu_count in range(1 , job.nr_menus+1): - im_bg = im_bg_orig.copy() - im_high = Image.new("P", (imgwidth, imgheight), 0) - im_high.putpalette(spu_palette) - draw_bg = ImageDraw.Draw(im_bg) - draw_high = ImageDraw.Draw(im_high) - - if menu_count == 1: - headline = s.name.getValue().decode("utf-8") - textsize = draw_bg.textsize(headline, font=font0) - if textsize[0] > imgwidth: - offset = (0 , 20) - else: - offset = (((imgwidth-textsize[0]) / 2) , 20) - draw_bg.text(offset, headline, fill=color_headline, font=font0) +class ImagePrepareTask(Task): + def __init__(self, job): + Task.__init__(self, job, _("please wait, loading picture...")) + self.postconditions.append(ImagingPostcondition()) + self.weighting = 20 + self.job = job + self.Menus = job.Menus - menubgpngfilename = job.workspace+"/dvd_menubg"+str(menu_count)+".png" - highlightpngfilename = job.workspace+"/dvd_highlight"+str(menu_count)+".png" - spuxml = """ - - - """ % (highlightpngfilename, spu_palette[0], spu_palette[1], spu_palette[2]) - s_top, s_rows, s_left = s.space.getValue() - rowheight = (fontsizes[1]+fontsizes[2]+s_rows) - menu_start_title = (menu_count-1)*titles_per_menu + 1 - menu_end_title = (menu_count)*titles_per_menu + 1 - if menu_end_title > nr_titles: - menu_end_title = nr_titles+1 - menu_i = 0 - for title_no in range( menu_start_title , menu_end_title ): - i = title_no-1 - top = s_top + ( menu_i * rowheight ) - menu_i += 1 - title = job.project.titles[i] - titleText = formatTitle(s.titleformat.getValue(), title, title_no) - draw_bg.text((s_left,top), titleText, fill=color_button, font=font1) - draw_high.text((s_left,top), titleText, fill=1, font=font1) - subtitleText = formatTitle(s.subtitleformat.getValue(), title, title_no) - draw_bg.text((s_left,top+36), subtitleText, fill=color_button, font=font2) - bottom = top+rowheight - if bottom > imgheight: - bottom = imgheight - spuxml += """ -