amy/label_printer
1
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

modify gen_image to allow for n number of lines

Browse source
This commit is contained in:
TT-392 2025-10-31 14:58:23 +00:00
parent c7b5ac9f5f
commit eb1056144c
1 changed files with 30 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
gen_image.py
View file

@ -2,38 +2,39 @@
from PIL import Image, ImageDraw, ImageFont
from colorama import Fore, Style
from config import *
from fonts import fonts
#font = "resources/ComicMono.ttf"
def gen_image(height, text1, text2, font):
font1 = ImageFont.truetype(font["path"], size=text1["size"])
font2 = ImageFont.truetype(font["path"], size=text2["size"])
def gen_image(height, lines, font):
text_widths = []
for text in lines:
pil_font = ImageFont.truetype(font["path"], size=text["size"])
text1["string"] = text1["string"].replace("\r\n", "\n")
text2["string"] = text2["string"].replace("\r\n", "\n")
text["string"] = text["string"].replace("\r\n", "\n")
lines1 = text1["string"].split('\n')
lines2 = text2["string"].split('\n')
sub_lines = text["string"].split('\n')
text1_width = max([font1.getbbox(line)[2] for line in lines1])
text2_width = max([font2.getbbox(line)[2] for line in lines2])
text_widths.append(max([pil_font.getbbox(sub_line)[2] for sub_line in sub_lines]))
text_max_start = max([pil_font.getbbox(sub_line)[0] for sub_line in sub_lines])
text1_max_start = max([font1.getbbox(line)[0] for line in lines1])
text2_max_start = max([font2.getbbox(line)[0] for line in lines2])
# I am assuming the left corner of the bbox to always be 0, I have yet to
# encounter a situation when this isn't the case
if (text_max_start != 0):
print(Fore.YELLOW + "Warning, found situation where left corner of bbox of text != 0,", "text1_box:", text1, "text2_box", text2)
# I am assuming the left corner of the bbox to always be 0, I have yet to
# encounter a situation when this isn't the case
if (text1_max_start != 0 or text2_max_start != 0):
print(Fore.YELLOW + "Warning, found situation where left corner of bbox of text != 0,", "text1_box:", text1, "text2_box", text2)
# the 1 here assures we don't get a 0 width image
width = max(text1_width, text2_width, 1)
text_widths.append(1)
width = max(text_widths)
# '1' is 1 bit bitdepth
img = Image.new('1', (width, height), color=1)
draw = ImageDraw.Draw(img)
draw.text((0, text1["pos"]), text1["string"], font=font1, stroke_width = font["stroke_width_bold"] if text1["bold"] else font["stroke_width"], stroke_fill="black")
draw.text((0, text2["pos"]), text2["string"], font=font2, stroke_width = font["stroke_width_bold"] if text2["bold"] else font["stroke_width"], stroke_fill="black")
for line in lines:
pil_font = ImageFont.truetype(font["path"], size=text["size"])
draw.text((0, line["pos"]), line["string"], font=pil_font, stroke_width = font["stroke_width_bold"] if line["bold"] else font["stroke_width"], stroke_fill="black")
if (width > max_label_length):
@ -49,11 +50,22 @@ Text1 = {
"string": "Hello world",
"size": 30,
"pos": 10,
"bold": False,
}
Text2 = {
"string": "Hello worhfeiheifhefld\nafefe",
"size": 20,
"pos": 40,
"bold": False,
}
Text3 = {
"string": "Hello worhfeiheifhefld\nafefe",
"size": 20,
"pos": 80,
"bold": False,
}
gen_image(200, [Text1, Text2, Text3], fonts()["CYBER"])[2].save("aaa.png")