#! /usr/bin/python ''' Copyright (C) 2017 Artem Synytsyn a.synytsyn@gmail.com #TODO: Code cleaning and refactoring 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 import sys from lxml import etree from math import * import re from typing import Any import wikipedia import openpyxl import math, decimal, datetime dec = decimal.Decimal # ['NAMESPACE', 'PARSER', 'TAG', 'WRAPPED_ATTRS', '__abstractmethods__', '__bool__', # '__class__', '__contains__', '__copy__', '__deepcopy__', '__delattr__', '__delitem__', # '__dict__', '__dir__', '__doc__', '__eq__', '__format__', '__ge__', '__getattr__', # '__getattribute__', '__getitem__', '__getstate__', '__gt__', '__hash__', '__init__', # '__init_subclass__', '__iter__', '__le__', '__len__', '__lt__', '__module__', '__ne__', # '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__reversed__', '__setattr__', # '__setitem__', '__sizeof__', '__slots__', '__str__', '__subclasshook__', '__weakref__', # '_abc_impl', '_ancestors', '_init', '_is_visible', 'add', 'add_unit', 'addnext', # 'addprevious', 'ancestors', 'append', 'attrib', 'backlinks', 'bake_transforms_recursively', # 'base', 'bounding_box', 'cascaded_style', 'clear', 'clip', 'composed_style', # 'composed_transform', 'copy', 'cssselect', 'delete', 'desc', 'descendants', # 'description', 'duplicate', 'effective_style', 'eid', 'extend', 'find', 'findall', # 'findone', 'findtext', 'get', 'get_id', 'get_line_height_uu', 'get_or_create', # 'get_path', 'getchildren', 'getiterator', 'getnext', 'getparent', 'getprevious', # 'getroottree', 'groupmode', 'href', 'index', 'insert', 'is_class_element', # 'is_sensitive', 'is_visible', 'items', 'iter', 'iterancestors', 'iterchildren', # 'iterdescendants', 'iterfind', 'itersiblings', 'itertext', 'keys', 'label', # 'makeelement', 'new', 'nsmap', 'path', 'pop', 'prefix', 'presentation_style', # 'remove', 'remove_all', 'replace', 'replace_with', 'root', 'set', 'set_id', # 'set_path', 'set_random_id', 'set_random_ids', 'set_sensitive', 'shape_box', # 'sourceline', 'specified_style', 'tag', 'tag_name', 'tail', 'text', 'title', # 'to_dimensional', 'to_dimensionless', 'to_path_element', 'tostring', 'typename', # 'unit', 'unit_to_viewport', 'unittouu', 'update', 'uutounit', 'values', # 'viewport_to_unit', 'wrapped_attrs', 'wrapped_props', 'xml_path', 'xpath'] # ___ _ _ _ _ _ # |_ _|_ __ | | _____ ___ __ _ _ __ ___| | | |_ __ _ __(_)_ __ | |_ # | || '_ \| |/ / __|/ __/ _` | '_ \ / _ \ | | | '_ \| '__| | '_ \| __| # | || | | | <\__ \ (_| (_| | |_) | __/ |_| | |_) | | | | | | | |_ # |___|_| |_|_|\_\___/\___\__,_| .__/ \___|\___/| .__/|_| |_|_| |_|\__| # |_| |_| # class _Spacing(): # size = 0. # padding = 0. # margin_top = 0. # margin_bottom = 0. # margin_left = 0. # margin_right = 0. class InkscapeUprint(inkex.Effect): def __init__(self): inkex.Effect.__init__(self) # General settings self.arg_parser.add_argument("--x", type=float, dest="x", default=0.0, help="Center X") self.arg_parser.add_argument("--y", type=float, dest="y", default=0.0, help="Center Y") self.arg_parser.add_argument("-u", "--units", type=str, dest="units", default="mm", help="units to measure size of knob") # Dummy self.arg_parser.add_argument("--tab") # # Label settings self.arg_parser.add_argument("--subst_file", type=str, dest="subst_file", default='', help="xlsx file to load substitutions") self.__element_spacing = { 'size' : 0, 'padding' : 0, 'margin-top' : 0, 'margin-bottom' : 0, 'margin-left' : 0, 'margin-right' : 0 } self.__page_spacing = { 'size' : 0, 'padding' : 0, 'margin-top' : 0, 'margin-bottom' : 0, 'margin-left' : 0, 'margin-right' : 0 } for sk in self.__element_spacing.keys(): sk = sk.replace('-','_') self.arg_parser.add_argument("--element_" + sk, type=float, dest="element_" + sk, help="Spacing element" + " " + sk) for sk in self.__page_spacing.keys(): sk = sk.replace('-','_') self.arg_parser.add_argument("--page_" + sk, type=float, dest="page_" + sk, help="Spacing page" + " " + sk) def svg_traverse_node(self, node, tag=None, exclude=None): def _tag(t): return re.sub(r'\{.*?\}', r'', t) l = [] if isinstance(node, etree._Element): if (tag is None or _tag(node.tag) == tag) and (exclude is None or _tag(node.tag) != exclude): l.append(node) for child in node.getchildren(): l.extend(self.svg_traverse_node(child)) elif isinstance(node, etree._Comment): pass elif isinstance(node, list): for child in node: l.extend(self.svg_traverse_node(child)) return l def replace_text(self, selected, substs): import re for node in selected: tspans = [ n for n in self.svg_traverse_node(node, tag='tspan')] for k,v in substs.items(): patt = re.compile(r'\{\{' + re.escape(k) + r'\}\}') for t in tspans: if t.text is None: continue t.text = patt.sub(v, t.text) # multilines = t.text.split('\n') # if len(multilines) > 1: # t.text = multilines[0] # for i in range(1, len(multilines)): # new_t = etree.Element('tspan', nsmap=t.nsmap) # new_t.text = multilines[i] # new_t.tail = t.tail # new_t.attrib.update(t.attrib) # copy style # p = t.getparent() # p.insert(p.index(t) + 1, new_t) def duplicate_object(self, parent, selected): duplicated = [] import copy for node in selected: new_node = node.duplicate() duplicated.append(new_node) # duplicated.append(etree.Element(node.tag, node.attrib, nsmap=node.nsmap)) # # # inkex.utils.debug(' node.text = ' + str(type(node.text))) # duplicated[-1].tail = node.tail # duplicated[-1].text = node.text # self.duplicate_object(duplicated[-1], node.getchildren()) parent.append(duplicated[-1]) return duplicated def translate_object(self, obj, x, y): trans = 'translate(' + str(float(x)) + ',' + str(float(y)) + ')' for d in obj: if 'transform' in d.attrib: d.set('transform', d.get('transform') + ' ' + trans) else: d.set('transform', trans) def place_in_pages(self, elements = None): if elements is None: return first_page = self.svg.getElement("//inkscape:page") if first_page is not None: page_width = float(first_page.get('width')) page_height = float(first_page.get('height')) view = first_page.getparent() else: view = self.svg.getElement("//sodipodi:namedview") if view is None: inkex.utils.errormsg('No view found') return page_width = self.svg.get_page_bbox().width page_height = self.svg.get_page_bbox().height pages = [ node for node in self.svg_traverse_node(view, tag='inkscape:page')] # the reference page is always the first one pgbbox = {"width": page_width, "height": page_height} elbbox = self.bbox nx = int (( float(pgbbox['width']) - self.__page_spacing['margin-left'] - self.__page_spacing['margin-right'] - self.__element_spacing['padding']) / \ ( elbbox.width + self.__element_spacing['padding'] )) ny = int (( float(pgbbox['height']) - self.__page_spacing['margin-top'] - self.__page_spacing['margin-bottom'] - self.__element_spacing['padding'] ) / \ ( elbbox.height + self.__element_spacing['padding'] )) # remove model position correct_x = self.bbox.left correct_y = self.bbox.top # compute page position zero if len(pages) > 0: rightmost_page = max(pages, key=lambda p: p.get('x')) px = float(rightmost_page.get('x')) py = float(rightmost_page.get('y')) else: px = 0. py = 0. # place elements for i, el in enumerate(elements): # create new page if needed # inkex.utils.debug('nx: '+str(nx) + ' ny: '+str(ny)) if i % int(nx * ny) == 0: new_page = etree.Element('{http://www.inkscape.org/namespaces/inkscape}page', nsmap=view.nsmap) new_page.set('x', str(px + ( page_width + self.__page_spacing['padding'] ))) new_page.set('y', str(py) ) new_page.set('width', str(page_width)) new_page.set('height', str(page_height)) view.append(new_page) rightmost_page = new_page px = float(rightmost_page.get('x')) py = float(rightmost_page.get('y')) x = px + (i % nx) * ( elbbox.width + self.__element_spacing['padding'] ) \ + self.__page_spacing['margin-left'] + self.__element_spacing['padding'] y = py + (i % (nx*ny) // nx) * ( elbbox.height + self.__element_spacing['padding'] ) \ + self.__page_spacing['margin-top'] + self.__element_spacing['padding'] self.translate_object(el, x - correct_x, y - correct_y) def compute_bounding_box(self, selected): _tag = lambda t: re.sub(r'\{.*?\}', r'', t) objs = [ n for n in self.svg_traverse_node(selected[0]) if _tag(n.tag) not in ['text', 'tspan','g'] ] l = inkex.elements._selected.ElementList(self.svg, objs) return l.bounding_box() def effect(self): # inkex.utils.debug('inkex: ' + inkex.__file__) parent = self.svg.get_current_layer() selected = self.svg.selected for k in self.__page_spacing.keys(): self.__page_spacing[k] = self.svg.unittouu( str(self.options.__dict__['page_'+k.replace('-','_')]) + self.options.units ) for k in self.__element_spacing.keys(): self.__element_spacing[k] = self.svg.unittouu(str(self.options.__dict__['element_'+k.replace('-','_')]) + self.options.units) if selected: self.bbox = self.compute_bounding_box(selected) if self.bbox is None: inkex.utils.errormsg('bbox is None') # self.x_offset = self.svg.unittouu(str(self.options.x) + self.options.units) # self.y_offset = self.svg.unittouu(str(self.options.y) + self.options.units) else: inkex.utils.errormsg("Error - No object selected") return wb = openpyxl.load_workbook(self.options.subst_file) ws = wb.active headings = [cell.value for cell in ws[1]] substs = [] for row in ws.iter_rows(min_row=2, values_only=True): if all([cell is None or cell.strip() == '' for cell in row]): continue substs.append(dict(zip(headings, [cell if cell is not None else '' for cell in row]))) new_objects = [] for i, subst in enumerate(substs): dup = self.duplicate_object(parent, selected) self.replace_text(dup, subst) new_objects.append(dup) # populating spacing with options self.place_in_pages(new_objects) if __name__ == '__main__': e = InkscapeUprint() e.run()