138 lines
6.0 KiB
Python
138 lines
6.0 KiB
Python
#!/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 ColorizePathLengths(inkex.EffectExtension):
|
|
|
|
def add_arguments(self, pars):
|
|
pars.add_argument("-s", "--selection", default=True, help="select path with length or slant")
|
|
pars.add_argument("-1", "--len1", type=float, default=12)
|
|
pars.add_argument("-2", "--len2", type=float, default=25)
|
|
pars.add_argument("-3", "--len3", type=float, default=40)
|
|
pars.add_argument("-4", "--len4", type=float, default=60)
|
|
pars.add_argument("-5", "--len5", type=float, default=60)
|
|
pars.add_argument("-6", "--hor", type=float, default=0.2)
|
|
pars.add_argument("-7", "--ver", type=float, default=10)
|
|
|
|
def effect(self):
|
|
# loop over all selected paths
|
|
if self.options.selection=="path_lengthselection":
|
|
if len(self.svg.selected) > 0:
|
|
for node in self.svg.selected.values():
|
|
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)
|
|
else:
|
|
self.msg('Please select some paths first.')
|
|
return
|
|
|
|
if self.options.selection=="path_slantselection":
|
|
if len(self.svg.selected) > 0:
|
|
for node in self.svg.selected.values():
|
|
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)
|
|
else:
|
|
self.msg('Please select some paths first.')
|
|
return
|
|
|
|
if __name__ == '__main__':
|
|
ColorizePathLengths().run() |