mightyscape-1.2/extensions/fablabchemnitz/move_path_node/move_path_node.py

184 lines
8.6 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
'''
Author: Mario Voigt / FabLab Chemnitz
Mail: mario.voigt@stadtfabrikanten.org
Date: 19.05.2021
Last patch: 19.05.2021
License: GNU GPL v3
'''
import copy
import inkex
from inkex import Circle, TextElement, PathElement
from inkex.paths import CubicSuperPath, Path
class MovePathNode(inkex.EffectExtension):
def modify(self, element):
raw = element.path.to_arrays()
subpaths, prev = [], 0
for i in range(len(raw)): # Breaks compound paths into simple paths
if raw[i][0] == 'M' and i != 0:
subpaths.append(raw[prev:i])
prev = i
subpaths.append(raw[prev:])
if self.options.debug is True:
if len(subpaths) == 0:
self.msg("{} has no subpaths").format(element.get('id'))
else:
self.msg("{} has {} subpath(s)".format(element.get('id'), len(subpaths)))
subpathNr = 0
for path in subpaths:
subpathNr += 1
newSubpaths = subpaths #we will overwrite them later
pathIsClosed = False
if path[-1][0] == 'Z' or \
(path[-1][0] == 'L' and path[0][1] == path[-1][1]) or \
(path[-1][0] == 'C' and path[0][1] == [path[-1][1][-2], path[-1][1][-1]]) \
: #if first is last point the path is also closed. The "Z" command is not required
pathIsClosed = True
if self.options.debug is True:
self.msg("pathIsClosed = " + str(pathIsClosed))
self.msg("nodes = " + str(len(path)))
if self.options.closed_only is True and pathIsClosed is False:
if len(subpaths) == 0:
self.msg("{}/subpath {} is not closed! Skipping ...".format(element.get('id'), subpathNr))
else:
self.msg("{} is not closed! Skipping ...".format(element.get('id')))
continue #skip this open path
if len(path) == 2:
continue #skip this open path (special case of straight line segment)
if path[-1][0] == 'Z': #replace Z with another L command (which moves to the coordinates of the first M command in path) to have better overview
path[-1][0] = 'L'
path[-1][1] = path[0][1]
#adjust if entered move number is higher than actual node count. We handle as infinite looping
moves = (self.options.movenode) % len(path)
if pathIsClosed is True: #if closed start and end collapse and "duplicate"
moves = (self.options.movenode) % (len(path) - 1)
if self.options.movenode == 0: #special handling for 0 is required
moves = 0
if self.options.debug is True:
self.msg("moves to perform = " + str(moves))
self.msg("root path:")
self.msg(path)
self.msg("-"*25)
for i in range(moves):
if len(path) > 2: #the path needs at least more than two segments, else we might just get a "pointy path" on an open path
#we move the first segment to the end of the list
move = path[0]
del path[0]
path.append(move)
oldseg = copy.deepcopy(path[0]) #if we assign like "oldseg = path[0]", it will get overwritten. So we need copy
if self.options.debug is True:
self.msg("moved path (move no. {}):".format(i+1))
self.msg(path)
self.msg("-"*25)
#Now we messed the integrity of the path. It does not begin with 'M' now. But we need an 'M'.
#It now either starts with L or C. H, V, Z cannot occure here.
if path[0][0] == 'C': #and path[-1][0] == 'M':
#self.msg("C to M")
path[0][1] = [path[0][1][-2], path[0][1][-1]]
elif path[0][0] == 'L': #and path[-1][0] == 'M':
#self.msg("L to M")
path[0][1] = [path[0][1][0], path[0][1][1]]
#else:
# self.msg("no idea")
path[0][0] = 'M' #we really need M. Does not matter if 'L' or 'C'.
if pathIsClosed is True:
if path[-1][0] == 'M' and len(oldseg[1]) == 2: #data of an 'L' command
path[-1][0] = 'L'
path[-1][1] = path[0][1]
elif path[-1][0] == 'M' and len(oldseg[1]) > 2: #data of an 'C' command
path[-1][0] = 'C'
path[-1][1] = oldseg[1]
else:
if path[-1][0] == 'M': #if open path we just drop the dangling 'M' command completely
del path[-1]
if self.options.debug is True:
self.msg("final path:")
self.msg(path)
self.msg("-"*25)
newSubpaths[subpathNr - 1] = path
else:
inkex.utils.debug("More moves entered than possible to apply. Path result would be a point, not a line")
#return
composedPath = inkex.Path()
for newSubpath in newSubpaths:
composedPath.extend(newSubpath)
if self.options.debug is True:
self.msg("Composed path = " + str(composedPath))
element.path = composedPath
def visualizeFirstTwo(self, element):
"""Add a dot label for this path element"""
group = element.getparent().add(inkex.Group(id="visualize-group-" + element.get('id')))
dot_group = group.add(inkex.Group(id="dot-group-" + element.get('id')))
num_group = group.add(inkex.Group(id="num-group-" + element.get('id')))
group.transform = element.transform
radius = self.svg.unittouu(self.options.dotsize) / 2
count = 0
for step, (x, y) in enumerate(element.path.end_points):
count += 1
circle = Circle(cx=str(x), cy=str(y), r=str(radius), id="circle-" + element.get('id') + "-" + str(count))
circle.style = inkex.Style({'stroke': 'none', 'fill': '#000'})
text = TextElement(x=str(x + radius), y=str(y - radius), id="text-" + element.get('id') + "-" + str(count))
text.text = str(count) #we start with #1
text.style = inkex.Style({'font-size': self.svg.unittouu(self.options.fontsize), 'fill-opacity': '1.0', 'stroke': 'none',
'font-weight': 'normal', 'font-style': 'normal', 'fill': '#999'})
dot_group.append(circle)
num_group.append(text)
if count > 1: #we only display first two points to see the position of the first node and the path direction
break
def add_arguments(self, pars):
pars.add_argument("--tab")
pars.add_argument("--closed_only", type=inkex.Boolean, default=False, help="If disabled we also apply on open (sub)path. Warning: This REMOVES segments!")
pars.add_argument("--movenode", type=int, default=0, help="Move starting node n nodes further")
pars.add_argument('--visualize_result', type=inkex.Boolean, default=False, help="If enabled each node gets a number and a dot")
pars.add_argument("--dotsize", default="10px", help="Size of the dots on the path nodes")
pars.add_argument("--fontsize", default="20px", help="Size of node labels")
pars.add_argument("--debug", type=inkex.Boolean, default=False, help="Debug Output")
def effect(self):
if len(self.svg.selected) > 0:
elements = self.svg.selection.filter(PathElement).values()
if len(elements) > 0:
for element in elements:
#move starting element / change direction
self.modify(element)
#finally apply dots to visualize the result
if self.options.visualize_result is True:
self.visualizeFirstTwo(element)
else:
inkex.errormsg('Selection seems not to contain path elements. Maybe you have selected a group instead?')
return
else:
inkex.errormsg('Please select some objects first.')
return
if __name__ == '__main__':
MovePathNode().run()