# png_function_art.py

# png_function_art.py

from Tkinter import *
from PIL import Image, ImageTk
import math
import random

ww = 300
hh = 300

random.seed()

class X:
    def eval(self, x, y):
        return x

    def __str__(self):
        return "x"

class Y:
    def eval(self, x, y):
        return y

    def __str__(self):
        return "y"

class SinPi:
    def __init__(self, prob):
        self.arg = buildExpr(prob * prob)

    def __str__(self):
        return "sin(pi*" + str(self.arg) + ")"

    def eval(self, x, y):
        return math.sin(math.pi * self.arg.eval(x,y))

class CosPi:
    def __init__(self, prob):
        self.arg = buildExpr(prob * prob)

    def __str__(self):
        return "cos(pi*" + str(self.arg) + ")"

    def eval(self, x, y):
        return math.cos(math.pi * self.arg.eval(x,y))

class Times:
    def __init__(self, prob):
        self.lhs = buildExpr(prob * prob)
        self.rhs = buildExpr(prob * prob)

    def __str__(self):
        return str(self.lhs) + "*" + str(self.rhs)

    def eval(self, x, y):
        return self.lhs.eval(x,y) * self.rhs.eval(x,y)

def buildExpr(prob = 0.99):
    if random.random() < prob:
        return random.choice([SinPi, CosPi, Times])(prob)
    else:
        return random.choice([X, Y])()

def plotIntensity(exp, pixelsPerUnit = 149):
     canvasWidth = 2 * pixelsPerUnit + 1
     img = Image.new("L",(ww, hh))

     for py in range(canvasWidth):
          for px in range(canvasWidth):
                # Convert pixel location to [-1,1] coordinates
                x = float(px - pixelsPerUnit) / pixelsPerUnit
                y = -float(py - pixelsPerUnit) / pixelsPerUnit
                z = exp.eval(x,y)

                # Scale [-1,1] result to [0,255].
                intensity = int(z * 127.5 + 127.5)
                img.putpixel((px,py), intensity)

     return img

def plotColor(redExp, greenExp, blueExp, pixelsPerUnit = 149):
     redPlane   = plotIntensity(redExp, pixelsPerUnit)
     greenPlane = plotIntensity(greenExp, pixelsPerUnit)
     bluePlane  = plotIntensity(blueExp, pixelsPerUnit)
     return Image.merge("RGB", (redPlane, greenPlane, bluePlane))

def makeImage(numPics = 20):
    with open("C:/py/artwork/eqns.txt", 'w') as eqnsFile:
        for i in range(numPics):
            redExp = buildExpr()
            greenExp = buildExpr()
            blueExp = buildExpr()

            eqnsFile.write("img" + str(i) + ":\n")
            eqnsFile.write("red = " + str(redExp) + "\n")
            eqnsFile.write("green = " + str(greenExp) + "\n")
            eqnsFile.write("blue = " + str(blueExp) + "\n\n")

            image = plotColor(redExp, greenExp, blueExp)
            id = "C:/py/artwork/" + str(i) + ".png"
            print id
            image.save(id, "PNG")


makeImage(50)