re-commit new NFIFlash plugin (bug #480)
[enigma2.git] / lib / python / Plugins / Extensions / DVDBurn / DVDProject.py
index b0b8197a5e606a8d692333987ebcba533f3a7237..7f755db41ddab6d12ec8e6e5a8fda02181cfc959 100644 (file)
@@ -1,16 +1,21 @@
 from Tools.Directories import fileExists
 from Components.config import config, ConfigSubsection, ConfigInteger, ConfigText, ConfigSelection, getConfigListEntry, ConfigSequence, ConfigSubList
+import DVDTitle
+import xml.dom.minidom
 
 class ConfigColor(ConfigSequence):
-       def __init__(self):
-               ConfigSequence.__init__(self, seperator = "#", limits = [(0,255),(0,255),(0,255)])
+       def __init__(self, default = [128,128,128]):
+               ConfigSequence.__init__(self, seperator = "#", limits = [(0,255),(0,255),(0,255)], default = default)
 
 class ConfigFilename(ConfigText):
        def __init__(self):
                ConfigText.__init__(self, default = "", fixed_size = True, visible_width = False)
 
        def getMulti(self, selected):
-               filename = (self.text.rstrip("/").rsplit("/",1))[1].encode("utf-8")[:40] + " "
+               if self.text == "":
+                       return ("mtext"[1-selected:], "", 0)
+               cut_len = min(len(self.text),40)
+               filename = (self.text.rstrip("/").rsplit("/",1))[1].encode("utf-8")[:cut_len] + " "
                if self.allmarked:
                        mark = range(0, len(filename))
                else:
@@ -18,6 +23,8 @@ class ConfigFilename(ConfigText):
                return ("mtext"[1-selected:], filename, mark)
        
 class DVDProject:
+       MAX_SL = 4480
+       MAX_DL = 8150
        def __init__(self):
                self.titles = [ ]
                self.target = None
@@ -32,17 +39,17 @@ class DVDProject:
                self.settings.vmgm = ConfigFilename()
                self.filekeys = ["vmgm", "isopath", "menutemplate"]
                self.menutemplate = MenuTemplate()
+               self.error = ""
+               self.session = None
 
        def addService(self, service):
-               import DVDTitle
-               title = DVDTitle.DVDTitle()
+               title = DVDTitle.DVDTitle(self)
                title.addService(service)
                self.titles.append(title)
                return title
 
        def saveProject(self, path):
-               import xml.dom.minidom
-               from Tools.XMLTools import elementsWithTag, mergeText, stringToXML
+               from Tools.XMLTools import stringToXML
                list = []
                list.append('<?xml version="1.0" encoding="utf-8" ?>\n')
                list.append('<DreamDVDBurnerProject>\n')
@@ -99,47 +106,86 @@ class DVDProject:
                return ret
 
        def loadProject(self, filename):
-               import xml.dom.minidom
-               try:
+               #try:
                        if not fileExists(filename):
                                self.error = "xml file not found!"
-                               raise AttributeError
-                       else:
-                               self.error = ""
+                               #raise AttributeError
                        file = open(filename, "r")
                        data = file.read().decode("utf-8").replace('&',"&amp;").encode("ascii",'xmlcharrefreplace')
                        file.close()
                        projectfiledom = xml.dom.minidom.parseString(data)
-                       for project in projectfiledom.childNodes[0].childNodes:
-                         if project.nodeType == xml.dom.minidom.Element.nodeType:
-                           if project.tagName == 'settings':
-                               i = 0
-                               if project.attributes.length < len(self.settings.dict()):
-                                       self.error = "project attributes missing"
-                                       raise AttributeError
-                               while i < project.attributes.length:
-                                       item = project.attributes.item(i)
-                                       key = item.name.encode("utf-8")
-                                       try:
-                                               val = eval(item.nodeValue)
-                                       except (NameError, SyntaxError):
-                                               val = item.nodeValue.encode("utf-8")
-                                       try:
-                                               self.settings.dict()[key].setValue(val)
-                                       except (KeyError):
-                                               self.error = "unknown attribute '%s'" % (key)
-                                               raise AttributeError
-                                       i += 1
+                       for node in projectfiledom.childNodes[0].childNodes:
+                         print "node:", node
+                         if node.nodeType == xml.dom.minidom.Element.nodeType:
+                           if node.tagName == 'settings':
+                               self.xmlAttributesToConfig(node, self.settings)
+                           elif node.tagName == 'titles':
+                               self.xmlGetTitleNodeRecursive(node)
+                               
                        for key in self.filekeys:
                                val = self.settings.dict()[key].getValue()
                                if not fileExists(val):
                                        self.error += "\n%s '%s' not found" % (key, val)
-                       if len(self.error):
-                               raise AttributeError
+               #except AttributeError:
+                       #print "loadProject AttributeError", self.error
+                       #self.error += (" in project '%s'") % (filename)
+                       #return False
+                       return True
+
+       def xmlAttributesToConfig(self, node, config):
+               try:
+                       i = 0
+                       #if node.attributes.length < len(config.dict())-1:
+                               #self.error = "project attributes missing"
+                               #raise AttributeError
+                       while i < node.attributes.length:
+                               item = node.attributes.item(i)
+                               key = item.name.encode("utf-8")
+                               try:
+                                       val = eval(item.nodeValue)
+                               except (NameError, SyntaxError):
+                                       val = item.nodeValue.encode("utf-8")
+                               try:
+                                       print "config[%s].setValue(%s)" % (key, val)
+                                       config.dict()[key].setValue(val)
+                               except (KeyError):
+                                       self.error = "unknown attribute '%s'" % (key)
+                                       print "KeyError", self.error
+                                       raise AttributeError
+                               i += 1
                except AttributeError:
-                       self.error += (" in project '%s'") % (filename)
+                       self.error += (" XML attribute error '%s'") % node.toxml()
                        return False
-               return True
+
+       def xmlGetTitleNodeRecursive(self, node, title_idx = -1):
+               print "[xmlGetTitleNodeRecursive]", title_idx, node
+               print node.childNodes
+               for subnode in node.childNodes:
+                 print "xmlGetTitleNodeRecursive subnode:", subnode
+                 if subnode.nodeType == xml.dom.minidom.Element.nodeType:
+                   if subnode.tagName == 'title':
+                       title_idx += 1
+                       title = DVDTitle.DVDTitle(self)
+                       self.titles.append(title)
+                       self.xmlGetTitleNodeRecursive(subnode, title_idx)
+                   if subnode.tagName == 'path':
+                       print "path:", subnode.firstChild.data
+                       filename = subnode.firstChild.data
+                       self.titles[title_idx].addFile(filename.encode("utf-8"))
+                   if subnode.tagName == 'properties':
+                       self.xmlAttributesToConfig(node, self.titles[title_idx].properties)
+                   if subnode.tagName == 'audiotracks':
+                       self.xmlGetTitleNodeRecursive(subnode, title_idx)
+                   if subnode.tagName == 'audiotrack':
+                       print "audiotrack...", subnode.toxml()
+
+       def getSize(self):
+               totalsize = 0
+               for title in self.titles:
+                       totalsize += title.estimatedDiskspace
+               return totalsize
+
+       size = property(getSize)
 
 class MenuTemplate(DVDProject):
        def __init__(self):
@@ -176,6 +222,9 @@ class MenuTemplate(DVDProject):
                self.settings.thumb_size = ConfigSequence(seperator = ',', default = [200,158], limits = [(0,576),(-1,720)])
                self.settings.thumb_border = ConfigInteger(default = 2, limits = (0, 20))
                self.filekeys = ["menubg", "menuaudio", "fontface_headline", "fontface_title", "fontface_subtitle"]
+               from TitleProperties import languageChoices
+               self.settings.menulang = ConfigSelection(choices = languageChoices.choices, default=languageChoices.choices[1][0])
+               self.error = ""
 
        def loadTemplate(self, filename):
                ret = DVDProject.loadProject(self, filename)