Code

noop: Add vim modeline for all share/extensions/*.py files that use four-space indent...
[inkscape.git] / share / extensions / perspective.py
index 7f95ea22742dc907ad1a7188eb76dbc2c745271b..26a23dc0a8baf55788fc3cb319ee89bbaec0250c 100755 (executable)
@@ -20,8 +20,12 @@ Perspective approach & math by Dmitry Platonov, shadowjack@mail.ru, 2006
 """
 import sys, inkex, os, re, simplepath, cubicsuperpath 
 from ffgeom import *
-from numpy import *
-from numpy.linalg import *
+try:
+    from numpy import *
+    from numpy.linalg import *
+except:
+    inkex.debug("Failed to import the numpy or numpy.linalg modules. These modules are required by this extension. Please install them and try again.  On a Debian-like system this can be done with the command, sudo apt-get install python-numpy.")
+    sys.exit()
 
 uuconv = {'in':90.0, 'pt':1.25, 'px':1, 'mm':3.5433070866, 'cm':35.433070866, 'pc':15.0}
 def unittouu(string):
@@ -43,81 +47,84 @@ def unittouu(string):
 
 class Project(inkex.Effect):
     def __init__(self):
-            inkex.Effect.__init__(self)
+        inkex.Effect.__init__(self)
     def effect(self):
         if len(self.options.ids) < 2:
             inkex.debug("Requires two selected paths. The second must be exactly four nodes long.")
-            exit()            
+            sys.exit()            
             
         #obj is selected second
         obj = self.selected[self.options.ids[0]]
         envelope = self.selected[self.options.ids[1]]
-        if (obj.tagName == 'path' or obj.tagName == 'g') and envelope.tagName == 'path':
-            path = cubicsuperpath.parsePath(envelope.attributes.getNamedItem('d').value)
+        if (obj.tag == inkex.addNS('path','svg') or obj.tag == inkex.addNS('g','svg')) and envelope.tag == inkex.addNS('path','svg'):
+            path = cubicsuperpath.parsePath(envelope.get('d'))
             dp = zeros((4,2), dtype=float64)
-           for i in range(4):
+            for i in range(4):
                 dp[i][0] = path[0][i][1][0]
-               dp[i][1] = path[0][i][1][1]
+                dp[i][1] = path[0][i][1][1]
 
             #query inkscape about the bounding box of obj
             q = {'x':0,'y':0,'width':0,'height':0}
             file = self.args[-1]
             id = self.options.ids[0]
             for query in q.keys():
-                f = os.popen("inkscape --query-%s --query-id=%s %s" % (query,id,file))
+                _,f,err = os.popen3('inkscape --query-%s --query-id=%s "%s"' % (query,id,file))
                 q[query] = float(f.read())
                 f.close()
+                err.close()
             sp = array([[q['x'], q['y']+q['height']],[q['x'], q['y']],[q['x']+q['width'], q['y']],[q['x']+q['width'], q['y']+q['height']]], dtype=float64)
 
-           solmatrix = zeros((8,8), dtype=float64)
-           free_term = zeros((8), dtype=float64)
-           for i in (0,1,2,3):
-               solmatrix[i][0] = sp[i][0]
-               solmatrix[i][1] = sp[i][1]
-               solmatrix[i][2] = 1
-               solmatrix[i][6] = -dp[i][0]*sp[i][0]
-               solmatrix[i][7] = -dp[i][0]*sp[i][1]
-               solmatrix[i+4][3] = sp[i][0]
-               solmatrix[i+4][4] = sp[i][1]
-               solmatrix[i+4][5] = 1
-               solmatrix[i+4][6] = -dp[i][1]*sp[i][0]
-               solmatrix[i+4][7] = -dp[i][1]*sp[i][1]
-               free_term[i] = dp[i][0]
-               free_term[i+4] = dp[i][1]
-
-           res = solve(solmatrix, free_term)
-           projmatrix = array([[res[0],res[1],res[2]],[res[3],res[4],res[5]],[res[6],res[7],1.0]],dtype=float64)
-           if obj.tagName == "path":
-               self.process_path(obj,projmatrix)
-           if obj.tagName == "g":
-               self.process_group(obj,projmatrix)
+        solmatrix = zeros((8,8), dtype=float64)
+        free_term = zeros((8), dtype=float64)
+        for i in (0,1,2,3):
+            solmatrix[i][0] = sp[i][0]
+            solmatrix[i][1] = sp[i][1]
+            solmatrix[i][2] = 1
+            solmatrix[i][6] = -dp[i][0]*sp[i][0]
+            solmatrix[i][7] = -dp[i][0]*sp[i][1]
+            solmatrix[i+4][3] = sp[i][0]
+            solmatrix[i+4][4] = sp[i][1]
+            solmatrix[i+4][5] = 1
+            solmatrix[i+4][6] = -dp[i][1]*sp[i][0]
+            solmatrix[i+4][7] = -dp[i][1]*sp[i][1]
+            free_term[i] = dp[i][0]
+            free_term[i+4] = dp[i][1]
+
+        res = solve(solmatrix, free_term)
+        projmatrix = array([[res[0],res[1],res[2]],[res[3],res[4],res[5]],[res[6],res[7],1.0]],dtype=float64)
+        if obj.tag == inkex.addNS("path",'svg'):
+            self.process_path(obj,projmatrix)
+        if obj.tag == inkex.addNS("g",'svg'):
+            self.process_group(obj,projmatrix)
 
 
     def process_group(self,group,m):
-        for node in group.childNodes:
-            if node.nodeType==node.ELEMENT_NODE:
-                if node.tagName == 'path':
-                     self.process_path(node,m)
-                if node.tagName == 'g':
-                     self.process_group(node,m)        
-       
+        for node in group:
+            if node.tag == inkex.addNS('path','svg'):
+                self.process_path(node,m)
+            if node.tag == inkex.addNS('g','svg'):
+                self.process_group(node,m)    
+
 
     def process_path(self,path,m):
-        d = path.attributes.getNamedItem('d')
-        p = cubicsuperpath.parsePath(d.value)
+        d = path.get('d')
+        p = cubicsuperpath.parsePath(d)
         for subs in p:
             for csp in subs:
                 csp[0] = self.project_point(csp[0],m)
                 csp[1] = self.project_point(csp[1],m)
                 csp[2] = self.project_point(csp[2],m)
-        d.value = cubicsuperpath.formatPath(p)
+        path.set('d',cubicsuperpath.formatPath(p))
 
 
 
     def project_point(self,p,m):
-       x = p[0]
-       y = p[1]
-       return [(x*m[0][0] + y*m[0][1] + m[0][2])/(x*m[2][0]+y*m[2][1]+m[2][2]),(x*m[1][0] + y*m[1][1] + m[1][2])/(x*m[2][0]+y*m[2][1]+m[2][2])]
+        x = p[0]
+        y = p[1]
+        return [(x*m[0][0] + y*m[0][1] + m[0][2])/(x*m[2][0]+y*m[2][1]+m[2][2]),(x*m[1][0] + y*m[1][1] + m[1][2])/(x*m[2][0]+y*m[2][1]+m[2][2])]
 
 e = Project()
 e.affect()
+
+
+# vim: expandtab shiftwidth=4 tabstop=8 softtabstop=4 encoding=utf-8 textwidth=99