"""
sketch 72 180313 - Alexandre B A Villares
https://abav.lugaralgum.com/sketch-a-day
"""

from __future__ import division
from slider import Slider
add_library('gifAnimation')
from gif_exporter import gif_export

# slider_object = Slider(lowest, highest, initial value)
i = Slider(PI / 6, TWO_PI / 3, QUARTER_PI)  # angle, changes number of sides
j = Slider(1, 4, 2)  # depth, changes number of recursions (fiddled afterwards)
k = Slider(10, 100, 50)  # scale
l = Slider(0, TWO_PI, 0)  # rotation

I, J, K, L = 1, 1, 1, 1  # dummy initial slider global values
# ^ I need these globals because I want to read/draw sliders
#   at the end of the draw loop so they draw on top
#   I should 'decouple' reading and drawing the sliders...

def setup():
    size(600, 600)
    colorMode(HSB)  # makes it easy to cycle colors through Hues...
    noFill()
    # Position Sliders (x, y)
    i.position(20, 20)
    j.position(20, 60)
    k.position(20, 100)
    l.position(20, 140)
    frameRate(10)

def draw():
    global I, J, K, L
    background(200)
    poly_shape(width / 2, height / 2, I, J, K, L)
    # Read and update sliders
    I = i.value()  # from PI/6 to TWO_PI/3.
    J = j.value()  # 1 to 3 + I * 1.5
    K = k.value()  # 10 to 50
    L = l.value()  # 0 to TWO_PI
    # uncomment the next line to save frames!
    #if not frameCount % 10: gif_export(GifMaker, frames=3000)


def poly_shape(x, y, angle, D, sat, rotation):
    if int(D) % 2: stroke(0)
    else: stroke(255)
    strokeWeight(D*2)
    with pushMatrix():
        translate(x, y)
        rotate(rotation)
        radius = D * K
        # create a polygon on a ps PShape object
        ps = createShape()
        ps.beginShape()
        a = 0
        while a < TWO_PI:
            sx = cos(a) * radius
            sy = sin(a) * radius
            ps.vertex(sx , sy)
            a += angle
            ps.stroke(int(sx) % 256, 255, 255)
        ps.endShape(CLOSE)  # end of PShape creation
        #shape(ps, 0, 0)  # Draw the PShape
        if D > 1:  # if the recursion 'distance'/'depth' allows...
            for i in range(ps.getVertexCount()):
                # for each vertex
                pv = ps.getVertex(i)  # gets vertex as a PVector
                arrow(0, 0, pv.x, pv.y,
                  shorter=D*3,
                  tail_func=ellipse,
                  tail_size=D*3)
               
                # recusively call poly_shape with a smaller D
                poly_shape(pv.x, pv.y, angle, D - .90, sat, rotation)
    
            
                                    
def arrow(x1, y1, x2, y2, shorter=0, head=None,
    tail_func=None, tail_size=None):
    """
    O código para fazer as setas, dois pares (x, y),
    um parâmetro de encurtamento: shorter
    e para o tamanho da cabeça da seta: head
    """
    rectMode(CENTER)
    L = dist(x1, y1, x2, y2)
    if not head:
        head = max(L / 10, 10)
    with pushMatrix():
        translate(x1, y1)
        angle = atan2(x1 - x2, y2 - y1)
        rotate(angle)
        offset = shorter / 2
        strokeCap(ROUND)
        line(0, L - offset, -head / 3, L - offset - head)
        line(0, L - offset, head / 3, L - offset - head)
        strokeCap(SQUARE)
        line(0, offset, 0, L - offset)
        
        if tail_func:
            if not tail_size: tail_size = head
            tail_func(0, 0, tail_size, tail_size)