# Alexandre B A Villares - https://abav.lugaralgum.com/sketch-a-day
# Trying rules by Leopoldo Leal

add_library('VideoExport')

from random import choice, shuffle, sample
from itertools import product, permutations, combinations
from forms import b_poly_arc_augmented
from line_geometry import *

NUM_POINTS = 6
BORDER = 150
SIZE = 100
RDS = 50
save_frame = False
ensambles = []
st = 0

def setup():
    size(600, 600, P3D)
    ensambles[:] = create_points()
    frameRate(15)
    # Video Export
    global ve
    ve = VideoExport(this)
    ve.setFrameRate(15)
    ve.startMovie()

def create_points():
    """ non intersecting poly """
    global grid
    good = False
    grid = list(product(range(BORDER, width - BORDER + 1, SIZE),
                   range(BORDER, height - BORDER + 1, SIZE)))
    points = sample(grid, NUM_POINTS)
    combos = set(permutations(points, NUM_POINTS))
    for pts in combos:
        for _ in range(len(pts)-1):
            pts = pts[1:] + (pts[0],)
            combos.discard(pts)
            combos.discard(pts[::-1])            
    ens = []
    for pts in combos:
        if not intersecting(pts):
            ens.append(pts) 
    # ens = list(pts)
    print(len(ens))
    shuffle(ens)
    return ens

def draw():
    background(0)
    translate(height/2, width/2)
    rotateY(frameCount / 100.)
    translate(-height/2, -width/2)
    if ensambles:
        draw_ensamble(ensambles[st])
    if save_frame:
        ve.saveFrame()
        
def draw_ensamble(pts): 
    noFill()
    stroke(255)
    strokeWeight(4)
    translate(0, 0, -8 * 10)
    for i in range(8):
        colorMode(HSB)
        stroke(i * 16, 255, 255)
        translate(0, 0, 20)
        b_poly_arc_augmented(pts, [i * 10] * NUM_POINTS)
        # for p in pts:
        #     ellipse(p[0], p[1], i * 10 , i * 10)

def keyPressed():
    global st, save_frame
    if key == "p":
        saveFrame("####.png")
    if key == " ":
        ensambles[:] = create_points()
        st = 0
    if key == "m":
        st = (st + 1) % len(ensambles)
    if key == "e":
        ve.endMovie()
    if key == "s":
        save_frame = not save_frame
        println("aave frame: {}".format(save_frame))


def settings():
    """ print markdown to add at the sketc-a-day page"""
    from os import path
    global SKETCH_NAME
    SKETCH_NAME = path.basename(sketchPath())
    OUTPUT = ".png"
    println(
        """
![{0}]({2}/{0}/{0}{1})

[{0}](https://github.com/villares/sketch-a-day/tree/master/{2}/{0}) [[Py.Processing](https://villares.github.io/como-instalar-o-processing-modo-python/index-EN)]
""".format(SKETCH_NAME, OUTPUT, year())
    )