python text to png to svg

技术 · 2023-03-03 · 478 人浏览

python text to png to svg

使用的是GitHub上开源的代码:GitHub

我还在进行修改先将基础功能放出来

text to png

from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
import img_converter

class TextToPng:
    def __init__(self, fontname, fontsize, text_color, background_color, image_border_size=4, text_border_size=3):
        self.fontname = fontname
        self.fontsize = fontsize
        self.text_color = text_color
        self.background_color = background_color
        self.image_border_size = image_border_size
        self.text_border_size = text_border_size

    def estimate_text_size(self, txt, font):
        testImg = Image.new('RGB', (1, 1))
        testDraw = ImageDraw.Draw(testImg)
        return testDraw.textsize(txt, font)

    def image_draw_text(self, text_input, output_filename):
        font = ImageFont.truetype(self.fontname, self.fontsize)
        image_width, image_height = self.estimate_text_size(text_input, font)
        img = Image.new('RGB', (image_width + self.image_border_size, image_height + self.image_border_size), self.background_color)
        drawing = ImageDraw.Draw(img)
        drawing.text((2, 2), text_input, fill=self.text_color, font=font)
        img.save(output_filename+'.png')
        img_converter.main(output_filename)  # 直接生成svg


if __name__ == "__main__":
    text_input = 'سخيفة 250 '
    text_color = "black"
    output_filename = "test_text"

    fontname = "../font/arial.ttf"
    fontsize = 40
    background_color = "white"

    TextToSVGInstance = TextToPng(fontname, fontsize, text_color, background_color)
    TextToSVGInstance.image_draw_text(text_input, output_filename)

png to svg

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from PIL import Image
import operator
from collections import deque
from io import StringIO


def add_tuple(a, b):
    return tuple(map(operator.add, a, b))


def sub_tuple(a, b):
    return tuple(map(operator.sub, a, b))


def neg_tuple(a):
    return tuple(map(operator.neg, a))


def direction(edge):
    return sub_tuple(edge[1], edge[0])


def magnitude(a):
    return int(pow(pow(a[0], 2) + pow(a[1], 2), .5))


def normalize(a):
    mag = magnitude(a)
    assert mag > 0, "Cannot normalize a zero-length vector"
    return tuple(map(operator.truediv, a, [mag] * len(a)))


def svg_header(width, height):
    return """<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" 
  "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
<svg width="%d" height="%d"
     xmlns="http://www.w3.org/2000/svg" version="1.1">
""" % (width, height)


def joined_edges(assorted_edges, keep_every_point=False):
    pieces = []
    piece = []
    directions = deque([
        (0, 1),
        (1, 0),
        (0, -1),
        (-1, 0),
    ])
    while assorted_edges:
        if not piece:
            piece.append(assorted_edges.pop())
        current_direction = normalize(direction(piece[-1]))
        while current_direction != directions[2]:
            directions.rotate()
        for i in range(1, 4):
            next_end = add_tuple(piece[-1][1], directions[i])
            next_edge = (piece[-1][1], next_end)
            if next_edge in assorted_edges:
                assorted_edges.remove(next_edge)
                if i == 2 and not keep_every_point:
                    # same direction
                    piece[-1] = (piece[-1][0], next_edge[1])
                else:
                    piece.append(next_edge)
                if piece[0][0] == piece[-1][1]:
                    if not keep_every_point and normalize(direction(piece[0])) == normalize(direction(piece[-1])):
                        piece[-1] = (piece[-1][0], piece.pop(0)[1])
                        # same direction
                    pieces.append(piece)
                    piece = []
                break
        else:
            raise Exception("Failed to find connecting edge")
    return pieces


def rgba_image_to_svg_contiguous(im, opaque=None, keep_every_point=False):
    # collect contiguous pixel groups

    adjacent = ((1, 0), (0, 1), (-1, 0), (0, -1))
    visited = Image.new("1", im.size, 0)

    color_pixel_lists = {}

    width, height = im.size
    for x in range(width):
        for y in range(height):
            here = (x, y)
            if visited.getpixel(here):
                continue
            rgba = im.getpixel((x, y))
            if opaque and not rgba[3]:
                continue
            piece = []
            queue = [here]
            visited.putpixel(here, 1)
            while queue:
                here = queue.pop()
                for offset in adjacent:
                    neighbour = add_tuple(here, offset)
                    if not (0 <= neighbour[0] < width) or not (0 <= neighbour[1] < height):
                        continue
                    if visited.getpixel(neighbour):
                        continue
                    neighbour_rgba = im.getpixel(neighbour)
                    if neighbour_rgba != rgba:
                        continue
                    queue.append(neighbour)
                    visited.putpixel(neighbour, 1)
                piece.append(here)

            if not rgba in color_pixel_lists:
                color_pixel_lists[rgba] = []
            color_pixel_lists[rgba].append(piece)

    del adjacent
    del visited

    # calculate clockwise edges of pixel groups

    edges = {
        (-1, 0): ((0, 0), (0, 1)),
        (0, 1): ((0, 1), (1, 1)),
        (1, 0): ((1, 1), (1, 0)),
        (0, -1): ((1, 0), (0, 0)),
    }

    color_edge_lists = {}

    for rgba, pieces in color_pixel_lists.items():
        for piece_pixel_list in pieces:
            edge_set = set([])
            for coord in piece_pixel_list:
                for offset, (start_offset, end_offset) in edges.items():
                    neighbour = add_tuple(coord, offset)
                    start = add_tuple(coord, start_offset)
                    end = add_tuple(coord, end_offset)
                    edge = (start, end)
                    if neighbour in piece_pixel_list:
                        continue
                    edge_set.add(edge)
            if not rgba in color_edge_lists:
                color_edge_lists[rgba] = []
            color_edge_lists[rgba].append(edge_set)

    del color_pixel_lists
    del edges

    # join edges of pixel groups

    color_joined_pieces = {}

    for color, pieces in color_edge_lists.items():
        color_joined_pieces[color] = []
        for assorted_edges in pieces:
            color_joined_pieces[color].append(joined_edges(assorted_edges, keep_every_point))

    s = StringIO()
    s.write(svg_header(*im.size))

    for color, shapes in color_joined_pieces.items():
        for shape in shapes:
            s.write(""" <path d=" """)
            for sub_shape in shape:
                here = sub_shape.pop(0)[0]
                s.write(""" M %d,%d """ % here)
                for edge in sub_shape:
                    here = edge[0]
                    s.write(""" L %d,%d """ % here)
                s.write(""" Z """)
            s.write(
                """ " style="fill:rgb%s; fill-opacity:%.3f; stroke:none;" />\n""" % (color[0:3], float(color[3]) / 255))

    s.write("""</svg>\n""")
    return s.getvalue()


def rgba_image_to_svg_pixels(im, opaque=None):
    s = StringIO()
    s.write(svg_header(*im.size))

    width, height = im.size
    for x in range(width):
        for y in range(height):
            here = (x, y)
            rgba = im.getpixel(here)
            if opaque and not rgba[3]:
                continue
            s.write(
                """  <rect x="%d" y="%d" width="1" height="1" style="fill:rgb%s; fill-opacity:%.3f; stroke:none;" />\n""" % (
                x, y, rgba[0:3], float(rgba[2]) / 255))
    s.write("""</svg>\n""")
    return s.getvalue()


def main(filename):
    image = Image.open(filename+'.png').convert('RGBA')
    svg_image = rgba_image_to_svg_contiguous(image)
    # svg_image = rgba_image_to_svg_pixels(image)
    with open(filename+".svg", "w") as text_file:
        text_file.write(svg_image)


if __name__ == '__main__':
    main('./en')

这个完全不会修改 看不懂 哈哈哈哈

python text to image image to svg
Theme Jasmine by Kent Liao