ImageFactory.py

import sys
# from os.path import isfile, join
import sys
import glob
import time # for timing
# from math import sqrt, tan, sin, cos, pi, ceil, floor, acos, atan, asin, degrees, radians, log, atan2
from random import *

from PIL import Image, ImageDraw

from ImageTools import *

''' This module is for producing procedural images
	Pass in a canvas to be drawn over.
	1. intermediate images are cached to the file system (Note: manual maintenance required. Track file system usage.
	2. final images are cached too
	3. cached images can be selected
	4. the image generator is a python module
	
	Extends work previously done on ABrush
'''

def cacheImage(img):
	pathImages = "images/"
	filename =  pathImages+"image_"+str(randint(10000000000,99999999999))
	img.save(filename+".png") # cache it

def cacheImageRecover():
	extension = "images/*.png"

	# get a list of the available images (from the file system)
	images = glob.glob(extension)
	imageName = images[randint(0,len(images)-1)]
	print "Recovering cached image: "+imageName
	return Image.open(imageName)
	
def makeRandomImage(width,height):

	img = None
	# Recover an image from the cache
	if random() < 0.5:
		try:
			img = cacheImageRecover()
		except:
			print sys.exc_info()

	# Otherwise make a new one
	if img == None:
		img = Image.new('RGBA', size=(width, height), color=(0, 0, 0, 255))

	try:
		drawImageRandom(img)
		# img.save(filename+"_blendtest.png")
		imageNormalize(img)
		if random() > 0.95:
			colour = (0,0,0,0)
			imageCarveCircle(img,colour)
		# Post-processing
		if random() > 0.95:
			img = imageAvgDiff(img) # Edge detection
			imageNormalize(img)
		cacheImage(img)
	except:
		print sys.exc_info()


	return img

def drawImageRandom(canvas):
	''' Choose a generator at random from the file system, invoke it, and get an image back in return
	'''

	extension = "Gen_*.py"
	
	# get a list of the available generators (from the file system)
	imageGenerators = glob.glob(extension)
	print 'Found %s image generators:' % (len(imageGenerators))
	if random() > 0.9:
		for fileName in imageGenerators:
			print fileName
	# call one of them
	theModule = imageGenerators[randint(0,len(imageGenerators)-1)]
	print "Selected image generator: "+theModule
	
	module = __import__(theModule[:-3])
	#module = __import__(path+theModule[:-3]) # Courtesy @CodeWarrior0 via https://github.com/mcedit/mcedit/blob/master/editortools/filter.py
	# Return the resulting image
	width = canvas.size[0]
	height = canvas.size[1]
	img = Image.new('RGBA', size=(width, height), color=(0,0,0, 255))
	module.draw(img)
	imageBlend(canvas,img)
	imageNormalize(canvas)
	return canvas

'''
pathImages = "images/"
while(True):
	img = Image.new('RGBA', size=(400, 400), color=(0, 0, 0, 255))
	drawImageRandom(img)
	drawImageRandom(img)
	filename =  pathImages+"TestImage_"+str(randint(1000000000,9999999999))
	# img.save(filename+"_blendtest.png")
	imageNormalize(img)
	colour = (0,0,0,0)
	imageCarveCircle(img,colour)
	img.save(filename+"_normalized.png")
'''