#define F_CPU 16000000UL
#include <stdint.h>
#include <avr/io.h>
#include <avr/interrupt.h>
#include <util/delay_basic.h>
#include "light_ws2812.h"
#include "light_ws2812.c"
#include "ws2812_config.h"

typedef char s8;
typedef int16_t s16;
typedef int s32;
typedef int64_t s64;
typedef uint8_t u8;
typedef uint16_t u16;
typedef uint32_t u32;
typedef uint64_t u64;

#define MAGIC 2500
#define LEDS 18
struct cRGB leds[LEDS]={{0}};

void gpioSetup(void)
{
	DDRB=(1<<ws2812_pin);
	DDRC=(1<<PC3);
	DDRD=(1<<PD1);
	PORTC=(1<<PC3);
}

void timerSetup(void)
{
	TCCR1B=(1<<CS11)|(1<<CS10);
}

void adcSetup(void)
{
	ADCSRA=(1<<ADEN)|(1<<ADPS2);
}

u16 adcRead(u8 channel)
{
	ADMUX=(1<<REFS0)|channel;
	ADCSRA|=(1<<ADSC);
	while(ADCSRA&(1<<ADSC));
	return(ADC);
}

void ledSetup(void)
{
	u8 i;
	for(i=0; i<LEDS; i++)
	{
		leds[i].r=0;
		leds[i].g=0;
		leds[i].b=0;
	}
	ws2812_setleds(leds, LEDS);
}

struct hp
{
	s16 cy;
	s16 ly;
	s16 lx;
};

struct lp
{
	s16 cy;
	s16 ly;
};

s16 highPass(s16 cx, struct hp *h)
{
	cx*=1;
	(h->cy)=cx-(h->lx);
	(h->cy)+=(h->ly);
	(h->cy)*=64;
	(h->cy)/=128;
	(h->ly)=(h->cy);
	(h->lx)=cx;
	return(h->cy);
}

s16 lowPass(s16 cx, struct lp *l)
{
	cx*=1;
	(l->cy)=cx-(l->ly);
	(l->cy)*=1;
	(l->cy)/=8;
	(l->cy)+=(l->ly);
	(l->ly)=(l->cy);
	return(l->cy);
}

u8 scale(s16 cx)
{
	cx=(cx>0)?cx:-cx;
	cx/=1;
	cx=cx>255?255:cx;
	return(cx);
}

s32 main(void)
{
	u8 i;
	struct hp fh[3]={{0}};
	struct lp fl[3]={{0}};
	gpioSetup();
	timerSetup();
	adcSetup();
	ledSetup();
	while(1)
	{
		if(TCNT1<MAGIC)
		{
			continue;
		}
		i=LEDS;
		while(--i)
		{
			leds[i].r=leds[i-1].r;
			leds[i].g=leds[i-1].g;
			leds[i].b=leds[i-1].b;
		}
		leds[0].r=scale(highPass(lowPass(adcRead(0), &fl[0]), &fh[0]));
		leds[0].g=scale(highPass(lowPass(adcRead(1), &fl[1]), &fh[1]));
		leds[0].b=scale(highPass(lowPass(adcRead(2), &fl[2]), &fh[2]));
		ws2812_setleds(leds, LEDS);
		TCNT1=0;
	}
}