#include <avr/io.h>
#include <util/delay.h>
#include <avr/sleep.h>
#include <avr/interrupt.h>
#include <avr/power.h>
#include <avr/wdt.h>

#include "lights.h"

static void hw_setup()
{
	init_battery();
	init_pwm();
	init_adc();
	init_tmr();
	init_buttons();

	init_pwmled();
	init_gpio();
	init_ambient();
	init_pattern();
	init_control();

	set_sleep_mode(SLEEP_MODE_IDLE);
}

static void inline hw_suspend()
{
	susp_pwm();
	susp_adc();
	susp_tmr();
	susp_gpio();
	susp_ambient();
	susp_buttons();
}

void power_down(unsigned char err)
{
	hw_suspend();

	do {
		if (err)
			gpio_set(0, 1);

		// G'night
		set_sleep_mode(SLEEP_MODE_PWR_DOWN);
		sleep_enable();
		sleep_bod_disable();
		sei();
		sleep_cpu();

		// G'morning
		cli();
		sleep_disable();

		// allow wakeup by long button-press only
	} while (!buttons_wait_for_release());

	// ok, so I will wake up
	hw_setup();
}

static void inline first_boot()
{
	unsigned char mcusr_save;

	// disable the WDT if running
	wdt_reset();
	mcusr_save = MCUSR;
	MCUSR = 0;
	wdt_disable();

	if (mcusr_save & _BV(WDRF)) // was watchdog reset?
		gpio_set(0, 1);

	init_log(mcusr_save);

	power_usi_disable(); // Once for lifetime
	ACSRA |= _BV(ACD);   // disable analog comparator

	log_set_state(3);

	hw_setup();
	power_down(mcusr_save & _BV(WDRF));

	sei();
}

static void inline main_loop_iteration()
{
	cli();
	if (TIMER1_IS_ON()) {
		set_sleep_mode(SLEEP_MODE_IDLE);
	} else if (adc_is_on) {
		set_sleep_mode(SLEEP_MODE_ADC);
	} else {
		set_sleep_mode(SLEEP_MODE_PWR_DOWN);
	}

	sleep_enable();
	// keep BOD active, no sleep_bod_disable();
	sei();
	sleep_cpu();
	sleep_disable();
}

int main(void)
{
	first_boot();

	while (1)
		main_loop_iteration();

#if 0
	DDRB |= _BV(PB2);
	while (1) {
		PORTB |=  _BV( PB2 );
		_delay_ms(200);
		PORTB &=~ _BV( PB2 );
		_delay_ms(200);
	}
#endif
}