Code

Created new Arrange submenu for effects extensions
[inkscape.git] / share / extensions / embedimage.py
index 5277f0fa47cd85da2c34d68abd7e23f24e73880c..263cf592505a6489c307041ab4108d445208b2bf 100644 (file)
@@ -1,6 +1,6 @@
 #!/usr/bin/env python 
 '''
-Copyright (C) 2005 Aaron Spike, aaron@ekips.org
+Copyright (C) 2005,2007 Aaron Spike, aaron@ekips.org
 
 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -19,7 +19,7 @@ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 import inkex, os, base64
 
-class MyEffect(inkex.Effect):
+class Embedder(inkex.Effect):
     def __init__(self):
         inkex.Effect.__init__(self)
         self.OptionParser.add_option("-s", "--selectedonly",
@@ -28,26 +28,34 @@ class MyEffect(inkex.Effect):
             help="embed only selected images")
 
     def effect(self):
-        ctx = inkex.xml.xpath.Context.Context(self.document,processorNss=inkex.NSS)
-
         # if slectedonly is enabled and there is a selection only embed selected
         # images. otherwise embed all images
         if (self.options.selectedonly):
-            if (self.options.ids):
-                for id, node in self.selected.iteritems():
-                    if node.tagName == 'image':
-                        self.embedImage(node)
+            self.embedSelected(self.document, self.selected)
         else:
-            path = '//image'
-            for node in inkex.xml.xpath.Evaluate(path,self.document, context=ctx):
-                self.embedImage(node)
+            self.embedAll(self.document)
+
+    def embedSelected(self, document, selected):
+        self.document=document
+        self.selected=selected
+        if (self.options.ids):
+            for id, node in selected.iteritems():
+                if node.tag == inkex.addNS('image','svg'):
+                    self.embedImage(node)
+
+    def embedAll(self, document):
+        self.document=document #not that nice... oh well
+        path = '//svg:image'
+        for node in self.document.getroot().xpath(path, namespaces=inkex.NSS):
+            self.embedImage(node)
+
     def embedImage(self, node):
-        xlink = node.attributes.getNamedItemNS(inkex.NSS[u'xlink'],'href')
-        if (xlink.value[:4]!='data'):
-            absref=node.attributes.getNamedItemNS(inkex.NSS[u'sodipodi'],'absref')
-            href=node.attributes.getNamedItemNS(inkex.NSS[u'xlink'],'href')
-            svg=self.document.getElementsByTagName('svg')[0]
-            docbase=svg.attributes.getNamedItemNS(inkex.NSS[u'sodipodi'],'docbase')
+        xlink = node.get(inkex.addNS('href','xlink'))
+        if (xlink[:4]!='data'):
+            absref=node.get(inkex.addNS('absref','sodipodi'))
+            href=xlink
+            svg=self.document.getroot().xpath('/svg:svg', namespaces=inkex.NSS)[0]
+            docbase=svg.get(inkex.addNS('docbase','sodipodi'))
 
             path=''
             #path selection strategy:
@@ -56,15 +64,15 @@ class MyEffect(inkex.Effect):
             # 3. realpath-ified href
             # 4. absref, only if the above does not point to a file
             if (href != None):
-                if (os.path.isabs(href.value)):
-                    path=os.path.realpath(href.value)
+                if (os.path.isabs(href)):
+                    path=os.path.realpath(href)
                 elif (docbase != None):
-                    path=os.path.join(docbase.value,href.value)
+                    path=os.path.join(docbase,href)
                 else:
-                    path=os.path.realpath(href.value)
+                    path=os.path.realpath(href)
             if (not os.path.isfile(path)):
                 if (absref != None):
-                    path=absref.value
+                    path=absref
             if (not os.path.isfile(path)):
                 inkex.debug('No xlink:href or sodipodi:absref attributes found, or they do not point to an existing file! Unable to embed image.')
             
@@ -85,11 +93,14 @@ class MyEffect(inkex.Effect):
                 else:
                     embed=False
                 if (embed):
-                    xlink.value = 'data:%s;base64,%s' % (type, base64.encodestring(file))
-                    node.removeAttributeNS(inkex.NSS[u'sodipodi'],'absref')
+                    node.set(inkex.addNS('href','xlink'), 'data:%s;base64,%s' % (type, base64.encodestring(file)))
+                    if (absref != None):
+                        del node.attrib[inkex.addNS('absref',u'sodipodi')]
                 else:
                     inkex.debug("%s is not of type image/png, image/jpeg, image/bmp, image/gif or image/x-icon" % path)
             else:
                 inkex.debug("Sorry we could not locate %s" % path)
-e = MyEffect()
-e.affect()
+
+if __name__ == '__main__':
+    e = Embedder()
+    e.affect()