tmk_keyboard_custom/keyboard/RedScarfIII/rgb.c

/*
Copyright 2014 Kai Ryu <kai1103@gmail.com>

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/eeprom.h>
#include "yc059.h"
#include "rgb.h"

static rgb_config_t rgb_config;

void rgb_init(void)
{
    DDRE |= (1<<PE2);
    yc059_init();
    rgb_read_config();
    if (rgb_config.raw == RGB_UNCONFIGURED) {
        rgb_config.enable = 0;
        rgb_config.mode = RGB_FIXED;
        rgb_config.id = RGB_FIXED_WHITE;
        rgb_write_config();
    }
    yc059_send(rgb_config.enable ? YC059_ON : YC059_OFF);
    rgb_resume();
    PORTE |= (1<<PE2);
}

void rgb_resume(void)
{
    yc059_send(rgb_to_yc059(rgb_config.mode, rgb_config.id));
}

void rgb_read_config(void)
{
    rgb_config.raw = eeprom_read_byte(EECONFIG_RGB);
}

void rgb_write_config(void)
{
    eeprom_write_byte(EECONFIG_RGB, rgb_config.raw);
}

void rgb_on(void)
{
    rgb_config.enable = 1;
    PORTE &= ~(1<<PE2);
    yc059_send(YC059_ON);
    rgb_resume();
    PORTE |= (1<<PE2);
    rgb_write_config();
}

void rgb_off(void)
{
    rgb_config.enable = 0;
    yc059_send(YC059_OFF);
    rgb_write_config();
}

void rgb_toggle(void)
{
    if (rgb_config.enable) {
        rgb_off();
    }
    else {
        rgb_on();
    }
}

void rgb_increase(void)
{
    yc059_send(YC059_INCREASE);
}

void rgb_decrease(void)
{
    yc059_send(YC059_DECREASE);
}

void rgb_set(uint8_t mode, uint8_t id)
{
    if (rgb_config.enable) {
        rgb_config.mode = mode;
        rgb_config.id = id;
        rgb_resume();
        rgb_write_config();
    }
}

void rgb_step(uint8_t mode)
{
    if (rgb_config.enable) {
        if (rgb_config.mode != mode) {
            rgb_config.mode = mode;
            rgb_config.id = 0;
        }
        else {
            uint8_t count;
            switch (mode) {
                case RGB_FIXED:
                    count = RGB_FIXED_COUNT;
                    break;
                case RGB_VARIABLE:
                    count = RGB_VARIABLE_COUNT;
                    break;
                default:
                    count = 1;
                    break;
            }
            rgb_config.id = (rgb_config.id + 1) % count;
        }
        rgb_resume();
        rgb_write_config();
    }
}

uint8_t rgb_to_yc059(uint8_t mode, uint8_t id)
{
    switch (mode) {
        case RGB_FIXED:
            switch (id) {
                case RGB_FIXED_WHITE:
                    return YC059_FIXED_WHITE;
                case RGB_FIXED_RED:
                    return YC059_FIXED_RED;
                case RGB_FIXED_GREEN:
                    return YC059_FIXED_GREEN;
                case RGB_FIXED_BLUE:
                    return YC059_FIXED_BLUE;
                case RGB_FIXED_RED_1:
                    return YC059_FIXED_RED_1;
                case RGB_FIXED_GREEN_1:
                    return YC059_FIXED_GREEN_1;
                case RGB_FIXED_BLUE_1:
                    return YC059_FIXED_BLUE_1;
                case RGB_FIXED_RED_2:
                    return YC059_FIXED_RED_2;
                case RGB_FIXED_GREEN_2:
                    return YC059_FIXED_GREEN_2;
                case RGB_FIXED_BLUE_2:
                    return YC059_FIXED_BLUE_2;
                case RGB_FIXED_RED_3:
                    return YC059_FIXED_RED_3;
                case RGB_FIXED_GREEN_3:
                    return YC059_FIXED_GREEN_3;
                case RGB_FIXED_BLUE_3:
                    return YC059_FIXED_BLUE_3;
                case RGB_FIXED_RED_4:
                    return YC059_FIXED_RED_4;
                case RGB_FIXED_GREEN_4:
                    return YC059_FIXED_GREEN_4;
                case RGB_FIXED_BLUE_4:
                    return YC059_FIXED_BLUE_4;
            }
        case RGB_VARIABLE:
            switch (id) {
                case RGB_VARIABLE_FLASH:
                    return YC059_FLASH;
                case RGB_VARIABLE_STROBE:
                    return YC059_STROBE;
                case RGB_VARIABLE_FADE:
                    return YC059_FADE;
                case RGB_VARIABLE_SMOOTH:
                    return YC059_SMOOTH;
            }
        default:
            return YC059_OFF;
    }
}