#include <avr/io.h>
#include <avr/interrupt.h>
#include <avr/pgmspace.h>
#include "led.h"
#include "softpwm_led.h"
#include "debug.h"

#define SOFTPWM_LED_FREQ 64
#define SOFTPWM_LED_TIMER_TOP F_CPU / (256 * SOFTPWM_LED_FREQ)

static uint8_t softpwm_led_state = 0;
static uint8_t softpwm_led_ocr[LED_COUNT] = {0};
static uint8_t softpwm_led_ocr_buff[LED_COUNT] = {0};

#ifdef FADING_LED_ENABLE
static void fading_led_proc(void);
#else
#define fading_led_proc()
#endif

#ifdef BREATHING_LED_ENABLE
static void breathing_led_proc(void);
#else
#define fading_led_proc()
#endif

void softpwm_init(void)
{
#ifdef SOFTPWM_LED_TIMER3
    /* Timer3 setup */
    /* CTC mode */
    TCCR3B |= (1<<WGM32);
    /* Clock selelct: clk/8 */
    TCCR3B |= (1<<CS30);
    /* Set TOP value */
    uint8_t sreg = SREG;
    cli();
    OCR3AH = (SOFTPWM_LED_TIMER_TOP >> 8) & 0xff;
    OCR3AL = SOFTPWM_LED_TIMER_TOP & 0xff;
    SREG = sreg;
#else
    /* Timer1 setup */
    /* CTC mode */
    TCCR1B |= (1<<WGM12);
    /* Clock selelct: clk/8 */
    TCCR1B |= (1<<CS10);
    /* Set TOP value */
    uint8_t sreg = SREG;
    cli();
    OCR1AH = (SOFTPWM_LED_TIMER_TOP >> 8) & 0xff;
    OCR1AL = SOFTPWM_LED_TIMER_TOP & 0xff;
    SREG = sreg;
#endif
    softpwm_led_init();
}

void softpwm_led_enable(void)
{
    /* Enable Compare Match Interrupt */
#ifdef SOFTPWM_LED_TIMER3
    TIMSK3 |= (1<<OCIE3A);
    //dprintf("softpwm led on: %u\n", TIMSK3 & (1<<OCIE3A));
#else
    TIMSK1 |= (1<<OCIE1A);
    //dprintf("softpwm led on: %u\n", TIMSK1 & (1<<OCIE1A));
#endif
    softpwm_led_state = 1;
#ifdef LEDMAP_ENABLE
    softpwm_led_state_change(softpwm_led_state);
#endif
}

void softpwm_led_disable(void)
{
    /* Disable Compare Match Interrupt */
#ifdef SOFTPWM_LED_TIMER3
    TIMSK3 &= ~(1<<OCIE3A);
    //dprintf("softpwm led off: %u\n", TIMSK3 & (1<<OCIE3A));
#else
    TIMSK1 &= ~(1<<OCIE1A);
    //dprintf("softpwm led off: %u\n", TIMSK1 & (1<<OCIE1A));
#endif
    softpwm_led_state = 0;
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        softpwm_led_off(i);
    }
#ifdef LEDMAP_ENABLE
    softpwm_led_state_change(softpwm_led_state);
#endif
}

void softpwm_led_toggle(void)
{
    if (softpwm_led_state) {
        softpwm_led_disable();
    }
    else {
        softpwm_led_enable();
    }
}

void softpwm_led_set(uint8_t index, uint8_t val)
{
    softpwm_led_ocr_buff[index] = val;
}

void softpwm_led_set_all(uint8_t val)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        softpwm_led_ocr_buff[i] = val;
    }
}

void softpwm_led_increase(uint8_t index, uint8_t offset)
{
    if (softpwm_led_ocr_buff[index] > 0xFF - offset) {
        softpwm_led_ocr_buff[index] = 0xFF;
    }
    else {
        softpwm_led_ocr_buff[index] += offset;
    }
}

void softpwm_led_increase_all(uint8_t offset)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        softpwm_led_increase(i, offset);
    }
}

void softpwm_led_decrease(uint8_t index, uint8_t offset)
{
    if (softpwm_led_ocr_buff[index] < offset) {
        softpwm_led_ocr_buff[index] = 0;
    }
    else {
        softpwm_led_ocr_buff[index] -= offset;
    }
}

void softpwm_led_decrease_all(uint8_t offset)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        softpwm_led_decrease(i, offset);
    }
}

inline uint8_t softpwm_led_get_state(void)
{
    return softpwm_led_state;
}

#ifdef FADING_LED_ENABLE

static led_pack_t fading_led_state = 0;
static led_pack_t fading_led_direction = 0;
static uint8_t fading_led_duration = 0;
static uint8_t fading_led_delay[LED_COUNT] = {0};

void fading_led_enable(uint8_t index)
{
    LED_BIT_SET(fading_led_state, index);
}

void fading_led_enable_all(void)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        LED_BIT_SET(fading_led_state, i);
    }
}

void fading_led_disable(uint8_t index)
{
    LED_BIT_CLEAR(fading_led_state, index);
}

void fading_led_disable_all(void)
{
    fading_led_state = 0;
}

void fading_led_toggle(uint8_t index)
{
    LED_BIT_XOR(fading_led_state, index);
}

void fading_led_toggle_all(void)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        LED_BIT_XOR(fading_led_state, i);
    }
}

void fading_led_set_direction(uint8_t index, uint8_t dir)
{
    if (dir) {
        LED_BIT_SET(fading_led_direction, index);
    }
    else {
        LED_BIT_CLEAR(fading_led_direction, index);
    }
}

void fading_led_set_direction_all(uint8_t dir)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        fading_led_set_direction(i, dir);
    }
}

void fading_led_set_duration(uint8_t dur)
{
    fading_led_duration = dur;
}

void fading_led_set_delay(uint8_t index, uint8_t delay)
{
    fading_led_delay[index] = delay;
}

void fading_led_set_delay_all(uint8_t delay)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        fading_led_delay[i] = delay;
    }
}

#endif

#ifdef BREATHING_LED_ENABLE

/* Breathing LED brighness(PWM On period) table
 *
 * http://www.wolframalpha.com/input/?i=Table%5Bfloor%28%28exp%28sin%28x%2F256*2*pi%2B3%2F2*pi%29%29-1%2Fe%29*%28256%2F%28e-1%2Fe%29%29%29%2C+%7Bx%2C0%2C255%2C1%7D%5D
 * Table[floor((exp(sin(x/256*2*pi+3/2*pi))-1/e)*(256/(e-1/e))), {x,0,255,1}]
 * (0..255).each {|x| print ((exp(sin(x/256.0*2*PI+3.0/2*PI))-1/E)*(256/(E-1/E))).to_i, ', ' }
 */
static const uint8_t breathing_table[128] PROGMEM = {
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 8, 8, 9, 10, 11, 11, 12, 13, 14, 15, 16, 17, 18, 19, 21, 22, 23, 25, 26, 27, 29, 30, 32, 34, 35, 37, 39, 41, 43, 45, 47, 49, 51, 53, 56, 58, 61, 63, 66, 68, 71, 74, 77, 80, 83, 86, 89, 92, 95, 98, 102, 105, 108, 112, 116, 119, 123, 126, 130, 134, 138, 142, 145, 149, 153, 157, 161, 165, 169, 173, 176, 180, 184, 188, 192, 195, 199, 203, 206, 210, 213, 216, 219, 223, 226, 228, 231, 234, 236, 239, 241, 243, 245, 247, 248, 250, 251, 252, 253, 254, 255, 255, 255
};

static led_pack_t breathing_led_state = 0;
static led_pack_t breathing_led_once = 0;
static uint8_t breathing_led_duration = 0;
static uint8_t breathing_led_index[LED_COUNT] = {0};
static led_pack_t breathing_led_direction = 0;

void breathing_led_enable(uint8_t index)
{
    LED_BIT_SET(breathing_led_state, index);
}

void breathing_led_enable_once(uint8_t index)
{
    LED_BIT_SET(breathing_led_state, index);
    LED_BIT_SET(breathing_led_once, index);
}

void breathing_led_enable_all(void)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        LED_BIT_SET(breathing_led_state, i);
    }
}

void breathing_led_disable(uint8_t index)
{
    LED_BIT_CLEAR(breathing_led_state, index);
}

void breathing_led_disable_all(void)
{
    breathing_led_state = 0;
}

void breathing_led_toggle(uint8_t index)
{
    LED_BIT_XOR(breathing_led_state, index);
}

void breathing_led_toggle_all(void)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        LED_BIT_XOR(breathing_led_state, i);
    }
}

void breathing_led_set_duration(uint8_t dur)
{
    breathing_led_duration = dur;
}

void breathing_led_set_index(uint8_t index, uint8_t value)
{
    if (value & 0x80) {
        LED_BIT_SET(breathing_led_direction, index);
    }
    else {
        LED_BIT_CLEAR(breathing_led_direction, index);
    }
    breathing_led_index[index] = value & 0x7F;
}

void breathing_led_set_index_all(uint8_t value)
{
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        breathing_led_set_index(i, value);
    }
}

#endif

#ifdef SOFTPWM_LED_TIMER3
ISR(TIMER3_COMPA_vect)
#else
ISR(TIMER1_COMPA_vect)
#endif
{
    static uint8_t pwm = 0;
    pwm++;
    for (uint8_t i = 0; i < LED_COUNT; i++) {
        // LED on
        if (pwm == 0) {
            if (softpwm_led_ocr[i]) softpwm_led_on(i);
            softpwm_led_ocr[i] = softpwm_led_ocr_buff[i];
        }
        // LED off
        if (pwm == softpwm_led_ocr[i]) {
            softpwm_led_off(i);
        }
    }

#if defined(FADING_LED_ENABLE) || defined(BREATHING_LED_ENABLE) || defined(CUSTOM_LED_ENABLE)
    static uint8_t counter = 0;
    if (++counter >= SOFTPWM_LED_FREQ) {
        counter = 0;
#ifdef FADING_LED_ENABLE
        fading_led_proc();
#endif
#ifdef BREATHING_LED_ENABLE
        breathing_led_proc();
#endif
#if CUSTOM_LED_ENABLE
        softpwm_led_custom();
#endif
    }
#endif
}

#ifdef FADING_LED_ENABLE
void fading_led_proc(void)
{
    static uint8_t step = 0;
    if (fading_led_state) {
        if (++step > fading_led_duration) {
            step = 0;
            for (uint8_t i = 0; i < LED_COUNT; i++) {
                if (fading_led_state & LED_BIT(i)) {
                    if (fading_led_delay[i]) {
                        fading_led_delay[i]--;
                    }
                    else {
                        if (fading_led_direction & LED_BIT(i)) {
                            softpwm_led_decrease(i, 1);
                        }
                        else {
                            softpwm_led_increase(i, 1);
                        }
                    }
                }
            }
#ifdef CUSTOM_LED_ENABLE
            fading_led_custom(softpwm_led_ocr);
#endif
        }
    }
}
#endif

#ifdef BREATHING_LED_ENABLE
void breathing_led_proc(void)
{
    static uint8_t step = 0;
    if (breathing_led_state) {
        if (++step > breathing_led_duration) {
            step = 0;
            for (uint8_t i = 0; i < LED_COUNT; i++) {
                if (breathing_led_state & LED_BIT(i)) {
                    softpwm_led_ocr_buff[i] = pgm_read_byte(&breathing_table[breathing_led_index[i]]);
                    if (breathing_led_direction & LED_BIT(i)) {
                        if (breathing_led_index[i] == 0) {
                            LED_BIT_CLEAR(breathing_led_direction, i);
                            if (breathing_led_once & LED_BIT(i)) {
                                LED_BIT_CLEAR(breathing_led_state, i);
                                LED_BIT_CLEAR(breathing_led_once, i);
                            }
                        }
                        else {
                            breathing_led_index[i]--;
                        }

                    }
                    else {
                        if (breathing_led_index[i] == 127) {
                            LED_BIT_SET(breathing_led_direction, i);
                        }
                        else {
                            breathing_led_index[i]++;
                        }
                    }
                }
            }
#ifdef CUSTOM_LED_ENABLE
            breathing_led_custom(softpwm_led_ocr);
#endif
        }
    }
}
#endif