#include <avr/io.h>
#include <avr/interrupt.h>
#include <util/delay.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdlib.h>

#include "lib/meta.h"
#include "lib/arduino_pins.h"
#include "lib/calc.h"
#include "lib/colors.h"
#include "lib/ws2812.h"
#include "lib/adc.h"

#define DEBO_CHANNELS 6
#define DEBO_TICKS 1  // in 0.01s

#include "lib/debounce.h"

#define WIDTH 6
#define HEIGHT 5

// number of cards
#define CARD_COUNT (WIDTH * HEIGHT)

// when the "small" pin is DOWN, only this many cards are dealt - on the same board size
#define CARD_COUNT_SMALL WIDTH * 2


// --- Pin assignments (see pins.h) ---

// RGB LED strip data line
#define WS1   D10

// Buttons (to ground)
#define BTN_LEFT     D2
#define BTN_RIGHT    D3
#define BTN_UP       D4
#define BTN_DOWN     D5
#define BTN_SELECT   D6
#define BTN_RESTART  D7

// Connect to GROUND to get small board size (for kids ^_^)
#define FLAG_SMALL   D12

// Debouncer channels for buttons
// (Must be added in this order to debouncer)
#define D_LEFT      0
#define D_RIGHT     1
#define D_UP        2
#define D_DOWN      3
#define D_SELECT    4
#define D_RESTART   5

// [ IMPORTANT ]
// Pin A0 must not be connected, it is used to get
// entropy for the random number generator



// number of pairs
#define PAIR_COUNT (CARD_COUNT / 2)

// assert valid board size
#if CARD_COUNT % 2 == 1
# error "CARD_COUNT is not even!"
#endif

#if CARD_COUNT_SMALL % 2 == 1
# error "CARD_COUNT_SMALL is not even!"
#endif

// color palette
const xrgb_t COLORS[] = {
	rgb24_xrgbc(0x0000CC), // full blue
	rgb24_xrgbc(0x0BEE00), // green
	rgb24_xrgbc(0xFFFF00), // yellow
	rgb24_xrgbc(0xFF0000), // red
	rgb24_xrgbc(0x00CCFF), // cyan
	rgb24_xrgbc(0xFF3300), // orange
	rgb24_xrgbc(0xFF00FF), // magenta
	rgb24_xrgbc(0x00FF88), // emerald
	rgb24_xrgbc(0x5FBA00), // yellow-green
	rgb24_xrgbc(0xFF7700), // tangerine yellow/orange
	rgb24_xrgbc(0x4400FF), // blue-purple
	rgb24_xrgbc(0xD70053), // wine
	rgb24_xrgbc(0xED1B24), // firetruck red
	rgb24_xrgbc(0xFF6D55), // salmon?
	rgb24_xrgbc(0xFF4C4C), // skin pink
};


// Prototypes
void render();
void update();
void deal_cards();
void safe_press_arrow_key(uint8_t n);


/** Program initialization */
void SECTION(".init8") init()
{
	// Randomize RNG
	adc_init();
	srand(adc_read_word(0));

	// led strip data
	as_output(WS1);

	// gamepad buttons
	as_input_pu(BTN_LEFT);
	as_input_pu(BTN_RIGHT);
	as_input_pu(BTN_UP);
	as_input_pu(BTN_DOWN);
	as_input_pu(BTN_SELECT);
	as_input_pu(BTN_RESTART);

	as_input_pu(FLAG_SMALL); // when LOW, use smaller board.

	// add buttons to debouncer
	debo_add_rev(BTN_LEFT);
	debo_add_rev(BTN_RIGHT);
	debo_add_rev(BTN_UP);
	debo_add_rev(BTN_DOWN);
	debo_add_rev(BTN_SELECT);
	debo_add_rev(BTN_RESTART);

	// setup timer
	TCCR0A = _BV(WGM01); // CTC
	TCCR0B = _BV(CS02) | _BV(CS00);  // prescaler 1024
	OCR0A = 156;  // interrupt every 10 ms
	sbi(TIMSK0, OCIE0A);

	// prepare game board
	deal_cards();

	// enable timer interrupts (update & render)
	sei();
}



/** Tile state enum */
typedef enum {
	SECRET,
	REVEALED,
	GONE
} tilestate_t;


/** Tile struct */
typedef struct {
	uint8_t color;  // color index from COLORS[]
	tilestate_t state;  // state of the tile (used for render)
} tile_t;


// board tiles
tile_t board[CARD_COUNT];


/** Randomly place pairs of cards on the board */
void deal_cards()
{
	// clear the board
	for (uint8_t i = 0; i < CARD_COUNT; ++i) {
		board[i] = (tile_t) { .color = 0, .state = GONE };
	}

	const uint8_t dealt_cards = get_pin(FLAG_SMALL) ? CARD_COUNT : CARD_COUNT_SMALL;

	// for all pair_COUNT
	for (uint8_t i = 0; i < (dealt_cards / 2); ++i) {
		// for both cards in pair
		for (uint8_t j = 0; j < 2; j++) {
			// loop until empty slot is found
			while(1) {
				const uint8_t pos = rand() % dealt_cards;

				if (board[pos].state == GONE) {
					board[pos] = (tile_t) { .color = i, .state = SECRET };
					break;
				}
			}
		}
	}
}


/** timer 0 interrupt vector */
ISR(TIMER0_COMPA_vect)
{
	debo_tick();  // poll debouncer
	update();  // update game state
	render();
}


// player cursor position
uint8_t pos_x = 0;
uint8_t pos_y = 0;
#define Cursor (pos_y * WIDTH + pos_x)

uint8_t animframe = 0;

bool hide_timeout_match;
uint8_t hide_timeout = 0;

// Game state
uint8_t tiles_revealed = 0;
uint8_t tile1;
uint8_t tile2;

// length of pulse animation (in 10ms)
#define F_ANIM_LEN 20
#define HIDE_TIME 100

// length of button holding before it's repeated (in 10ms)
#define BTNHOLD_REPEAT 20

uint8_t btn_hold_cnt[DEBO_CHANNELS];


/** Handle a button press event */
void button_click(uint8_t n)
{
	switch (n) {
		case D_UP:
			dec_wrap(pos_y, 0, HEIGHT);
			break;

		case D_DOWN:
			inc_wrap(pos_y, 0, HEIGHT);
			break;

		case D_LEFT:
			if (pos_x == 0) dec_wrap(pos_y, 0, HEIGHT);  // go to previous row
			dec_wrap(pos_x, 0, WIDTH);
			break;

		case D_RIGHT:
			if (pos_x == WIDTH - 1) inc_wrap(pos_y, 0, HEIGHT);  // go to next row
			inc_wrap(pos_x, 0, WIDTH);
			break;

		case D_SELECT:
			if (tiles_revealed == 2) break;  // two already shown
			if (board[Cursor].state != SECRET) break;  // selected tile not secret

			// reveal a tile
			if (tiles_revealed < 2) {
				board[Cursor].state = REVEALED;
				tiles_revealed++;

				if(tiles_revealed == 1) {
					tile1 = Cursor;
				} else {
					tile2 = Cursor;
				}
			}

			// Check equality if it's the second
			if (tiles_revealed == 2) {
				hide_timeout_match = (board[tile1].color == board[tile2].color);
				hide_timeout = HIDE_TIME;
			}

			break;

		case D_RESTART:
			deal_cards();
			if (board[Cursor].state == GONE)
				safe_press_arrow_key(D_RIGHT);

			break;
	}
}


/** Press arrow key, skip empty tiles */
void safe_press_arrow_key(uint8_t n)
{
	// preserve old position
	const uint8_t x1 = pos_x;
	const uint8_t y1 = pos_y;

	if (n == D_LEFT || n == D_RIGHT) {

		// traverse all buttons
		for (uint8_t i = 0; i < CARD_COUNT; i++) {
			button_click(n);

			if (board[Cursor].state != GONE) return;
		}

	} else {

		for (uint8_t i = 0; i < HEIGHT; i++) {

			// Go up/down
			button_click(n);
			if (board[Cursor].state != GONE) return;

			for (int8_t x = 0; x < WIDTH; x++) {

				if ((int8_t) pos_x - x >= 0) {
					if (board[Cursor - x].state != GONE) {
						pos_x -= x;
						return;
					}
				}

				if ((int8_t) pos_x + x < WIDTH) {
					if (board[Cursor + x].state != GONE) {
						pos_x += x;
						return;
					}
				}
			}
		}
	}

	// restore original position
	pos_x = x1;
	pos_y = y1;
}


#define is_arrow_key(id) ((id) == D_LEFT || (id) == D_RIGHT || (id) == D_UP || (id) == D_DOWN)


/** Update game (every 10 ms) */
void update()
{
	// handle buttons (with repeating when held down)
	for (uint8_t i = 0; i < DEBO_CHANNELS; i++) {
		if (debo_get_pin(i)) {
			if (btn_hold_cnt[i] == 0) {
				if (is_arrow_key(i)) {
					safe_press_arrow_key(i);
				} else {
					button_click(i);
				}
			}

			// non-arrows wrap to 1 -> do not generate repeated clicks
			inc_wrap(btn_hold_cnt[i], is_arrow_key(i) ? 0 : 1, BTNHOLD_REPEAT);

		} else {
			btn_hold_cnt[i] = 0;
		}
	}

	// game logic - hide or remove cards when time is up
	if (hide_timeout > 0) {
		if (--hide_timeout == 0) {
			if (hide_timeout_match) {
				// Tiles removed from board
				board[tile1].state = GONE;
				board[tile2].state = GONE;

				if (board[Cursor].state == GONE) {
					// move to some other tile
					// try not to change row if possible
					if ((Cursor % WIDTH) == (WIDTH - 1))
						safe_press_arrow_key(D_LEFT);
					else
						safe_press_arrow_key(D_RIGHT);
				}
			} else {
				// Tiles made secret again
				board[tile1].state = SECRET;
				board[tile2].state = SECRET;
			}

			tiles_revealed = 0; // no revealed
		}
	}

	// Animation for pulsing the active color
	inc_wrapi(animframe, 0, F_ANIM_LEN * 2);
}

// LED off
#define BLACK rgb24_xrgb(0x000000)
// LED on - secret tile
#define WHITE rgb24_xrgb(0x555555)

// colors to be displayed
xrgb_t screen[CARD_COUNT];


/** Update screen[] and send to display */
void render()
{
	// Prepare screen (compute colors)
	for (uint8_t i = 0; i < CARD_COUNT; i++) {
		// get tile color to show
		switch (board[i].state) {
			case SECRET:
				screen[i] = WHITE;
				break;

			case REVEALED:
				screen[i] = COLORS[board[i].color];
				break;

			default:
			case GONE:
				screen[i] = BLACK;
				break;
		}

		// pulse active tile
		if (i == Cursor) {
			uint16_t mult;

			if (animframe < F_ANIM_LEN) {
				mult = animframe;
			} else {
				mult = (F_ANIM_LEN * 2) - animframe;
			}

			screen[i] = (xrgb_t) {
				.r = (uint8_t) ((((uint16_t) screen[i].r) * mult) / F_ANIM_LEN),
				.g = (uint8_t) ((((uint16_t) screen[i].g) * mult) / F_ANIM_LEN),
				.b = (uint8_t) ((((uint16_t) screen[i].b) * mult) / F_ANIM_LEN),
			};
		}
	}

	// Send to LEDs
	ws_send_xrgb_array(WS1, screen, CARD_COUNT);
	ws_show();
}


void main()
{
	while(1);  // Timer does everything
}