#include <inttypes.h>
#include <stdlib.h> // for NULL

#include "lights.h"

static pattern_t panic_pattern[] = {
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_1 },
	PATTERN_END
};

static pattern_t slow_pattern[] = {
	{ PWM_PAT(1, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 1), D_1 },
	{ PWM_PAT(0, 0, 0), D_13 },
	PATTERN_END
};

static pattern_t fast_pattern[] = {
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(2, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 2), D_1 },
	{ PWM_PAT(0, 0, 0), D_8 },
	{ PWM_PAT(1, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 1), D_1 },
	{ PWM_PAT(1, 0, 0), D_1 },
	{ PWM_PAT(0, 0, 1), D_1 },
	{ PWM_PAT(0, 0, 0), D_8 },
	PATTERN_END
};

static pattern_t night_pattern[] = {
	{ PWM_PAT(3, 0, 1), D_3 },
	{ PWM_PAT(2, 0, 0), D_8 },
	{ PWM_PAT(3, 0, 1), D_1 },
	{ PWM_PAT(2, 0, 0), D_2 },
	{ PWM_PAT(3, 0, 1), D_1 },
	{ PWM_PAT(2, 0, 0), D_8 },
	PATTERN_END
};

pattern_t on_pattern[] = {
	{ 1, D_8 },
	PATTERN_END
};

// #define TEST_PATTERN 1
#ifdef TEST_PATTERN
pattern_t test_pattern[] = {
	{ PWM_PAT(1, 0, 0), D_13 },
	{ PWM_PAT(2, 0, 0), D_13 },
	{ PWM_PAT(0, 0, 1), D_13 },
	{ PWM_PAT(0, 0, 2), D_13 },
	PATTERN_END
};
#endif

volatile unsigned char braking;
static unsigned char dim_mode, towbar_mode;

void init_control()
{
	dim_mode = 0;
	towbar_mode = 0;
	braking = 0;

	pwmled_select_brightness();
}

void brake_on()
{
	braking = 1;
	gpio_set(0, 1);
	led_set_pattern(N_STATUS_LED, status_led_pattern_select());
	pwmleds_update_mode();
}

void brake_off()
{
	braking = 0;
	gpio_set(0, 0);
	led_set_pattern(N_STATUS_LED, status_led_pattern_select());
	pwmleds_update_mode();
}

void toggle_dim_mode()
{
	dim_mode = !dim_mode;
	pwmled_select_brightness();
	pattern_reload();
}

void set_panic_mode()
{
	led_set_pattern(0, panic_pattern);
	led_set_pattern(N_ILLUM_LED, panic_pattern);
}

pattern_t *pwmled_pattern_select()
{
#ifdef TEST_PATTERN
	return tmp_pattern;
#endif
	if (battery_critical)
		return slow_pattern;

	switch (ambient_zone) {
	case 0: return night_pattern;
	case 1:
	case 2:
	case 3:
	default:
		return dim_mode ? slow_pattern : fast_pattern;
	}
}

pattern_t *status_led_pattern_select()
{
	if (braking)
		return on_pattern;

	if (buttons_setup_in_progress())
		return buttons_setup_status0_pattern_select();

	// FIXME: do something sane
	return number_pattern(battery_gauge(), 0);
}

pattern_t *illumination_led_pattern_select()
{
	if (battery_critical)
		return NULL;

	switch (ambient_zone) {
	case 0: return dim_mode
		? number_pattern(1, 1)
		: on_pattern;
	case 1: return dim_mode
		? number_pattern(2, 1)
		: number_pattern(3, 1);
	case 2: return dim_mode
		? number_pattern(1, 0)
		: number_pattern(2, 0);
	case 3:
	default: return dim_mode
		? number_pattern(3, 0)
		: number_pattern(4, 0);
	}
}

pattern_t *laser_pattern_select()
{
	if (!dim_mode && ambient_zone <= 1)
		return number_pattern(2, 1);
	else
		return NULL;
}

void pwmled_select_brightness()
{
	uint16_t brightness = PWMLED_BRIGHTNESS(0, 2, 1, 0, 2); // default

	if (battery_critical) {
		brightness = PWMLED_BRIGHTNESS(0, 0, 0, 0, 0);
	} else if (ambient_zone < 2) {
		if (dim_mode)
			brightness = PWMLED_BRIGHTNESS(0, 1, 0, 0, 1);
		else
			brightness = PWMLED_BRIGHTNESS(0, 2, 1, 0, 2);
	} else if (ambient_zone == 2) {
		brightness = PWMLED_BRIGHTNESS(1, 3, 2, 1, 3);
	} else if (ambient_zone == 3) {
		brightness = PWMLED_BRIGHTNESS(2, 4, 2, 2, 4);
	}

	pwmled_set_brightness(brightness);
}