56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
from io import StringIO
|
|
import inkex
|
|
from lxml import etree
|
|
|
|
class SliderElectrodes(inkex.Effect):
|
|
def __init__(self):
|
|
inkex.Effect.__init__(self)
|
|
self.arg_parser.add_argument("-c", "--count", type=int, default=5, help="Number of electrodes")
|
|
self.arg_parser.add_argument("-s", "--spikes", type=int, default=5, help="Number of spikes")
|
|
|
|
def genPathString(self, bounds, spikeWidth, first=False, last=False):
|
|
s = StringIO()
|
|
cx = bounds[0]
|
|
cy = bounds[1]
|
|
stepx = spikeWidth
|
|
stepy = (bounds[3] - bounds[1]) / (2.0 * self.options.spikes)
|
|
s.write(" M %f, %f " % (bounds[0], bounds[1]))
|
|
if first:
|
|
s.write(" L %f, %f " % (bounds[0], bounds[3]))
|
|
else:
|
|
for i in range(self.options.spikes):
|
|
s.write(" L %f, %f " % (bounds[0] + stepx, bounds[1] + (2 * i + 1) * stepy))
|
|
s.write(" L %f, %f " % (bounds[0], bounds[1] + (2 * i + 2) * stepy))
|
|
if last:
|
|
s.write(" L %f, %f " % (bounds[2], bounds[3]))
|
|
s.write(" L %f, %f " % (bounds[2], bounds[1]))
|
|
else:
|
|
s.write(" L %f, %f " % (bounds[2] - stepx, bounds[3]))
|
|
for i in range(self.options.spikes):
|
|
s.write(" L %f, %f " % (bounds[2], bounds[3] - (2 * i + 1) * stepy))
|
|
s.write(" L %f, %f " % (bounds[2] - stepx, bounds[3] - (2 * i + 2) * stepy))
|
|
s.write(" Z ")
|
|
return s.getvalue()
|
|
|
|
def effect(self):
|
|
svg = self.document.getroot()
|
|
width = self.svg.unittouu(self.document.getroot().get('width'))
|
|
height = self.svg.unittouu(self.document.getroot().get('height'))
|
|
|
|
group = etree.SubElement(self.svg.get_current_layer(), 'g', {inkex.addNS('label', 'inkscape') : 'Slider electrodes'})
|
|
|
|
eWidth = width / self.options.count
|
|
spikeWidth = 0.6 * eWidth
|
|
|
|
for eid in range(self.options.count):
|
|
if eid == 0:
|
|
path = self.genPathString((eid * eWidth, 0, (eid + 1) * eWidth + 0.4 * spikeWidth, height), spikeWidth, first=True)
|
|
elif eid == self.options.count - 1:
|
|
path = self.genPathString((eid * eWidth - 0.4 * spikeWidth, 0, (eid + 1) * eWidth, height), spikeWidth, last=True)
|
|
else:
|
|
path = self.genPathString((eid * eWidth - 0.4 * spikeWidth, 0, (eid + 1) * eWidth + 0.4 * spikeWidth, height), spikeWidth)
|
|
e = etree.SubElement(group, inkex.addNS('path', 'svg'), {'style':str(inkex.Style({'stroke':'none','fill' :'#000000'})),'d' : path})
|
|
|
|
if __name__ == '__main__':
|
|
effect = SliderElectrodes().run() |