71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
#! /usr/bin/env python3
|
|
'''
|
|
Copyright (C) 2019 Grant Patterson <grant@revoltlabs.co>
|
|
|
|
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
|
|
'''
|
|
|
|
import gettext
|
|
import sys
|
|
import inkex
|
|
import rotate_helper
|
|
from inkex import Transform
|
|
import copy
|
|
|
|
debug = False
|
|
|
|
error = lambda msg: inkex.errormsg(gettext.gettext(msg))
|
|
if debug:
|
|
stderr = lambda msg: sys.stderr.write(msg + '\n')
|
|
else:
|
|
stderr = lambda msg: None
|
|
|
|
class RotationsFindAllOptimal(inkex.EffectExtension):
|
|
|
|
def add_arguments(self, pars):
|
|
pars.add_argument("--precision", type=int, default=3, help="Precision")
|
|
|
|
def effect(self):
|
|
|
|
def duplicateNodes(aList):
|
|
clones={}
|
|
for id, node in aList.items():
|
|
clone = copy.deepcopy(node)
|
|
myid = node.tag.split('}')[-1]
|
|
clone.set("id", self.svg.get_unique_id(myid))
|
|
node.getparent().append(clone)
|
|
node.delete()
|
|
clones[clone.get("id")]=clone
|
|
return(clones)
|
|
|
|
for nid, node in self.svg.selected.items():
|
|
# set() removes duplicates
|
|
angles = set(
|
|
# and remove Nones
|
|
[x for x in rotate_helper.optimal_rotations(node, self.options.precision)
|
|
if x is not None])
|
|
# Go backwards so we know if we need to duplicate the node for
|
|
# multiple rotations. (We don't want to rotate the main node
|
|
# before duplicating it.)
|
|
for i, angle in reversed(list(enumerate(angles))):
|
|
if i > 0:
|
|
# Rotate a duplicate of the node
|
|
rotate_node = list(duplicateNodes({nid: node}).items())[0][1]
|
|
else:
|
|
rotate_node = node
|
|
rotate_node.transform = rotate_helper.rotate_matrix(rotate_node, angle) @ rotate_node.transform
|
|
|
|
if __name__ == '__main__':
|
|
RotationsFindAllOptimal().run() |