q11.c (5159B)
1 /* Copyright 2023 @ Keychron (https://www.keychron.com) 2 * 3 * This program is free software: you can redistribute it and/or modify 4 * it under the terms of the GNU General Public License as published by 5 * the Free Software Foundation, either version 2 of the License, or 6 * (at your option) any later version. 7 * 8 * This program is distributed in the hope that it will be useful, 9 * but WITHOUT ANY WARRANTY; without even the implied warranty of 10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 11 * GNU General Public License for more details. 12 * 13 * You should have received a copy of the GNU General Public License 14 * along with this program. If not, see <http://www.gnu.org/licenses/>. 15 */ 16 17 #include "quantum.h" 18 19 #ifdef DIP_SWITCH_ENABLE 20 bool dip_switch_update_kb(uint8_t index, bool active) { 21 if (!dip_switch_update_user(index, active)) { 22 return false; 23 } 24 if (index == 0) { 25 default_layer_set(1UL << (active ? 0 : 2)); 26 } 27 return true; 28 } 29 #endif 30 31 #if defined(RGB_MATRIX_ENABLE) && defined(CAPS_LOCK_LED_INDEX) 32 bool process_record_kb(uint16_t keycode, keyrecord_t *record) { 33 if (!process_record_user(keycode, record)) { 34 return false; 35 } 36 switch (keycode) { 37 case QK_RGB_MATRIX_TOGGLE: 38 if (record->event.pressed) { 39 switch (rgb_matrix_get_flags()) { 40 case LED_FLAG_ALL: { 41 rgb_matrix_set_flags(LED_FLAG_NONE); 42 rgb_matrix_set_color_all(0, 0, 0); 43 } break; 44 default: { 45 rgb_matrix_set_flags(LED_FLAG_ALL); 46 } break; 47 } 48 } 49 if (!rgb_matrix_is_enabled()) { 50 rgb_matrix_set_flags(LED_FLAG_ALL); 51 rgb_matrix_enable(); 52 } 53 return false; 54 } 55 return true; 56 } 57 58 bool rgb_matrix_indicators_advanced_kb(uint8_t led_min, uint8_t led_max) { 59 if (!rgb_matrix_indicators_advanced_user(led_min, led_max)) { 60 return false; 61 } 62 // RGB_MATRIX_INDICATOR_SET_COLOR(index, red, green, blue); 63 # if defined(CAPS_LOCK_LED_INDEX) 64 if (host_keyboard_led_state().caps_lock) { 65 RGB_MATRIX_INDICATOR_SET_COLOR(CAPS_LOCK_LED_INDEX, 255, 255, 255); 66 } else { 67 if (!rgb_matrix_get_flags()) { 68 RGB_MATRIX_INDICATOR_SET_COLOR(CAPS_LOCK_LED_INDEX, 0, 0, 0); 69 } 70 } 71 # endif // CAPS_LOCK_LED_INDEX 72 # if defined(NUM_LOCK_LED_INDEX) 73 if (host_keyboard_led_state().num_lock) { 74 RGB_MATRIX_INDICATOR_SET_COLOR(NUM_LOCK_LED_INDEX, 255, 255, 255); 75 } else { 76 if (!rgb_matrix_get_flags()) { 77 RGB_MATRIX_INDICATOR_SET_COLOR(NUM_LOCK_LED_INDEX, 0, 0, 0); 78 } 79 } 80 # endif // NUM_LOCK_LED_INDEX 81 return true; 82 } 83 #endif 84 85 #define ADC_BUFFER_DEPTH 1 86 #define ADC_NUM_CHANNELS 1 87 #define ADC_SAMPLING_RATE ADC_SMPR_SMP_12P5 88 #define ADC_RESOLUTION ADC_CFGR_RES_10BITS 89 90 static int16_t analogReadPin_my(pin_t pin) { 91 ADCConfig adcCfg = {}; 92 adcsample_t sampleBuffer[ADC_NUM_CHANNELS * ADC_BUFFER_DEPTH]; 93 ADCDriver *targetDriver = &ADCD1; 94 ADCConversionGroup adcConversionGroup = { 95 .circular = FALSE, 96 .num_channels = (uint16_t)(ADC_NUM_CHANNELS), 97 .cfgr = ADC_RESOLUTION, 98 }; 99 100 palSetLineMode(pin, PAL_MODE_INPUT_ANALOG); 101 switch (pin) { 102 case B0: 103 adcConversionGroup.smpr[2] = ADC_SMPR2_SMP_AN15(ADC_SAMPLING_RATE); 104 adcConversionGroup.sqr[0] = ADC_SQR1_SQ1_N(ADC_CHANNEL_IN15); 105 sampleBuffer[0] = 0; 106 break; 107 case B1: 108 adcConversionGroup.smpr[2] = ADC_SMPR2_SMP_AN16(ADC_SAMPLING_RATE); 109 adcConversionGroup.sqr[0] = ADC_SQR1_SQ1_N(ADC_CHANNEL_IN16); 110 sampleBuffer[0] = 0; 111 break; 112 default: 113 return 0; 114 } 115 adcStart(targetDriver, &adcCfg); 116 if (adcConvert(targetDriver, &adcConversionGroup, &sampleBuffer[0], ADC_BUFFER_DEPTH) != MSG_OK) { 117 return 0; 118 } 119 120 return *sampleBuffer; 121 } 122 123 void keyboard_post_init_kb(void) { 124 // 1. The pin A5/B5 of the USB C interface in the left hand is connected to the pin A0 of MCU, 125 // A0 will be set to output and write high when keyboard initial. 126 // 2. The same pin in the right hand is connected to the pin B0 and B1 of MCU respectively, 127 // and the ADC function of B0 and B1 will be enabled when keyboard initial. 128 // 3. because the serial usart RXD and TXD is multiplexed on USB's D+ and D- in the right hand. 129 // So detect the voltage on the pin A5/B5 of the USB C interface by ADC, 130 // and disable USB connectivity when the ADC value exceeds 1000, 131 // to avoid affecting the serial usart communication between the left hand and the right hand. 132 if (is_keyboard_left()) { 133 gpio_set_pin_output(A0); 134 gpio_write_pin_high(A0); 135 } else { 136 if ((analogReadPin_my(B0) > 1000) || (analogReadPin_my(B1) > 1000)) { 137 gpio_set_pin_input(A11); 138 gpio_set_pin_input(A12); 139 } 140 } 141 142 keyboard_post_init_user(); 143 }