You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

192 lines
5.8 KiB

/* Copyright 2023 ArthurCyy <https://github.com/ArthurCyy>
*
* 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 "blackio83.h"
#include "usb_main.h"
#include "usb_util.h"
#define LOOP_10HZ_PERIOD 100
deferred_token loop10hz_token = INVALID_DEFERRED_TOKEN;
uint32_t loop_10Hz(uint32_t trigger_time, void *cb_arg);
static const SerialConfig mwproto_uart_config = {
.speed = MWPROTO_BITRATE,
};
static void POWER_EnterSleep(void) {
/* Clear Wake-up flag */
PWR->CR |= PWR_CR_CWUF | PWR_CR_CSBF;
/* Select Sleep mode */
/* PWR->CR |= PWR_CR_PDDS | PWR_CR_LPDS; */
/* Set SLEEPDEEP bit of Cortex System Control Register */
SCB->SCR |= SCB_SCR_SLEEPDEEP_Msk;
/* Request Wait For Interrupt */
__WFI();
/* Reset SLEEPDEEP bit of Cortex System Control Register */
SCB->SCR &= ~SCB_SCR_SLEEPDEEP_Msk;
NVIC_SystemReset();
}
extern void ws2812_poweron(void);
extern void ws2812_poweroff(void);
static bool p_setup = false;
static bool s_init = false;
void ws2812_poweron(void) {
if(p_setup) return;
p_setup = true;
s_init = false;
gpio_set_pin_output(RGB_EN_PIN);
gpio_write_pin_high(RGB_EN_PIN);
}
void ws2812_poweroff(void) {
if(!p_setup) return;
p_setup = false;
setPinInputLow(WS2812_DI_PIN);
gpio_write_pin_low(RGB_EN_PIN);
}
void keyboard_pre_init_kb() {
keyboard_pre_init_user();
setPinInputLow(MWPROTO_STATUS_PIN);
gpio_set_pin_output(MWPROTO_WAKEUP_PIN);
gpio_write_pin_low(MWPROTO_WAKEUP_PIN);
wait_ms(2);
gpio_write_pin_high(MWPROTO_WAKEUP_PIN);
palSetLineMode(MWPROTO_TX_PIN, PAL_MODE_ALTERNATE(MWPROTO_TX_PAL_MODE) | PAL_OUTPUT_TYPE_OPENDRAIN);
sdStart(&MWPROTO_DRIVER, &mwproto_uart_config);
}
void keyboard_post_init_kb(void) {
keyboard_post_init_user();
print(/* clang-format off */
"\n<--QMK Keyboard-->\n"
"Vendor: " STR(MANUFACTURER) "(" STR(VENDOR_ID) ")\n"
"Product: " STR(PRODUCT) " (" STR(PRODUCT_ID) ")\n"
"Version: " STR(DEVICE_VER) "\n"
"BUILD: " __DATE__ "\n"
); /* clang-format on */
gpio_write_pin_low(MWPROTO_WAKEUP_PIN);
wait_ms(50);
sdPutI(&MWPROTO_DRIVER, 0xA5);
sdPutI(&MWPROTO_DRIVER, 0x12);
sdPutI(&MWPROTO_DRIVER, 0x01);
sdPutI(&MWPROTO_DRIVER, 0x02);
sdPutI(&MWPROTO_DRIVER, 0xB4);
gpio_write_pin_high(MWPROTO_WAKEUP_PIN);
ws2812_poweron();
loop10hz_token = defer_exec(LOOP_10HZ_PERIOD, loop_10Hz, NULL);
}
bool shutdown_kb(bool jump_to_bootloader) {
if (shutdown_user(jump_to_bootloader)) {
#ifdef RGB_MATRIX_ENABLE
rgb_matrix_set_suspend_state(true);
#endif // RGB_MATRIX_ENABLE
wait_ms(10);
}
ws2812_poweroff();
return true;
}
#ifdef DIP_SWITCH_ENABLE
bool dip_switch_update_mask_kb(uint32_t state) {
if (!dip_switch_update_mask_user(state)) { return false; }
if(state & 0x01) {
led_suspend();
usbDisconnectBus(&USB_DRIVER);
usbStop(&USB_DRIVER);
shutdown_user(true);
gpio_set_pin_input_high(POWER_SWITCH_PIN);
palEnableLineEvent(POWER_SWITCH_PIN, PAL_EVENT_MODE_RISING_EDGE);
POWER_EnterSleep();
}
return true;
}
#endif
uint32_t loop_10Hz(uint32_t trigger_time, void *cb_arg) {
if(last_input_activity_elapsed() > 1000) {
static uint32_t pmu_timer = 0;
if(timer_elapsed32(pmu_timer) > 3000) {
pmu_timer = timer_read32();
gpio_write_pin_low(MWPROTO_WAKEUP_PIN);
if(gpio_read_pin(MWPROTO_STATUS_PIN))
wait_us(500);
else
wait_us(1500);
sdPutI(&MWPROTO_DRIVER, 0xA5);
sdPutI(&MWPROTO_DRIVER, 0x28);
sdPutI(&MWPROTO_DRIVER, 0x00);
sdPutI(&MWPROTO_DRIVER, 0x8D);
gpio_write_pin_high(MWPROTO_WAKEUP_PIN);
}
}
extern matrix_row_t matrix[MATRIX_ROWS];
static uint32_t restore_tick = 0;
if(matrix[0] == 0x4000 && matrix[1] == 0 &&
matrix[2] == 0 && matrix[3] == 0 && matrix[4] == 0 && matrix[5] == 0x201) {
if(restore_tick++ > 50) {
restore_tick = 0;
gpio_write_pin_low(MWPROTO_WAKEUP_PIN);
if(gpio_read_pin(MWPROTO_STATUS_PIN))
wait_us(500);
else
wait_us(1500);
sdPutI(&MWPROTO_DRIVER, 0xA5);
sdPutI(&MWPROTO_DRIVER, 0x1F);
sdPutI(&MWPROTO_DRIVER, 0x01);
sdPutI(&MWPROTO_DRIVER, 0x0F);
sdPutI(&MWPROTO_DRIVER, 0xB4);
gpio_write_pin_high(MWPROTO_WAKEUP_PIN);
wait_ms(50);
eeconfig_init();
#ifdef RGB_MATRIX_ENABLE
extern void rgb_matrix_update_pwm_buffers(void);
for(int i = 0; i < 5; i++) {
rgb_matrix_set_color_all(RGB_WHITE);
rgb_matrix_update_pwm_buffers();
wait_ms(500);
rgb_matrix_set_color_all(RGB_OFF);
rgb_matrix_update_pwm_buffers();
wait_ms(500);
}
#endif
wait_ms(500);
soft_reset_keyboard();
}
} else {
restore_tick = 0;
}
static uint32_t debug_tick = 0;
if (debug_tick++ > 9 ) {
dprintf("trigger %d\n", trigger_time);
debug_tick = 0;
}
return LOOP_10HZ_PERIOD;
}