158 lines
4.5 KiB
C++
158 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"
|
|
#include "action_layer.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
|