else/PixelsToSVG/pixelstosvg.py

135 lines
3.2 KiB
Python
Raw Normal View History

2017-05-07 01:29:10 +00:00
import argparse
import os
import PIL.Image
import random
import sys
SVG_TEMPLATE = '''
<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<svg
xmlns:dc="http://purl.org/dc/elements/1.1/"
xmlns:cc="http://creativecommons.org/ns#"
xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:svg="http://www.w3.org/2000/svg"
xmlns="http://www.w3.org/2000/svg"
id="svg5633"
version="1.1"
viewBox="0 0 {width} {height}"
height="{height}mm"
width="{width}mm">
<defs
id="defs5627" />
<metadata
id="metadata5630">
<rdf:RDF>
<cc:Work
rdf:about="">
<dc:format>image/svg+xml</dc:format>
<dc:type
rdf:resource="http://purl.org/dc/dcmitype/StillImage" />
<dc:title></dc:title>
</cc:Work>
</rdf:RDF>
</metadata>
<g
id="layer1">
{rectangles}
</g>
</svg>
'''.strip()
RECTANGLE_TEMPLATE = '''
<rect
y="{y}"
x="{x}"
height="{height}"
width="{width}"
id="{id}"
style=
"opacity:{opacity};
fill:{fill};
fill-opacity:{opacity};
fill-rule:nonzero;
stroke:none;
stroke-width:0;
stroke-linecap:round;
stroke-linejoin:miter;
stroke-miterlimit:4;
stroke-dasharray:none;
stroke-dashoffset:0;
stroke-opacity:1"
/>
'''.strip()
def normalize_rgb(rgb, mode):
if mode == '1':
v = rgb * 255
return (v, v, v, 255)
if mode == 'L':
return (rgb, rgb, rgb, 255)
if mode == 'RGB':
return rgb + (255, )
if mode == 'RGBA':
return rgb
def int_to_hex(i):
return hex(i)[2:].rjust(2, '0')
def rgb_to_hex(rgb):
return '#' + ''.join(int_to_hex(x) for x in rgb)
def make_svg(image):
rectangles = []
(width, height) = image.size
for y in range(height):
for x in range(width):
pixel = image.getpixel((x, y))
pixel = normalize_rgb(pixel, image.mode)
(rgb, opacity) = (pixel[:-1], pixel[-1])
fill = rgb_to_hex(rgb)
#print(fill)
opacity = opacity / 255
rectangle = RECTANGLE_TEMPLATE.format(
x=x,
y=y,
width=1,
height=1,
id=str(random.random()),
fill=fill,
opacity=opacity,
)
rectangles.append(rectangle)
#print(rectangle)
rectangles = '\n'.join(rectangles)
svg = SVG_TEMPLATE.format(width=width, height=height, rectangles=rectangles)
return svg
def image_to_svg(image_filename, svg_filename=None):
svg_filename = svg_filename or ''
if not svg_filename:
svg_filename = image_filename + '.svg'
image = PIL.Image.open(image_filename)
svg = make_svg(image)
with open(svg_filename, 'w') as handle:
handle.write(svg)
def image_to_svg_argparse(args):
return image_to_svg(
image_filename=args.image_filename,
svg_filename=args.svg_filename
)
def main(argv):
parser = argparse.ArgumentParser()
parser.add_argument('image_filename')
parser.add_argument('svg_filename', nargs='?', default=None)
parser.set_defaults(func=image_to_svg_argparse)
args = parser.parse_args(argv)
return args.func(args)
2017-05-07 01:29:10 +00:00
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))