mightyscape-1.1-deprecated/extensions/fablabchemnitz/pathselection.py

#!/usr/bin/env python3
'''
pathselection.py

Sunabe kazumichi 2009/9/29
http://dp48069596.lolipop.jp/

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
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

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 inkex, locale
import math
from lxml import etree
from inkex import paths
from inkex import bezier

# Set current system locale
locale.setlocale(locale.LC_ALL, '')

def cspseglength(sp1,sp2, tolerance = 0.001):
    bez = (sp1[1][:],sp1[2][:],sp2[0][:],sp2[1][:])    
    return bezier.bezierlength(bez, tolerance)    
def csplength(csp):
    lengths = []
    for sp in csp:
        lengths.append([])
        for i in range(1,len(sp)):
            l = cspseglength(sp[i-1],sp[i])
            lengths[-1].append(l)      
    return lengths

def roughBBox(path):
    xmin,xMax,ymin,yMax=path[0][0][0],path[0][0][0],path[0][0][1],path[0][0][1]
    for pathcomp in path:
        for pt in pathcomp:
               xmin=min(xmin,pt[0])
               xMax=max(xMax,pt[0])
               ymin=min(ymin,pt[1])
               yMax=max(yMax,pt[1])
               if xMax-xmin==0:
                   tn=0
               else :
                   tn=(yMax-ymin)/(xMax-xmin)
    return tn

class Length(inkex.Effect):
    def __init__(self):
        inkex.Effect.__init__(self)
        self.arg_parser.add_argument("-s", "--selection", default=True, help="select path with length or slant")
        self.arg_parser.add_argument("-1", "--len1", type=float, default=12)
        self.arg_parser.add_argument("-2", "--len2", type=float, default=25)
        self.arg_parser.add_argument("-3", "--len3", type=float, default=40)
        self.arg_parser.add_argument("-4", "--len4", type=float, default=60)
        self.arg_parser.add_argument("-5", "--len5", type=float, default=60)
        self.arg_parser.add_argument("-6", "--hor", type=float, default=0.2)
        self.arg_parser.add_argument("-7", "--ver", type=float, default=10)
  
    def effect(self):
        # loop over all selected paths
        if self.options.selection=="Path_lengthselection":
            for id, node in self.svg.selected.items():
                if node.tag == inkex.addNS('path','svg'):  
                    l1,l2,l3,l4,l5=[],[],[],[],[]
                    p = paths.CubicSuperPath(inkex.paths.Path(node.get('d')))
                    slengths= csplength(p)
                    b=[slengths, p]
                    
                    # path length select
                    for x in range(0,len(slengths)):
                        if sum(b[0][x])<self.options.len1:
                            l1.append(b[1][x])
                        if self.options.len2>sum(b[0][x])>=self.options.len1 :
                            l2.append(b[1][x])
                        if self.options.len3>sum(b[0][x])>=self.options.len2 :
                            l3.append(b[1][x])
                        if self.options.len4>sum(b[0][x])>=self.options.len3 :
                            l4.append(b[1][x])                        
                        if sum(b[0][x])>=self.options.len4 :
                            l5.append(b[1][x])

                # make path
                lensel=[l1,l2,l3,l4,l5]
                strlen=['#FF0001','#00FF02','#AAFF03','#87CEE4','#000FF5']                        
                for i,x in zip(strlen,lensel):
                        s = {'stroke-linejoin': 'miter', 'stroke-width': '0.5px', 
                            'stroke-opacity': '1.0', 'fill-opacity': '1.0', 
                            'stroke': i, 'stroke-linecap': 'butt', 'fill': 'none'}
                        attribs={'style':str(inkex.Style(s)),'d':str(paths.Path(paths.CubicSuperPath(x).to_path().to_arrays()))}
                        etree.SubElement(node.getparent(),inkex.addNS('path','svg'),attribs)

        if self.options.selection=="Path_slantselection":
            for id, node in self.svg.selected.items():
                if node.tag == inkex.addNS('path','svg'):  
                    hor1,ver2,slan3=[],[],[]
                    p = paths.CubicSuperPath(inkex.paths.Path(node.get('d')))

                    # path slant select
                    for i,x in enumerate(p):
                        tn=roughBBox(x)
                        if tn<self.options.hor:
                            hor1.append(p[i])
                        elif tn>self.options.ver:
                            ver2.append(p[i])
                        else:
                            slan3.append(p[i])

                # make path
                slnsel=[hor1,ver2,slan3]
                strsln=['#FF0001','#00FF02','#000FF5']                        
                for i,x in zip(strsln,slnsel):
                        s = {'stroke-linejoin': 'miter', 'stroke-width': '0.5px', 
                            'stroke-opacity': '1.0', 'fill-opacity': '1.0', 
                            'stroke': i, 'stroke-linecap': 'butt', 'fill': 'none'}
                        attribs={'style':str(inkex.Style(s)),'d':str(paths.Path(paths.CubicSuperPath(x).to_path().to_arrays()))}
                        etree.SubElement(node.getparent(),inkex.addNS('path','svg'),attribs)

if __name__ == '__main__':
    Length().run()