qmk_firmware/keyboards/nuphy/air75_v2/ansi/user_kb.c
2023-12-28 01:03:37 +08:00

416 lines
12 KiB
C

/*
Copyright 2023 @ Nuphy <https://nuphy.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 "user_kb.h"
#include "ansi.h"
#include "usb_main.h"
user_config_t user_config;
DEV_INFO_STRUCT dev_info = {
.rf_baterry = 100,
.link_mode = LINK_USB,
.rf_state = RF_IDLE,
};
bool f_bat_hold = 0;
bool f_sys_show = 0;
bool f_sleep_show = 0;
bool f_send_channel = 0;
bool f_dial_sw_init_ok = 0;
bool f_rf_sw_press = 0;
bool f_dev_reset_press = 0;
bool f_rgb_test_press = 0;
bool f_bat_num_show = 0;
uint8_t rf_blink_cnt = 0;
uint8_t rf_sw_temp = 0;
uint8_t host_mode = 0;
uint16_t rf_linking_time = 0;
uint16_t rf_link_show_time = 0;
uint16_t no_act_time = 0;
uint16_t dev_reset_press_delay = 0;
uint16_t rf_sw_press_delay = 0;
uint16_t rgb_test_press_delay = 0;
host_driver_t *m_host_driver = 0;
extern bool f_rf_new_adv_ok;
extern report_keyboard_t *keyboard_report;
extern report_nkro_t *nkro_report;
extern uint8_t bitkb_report_buf[32];
extern uint8_t bytekb_report_buf[8];
extern uint8_t side_mode;
extern uint8_t side_light;
extern uint8_t side_speed;
extern uint8_t side_rgb;
extern uint8_t side_colour;
/**
* @brief gpio initial.
*/
void gpio_init(void) {
/* enable led power driver */
setPinOutput(DRIVER_LED_CS_PIN);
setPinOutput(DRIVER_SIDE_CS_PIN);
writePinLow(DRIVER_LED_CS_PIN);
writePinLow(DRIVER_SIDE_CS_PIN);
/* set side led pin output low */
setPinOutput(DRIVER_SIDE_PIN);
writePinLow(DRIVER_SIDE_PIN);
/* config RF module pin */
setPinOutput(NRF_WAKEUP_PIN);
writePinHigh(NRF_WAKEUP_PIN);
setPinInputHigh(NRF_TEST_PIN);
/* reset RF module */
setPinOutput(NRF_RESET_PIN);
writePinLow(NRF_RESET_PIN);
wait_ms(50);
writePinHigh(NRF_RESET_PIN);
/* config dial switch pin */
setPinInputHigh(DEV_MODE_PIN);
setPinInputHigh(SYS_MODE_PIN);
/* open led DC driver */
setPinOutput(DC_BOOST_PIN);
writePinHigh(DC_BOOST_PIN);
}
/**
* @brief long press key process.
*/
void long_press_key(void) {
static uint32_t long_press_timer = 0;
if (timer_elapsed32(long_press_timer) < 100) return;
long_press_timer = timer_read32();
// Open a new RF device
if (f_rf_sw_press) {
rf_sw_press_delay++;
if (rf_sw_press_delay >= RF_LONG_PRESS_DELAY) {
f_rf_sw_press = 0;
dev_info.link_mode = rf_sw_temp;
dev_info.rf_channel = rf_sw_temp;
dev_info.ble_channel = rf_sw_temp;
uint8_t timeout = 5;
while (timeout--) {
uart_send_cmd(CMD_NEW_ADV, 0, 1);
wait_ms(20);
uart_receive_pro();
if (f_rf_new_adv_ok) break;
}
}
} else {
rf_sw_press_delay = 0;
}
// The device is restored to factory Settings
if (f_dev_reset_press) {
dev_reset_press_delay++;
if (dev_reset_press_delay >= DEV_RESET_PRESS_DELAY) {
f_dev_reset_press = 0;
if (dev_info.link_mode != LINK_USB) {
if (dev_info.link_mode != LINK_RF_24) {
dev_info.link_mode = LINK_BT_1;
dev_info.ble_channel = LINK_BT_1;
dev_info.rf_channel = LINK_BT_1;
}
} else {
dev_info.ble_channel = LINK_BT_1;
dev_info.rf_channel = LINK_BT_1;
}
uart_send_cmd(CMD_SET_LINK, 10, 10);
wait_ms(500);
uart_send_cmd(CMD_CLR_DEVICE, 10, 10);
void device_reset_show(void);
void device_reset_init(void);
eeconfig_init();
device_reset_show();
device_reset_init();
if (dev_info.sys_sw_state == SYS_SW_MAC) {
default_layer_set(1 << 0);
keymap_config.nkro = 0;
} else {
default_layer_set(1 << 2);
keymap_config.nkro = 1;
}
}
} else {
dev_reset_press_delay = 0;
}
// Enter the RGB test mode
if (f_rgb_test_press) {
rgb_test_press_delay++;
if (rgb_test_press_delay >= RGB_TEST_PRESS_DELAY) {
f_rgb_test_press = 0;
rgb_test_show();
}
} else {
rgb_test_press_delay = 0;
}
}
/**
* @brief Release all keys, clear keyboard report.
*/
void break_all_key(void) {
uint8_t report_buf[16];
bool nkro_temp = keymap_config.nkro;
clear_weak_mods();
clear_mods();
clear_keyboard();
keymap_config.nkro = 1;
memset(keyboard_report, 0, sizeof(report_keyboard_t));
host_keyboard_send(keyboard_report);
wait_ms(10);
keymap_config.nkro = 0;
memset(keyboard_report, 0, sizeof(report_keyboard_t));
host_keyboard_send(keyboard_report);
wait_ms(10);
keymap_config.nkro = nkro_temp;
if (dev_info.link_mode != LINK_USB) {
memset(report_buf, 0, 16);
uart_send_report(CMD_RPT_BIT_KB, report_buf, 16);
wait_ms(10);
uart_send_report(CMD_RPT_BYTE_KB, report_buf, 8);
wait_ms(10);
}
memset(bitkb_report_buf, 0, sizeof(bitkb_report_buf));
memset(bytekb_report_buf, 0, sizeof(bytekb_report_buf));
}
/**
* @brief switch device link mode.
* @param mode : link mode
*/
void switch_dev_link(uint8_t mode) {
if (mode > LINK_USB) return;
break_all_key();
dev_info.link_mode = mode;
dev_info.rf_state = RF_IDLE;
f_send_channel = 1;
if (mode == LINK_USB) {
host_mode = HOST_USB_TYPE;
host_set_driver(m_host_driver);
rf_link_show_time = 0;
} else {
host_mode = HOST_RF_TYPE;
host_set_driver(0);
}
}
/**
* @brief scan dial switch.
*/
void dial_sw_scan(void) {
uint8_t dial_scan = 0;
static uint8_t dial_save = 0xf0;
static uint8_t debounce = 0;
static uint32_t dial_scan_timer = 0;
static bool f_first = true;
if (!f_first) {
if (timer_elapsed32(dial_scan_timer) < 20) return;
}
dial_scan_timer = timer_read32();
setPinInputHigh(DEV_MODE_PIN);
setPinInputHigh(SYS_MODE_PIN);
if (readPin(DEV_MODE_PIN)) dial_scan |= 0X01;
if (readPin(SYS_MODE_PIN)) dial_scan |= 0X02;
if (dial_save != dial_scan) {
break_all_key();
no_act_time = 0;
rf_linking_time = 0;
dial_save = dial_scan;
debounce = 25;
f_dial_sw_init_ok = 0;
return;
} else if (debounce) {
debounce--;
return;
}
if (dial_scan & 0x01) {
if (dev_info.link_mode != LINK_USB) {
switch_dev_link(LINK_USB);
}
} else {
if (dev_info.link_mode != dev_info.rf_channel) {
switch_dev_link(dev_info.rf_channel);
}
}
if (dial_scan & 0x02) {
if (dev_info.sys_sw_state != SYS_SW_MAC) {
f_sys_show = 1;
default_layer_set(1 << 0);
dev_info.sys_sw_state = SYS_SW_MAC;
keymap_config.nkro = 0;
break_all_key();
}
} else {
if (dev_info.sys_sw_state != SYS_SW_WIN) {
f_sys_show = 1;
default_layer_set(1 << 2);
dev_info.sys_sw_state = SYS_SW_WIN;
keymap_config.nkro = 1;
break_all_key();
}
}
if (f_dial_sw_init_ok == 0) {
f_dial_sw_init_ok = 1;
f_first = false;
if (dev_info.link_mode != LINK_USB) {
host_set_driver(0);
}
}
}
/**
* @brief power on scan dial switch.
*/
void dial_sw_fast_scan(void) {
{
uint8_t dial_scan_dev = 0;
uint8_t dial_scan_sys = 0;
uint8_t dial_check_dev = 0;
uint8_t dial_check_sys = 0;
uint8_t debounce = 0;
setPinInputHigh(DEV_MODE_PIN);
setPinInputHigh(SYS_MODE_PIN);
// Debounce to get a stable state
for(debounce=0; debounce<10; debounce++) {
dial_scan_dev = 0;
dial_scan_sys = 0;
if (readPin(DEV_MODE_PIN)) dial_scan_dev = 0x01;
else dial_scan_dev = 0;
if (readPin(SYS_MODE_PIN)) dial_scan_sys = 0x01;
else dial_scan_sys = 0;
if((dial_scan_dev != dial_check_dev)||(dial_scan_sys != dial_check_sys))
{
dial_check_dev = dial_scan_dev;
dial_check_sys = dial_scan_sys;
debounce = 0;
}
wait_ms(1);
}
// RF link mode
if (dial_scan_dev) {
if (dev_info.link_mode != LINK_USB) {
switch_dev_link(LINK_USB);
}
} else {
if (dev_info.link_mode != dev_info.rf_channel) {
switch_dev_link(dev_info.rf_channel);
}
}
// Win or Mac
if (dial_scan_sys) {
if (dev_info.sys_sw_state != SYS_SW_MAC) {
default_layer_set(1 << 0);
dev_info.sys_sw_state = SYS_SW_MAC;
keymap_config.nkro = 0;
break_all_key();
}
} else {
if (dev_info.sys_sw_state != SYS_SW_WIN) {
//f_sys_show = 1;
default_layer_set(1 << 2);
dev_info.sys_sw_state = SYS_SW_WIN;
keymap_config.nkro = 1;
break_all_key();
}
}
}
}
/**
* @brief timer process.
*/
void timer_pro(void) {
static uint32_t interval_timer = 0;
static bool f_first = true;
if (f_first) {
f_first = false;
interval_timer = timer_read32();
m_host_driver = host_get_driver();
}
if (timer_elapsed32(interval_timer) < 10)
return;
else
interval_timer = timer_read32();
if (rf_link_show_time < RF_LINK_SHOW_TIME) rf_link_show_time++;
if (no_act_time < 0xffff) no_act_time++;
if (rf_linking_time < 0xffff) rf_linking_time++;
}
/**
* @brief londing eeprom data.
*/
void londing_eeprom_data(void) {
eeconfig_read_user_datablock(&user_config);
if (user_config.default_brightness_flag != 0xA5) {
/* first power on, set rgb matrix brightness at middle level*/
rgb_matrix_sethsv(255, 255, RGB_MATRIX_MAXIMUM_BRIGHTNESS - RGB_MATRIX_VAL_STEP * 2);
user_config.default_brightness_flag = 0xA5;
user_config.ee_side_mode = side_mode;
user_config.ee_side_light = side_light;
user_config.ee_side_speed = side_speed;
user_config.ee_side_rgb = side_rgb;
user_config.ee_side_colour = side_colour;
user_config.sleep_enable = true;
eeconfig_update_user_datablock(&user_config);
} else {
side_mode = user_config.ee_side_mode;
side_light = user_config.ee_side_light;
side_speed = user_config.ee_side_speed;
side_rgb = user_config.ee_side_rgb;
side_colour = user_config.ee_side_colour;
}
}