# Alexandre B A Villares - https://abav.lugaralgum.com/sketch-a-day
SKETCH_NAME, OUTPUT = "sketch_190217a", ".gif"
"""
Like 190215 but using the rounded poly from 190216
"""
from collections import namedtuple
import random as rnd
import copy as cp
from gif_exporter import gif_export
from arcs import var_bar, poly_rounded2
add_library('peasycam')
add_library('GifAnimation')

SPACING, MARGIN = 200, 100
X_LIST, Y_LIST = [], []  # listas de posições para elementos
desenho_atual, outro_desenho, desenho_inter, desenho_inicial = [], [], [], []
NUM_NODES = 2  # número de elementos do desenho / number of nodes
Node = namedtuple(
    'Node', 'x y tam points_to')
save_frames = False
strips = 10

def setup():
    smooth(16)
    size(600, 600, P3D)
    colorMode(HSB)
    rectMode(CENTER)
    noFill()
    cam = PeasyCam(this, 600)
    X_LIST[:] = [x for x in range(MARGIN, 1 + width - MARGIN, SPACING)]
    Y_LIST[:] = [y for y in range(MARGIN, 1 + height - MARGIN, SPACING)]
    novo_desenho(desenho_atual)
    desenho_inicial[:] = cp.deepcopy(desenho_atual)
    println("'s' to save, and 'n' for a new drawing")

def draw():
    global strips, desenho_atual, outro_desenho
    translate(-width/2, -height/2)
    background(200)
    fc = -1 # frameCount % 200 - 100
    if fc < 0:
        desenho = desenho_atual
    elif 0 <= fc < 99:
        # if frameCount % 10 == 0:.
        make_inter_nodes(map(fc, 0, 100, 0, 1))
        desenho = desenho_inter
    elif fc == 99:
        desenho_atual, outro_desenho = outro_desenho, desenho_atual
        desenho = desenho_atual
        if not mousePressed:
            make_nodes_point(outro_desenho)
        else:
            print("will reset")
            outro_desenho[:] = cp.deepcopy(desenho_inicial)

    desenho_plot(desenho)
    # if frameCount % 4 == 0:
    #    gif_export(GifMaker, filename=SKETCH_NAME, delay=200)
    # #strips += 1

def keyPressed():
    global save_frames
    if key == 'g':
        gif_export(GifMaker, filename=SKETCH_NAME)
        # save_frames = not save_frames
        # print "Saving "+repr(save_frames)
    if key == 'r':
        make_nodes_point(desenho_atual)
    if key == 'n':
        novo_desenho(desenho_atual)
    if key == ' ':
        background(200)

def novo_desenho(desenho):
    """
    esvazia a lista elementos (setas e linhas) do desenho anterior
    clears the list of nodes and creates a a new drawing appending desenho_atual,
    a list of nodes/drawing elements: specials, connecting lines and lonely nodes
    """
    desenho[:] = []
    for _ in range(NUM_NODES):
        desenho.append(new_node())
    make_nodes_point(desenho)
    outro_desenho[:] = cp.deepcopy(desenho)
    make_nodes_point(outro_desenho)


def new_node():
    return Node(                   # elemento/"nó" uma namedtuple com:
        rnd.choice(X_LIST),        # x
        rnd.choice(Y_LIST),        # y
        rnd.choice([2, 4, 6]),     # tam
        []  # points_to... (lista com ref. a outro elem.))
    )

def make_nodes_point(desenho):
    for node in desenho:  # para cada elemento do desenho
        node.points_to[:] = []
        node.points_to.append(new_node())
        node.points_to.append(new_node())

def desenho_plot(d):
    for node in d:
        p1, p2 = node.points_to  # se estiver apontando para alguém
        # strokeWeight(node.tam)
        with pushMatrix():
            for i in range(10):
                # stroke(0 + i * 8)
                strokeWeight(2)
                translate(0, 0, -node.tam*5)
                rs = [(node.tam + i) * strips,
                (p1.tam + i) * strips,
                (p2.tam + i) * strips,]
                poly_rounded2([node, p1, p2], rs)

def mouseWheel(E):
    global strips
    strips += int(E.getAmount())
    print(strips)

def make_inter_nodes(amt):
    desenho_inter[:] = []
    for n1, n2 in zip(desenho_atual, outro_desenho):
        desenho_inter.append(Node(                   # elemento/"nó" uma namedtuple com:
            n1.x,        # x
            n1.y,        # y
            n1.tam,     # tam 
            # cp.deepcopy(n1.points_to)
            [PVector(lerp(p1.x, p2.x, amt), lerp(p1.y, p2.y, amt))
             for p1, p2 in zip(n1.points_to, n2.points_to)]
        ))

# print text to add to the project's README.md
def settings():
    println(
        """
![{0}](2019/{0}/{0}{1})

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