logo

qmk_firmware

custom branch of QMK firmware git clone https://anongit.hacktivis.me/git/qmk_firmware.git

mux.c (2533B)

    

  1. // Copyright 2023 Massdrop, Inc.
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  3. #include "keyboard.h"
  4. #include "analog.h"
  5. #include "gpio.h"
  6. #include "wait.h"
  8. #define S_UP        B12
  9. #define E_UP_N      B13
  10. #define S_DN1       B14
  11. #define E_DN1_N     B15
  12. #define E_VBUS_1    A8
  13. #define E_VBUS_2    A15
  14. #define SRC_1       B8
  15. #define SRC_2       B9
  16. #define HUB_RESET_N D2
  17. #define IRST        B1
  18. #define SDB_N       B2
  19. #define C1_A5_SENSE A4
  20. #define C1_B5_SENSE A7
  21. #define C2_A5_SENSE C4
  22. #define C2_B5_SENSE B0
  24. static inline void digital_write(pin_t pin, uint8_t level) {
  25.     gpio_set_pin_output(pin);
  26.     gpio_write_pin(pin, level);
  27. }
  29. uint16_t v_con_1 = 0;
  30. uint16_t v_con_2 = 0;
  31. uint16_t v_con_3 = 0;
  32. uint16_t v_con_4 = 0;
  34. void keyboard_USB_enable(void) {
  35.     // initial state
  36.     digital_write(S_UP, 0);
  37.     digital_write(E_UP_N, 1);
  38.     digital_write(S_DN1, 1);
  39.     digital_write(E_DN1_N, 1);
  40.     digital_write(E_VBUS_1, 0);
  41.     digital_write(E_VBUS_2, 0);
  42.     digital_write(SRC_1, 1);
  43.     digital_write(SRC_2, 1);
  45.     gpio_set_pin_input(C1_A5_SENSE);
  46.     gpio_set_pin_input(C1_B5_SENSE);
  47.     gpio_set_pin_input(C2_A5_SENSE);
  48.     gpio_set_pin_input(C2_B5_SENSE);
  50.     // reset hub
  51.     digital_write(HUB_RESET_N, 0);
  52.     wait_ms(100);
  53.     digital_write(HUB_RESET_N, 1);
  55.     wait_ms(500); // Allow power dissipation time on CC lines
  57.     v_con_1 = analogReadPin(C1_A5_SENSE);
  58.     v_con_2 = analogReadPin(C1_B5_SENSE);
  59.     v_con_3 = analogReadPin(C2_A5_SENSE);
  60.     v_con_4 = analogReadPin(C2_B5_SENSE);
  62.     // TODO: dynamic port port configure logic?
  63.     digital_write(E_UP_N, 0);   // HOST enable
  64.     digital_write(E_DN1_N, 0);  // EXTRA enable
  65.     digital_write(E_VBUS_1, 1); // USBC-1 enable full power I/O
  66.     digital_write(E_VBUS_2, 1); // USBC-2 enable full power I/O
  68.     if ((v_con_1 + v_con_2) > (v_con_3 + v_con_4)) {
  69.         digital_write(S_UP, 0);  // HOST to USBC-1
  70.         digital_write(S_DN1, 1); // EXTRA to USBC-2
  71.         digital_write(SRC_1, 1); // HOST on USBC-1
  72.         digital_write(SRC_2, 0); // EXTRA available on USBC-2
  73.     } else {
  74.         digital_write(S_UP, 1);  // EXTRA to USBC-1
  75.         digital_write(S_DN1, 0); // HOST to USBC-2
  76.         digital_write(SRC_1, 0); // EXTRA available on USBC-1
  77.         digital_write(SRC_2, 1); // HOST on USBC-2
  78.     }
  79. }
  81. void keyboard_ISSI_enable(void) {
  82. #ifdef RGB_MATRIX_ENABLE
  83.     digital_write(IRST, 0);
  84.     digital_write(SDB_N, 1);
  85. #endif
  86. }
  89. void keyboard_pre_init_kb(void) {
  90.     keyboard_USB_enable();
  91.     keyboard_ISSI_enable();
  93.     keyboard_pre_init_user();
  94. }