/* * Copyright 2020 Michael Schwingen * 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 <util/delay.h> #include "modelm.h" #include "uart.h" #include "print.h" #include "sendchar.h" #include "ws2812.h" #include "modelm.h" #include "sleep_led.h" #ifdef UART_DEBUG # undef sendchar static int8_t capture_sendchar(uint8_t c) { // sendchar(c); uart_write(c); return 0; } #endif static uint16_t blink_cycle_timer; static bool blink_state = false; static uint8_t isRecording = 0; #ifdef KEYBOARD_mschwingen_modelm_led_ws2812 # if RGBLED_NUM < 3 # error we need at least 3 RGB LEDs! # endif static cRGB led[RGBLED_NUM] = {{255, 255, 255}, {255, 255, 255}, {255, 255, 255}}; # define BRIGHT 32 # define DIM 6 static const cRGB black = {.r = 0, .g = 0, .b = 0}; static const __attribute__((unused)) cRGB green = {.r = 0, .g = BRIGHT, .b = 0}; static const __attribute__((unused)) cRGB lgreen = {.r = 0, .g = DIM, .b = 0}; static const __attribute__((unused)) cRGB red = {.r = BRIGHT, .g = 0, .b = 0}; static const __attribute__((unused)) cRGB lred = {.r = DIM, .g = 0, .b = 0}; static const __attribute__((unused)) cRGB blue = {.r = 0, .g = 0, .b = BRIGHT}; static const __attribute__((unused)) cRGB lblue = {.r = 0, .g = 0, .b = DIM}; static const __attribute__((unused)) cRGB turq = {.r = 0, .g = BRIGHT, .b = BRIGHT}; static const __attribute__((unused)) cRGB lturq = {.r = 0, .g = DIM, .b = DIM}; static const __attribute__((unused)) cRGB white = {.r = BRIGHT, .g = BRIGHT, .b = BRIGHT}; static led_t led_state; static uint8_t layer; static uint8_t default_layer; #endif // we need our own sleep_led_* implementation to get callbacks on USB // suspend/resume in order to completely turn off WS2812 LEDs static bool suspend_active = false; void sleep_led_init(void) {} void sleep_led_toggle(void) {} void sleep_led_disable(void) { suspend_active = false; writePinHigh(QMK_LED); } void sleep_led_enable(void) { suspend_active = true; writePinLow(QMK_LED); #ifdef KEYBOARD_mschwingen_modelm_led_ws2812 led[0] = black; led[1] = black; led[2] = black; ws2812_setleds(led, RGBLED_NUM); #endif } void keyboard_pre_init_kb(void) { #ifdef KEYBOARD_mschwingen_modelm_led_ws2812 ws2812_setleds(led, RGBLED_NUM); #else /* Set status LEDs pins to output and Low (on) */ setPinOutput(MODELM_LED_CAPSLOCK); setPinOutput(MODELM_LED_SCROLLOCK); setPinOutput(MODELM_LED_NUMLOCK); writePinLow(MODELM_LED_CAPSLOCK); writePinLow(MODELM_LED_SCROLLOCK); writePinLow(MODELM_LED_NUMLOCK); #endif setPinOutput(QMK_LED); writePinHigh(QMK_LED); _delay_ms(50); #ifdef UART_DEBUG uart_init(115200); print_set_sendchar(capture_sendchar); uprintf("\r\nHello world!\r\n"); #endif setPinOutput(SR_LOAD_PIN); setPinOutput(SR_CLK_PIN); setPinOutput(SR_DOUT_PIN); // MOSI - unused writePinLow(SR_CLK_PIN); } #ifdef KEYBOARD_mschwingen_modelm_led_ws2812 static void led_update_rgb(void) { if (isRecording && blink_state) { led[0] = white; } else { switch (default_layer) { case 0: led[0] = led_state.num_lock ? blue : lblue; break; case 1: led[0] = led_state.num_lock ? green : black; break; } } led[1] = led_state.caps_lock ? green : black; switch (layer) { case 0: case 1: default: led[2] = led_state.scroll_lock ? green : black; break; case 2: led[2] = led_state.scroll_lock ? red : lred; break; case 3: led[2] = led_state.scroll_lock ? turq : lturq; break; } if (!suspend_active) { ws2812_setleds(led, RGBLED_NUM); } } bool led_update_kb(led_t state) { dprintf("LED Update: %d %d %d", led_state.num_lock, led_state.caps_lock, led_state.scroll_lock); led_state = state; led_update_rgb(); return true; } void update_layer_leds(void) { static uint8_t old_layer = 255; static uint8_t old_default_layer = 255; layer = get_highest_layer(layer_state); default_layer = get_highest_layer(default_layer_state); if (isRecording && timer_elapsed(blink_cycle_timer) > 150) { blink_state = !blink_state; blink_cycle_timer = timer_read(); old_layer = 255; // fallthrough next check } if (layer == old_layer && default_layer == old_default_layer) { return; } old_layer = layer; old_default_layer = default_layer; dprintf("Layer change: %d %d", default_layer, layer); led_update_rgb(); } /*****************************************************************************/ #else // classic LEDs on GPIO bool led_update_kb(led_t led_state) { dprintf("LED Update: %d %d %d", led_state.num_lock, led_state.caps_lock, led_state.scroll_lock); if (led_update_user(led_state)) { if (!isRecording) writePin(MODELM_LED_NUMLOCK, !led_state.num_lock); writePin(MODELM_LED_CAPSLOCK, !led_state.caps_lock); writePin(MODELM_LED_SCROLLOCK, !led_state.scroll_lock); } return true; } void update_layer_leds(void) { if (isRecording && timer_elapsed(blink_cycle_timer) > 150) { blink_state = !blink_state; blink_cycle_timer = timer_read(); writePin(MODELM_LED_NUMLOCK, blink_state); } } #endif void dynamic_macro_record_start_user(void) { isRecording++; blink_cycle_timer = timer_read(); } void dynamic_macro_record_end_user(int8_t direction) { if (isRecording) isRecording--; }