66 lines
2.4 KiB
C
66 lines
2.4 KiB
C
// Copyright 2024 Dasky (@daskygit)
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#include <hal.h>
|
|
#include "chibios_config.h"
|
|
#include "gpio.h"
|
|
#include "color.h"
|
|
#include "rgblight_drivers.h"
|
|
|
|
static PWMConfig pwmCFG = {
|
|
.frequency = 0xFFFF,
|
|
.period = 256,
|
|
};
|
|
|
|
rgb_t prime_leds[RGBLIGHT_LED_COUNT];
|
|
|
|
void init_custom(void) {
|
|
palSetPadMode(PAL_PORT(RGB_RED_PIN), PAL_PAD(RGB_RED_PIN), PAL_MODE_ALTERNATE_PUSHPULL);
|
|
palSetPadMode(PAL_PORT(RGB_GREEN_PIN), PAL_PAD(RGB_GREEN_PIN), PAL_MODE_ALTERNATE_PUSHPULL);
|
|
palSetPadMode(PAL_PORT(RGB_BLUE_PIN), PAL_PAD(RGB_BLUE_PIN), PAL_MODE_ALTERNATE_PUSHPULL);
|
|
pwmCFG.channels[RGB_RED_PWM_CHANNEL - 1].mode = PWM_OUTPUT_ACTIVE_HIGH;
|
|
pwmCFG.channels[RGB_GREEN_PWM_CHANNEL - 1].mode = PWM_OUTPUT_ACTIVE_HIGH;
|
|
pwmCFG.channels[RGB_BLUE_PWM_CHANNEL - 1].mode = PWM_OUTPUT_ACTIVE_HIGH;
|
|
pwmStart(&RGB_PWM_DRIVER, &pwmCFG);
|
|
}
|
|
|
|
void set_color_custom(int index, uint8_t red, uint8_t green, uint8_t blue) {
|
|
prime_leds[index].r = red;
|
|
prime_leds[index].g = green;
|
|
prime_leds[index].b = blue;
|
|
}
|
|
|
|
void set_color_all_custom(uint8_t red, uint8_t green, uint8_t blue) {
|
|
for (int i = 0; i < RGBLIGHT_LED_COUNT; i++) {
|
|
set_color_custom(i, red, green, blue);
|
|
}
|
|
}
|
|
|
|
void flush_custom(void) {
|
|
if (prime_leds[0].r == 0) {
|
|
pwmDisableChannel(&RGB_PWM_DRIVER, RGB_RED_PWM_CHANNEL - 1);
|
|
} else {
|
|
uint32_t duty = ((uint32_t)0xFFFF * prime_leds[0].r) / 0xFF;
|
|
pwmEnableChannel(&RGB_PWM_DRIVER, RGB_RED_PWM_CHANNEL - 1, PWM_FRACTION_TO_WIDTH(&RGB_PWM_DRIVER, 0xFFFF, duty));
|
|
}
|
|
if (prime_leds[0].g == 0) {
|
|
pwmDisableChannel(&RGB_PWM_DRIVER, RGB_GREEN_PWM_CHANNEL - 1);
|
|
} else {
|
|
uint32_t duty = ((uint32_t)0xFFFF * prime_leds[0].g) / 0xFF;
|
|
pwmEnableChannel(&RGB_PWM_DRIVER, RGB_GREEN_PWM_CHANNEL - 1, PWM_FRACTION_TO_WIDTH(&RGB_PWM_DRIVER, 0xFFFF, duty));
|
|
}
|
|
if (prime_leds[0].b == 0) {
|
|
pwmDisableChannel(&RGB_PWM_DRIVER, RGB_BLUE_PWM_CHANNEL - 1);
|
|
} else {
|
|
uint32_t duty = ((uint32_t)0xFFFF * prime_leds[0].b) / 0xFF;
|
|
pwmEnableChannel(&RGB_PWM_DRIVER, RGB_BLUE_PWM_CHANNEL - 1, PWM_FRACTION_TO_WIDTH(&RGB_PWM_DRIVER, 0xFFFF, duty));
|
|
}
|
|
}
|
|
|
|
const rgblight_driver_t rgblight_driver = {
|
|
.init = init_custom,
|
|
.set_color = set_color_custom,
|
|
.set_color_all = set_color_all_custom,
|
|
.flush = flush_custom,
|
|
};
|