mirror of
https://github.com/Keychron/qmk_firmware.git
synced 2024-11-23 17:06:52 +06:00
9632360caa
* Add ARRAY_SIZE and CEILING utility macros * Apply a coccinelle patch to use ARRAY_SIZE * fix up some straggling items * Fix 'make test:secure' * Enhance ARRAY_SIZE macro to reject acting on pointers The previous definition would not produce a diagnostic for ``` int *p; size_t num_elem = ARRAY_SIZE(p) ``` but the new one will. * explicitly get definition of ARRAY_SIZE * Convert to ARRAY_SIZE when const is involved The following spatch finds additional instances where the array is const and the division is by the size of the type, not the size of the first element: ``` @ rule5a using "empty.iso" @ type T; const T[] E; @@ - (sizeof(E)/sizeof(T)) + ARRAY_SIZE(E) @ rule6a using "empty.iso" @ type T; const T[] E; @@ - sizeof(E)/sizeof(T) + ARRAY_SIZE(E) ``` * New instances of ARRAY_SIZE added since initial spatch run * Use `ARRAY_SIZE` in docs (found by grep) * Manually use ARRAY_SIZE hs_set is expected to be the same size as uint16_t, though it's made of two 8-bit integers * Just like char, sizeof(uint8_t) is guaranteed to be 1 This is at least true on any plausible system where qmk is actually used. Per my understanding it's universally true, assuming that uint8_t exists: https://stackoverflow.com/questions/48655310/can-i-assume-that-sizeofuint8-t-1 * Run qmk-format on core C files touched in this branch Co-authored-by: Stefan Kerkmann <karlk90@pm.me>
384 lines
12 KiB
C
384 lines
12 KiB
C
/* Copyright 2018 James Laird-Wah
|
|
*
|
|
* 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 "rgb_matrix.h"
|
|
|
|
/* Each driver needs to define the struct
|
|
* const rgb_matrix_driver_t rgb_matrix_driver;
|
|
* All members must be provided.
|
|
* Keyboard custom drivers can define this in their own files, it should only
|
|
* be here if shared between boards.
|
|
*/
|
|
|
|
#if defined(IS31FL3731) || defined(IS31FL3733) || defined(IS31FL3737) || defined(IS31FL3741) || defined(IS31FLCOMMON) || defined(CKLED2001)
|
|
# include "i2c_master.h"
|
|
|
|
// TODO: Remove this at some later date
|
|
# if defined(DRIVER_ADDR_1) && defined(DRIVER_ADDR_2)
|
|
# if DRIVER_ADDR_1 == DRIVER_ADDR_2
|
|
# error "Setting DRIVER_ADDR_2 == DRIVER_ADDR_1 is obsolete. If you are only using one ISSI driver, set DRIVER_COUNT to 1 and remove DRIVER_ADDR_2"
|
|
# endif
|
|
# endif
|
|
|
|
static void init(void) {
|
|
i2c_init();
|
|
|
|
# if defined(IS31FL3731)
|
|
IS31FL3731_init(DRIVER_ADDR_1);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3731_init(DRIVER_ADDR_2);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL3731_init(DRIVER_ADDR_3);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL3731_init(DRIVER_ADDR_4);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
# elif defined(IS31FL3733)
|
|
# if !defined(DRIVER_SYNC_1)
|
|
# define DRIVER_SYNC_1 0
|
|
# endif
|
|
IS31FL3733_init(DRIVER_ADDR_1, DRIVER_SYNC_1);
|
|
# if defined(DRIVER_ADDR_2)
|
|
# if !defined(DRIVER_SYNC_2)
|
|
# define DRIVER_SYNC_2 0
|
|
# endif
|
|
IS31FL3733_init(DRIVER_ADDR_2, DRIVER_SYNC_2);
|
|
# if defined(DRIVER_ADDR_3)
|
|
# if !defined(DRIVER_SYNC_3)
|
|
# define DRIVER_SYNC_3 0
|
|
# endif
|
|
IS31FL3733_init(DRIVER_ADDR_3, DRIVER_SYNC_3);
|
|
# if defined(DRIVER_ADDR_4)
|
|
# if !defined(DRIVER_SYNC_4)
|
|
# define DRIVER_SYNC_4 0
|
|
# endif
|
|
IS31FL3733_init(DRIVER_ADDR_4, DRIVER_SYNC_4);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
# elif defined(IS31FL3737)
|
|
IS31FL3737_init(DRIVER_ADDR_1);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3737_init(DRIVER_ADDR_2);
|
|
# endif
|
|
|
|
# elif defined(IS31FL3741)
|
|
IS31FL3741_init(DRIVER_ADDR_1);
|
|
|
|
# elif defined(IS31FLCOMMON)
|
|
IS31FL_common_init(DRIVER_ADDR_1, ISSI_SSR_1);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL_common_init(DRIVER_ADDR_2, ISSI_SSR_2);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL_common_init(DRIVER_ADDR_3, ISSI_SSR_3);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL_common_init(DRIVER_ADDR_4, ISSI_SSR_4);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
# elif defined(CKLED2001)
|
|
CKLED2001_init(DRIVER_ADDR_1);
|
|
# if defined(DRIVER_ADDR_2)
|
|
CKLED2001_init(DRIVER_ADDR_2);
|
|
# if defined(DRIVER_ADDR_3)
|
|
CKLED2001_init(DRIVER_ADDR_3);
|
|
# if defined(DRIVER_ADDR_4)
|
|
CKLED2001_init(DRIVER_ADDR_4);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
for (int index = 0; index < DRIVER_LED_TOTAL; index++) {
|
|
bool enabled = true;
|
|
|
|
// This only caches it for later
|
|
# if defined(IS31FL3731)
|
|
IS31FL3731_set_led_control_register(index, enabled, enabled, enabled);
|
|
# elif defined(IS31FL3733)
|
|
IS31FL3733_set_led_control_register(index, enabled, enabled, enabled);
|
|
# elif defined(IS31FL3737)
|
|
IS31FL3737_set_led_control_register(index, enabled, enabled, enabled);
|
|
# elif defined(IS31FL3741)
|
|
IS31FL3741_set_led_control_register(index, enabled, enabled, enabled);
|
|
# elif defined(IS31FLCOMMON)
|
|
IS31FL_RGB_set_scaling_buffer(index, enabled, enabled, enabled);
|
|
# elif defined(CKLED2001)
|
|
CKLED2001_set_led_control_register(index, enabled, enabled, enabled);
|
|
# endif
|
|
}
|
|
|
|
// This actually updates the LED drivers
|
|
# if defined(IS31FL3731)
|
|
IS31FL3731_update_led_control_registers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3731_update_led_control_registers(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL3731_update_led_control_registers(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL3731_update_led_control_registers(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
# elif defined(IS31FL3733)
|
|
IS31FL3733_update_led_control_registers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3733_update_led_control_registers(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL3733_update_led_control_registers(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL3733_update_led_control_registers(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
# elif defined(IS31FL3737)
|
|
IS31FL3737_update_led_control_registers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3737_update_led_control_registers(DRIVER_ADDR_2, 1);
|
|
# endif
|
|
|
|
# elif defined(IS31FL3741)
|
|
IS31FL3741_update_led_control_registers(DRIVER_ADDR_1, 0);
|
|
|
|
# elif defined(IS31FLCOMMON)
|
|
# ifdef ISSI_MANUAL_SCALING
|
|
IS31FL_set_manual_scaling_buffer();
|
|
# endif
|
|
IS31FL_common_update_scaling_register(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL_common_update_scaling_register(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL_common_update_scaling_register(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL_common_update_scaling_register(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
|
|
# elif defined(CKLED2001)
|
|
CKLED2001_update_led_control_registers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
CKLED2001_update_led_control_registers(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
CKLED2001_update_led_control_registers(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
CKLED2001_update_led_control_registers(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
# endif
|
|
}
|
|
|
|
# if defined(IS31FL3731)
|
|
static void flush(void) {
|
|
IS31FL3731_update_pwm_buffers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3731_update_pwm_buffers(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL3731_update_pwm_buffers(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL3731_update_pwm_buffers(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = IS31FL3731_set_color,
|
|
.set_color_all = IS31FL3731_set_color_all,
|
|
};
|
|
|
|
# elif defined(IS31FL3733)
|
|
static void flush(void) {
|
|
IS31FL3733_update_pwm_buffers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3733_update_pwm_buffers(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL3733_update_pwm_buffers(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL3733_update_pwm_buffers(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = IS31FL3733_set_color,
|
|
.set_color_all = IS31FL3733_set_color_all,
|
|
};
|
|
|
|
# elif defined(IS31FL3737)
|
|
static void flush(void) {
|
|
IS31FL3737_update_pwm_buffers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3737_update_pwm_buffers(DRIVER_ADDR_2, 1);
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = IS31FL3737_set_color,
|
|
.set_color_all = IS31FL3737_set_color_all,
|
|
};
|
|
|
|
# elif defined(IS31FL3741)
|
|
static void flush(void) {
|
|
IS31FL3741_update_pwm_buffers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL3741_update_pwm_buffers(DRIVER_ADDR_2, 1);
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = IS31FL3741_set_color,
|
|
.set_color_all = IS31FL3741_set_color_all,
|
|
};
|
|
|
|
# elif defined(IS31FLCOMMON)
|
|
static void flush(void) {
|
|
IS31FL_common_update_pwm_register(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
IS31FL_common_update_pwm_register(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
IS31FL_common_update_pwm_register(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
IS31FL_common_update_pwm_register(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = IS31FL_RGB_set_color,
|
|
.set_color_all = IS31FL_RGB_set_color_all,
|
|
};
|
|
|
|
# elif defined(CKLED2001)
|
|
static void flush(void) {
|
|
CKLED2001_update_pwm_buffers(DRIVER_ADDR_1, 0);
|
|
# if defined(DRIVER_ADDR_2)
|
|
CKLED2001_update_pwm_buffers(DRIVER_ADDR_2, 1);
|
|
# if defined(DRIVER_ADDR_3)
|
|
CKLED2001_update_pwm_buffers(DRIVER_ADDR_3, 2);
|
|
# if defined(DRIVER_ADDR_4)
|
|
CKLED2001_update_pwm_buffers(DRIVER_ADDR_4, 3);
|
|
# endif
|
|
# endif
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = CKLED2001_set_color,
|
|
.set_color_all = CKLED2001_set_color_all,
|
|
};
|
|
# endif
|
|
|
|
#elif defined(AW20216)
|
|
# include "spi_master.h"
|
|
|
|
static void init(void) {
|
|
spi_init();
|
|
|
|
AW20216_init(DRIVER_1_CS, DRIVER_1_EN);
|
|
# if defined(DRIVER_2_CS)
|
|
AW20216_init(DRIVER_2_CS, DRIVER_2_EN);
|
|
# endif
|
|
}
|
|
|
|
static void flush(void) {
|
|
AW20216_update_pwm_buffers(DRIVER_1_CS, 0);
|
|
# if defined(DRIVER_2_CS)
|
|
AW20216_update_pwm_buffers(DRIVER_2_CS, 1);
|
|
# endif
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = AW20216_set_color,
|
|
.set_color_all = AW20216_set_color_all,
|
|
};
|
|
|
|
#elif defined(WS2812)
|
|
# if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_CUSTOM_DRIVER)
|
|
# pragma message "Cannot use RGBLIGHT and RGB Matrix using WS2812 at the same time."
|
|
# pragma message "You need to use a custom driver, or re-implement the WS2812 driver to use a different configuration."
|
|
# endif
|
|
|
|
// LED color buffer
|
|
LED_TYPE rgb_matrix_ws2812_array[DRIVER_LED_TOTAL];
|
|
|
|
static void init(void) {}
|
|
|
|
static void flush(void) {
|
|
// Assumes use of RGB_DI_PIN
|
|
ws2812_setleds(rgb_matrix_ws2812_array, DRIVER_LED_TOTAL);
|
|
}
|
|
|
|
// Set an led in the buffer to a color
|
|
static inline void setled(int i, uint8_t r, uint8_t g, uint8_t b) {
|
|
# if defined(RGB_MATRIX_ENABLE) && defined(RGB_MATRIX_SPLIT)
|
|
const uint8_t k_rgb_matrix_split[2] = RGB_MATRIX_SPLIT;
|
|
if (!is_keyboard_left()) {
|
|
if (i >= k_rgb_matrix_split[0]) {
|
|
i -= k_rgb_matrix_split[0];
|
|
} else {
|
|
return;
|
|
}
|
|
} else if (i >= k_rgb_matrix_split[0]) {
|
|
return;
|
|
}
|
|
# endif
|
|
|
|
rgb_matrix_ws2812_array[i].r = r;
|
|
rgb_matrix_ws2812_array[i].g = g;
|
|
rgb_matrix_ws2812_array[i].b = b;
|
|
# ifdef RGBW
|
|
convert_rgb_to_rgbw(&rgb_matrix_ws2812_array[i]);
|
|
# endif
|
|
}
|
|
|
|
static void setled_all(uint8_t r, uint8_t g, uint8_t b) {
|
|
for (int i = 0; i < ARRAY_SIZE(rgb_matrix_ws2812_array); i++) {
|
|
setled(i, r, g, b);
|
|
}
|
|
}
|
|
|
|
const rgb_matrix_driver_t rgb_matrix_driver = {
|
|
.init = init,
|
|
.flush = flush,
|
|
.set_color = setled,
|
|
.set_color_all = setled_all,
|
|
};
|
|
#endif
|