123 lines
		
	
	
	
		
			4.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			123 lines
		
	
	
	
		
			4.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // Copyright 2020 Christopher Courtney, aka Drashna Jael're  (@drashna) <drashna@live.com>
 | |
| // SPDX-License-Identifier: GPL-2.0-or-later
 | |
| 
 | |
| #include "drashna.h"
 | |
| #include "rgb_stuff.h"
 | |
| #include "eeprom.h"
 | |
| 
 | |
| bool has_initialized;
 | |
| 
 | |
| void rgblight_sethsv_default_helper(uint8_t index) {
 | |
|     rgblight_sethsv_at(rgblight_get_hue(), rgblight_get_sat(), rgblight_get_val(), index);
 | |
| }
 | |
| void rgblight_set_hsv_and_mode(uint8_t hue, uint8_t sat, uint8_t val, uint8_t mode) {
 | |
|     rgblight_sethsv_noeeprom(hue, sat, val);
 | |
|     // wait_us(175);  // Add a slight delay between color and mode to ensure it's processed correctly
 | |
|     rgblight_mode_noeeprom(mode);
 | |
| }
 | |
| 
 | |
| bool process_record_user_rgb_light(uint16_t keycode, keyrecord_t *record) {
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| #if defined(RGBLIGHT_STARTUP_ANIMATION)
 | |
| static bool    is_enabled;
 | |
| static bool    is_rgblight_startup;
 | |
| static HSV     old_hsv;
 | |
| static uint8_t old_mode;
 | |
| deferred_token rgb_startup_token;
 | |
| 
 | |
| uint32_t rgb_startup_animation(uint32_t triger_time, void *cb_arg) {
 | |
|     if (is_rgblight_startup && is_keyboard_master()) {
 | |
|         static uint8_t counter = 0;
 | |
|         counter++;
 | |
|         rgblight_sethsv_noeeprom((counter + old_hsv.h) % 255, 255, 255);
 | |
|         if (counter >= 255) {
 | |
|             is_rgblight_startup = false;
 | |
|             if (userspace_config.rgb_layer_change) {
 | |
|                 layer_state_set_rgb_light(layer_state);
 | |
|             } else {
 | |
|                 rgblight_set_hsv_and_mode(old_hsv.h, old_hsv.s, old_hsv.v, old_mode);
 | |
|             }
 | |
|             if (!is_enabled) {
 | |
|                 rgblight_disable_noeeprom();
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     return is_rgblight_startup ? 10 : 0;
 | |
| }
 | |
| #endif
 | |
| 
 | |
| void keyboard_post_init_rgb_light(void) {
 | |
| #if defined(RGBLIGHT_STARTUP_ANIMATION)
 | |
|     is_enabled = rgblight_is_enabled();
 | |
|     if (userspace_config.rgb_layer_change) {
 | |
|         layer_state_set_rgb_light(layer_state);
 | |
|     }
 | |
|     old_hsv  = rgblight_get_hsv();
 | |
|     old_mode = rgblight_get_mode();
 | |
|     rgblight_mode_noeeprom(RGBLIGHT_MODE_STATIC_LIGHT);
 | |
|     is_rgblight_startup = true;
 | |
|     rgb_startup_token   = defer_exec(300, rgb_startup_animation, NULL);
 | |
| #endif
 | |
|     if (userspace_config.rgb_layer_change) {
 | |
|         layer_state_set_rgb_light(layer_state);
 | |
|     }
 | |
| }
 | |
| 
 | |
| layer_state_t layer_state_set_rgb_light(layer_state_t state) {
 | |
| #ifdef RGBLIGHT_ENABLE
 | |
|     if (userspace_config.rgb_layer_change) {
 | |
|         switch (get_highest_layer(state & ~((layer_state_t)1 << _MOUSE))) {
 | |
|             case _MEDIA:
 | |
|                 rgblight_set_hsv_and_mode(HSV_CHARTREUSE, RGBLIGHT_MODE_KNIGHT + 1);
 | |
|                 break;
 | |
|             case _GAMEPAD:
 | |
|                 rgblight_set_hsv_and_mode(HSV_ORANGE, RGBLIGHT_MODE_SNAKE + 2);
 | |
|                 break;
 | |
|             case _DIABLO:
 | |
|             case _DIABLOII:
 | |
|                 rgblight_set_hsv_and_mode(HSV_RED, RGBLIGHT_MODE_BREATHING + 3);
 | |
|                 break;
 | |
|             case _RAISE:
 | |
|                 rgblight_set_hsv_and_mode(HSV_YELLOW, RGBLIGHT_MODE_BREATHING + 3);
 | |
|                 break;
 | |
|             case _LOWER:
 | |
|                 rgblight_set_hsv_and_mode(HSV_GREEN, RGBLIGHT_MODE_BREATHING + 3);
 | |
|                 break;
 | |
|             case _ADJUST:
 | |
|                 rgblight_set_hsv_and_mode(HSV_RED, RGBLIGHT_MODE_KNIGHT + 2);
 | |
|                 break;
 | |
|             default:
 | |
|                 if (layer_state_cmp(state, _MOUSE)) {
 | |
| #    if defined(RGBLIGHT_EFFECT_TWINKLE)
 | |
|                     rgblight_set_hsv_and_mode(HSV_CHARTREUSE, RGBLIGHT_MODE_TWINKLE + 5);
 | |
| #    else
 | |
|                     rgblight_set_hsv_and_mode(HSV_CHARTREUSE, RGBLIGHT_MODE_BREATHING + 3);
 | |
| #    endif
 | |
|                 } else {
 | |
|                     default_layer_state_set_rgb_light(default_layer_state);
 | |
|                 }
 | |
|         }
 | |
| #endif // RGBLIGHT_ENABLE
 | |
|     }
 | |
|     return state;
 | |
| }
 | |
| 
 | |
| layer_state_t default_layer_state_set_rgb_light(layer_state_t state) {
 | |
|     switch (get_highest_layer(state)) {
 | |
|         case _DEFAULT_LAYER_1:
 | |
|             rgblight_set_hsv_and_mode(DEFAULT_LAYER_1_HSV, RGBLIGHT_MODE_STATIC_LIGHT);
 | |
|             break;
 | |
|         case _DEFAULT_LAYER_2:
 | |
|             rgblight_set_hsv_and_mode(DEFAULT_LAYER_2_HSV, RGBLIGHT_MODE_STATIC_LIGHT);
 | |
|             break;
 | |
|         case _DEFAULT_LAYER_3:
 | |
|             rgblight_set_hsv_and_mode(DEFAULT_LAYER_3_HSV, RGBLIGHT_MODE_STATIC_LIGHT);
 | |
|             break;
 | |
|         case _DEFAULT_LAYER_4:
 | |
|             rgblight_set_hsv_and_mode(DEFAULT_LAYER_4_HSV, RGBLIGHT_MODE_STATIC_LIGHT);
 | |
|             break;
 | |
|     }
 | |
|     return state;
 | |
| }
 | 
