User:Dyne

#!/usr/bin/python
'''This simple script takes the filename of an image that is a multiple of the tilesize (256 by default), and splits it into tiles of that size.

USAGE: python terrain-split.py infile1 [infile2 ... infileN]

Note that any extra pixels in the original image beyond an even multiple of the tilesize are ignored.
'''

__interactive__ = 1

import Image, ImageOps
import os.path, sys

tilesize=256

for infile in sys.argv[1:]:
	try:
		myimage = Image.open(infile)
		path, filename = os.path.split(infile)
		basename, extension = os.path.splitext(filename)
		(width,height) = myimage.size
		tileswide = width // tilesize
		tileshigh = height // tilesize
		twremain = width % tilesize  # leftover pixels
		thremain = height % tilesize  # leftover pixels

		print
		print "---------------------------------------------------------"
		print "Input file: " + infile
		print "Image dimensions: %s x %s pixels" % (width,height)
		print "Output: %s x %s tiles (%s files) of %s pixels square" % (tileswide, tileshigh, tileswide*tileshigh, tilesize)
		print "Pixels truncated: %s across, %s down" % (twremain, thremain)
		if __interactive__:
                        print
                        print "Note: Values in [brackets] are defaults."
                        print
                        xin = raw_input("What is the desired grid X coordinate of the lower-left sim [1000]? ")
                        if xin == "":
                                basegridX = 1000
                        else:
                                basegridX = int(xin)
                        yin = raw_input("What is the desired grid Y coordinate of the lower-left sim [1000]? ")
                        if yin == "":
                                basegridY = 1000
                        else:
                                basegridY = int(yin)
		for y in range(tileshigh):
                        print
			print "Tiling Row #%s" % str(y+1)
			for x in range(tileswide):
				# Compute the upper left corner
				xnew = x * tilesize
				ynew = y * tilesize
				yrelative = tileshigh-y-1

				# Decide how to name the file
				if __interactive__:  # the filename will be based on individual names and will include the actual grid coordinates
                                        nameprompt = "What is the name of the sim at row %s, column %s [%s]? " % (y+1, x+1, basename)
                                        simname = raw_input(nameprompt)
                                        if simname == "":
                                                simname = basename
        				newfile = simname + "." + str(x + basegridX) + "." + str(basegridY+yrelative) + extension
                                else:  # the filename of our output tile is based on the input file and includes relative positions
        				newfile = basename + "." + str(x) + "." + str(yrelative) + extension
        			outfile = os.path.join(path, newfile)
				
				# Create the tile
				# Flip the tile along the way (transpose) to account for inverted Y axis.
				borders = (xnew, ynew, xnew+tilesize, ynew+tilesize)
				tile = Image.new("RGB", (tilesize, tilesize))
				region = myimage.crop(borders).transpose(Image.FLIP_TOP_BOTTOM)
				tile.paste(region, (0,0) )

				# Attempt to save it				
				try:
					tile.save(outfile)
					print "Writing: File %s with coords %s - %s" % (outfile, (xnew,ynew), (xnew+tilesize-1, ynew+tilesize-1) )
				except IOError:
					print "Cannot open output file ", outfile
	except IOError:
		print "Cannot open file ", infile
	print "---------------------------------------------------------"