mirror of
https://github.com/Keychron/qmk_firmware.git
synced 2024-12-03 14:16:35 +06:00
278 lines
8.0 KiB
C
278 lines
8.0 KiB
C
/* Copyright 2022~2023 @ lokher (https://www.keychron.com)
|
|
*
|
|
* 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 "quantum.h"
|
|
#include "wireless.h"
|
|
#include "indicator.h"
|
|
#include "lpm.h"
|
|
#include "mousekey.h"
|
|
#if defined(PROTOCOL_CHIBIOS)
|
|
# include <usb_main.h>
|
|
#endif
|
|
#include "transport.h"
|
|
#include "lkbt51.h"
|
|
|
|
#ifndef REINIT_LED_DRIVER
|
|
# define REINIT_LED_DRIVER 0
|
|
#endif
|
|
|
|
#if defined(PROTOCOL_CHIBIOS)
|
|
extern host_driver_t chibios_driver;
|
|
#endif
|
|
extern host_driver_t wireless_driver;
|
|
extern keymap_config_t keymap_config;
|
|
extern wt_func_t wireless_transport;
|
|
|
|
static transport_t transport = TRANSPORT_NONE;
|
|
|
|
#ifdef NKRO_ENABLE
|
|
nkro_t nkro = {false, false};
|
|
#endif
|
|
|
|
static void transport_changed(transport_t new_transport);
|
|
|
|
__attribute__((weak)) void bt_transport_enable(bool enable) {
|
|
if (enable) {
|
|
// if (host_get_driver() != &wireless_driver) {
|
|
host_set_driver(&wireless_driver);
|
|
|
|
/* Disconnect and reconnect to sync the wireless state
|
|
* TODO: query wireless state to sync
|
|
*/
|
|
wireless_disconnect();
|
|
|
|
uint32_t t = timer_read32();
|
|
while (timer_elapsed32(t) < 50) {
|
|
wireless_transport.task();
|
|
}
|
|
// wireless_connect();
|
|
wireless_connect_ex(30, 0);
|
|
// TODO: Clear USB report
|
|
//}
|
|
} else {
|
|
indicator_stop();
|
|
|
|
if (wireless_get_state() == WT_CONNECTED && transport == TRANSPORT_BLUETOOTH) {
|
|
report_keyboard_t empty_report = {0};
|
|
wireless_driver.send_keyboard(&empty_report);
|
|
}
|
|
}
|
|
}
|
|
|
|
__attribute__((weak)) void p24g_transport_enable(bool enable) {
|
|
if (enable) {
|
|
// if (host_get_driver() != &wireless_driver) {
|
|
host_set_driver(&wireless_driver);
|
|
|
|
/* Disconnect and reconnect to sync the wireless state
|
|
* TODO: query bluetooth state to sync
|
|
*/
|
|
wireless_disconnect();
|
|
|
|
uint32_t t = timer_read32();
|
|
while (timer_elapsed32(t) < 50) {
|
|
wireless_transport.task();
|
|
}
|
|
wireless_connect_ex(P24G_INDEX, 0);
|
|
// wireless_connect();
|
|
// TODO: Clear USB report
|
|
//}
|
|
} else {
|
|
indicator_stop();
|
|
|
|
if (wireless_get_state() == WT_CONNECTED && transport == TRANSPORT_P2P4) {
|
|
report_keyboard_t empty_report = {0};
|
|
wireless_driver.send_keyboard(&empty_report);
|
|
}
|
|
}
|
|
}
|
|
|
|
__attribute__((weak)) void usb_power_connect(void) {}
|
|
__attribute__((weak)) void usb_power_disconnect(void) {}
|
|
|
|
__attribute__((weak)) void usb_transport_enable(bool enable) {
|
|
if (enable) {
|
|
if (host_get_driver() != &chibios_driver) {
|
|
#if !defined(KEEP_USB_CONNECTION_IN_WIRELESS_MODE)
|
|
usb_power_connect();
|
|
usb_start(&USBD1);
|
|
#endif
|
|
host_set_driver(&chibios_driver);
|
|
}
|
|
} else {
|
|
if (USB_DRIVER.state == USB_ACTIVE) {
|
|
report_keyboard_t empty_report = {0};
|
|
chibios_driver.send_keyboard(&empty_report);
|
|
}
|
|
|
|
#if !defined(KEEP_USB_CONNECTION_IN_WIRELESS_MODE)
|
|
usbStop(&USBD1);
|
|
usbDisconnectBus(&USBD1);
|
|
usb_power_disconnect();
|
|
#endif
|
|
}
|
|
}
|
|
|
|
void set_transport(transport_t new_transport) {
|
|
if (transport != new_transport) {
|
|
if (transport == TRANSPORT_USB || ((transport != TRANSPORT_USB) && wireless_get_state() == WT_CONNECTED)) clear_keyboard();
|
|
|
|
transport = new_transport;
|
|
|
|
switch (transport) {
|
|
case TRANSPORT_USB:
|
|
usb_transport_enable(true);
|
|
bt_transport_enable(false);
|
|
wait_ms(5);
|
|
p24g_transport_enable(false);
|
|
wireless_disconnect();
|
|
lpm_timer_stop();
|
|
#ifdef NKRO_ENABLE
|
|
# if defined(WIRELESS_NKRO_ENABLE)
|
|
nkro.bluetooth = keymap_config.nkro;
|
|
# endif
|
|
keymap_config.nkro = nkro.usb;
|
|
#endif
|
|
break;
|
|
|
|
case TRANSPORT_BLUETOOTH:
|
|
p24g_transport_enable(false);
|
|
wait_ms(1);
|
|
bt_transport_enable(true);
|
|
usb_transport_enable(false);
|
|
lpm_timer_reset();
|
|
#if defined(NKRO_ENABLE)
|
|
nkro.usb = keymap_config.nkro;
|
|
# if defined(WIRELESS_NKRO_ENABLE)
|
|
keymap_config.nkro = nkro.bluetooth;
|
|
# else
|
|
keymap_config.nkro = FALSE;
|
|
# endif
|
|
#endif
|
|
break;
|
|
|
|
case TRANSPORT_P2P4:
|
|
bt_transport_enable(false);
|
|
wait_ms(1);
|
|
p24g_transport_enable(true);
|
|
usb_transport_enable(false);
|
|
lpm_timer_reset();
|
|
#if defined(NKRO_ENABLE)
|
|
nkro.usb = keymap_config.nkro;
|
|
# if defined(WIRELESS_NKRO_ENABLE)
|
|
keymap_config.nkro = nkro.bluetooth;
|
|
# else
|
|
keymap_config.nkro = FALSE;
|
|
# endif
|
|
#endif
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
transport_changed(transport);
|
|
}
|
|
}
|
|
|
|
transport_t get_transport(void) {
|
|
return transport;
|
|
}
|
|
|
|
#if (REINIT_LED_DRIVER)
|
|
/* Changing transport may cause bronw-out reset of led driver
|
|
* withoug MCU reset, which lead backlight to not work,
|
|
* reinit the led driver workgound this issue */
|
|
static void reinit_led_drvier(void) {
|
|
/* Wait circuit to discharge for a while */
|
|
systime_t start = chVTGetSystemTime();
|
|
while (chTimeI2MS(chVTTimeElapsedSinceX(start)) < 100) {
|
|
};
|
|
|
|
# ifdef LED_MATRIX_ENABLE
|
|
led_matrix_init();
|
|
# endif
|
|
# ifdef RGB_MATRIX_ENABLE
|
|
rgb_matrix_init();
|
|
# endif
|
|
}
|
|
#endif
|
|
|
|
void transport_changed(transport_t new_transport) {
|
|
kc_printf("transport_changed %d\n\r", new_transport);
|
|
indicator_init();
|
|
|
|
#if (REINIT_LED_DRIVER)
|
|
reinit_led_drvier();
|
|
#endif
|
|
|
|
#if defined(RGB_MATRIX_ENABLE) && defined(RGB_MATRIX_TIMEOUT)
|
|
# if (RGB_MATRIX_TIMEOUT > 0)
|
|
rgb_matrix_disable_timeout_set(RGB_MATRIX_TIMEOUT_INFINITE);
|
|
rgb_matrix_disable_time_reset();
|
|
# endif
|
|
#endif
|
|
#if defined(LED_MATRIX_ENABLE) && defined(LED_MATRIX_TIMEOUT)
|
|
# if (LED_MATRIX_TIMEOUT > 0)
|
|
led_matrix_disable_timeout_set(LED_MATRIX_TIMEOUT_INFINITE);
|
|
led_matrix_disable_time_reset();
|
|
# endif
|
|
#endif
|
|
}
|
|
|
|
void usb_remote_wakeup(void) {
|
|
if (USB_DRIVER.state == USB_SUSPENDED) {
|
|
while (USB_DRIVER.state == USB_SUSPENDED) {
|
|
wireless_pre_task();
|
|
if (get_transport() != TRANSPORT_USB) {
|
|
suspend_wakeup_init_quantum();
|
|
return;
|
|
}
|
|
/* Do this in the suspended state */
|
|
suspend_power_down(); // on AVR this deep sleeps for 15ms
|
|
/* Remote wakeup */
|
|
if (suspend_wakeup_condition()) {
|
|
usbWakeupHost(&USB_DRIVER);
|
|
wait_ms(300);
|
|
#ifdef MOUSEKEY_ENABLE
|
|
// Wiggle to wakeup
|
|
mousekey_on(KC_MS_LEFT);
|
|
mousekey_send();
|
|
wait_ms(10);
|
|
mousekey_on(KC_MS_RIGHT);
|
|
mousekey_send();
|
|
wait_ms(10);
|
|
mousekey_off((KC_MS_RIGHT));
|
|
mousekey_send();
|
|
#else
|
|
set_mods(0x02);
|
|
send_keyboard_report();
|
|
wait_ms(10);
|
|
del_mods(0x02);
|
|
send_keyboard_report();
|
|
#endif
|
|
}
|
|
}
|
|
/* Woken up */
|
|
// variables has been already cleared by the wakeup hook
|
|
send_keyboard_report();
|
|
#ifdef MOUSEKEY_ENABLE
|
|
mousekey_send();
|
|
#endif /* MOUSEKEY_ENABLE */
|
|
usb_event_queue_task();
|
|
}
|
|
}
|