qmk_firmware/keyboards/system76/launch_1/rgb_matrix_kb.inc

157 lines
4.5 KiB
C++

/*
* Copyright (C) 2021 System76
*
* 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 3 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 <https://www.gnu.org/licenses/>.
*/
RGB_MATRIX_EFFECT(active_keys)
RGB_MATRIX_EFFECT(raw_rgb)
RGB_MATRIX_EFFECT(unlocked)
#ifdef RGB_MATRIX_CUSTOM_EFFECT_IMPLS
#include "dynamic_keymap.h"
static bool active_keys_initialized = false;
static uint8_t active_keys_table[RGB_MATRIX_LED_COUNT] = {0};
static void active_keys_initialize(void) {
for (uint8_t row = 0; row < MATRIX_ROWS; row++) {
for (uint8_t col = 0; col < MATRIX_COLS; col++) {
uint8_t led = g_led_config.matrix_co[row][col];
if (led < RGB_MATRIX_LED_COUNT && row < 16 && col < 16) {
active_keys_table[led] = (row << 4) | col;
}
}
}
active_keys_initialized = true;
}
static bool active_keys(effect_params_t* params) {
if (!active_keys_initialized) {
active_keys_initialize();
}
RGB_MATRIX_USE_LIMITS(led_min, led_max);
uint8_t layer = get_highest_layer(layer_state);
RGB rgb = hsv_to_rgb(rgb_matrix_config.hsv);
for (uint8_t i = led_min; i < led_max; i++) {
RGB_MATRIX_TEST_LED_FLAGS();
uint8_t rowcol = active_keys_table[i];
uint8_t row = rowcol >> 4;
uint8_t col = rowcol & 0xF;
uint16_t keycode = dynamic_keymap_get_keycode(layer, row, col);
switch (keycode) {
case KC_NO:
case KC_TRNS:
rgb_matrix_set_color(i, 0, 0, 0);
break;
default:
rgb_matrix_set_color(i, rgb.r, rgb.g, rgb.b);
break;
}
}
return led_max < RGB_MATRIX_LED_COUNT;
}
RGB raw_rgb_data[RGB_MATRIX_LED_COUNT] = {0};
static uint8_t normalize_component(uint8_t component) {
uint16_t x = (uint16_t)component;
x *= rgb_matrix_config.hsv.v; // Multiply by current brightness
x /= 255; // Divide by maximum brightness
return (uint8_t)x;
}
static RGB normalize_index(uint8_t i) {
RGB raw = raw_rgb_data[i];
RGB rgb = {
.r = normalize_component(raw.r),
.g = normalize_component(raw.g),
.b = normalize_component(raw.b),
};
return rgb;
}
static bool raw_rgb(effect_params_t* params) {
RGB_MATRIX_USE_LIMITS(led_min, led_max);
for (uint8_t i = led_min; i < led_max; i++) {
RGB_MATRIX_TEST_LED_FLAGS();
RGB rgb = normalize_index(i);
rgb_matrix_set_color(i, rgb.r, rgb.g, rgb.b);
}
return led_max < RGB_MATRIX_LED_COUNT;
}
static uint8_t unlocked_keys[8][2] = {
{2, 7}, // U
{4, 6}, // N
{3, 9}, // L
{2, 9}, // O
{4, 3}, // C
{3, 8}, // K
{2, 3}, // E
{3, 3}, // D
};
static uint8_t unlocked_ticks = 0;
static uint8_t unlocked_i = 0;
static uint8_t unlocked_leds_count = 0;
static uint8_t unlocked_leds[2] = {0, 0};
static bool unlocked(effect_params_t* params) {
RGB_MATRIX_USE_LIMITS(led_min, led_max);
unlocked_ticks++;
if (params->init) {
unlocked_ticks = 0;
unlocked_i = 0;
}
if (unlocked_ticks == 0) {
if (unlocked_i == 8) {
unlocked_leds_count = 0;
unlocked_i = 0;
} else {
unlocked_leds_count = rgb_matrix_map_row_column_to_led(unlocked_keys[unlocked_i][0], unlocked_keys[unlocked_i][1], unlocked_leds);
unlocked_i++;
}
}
for (uint8_t i = led_min; i < led_max; i++) {
RGB_MATRIX_TEST_LED_FLAGS();
HSV hsv = {
.h = i + unlocked_ticks,
.s = 0xFF,
.v = 0x70,
};
for (uint8_t j = 0; j < unlocked_leds_count; j++) {
if (i == unlocked_leds[j]) {
hsv.s = 0;
hsv.v = 0xFF;
}
}
RGB rgb = hsv_to_rgb(hsv);
rgb_matrix_set_color(i, rgb.r, rgb.g, rgb.b);
}
return led_max < RGB_MATRIX_LED_COUNT;
}
#endif // RGB_MATRIX_CUSTOM_EFFECT_IMPLS