cleanup names, rgb matrix

This commit is contained in:
Jack Humbert 2018-04-28 17:25:02 -04:00
parent ea7b1b389d
commit 1d44465dbd
4 changed files with 229 additions and 290 deletions

View File

@ -19,7 +19,8 @@
const is31_led g_is31_leds[DRIVER_LED_TOTAL] = { const is31_led g_is31_leds[DRIVER_LED_TOTAL] = {
/* driver /* Refer to IS31 manual for these locations
* driver
* | R location * | R location
* | | G location * | | G location
* | | | B location * | | | B location
@ -140,12 +141,6 @@ const rgb_led g_rgb_leds[DRIVER_LED_TOTAL] = {
void matrix_init_kb(void) { void matrix_init_kb(void) {
// Initialize LED drivers for backlight.
backlight_init_drivers();
backlight_timer_init();
backlight_timer_enable();
// Turn status LED on // Turn status LED on
DDRD |= (1<<6); DDRD |= (1<<6);
PORTD |= (1<<6); PORTD |= (1<<6);
@ -155,44 +150,26 @@ void matrix_init_kb(void) {
bool process_record_kb(uint16_t keycode, keyrecord_t *record) bool process_record_kb(uint16_t keycode, keyrecord_t *record)
{ {
// Record keypresses for backlight effects
if ( record->event.pressed ) {
backlight_set_key_hit( record->event.key.row, record->event.key.col );
} else {
// backlight_unset_key_hit( record->event.key.row, record->event.key.col );
}
return process_record_user(keycode, record); return process_record_user(keycode, record);
} }
uint16_t backlight_task_counter = 0;
void matrix_scan_kb(void) void matrix_scan_kb(void)
{ {
// if (backlight_task_counter == 0)
backlight_rgb_task();
// backlight_effect_single_LED_test();
// backlight_task_counter = ((backlight_task_counter + 1) % 5);
// This only updates the LED driver buffers if something has changed.
backlight_update_pwm_buffers();
matrix_scan_user(); matrix_scan_user();
} }
void led_set_kb(uint8_t usb_led) void led_set_kb(uint8_t usb_led)
{ {
backlight_set_indicator_state(usb_led); rgb_matrix_set_indicator_state(usb_led);
//backlight_debug_led(usb_led & (1<<USB_LED_CAPS_LOCK)); //rgb_matrix_debug_led(usb_led & (1<<USB_LED_CAPS_LOCK));
} }
void suspend_power_down_kb(void) void suspend_power_down_kb(void)
{ {
backlight_set_suspend_state(true); rgb_matrix_set_suspend_state(true);
} }
void suspend_wakeup_init_kb(void) void suspend_wakeup_init_kb(void)
{ {
backlight_set_suspend_state(false); rgb_matrix_set_suspend_state(false);
} }

View File

@ -202,6 +202,9 @@ bool process_record_quantum(keyrecord_t *record) {
process_key_lock(&keycode, record) && process_key_lock(&keycode, record) &&
#endif #endif
process_record_kb(keycode, record) && process_record_kb(keycode, record) &&
#if defined(RGB_MATRIX_ENABLE) && defined(RGB_MATRIX_KEYPRESSES)
process_rgb_matrix(keycode, record) &&
#endif
#if defined(MIDI_ENABLE) && defined(MIDI_ADVANCED) #if defined(MIDI_ENABLE) && defined(MIDI_ADVANCED)
process_midi(keycode, record) && process_midi(keycode, record) &&
#endif #endif
@ -737,9 +740,14 @@ void matrix_init_quantum() {
#ifdef AUDIO_ENABLE #ifdef AUDIO_ENABLE
audio_init(); audio_init();
#endif #endif
#ifdef RGB_MATRIX_ENABLE
rgb_matrix_init_drivers();
#endif
matrix_init_kb(); matrix_init_kb();
} }
// uint16_t rgb_matrix_task_counter = 0;
void matrix_scan_quantum() { void matrix_scan_quantum() {
#ifdef AUDIO_ENABLE #ifdef AUDIO_ENABLE
matrix_scan_music(); matrix_scan_music();
@ -757,6 +765,13 @@ void matrix_scan_quantum() {
backlight_task(); backlight_task();
#endif #endif
#ifdef RGB_MATRIX_ENABLE
// if (rgb_matrix_task_counter == 0)
rgb_matrix_task();
// rgb_matrix_task_counter = ((rgb_matrix_task_counter + 1) % 5);
rgb_matrix_update_pwm_buffers();
#endif
matrix_scan_kb(); matrix_scan_kb();
} }

View File

@ -27,8 +27,6 @@
#include "lufa.h" #include "lufa.h"
#include <math.h> #include <math.h>
#define RGB_MATRIX_EFFECT_MAX 17
rgb_config_t rgb_matrix_config; rgb_config_t rgb_matrix_config;
#ifndef RGB_DISABLE_AFTER_TIMEOUT #ifndef RGB_DISABLE_AFTER_TIMEOUT
@ -55,25 +53,25 @@ uint32_t g_any_key_hit = 0;
#define PI 3.14159265 #define PI 3.14159265
#endif #endif
uint32_t eeconfig_read_rgblight(void) { uint32_t eeconfig_read_rgb_matrix(void) {
return eeprom_read_dword(EECONFIG_RGBLIGHT); return eeprom_read_dword(EECONFIG_RGBLIGHT);
} }
void eeconfig_update_rgblight(uint32_t val) { void eeconfig_update_rgb_matrix(uint32_t val) {
eeprom_update_dword(EECONFIG_RGBLIGHT, val); eeprom_update_dword(EECONFIG_RGBLIGHT, val);
} }
void eeconfig_update_rgblight_default(void) { void eeconfig_update_rgb_matrix_default(void) {
dprintf("eeconfig_update_rgblight_default\n"); dprintf("eeconfig_update_rgb_matrix_default\n");
rgb_matrix_config.enable = 1; rgb_matrix_config.enable = 1;
rgb_matrix_config.mode = 7; rgb_matrix_config.mode = RGB_MATRIX_CYCLE_LEFT_RIGHT;
rgb_matrix_config.hue = 0; rgb_matrix_config.hue = 0;
rgb_matrix_config.sat = 255; rgb_matrix_config.sat = 255;
rgb_matrix_config.val = 255; rgb_matrix_config.val = 255;
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void eeconfig_debug_rgblight(void) { void eeconfig_debug_rgb_matrix(void) {
dprintf("rgb_matrix_config eprom\n"); dprintf("rgb_matrix_config eprom\n");
dprintf("rgb_matrix_config.enable = %d\n", rgb_matrix_config.enable); dprintf("rgb_matrix_config.enable = %d\n", rgb_matrix_config.enable);
dprintf("rghlight_config.mode = %d\n", rgb_matrix_config.mode); dprintf("rgb_matrix_config.mode = %d\n", rgb_matrix_config.mode);
dprintf("rgb_matrix_config.hue = %d\n", rgb_matrix_config.hue); dprintf("rgb_matrix_config.hue = %d\n", rgb_matrix_config.hue);
dprintf("rgb_matrix_config.sat = %d\n", rgb_matrix_config.sat); dprintf("rgb_matrix_config.sat = %d\n", rgb_matrix_config.sat);
dprintf("rgb_matrix_config.val = %d\n", rgb_matrix_config.val); dprintf("rgb_matrix_config.val = %d\n", rgb_matrix_config.val);
@ -84,8 +82,7 @@ void eeconfig_debug_rgblight(void) {
uint8_t g_last_led_hit[LED_HITS_TO_REMEMBER] = {255}; uint8_t g_last_led_hit[LED_HITS_TO_REMEMBER] = {255};
uint8_t g_last_led_count = 0; uint8_t g_last_led_count = 0;
void map_row_column_to_led( uint8_t row, uint8_t column, uint8_t *led_i, uint8_t *led_count) void map_row_column_to_led( uint8_t row, uint8_t column, uint8_t *led_i, uint8_t *led_count) {
{
rgb_led led; rgb_led led;
*led_count = 0; *led_count = 0;
@ -100,120 +97,78 @@ void map_row_column_to_led( uint8_t row, uint8_t column, uint8_t *led_i, uint8_t
} }
void backlight_update_pwm_buffers(void) void rgb_matrix_update_pwm_buffers(void) {
{
IS31FL3731_update_pwm_buffers( DRIVER_ADDR_1, DRIVER_ADDR_2 ); IS31FL3731_update_pwm_buffers( DRIVER_ADDR_1, DRIVER_ADDR_2 );
IS31FL3731_update_led_control_registers( DRIVER_ADDR_1, DRIVER_ADDR_2 ); IS31FL3731_update_led_control_registers( DRIVER_ADDR_1, DRIVER_ADDR_2 );
} }
void backlight_set_color( int index, uint8_t red, uint8_t green, uint8_t blue ) void rgb_matrix_set_color( int index, uint8_t red, uint8_t green, uint8_t blue ) {
{
IS31FL3731_set_color( index, red, green, blue ); IS31FL3731_set_color( index, red, green, blue );
} }
void backlight_set_color_all( uint8_t red, uint8_t green, uint8_t blue ) void rgb_matrix_set_color_all( uint8_t red, uint8_t green, uint8_t blue ) {
{
IS31FL3731_set_color_all( red, green, blue ); IS31FL3731_set_color_all( red, green, blue );
} }
void backlight_set_key_hit(uint8_t row, uint8_t column)
{ bool process_rgb_matrix(uint16_t keycode, keyrecord_t *record) {
uint8_t led[8], led_count; if ( record->event.pressed ) {
map_row_column_to_led(row,column,led,&led_count); uint8_t led[8], led_count;
if (led_count > 0) { map_row_column_to_led(record->event.key.row, record->event.key.col, led, &led_count);
for (uint8_t i = LED_HITS_TO_REMEMBER; i > 1; i--) { if (led_count > 0) {
g_last_led_hit[i - 1] = g_last_led_hit[i - 2]; for (uint8_t i = LED_HITS_TO_REMEMBER; i > 1; i--) {
g_last_led_hit[i - 1] = g_last_led_hit[i - 2];
}
g_last_led_hit[0] = led[0];
g_last_led_count = MIN(LED_HITS_TO_REMEMBER, g_last_led_count + 1);
} }
g_last_led_hit[0] = led[0]; for(uint8_t i = 0; i < led_count; i++)
g_last_led_count = MIN(LED_HITS_TO_REMEMBER, g_last_led_count + 1); g_key_hit[led[i]] = 0;
g_any_key_hit = 0;
} else {
#ifdef RGB_MATRIX_KEYRELEASES
uint8_t led[8], led_count;
map_row_column_to_led(record->event.key.row, record->event.key.col, led, &led_count);
for(uint8_t i = 0; i < led_count; i++)
g_key_hit[led[i]] = 255;
g_any_key_hit = 255;
#endif
} }
for(uint8_t i = 0; i < led_count; i++) return true;
g_key_hit[led[i]] = 0;
g_any_key_hit = 0;
} }
void backlight_unset_key_hit(uint8_t row, uint8_t column) void rgb_matrix_set_suspend_state(bool state) {
{
uint8_t led[8], led_count;
map_row_column_to_led(row,column,led,&led_count);
for(uint8_t i = 0; i < led_count; i++)
g_key_hit[led[i]] = 255;
g_any_key_hit = 255;
}
// This is (F_CPU/1024) / 20 Hz
// = 15625 Hz / 20 Hz
// = 781
// #define TIMER3_TOP 781
void backlight_timer_init(void)
{
static uint8_t backlight_timer_is_init = 0;
if ( backlight_timer_is_init )
{
return;
}
backlight_timer_is_init = 1;
// Timer 3 setup
//TCCR3B = _BV(WGM32) | // CTC mode OCR3A as TOP
// _BV(CS32) | _BV(CS30); // prescale by /1024
// Set TOP value
//uint8_t sreg = SREG;
//cli();
//OCR3AH = (TIMER3_TOP >> 8) & 0xff;
//OCR3AL = TIMER3_TOP & 0xff;
//SREG = sreg;
}
void backlight_timer_enable(void)
{
//TIMSK3 |= _BV(OCIE3A);
}
void backlight_timer_disable(void)
{
//TIMSK3 &= ~_BV(OCIE3A);
}
void backlight_set_suspend_state(bool state)
{
g_suspend_state = state; g_suspend_state = state;
} }
void backlight_set_indicator_state(uint8_t state) void rgb_matrix_set_indicator_state(uint8_t state) {
{
g_indicator_state = state; g_indicator_state = state;
} }
void backlight_effect_rgb_test(void) void rgb_matrix_test(void) {
{
// Mask out bits 4 and 5 // Mask out bits 4 and 5
// This 2-bit value will stay the same for 16 ticks. // This 2-bit value will stay the same for 16 ticks.
switch ( (g_tick & 0x30) >> 4 ) switch ( (g_tick & 0x30) >> 4 )
{ {
case 0: case 0:
{ {
backlight_set_color_all( 20, 0, 0 ); rgb_matrix_set_color_all( 20, 0, 0 );
break; break;
} }
case 1: case 1:
{ {
backlight_set_color_all( 0, 20, 0 ); rgb_matrix_set_color_all( 0, 20, 0 );
break; break;
} }
case 2: case 2:
{ {
backlight_set_color_all( 0, 0, 20 ); rgb_matrix_set_color_all( 0, 0, 20 );
break; break;
} }
case 3: case 3:
{ {
backlight_set_color_all( 20, 20, 20 ); rgb_matrix_set_color_all( 20, 20, 20 );
break; break;
} }
} }
@ -224,8 +179,7 @@ void backlight_effect_rgb_test(void)
// in the LED drivers, and leave them in an invalid // in the LED drivers, and leave them in an invalid
// state for other backlight effects. // state for other backlight effects.
// ONLY USE THIS FOR TESTING LEDS! // ONLY USE THIS FOR TESTING LEDS!
void backlight_effect_single_LED_test(void) void rgb_matrix_single_LED_test(void) {
{
static uint8_t color = 0; // 0,1,2 for R,G,B static uint8_t color = 0; // 0,1,2 for R,G,B
static uint8_t row = 0; static uint8_t row = 0;
static uint8_t column = 0; static uint8_t column = 0;
@ -256,27 +210,24 @@ void backlight_effect_single_LED_test(void)
uint8_t led[8], led_count; uint8_t led[8], led_count;
map_row_column_to_led(row,column,led,&led_count); map_row_column_to_led(row,column,led,&led_count);
for(uint8_t i = 0; i < led_count; i++) { for(uint8_t i = 0; i < led_count; i++) {
backlight_set_color_all( 40, 40, 40 ); rgb_matrix_set_color_all( 40, 40, 40 );
backlight_test_led( led[i], color==0, color==1, color==2 ); rgb_matrix_test_led( led[i], color==0, color==1, color==2 );
} }
} }
// All LEDs off // All LEDs off
void backlight_effect_all_off(void) void rgb_matrix_all_off(void) {
{ rgb_matrix_set_color_all( 0, 0, 0 );
backlight_set_color_all( 0, 0, 0 );
} }
// Solid color // Solid color
void backlight_effect_solid_color(void) void rgb_matrix_solid_color(void) {
{
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb = hsv_to_rgb( hsv ); RGB rgb = hsv_to_rgb( hsv );
backlight_set_color_all( rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color_all( rgb.r, rgb.g, rgb.b );
} }
void backlight_effect_solid_reactive(void) void rgb_matrix_solid_reactive(void) {
{
// Relies on hue being 8-bit and wrapping // Relies on hue being 8-bit and wrapping
for ( int i=0; i<DRIVER_LED_TOTAL; i++ ) for ( int i=0; i<DRIVER_LED_TOTAL; i++ )
{ {
@ -285,13 +236,13 @@ void backlight_effect_solid_reactive(void)
HSV hsv = { .h = rgb_matrix_config.hue+offset2, .s = 255, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue+offset2, .s = 255, .v = rgb_matrix_config.val };
RGB rgb = hsv_to_rgb( hsv ); RGB rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
// alphas = color1, mods = color2 // alphas = color1, mods = color2
void backlight_effect_alphas_mods(void) void rgb_matrix_alphas_mods(void) {
{
RGB rgb1 = hsv_to_rgb( (HSV){ .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val } ); RGB rgb1 = hsv_to_rgb( (HSV){ .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val } );
RGB rgb2 = hsv_to_rgb( (HSV){ .h = (rgb_matrix_config.hue + 180) % 360, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val } ); RGB rgb2 = hsv_to_rgb( (HSV){ .h = (rgb_matrix_config.hue + 180) % 360, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val } );
@ -301,18 +252,17 @@ void backlight_effect_alphas_mods(void)
if ( led.matrix_co.raw < 0xFF ) { if ( led.matrix_co.raw < 0xFF ) {
if ( led.modifier ) if ( led.modifier )
{ {
backlight_set_color( i, rgb2.r, rgb2.g, rgb2.b ); rgb_matrix_set_color( i, rgb2.r, rgb2.g, rgb2.b );
} }
else else
{ {
backlight_set_color( i, rgb1.r, rgb1.g, rgb1.b ); rgb_matrix_set_color( i, rgb1.r, rgb1.g, rgb1.b );
} }
} }
} }
} }
void backlight_effect_gradient_up_down(void) void rgb_matrix_gradient_up_down(void) {
{
int16_t h1 = rgb_matrix_config.hue; int16_t h1 = rgb_matrix_config.hue;
int16_t h2 = (rgb_matrix_config.hue + 180) % 360; int16_t h2 = (rgb_matrix_config.hue + 180) % 360;
int16_t deltaH = h2 - h1; int16_t deltaH = h2 - h1;
@ -346,12 +296,11 @@ void backlight_effect_gradient_up_down(void)
hsv.h = rgb_matrix_config.hue + ( deltaH * y ); hsv.h = rgb_matrix_config.hue + ( deltaH * y );
hsv.s = rgb_matrix_config.sat + ( deltaS * y ); hsv.s = rgb_matrix_config.sat + ( deltaS * y );
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
void backlight_effect_raindrops(bool initialize) void rgb_matrix_raindrops(bool initialize) {
{
int16_t h1 = rgb_matrix_config.hue; int16_t h1 = rgb_matrix_config.hue;
int16_t h2 = (rgb_matrix_config.hue + 180) % 360; int16_t h2 = (rgb_matrix_config.hue + 180) % 360;
int16_t deltaH = h2 - h1; int16_t deltaH = h2 - h1;
@ -389,13 +338,12 @@ void backlight_effect_raindrops(bool initialize)
hsv.v = rgb_matrix_config.val; hsv.v = rgb_matrix_config.val;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
} }
void backlight_effect_cycle_all(void) void rgb_matrix_cycle_all(void) {
{
uint8_t offset = g_tick & 0xFF; uint8_t offset = g_tick & 0xFF;
rgb_led led; rgb_led led;
@ -411,13 +359,12 @@ void backlight_effect_cycle_all(void)
HSV hsv = { .h = offset+offset2, .s = 255, .v = rgb_matrix_config.val }; HSV hsv = { .h = offset+offset2, .s = 255, .v = rgb_matrix_config.val };
RGB rgb = hsv_to_rgb( hsv ); RGB rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
} }
void backlight_effect_cycle_left_right(void) void rgb_matrix_cycle_left_right(void) {
{
uint8_t offset = g_tick & 0xFF; uint8_t offset = g_tick & 0xFF;
HSV hsv = { .h = 0, .s = 255, .v = rgb_matrix_config.val }; HSV hsv = { .h = 0, .s = 255, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
@ -436,13 +383,12 @@ void backlight_effect_cycle_left_right(void)
// Relies on hue being 8-bit and wrapping // Relies on hue being 8-bit and wrapping
hsv.h = point.x + offset + offset2; hsv.h = point.x + offset + offset2;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
} }
void backlight_effect_cycle_up_down(void) void rgb_matrix_cycle_up_down(void) {
{
uint8_t offset = g_tick & 0xFF; uint8_t offset = g_tick & 0xFF;
HSV hsv = { .h = 0, .s = 255, .v = rgb_matrix_config.val }; HSV hsv = { .h = 0, .s = 255, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
@ -461,13 +407,13 @@ void backlight_effect_cycle_up_down(void)
// Relies on hue being 8-bit and wrapping // Relies on hue being 8-bit and wrapping
hsv.h = point.y + offset + offset2; hsv.h = point.y + offset + offset2;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
} }
void backlight_effect_dual_beacon(void) { void rgb_matrix_dual_beacon(void) {
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
rgb_led led; rgb_led led;
@ -475,11 +421,11 @@ void backlight_effect_dual_beacon(void) {
led = g_rgb_leds[i]; led = g_rgb_leds[i];
hsv.h = ((led.point.y - 32.0)* cos(g_tick * PI / 128) / 32 + (led.point.x - 112.0) * sin(g_tick * PI / 128) / (112)) * (180) + rgb_matrix_config.hue; hsv.h = ((led.point.y - 32.0)* cos(g_tick * PI / 128) / 32 + (led.point.x - 112.0) * sin(g_tick * PI / 128) / (112)) * (180) + rgb_matrix_config.hue;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
void backlight_effect_rainbow_beacon(void) { void rgb_matrix_rainbow_beacon(void) {
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
rgb_led led; rgb_led led;
@ -487,11 +433,11 @@ void backlight_effect_rainbow_beacon(void) {
led = g_rgb_leds[i]; led = g_rgb_leds[i];
hsv.h = 1.5 * (led.point.y - 32.0)* cos(g_tick * PI / 128) + 1.5 * (led.point.x - 112.0) * sin(g_tick * PI / 128) + rgb_matrix_config.hue; hsv.h = 1.5 * (led.point.y - 32.0)* cos(g_tick * PI / 128) + 1.5 * (led.point.x - 112.0) * sin(g_tick * PI / 128) + rgb_matrix_config.hue;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
void backlight_effect_rainbow_pinwheels(void) { void rgb_matrix_rainbow_pinwheels(void) {
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
rgb_led led; rgb_led led;
@ -499,11 +445,11 @@ void backlight_effect_rainbow_pinwheels(void) {
led = g_rgb_leds[i]; led = g_rgb_leds[i];
hsv.h = 2 * (led.point.y - 32.0)* cos(g_tick * PI / 128) + 2 * (66 - abs(led.point.x - 112.0)) * sin(g_tick * PI / 128) + rgb_matrix_config.hue; hsv.h = 2 * (led.point.y - 32.0)* cos(g_tick * PI / 128) + 2 * (66 - abs(led.point.x - 112.0)) * sin(g_tick * PI / 128) + rgb_matrix_config.hue;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
void backlight_effect_rainbow_moving_chevron(void) { void rgb_matrix_rainbow_moving_chevron(void) {
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
rgb_led led; rgb_led led;
@ -513,13 +459,12 @@ void backlight_effect_rainbow_moving_chevron(void) {
uint8_t r = 32; uint8_t r = 32;
hsv.h = 1.5 * abs(led.point.y - 32.0)* sin(r * PI / 128) + 1.5 * (led.point.x - (g_tick / 256.0 * 224)) * cos(r * PI / 128) + rgb_matrix_config.hue; hsv.h = 1.5 * abs(led.point.y - 32.0)* sin(r * PI / 128) + 1.5 * (led.point.x - (g_tick / 256.0 * 224)) * cos(r * PI / 128) + rgb_matrix_config.hue;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
void backlight_effect_jellybean_raindrops( bool initialize ) void rgb_matrix_jellybean_raindrops( bool initialize ) {
{
HSV hsv; HSV hsv;
RGB rgb; RGB rgb;
@ -538,12 +483,12 @@ void backlight_effect_jellybean_raindrops( bool initialize )
hsv.v = rgb_matrix_config.val; hsv.v = rgb_matrix_config.val;
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
} }
} }
void backlight_effect_multisplash(void) { void rgb_matrix_multisplash(void) {
// if (g_any_key_hit < 0xFF) { // if (g_any_key_hit < 0xFF) {
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
@ -566,21 +511,21 @@ void backlight_effect_multisplash(void) {
hsv.h = (rgb_matrix_config.hue + c) % 256; hsv.h = (rgb_matrix_config.hue + c) % 256;
hsv.v = MAX(MIN(d, 255), 0); hsv.v = MAX(MIN(d, 255), 0);
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
// } else { // } else {
// backlight_set_color_all( 0, 0, 0 ); // rgb_matrix_set_color_all( 0, 0, 0 );
// } // }
} }
void backlight_effect_splash(void) { void rgb_matrix_splash(void) {
g_last_led_count = MIN(g_last_led_count, 1); g_last_led_count = MIN(g_last_led_count, 1);
backlight_effect_multisplash(); rgb_matrix_multisplash();
} }
void backlight_effect_solid_multisplash(void) { void rgb_matrix_solid_multisplash(void) {
// if (g_any_key_hit < 0xFF) { // if (g_any_key_hit < 0xFF) {
HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val }; HSV hsv = { .h = rgb_matrix_config.hue, .s = rgb_matrix_config.sat, .v = rgb_matrix_config.val };
RGB rgb; RGB rgb;
@ -601,23 +546,23 @@ void backlight_effect_solid_multisplash(void) {
// } // }
hsv.v = MAX(MIN(d, 255), 0); hsv.v = MAX(MIN(d, 255), 0);
rgb = hsv_to_rgb( hsv ); rgb = hsv_to_rgb( hsv );
backlight_set_color( i, rgb.r, rgb.g, rgb.b ); rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
} }
// } else { // } else {
// backlight_set_color_all( 0, 0, 0 ); // rgb_matrix_set_color_all( 0, 0, 0 );
// } // }
} }
void backlight_effect_solid_splash(void) { void rgb_matrix_solid_splash(void) {
g_last_led_count = MIN(g_last_led_count, 1); g_last_led_count = MIN(g_last_led_count, 1);
backlight_effect_solid_multisplash(); rgb_matrix_solid_multisplash();
} }
// Needs eeprom access that we don't have setup currently // Needs eeprom access that we don't have setup currently
void backlight_effect_custom(void) { void rgb_matrix_custom(void) {
// HSV hsv; // HSV hsv;
// RGB rgb; // RGB rgb;
// for ( int i=0; i<DRIVER_LED_TOTAL; i++ ) // for ( int i=0; i<DRIVER_LED_TOTAL; i++ )
@ -626,34 +571,30 @@ void backlight_effect_custom(void) {
// // Override brightness with global brightness control // // Override brightness with global brightness control
// hsv.v = rgb_matrix_config.val; // hsv.v = rgb_matrix_config.val;
// rgb = hsv_to_rgb( hsv ); // rgb = hsv_to_rgb( hsv );
// backlight_set_color( i, rgb.r, rgb.g, rgb.b ); // rgb_matrix_set_color( i, rgb.r, rgb.g, rgb.b );
// } // }
} }
void backlight_rgb_task(void) { void rgb_matrix_task(void) {
if (!rgb_matrix_config.enable) { if (!rgb_matrix_config.enable) {
backlight_effect_all_off(); rgb_matrix_all_off();
return; return;
} }
// delay 1 second before driving LEDs or doing anything else // delay 1 second before driving LEDs or doing anything else
static uint8_t startup_tick = 0; static uint8_t startup_tick = 0;
if ( startup_tick < 20 ) if ( startup_tick < 20 ) {
{
startup_tick++; startup_tick++;
return; return;
} }
g_tick++; g_tick++;
if ( g_any_key_hit < 0xFFFFFFFF ) if ( g_any_key_hit < 0xFFFFFFFF ) {
{
g_any_key_hit++; g_any_key_hit++;
} }
for ( int led = 0; led < DRIVER_LED_TOTAL; led++ ) for ( int led = 0; led < DRIVER_LED_TOTAL; led++ ) {
{ if ( g_key_hit[led] < 255 ) {
if ( g_key_hit[led] < 255 )
{
if (g_key_hit[led] == 254) if (g_key_hit[led] == 254)
g_last_led_count = MAX(g_last_led_count - 1, 0); g_last_led_count = MAX(g_last_led_count - 1, 0);
g_key_hit[led]++; g_key_hit[led]++;
@ -661,9 +602,8 @@ void backlight_rgb_task(void) {
} }
// Factory default magic value // Factory default magic value
if ( rgb_matrix_config.mode == 255 ) if ( rgb_matrix_config.mode == 255 ) {
{ rgb_matrix_test();
backlight_effect_rgb_test();
return; return;
} }
@ -683,68 +623,67 @@ void backlight_rgb_task(void) {
// this gets ticked at 20 Hz. // this gets ticked at 20 Hz.
// each effect can opt to do calculations // each effect can opt to do calculations
// and/or request PWM buffer updates. // and/or request PWM buffer updates.
switch ( effect ) switch ( effect ) {
{ case RGB_MATRIX_SOLID_COLOR:
case 0: rgb_matrix_solid_color();
backlight_effect_solid_color();
break; break;
case 1: case RGB_MATRIX_SOLID_REACTIVE:
backlight_effect_solid_reactive(); rgb_matrix_solid_reactive();
break; break;
case 2: case RGB_MATRIX_ALPHAS_MODS:
backlight_effect_alphas_mods(); rgb_matrix_alphas_mods();
break; break;
case 3: case RGB_MATRIX_DUAL_BEACON:
backlight_effect_dual_beacon(); rgb_matrix_dual_beacon();
break; break;
case 4: case RGB_MATRIX_GRADIENT_UP_DOWN:
backlight_effect_gradient_up_down(); rgb_matrix_gradient_up_down();
break; break;
case 5: case RGB_MATRIX_RAINDROPS:
backlight_effect_raindrops( initialize ); rgb_matrix_raindrops( initialize );
break; break;
case 6: case RGB_MATRIX_CYCLE_ALL:
backlight_effect_cycle_all(); rgb_matrix_cycle_all();
break; break;
case 7: case RGB_MATRIX_CYCLE_LEFT_RIGHT:
backlight_effect_cycle_left_right(); rgb_matrix_cycle_left_right();
break; break;
case 8: case RGB_MATRIX_CYCLE_UP_DOWN:
backlight_effect_cycle_up_down(); rgb_matrix_cycle_up_down();
break; break;
case 9: case RGB_MATRIX_RAINBOW_BEACON:
backlight_effect_rainbow_beacon(); rgb_matrix_rainbow_beacon();
break; break;
case 10: case RGB_MATRIX_RAINBOW_PINWHEELS:
backlight_effect_rainbow_pinwheels(); rgb_matrix_rainbow_pinwheels();
break; break;
case 11: case RGB_MATRIX_RAINBOW_MOVING_CHEVRON:
backlight_effect_rainbow_moving_chevron(); rgb_matrix_rainbow_moving_chevron();
break; break;
case 12: case RGB_MATRIX_JELLYBEAN_RAINDROPS:
backlight_effect_jellybean_raindrops( initialize ); rgb_matrix_jellybean_raindrops( initialize );
break; break;
case 13: #ifdef RGB_MATRIX_KEYPRESSES
backlight_effect_splash(); case RGB_MATRIX_SPLASH:
break; rgb_matrix_splash();
case 14: break;
backlight_effect_multisplash(); case RGB_MATRIX_MULTISPLASH:
break; rgb_matrix_multisplash();
case 15: break;
backlight_effect_solid_splash(); case RGB_MATRIX_SOLID_SPLASH:
break; rgb_matrix_solid_splash();
case 16: break;
backlight_effect_solid_multisplash(); case RGB_MATRIX_SOLID_MULTISPLASH:
break; rgb_matrix_solid_multisplash();
case 17: break;
#endif
default: default:
backlight_effect_custom(); rgb_matrix_custom();
break; break;
} }
if ( ! suspend_backlight ) if ( ! suspend_backlight ) {
{ rgb_matrix_indicators();
backlight_effect_indicators();
} }
} }
@ -766,8 +705,7 @@ void backlight_rgb_task(void) {
// } // }
// } // }
void backlight_init_drivers(void) void rgb_matrix_init_drivers(void) {
{
//sei(); //sei();
// Initialize TWI // Initialize TWI
@ -775,8 +713,7 @@ void backlight_init_drivers(void)
IS31FL3731_init( DRIVER_ADDR_1 ); IS31FL3731_init( DRIVER_ADDR_1 );
IS31FL3731_init( DRIVER_ADDR_2 ); IS31FL3731_init( DRIVER_ADDR_2 );
for ( int index = 0; index < DRIVER_LED_TOTAL; index++ ) for ( int index = 0; index < DRIVER_LED_TOTAL; index++ ) {
{
bool enabled = true; bool enabled = true;
// This only caches it for later // This only caches it for later
IS31FL3731_set_led_control_register( index, enabled, enabled, enabled ); IS31FL3731_set_led_control_register( index, enabled, enabled, enabled );
@ -787,36 +724,33 @@ void backlight_init_drivers(void)
// TODO: put the 1 second startup delay here? // TODO: put the 1 second startup delay here?
// clear the key hits // clear the key hits
for ( int led=0; led<DRIVER_LED_TOTAL; led++ ) for ( int led=0; led<DRIVER_LED_TOTAL; led++ ) {
{
g_key_hit[led] = 255; g_key_hit[led] = 255;
} }
if (!eeconfig_is_enabled()) { if (!eeconfig_is_enabled()) {
dprintf("backlight_init_drivers eeconfig is not enabled.\n"); dprintf("rgb_matrix_init_drivers eeconfig is not enabled.\n");
eeconfig_init(); eeconfig_init();
eeconfig_update_rgblight_default(); eeconfig_update_rgb_matrix_default();
} }
rgb_matrix_config.raw = eeconfig_read_rgblight(); rgb_matrix_config.raw = eeconfig_read_rgb_matrix();
if (!rgb_matrix_config.mode) { if (!rgb_matrix_config.mode) {
dprintf("backlight_init_drivers rgb_matrix_config.mode = 0. Write default values to EEPROM.\n"); dprintf("rgb_matrix_init_drivers rgb_matrix_config.mode = 0. Write default values to EEPROM.\n");
eeconfig_update_rgblight_default(); eeconfig_update_rgb_matrix_default();
rgb_matrix_config.raw = eeconfig_read_rgblight(); rgb_matrix_config.raw = eeconfig_read_rgb_matrix();
} }
eeconfig_debug_rgblight(); // display current eeprom values eeconfig_debug_rgb_matrix(); // display current eeprom values
} }
// Deals with the messy details of incrementing an integer // Deals with the messy details of incrementing an integer
uint8_t increment( uint8_t value, uint8_t step, uint8_t min, uint8_t max ) uint8_t increment( uint8_t value, uint8_t step, uint8_t min, uint8_t max ) {
{
int16_t new_value = value; int16_t new_value = value;
new_value += step; new_value += step;
return MIN( MAX( new_value, min ), max ); return MIN( MAX( new_value, min ), max );
} }
uint8_t decrement( uint8_t value, uint8_t step, uint8_t min, uint8_t max ) uint8_t decrement( uint8_t value, uint8_t step, uint8_t min, uint8_t max ) {
{
int16_t new_value = value; int16_t new_value = value;
new_value -= step; new_value -= step;
return MIN( MAX( new_value, min ), max ); return MIN( MAX( new_value, min ), max );
@ -851,8 +785,7 @@ uint8_t decrement( uint8_t value, uint8_t step, uint8_t min, uint8_t max )
// } // }
// } // }
void backlight_test_led( uint8_t index, bool red, bool green, bool blue ) void rgb_matrix_test_led( uint8_t index, bool red, bool green, bool blue ) {
{
for ( int i=0; i<DRIVER_LED_TOTAL; i++ ) for ( int i=0; i<DRIVER_LED_TOTAL; i++ )
{ {
if ( i == index ) if ( i == index )
@ -866,61 +799,62 @@ void backlight_test_led( uint8_t index, bool red, bool green, bool blue )
} }
} }
uint32_t backlight_get_tick(void) uint32_t rgb_matrix_get_tick(void) {
{
return g_tick; return g_tick;
} }
void rgblight_toggle(void) { void rgblight_toggle(void) {
rgb_matrix_config.enable ^= 1; rgb_matrix_config.enable ^= 1;
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_step(void) { void rgblight_step(void) {
rgb_matrix_config.mode = (rgb_matrix_config.mode + 1) % (RGB_MATRIX_EFFECT_MAX + 1); rgb_matrix_config.mode++;
eeconfig_update_rgblight(rgb_matrix_config.raw); if (rgb_matrix_config.mode >= RGB_MATRIX_EFFECT_MAX)
rgb_matrix_config.mode = 1;
eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_step_reverse(void) { void rgblight_step_reverse(void) {
if (rgb_matrix_config.mode > 1) { rgb_matrix_config.mode--;
rgb_matrix_config.mode = (rgb_matrix_config.mode - 1) % (RGB_MATRIX_EFFECT_MAX + 1); if (rgb_matrix_config.mode <= 1)
eeconfig_update_rgblight(rgb_matrix_config.raw); rgb_matrix_config.mode = (RGB_MATRIX_EFFECT_MAX - 1);
} eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_increase_hue(void) { void rgblight_increase_hue(void) {
rgb_matrix_config.hue = increment( rgb_matrix_config.hue, 8, 0, 255 ); rgb_matrix_config.hue = increment( rgb_matrix_config.hue, 8, 0, 255 );
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_decrease_hue(void) { void rgblight_decrease_hue(void) {
rgb_matrix_config.hue = decrement( rgb_matrix_config.hue, 8, 0, 255 ); rgb_matrix_config.hue = decrement( rgb_matrix_config.hue, 8, 0, 255 );
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_increase_sat(void) { void rgblight_increase_sat(void) {
rgb_matrix_config.sat = increment( rgb_matrix_config.sat, 8, 0, 255 ); rgb_matrix_config.sat = increment( rgb_matrix_config.sat, 8, 0, 255 );
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_decrease_sat(void) { void rgblight_decrease_sat(void) {
rgb_matrix_config.sat = decrement( rgb_matrix_config.sat, 8, 0, 255 ); rgb_matrix_config.sat = decrement( rgb_matrix_config.sat, 8, 0, 255 );
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_increase_val(void) { void rgblight_increase_val(void) {
rgb_matrix_config.val = increment( rgb_matrix_config.val, 8, 0, 255 ); rgb_matrix_config.val = increment( rgb_matrix_config.val, 8, 0, 255 );
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_decrease_val(void) { void rgblight_decrease_val(void) {
rgb_matrix_config.val = decrement( rgb_matrix_config.val, 8, 0, 255 ); rgb_matrix_config.val = decrement( rgb_matrix_config.val, 8, 0, 255 );
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
void rgblight_mode(uint8_t mode) { void rgblight_mode(uint8_t mode) {
rgb_matrix_config.mode = mode; rgb_matrix_config.mode = mode;
eeconfig_update_rgblight(rgb_matrix_config.raw); eeconfig_update_rgb_matrix(rgb_matrix_config.raw);
} }
uint32_t rgblight_get_mode(void) { uint32_t rgblight_get_mode(void) {

View File

@ -61,48 +61,61 @@ typedef union {
}; };
} rgb_config_t; } rgb_config_t;
enum rgb_matrix_effects {
RGB_MATRIX_SOLID_COLOR = 1,
RGB_MATRIX_SOLID_REACTIVE,
RGB_MATRIX_ALPHAS_MODS,
RGB_MATRIX_DUAL_BEACON,
RGB_MATRIX_GRADIENT_UP_DOWN,
RGB_MATRIX_RAINDROPS,
RGB_MATRIX_CYCLE_ALL,
RGB_MATRIX_CYCLE_LEFT_RIGHT,
RGB_MATRIX_CYCLE_UP_DOWN,
RGB_MATRIX_RAINBOW_BEACON,
RGB_MATRIX_RAINBOW_PINWHEELS,
RGB_MATRIX_RAINBOW_MOVING_CHEVRON,
RGB_MATRIX_JELLYBEAN_RAINDROPS,
#ifdef RGB_MATRIX_KEYPRESSES
RGB_MATRIX_SPLASH,
RGB_MATRIX_MULTISPLASH,
RGB_MATRIX_SOLID_SPLASH,
RGB_MATRIX_SOLID_MULTISPLASH,
#endif
RGB_MATRIX_EFFECT_MAX
};
// This runs after another backlight effect and replaces // This runs after another backlight effect and replaces
// colors already set // colors already set
__attribute__((weak)) __attribute__((weak))
void backlight_effect_indicators(void) {}; void rgb_matrix_indicators(void) {};
void backlight_effect_single_LED_test(void); void rgb_matrix_single_LED_test(void);
void backlight_config_set_alphas_mods( uint16_t *value ); void rgb_matrix_init_drivers(void);
void backlight_config_load(void);
void backlight_config_save(void);
void backlight_init_drivers(void); void rgb_matrix_set_suspend_state(bool state);
void rgb_matrix_set_indicator_state(uint8_t state);
void backlight_timer_init(void);
void backlight_timer_enable(void);
void backlight_timer_disable(void);
void backlight_set_suspend_state(bool state);
void backlight_set_indicator_state(uint8_t state);
void backlight_rgb_task(void); void rgb_matrix_task(void);
// This should not be called from an interrupt // This should not be called from an interrupt
// (eg. from a timer interrupt). // (eg. from a timer interrupt).
// Call this while idle (in between matrix scans). // Call this while idle (in between matrix scans).
// If the buffer is dirty, it will update the driver with the buffer. // If the buffer is dirty, it will update the driver with the buffer.
void backlight_update_pwm_buffers(void); void rgb_matrix_update_pwm_buffers(void);
void backlight_set_key_hit(uint8_t row, uint8_t col); bool process_rgb_matrix(uint16_t keycode, keyrecord_t *record);
void backlight_unset_key_hit(uint8_t row, uint8_t col);
void backlight_effect_increase(void); void rgb_matrix_increase(void);
void backlight_effect_decrease(void); void rgb_matrix_decrease(void);
void *backlight_get_key_color_eeprom_address(uint8_t led); // void *backlight_get_key_color_eeprom_address(uint8_t led);
void backlight_get_key_color( uint8_t led, HSV *hsv ); // void backlight_get_key_color( uint8_t led, HSV *hsv );
void backlight_set_key_color( uint8_t row, uint8_t column, HSV hsv ); // void backlight_set_key_color( uint8_t row, uint8_t column, HSV hsv );
void backlight_test_led( uint8_t index, bool red, bool green, bool blue ); void rgb_matrix_test_led( uint8_t index, bool red, bool green, bool blue );
uint32_t backlight_get_tick(void); uint32_t rgb_matrix_get_tick(void);
void backlight_debug_led(bool state);
void rgblight_toggle(void); void rgblight_toggle(void);
void rgblight_step(void); void rgblight_step(void);