#!/usr/bin/env python3

'''
Inkscape extension to copy length of the source path to the selected 
destination path(s)

Copyright (C) 2018  Shrinivas Kulkarni

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.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
'''

import inkex
from inkex import bezier
from inkex.paths import Path, CubicSuperPath

class PasteLengthEffect(inkex.Effect):

    def __init__(self):
        inkex.Effect.__init__(self)
        self.arg_parser.add_argument('--scale', type = float, default = '1', help = 'Additionally scale the length by')
        self.arg_parser.add_argument('--scaleFrom', default = 'center', help = 'Scale Path From')
        self.arg_parser.add_argument('--precision', type = int, default = '5', help = 'Number of significant digits')
        self.arg_parser.add_argument("--tab", default="sampling", help="Tab") 

    def scaleCubicSuper(self, cspath, scaleFactor, scaleFrom):
        bbox = Path(cspath).bounding_box()

        if(scaleFrom == 'topLeft'):
            oldOrigin= [bbox.left, bbox.bottom]
        elif(scaleFrom == 'topRight'):
            oldOrigin= [bbox.right, bbox.bottom]
        elif(scaleFrom == 'bottomLeft'):
            oldOrigin= [bbox.left, bbox.top]
        elif(scaleFrom == 'bottomRight'):
            oldOrigin= [bbox.right, bbox.top]
        else: #if(scaleFrom == 'center'):
            oldOrigin= [bbox.left + (bbox.right - bbox.left) / 2., bbox.bottom + (bbox.top - bbox.bottom) / 2.]
            
        newOrigin = [oldOrigin[0] * scaleFactor , oldOrigin[1] * scaleFactor ]
            
        for subpath in cspath:
            for bezierPt in subpath:
                for i in range(0, len(bezierPt)):
                    
                    bezierPt[i] = [bezierPt[i][0] * scaleFactor, 
                        bezierPt[i][1] * scaleFactor]
                        
                    bezierPt[i][0] += (oldOrigin[0] - newOrigin[0])
                    bezierPt[i][1] += (oldOrigin[1] - newOrigin[1])
                
    def getPartsFromCubicSuper(self, cspath):
        parts = []
        for subpath in cspath:
            part = []
            prevBezPt = None            
            for i, bezierPt in enumerate(subpath):
                if(prevBezPt != None):
                    seg = [prevBezPt[1], prevBezPt[2], bezierPt[0], bezierPt[1]]
                    part.append(seg)
                prevBezPt = bezierPt
            parts.append(part)
        return parts
        
    def getLength(self, cspath, tolerance):
        parts = self.getPartsFromCubicSuper(cspath)
        curveLen = 0
        for i, part in enumerate(parts):
            for j, seg in enumerate(part):
                curveLen += bezier.bezierlength((seg[0], seg[1], seg[2], seg[3]), tolerance = tolerance)
        return curveLen

    def effect(self):
        scale = self.options.scale
        scaleFrom = self.options.scaleFrom
        tolerance = 10 ** (-1 * self.options.precision)
        selections = self.svg.selected
        pathNodes = self.document.xpath('//svg:path',namespaces=inkex.NSS)
        outStrs = [str(len(pathNodes))]

        paths = [(pathNode.get('id'), CubicSuperPath(pathNode.get('d'))) for pathNode in pathNodes]

        if(len(paths) > 1):
            srcPath = paths[-1][1]
            srclen = self.getLength(srcPath, tolerance)
            paths = paths[:len(paths)-1]
            for key, cspath in paths:
                curveLen = self.getLength(cspath, tolerance)
                
                self.scaleCubicSuper(cspath, scaleFactor = scale * (srclen / curveLen), scaleFrom = scaleFrom)
                try: #we wrap this in try-except because if the elements are within groups it will cause errors
                    selections[key].set('d', CubicSuperPath(cspath))
                except:
                    pass
        else:
            inkex.errormsg("Please select at least two paths, with the path whose \
            length is to be copied at the top. You may have to convert the shape \
            to path with path->Object to Path.")

if __name__ == '__main__':
    PasteLengthEffect().run()