305 lines
11 KiB
C
305 lines
11 KiB
C
/* Copyright 2021 HorrorTroll <https://github.com/HorrorTroll>
|
|
*
|
|
* 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 "rgb_matrix.h"
|
|
#include "progmem.h"
|
|
#include "config.h"
|
|
#include "eeprom.h"
|
|
#include <string.h>
|
|
#include <math.h>
|
|
|
|
#include <lib/lib8tion/lib8tion.h>
|
|
|
|
#include "oled/oled_stuff.h"
|
|
|
|
// Each layer gets a name for readability, which is then used in the keymap matrix below.
|
|
// The underscores don't mean anything - you can have a layer called STUFF or any other name.
|
|
// Layer names don't all need to be of the same length, obviously, and you can also skip them
|
|
// entirely and just use numbers.
|
|
|
|
enum layer_names {
|
|
_BASE = 0,
|
|
_WAVE = 1,
|
|
_FN = 2
|
|
};
|
|
|
|
// For CUSTOM_GRADIENT
|
|
HSV gradient_0 = {205, 250, 255};
|
|
HSV gradient_100 = {140, 215, 125};
|
|
bool reflected_gradient = false;
|
|
uint8_t gp_i = 0;
|
|
|
|
typedef struct {
|
|
HSV gradient_0;
|
|
HSV gradient_1;
|
|
bool reflected;
|
|
} CUSTOM_PRESETS;
|
|
|
|
enum user_rgb_mode {
|
|
RGB_MODE_ALL,
|
|
RGB_MODE_NONE,
|
|
};
|
|
|
|
typedef union {
|
|
uint32_t raw;
|
|
struct {
|
|
uint8_t rgb_mode :8;
|
|
};
|
|
} user_config_t;
|
|
|
|
user_config_t user_config;
|
|
|
|
enum layer_keycodes {
|
|
//Custom Gradient control keycode
|
|
G1_HUI = SAFE_RANGE, //Custom gradient color 1 hue increase
|
|
G1_HUD, //Custom gradient color 1 hue decrease
|
|
G1_SAI, //Custom gradient color 1 saturation increase
|
|
G1_SAD, //Custom gradient color 1 saturation decrease
|
|
G1_VAI, //Custom gradient color 1 value increase
|
|
G1_VAD, //Custom gradient color 1 value decrease
|
|
G2_HUI, //Custom gradient color 2 hue increase
|
|
G2_HUD, //Custom gradient color 2 hue decrease
|
|
G2_SAI, //Custom gradient color 2 saturation increase
|
|
G2_SAD, //Custom gradient color 2 saturation decrease
|
|
G2_VAI, //Custom gradient color 2 value increase
|
|
G2_VAD, //Custom gradient color 2 value decrease
|
|
G_PRE, //Gradient presets
|
|
REF_G, //Toggle between linear and reflected gradient
|
|
G_FLIP, //Flip the gradient colors
|
|
|
|
//Custom led effect keycode
|
|
RGB_C_E, //Cycle user effect
|
|
};
|
|
|
|
void keyboard_post_init_kb(void) {
|
|
user_config.raw = eeconfig_read_user();
|
|
switch (user_config.rgb_mode) {
|
|
case RGB_MODE_ALL:
|
|
rgb_matrix_set_flags(LED_FLAG_ALL);
|
|
rgb_matrix_enable_noeeprom();
|
|
break;
|
|
case RGB_MODE_NONE:
|
|
rgb_matrix_set_flags(LED_FLAG_NONE);
|
|
rgb_matrix_set_color_all(0, 0, 0);
|
|
break;
|
|
}
|
|
}
|
|
|
|
bool process_record_user(uint16_t keycode, keyrecord_t *record) {
|
|
process_record_user_oled(keycode, record);
|
|
|
|
uint8_t color_adj_step = 5;
|
|
|
|
CUSTOM_PRESETS gradient_presets[] = {
|
|
{{41 , 255, 255}, {233, 245, 255}, false },
|
|
{{45 , 245, 155}, {160, 255, 80}, false },
|
|
{{173, 245, 40}, {41 , 255, 205}, true },
|
|
{{32 , 255, 165}, {217, 185, 70}, false },
|
|
{{240, 255, 145}, {115, 255, 245}, true },
|
|
{{118, 255, 255}, {242, 255, 255}, false },
|
|
{{212, 0 , 0}, {223, 235, 165}, true },
|
|
{{205, 250, 255}, {140, 215, 125}, false },
|
|
};
|
|
|
|
uint8_t gp_length = sizeof(gradient_presets)/sizeof(gradient_presets[0]);
|
|
|
|
switch (keycode) {
|
|
case G1_HUI:
|
|
if (record->event.pressed) {
|
|
gradient_0.h += color_adj_step;
|
|
dprintf("Gradient 0 HSV: %d, %d, %d\n", gradient_0.h, gradient_0.s, gradient_0.v);
|
|
}
|
|
return false;
|
|
case G1_HUD:
|
|
if (record->event.pressed) {
|
|
gradient_0.h -= color_adj_step;
|
|
dprintf("Gradient 0 HSV: %d, %d, %d\n", gradient_0.h, gradient_0.s, gradient_0.v);
|
|
}
|
|
return false;
|
|
case G1_SAI:
|
|
if (record->event.pressed) {
|
|
gradient_0.s = (gradient_0.s + color_adj_step * 2 <= 255) ? gradient_0.s + color_adj_step * 2 : 255;
|
|
dprintf("Gradient 0 HSV: %d, %d, %d\n", gradient_0.h, gradient_0.s, gradient_0.v);
|
|
}
|
|
return false;
|
|
case G1_SAD:
|
|
if (record->event.pressed) {
|
|
gradient_0.s = (gradient_0.s - color_adj_step * 2 >= 0) ? gradient_0.s - color_adj_step * 2 : 0;
|
|
dprintf("Gradient 0 HSV: %d, %d, %d\n", gradient_0.h, gradient_0.s, gradient_0.v);
|
|
}
|
|
return false;
|
|
case G1_VAI:
|
|
if (record->event.pressed) {
|
|
gradient_0.v = (gradient_0.v + color_adj_step * 2 <= 255) ? gradient_0.v + color_adj_step * 2 : 255;
|
|
dprintf("Gradient 0 HSV: %d, %d, %d\n", gradient_0.h, gradient_0.s, gradient_0.v);
|
|
}
|
|
return false;
|
|
case G1_VAD:
|
|
if (record->event.pressed) {
|
|
gradient_0.v = (gradient_0.v - color_adj_step * 2 >= 0) ? gradient_0.v - color_adj_step * 2 : 0;
|
|
dprintf("Gradient 0 HSV: %d, %d, %d\n", gradient_0.h, gradient_0.s, gradient_0.v);
|
|
}
|
|
return false;
|
|
case G2_HUI:
|
|
if (record->event.pressed) {
|
|
gradient_100.h += color_adj_step;
|
|
dprintf("Gradient 100 HSV: %d, %d, %d\n", gradient_100.h, gradient_100.s, gradient_100.v);
|
|
}
|
|
return false;
|
|
case G2_HUD:
|
|
if (record->event.pressed) {
|
|
gradient_100.h -= color_adj_step;
|
|
dprintf("Gradient 100 HSV: %d, %d, %d\n", gradient_100.h, gradient_100.s, gradient_100.v);
|
|
}
|
|
return false;
|
|
case G2_SAI:
|
|
if (record->event.pressed) {
|
|
gradient_100.s = (gradient_100.s + color_adj_step * 2 <= 255) ? gradient_100.s + color_adj_step * 2 : 255;
|
|
dprintf("Gradient 100 HSV: %d, %d, %d\n", gradient_100.h, gradient_100.s, gradient_100.v);
|
|
}
|
|
return false;
|
|
case G2_SAD:
|
|
if (record->event.pressed) {
|
|
gradient_100.s = (gradient_100.s - color_adj_step * 2 >= 0) ? gradient_100.s - color_adj_step * 2 : 0;
|
|
dprintf("Gradient 100 HSV: %d, %d, %d\n", gradient_100.h, gradient_100.s, gradient_100.v);
|
|
}
|
|
return false;
|
|
case G2_VAI:
|
|
if (record->event.pressed) {
|
|
gradient_100.v = (gradient_100.v + color_adj_step * 2 <= 255) ? gradient_100.v + color_adj_step * 2 : 255;
|
|
dprintf("Gradient 100 HSV: %d, %d, %d\n", gradient_100.h, gradient_100.s, gradient_100.v);
|
|
}
|
|
return false;
|
|
case G2_VAD:
|
|
if (record->event.pressed) {
|
|
gradient_100.v = (gradient_100.v - color_adj_step * 2 >= 0) ? gradient_100.v - color_adj_step * 2 : 0;
|
|
dprintf("Gradient 100 HSV: %d, %d, %d\n", gradient_100.h, gradient_100.s, gradient_100.v);
|
|
}
|
|
return false;
|
|
case G_PRE:
|
|
if (record->event.pressed) {
|
|
gp_i = (gp_i + gp_length ) % gp_length;
|
|
|
|
gradient_0 = gradient_presets[gp_i].gradient_0;
|
|
gradient_100 = gradient_presets[gp_i].gradient_1;
|
|
reflected_gradient = gradient_presets[gp_i].reflected;
|
|
|
|
gp_i += 1;
|
|
}
|
|
return false;
|
|
case REF_G:
|
|
if (record->event.pressed) {
|
|
reflected_gradient = !reflected_gradient;
|
|
}
|
|
return false;
|
|
case G_FLIP:
|
|
if (record->event.pressed) {
|
|
HSV temp_color = gradient_0;
|
|
gradient_0 = gradient_100;
|
|
gradient_100 = temp_color;
|
|
}
|
|
return false;
|
|
case RGB_C_E:
|
|
if (record->event.pressed) {
|
|
switch (rgb_matrix_get_mode()) {
|
|
case RGB_MATRIX_CUSTOM_CUSTOM_GRADIENT:
|
|
rgb_matrix_mode(RGB_MATRIX_CUSTOM_DIAGONAL);
|
|
return false;
|
|
case RGB_MATRIX_CUSTOM_DIAGONAL:
|
|
rgb_matrix_mode(RGB_MATRIX_CUSTOM_COOL_DIAGONAL);
|
|
return false;
|
|
case RGB_MATRIX_CUSTOM_COOL_DIAGONAL:
|
|
rgb_matrix_mode(RGB_MATRIX_CUSTOM_KITT);
|
|
return false;
|
|
case RGB_MATRIX_CUSTOM_KITT:
|
|
rgb_matrix_mode(RGB_MATRIX_CUSTOM_RANDOM_BREATH_RAINBOW);
|
|
return false;
|
|
default:
|
|
rgb_matrix_mode(RGB_MATRIX_CUSTOM_CUSTOM_GRADIENT);
|
|
return false;
|
|
}
|
|
}
|
|
return false;
|
|
case RGB_TOG:
|
|
if (record->event.pressed) {
|
|
switch (rgb_matrix_get_flags()) {
|
|
case LED_FLAG_ALL: {
|
|
rgb_matrix_set_flags(LED_FLAG_NONE);
|
|
rgb_matrix_set_color_all(0, 0, 0);
|
|
user_config.rgb_mode = RGB_MODE_NONE;
|
|
}
|
|
break;
|
|
default: {
|
|
rgb_matrix_set_flags(LED_FLAG_ALL);
|
|
rgb_matrix_enable_noeeprom();
|
|
user_config.rgb_mode = RGB_MODE_ALL;
|
|
}
|
|
break;
|
|
}
|
|
eeconfig_update_user(user_config.raw);
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void rgb_matrix_indicators_user(void) {
|
|
uint8_t side_leds_left[3] = {17, 18, 19};
|
|
uint8_t side_leds_right[3] = { 4, 5, 6};
|
|
HSV hsv = rgb_matrix_config.hsv;
|
|
uint8_t time = scale16by8(g_rgb_timer, qadd8(32, 1));
|
|
hsv.h = time;
|
|
RGB rgb = hsv_to_rgb(hsv);
|
|
|
|
if ((rgb_matrix_get_flags() & LED_FLAG_ALL)) {
|
|
if (host_keyboard_led_state().caps_lock) {
|
|
for (uint8_t i = 0; i < 3; i++)
|
|
{
|
|
rgb_matrix_set_color(side_leds_left[i], rgb.r, rgb.g, rgb.b);
|
|
}
|
|
}
|
|
if (host_keyboard_led_state().scroll_lock) {
|
|
for (uint8_t i = 0; i < 3; i++)
|
|
{
|
|
rgb_matrix_set_color(side_leds_right[i], rgb.r, rgb.g, rgb.b);
|
|
}
|
|
}
|
|
} else {
|
|
if (host_keyboard_led_state().caps_lock) {
|
|
for (uint8_t i = 0; i < 3; i++)
|
|
{
|
|
rgb_matrix_set_color(side_leds_left[i], rgb.r, rgb.g, rgb.b);
|
|
}
|
|
} else {
|
|
for (uint8_t i = 0; i < 3; i++)
|
|
{
|
|
rgb_matrix_set_color(side_leds_left[i], 0, 0, 0);
|
|
}
|
|
}
|
|
if (host_keyboard_led_state().scroll_lock) {
|
|
for (uint8_t i = 0; i < 3; i++)
|
|
{
|
|
rgb_matrix_set_color(side_leds_right[i], rgb.r, rgb.g, rgb.b);
|
|
}
|
|
} else {
|
|
for (uint8_t i = 0; i < 3; i++)
|
|
{
|
|
rgb_matrix_set_color(side_leds_right[i], 0, 0, 0);
|
|
}
|
|
}
|
|
}
|
|
}
|