chore: move led shutdown logic in rgb_matric.c

zsa · Aug 17, 2019 · 8f065c4 · 8f065c4
1 parent 76ba4c4
commit 8f065c4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/keyboards/ergodox_ez/ergodox_ez.c b/keyboards/ergodox_ez/ergodox_ez.c
@@ -222,9 +222,6 @@ const keypos_t hand_swap_config[MATRIX_ROWS][MATRIX_COLS] = {
 #ifdef RGB_MATRIX_ENABLE
 
 void suspend_power_down_kb(void) {
-#if RGB_DISABLE_WHEN_USB_SUSPENDED == true
-    rgb_matrix_set_color_all(0, 0, 0);
-#endif
     rgb_matrix_set_suspend_state(true);
     suspend_power_down_user();
 }

diff --git a/keyboards/planck/ez/ez.c b/keyboards/planck/ez/ez.c
@@ -22,9 +22,6 @@ keyboard_config_t keyboard_config;
 
 #ifdef RGB_MATRIX_ENABLE
 void suspend_power_down_kb(void) {
-#if RGB_DISABLE_WHEN_USB_SUSPENDED == true
-    rgb_matrix_set_color_all(0, 0, 0);
-#endif
     rgb_matrix_set_suspend_state(true);
     suspend_power_down_user();
 }

diff --git a/quantum/rgb_matrix.c b/quantum/rgb_matrix.c
@@ -436,6 +436,9 @@ void rgb_matrix_init(void) {
 }
 
 void rgb_matrix_set_suspend_state(bool state) {
+  if (state && RGB_DISABLE_WHEN_USB_SUSPENDED) {
+    rgb_matrix_set_color_all(0, 0, 0);
+  }
   g_suspend_state = state;
 }