/*
Copyright 2014 Kai Ryu <[email protected]>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <avr/io.h>
#include <avr/interrupt.h>
#include <avr/pgmspace.h>
#include "backlight.h"
#include "softpwm_led.h"
#include "action.h"
#include "keymap_common.h"

#ifdef BACKLIGHT_ENABLE

static uint8_t backlight_mode;
static const uint8_t backlight_brightness = 0xFF;

enum {
    LED_KEY_1 = 0,
    LED_KEY_2,
    LED_KEY_SIDE,
    LED_BOARD_SIDE
};

void backlight_set(uint8_t level)
{
    backlight_mode = level;
    softpwm_led_set_all(0);
    fading_led_set_duration(2);
    fading_led_set_direction(FADING_LED_FADE_OUT);
    breathing_led_set_duration(3);
    switch (level) {
        case 0:
            softpwm_led_disable();
            fading_led_disable_all();
            breathing_led_disable_all();
            softpwm_led_on(LED_KEY_SIDE);
            softpwm_led_on(LED_BOARD_SIDE);
            break;
        case 1:
            softpwm_led_set(LED_BOARD_SIDE, backlight_brightness);
            fading_led_disable_all();
            fading_led_enable(LED_KEY_SIDE);
            breathing_led_disable_all();
            softpwm_led_enable();
            break;
        case 2:
            softpwm_led_set(LED_KEY_SIDE, backlight_brightness);
            fading_led_disable_all();
            fading_led_enable(LED_BOARD_SIDE);
            breathing_led_disable_all();
            softpwm_led_enable();
            break;
        case 3:
            fading_led_disable_all();
            breathing_led_disable_all();
            breathing_led_enable(LED_KEY_SIDE);
            breathing_led_enable(LED_BOARD_SIDE);
            softpwm_led_enable();
            break;
        case 4:
            softpwm_led_set(LED_KEY_SIDE, backlight_brightness);
            breathing_led_disable_all();
            breathing_led_enable(LED_BOARD_SIDE);
            fading_led_disable_all();
            softpwm_led_enable();
            break;
        case 5:
            softpwm_led_set(LED_BOARD_SIDE, backlight_brightness);
            breathing_led_disable_all();
            breathing_led_enable(LED_KEY_SIDE);
            fading_led_disable_all();
            softpwm_led_enable();
            break;
        case 6:
            fading_led_disable_all();
            breathing_led_disable_all();
            softpwm_led_disable();
            break;
        case 7:
            fading_led_disable_all();
            breathing_led_disable_all();
            softpwm_led_disable();
            break;
        case 8:
            softpwm_led_set_all(16);
            fading_led_disable_all();
            breathing_led_disable_all();
            softpwm_led_enable();
            break;
    }
}

#ifndef LEDMAP_ENABLE
void softpwm_led_init(void)
{
#ifndef EXPERIMENTAL
    DDRD |= (1<<PD0);
    DDRB |= (1<<PB7);
    DDRC |= (1<<PD5 | 1<<PD6);
#else
    DDRB |= (1<<PB6);
#endif
}

void softpwm_led_on(uint8_t index)
{
#ifndef EXPERIMENTAL
    switch (index) {
        case LED_KEY_1:
            PORTC |= (1<<PC5);
            break;
        case LED_KEY_2:
            PORTC |= (1<<PC6);
            break;
        case LED_KEY_SIDE:
            PORTB |= (1<<PB7);
            break;
        case LED_BOARD_SIDE:
            PORTD |= (1<<PD0);
            break;
    }
#else
    PORTB |= (1<<PB6);
#endif
}

void softpwm_led_off(uint8_t index)
{
#ifndef EXPERIMENTAL
    switch (index) {
        case LED_KEY_1:
            PORTC &= ~(1<<PC5);
            break;
        case LED_KEY_2:
            PORTC &= ~(1<<PC6);
            break;
        case LED_KEY_SIDE:
            PORTB &= ~(1<<PB7);
            break;
        case LED_BOARD_SIDE:
            PORTD &= ~(1<<PD0);
            break;
    }
#else
    PORTB &= ~(1<<PB6);
#endif
}
#endif

extern uint8_t config_mode;

void action_keyevent(keyevent_t event)
{
    if (config_mode) return;
    if (event.key.col < 2) {
        if (event.pressed) {
            switch (backlight_mode) {
                case 0: case 6:
                    softpwm_led_on(event.key.col);
                    break;
                case 1 ... 5:
                    softpwm_led_set(event.key.col, backlight_brightness);
                    break;
            }
        }
        else {
            switch (backlight_mode) {
                case 0: case 6:
                    softpwm_led_off(event.key.col);
                    break;
                case 1 ... 5:
                    softpwm_led_set(event.key.col, 0);
                    break;
            }
        }
    }
    switch (backlight_mode) {
        case 1:
            if (event.pressed) {
                softpwm_led_increase(LED_KEY_SIDE, 32);
            }
            break;
        case 2:
            if (event.pressed) {
                softpwm_led_increase(LED_BOARD_SIDE, 32);
            }
            break;
    }
}

#endif