#!/usr/bin/env python3 ''' Copyright (C) 2015 Paco Garcia, www.arakne.es 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 os, sys, tempfile, math, inkex from lxml import etree def info(s, newLine="\n"): sys.stderr.write(s) #sys.stderr.write(s.encode("UTF-8")) sys.stderr.write(newLine) def tern(condition,val1,val2): return val1 if condition else val2 def _rads(n): return math.radians(n) def pow2(n): return math.pow(n, 2) # calcula la hipotenusa dados los catetos def triHipo(catA, catB): return math.sqrt(pow2(catA) + pow2(catB)) # calcula el cateto dada la hipotenusa y el otro cateto def triCat(Hipo, catA): return math.sqrt(pow2(Hipo) - pow2(catA)) class XY: """A class for work with 2d points""" def __init__(self, *args, **kwargs): self.co=[0.0,0.0] lArgs=len(args) if lArgs>0: if lArgs==1: if type(args[0])==XY: self.co=args[0].co else: self.co=[args[0],args[0]] if lArgs>1: self.co=[args[0],args[1]] def __add__(self,xy): self.co = [self.co[0] + xy.co[0],self.co[1] + xy.co[1]] return self def __sub__(self,xy): self.co=[self.co[0] - xy.co[0], self.co[1] - xy.co[1]] return self def __eq__(self, xy): return (self.co[0] == xy.x and self.co[1] == xy.y) def sub(self,xy): self.__sub__(xy) return self def mul(self,xy): if type(xy)==XY: co=[self.co[0] * xy.co[0],self.co[1] * xy.co[1]] else: co=[self.co[0] * xy,self.co[1] * xy] self.co = co return self def div(self,xy): if type(xy)==XY: co=[self.co[0] / xy.co[0], self.co[1] / xy.co[1]] else: co=[self.co[0] / xy, self.co[1] / xy] self.co = co return self def vlength(self): #return math.sqrt((self.co[0]*self.co[0])+(self.co[1]*self.co[1])) return triHipo(self.co[0], self.co[1]) def rot(self,ang): x,y,sa,ca= (self.co[0], self.co[1], math.sin(ang), math.cos(ang)) self.co=[ca * x - sa * y, sa * x + ca * y] return self def Rot(self,p,r): self.co=[math.cos(r)*p, math.sin(r)*p] return self def rotate(self,rot,cX=0.0,cY=0.0): cosRot = math.cos(rot) px = cX + (self.x-cX) * cosRot - (self.y-cY) * math.sin(rot) py = cY + (self.x-cX) * math.sin(rot) + (self.y-cY) * cosRot self.co = [px,py] return self def rotateD(self,rot,cX=0.0,cY=0.0): self.rotate(_rads(rot),cX,cY) return self def VDist(self,V2): tmp = XY(self.co[0],self.co[1]) tmp = tmp.sub(V2) return tmp.vlength() def st(self): return str(self.co[0])+','+str(self.co[1]) @property def x(self): return self.co[0] @property def sx(self): return str(self.co[0]) @property def y(self): return self.co[1] @property def sy(self): return str(self.co[1]) def hipo(self,xy): #return math.sqrt(math.pow(self.x-xy.x,2) + math.pow(self.y-xy.y,2) ) return triHipo(self.x-xy.x, self.y-xy.y) def angBetween2Lines(self,p1,p2): # pC punto comun return math.atan2(self.y - p1.y, self.x - p1.x) - math.atan2(self.y - p2.y, self.x - p2.x) def getAngle(self,b): return math.atan2(b.y - self.y, b.x - self.x) def getAngleD(self,b): return math.degrees(math.atan2(b.y - self.y, b.x - self.x)) # translada un punto hacia otro def atPercent(self, p2, percent): self.co = [(p2.x - self.x) * percent + self.x,(p2.y-self.y) * percent + self.y] return self def atMid(self, p2): return self.atPercent(p2,0.5) # ________________________________________________________________ # ________________________________________________________________ # ________________________________________________________________ class bezpnt(object): def __init__(self,pfixed=None,pprev=None,pnext=None): if isinstance(pfixed, list): self.fixed = XY(pfixed[0],pfixed[1]) else: self.fixed = pfixed if isinstance(pprev, list): self.prev = XY(pprev[0],pprev[1]) else: self.prev = pprev if isinstance(pnext, list): self.next = XY(pnext[0],pnext[1]) else: self.next = pnext return def translate(self,x,y): self.fixed + XY(x,y) if self.prev!=None:self.prev + XY(x,y) if self.next!=None:self.next + XY(x,y) return self def scale(self,x=1.0,y=1.0): self.fixed.scale(x,y) if self.prev!=None:self.prev.scale(x,y) if self.next!=None:self.next.scale(x,y) return self def rotate(self,rot,cX=0.0,cY=0.0): self.fixed.rotate(rot,cX,cY) if self.prev!=None:self.prev.rotate(rot,cX,cY) if self.next!=None:self.next.rotate(rot,cX,cY) return sel def skew(self,rotx,roty,cX=0.0,cY=0.0): self.fixed.skew(rotx,roty,cX,cY) if self.prev!=None:self.prev.skew(rotx,roty,cX,cY) if self.next!=None:self.next.skew(rotx,roty,cX,cY) return self def copy(self,bez2): try: self.fixed=XY().copy(bez2.fixed) self.prev = None if bez2.prev == None else XY().copy(bez2.prev) self.next = None if bez2.next == None else XY().copy(bez2.next) except Exception: gimp.message(str(Exception)) return self def arrXY(self): pts=[] if self.prev == None: pts+=self.fixed.arrXY(1) else: pts+=self.prev.arrXY(1) pts+=self.fixed.arrXY(1) if self.next==None: pts+=self.fixed.arrXY(1) else: pts+=self.next.arrXY(1) return pts def Prev(self): p = self.prev if p==None: p=self.fixed return p def Next(self): p = self.next if p==None: p=self.fixed return p def Fixed(self): return self.fixed def flip(self): p=self.prev n=self.next self.prev=n self.next=p def createSmallArcBez(r, a1, a2,rot): a = (a2 - a1) * 0.5 p4 = XY(r * math.cos(a), r * math.sin(a)) p1 = XY(p4.x, -p4.y) k = 0.5522847498 f = k * math.tan(a) p2 = XY(p1.x + f * p4.y, p1.y + f * p4.x) p3 = XY(p2.x,-p2.y) ar = a + a1 P1 = XY(r * math.cos(a1), r * math.sin(a1)).rotate(rot) P2 = XY(p2.x, p2.y).rotate(ar).rotate(rot) P3 = XY(p3.x, p3.y).rotate(ar).rotate(rot) P4 = XY(r * math.cos(a2),r * math.sin(a2)).rotate(rot) B1=bezpnt(P1,None,P2) B2=bezpnt(P4,P3) return [B1,B2] def createArcBez(rad, sAng, eAng): EPSILON = 0.0000000001 bezs =[] if eAng < sAng: eAng += 360.0 sAng = _rads(sAng) eAng = _rads(eAng) rot = sAng sAng = _rads(0) eAng = eAng - rot pi2 = math.pi * 2 sAng, eAng = (sAng % pi2, eAng % pi2) pi_2 = math.pi * 0.5 sign = 1 if (sAng < eAng) else -1 a1 = sAng totAng = min(math.pi * 2, abs(eAng - sAng)) while (totAng > EPSILON): a2 = a1 + sign * min(totAng, pi_2) bezs.extend(createSmallArcBez(rad, a1, a2,rot)) totAng = totAng - abs(a2 - a1) a1 = a2 return bezs def bezs2XYList(arc1, transform = None): pnts=[] bezs=[] for aa in arc1: if aa.prev is not None: bezs.append(XY(aa.prev)) bezs.append(XY(aa.fixed)) if aa.next is not None: bezs.append(XY(aa.next)) for i in range(len(bezs)): v = bezs[i] if transform: v = v + transform if i == 0: pnts.append(v) else: v2=pnts[-1] if (v2.x != v.x or v2.y != v.y): pnts.append(XY(v)) a=len(pnts) return pnts def XYList(lst, rot = 0.0, add = None): verts=[] for nn in range(len(lst)): v = lst[nn] if rot != 0.0: v = v.rotate(rot) if add: v = v + add verts.append([v.x,v.y]) return verts def XYListSt(lst, rot = 0.0, add = None): D2 = "" for nn in range(len(lst)): v = lst[nn] if rot != 0.0: v = v.rotate(rot) if add: v = v + add D2 += "%s%s " % (tern(nn==1,"C",""), v.st()) return D2 # circle by quadrants, A: 0>90, B: 90>180, C: 180>270, D: 270>360 def circQ(p,r,abcd="ABCD",inverse=0,xtra=None): aa = r * 0.551915024494 parts={ 'A':[XY(0,-r),XY(aa,-r), XY(r, -aa),XY(r,0)], 'B':[XY(r,0), XY(r, aa), XY(aa, r),XY(0,r)], 'C':[XY(0,r), XY(-aa,r), XY(-r, aa),XY(-r,0)], 'D':[XY(-r,0),XY(-r,-aa),XY(-aa,-r),XY(0,-r)]} pA = [XY(p)+N for N in parts[abcd[0]]] for aa in abcd[1:]: pA = pA + [XY(p)+N for N in parts[aa][1:]] if inverse==1: pA.reverse() listA = XYList(pA) if xtra: for n in xtra: listA[n].extend(xtra[n]) return listA def circleInCircle(c1,r1,c2,r2): return tern((r1 > (c1.hipo(c2) + r2)),True,False) def polar2cartesian(cX, cY, rad, ang): return XY(cX + (rad * math.cos(ang)), cY + (rad * math.sin(ang))) def setArc(x, y, rad, ang1, ang2, first): start = polar2cartesian(x, y, rad, ang2) end = polar2cartesian(x, y, rad, ang1) arcSweep = "0" if (ang2 - ang1 <= 180) else "1" d = " A%f,%f 0 %s 0 %s" % (rad, rad, arcSweep, end.st()) if first == 1: d = "M" + start.st() + " " + d return d def setArcs(x, y, rad, ang1, ang2, first): if ang2 < ang1: m = setArc(x, y, rad, 0, _rads(ang2), first) m = m +" "+ setArc(x, y, rad, _rads(ang1), _rads(360), 0) else: m = setArc(x, y, rad, _rads(ang1), _rads(ang2), first) return m def addChild(padre, type, props): hijo = etree.SubElement(padre, inkex.addNS(type,'svg')) for n in props: hijo.set(n,props[n]) return hijo def svgCircle(padre, r, cx, cy): return addChild(padre,'circle' ,{'r':str(r), 'cx': str(cx), 'cy': str(cy)}) def circleInscribedInTri(p1,p2,p3): d1, d2, d3 = [n.vlength() for n in [XY(p3)-p2, XY(p1)-p3, XY(p2)-p1]] p = d1 + d2 + d3 centro = XY( (p1.x*d1 + p2.x*d2 + p3.x*d3) / p, (p1.y*d1 + p2.y*d2 + p3.y*d3) / p) p = p / 2.0 radius = math.sqrt(p * (p - d1) * (p - d2) * (p - d3))/p return (radius, centro.x, centro.y) def TriInscribedInCircle(p1,p2,p3): x12, x13, x31, x21 = (p1.x - p2.x, p1.x - p3.x, p3.x - p1.x, p2.x - p1.x) y12, y13, y31, y21 = (p1.y - p2.y, p1.y - p3.y, p3.y - p1.y, p2.y - p1.y) sx13 = pow2(p1.x) - pow2(p3.x) sy13 = pow2(p1.y) - pow2(p3.y) sx21 = pow2(p2.x) - pow2(p1.x) sy21 = pow2(p2.y) - pow2(p1.y) f = (sx13*x12 + sy13*x12 + sx21*x13 + sy21*x13) / (2 * (y31*x12 - y21*x13)) g = (sx13*y12 + sy13*y12 + sx21*y13 + sy21*y13) / (2 * (x31*y12 - x21*y13)) c = - pow2(p1.x) - pow2(p1.y) - 2*g*p1.x - 2*f*p1.y r = math.sqrt(pow2(-g) + pow2(-f) - c) return (r, -g, -f) # 243