/* Copyright 2023 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 . */ #include "rev_0100.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; gpio_set_pin_input_low(WS2812_DI_PIN); gpio_write_pin_low(RGB_EN_PIN); } void keyboard_pre_init_kb() { keyboard_pre_init_user(); gpio_set_pin_input_low(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 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; }