Badger/generate.py at main · meetesh06/Badger · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
from PIL import Image, ImageDraw, ImageFont
import pandas as pd
import argparse
from pathlib import Path
import os
import shutil

def ensure_folder_exists_and_clear(folder_path):
    os.makedirs(folder_path, exist_ok=True)
    for filename in os.listdir(folder_path):
        file_path = os.path.join(folder_path, filename)
        try:
            if os.path.isfile(file_path) or os.path.islink(file_path):
                os.unlink(file_path)  # Remove file or symbolic link
            elif os.path.isdir(file_path):
                shutil.rmtree(file_path)  # Remove directory and contents
        except Exception as e:
            print(f"Failed to delete {file_path}: {e}")


def find_max_font_size(draw, text, font_path, max_font_size, rect):
    # Try decreasing font size until text fits within the rectangle
    for font_size in range(max_font_size, 0, -1):
        font = ImageFont.truetype(font_path, font_size)
        bbox = draw.textbbox((0, 0), text, font=font)
        text_width = bbox[2] - bbox[0]
        text_height = bbox[3] - bbox[1]
        if text_width <= (rect[2] - rect[0]) and text_height <= (rect[3] - rect[1]):
            return font_size  # Found the maximum font size that fits
    return 0

def plotBox(draw, rect, font_path, font_size, text_to_print):
    font = ImageFont.truetype(font_path, font_size)
    bbox = draw.textbbox((0, 0), text_to_print, font=font)
    text_width = bbox[2] - bbox[0]
    text_height = bbox[3] - bbox[1]
    text_x = (rect[0] + rect[2] - text_width) // 2
    text_y = (rect[1] + rect[3] - text_height) // 2
    text_position = (text_x, text_y)
    draw.text(text_position, text_to_print, font=font, fill="black")

def plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path):
    draw = ImageDraw.Draw(img)
    max_font_size = 68
    min_font_size = 50  # Ensure font size is at least 50
    vertical_spacing = 15

    # Split the name into lines if necessary
    words = name1.split(" ")
    lines = []
    current_line = words[0]

    for word in words[1:]:
        test_line = f"{current_line} {word}"
        font = ImageFont.truetype(font_path, max_font_size)
        bbox = draw.textbbox((0, 0), test_line, font=font)
        test_width = bbox[2] - bbox[0]
        if test_width <= (rect[2] - rect[0]):
            current_line = test_line
        else:
            lines.append(current_line)
            current_line = word
    lines.append(current_line)

    # Find the maximum font size that works for all lines
    for font_size in range(max_font_size, min_font_size - 1, -1):
        font = ImageFont.truetype(font_path, font_size)
        fits = True
        for line in lines:
            bbox = draw.textbbox((0, 0), line, font=font)
            text_width = bbox[2] - bbox[0]
            if text_width > (rect[2] - rect[0]):
                fits = False
                break
        if fits:
            break  # Found the maximum font size that works for all lines

    # Calculate total text height
    total_text_height = len(lines) * font_size + (len(lines) - 1) * vertical_spacing
    current_y = (rect[1] + rect[3] - total_text_height) // 2

    # Draw each line
    font = ImageFont.truetype(font_path, font_size)
    for line in lines:
        bbox = draw.textbbox((0, 0), line, font=font)
        text_width = bbox[2] - bbox[0]
        text_position = ((rect[0] + rect[2] - text_width) // 2, current_y)
        draw.text(text_position, line, font=font, fill="black")
        current_y += font_size + vertical_spacing

    # Dynamically adjust rect_affiliation to be below the name rectangle
    rect_affiliation[1] = rect[1] + total_text_height + vertical_spacing
    rect_affiliation[3] = rect_affiliation[1] + 50  # Adjust height for affiliation text

    # Ensure rect_affiliation does not overlap with the name rectangle
    if rect_affiliation[1] < rect[3]:
        rect_affiliation[1] = rect[3] + vertical_spacing
        rect_affiliation[3] = rect_affiliation[1] + 50

    plotBox(draw, rect_affiliation, font_path, 50, aff1)

def main():
    parser = argparse.ArgumentParser(description="PLλTO Badger, IITB")
    parser.add_argument("csv", type=str, help="path to CSV containing columns 'Name' and 'Affiliation'; all other columns will be ignored")
    parser.add_argument("template", type=str, help="path to template file to use, see PLACID2024.png for reference (we expect to print 6 ID's in an A4 Sheet)")
    parser.add_argument("font", type=str, help="path to the font file to use, Kollektif-Bold.ttf is a good choice...")
    parser.add_argument("--out", type=str, help="Path to the output folder (default='./outputs')", required=False, default="./outputs")
    parser.add_argument("--h-scale", type=float, help="Horizontal Scaling (default = 67.34006734)", required=False, default=67.34006734)
    parser.add_argument("--v-scale", type=float, help="Horizontal Scaling (default = 67.333333333)", required=False, default=67.333333333)
    args = parser.parse_args()
    csv_path = Path(args.csv)
    template_path = Path(args.template)
    font_path = Path(args.font)
    horizontal_scaling = args.h_scale
    vertical_scaling = args.v_scale
    outputs_path = args.out
    ensure_folder_exists_and_clear(outputs_path)
    df = pd.read_csv(csv_path)
    names = df['Name']
    affiliation = df['Affiliation']

    for x in range(0, len(names), 6):
        img = Image.open(template_path)
        name1 = names[x]
        aff1 = affiliation[x]
        rect = [1*horizontal_scaling, 4*vertical_scaling, 8.6*horizontal_scaling, 7.0*vertical_scaling]
        rect_affiliation = [1*horizontal_scaling, 6*vertical_scaling, 8.6*horizontal_scaling, 8.2*vertical_scaling]
        plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path)

        name1 = names[x + 1]
        aff1 = affiliation[x + 1]
        rect = [1*horizontal_scaling, 14.5*vertical_scaling, 8.6*horizontal_scaling, 17.5*vertical_scaling]
        rect_affiliation = [1*horizontal_scaling, 16.5*vertical_scaling, 8.6*horizontal_scaling, 18.7*vertical_scaling]
        plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path)

        name1 = names[x + 2]
        aff1 = affiliation[x + 2]
        rect = [11*horizontal_scaling, 4*vertical_scaling, 18.6*horizontal_scaling, 7*vertical_scaling]
        rect_affiliation = [11*horizontal_scaling, 6*vertical_scaling, 18.6*horizontal_scaling, 8.2*vertical_scaling]
        plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path)

        name1 = names[x + 3]
        aff1 = affiliation[x + 3]
        rect = [11*horizontal_scaling, 14.5*vertical_scaling, 18.6*horizontal_scaling, 17.5*vertical_scaling]
        rect_affiliation = [11*horizontal_scaling, 16.5*vertical_scaling, 18.6*horizontal_scaling, 18.7*vertical_scaling]
        plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path)

        name1 = names[x + 4]
        aff1 = affiliation[x + 4]
        rect = [21.1*horizontal_scaling, 4*vertical_scaling, 28.6*horizontal_scaling, 7*vertical_scaling]
        rect_affiliation = [21.1*horizontal_scaling, 6*vertical_scaling, 28.6*horizontal_scaling, 8.2*vertical_scaling]
        plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path)

        name1 = names[x + 5]
        aff1 = affiliation[x + 5]
        rect = [21.1*horizontal_scaling, 14.5*vertical_scaling, 28.6*horizontal_scaling, 17.5*vertical_scaling]
        rect_affiliation = [21.1*horizontal_scaling, 16.5*vertical_scaling, 28.6*horizontal_scaling, 18.7*vertical_scaling]
        plotForEntry(img, name1, aff1, rect, rect_affiliation, font_path)

        img.save(f"{outputs_path}/generated-{x}-{x+3}.png")

if __name__ == "__main__":
    main()