share/extensions/addnodes.py

   1 #!/usr/bin/env python
   2 '''
   3 Copyright (C) 2005 Aaron Spike, aaron@ekips.org
   4
   5 This program is free software; you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation; either version 2 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program; if not, write to the Free Software
  17 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18 '''
  19 import inkex, cubicsuperpath, simplestyle, copy, math, re, bezmisc
  20
  21 def numsegs(csp):
  22     return sum([len(p)-1 for p in csp])
  23 def tpoint((x1,y1), (x2,y2), t = 0.5):
  24     return [x1+t*(x2-x1),y1+t*(y2-y1)]
  25 def cspbezsplit(sp1, sp2, t = 0.5):
  26     m1=tpoint(sp1[1],sp1[2],t)
  27     m2=tpoint(sp1[2],sp2[0],t)
  28     m3=tpoint(sp2[0],sp2[1],t)
  29     m4=tpoint(m1,m2,t)
  30     m5=tpoint(m2,m3,t)
  31     m=tpoint(m4,m5,t)
  32     return [[sp1[0][:],sp1[1][:],m1], [m4,m,m5], [m3,sp2[1][:],sp2[2][:]]]
  33 def cspbezsplitatlength(sp1, sp2, l = 0.5, tolerance = 0.001):
  34     bez = (sp1[1][:],sp1[2][:],sp2[0][:],sp2[1][:])
  35     t = bezmisc.beziertatlength(bez, l, tolerance)
  36     return cspbezsplit(sp1, sp2, t)
  37 def cspseglength(sp1,sp2, tolerance = 0.001):
  38     bez = (sp1[1][:],sp1[2][:],sp2[0][:],sp2[1][:])
  39     return bezmisc.bezierlength(bez, tolerance)
  40 def csplength(csp):
  41     total = 0
  42     lengths = []
  43     for sp in csp:
  44         lengths.append([])
  45         for i in xrange(1,len(sp)):
  46             l = cspseglength(sp[i-1],sp[i])
  47             lengths[-1].append(l)
  48             total += l
  49     return lengths, total
  50 def numlengths(csplen):
  51     retval = 0
  52     for sp in csplen:
  53         for l in sp:
  54             if l > 0:
  55                 retval += 1
  56     return retval
  57
  58 class SplitIt(inkex.Effect):
  59     def __init__(self):
  60         inkex.Effect.__init__(self)
  61         self.OptionParser.add_option("-m", "--max",
  62                         action="store", type="float",
  63                         dest="max", default=0.0,
  64                         help="maximum segment length")
  65     def effect(self):
  66         for id, node in self.selected.iteritems():
  67             if node.tagName == 'path':
  68                 d = node.attributes.getNamedItem('d')
  69                 p = cubicsuperpath.parsePath(d.value)
  70
  71                 #lens, total = csplength(p)
  72                 #avg = total/numlengths(lens)
  73                 #inkex.debug("average segment length: %s" % avg)
  74
  75                 new = []
  76                 for sub in p:
  77                     new.append([sub[0][:]])
  78                     i = 1
  79                     while i <= len(sub)-1:
  80                         length = cspseglength(new[-1][-1], sub[i])
  81                         if length > self.options.max:
  82                             splits = math.ceil(length/self.options.max)
  83                             for s in xrange(int(splits),1,-1):
  84                                 new[-1][-1], next, sub[i] = cspbezsplitatlength(new[-1][-1], sub[i], 1.0/s)
  85                                 new[-1].append(next[:])
  86                         new[-1].append(sub[i])
  87                         i+=1
  88
  89                 d.value = cubicsuperpath.formatPath(new)
  90
  91 e = SplitIt()
  92 e.affect()