share/extensions/kochify.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 math, tempfile, copy, cPickle, inkex, simplepath
  20
  21 def findend(d):
  22     end = []
  23     subPathStart = []
  24     for cmd,params in d:
  25         if cmd == 'M':
  26             subPathStart = params[-2:]
  27         if cmd == 'Z':
  28             end = subPathStart[:]
  29         else:
  30             end = params[-2:]
  31     return end
  32
  33 class Kochify(inkex.Effect):
  34     def effect(self):
  35         try:
  36             f = open(tempfile.gettempdir() + '/kochify.bin', 'r')
  37             proto = cPickle.load(f)
  38             f.close()
  39         except:
  40             return False
  41         for id, node in self.selected.iteritems():
  42             if node.tagName == 'path':
  43                 d = node.attributes.getNamedItem('d')
  44                 p = simplepath.parsePath(d.value)
  45                 last = p[0][1][-2:]
  46                 subPathStart = []
  47                 cur = []
  48                 new = []
  49                 for i in range(len(p)):
  50                     rep = copy.deepcopy(proto['path'])
  51                     cmd, params = p[i]
  52                     if cmd == 'M':
  53                         subPathStart = params[-2:]
  54                         new.append(copy.deepcopy(p[i]))
  55                         continue
  56                     if cmd == 'Z':
  57                         cur = subPathStart[:]
  58                     else:
  59                         cur = params[-2:]
  60
  61                     if last == cur:
  62                         continue
  63
  64                     dx = cur[0]-last[0]
  65                     dy = cur[1]-last[1]
  66                     length = math.sqrt((dx**2) + (dy**2))
  67                     angle = math.atan2(dy,dx)
  68
  69                     scale = length / proto['length']
  70                     rotate = angle - proto['angle']
  71                     simplepath.scalePath(rep,scale,scale)
  72                     simplepath.rotatePath(rep, rotate)
  73                     repend = findend(rep)
  74                     transx = cur[0] - repend[0]
  75                     transy = cur[1] - repend[1]
  76                     simplepath.translatePath(rep, transx, transy)
  77
  78                     if proto['endsinz']:
  79                         new.extend(rep[:])
  80                     else:
  81                         new.extend(rep[1:])
  82                     last = cur[:]
  83
  84                 d.value = simplepath.formatPath(new)
  85 e = Kochify()
  86 e.affect()