/*
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/pgmspace.h>
#include "ledmap.h"

#ifdef LEDMAP_ENABLE

static const uint16_t ledmaps[LED_COUNT] PROGMEM = {
    [0] = LEDMAP_CAPS_LOCK | LEDMAP_BACKLIGHT,      // CapsLock
    [1] = LEDMAP_NUM_LOCK | LEDMAP_BACKLIGHT,       // NumLock
#ifdef REDSCARFII_PLUS
    [2] = LEDMAP_BACKLIGHT,                         // Backlight
#else
    [2] = LEDMAP_SCROLL_LOCK | LEDMAP_BACKLIGHT,    // Logo
    [3] = LEDMAP_BACKLIGHT,                         // Backlight
#endif
};

ledmap_t ledmap_get_code(uint8_t index)
{
    return (ledmap_t) { .code = pgm_read_word(&ledmaps[index]) };
}

/* LED pin configration
 *   CapsLock:  PC7
 *   NumLock:   PE6
 *   Logo:      PC6
 *   Backlight: PB7
 */
void ledmap_led_init(void)
{
    DDRC  |=  (1<<PC7);
    PORTC |=  (1<<PC7);
    DDRE  |=  (1<<PE6);
    PORTE |=  (1<<PE6);
#ifndef REDSCARFII_PLUS
    DDRC  |=  (1<<PC6);
    PORTC &= ~(1<<PC6);
#endif
    DDRB  |=  (1<<PB7);
    PORTB &= ~(1<<PB7);
}

void ledmap_led_on(uint8_t index)
{
    switch (index) {
        case 0:
            PORTC &= ~(1<<PC7);
            break;
        case 1:
            PORTE &= ~(1<<PE6);
            break;
#ifdef REDSCARFII_PLUS
        case 2:
            PORTB |=  (1<<PB7);
            break;
#else
        case 2:
            PORTC |=  (1<<PC6);
            break;
        case 3:
            PORTB |=  (1<<PB7);
            break;
#endif
    }
}

void ledmap_led_off(uint8_t index)
{
    switch (index) {
        case 0:
            PORTC |=  (1<<PC7);
            break;
        case 1:
            PORTE |=  (1<<PE6);
            break;
#ifdef REDSCARFII_PLUS
        case 2:
            PORTB &= ~(1<<PB7);
            break;
#else
        case 2:
            PORTC &= ~(1<<PC6);
            break;
        case 3:
            PORTB &= ~(1<<PB7);
            break;
#endif
    }
}

#endif