#include "drashna.h" #include "rgb_stuff.h" #include "eeprom.h" extern rgblight_config_t rgblight_config; extern userspace_config_t userspace_config; #ifdef RGBLIGHT_ENABLE void rgblight_sethsv_default_helper(uint8_t index) { rgblight_sethsv_at(rgblight_config.hue, rgblight_config.sat, rgblight_config.val, index); } #endif // RGBLIGHT_ENABLE #ifdef INDICATOR_LIGHTS uint8_t last_mod; uint8_t last_led; uint8_t last_osm; uint8_t current_mod; uint8_t current_led; uint8_t current_osm; void set_rgb_indicators(uint8_t this_mod, uint8_t this_led, uint8_t this_osm) { if (userspace_config.rgb_layer_change && biton32(layer_state) == 0) { if (this_mod & MODS_SHIFT_MASK || this_led & (1<<USB_LED_CAPS_LOCK) || this_osm & MODS_SHIFT_MASK) { #ifdef SHFT_LED1 rgblight_sethsv_at(120, 255, 255, SHFT_LED1); #endif // SHFT_LED1 #ifdef SHFT_LED2 rgblight_sethsv_at(120, 255, 255, SHFT_LED2); #endif // SHFT_LED2 } else { #ifdef SHFT_LED1 rgblight_sethsv_default_helper(SHFT_LED1); #endif // SHFT_LED1 #ifdef SHFT_LED2 rgblight_sethsv_default_helper(SHFT_LED2); #endif // SHFT_LED2 } if (this_mod & MODS_CTRL_MASK || this_osm & MODS_CTRL_MASK) { #ifdef CTRL_LED1 rgblight_sethsv_at(0, 255, 255, CTRL_LED1); #endif // CTRL_LED1 #ifdef CTRL_LED2 rgblight_sethsv_at(0, 255, 255, CTRL_LED2); #endif // CTRL_LED2 } else { #ifdef CTRL_LED1 rgblight_sethsv_default_helper(CTRL_LED1); #endif // CTRL_LED1 #ifdef CTRL_LED2 rgblight_sethsv_default_helper(CTRL_LED2); #endif // CTRL_LED2 } if (this_mod & MODS_GUI_MASK || this_osm & MODS_GUI_MASK) { #ifdef GUI_LED1 rgblight_sethsv_at(51, 255, 255, GUI_LED1); #endif // GUI_LED1 #ifdef GUI_LED2 rgblight_sethsv_at(51, 255, 255, GUI_LED2); #endif // GUI_LED2 } else { #ifdef GUI_LED1 rgblight_sethsv_default_helper(GUI_LED1); #endif // GUI_LED1 #ifdef GUI_LED2 rgblight_sethsv_default_helper(GUI_LED2); #endif // GUI_LED2 } if (this_mod & MODS_ALT_MASK || this_osm & MODS_ALT_MASK) { #ifdef ALT_LED1 rgblight_sethsv_at(240, 255, 255, ALT_LED1); #endif // ALT_LED1 #ifdef GUI_LED2 rgblight_sethsv_at(240, 255, 255, ALT_LED2); #endif // GUI_LED2 } else { #ifdef GUI_LED1 rgblight_sethsv_default_helper(ALT_LED1); #endif // GUI_LED1 #ifdef GUI_LED2 rgblight_sethsv_default_helper(ALT_LED2); #endif // GUI_LED2 } } } void matrix_scan_indicator(void) { current_mod = get_mods(); current_led = host_keyboard_leds(); current_osm = get_oneshot_mods(); set_rgb_indicators(current_mod, current_led, current_osm); last_mod = current_mod; last_led = current_led; last_osm = current_osm; } #endif //INDICATOR_LIGHTS #ifdef RGBLIGHT_TWINKLE static rgblight_fadeout lights[RGBLED_NUM]; __attribute__ ((weak)) bool rgblight_twinkle_is_led_used_keymap(uint8_t index) { return false; } bool rgblight_twinkle_is_led_used(uint8_t index) { switch (index) { #ifdef INDICATOR_LIGHTS #ifdef SHFT_LED1 case SHFT_LED1: return true; #endif //SHFT_LED1 #ifdef SHFT_LED2 case SHFT_LED2: return true; #endif //SHFT_LED2 #ifdef CTRL_LED1 case CTRL_LED1: return true; #endif //CTRL_LED1 #ifdef CTRL_LED2 case CTRL_LED2: return true; #endif //CTRL_LED2 #ifdef GUI_LED1 case GUI_LED1: return true; #endif //GUI_LED1 #ifdef GUI_LED2 case GUI_LED2: return true; #endif //GUI_LED2 #ifdef ALT_LED1 case ALT_LED1: return true; #endif //ALT_LED1 #ifdef ALT_LED2 case ALT_LED2: return true; #endif //ALT_LED2 #endif //INDICATOR_LIGHTS default: return rgblight_twinkle_is_led_used_keymap(index); } } void scan_rgblight_fadeout(void) { // Don't effing change this function .... rgblight_sethsv is supppppper intensive bool litup = false; for (uint8_t light_index = 0 ; light_index < RGBLED_NUM ; ++light_index ) { if (lights[light_index].enabled && timer_elapsed(lights[light_index].timer) > 10) { rgblight_fadeout *light = &lights[light_index]; litup = true; if (light->life) { light->life -= 1; if (biton32(layer_state) == 0) { sethsv(light->hue + rand() % 0xF, 255, light->life, (LED_TYPE *)&led[light_index]); } light->timer = timer_read(); } else { if (light->enabled && biton32(layer_state) == 0) { rgblight_sethsv_default_helper(light_index); } litup = light->enabled = false; } } } if (litup && biton32(layer_state) == 0) { rgblight_set(); } } void start_rgb_light(void) { uint8_t indices[RGBLED_NUM]; uint8_t indices_count = 0; uint8_t min_life = 0xFF; uint8_t min_life_index = -1; for (uint8_t index = 0 ; index < RGBLED_NUM ; ++index ) { if (rgblight_twinkle_is_led_used(index)) { continue; } if (lights[index].enabled) { if (min_life_index == -1 || lights[index].life < min_life) { min_life = lights[index].life; min_life_index = index; } continue; } indices[indices_count] = index; ++indices_count; } uint8_t light_index; if (!indices_count) { light_index = min_life_index; } else { light_index = indices[rand() % indices_count]; } rgblight_fadeout *light = &lights[light_index]; light->enabled = true; light->timer = timer_read(); light->life = 0xC0 + rand() % 0x40; light->hue = rgblight_config.hue + (rand() % 0xB4) - 0x54; rgblight_sethsv_at(light->hue, 255, light->life, light_index); } #endif bool process_record_user_rgb(uint16_t keycode, keyrecord_t *record) { switch (keycode) { #ifdef RGBLIGHT_TWINKLE case KC_A ... KC_SLASH: case KC_F1 ... KC_F12: case KC_INSERT ... KC_UP: case KC_KP_SLASH ... KC_KP_DOT: case KC_F13 ... KC_F24: case KC_AUDIO_MUTE ... KC_MEDIA_REWIND: if (record->event.pressed) { start_rgb_light(); } return true; break; #endif // RGBLIGHT_TWINKLE case KC_RGB_T: // This allows me to use underglow as layer indication, or as normal #ifdef RGBLIGHT_ENABLE if (record->event.pressed) { userspace_config.rgb_layer_change ^= 1; xprintf("rgblight layer change [EEPROM]: %u\n", userspace_config.rgb_layer_change); eeprom_update_byte(EECONFIG_USERSPACE, userspace_config.raw); if (userspace_config.rgb_layer_change) { layer_state_set(layer_state); // This is needed to immediately set the layer color (looks better) } } #endif // RGBLIGHT_ENABLE return false; break; #ifdef RGBLIGHT_ENABLE case RGB_MODE_FORWARD ... RGB_MODE_GRADIENT: // quantum_keycodes.h L400 for definitions if (record->event.pressed) { //This disables layer indication, as it's assumed that if you're changing this ... you want that disabled if (userspace_config.rgb_layer_change) { userspace_config.rgb_layer_change = false; xprintf("rgblight layer change [EEPROM]: %u\n", userspace_config.rgb_layer_change); eeprom_update_byte(EECONFIG_USERSPACE, userspace_config.raw); } } return true; break; #endif // RGBLIGHT_ENABLE } return true; } void matrix_init_rgb(void) { #ifdef INDICATOR_LIGHTS current_mod = last_mod = get_mods(); current_led = last_led = host_keyboard_leds(); current_osm = last_osm = get_oneshot_mods(); #endif if (userspace_config.rgb_layer_change) { rgblight_enable_noeeprom(); switch (biton32(eeconfig_read_default_layer())) { case _COLEMAK: rgblight_sethsv_noeeprom_magenta(); break; case _DVORAK: rgblight_sethsv_noeeprom_springgreen(); break; case _WORKMAN: rgblight_sethsv_noeeprom_goldenrod(); break; default: rgblight_sethsv_noeeprom_cyan(); break; } } } void matrix_scan_rgb(void) { #ifdef RGBLIGHT_TWINKLE scan_rgblight_fadeout(); #endif // RGBLIGHT_ENABLE #ifdef INDICATOR_LIGHTS matrix_scan_indicator(); #endif } uint32_t layer_state_set_rgb(uint32_t state) { #ifdef RGBLIGHT_ENABLE if (userspace_config.rgb_layer_change) { switch (biton32(state)) { case _MACROS: rgblight_sethsv_noeeprom_orange(); userspace_config.is_overwatch ? rgblight_mode_noeeprom(17) : rgblight_mode_noeeprom(18); break; case _MEDIA: rgblight_sethsv_noeeprom_chartreuse(); rgblight_mode_noeeprom(22); break; case _GAMEPAD: rgblight_sethsv_noeeprom_orange(); rgblight_mode_noeeprom(17); break; case _DIABLO: rgblight_sethsv_noeeprom_red(); rgblight_mode_noeeprom(5); break; case _RAISE: rgblight_sethsv_noeeprom_yellow(); rgblight_mode_noeeprom(5); break; case _LOWER: rgblight_sethsv_noeeprom_green(); rgblight_mode_noeeprom(5); break; case _ADJUST: rgblight_sethsv_noeeprom_red(); rgblight_mode_noeeprom(23); break; default: // for any other layers, or the default layer switch (biton32(default_layer_state)) { case _COLEMAK: rgblight_sethsv_noeeprom_magenta(); break; case _DVORAK: rgblight_sethsv_noeeprom_springgreen(); break; case _WORKMAN: rgblight_sethsv_noeeprom_goldenrod(); break; default: rgblight_sethsv_noeeprom_cyan(); break; } biton32(state) == _MODS ? rgblight_mode_noeeprom(2) : rgblight_mode_noeeprom(1); // if _MODS layer is on, then breath to denote it break; } // layer_state_set_indicator(); // Runs every scan, so need to call this here .... since I can't get it working "right" anyhow } #endif // RGBLIGHT_ENABLE return state; }