mightyscape-1.1-deprecated/extensions/fablabchemnitz/move_path_node/move_path_node.py

#!/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, Path, PathElement, CubicSuperPath

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 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 - 1) % len(path)
            if pathIsClosed is True: #if closed start and end collapse and "duplicate"
                moves = (self.options.movenode - 1) % (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
                    
                    #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")
                
        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()