Pythonで細切れになった画像を一枚にするやつ

この前の逆のやつです。某CTFで使ったのでメモ。

# -*- coding: utf-8 -*-

from PIL import Image
import os
import sys
import math

# 細切れになったファイルが入ってるフォルダのパス
ORIGINAL_FOLDER_PATH = "./pieces/"
# 並べる画像の枚数
IMG_HEIGHT_COUNT = 45
IMG_WIDTH_COUNT = 45
# 1枚のPixel数
IMG_H_PIXEL = 4
IMG_W_PIXEL = 4

# ファイルリストの取得
file_list = os.listdir(ORIGINAL_FOLDER_PATH)
# 生成する画像の生成
dst = Image.new('RGB', (IMG_WIDTH_COUNT * IMG_W_PIXEL, IMG_HEIGHT_COUNT * IMG_H_PIXEL))

count = 0

for list in file_list:
    print("now:" + list)

    # 一コマ分の画像を開く
    im = Image.open(ORIGINAL_FOLDER_PATH + list)

    # 座標計算して貼り付ける
    x = count % IMG_WIDTH_COUNT * IMG_W_PIXEL
    y = math.floor(count/IMG_HEIGHT_COUNT) * IMG_H_PIXEL
    dst.paste(im, (x, y))
    count += 1

# 保存
dst.save('output.png')