#!/usr/bin/env python3 ''' Copyright (C) 2010 Jonathan Manton (jmanton @ illinois.edu) Copyright (C) 2007 Aaron Spike (aaron @ ekips.org) Copyright (C) 2007 Tavmjong Bah (tavmjong @ free.fr) 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 ''' # this program is loosely based on the gear extension for inkscape import inkex from polyhedrondata import angleOverride, polyhedronData from math import * from lxml import etree def points_to_svgd(p): f = p[0] p = p[1:] svgd = 'M%.3f,%.3f' % f for x in p: svgd += 'L%.3f,%.3f' % x return svgd class Polyhedra(inkex.EffectExtension): no_tab = { 'cut' : ['m 0,0 100,0'], 'perf' : []} def add_arguments(self, pars): pars.add_argument("-p", "--poly", default="Cube", help="Polygon net to render") pars.add_argument("-s", "--size", type=float, default=100.0, help="Size of first edge") pars.add_argument("-u", "--unit", default="mm", help="Units") pars.add_argument("-m", "--material_thickness", type=float, default=1.0, help="Material thickness") pars.add_argument("-t", "--tabs", type=int, default=0, help="Tab style") def get_tab(self, limitAngle): return(self.get_connector('tab', limitAngle)) def get_slot(self, limitAngle): return(self.get_connector('slot', limitAngle)) def get_slot_tab_style(self): # these are designed to be 100px wide - they have to be scaled notab = {0 : no_tab} simpletab = {0: {'cut': ['m 0,0 20,19 60,0 l 20,-19'], 'perf' : ['m 0,0 100,0']}} # tab array - 0 = slot/tab, 1 = two tabs, 2 = single tab (alternate tab and nothing), 3 = none mt = self.options.material_thickness tabStyle = [{ \ 50 : {'cut': ['m 0,0 35,0 0,{} -6,0 c 0,0 1,8 6,12 8,6 22,6 30,0 5,-4 6,-12 6,-12 l -6,0 0,{} 35,0'.format(mt,-mt)], 'perf' : ['M 42,0 58,0']}, \ 24 : {'cut': ['m 0,0 55,0 0,{} -6,0 c 0,0 1,3 6,6 8,3 19,6 27,8 5,-4 9,-14 9,-14 l -6,0 0,{} 15,0'.format(mt,-mt)], 'perf' : ['M 62,0 78,0']}, \ 18 : {'cut': ['m 0,0 65,0 0,{} -6,0 c 0,0 1,3 6,6 8,2 11,2 19,3 5,-4 9,-10 7,-9 l -6,0 0,{} 15,0'.format(mt,-mt)], 'perf' : ['m 72,0 6,0']}, \ 6 : {'cut': ['m 0,0 70,0 {},7 10,1 5,-6 -2,0 0,{} 15,0'.format(mt,-mt)], 'perf' : ['m 74,0 7,0']}, 0 : {'cut': ['m 0,0 100,0'], 'perf' : []}}, simpletab, simpletab, notab] slotStyle = [{\ 50 : {'cut': ['m 0,0 20,19 60,0 l 20,-19', 'm 28,{} 4,{} 36,0 4,{}'.format(-mt, mt, -(mt))], 'perf' : ['M 0,0 28,0','M 100,0 72,0']}, \ 24 : {'cut': ['M 100,0 90,18 30,11 20,0 0,0', 'm 92,{} -4,{} -36,0 -4,{}'.format(-mt, mt, -(mt))], 'perf' : ['M 100,0 92,0', 'M 20,0 48,0']}, \ 18 : {'cut': ['M 100,0 90,16 40,9 35,0 0,0', 'm 92,{} -4,{} -26,0 -4,{}'.format(-mt, mt, -(mt))], 'perf' : ['M 100,0 92,0', 'M 35,0 58,0']}, \ 6 : {'cut': ['M 100,0 98,10 88,9 84,2 86,2 86,0 0,0'], 'perf' : ['m 96,0 -6,0']}, 0 : {'cut': ['m 0,0 100,0'], 'perf' : [] }}, simpletab, notab, notab] return [tabStyle, slotStyle] def get_connector(self, type, limitAngle): if(self.options.tabs == 1): # two tabs return(self.gen_tab(limitAngle/2)) if(self.options.tabs == 2): # one tab if(type == 'tab'): return(self.gen_tab(limitAngle)) else: return(self.no_tab) if(self.options.tabs == 3): # no tabs or slots return(self.no_tab) # otherwise, get stuff from the array of specially modified tab/slots if(type == 'tab'): source = self.get_slot_tab_style()[0] else: source = self.get_slot_tab_style()[1] cuttable = source[self.options.tabs].keys() sorted(cuttable) # sorts in-place. Ugh. reversed(cuttable) # in-place. Ugh. for angle in cuttable: if(limitAngle >= angle): return(source[self.options.tabs][angle]) def gen_tab(self, theta_degrees): theta = radians(min(theta_degrees,90)) minOffset = 8 tabAngle = radians(40) minTabLength = 40 tabHeight = 19 # must be >= minOffset tabWidth = 100 # determine if we need to sqash down side of tab by limit angle bX = minOffset * cos(theta) / sin(theta) aX = bX - (minOffset * cos(tabAngle) / sin(tabAngle)) if(theta >= tabAngle): tab_cut = [(0,0), (tabHeight/sin(tabAngle) * cos(tabAngle), tabHeight)] tab_perf = [(0,0), (tabWidth,0)] else: if (aX > (tabWidth - minTabLength)): # too stubby, don't put a tab return(no_tab) tab_cut = [(0,0), (aX,0), (bX, minOffset)] tab_perf = [(aX,0), (tabWidth,0)] # now see if we also have to squash the other side of the tab - this happens # for very small angles where the tab height on the far side would still be # too high and would intersect with the other tab # first, find where the tab would intersect the limit angle if we just put # out a line of infinite length at the tab angle. bmult = tabWidth / (cos(theta) * sin(tabAngle) / sin(theta) + cos(tabAngle)) tiY = bmult * sin(tabAngle) # now see if that intersection is too low. If so, limit the tab's height tiY = min(tiY, tabHeight) tab_cut.append((tabWidth - tiY / sin(tabAngle) * cos(tabAngle), tiY)) tab_cut.append((tabWidth, 0)) return({'cut' : [points_to_svgd(tab_cut)], 'perf' : [points_to_svgd(tab_perf)]}) def effect(self): poly = self.options.poly size = self.svg.unittouu(str(self.options.size) + self.options.unit) eC = polyhedronData[poly]['edgeCoordinates'] iEI = polyhedronData[poly]['insideEdgeIndices'] oEI = polyhedronData[poly]['outsideEdgeIndices'] oED = polyhedronData[poly]['outsideEdgeDegrees'] sidelen = sqrt((eC[oEI[0][0]-1][0] - eC[oEI[0][1]-1][0])**2 + (eC[oEI[0][0]-1][1] - eC[oEI[0][1]-1][1])**2) scale = size / sidelen # Translate group, Rotate path. t = 'translate(' + str( self.svg.namedview.center[0] ) + ',' + str( self.svg.namedview.center[1] ) + ')' g_attribs = {inkex.addNS('label','inkscape'):'Polygon ' + str( poly ), 'transform':t } g = etree.SubElement(self.svg.get_current_layer(), 'g', g_attribs) gsub_attribs = {inkex.addNS('label','inkscape'):'Polygon ' + str( poly ) + 'border' } gsub = etree.SubElement(g, 'g', gsub_attribs) # Create SVG Path cutStyle = { 'stroke': '#0000FF', 'stroke-width': self.svg.unittouu("1px"), 'fill': 'none' } perfStyle = { 'stroke': '#FF0000', 'stroke-width': self.svg.unittouu("1px"), 'fill': 'none' } textStyle = { 'font-size': str( size/4 ), 'font-family': 'arial', 'text-anchor': 'middle', 'text-align': 'center', 'fill': '#222' } numOEI = len(oEI) for edgeIndex in range(numOEI): eX1 = eC[oEI[edgeIndex][0]-1][0] eY1 = eC[oEI[edgeIndex][0]-1][1] eX2 = eC[oEI[edgeIndex][1]-1][0] eY2 = eC[oEI[edgeIndex][1]-1][1] origin = (eX1 * scale, eY1 * scale) edgelen = sqrt((eX1 - eX2)**2 + (eY1 - eY2)**2) edgesize = size * (edgelen / sidelen) prevAngle = (oED[(edgeIndex - 1) % numOEI] + 180 - oED[edgeIndex]) % 360 nextAngle = (oED[edgeIndex] + 180 - oED[(edgeIndex + 1) % numOEI]) % 360 if str(angleOverride) in poly: if angleOverride[poly].has_key('prev'): if angleOverride[poly]['prev'].has_key(edgeIndex): prevAngle = angleOverride[poly]['prev'][edgeIndex] if angleOverride[poly].has_key('next'): if angleOverride[poly]['next'].has_key(edgeIndex): nextAngle = angleOverride[poly]['next'][edgeIndex] trans = 'translate(' + str(origin[0]) + ',' + str(origin[1]) + ') rotate(' + str(oED[edgeIndex] ) + ') scale(' + str(edgesize/100.0) + ')' limitAngle = min(prevAngle, nextAngle) tab = self.get_tab(limitAngle) slot = self.get_slot(limitAngle) # the "special" tabs are all skewed one way or the other, to # make room for tight turns. This determines if it is the previous # or next angle that is the tight one, and flips the tab or slot if # needed. if(nextAngle == limitAngle): trans += ' translate(100.0, 0) scale(-1,1)' tab_group_attribs = {inkex.addNS('label','inkscape'):'tab', 'transform': trans } tab_group = etree.SubElement(gsub, 'g', tab_group_attribs) midX = (eX2 - eX1)/2 + eX1 midY = (eY2 - eY1)/2 + eY1 text_attrib = {'style': str(inkex.Style(textStyle)), 'x' : str(midX * scale), 'y' : str(midY * scale) } # etree.SubElement(gsub, 'text', text_attrib).text = str(edgeIndex) if ((edgeIndex % 2) == 0): edgetype = tab else: edgetype = slot for path in edgetype['cut']: gear_attribs = {'style':str(inkex.Style(cutStyle)), 'd':path} gear = etree.SubElement(tab_group, inkex.addNS('path','svg'), gear_attribs ) for path in edgetype['perf']: gear_attribs = {'style':str(inkex.Style(perfStyle)), 'd':path} gear = etree.SubElement(tab_group, inkex.addNS('path','svg'), gear_attribs ) gsub_attribs = {inkex.addNS('label','inkscape'):'Polygon ' + str( poly ) + 'inside' } gsub = etree.SubElement(g, 'g', gsub_attribs) for edge in iEI: points = [(eC[edge[0]-1][0] * scale, eC[edge[0]-1][1] * scale), (eC[edge[1]-1][0] * scale, eC[edge[1]-1][1] * scale)] path = points_to_svgd( points ) perf_attribs = {'style':str(inkex.Style(perfStyle)), 'd':path} gear = etree.SubElement(gsub, inkex.addNS('path','svg'), perf_attribs ) if __name__ == '__main__': Polyhedra().run()