#ShowMondrian.py
""" Illustrates the recursive function Mondrian that produces
randomly colored, randomly tiled rectangles.
"""

from simpleGraphicsE import *
from random import uniform as randu
from random import randint as randi

def RandomColor():
    """ Returns a randomly selected color
    """
    listOfColors = [RED,BLUE,CYAN,MAGENTA]
    return listOfColors[randi(0,len(listOfColors)-1)]

def Mondrian(x,y,L,W,level):
    """ Draws a level-L Mondrian in an LxW rectangle
    with center (x,y).
    """
    if level ==0:
        # Base case. Draw the rectangle with a random color.
        DrawRect(x,y,L,W,stroke=4,color=RandomColor())
    else:
        # Compute the L's and W's and centers of the 4 subrectangles.
        xc = randu(x-L/4,x+L/4)
        L1 = xc-(x-L/2)
        L2 = L-L1
        yc = randu(y-W/4,y+W/4)
        W1 = yc-(y-W/2)
        W2 = W-W1
        # The "southwest" subrectangle:
        Mondrian(x-L/2+L1/2,y-W/2+W1/2 ,L1,W1,level-1)
        # The "northwest" subrectangle:
        Mondrian(x-L/2+L1/2,y+W/2-W2/2 ,L1,W2,level-1)
        # The "northeast" subrectangle:
        Mondrian(x+L/2-L2/2,y+W/2-W2/2 ,L2,W2,level-1)
        # The "Southeas" subrectabngle 
        Mondrian(x+L/2-L2/2,y-W/2+W1/2 ,L2,W1,level-1)

if __name__ == '__main__':
        # Draw a level-1, level-2, level-3, and level4 Mondrian
        MakeWindow(2,labels=False)
        Mondrian(0,0,3.,3.,1)
        MakeWindow(2,labels=False)
        Mondrian(0,0,3.,3.,2)
        MakeWindow(2,labels=False)
        Mondrian(0,0,3.,3.,3)
        MakeWindow(2,labels=False)
        Mondrian(0,0,3.,3.,4)
        ShowWindow()