rC3-WA/tiles_png/community/cert/scripts/check_imgs.py

89 lines
2.8 KiB
Python
Raw Normal View History

2020-12-21 21:13:39 +01:00
#!/usr/bin/env python3
from PIL import Image
import sys
import os
BASE_DIR = 'imgs'
COMPOSE_TILESHEETS = True
TILESHEET_WIDTH = 512
TILESHEET_HEIGTH = 2048
class Row():
def __init__(self, width=TILESHEET_WIDTH, height=32):
self.width = width
self.height = height
self.remainingWidth = width
self.sprites = []
def append(self, Img):
x,y = Img.size
if (self.remainingWidth - x) < 0:
raise Exception('ImageTooLarge')
self.height = max(self.height, y)
if self.height > TILESHEET_HEIGTH:
raise Exception("Tilesheet full")
self.remainingWidth = self.remainingWidth - x
self.sprites.append(Img.convert(mode='RGBA'))
return self.remainingWidth
def render(self):
row = Image.new('RGBA', (self.width, self.height),(255,255,255,0))
x_offset = 0
for img in self.sprites:
x,y = img.size
row.alpha_composite(img, (x_offset,0))
x_offset += x
return row
class Spritesheet():
def __init__(self, width=TILESHEET_WIDTH, height=TILESHEET_HEIGTH):
self.width = width
self.height = height
self.rows = []
def append(self, Img):
x,y = Img.size
# self.height += y
self.rows.append(Img)
def render(self):
sheet = Image.new('RGBA', (self.width, self.height),(255,255,255,0))
y_offset = 0
for img in self.rows:
x,y = img.size
sheet.alpha_composite(img, (0,y_offset))
y_offset += y
return sheet
if __name__ == "__main__":
print("Check images dimensions")
print('Absolute path:', os.getcwd() + '/imgs')
for root, dirs, files in os.walk(BASE_DIR):
for file in files:
if file.endswith('.png'):
x,y = Image.open(os.path.join(root, file)).size
if not(x % 32 == y % 32 == 0 and x <= TILESHEET_WIDTH):
sys.exit("%s has wrong dimensions." % file)
if COMPOSE_TILESHEETS:
print("Generate Tilesheets")
for dir in os.listdir(BASE_DIR):
if dir != "tilesheets":
name = dir
print("Processing:", dir)
sheet = Spritesheet()
row = Row()
for file in sorted(os.listdir(os.path.join(BASE_DIR, dir)),key=str.lower):
try:
if file.endswith('.png'):
row.append(Image.open(os.path.join(BASE_DIR, dir, file)))
except:
# new row
sheet.append(row.render())
row = Row()
row.append(Image.open(os.path.join(BASE_DIR, dir, file)))
sheet.append(row.render())
sheet.render().save(os.path.join(BASE_DIR, 'tilesheets', name + '.png'))