78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
# Copyright 2015 Scott Bezek
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import logging
|
|
import re
|
|
import xml
|
|
from xml.dom import minidom
|
|
|
|
"""
|
|
Processes SVG files generated by pcbnew to colorize and merge
|
|
"""
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class SvgProcessor(object):
|
|
|
|
def __init__(self, input_file):
|
|
self.dom = minidom.parse(input_file)
|
|
self.svg_node = self.dom.documentElement
|
|
|
|
def apply_color_transform(self, transform_function):
|
|
# Set fill and stroke on all groups
|
|
for group in self.svg_node.getElementsByTagName('g'):
|
|
SvgProcessor._apply_transform(group, {
|
|
'fill': transform_function,
|
|
'stroke': transform_function,
|
|
})
|
|
|
|
def import_groups(self, from_svg_processor):
|
|
for child in from_svg_processor.svg_node.childNodes:
|
|
if child.nodeType != child.ELEMENT_NODE or child.tagName != 'g':
|
|
continue
|
|
group = child
|
|
output_node = self.dom.importNode(group, True)
|
|
self.svg_node.appendChild(output_node)
|
|
|
|
def write(self, filename):
|
|
with open(filename, 'wb') as output_file:
|
|
self.svg_node.writexml(output_file)
|
|
|
|
def wrap_with_group(self, attrs):
|
|
parent = self.svg_node
|
|
wrapper = self.dom.createElement("g")
|
|
for k,v in attrs.items():
|
|
wrapper.setAttribute(k,v)
|
|
|
|
for child in parent.getElementsByTagName('g'):
|
|
parent.removeChild(child)
|
|
wrapper.appendChild(child)
|
|
|
|
parent.appendChild(wrapper)
|
|
|
|
@staticmethod
|
|
def _apply_transform(node, values):
|
|
original_style = node.attributes['style'].value
|
|
for (k,v) in values.items():
|
|
escaped_key = re.escape(k)
|
|
m = re.search(r'\b' + escaped_key + r':(?P<value>[^;]*);', original_style)
|
|
if m:
|
|
transformed_value = v(m.group('value'))
|
|
original_style = re.sub(
|
|
r'\b' + escaped_key + r':[^;]*;',
|
|
k + ':' + transformed_value + ';',
|
|
original_style)
|
|
node.attributes['style'] = original_style
|
|
|