qmk-firmware/keyboards/mschwingen/modelm/modelm.c
Ryan 04b51e381e
Update UART driver API ()
* Add uart_puts() and uart_gets()

* Add some docs

* Rework API

* Formatting

* Update docs/uart_driver.md

Co-authored-by: Sergey Vlasov <sigprof@gmail.com>

* Simplify a uart_write() loop

* Update platforms/avr/drivers/uart.c

Co-authored-by: Joel Challis <git@zvecr.com>

Co-authored-by: Sergey Vlasov <sigprof@gmail.com>
Co-authored-by: Joel Challis <git@zvecr.com>
2021-11-13 18:23:14 +00:00

215 lines
6.1 KiB
C

/*
* 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 = biton32(layer_state);
default_layer = biton32(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--;
}