import time

import board
import busio
from adafruit_neotrellis.neotrellis import NeoTrellis
from adafruit_neotrellis.multitrellis import MultiTrellis
from neotrellism4 import NeoTrellisM4
import conway
import adafruit_adxl34x

ACCEL_THRESHOLD = 100


#create the i2c object for the trellis
I2C = busio.I2C(board.SCL, board.SDA)

"""create the trellis. This is for a 2x2 array of TrellisM4 (first row) with
2 Neotrellis (second row).

 [ NeoM4_left | NeoM4_right ]
  neotrellis0 | neotrellis1
"""

trellim4_left = NeoTrellisM4()
trellim4_right = NeoTrellisM4(left_part=trellim4_left)
trelli = [
    [trellim4_left, trellim4_right],
    [NeoTrellis(I2C, False, addr=0x2F), NeoTrellis(I2C, False, addr=0x2E)]
    ]

trellis = MultiTrellis(trelli)

#some color definitions
OFF = (0, 0, 0)
ON = (100, 100, 80)
# RED = (127, 0, 0)
# YELLOW = (127, 75, 0)
# GREEN = (0, 127, 0)
# CYAN = (0, 127, 127)
# BLUE = (0, 0, 127)
# PURPLE = (90, 0, 127)


I2C_ACCEL = busio.I2C(board.ACCELEROMETER_SCL, board.ACCELEROMETER_SDA)
accelerometer = adafruit_adxl34x.ADXL343(I2C_ACCEL)

# Init universe
univers = [[False for x in range(8)] for y in range(8)]

universe = conway.GLIDER

#this will be called when button events are received
def genesis_callback(xcoord, ycoord, edge):
    #turn the LED on when a rising edge is detected
    if edge == NeoTrellis.EDGE_RISING:
        universe[xcoord][ycoord] = not(universe[xcoord][ycoord])
        if univers[xcoord][ycoord]:
            trellis.color(xcoord, ycoord, ON)
        else:
            trellis.color(xcoord, ycoord, OFF)
    conway.draw_universe(trellis, universe)


def evo_callback(xcoord, ycoord, edge):
    pass


# Init. all the keys for Genesis
for x in range(8):
    for y in range(8):
        #activate rising edge events on all keys
        trellis.activate_key(x, y, NeoTrellis.EDGE_RISING)
        trellis.set_callback(x, y, genesis_callback)

trellis.sync()
conway.draw_universe(trellis, universe)

accelerometer.enable_motion_detection(threshold=ACCEL_THRESHOLD)

# Genesis
while not(accelerometer.events["motion"]):
     #the trellis can only be read every 17 millisecons or so
     trellis.sync()
     time.sleep(.02)

print("The end of Genesis, time to evolve...")

# Init. all the keys for Evolution
for x in range(8):
    for y in range(8):
        #activate rising edge events on all keys
        trellis.activate_key(x, y, NeoTrellis.EDGE_RISING)
        trellis.set_callback(x, y, evo_callback)

trellis.sync()
conway.draw_universe(trellis, universe)

generation = 0

#Evolution
while True:
    new_universe = [[False for x in range(8)] for y in range(8)]
    ext_universe = conway.extend_universe(universe)
    for x in range(8):
        for y in range(8):
            n = conway.living_neighbour_count(x, y, ext_universe)
            if n == 3: 
                new_universe[x][y] = True
                # print(x, y, n, "alive")
            elif n == 2 and universe[x][y]:
                new_universe[x][y] = True
                # print(x, y, n, "still alive")
            else:
                new_universe[x][y] = False
                # print(x, y, n, "dead")
    generation += 1
    print(generation)
    universe = new_universe.copy()

    conway.draw_universe(trellis, universe)