MTSM/mods/3d_armor/preview_gen.py

#!/usr/bin/python

import os
import sys
import Image

try :
	arg = sys.argv[1]
except IndexError :
	print "Usage: preview_gen.py <index_file>"
	sys.exit(1)

try :
	index = open(arg, "r")
except IOError :
	print "Failed to open index file%s" %s (arg)
	sys.exit(1)

preview = []

for line in index.readlines() :
	if ":" in line :
		line = line.rstrip('\n')
		preview.append(line.split(':'))

print "Generating preview images..."
for fn, place in preview :
	try :
		imi = Image.open(fn)
	except IOError :
		print "Failed to open %s" % (fn)
		sys.exit(1)

	w, h = imi.size
	if h != w / 2:
		print "Incompatible texture size %s" % (fn)
		sys.exit(1)

	s = w / 64
	imo = Image.new("RGBA", (16 * s, 32 * s))
	
	if place == "all" or place == "head" :
		face = (40 * s, 8 * s, 48 * s, 16 * s)
		side_l = (56 * s, 8 * s, 57 * s, 16 * s)
		side_r = (63 * s, 8 * s, 64 * s, 16 * s)
		imo.paste(imi.crop(side_l), (4 * s, 0, 5 * s, 8 * s))
		imo.paste(imi.crop(side_r), (11 * s, 0, 12 * s, 8 * s))
		imo.paste(imi.crop(face), (4 * s, 0, 12 * s, 8 * s))

	if place == "all" or place == "torso" :
		arm = (44 * s, 20 * s, 48 * s, 32 * s)
		body = (20 * s, 20 * s, 28 * s, 32 * s)
		imo.paste(imi.crop(arm), (0 * s, 8 * s, 4 * s, 20 * s))
		imo.paste(imi.crop(arm).transpose(Image.FLIP_LEFT_RIGHT),
				(12 * s, 8 * s, 16 * s, 20 * s))
		imo.paste(imi.crop(body), (4 * s, 8 * s, 12 * s, 20 * s))

	if place == "all" or place == "legs" :
		leg = (4 * s, 20 * s, 8 * s, 32 * s)
		imo.paste(imi.crop(leg), (4 * s, 20 * s, 8 * s, 32 * s))
		imo.paste(imi.crop(leg).transpose(Image.FLIP_LEFT_RIGHT),
				(8 * s, 20 * s, 12 * s, 32 * s))

	if place == "all" or place == "feet" :
		boot = (20 * s, 4 * s, 24 * s, 11 * s)
		imo.paste(imi.crop(boot), (4 * s, 25 * s, 8 * s, 32 * s))
		imo.paste(imi.crop(boot).transpose(Image.FLIP_LEFT_RIGHT),
				(8 * s, 25 * s, 12 * s, 32 * s))

	size = (32 * s, 64 * s)
	imo = imo.resize(size)

	if place == "shield" :
		shield = (0, 0, 16 * s, 16 * s)
		imo.paste(imi.crop(shield), (16 * s, 32 * s, 32 * s, 48 * s))

	outfile = fn.replace(".png", "_preview.png")
	imo.save(outfile)
	print outfile