Code

fix bug [ 1685070 ] PathAlongPath - Error message when no paths selected
[inkscape.git] / share / extensions / inkex.py
index 041f7bff77edadc3a477d6ee5b497eb4d11752f2..8be7b5282869fa4355d7ae6d30a28730a68d0a25 100755 (executable)
@@ -19,37 +19,58 @@ You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 """
-import sys, copy, optparse
+import sys, copy, optparse, random, re
 
 #a dictionary of all of the xmlns prefixes in a standard inkscape doc
 NSS = {
-u'sodipodi'    :u'http://inkscape.sourceforge.net/DTD/sodipodi-0.dtd',
-u'cc'          :u'http://web.resource.org/cc/',
-u'svg'         :u'http://www.w3.org/2000/svg',
-u'dc'          :u'http://purl.org/dc/elements/1.1/',
-u'rdf'         :u'http://www.w3.org/1999/02/22-rdf-syntax-ns#',
-u'inkscape'    :u'http://www.inkscape.org/namespaces/inkscape',
-u'xlink'       :u'http://www.w3.org/1999/xlink'
+u'sodipodi'    :u'http://sodipodi.sourceforge.net/DTD/sodipodi-0.dtd',
+u'cc'        :u'http://web.resource.org/cc/',
+u'svg'        :u'http://www.w3.org/2000/svg',
+u'dc'        :u'http://purl.org/dc/elements/1.1/',
+u'rdf'        :u'http://www.w3.org/1999/02/22-rdf-syntax-ns#',
+u'inkscape'    :u'http://www.inkscape.org/namespaces/inkscape',
+u'xlink'    :u'http://www.w3.org/1999/xlink'
 }
 
+#a dictionary of unit to user unit conversion factors
+uuconv = {'in':90.0, 'pt':1.25, 'px':1, 'mm':3.5433070866, 'cm':35.433070866, 'pc':15.0}
+def unittouu(string):
+    '''Returns userunits given a string representation of units in another system'''
+    unit = re.compile('(%s)$' % '|'.join(uuconv.keys()))
+    param = re.compile(r'(([-+]?[0-9]+(\.[0-9]*)?|[-+]?\.[0-9]+)([eE][-+]?[0-9]+)?)')
+
+    p = param.match(string)
+    u = unit.search(string)    
+    if p:
+        retval = float(p.string[p.start():p.end()])
+    else:
+        retval = 0.0
+    if u:
+        try:
+            return retval * uuconv[u.string[u.start():u.end()]]
+        except KeyError:
+            pass
+    return retval
+
 try:
-       import xml.dom.ext
-       import xml.dom.ext.reader.Sax2
-       import xml.xpath
+    import xml.dom.ext
+    import xml.dom.minidom
+    import xml.dom.ext.reader.Sax2
+    import xml.xpath
 except:
-       sys.exit('The inkex.py module requires PyXML. Please download the latest version from <http://pyxml.sourceforge.net/>.')
+    sys.exit('The inkex.py module requires PyXML. Please download the latest version from <http://pyxml.sourceforge.net/>.')
 
 def debug(what):
-       sys.stderr.write(str(what) + "\n")
-       return what
+    sys.stderr.write(str(what) + "\n")
+    return what
 
 def check_inkbool(option, opt, value):
-       if str(value).capitalize() == 'True':
-               return True
-       elif str(value).capitalize() == 'False':
-               return False
-       else:
-               raise OptionValueError("option %s: invalid inkbool value: %s" % (opt, value))
+    if str(value).capitalize() == 'True':
+        return True
+    elif str(value).capitalize() == 'False':
+        return False
+    else:
+        raise OptionValueError("option %s: invalid inkbool value: %s" % (opt, value))
 
 class InkOption(optparse.Option):
     TYPES = optparse.Option.TYPES + ("inkbool",)
@@ -58,46 +79,97 @@ class InkOption(optparse.Option):
 
 
 class Effect:
-       """A class for creating Inkscape SVG Effects"""
-       def __init__(self):
-               self.document=None
-               self.selected={}
-               self.options=None
-               self.args=None
-               self.OptionParser = optparse.OptionParser(usage="usage: %prog [options] SVGfile",option_class=InkOption)
-               self.OptionParser.add_option("--id",
-                                               action="append", type="string", dest="ids", default=[], 
-                                               help="id attribute of object to manipulate")
-       def effect(self):
-               pass
-       def getoptions(self,args=sys.argv[1:]):
-               """Collect command line arguments"""
-               self.options, self.args = self.OptionParser.parse_args(args)
-       def parse(self,file=None):
-               """Parse document in specified file or on stdin"""
-               reader = xml.dom.ext.reader.Sax2.Reader()
-               try:
-                       try:
-                               stream = open(file,'r')
-                       except:
-                               stream = open(self.args[-1],'r')
-               except:
-                       stream = sys.stdin
-               self.document = reader.fromStream(stream)
-               stream.close()
-       def getselected(self):
-               """Collect selected nodes"""
-               for id in self.options.ids:
-                       path = '//*[@id="%s"]' % id
-                       for node in xml.xpath.Evaluate(path,self.document):
-                               self.selected[id] = node
-       def output(self):
-               """Serialize document into XML on stdout"""
-               xml.dom.ext.Print(self.document)
-       def affect(self):
-               """Affect an SVG document with a callback effect"""
-               self.getoptions()
-               self.parse()
-               self.getselected()
-               self.effect()
-               self.output()
+    """A class for creating Inkscape SVG Effects"""
+    def __init__(self, *args, **kwargs):
+        self.id_characters = '0123456789abcdefghijklmnopqrstuvwkyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
+        self.document=None
+        self.ctx=None
+        self.selected={}
+        self.doc_ids={}
+        self.options=None
+        self.args=None
+        self.use_minidom=kwargs.pop("use_minidom", False)
+        self.OptionParser = optparse.OptionParser(usage="usage: %prog [options] SVGfile",option_class=InkOption)
+        self.OptionParser.add_option("--id",
+                        action="append", type="string", dest="ids", default=[], 
+                        help="id attribute of object to manipulate")
+    def effect(self):
+        pass
+    def getoptions(self,args=sys.argv[1:]):
+        """Collect command line arguments"""
+        self.options, self.args = self.OptionParser.parse_args(args)
+    def parse(self,file=None):
+        """Parse document in specified file or on stdin"""
+        reader = xml.dom.ext.reader.Sax2.Reader()
+        try:
+            try:
+                stream = open(file,'r')
+            except:
+                stream = open(self.args[-1],'r')
+        except:
+            stream = sys.stdin
+        if self.use_minidom:
+            self.document = xml.dom.minidom.parse(stream)
+        else:
+            self.document = reader.fromStream(stream)
+        self.ctx = xml.xpath.Context.Context(self.document,processorNss=NSS)
+        stream.close()
+    def getposinlayer(self):
+        ctx = xml.xpath.Context.Context(self.document,processorNss=NSS)
+        #defaults
+        self.current_layer = self.document.documentElement
+        self.view_center = (0.0,0.0)
+
+        layerattr = xml.xpath.Evaluate('//sodipodi:namedview/@inkscape:current-layer',self.document,context=ctx)
+        if layerattr:
+            layername = layerattr[0].value
+            layer = xml.xpath.Evaluate('//g[@id="%s"]' % layername,self.document,context=ctx)
+            if layer:
+                self.current_layer = layer[0]
+
+        xattr = xml.xpath.Evaluate('//sodipodi:namedview/@inkscape:cx',self.document,context=ctx)
+        yattr = xml.xpath.Evaluate('//sodipodi:namedview/@inkscape:cy',self.document,context=ctx)
+        doc_height = unittouu(self.document.documentElement.getAttribute('height'))
+        if xattr and yattr:
+            x = xattr[0].value
+            y = yattr[0].value
+            if x and y:
+                self.view_center = (float(x), doc_height - float(y)) # FIXME: y-coordinate flip, eliminate it when it's gone in Inkscape
+    def getselected(self):
+        """Collect selected nodes"""
+        for id in self.options.ids:
+            path = '//*[@id="%s"]' % id
+            for node in xml.xpath.Evaluate(path,self.document):
+                self.selected[id] = node
+    def getdocids(self):
+        docIdNodes = xml.xpath.Evaluate('//@id',self.document,context=self.ctx)
+        for m in docIdNodes:
+            self.doc_ids[m.value] = 1
+    def output(self):
+        """Serialize document into XML on stdout"""
+        xml.dom.ext.Print(self.document)
+    def affect(self):
+        """Affect an SVG document with a callback effect"""
+        self.getoptions()
+        self.parse()
+        self.getposinlayer()
+        self.getselected()
+        self.getdocids()
+        self.effect()
+        self.output()
+        
+    def uniqueId(self, old_id, make_new_id = True):
+        new_id = old_id
+        if make_new_id:
+            while new_id in self.doc_ids:
+                new_id = "%s%s" % (new_id,random.choice(self.id_characters))
+            self.doc_ids[new_id] = 1
+        return new_id
+    def xpathSingle(self, path):
+        try:
+            retval = xml.xpath.Evaluate(path,self.document,context=self.ctx)[0]
+        except:
+            debug("No matching node for expression: %s" % path)
+            retval = None
+        return retval
+