# Alexandre B A Villares - https://abav.lugaralgum.com/sketch-a-day
SKETCH_NAME = "s129"  # 180509
GRID_SIDE = 50  # colunas na grade

add_library('gifAnimation')
from gif_exporter import *

def setup():
    global input, GIF_EXPORT
    size(600, 600, P2D)
    frameRate(3)
    # textAlign(CENTER, CENTER)
    noFill()
    Cell.CELLS = []
    GIF_EXPORT = False
    create_grid()
    
def draw():
    # background(64, 127, 127, 128)
    fill(64, 64, 127, 128)
    rect(0, 0, width, height)

    for cell in Cell.CELLS:
        cell.draw_()
    # uncomment next lines to export GIF
    global GIF_EXPORT
    if GIF_EXPORT:
        GIF_EXPORT = gif_export(GifMaker,
                                frames=50,
                                delay=200,
                                filename=SKETCH_NAME)
    Cell.update()


def create_grid():
    global SPAC_SIZE
    # espaçamento entre os elementos
    SPAC_SIZE = int(width / GRID_SIDE)
    # empty list
    Cell.CELLS[:] = []
    # append new cells to grid
    for x in range(GRID_SIDE):  # um x p/ cada coluna
        for y in range(GRID_SIDE):  # um y p/ cada linha
            new_cell = Cell(x, y)
            if random(10) > 2:
                new_cell.status = 1
            Cell.CELLS.append(new_cell)
    Cell.set_neighbours()

def keyPressed():
    global GIF_EXPORT
    if key == 'p':  # save PNG
        saveFrame("####.png")
    if key == 'g':  # save GIF
        GIF_EXPORT = True
    if key == 'h':
        input.help()
    if key == 'r':
        create_grid()


class Cell():
    CELLS = []

    def __init__(self, x, y):

        self.x = x
        self.y = y
        self.status = 0
        self.next = 0
        self.nc = 6
        self.nl = []
        self.color_ = color(255, 0, 0)

    def update_nc(self):
        '''
        update neighbour count (self.nc)
        AND apply the 'game' rule
        '''
        self.nc = self.count_neighbours()
        # cells with 1, 4, 5 or 6 neighbours die
        if self.nc < 2:
            self.next = 0
        elif self.nc > 5:
            self.next = 0
        else:
            self.next = 1

    @classmethod
    def set_neighbours(cls):
        for cell in cls.CELLS:
            cell.nl = cell.neighbours()

    @classmethod
    def update(cls):
        for cell in cls.CELLS:
            cell.status = cell.next
        for cell in cls.CELLS:
            cell.update_nc()
            cell.color_ = map(cell.nc, 0, 6, 0, 255)

    def draw_(self):
        if self.status:
            fill(self.color_)
            noStroke()
            s = SPAC_SIZE * 1.2
            v = SPAC_SIZE * 1.5
            h = SPAC_SIZE * sqrt(3)
            if self.y % 2:
                x = self.x * h + h / 4
            else:
                x = self.x * h - h / 4
            y = self.y * v
            ellipse(x, y, s, s)
            # pointy_hexagon(x, y, s)
            if dist(mouseX, mouseY, x, y) < s:
                fill(255)
                text(str(self.nc), x, y)
                println(self.nl)

    def count_neighbours(self):
        count = 0
        for cell in self.nl:
            if cell and cell.status:
                count += 1
        return count

    def neighbours(self):
        neighbour_list = []
        offset_list = [(-1, 0), (0, -1), (0, 1), (1, 0), (1, -1), (1, 1)]
        for offset_x, offset_y in offset_list:
            neighbour = Cell.item_at_x_y(self.x + offset_x,
                                         self.y + offset_y,
                                         Cell.CELLS,
                                         GRID_SIDE, GRID_SIDE)
            neighbour_list.append(neighbour)
        return neighbour_list

    @staticmethod
    def item_at_x_y(x, y, collection, w, h):
        if 0 < x < w and 0 < y < h:
            return collection[x + y * w]
        return None