|
from PIL import Image
|
|
from pathlib import Path
|
|
import functools
|
|
import sys
|
|
|
|
printr = functools.partial(print, file=sys.stderr)
|
|
|
|
per_image_width = 480
|
|
per_image_height = 320
|
|
cols = 2
|
|
|
|
images = list(Path("./export/").glob("*.png"))
|
|
images.sort()
|
|
image_count = len(images)
|
|
total_image_height = image_count / cols * per_image_height
|
|
if image_count / cols * per_image_height > total_image_height:
|
|
total_image_height += per_image_height
|
|
total_image_width = image_count / cols * per_image_width
|
|
|
|
resized_images = []
|
|
|
|
for path in images:
|
|
image = Image.open(path)
|
|
image = image.resize((per_image_width, per_image_height))
|
|
resized_images.append((path, image))
|
|
|
|
new_image = Image.new(
|
|
"RGB",
|
|
(per_image_width * cols, int(per_image_height * image_count / cols)),
|
|
(250, 250, 250),
|
|
)
|
|
|
|
current_col = 0
|
|
current_row = 0
|
|
current_image_number = 0
|
|
for path, image in resized_images:
|
|
printr("Merging image {}".format(path))
|
|
current_row = int(current_image_number / cols)
|
|
left = int((current_col) * per_image_width)
|
|
top = int(per_image_height * current_row)
|
|
new_image.paste(image, (left, top))
|
|
new_image.save("export/1_graphs_compliation.png")
|
|
|
|
current_col += 1
|
|
current_image_number += 1
|
|
if current_col == cols:
|
|
current_col = 0
|