#include <avr/io.h>

#include "lights.h"

typedef struct {
	unsigned char mode: 3;
	unsigned char duration: 5;
} pattern_t;

static unsigned char led_counters[N_LEDS];
static pattern_t *led_patterns[N_LEDS];

#define PATTERN_END { 0, 0 }
pattern_t off_pattern[] = {
	{ 0, 0x1F },
	PATTERN_END
};

pattern_t blink_pattern[] = {
	{ 1, 0x4 },
	{ 0, 0x8 },
	PATTERN_END
};

pattern_t mode1_pattern[] = {
	{ 1, 0x1 },
	{ 0, 0x1 },
	{ 4, 0x1 },
	{ 0, 0x1 },
	{ 1, 0x1 },
	PATTERN_END
};

pattern_t boot_pattern[] = {
	{ 1, 0x6 },
	{ 0, 0x6 },
	{ 1, 0x3 },
	{ 0, 0x3 },
	{ 1, 0x2 },
	{ 0, 0x2 },
	{ 1, 0x1 },
	{ 0, 0x1 },
	{ 1, 0x1 },
	{ 0, 0x1 },
	{ 1, 0x1 },
	{ 0, 0x1 },
	{ 1, 0x1 },
	{ 0, 0x1 },
	{ 1, 0x10 },
	{ 0, 0x10 },
	PATTERN_END
};

pattern_t pattern_num[] = {
	{ 1, 0x1 }, /* 10 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  9 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  8 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  7 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  6 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  5 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  4 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  3 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  2 */
	{ 0, 0x4 },
	{ 1, 0x1 }, /*  1 */
	{ 0, 0x1F },
	PATTERN_END
};

static unsigned char test_running;

void pattern_init()
{
	unsigned char i;

	for (i = 0; i < N_LEDS; i++) {
		led_counters[i] = 0;
		led_patterns[i] = off_pattern;
	}
	led_patterns[N_PWMLEDS] = boot_pattern;
	led_counters[N_PWMLEDS] = boot_pattern->duration;
	gpio_set(GPIO_LED2, 1);
	test_running = 0;
}

static pattern_t *number_pattern(unsigned char num)
{
	if (num >= 9)
		num = 9;

	return pattern_num + sizeof(pattern_num)/sizeof(pattern_t)
		- 1 - 2*num;
}

static inline pattern_t *pattern_select(unsigned char n)
{
	if (n < N_PWMLEDS && !pwmled_enabled(n))
		return off_pattern; // Don't mess with non-enabled LEDs

	if (n == 2) {
		if (test_running) {
			log_byte(0xF7);
			log_flush();
			return off_pattern;
		} else {
			test_running = 1;
			log_byte(0xF6);
			return mode1_pattern;
		}
	}
	return number_pattern(1+ambient_zone);
}

static void led_set_mode(unsigned char n, unsigned char mode)
{
	if (n < N_PWMLEDS) {
		pwmled_set_mode(n, mode);
	} else if (n == N_PWMLEDS) {
		gpio_set(GPIO_LED2, mode);
	}
	// TODO LED 1
}

static void inline led_set_level(unsigned char n, unsigned char level)
{
	if (n == N_PWMLEDS) {
		gpio_set(GPIO_LED2, 1);
	}
}

void patterns_next_tick()
{
	unsigned char i;

	for (i = 0; i < N_LEDS; i++) {
		if (led_counters[i] == 0) {
			led_patterns[i]++;
			if (led_patterns[i]->duration == 0) { // END
				led_patterns[i] = pattern_select(i);
			}
			led_counters[i] = led_patterns[i]->duration;
			led_set_mode(i, led_patterns[i]->mode);
		}

		led_counters[i]--;
	}
}