Add prototype code for big text labels

master
Tanner Collin 10 months ago
parent c4643daa36
commit 373acb67de
  1. 93
      big_text_labels.py

@ -0,0 +1,93 @@
#!/usr/bin/env python3
from PIL import Image, ImageEnhance, ImageFont, ImageDraw
from os import path
def fit_text(text, script_dir, font_path, label_path, font_size, anchor="lt"):
img = Image.open(label_path)
draw = ImageDraw.Draw(img)
print(f"img.size: {img.size}")
font = ImageFont.truetype(font_path, size=font_size)
space_bbox = font.getbbox(" ", anchor=anchor)
[space_left, space_upper, space_right, space_lower] = space_bbox
space_width = abs(space_left) + abs(space_right)
space_height = abs(space_upper) + abs(space_lower)
print(f"font_size: {font_size} space_bbox: {space_bbox}")
text_cur = [0, 0]
line_bottom = 0
for word in text.split():
word_bbox = font.getbbox(word, anchor=anchor)
[word_left, word_upper, word_right, word_lower] = word_bbox
word_width = abs(word_left) + abs(word_right)
word_height = abs(word_upper) + abs(word_lower)
word_rec = [
word_left + text_cur[0],
word_upper + text_cur[1],
word_right + text_cur[0],
word_lower + text_cur[1],
]
[rec_left, rec_upper, rec_right, rec_lower] = word_rec
line_bottom = max(rec_lower, line_bottom)
if rec_right > img.size[0]:
# new line, move down, start at left
text_cur = [0, line_bottom]
word_rec = [
word_left + text_cur[0],
word_upper + text_cur[1],
word_right + text_cur[0],
word_lower + text_cur[1],
]
[rec_left, rec_upper, rec_right, rec_lower] = word_rec
line_bottom = max(rec_lower, line_bottom)
if rec_right > img.size[0]:
# word is currently extending off the right side of the page.
# despite doing a new line.
return False
# print(f"{word:>10}, word_bbox: {word_bbox}, word_rec: {word_rec}")
draw.text(text_cur, word, font=font, fill="black", anchor=anchor)
draw.rectangle(
word_rec, outline="#663399"
) # optional. draw a border around the text.
# move over X direction
text_cur[0] += word_width + space_width
img.save(path.join(script_dir, "tmp.png"))
return line_bottom <= img.size[1]
def main():
# I'm trying to make text as big as possible within a rectangle for generating labels
text = "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum."
#text = "UDIA"
text = 'a'*100
text = 'hello world sample sentence test'
script_dir = path.dirname(path.abspath(__file__))
label_path = path.join(script_dir, "label.png")
font_path = '/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf'
font_size_range = [1, 1000]
font_size = sum(font_size_range) // len(font_size_range)
while abs(font_size_range[0] - font_size_range[1]) > 1:
image_fit = fit_text(text, script_dir, font_path, label_path, font_size)
print(f"image_fit: {image_fit}, font_size: {font_size}, font_size_range={font_size_range}")
if image_fit:
font_size_range = [font_size, font_size_range[1]]
else:
font_size_range = [font_size_range[0], font_size]
font_size = sum(font_size_range) // len(font_size_range)
if __name__ == "__main__":
main()
Loading…
Cancel
Save