"""Grid study"""

from random import choice

add_library('GifAnimation')
from gif_exporter import *

from grid import Grid

def setup():
    size(500, 500)
    rectMode(CENTER)
    colorMode(HSB)
    strokeJoin(ROUND)
    strokeWeight(2)
    create_grids()
    frameRate(5)

def create_grids():
    global grids
    grids = []    
    for i in range(10):
        d = int(random(4, 11))
        sp = 20
        x = int(random(-7, 8)) * sp
        y = int(random(-7, 8)) * sp
        si = random(10, 25)
        sh = choice((ELLIPSE, ELLIPSE, RECT, RECT, TRIANGLE, TRIANGLES))
        grids.append(Grid(pos=(x, y),
                           dims=(d, d),
                           space=sp,
                           elem=(sh, si))
                      )
    global ix, iy
    ix, iy, _ = grids[0].pos
    print ix, iy   
        
def draw():
    background(0)
    translate(width / 2., height / 2.)
    # scale(.5, .5)
    
    for g in grids:
        g.update()
        
    saveFrame("###.png")
    fx, fy, _ = grids[0].pos
    if (ix, iy) == (fx, fy):
        exit()

def keyPressed():
    if key == "s":
        saveFrame("####.png")
    if key == " ":
        create_grids()