/*
 * Led.h
 *
 * Created: 03.11.2013 17:11:58
 * Author: BlubbFish
 */ 

#ifndef STRIPE_H_
#define STRIPE_H_

#include "hardware/pin.hpp"

#include <util/delay.h>

template <typename PortR, int pin_red, typename PortG, int pin_green, typename PortB, int pin_blue>
class Stripe {
	public:
		Stripe() {
			this->init();
		}
		void color(uint8_t r, uint8_t g, uint8_t b) {
			OCR0B=r; //ROT
			OCR2B=g; //GRÜN
			OCR2A=b; //BLAU
		}
		void setcolor(uint8_t mask) {
			(mask & (1<<0)) ? OCR2A=0xFF : OCR2A=0x00;
			(mask & (1<<1)) ? OCR2B=0xFF : OCR2B=0x00;
			(mask & (1<<2)) ? OCR0B=0xFF : OCR0B=0x00;
		}
		static const uint8_t BLACK = 0;
		static const uint8_t BLUE = 1;
		static const uint8_t GREEN = 2;
		static const uint8_t CYAN = 3;
		static const uint8_t RED = 4;
		static const uint8_t MAGENTA = 5;
		static const uint8_t YELLOW = 6;
		static const uint8_t WHITE = 7;
		void off() {
			this->timerOff();
			stripered::make_low();
			stripegreen::make_low();
			stripeblue::make_low();
		}
		void on() {
			this->timerOn();
		}
		void lower() {
			if(OCR0B >= 5) {
				OCR0B = OCR0B - 5;
			} else {
				OCR0B = 0;
			}
			if(OCR2B >= 5) {
				OCR2B = OCR2B - 5;
			} else {
				OCR2B = 0;
			}
			if(OCR2A >= 5) {
				OCR2A = OCR2A - 5;
			} else {
				OCR2A = 0;
			}
		}
		void higher() {
			if(OCR0B <= 250) {
				OCR0B = OCR0B + 5;
				} else {
				OCR0B = 0;
			}
			if(OCR2B <= 250) {
				OCR2B = OCR2B + 5;
				} else {
				OCR2B = 0;
			}
			if(OCR2A <= 250) {
				OCR2A = OCR2A + 5;
				} else {
				OCR2A = 0;
			}
		}
		void fadeto(uint8_t red, uint8_t green, uint8_t blue, uint8_t time) {
			this->setfadecolor(0, red, green, blue, time);
			this->fade_zylk = 1;
		}
		void setfadecolor(uint8_t index, uint8_t red, uint8_t green, uint8_t blue, uint8_t time) {
			this->fademap[0][index] = red;
			this->fademap[1][index] = green;
			this->fademap[2][index] = blue;
			this->fademap[3][index] = time;
		}
		void interrupt() {
			if(this->fade_zylk > 0) {
				if(this->fadesteps[0] == 0 && this->fadesteps[1] == 0 && this->fadesteps[2] == 0) {
					this->fadesteps[0] = (this->fademap[0][this->fade_zylk-1]-OCR0B)/this->fademap[3][this->fade_zylk-1]; //Calc Red;
					this->fadesteps[1] = (this->fademap[1][this->fade_zylk-1]-OCR2B)/this->fademap[3][this->fade_zylk-1]; //Calc Green;
					this->fadesteps[2] = (this->fademap[2][this->fade_zylk-1]-OCR2A)/this->fademap[3][this->fade_zylk-1]; //Calc Blue;
				} else {
					if((this->fadesteps[0] > 0 && (OCR0B + this->fadesteps[0]) <= this->fademap[0][this->fade_zylk-1]) || (this->fadesteps[0] < 0 && (OCR0B + this->fadesteps[0]) >= this->fademap[0][this->fade_zylk-1])) {
						OCR0B = OCR0B + this->fadesteps[0];
					} else {
						OCR0B = this->fademap[0][this->fade_zylk-1];
						this->fadesteps[0] = 0;
					}
					if((this->fadesteps[1] > 0 && (OCR2B + this->fadesteps[1]) <= this->fademap[1][this->fade_zylk-1]) || (this->fadesteps[1] < 0 && (OCR2B + this->fadesteps[1]) >= this->fademap[1][this->fade_zylk-1])) {
						OCR2B = OCR2B + this->fadesteps[1];
					} else {
						OCR2B = this->fademap[1][this->fade_zylk-1];
						this->fadesteps[1] = 0;
					}
					if((this->fadesteps[2] > 0 && (OCR2A + this->fadesteps[2]) <= this->fademap[2][this->fade_zylk-1]) || (this->fadesteps[2] < 0 && (OCR2A + this->fadesteps[2]) >= this->fademap[2][this->fade_zylk-1])) {
						OCR2A = OCR2A + this->fadesteps[2];
					} else {
						OCR2A = this->fademap[2][this->fade_zylk-1];
						this->fadesteps[2] = 0;
					}
				}
			}
		}
	private:
		uint8_t fade_zylk;
		uint8_t fademap[4][5];
		float fadesteps[3];
		void init() {
			stripered::make_output();
			stripegreen::make_output();
			stripeblue::make_output();
			this->fade_zylk = 0;
			this->initTimer();
			this->color(0, 0, 0);
		}
		void timerOff() {
			// OC0B output: Disconnected
			// OC2A output: Disconnected
			// OC2B output: Disconnected
			TCCR0A &= ~(1<<COM0B1);
			TCCR2A &= ~((1<<COM2A1) | (1<<COM2B1));
		}
		void timerOn() {
			// OC0B output: Non-Inverted PWM
			// OC2A output: Non-Inverted PWM
			// OC2B output: Non-Inverted PWM
			TCCR0A |= (1<<COM0B1);
			TCCR2A |= (1<<COM2A1) | (1<<COM2B1);
		}
		void uninit();
		void initTimer() {
			// Timer/Counter 0 initialization
			// Clock source: System Clock
			// Clock value: 8000,000 kHz
			// Mode: Phase correct PWM top=0xFF
			// OC0A output: Disconnected
			// Timer Period: 0,06375 ms
			// Output Pulse(s):
			// OC0B Period: 0,06375 ms Width: 0 us
			TCCR0A=(0<<COM0A1) | (0<<COM0A0) | (0<<COM0B0) | (0<<WGM01) | (1<<WGM00);
			TCCR0B=(0<<WGM02) | (1<<CS02) | (0<<CS01) | (0<<CS00);
			TCNT0=0x00;
			// Timer/Counter 1 initialization
			// Clock source: System Clock
			// Clock value: 20000,000 kHz
			// Mode: Normal top=0xFFFF
			// OC1A output: Disconnected
			// OC1B output: Disconnected
			// Noise Canceler: Off
			// Input Capture on Falling Edge
			// Timer Period: 3,2768 ms
			// Timer1 Overflow Interrupt: On
			// Input Capture Interrupt: Off
			// Compare A Match Interrupt: Off
			// Compare B Match Interrupt: Off
			TCCR1A=(0<<COM1A1) | (0<<COM1A0) | (0<<COM1B1) | (0<<COM1B0) | (0<<WGM11) | (0<<WGM10);
			TCCR1B=(0<<ICNC1) | (0<<ICES1) | (0<<WGM13) | (0<<WGM12) | (0<<CS12) | (1<<CS11) | (0<<CS10);
			TCNT1H=0x00;
			TCNT1L=0x00;
			// Timer/Counter 2 initialization
			// Clock source: System Clock
			// Clock value: 8000,000 kHz
			// Mode: Phase correct PWM top=0xFF
			// Timer Period: 0,06375 ms
			// Output Pulse(s):
			// OC2A Period: 0,06375 ms Width: 0 us
			// OC2B Period: 0,06375 ms Width: 0 us
			ASSR=(0<<EXCLK) | (0<<AS2);
			TCCR2A=(0<<COM2A0) | (0<<COM2B0) | (0<<WGM21) | (1<<WGM20);
			TCCR2B=(0<<WGM22) | (1<<CS22) | (0<<CS21) | (1<<CS20);
			TCNT2=0x00;
			// Timer/Counter 0 Interrupt(s) initialization
			TIMSK0=(0<<OCIE0B) | (0<<OCIE0A) | (0<<TOIE0);
			// Timer/Counter 1 Interrupt(s) initialization
			TIMSK1=(0<<ICIE1) | (0<<OCIE1B) | (0<<OCIE1A) | (1<<TOIE1);
			// Timer/Counter 2 Interrupt(s) initialization
			TIMSK2=(0<<OCIE2B) | (0<<OCIE2A) | (0<<TOIE2);
			this->timerOn();
		}
		const typedef avrlib::pin<PortR, pin_red> stripered;
		const typedef avrlib::pin<PortG, pin_green> stripegreen;
		const typedef avrlib::pin<PortB, pin_blue> stripeblue;
};
#endif /* STRIPE_H_ */