code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Plugins/Extensions/DVDBurn/DVDProject.py

   1 from Tools.Directories import fileExists
   2 from Components.config import config, ConfigSubsection, ConfigInteger, ConfigText, ConfigSelection, getConfigListEntry, ConfigSequence, ConfigSubList
   3 import DVDTitle
   4 import xml.dom.minidom
   5 from Tools.Directories import resolveFilename, SCOPE_PLUGINS, SCOPE_FONTS
   6
   7 class ConfigColor(ConfigSequence):
   8         def __init__(self, default = [128,128,128]):
   9                 ConfigSequence.__init__(self, seperator = "#", limits = [(0,255),(0,255),(0,255)], default = default)
  10
  11 class ConfigFilename(ConfigText):
  12         def __init__(self):
  13                 ConfigText.__init__(self, default = "", fixed_size = True, visible_width = False)
  14
  15         def getMulti(self, selected):
  16                 if self.text == "":
  17                         return ("mtext"[1-selected:], "", 0)
  18                 cut_len = min(len(self.text),40)
  19                 filename = (self.text.rstrip("/").rsplit("/",1))[1].encode("utf-8")[:cut_len] + " "
  20                 if self.allmarked:
  21                         mark = range(0, len(filename))
  22                 else:
  23                         mark = [filename]
  24                 return ("mtext"[1-selected:], filename, mark)
  25
  26 class DVDProject:
  27         MAX_SL = 4480
  28         MAX_DL = 8150
  29         def __init__(self):
  30                 self.titles = [ ]
  31                 self.target = None
  32                 self.settings = ConfigSubsection()
  33                 self.settings.name = ConfigText(fixed_size = False, visible_width = 40)
  34                 self.settings.authormode = ConfigSelection(choices = [("menu_linked", _("Linked titles with a DVD menu")), ("just_linked", _("Direct playback of linked titles without menu")), ("menu_seperate", _("Seperate titles with a main menu")), ("data_ts", _("Dreambox format data DVD (HDTV compatible)"))])
  35                 self.settings.titlesetmode = ConfigSelection(choices = [("single", _("Simple titleset (compatibility for legacy players)")), ("multi", _("Complex (allows mixing audio tracks and aspects)"))], default="multi")
  36                 self.settings.output = ConfigSelection(choices = [("iso", _("Create DVD-ISO")), ("dvd", _("Burn DVD"))])
  37                 self.settings.isopath = ConfigText(fixed_size = False, visible_width = 40)
  38                 self.settings.dataformat = ConfigSelection(choices = [("iso9660_1", ("ISO9660 Level 1")), ("iso9660_4", ("ISO9660 version 2")), ("udf", ("UDF"))])
  39                 self.settings.menutemplate = ConfigFilename()
  40                 self.settings.vmgm = ConfigFilename()
  41                 self.filekeys = ["vmgm", "isopath", "menutemplate"]
  42                 self.menutemplate = MenuTemplate()
  43                 self.error = ""
  44                 self.session = None
  45
  46         def addService(self, service):
  47                 title = DVDTitle.DVDTitle(self)
  48                 title.addService(service)
  49                 self.titles.append(title)
  50                 return title
  51
  52         def saveProject(self, path):
  53                 from Tools.XMLTools import stringToXML
  54                 list = []
  55                 list.append('<?xml version="1.0" encoding="utf-8" ?>\n')
  56                 list.append('<DreamDVDBurnerProject>\n')
  57                 list.append('\t<settings ')
  58                 for key, val in self.settings.dict().iteritems():
  59                         list.append( key + '="' + str(val.getValue()) + '" ' )
  60                 list.append('/>\n')
  61                 list.append('\t<titles>\n')
  62                 for title in self.titles:
  63                         list.append('\t\t<title>\n')
  64                         list.append('\t\t\t<path>')
  65                         list.append(stringToXML(title.source.getPath()))
  66                         list.append('</path>\n')
  67                         list.append('\t\t\t<properties ')
  68                         audiotracks = []
  69                         for key, val in title.properties.dict().iteritems():
  70                                 if type(val) is ConfigSubList:
  71                                         audiotracks.append('\t\t\t<audiotracks>\n')
  72                                         for audiotrack in val:
  73                                                 audiotracks.append('\t\t\t\t<audiotrack ')
  74                                                 for subkey, subval in audiotrack.dict().iteritems():
  75                                                         audiotracks.append( subkey + '="' + str(subval.getValue()) + '" ' )
  76                                                 audiotracks.append(' />\n')
  77                                         audiotracks.append('\t\t\t</audiotracks>\n')
  78                                 else:
  79                                         list.append( key + '="' + str(val.getValue()) + '" ' )
  80                         list.append('/>\n')
  81                         for line in audiotracks:
  82                                 list.append(line)
  83                         list.append('\t\t</title>\n')
  84                 list.append('\t</titles>\n')
  85                 list.append('</DreamDVDBurnerProject>\n')
  86
  87                 name = self.settings.name.getValue()
  88                 i = 0
  89                 filename = path + name + ".ddvdp.xml"
  90                 while fileExists(filename):
  91                         i = i+1
  92                         filename = path + name + str(i).zfill(3) + ".ddvdp.xml"
  93                 try:
  94                         file = open(filename, "w")
  95                         for x in list:
  96                                 file.write(x)
  97                         file.close()
  98                 except:
  99                         return False
 100                 return filename
 101
 102         def load(self, filename):
 103                 ret = self.loadProject(filename)
 104                 if ret:
 105                         ret = self.menutemplate.loadTemplate(self.settings.menutemplate.getValue())
 106                         self.error += self.menutemplate.error
 107                 return ret
 108
 109         def loadProject(self, filename):
 110                 #try:
 111                         if not fileExists(filename):
 112                                 self.error = "xml file not found!"
 113                                 #raise AttributeError
 114                         file = open(filename, "r")
 115                         data = file.read().decode("utf-8").replace('&',"&amp;").encode("ascii",'xmlcharrefreplace')
 116                         file.close()
 117                         projectfiledom = xml.dom.minidom.parseString(data)
 118                         for node in projectfiledom.childNodes[0].childNodes:
 119                           print "node:", node
 120                           if node.nodeType == xml.dom.minidom.Element.nodeType:
 121                             if node.tagName == 'settings':
 122                                 self.xmlAttributesToConfig(node, self.settings)
 123                             elif node.tagName == 'titles':
 124                                 self.xmlGetTitleNodeRecursive(node)
 125
 126                         for key in self.filekeys:
 127                                 val = self.settings.dict()[key].getValue()
 128                                 if not fileExists(val):
 129                                         if val[0] != "/":
 130                                                 if key.find("font") == 0:
 131                                                         val = resolveFilename(SCOPE_FONTS)+val
 132                                                 else:
 133                                                         val = resolveFilename(SCOPE_PLUGINS)+"Extensions/DVDBurn/"+val
 134                                                 if fileExists(val):
 135                                                         self.settings.dict()[key].setValue(val)
 136                                                         continue
 137                                         self.error += "\n%s '%s' not found" % (key, val)
 138                 #except AttributeError:
 139                         #print "loadProject AttributeError", self.error
 140                         #self.error += (" in project '%s'") % (filename)
 141                         #return False
 142                         return True
 143
 144         def xmlAttributesToConfig(self, node, config):
 145                 try:
 146                         i = 0
 147                         #if node.attributes.length < len(config.dict())-1:
 148                                 #self.error = "project attributes missing"
 149                                 #raise AttributeError
 150                         while i < node.attributes.length:
 151                                 item = node.attributes.item(i)
 152                                 key = item.name.encode("utf-8")
 153                                 try:
 154                                         val = eval(item.nodeValue)
 155                                 except (NameError, SyntaxError):
 156                                         val = item.nodeValue.encode("utf-8")
 157                                 try:
 158                                         print "config[%s].setValue(%s)" % (key, val)
 159                                         config.dict()[key].setValue(val)
 160                                 except (KeyError):
 161                                         self.error = "unknown attribute '%s'" % (key)
 162                                         print "KeyError", self.error
 163                                         raise AttributeError
 164                                 i += 1
 165                 except AttributeError:
 166                         self.error += (" XML attribute error '%s'") % node.toxml()
 167                         return False
 168
 169         def xmlGetTitleNodeRecursive(self, node, title_idx = -1):
 170                 print "[xmlGetTitleNodeRecursive]", title_idx, node
 171                 print node.childNodes
 172                 for subnode in node.childNodes:
 173                   print "xmlGetTitleNodeRecursive subnode:", subnode
 174                   if subnode.nodeType == xml.dom.minidom.Element.nodeType:
 175                     if subnode.tagName == 'title':
 176                         title_idx += 1
 177                         title = DVDTitle.DVDTitle(self)
 178                         self.titles.append(title)
 179                         self.xmlGetTitleNodeRecursive(subnode, title_idx)
 180                     if subnode.tagName == 'path':
 181                         print "path:", subnode.firstChild.data
 182                         filename = subnode.firstChild.data
 183                         self.titles[title_idx].addFile(filename.encode("utf-8"))
 184                     if subnode.tagName == 'properties':
 185                         self.xmlAttributesToConfig(node, self.titles[title_idx].properties)
 186                     if subnode.tagName == 'audiotracks':
 187                         self.xmlGetTitleNodeRecursive(subnode, title_idx)
 188                     if subnode.tagName == 'audiotrack':
 189                         print "audiotrack...", subnode.toxml()
 190
 191         def getSize(self):
 192                 totalsize = 0
 193                 for title in self.titles:
 194                         totalsize += title.estimatedDiskspace
 195                 return totalsize
 196
 197         size = property(getSize)
 198
 199 class MenuTemplate(DVDProject):
 200         def __init__(self):
 201                 self.settings = ConfigSubsection()
 202                 self.settings.titleformat = ConfigText(fixed_size = False, visible_width = 40)
 203                 self.settings.subtitleformat = ConfigText(fixed_size = False, visible_width = 40)
 204                 self.settings.menubg = ConfigFilename()
 205                 self.settings.menuaudio = ConfigFilename()
 206                 self.settings.dimensions = ConfigSequence(seperator = ',', default = [576,720], limits = [(352,720),(480,576)])
 207                 self.settings.rows = ConfigInteger(default = 4, limits = (1, 10))
 208                 self.settings.cols = ConfigInteger(default = 1, limits = (1, 4))
 209                 self.settings.color_headline = ConfigColor()
 210                 self.settings.color_headline = ConfigColor()
 211                 self.settings.color_highlight = ConfigColor()
 212                 self.settings.color_button = ConfigColor()
 213                 self.settings.fontface_headline = ConfigFilename()
 214                 self.settings.fontface_title = ConfigFilename()
 215                 self.settings.fontface_subtitle = ConfigFilename()
 216                 self.settings.fontsize_headline = ConfigInteger(default = 46, limits = (0, 199))
 217                 self.settings.fontsize_title = ConfigInteger(default = 24, limits = (0, 199))
 218                 self.settings.fontsize_subtitle = ConfigInteger(default = 14, limits = (0, 199))
 219                 self.settings.margin_top = ConfigInteger(default = 120, limits = (0, 500))
 220                 self.settings.margin_bottom = ConfigInteger(default = 40, limits = (0, 500))
 221                 self.settings.margin_left = ConfigInteger(default = 56, limits = (0, 500))
 222                 self.settings.margin_right = ConfigInteger(default = 56, limits = (0, 500))
 223                 self.settings.space_rows = ConfigInteger(default = 32, limits = (0, 500))
 224                 self.settings.space_cols = ConfigInteger(default = 24, limits = (0, 500))
 225                 self.settings.prev_page_text = ConfigText(default = "<<<", fixed_size = False)
 226                 self.settings.next_page_text = ConfigText(default = ">>>", fixed_size = False)
 227                 self.settings.offset_headline = ConfigSequence(seperator = ',', default = [0,0], limits = [(-1,500),(-1,500)])
 228                 self.settings.offset_title = ConfigSequence(seperator = ',', default = [0,0], limits = [(-1,500),(-1,500)])
 229                 self.settings.offset_subtitle = ConfigSequence(seperator = ',', default = [20,0], limits = [(-1,500),(-1,500)])
 230                 self.settings.offset_thumb = ConfigSequence(seperator = ',', default = [40,0], limits = [(-1,500),(-1,500)])
 231                 self.settings.thumb_size = ConfigSequence(seperator = ',', default = [200,158], limits = [(0,576),(-1,720)])
 232                 self.settings.thumb_border = ConfigInteger(default = 2, limits = (0, 20))
 233                 self.filekeys = ["menubg", "menuaudio", "fontface_headline", "fontface_title", "fontface_subtitle"]
 234                 from TitleProperties import languageChoices
 235                 self.settings.menulang = ConfigSelection(choices = languageChoices.choices, default=languageChoices.choices[1][0])
 236                 self.error = ""
 237
 238         def loadTemplate(self, filename):
 239                 ret = DVDProject.loadProject(self, filename)
 240                 DVDProject.error = self.error
 241                 return ret