logo

qmk_firmware

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

abelx.c (2986B)

    

  1. /**
  2.  * abelx.c
  3.  *
  4.  * Copyright 2020 astro <yuleiz@gmail.com>
  5.  *
  6.  * This program is free software: you can redistribute it and/or modify
  7.  * it under the terms of the GNU General Public License as published by
  8.  * the Free Software Foundation, either version 2 of the License, or
  9.  * (at your option) any later version.
  10.  *
  11.  * This program is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.  * GNU General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU General Public License
  17.  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18.  */
  20. #include "abelx.h"
  21. #include "tca6424.h"
  22. #include "aw9523b.h"
  24. void set_pin(uint16_t pin)
  25. {
  26.     uint8_t data = tca6424_read_port(GET_PORT(pin));
  27.     data |= ( 1 << GET_PIN(pin));
  28.     tca6424_write_port(GET_PORT(pin), data);
  29. }
  31. void clear_pin(uint16_t pin)
  32. {
  33.     uint8_t data = tca6424_read_port(GET_PORT(pin));
  34.     data &= ~( 1 << GET_PIN(pin));
  35.     tca6424_write_port(GET_PORT(pin), data);
  36. }
  38. uint8_t read_pin(uint16_t pin)
  39. {
  40.     uint8_t data = tca6424_read_port(GET_PORT(pin));
  41.     return (data & (1<<GET_PIN(pin))) ? 1 : 0;
  42. }
  44. #ifdef RGBLIGHT_ENABLE
  45. #include "rgblight.h"
  46. #include "ws2812.h"
  47. #include "i2c_master.h"
  49. const aw9523b_led g_aw9523b_leds[AW9523B_RGB_NUM] = {
  50.     {AW9523B_P12_PWM, AW9523B_P11_PWM, AW9523B_P10_PWM},
  51.     {AW9523B_P01_PWM, AW9523B_P00_PWM, AW9523B_P13_PWM},
  52.     {AW9523B_P04_PWM, AW9523B_P03_PWM, AW9523B_P02_PWM},
  53.     {AW9523B_P07_PWM, AW9523B_P06_PWM, AW9523B_P05_PWM},
  54. };
  56. void init_custom(void) {
  57.     aw9523b_init(AW9523B_ADDR);
  58.     ws2812_init();
  59. }
  61. void set_color_custom(int index, uint8_t red, uint8_t green, uint8_t blue) {
  62.     if (index < AW9523B_RGB_NUM) {
  63.         aw9523b_set_color(index, red, green, blue);
  64.     } else {
  65.         ws2812_set_color(index - AW9523B_RGB_NUM, red, green, blue);
  66.     }
  67. }
  69. void set_color_all_custom(uint8_t red, uint8_t green, uint8_t blue) {
  70.     for (int i = 0; i < RGBLIGHT_LED_COUNT; i++) {
  71.         set_color_custom(i, red, green, blue);
  72.     }
  73. }
  75. void flush_custom(void) {
  76.     aw9523b_update_pwm_buffers(AW9523B_ADDR);
  77.     ws2812_flush();
  78. }
  80. const rgblight_driver_t rgblight_driver = {
  81.     .init          = init_custom,
  82.     .set_color     = set_color_custom,
  83.     .set_color_all = set_color_all_custom,
  84.     .flush         = flush_custom,
  85. };
  87. #endif
  89. static uint16_t caps_lock_pin = DEF_PIN(TCA6424_PORT2, 3);
  90. static uint16_t scroll_lock_pin = DEF_PIN(TCA6424_PORT0, 0);
  92. bool led_update_kb(led_t led_state) {
  93.     bool res = led_update_user(led_state);
  94.     if (res) {
  95.         led_state.caps_lock ? set_pin(caps_lock_pin) : clear_pin(caps_lock_pin);
  96.         led_state.scroll_lock ? set_pin(scroll_lock_pin) : clear_pin(scroll_lock_pin);
  97.     }
  98.     return res;
  99. }
  101. #define REBOOT_MAGIC 0x41544B42
  103. void bootloader_jump(void) {
  104.     *(uint32_t *)(&(RTCD1.rtc->BKP0R)) = REBOOT_MAGIC;
  105.     NVIC_SystemReset();
  106. }