feat: 添加LED灯带模块支持
- 在CMakeLists.txt中添加led_strip_module.c源文件和led_strip_en_event.c事件文件 - 在设备树配置中添加SPI1接口的WS2812灯带引脚控制配置 - 在板级配置文件中添加LED灯带设备节点和别名定义 - 新增led_strip_en_event事件头文件和实现，用于控制灯带使能状态 - 配置prj.conf启用LED Strip和WS2812 SPI驱动 - 实现完整的LED灯带模块功能，包括： - 初始化和电源管理 - RGB色彩效果渲染 - 通过GPIO控制灯带供电 - 响应应用事件进行启停控制
This commit is contained in:
281
src/led_strip_module.c
Normal file
281
src/led_strip_module.c
Normal file
@@ -0,0 +1,281 @@
|
||||
#include <errno.h>
|
||||
#include <stdbool.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <app_event_manager.h>
|
||||
|
||||
#define MODULE led_strip_module
|
||||
#include <caf/events/module_state_event.h>
|
||||
#include <caf/events/power_event.h>
|
||||
|
||||
#include <zephyr/device.h>
|
||||
#include <zephyr/drivers/gpio.h>
|
||||
#include <zephyr/drivers/led_strip.h>
|
||||
#include <zephyr/kernel.h>
|
||||
#include <zephyr/logging/log.h>
|
||||
|
||||
#include "led_strip_en_event.h"
|
||||
|
||||
LOG_MODULE_REGISTER(MODULE, LOG_LEVEL_INF);
|
||||
|
||||
#define LED_STRIP_NODE DT_CHOSEN(zephyr_led_strip)
|
||||
#define LED_STRIP_NUM_PIXELS DT_PROP(LED_STRIP_NODE, chain_length)
|
||||
#define LED_STRIP_EFFECT_INTERVAL K_MSEC(400)
|
||||
|
||||
BUILD_ASSERT(DT_NODE_HAS_STATUS(LED_STRIP_NODE, okay),
|
||||
"Missing zephyr,led-strip chosen node");
|
||||
BUILD_ASSERT(DT_NODE_HAS_PROP(LED_STRIP_NODE, supply_gpios),
|
||||
"Missing supply-gpios on zephyr,led-strip node");
|
||||
|
||||
static const struct device *const strip = DEVICE_DT_GET(LED_STRIP_NODE);
|
||||
static const struct gpio_dt_spec strip_en =
|
||||
GPIO_DT_SPEC_GET(LED_STRIP_NODE, supply_gpios);
|
||||
|
||||
static struct led_rgb pixels[LED_STRIP_NUM_PIXELS];
|
||||
static struct k_work_delayable effect_work;
|
||||
static bool initialized;
|
||||
static bool running;
|
||||
static bool enabled = true;
|
||||
static uint8_t effect_step;
|
||||
|
||||
static void clear_pixels(void)
|
||||
{
|
||||
memset(pixels, 0, sizeof(pixels));
|
||||
}
|
||||
|
||||
static void fill_pixels(uint8_t r, uint8_t g, uint8_t b)
|
||||
{
|
||||
for (size_t i = 0; i < ARRAY_SIZE(pixels); i++) {
|
||||
pixels[i].r = r;
|
||||
pixels[i].g = g;
|
||||
pixels[i].b = b;
|
||||
}
|
||||
}
|
||||
|
||||
static int submit_frame(void)
|
||||
{
|
||||
int err;
|
||||
|
||||
err = led_strip_update_rgb(strip, pixels, ARRAY_SIZE(pixels));
|
||||
if (err) {
|
||||
LOG_WRN("led_strip_update_rgb failed (%d)", err);
|
||||
}
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
static int set_strip_power(bool on)
|
||||
{
|
||||
int err;
|
||||
|
||||
err = gpio_pin_set_dt(&strip_en, on ? 1 : 0);
|
||||
if (err) {
|
||||
LOG_WRN("LED strip EN set failed (%d)", err);
|
||||
}
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
static void render_effect_step(void)
|
||||
{
|
||||
switch (effect_step) {
|
||||
case 0U:
|
||||
fill_pixels(0x40, 0x00, 0x00);
|
||||
break;
|
||||
|
||||
case 1U:
|
||||
fill_pixels(0x00, 0x40, 0x00);
|
||||
break;
|
||||
|
||||
default:
|
||||
fill_pixels(0x00, 0x00, 0x40);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void effect_work_handler(struct k_work *work)
|
||||
{
|
||||
ARG_UNUSED(work);
|
||||
|
||||
if (!running || !enabled) {
|
||||
return;
|
||||
}
|
||||
|
||||
render_effect_step();
|
||||
(void)submit_frame();
|
||||
effect_step = (uint8_t)((effect_step + 1U) % 3U);
|
||||
k_work_reschedule(&effect_work, LED_STRIP_EFFECT_INTERVAL);
|
||||
}
|
||||
|
||||
static int enable_effect_output(void)
|
||||
{
|
||||
int err;
|
||||
|
||||
err = set_strip_power(true);
|
||||
if (err) {
|
||||
return err;
|
||||
}
|
||||
|
||||
render_effect_step();
|
||||
err = submit_frame();
|
||||
if (err) {
|
||||
return err;
|
||||
}
|
||||
|
||||
k_work_reschedule(&effect_work, LED_STRIP_EFFECT_INTERVAL);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void disable_effect_output(void)
|
||||
{
|
||||
k_work_cancel_delayable(&effect_work);
|
||||
clear_pixels();
|
||||
(void)submit_frame();
|
||||
(void)set_strip_power(false);
|
||||
}
|
||||
|
||||
static int module_init(void)
|
||||
{
|
||||
int err;
|
||||
|
||||
if (!device_is_ready(strip)) {
|
||||
LOG_ERR("LED strip device %s not ready", strip->name);
|
||||
return -ENODEV;
|
||||
}
|
||||
|
||||
if (!gpio_is_ready_dt(&strip_en)) {
|
||||
LOG_ERR("LED strip EN GPIO not ready");
|
||||
return -ENODEV;
|
||||
}
|
||||
|
||||
err = gpio_pin_configure_dt(&strip_en, GPIO_OUTPUT_INACTIVE);
|
||||
if (err) {
|
||||
LOG_ERR("Failed to configure LED strip EN (%d)", err);
|
||||
return err;
|
||||
}
|
||||
|
||||
k_work_init_delayable(&effect_work, effect_work_handler);
|
||||
clear_pixels();
|
||||
effect_step = 0U;
|
||||
|
||||
LOG_INF("LED strip ready len:%u", (uint32_t)led_strip_length(strip));
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int module_start(void)
|
||||
{
|
||||
int err;
|
||||
|
||||
if (running) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
running = true;
|
||||
|
||||
if (!enabled) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
err = enable_effect_output();
|
||||
if (err) {
|
||||
running = false;
|
||||
}
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
static void module_pause(void)
|
||||
{
|
||||
if (!running) {
|
||||
return;
|
||||
}
|
||||
|
||||
disable_effect_output();
|
||||
running = false;
|
||||
}
|
||||
|
||||
static bool handle_led_strip_en_event(const struct led_strip_en_event *event)
|
||||
{
|
||||
enabled = event->enabled;
|
||||
|
||||
if (!running) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (enabled) {
|
||||
int err = enable_effect_output();
|
||||
|
||||
if (err) {
|
||||
module_set_state(MODULE_STATE_ERROR);
|
||||
}
|
||||
} else {
|
||||
disable_effect_output();
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
static bool app_event_handler(const struct app_event_header *aeh)
|
||||
{
|
||||
if (is_led_strip_en_event(aeh)) {
|
||||
return handle_led_strip_en_event(cast_led_strip_en_event(aeh));
|
||||
}
|
||||
|
||||
if (is_module_state_event(aeh)) {
|
||||
const struct module_state_event *event = cast_module_state_event(aeh);
|
||||
|
||||
if (check_state(event, MODULE_ID(main), MODULE_STATE_READY)) {
|
||||
int err;
|
||||
|
||||
if (!initialized) {
|
||||
err = module_init();
|
||||
if (err) {
|
||||
module_set_state(MODULE_STATE_ERROR);
|
||||
return false;
|
||||
}
|
||||
|
||||
initialized = true;
|
||||
}
|
||||
|
||||
err = module_start();
|
||||
if (err) {
|
||||
module_set_state(MODULE_STATE_ERROR);
|
||||
} else {
|
||||
module_set_state(MODULE_STATE_READY);
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if (is_power_down_event(aeh)) {
|
||||
if (initialized) {
|
||||
module_pause();
|
||||
module_set_state(MODULE_STATE_STANDBY);
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if (is_wake_up_event(aeh)) {
|
||||
if (initialized) {
|
||||
int err = module_start();
|
||||
|
||||
if (err) {
|
||||
module_set_state(MODULE_STATE_ERROR);
|
||||
} else {
|
||||
module_set_state(MODULE_STATE_READY);
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
APP_EVENT_LISTENER(MODULE, app_event_handler);
|
||||
APP_EVENT_SUBSCRIBE(MODULE, led_strip_en_event);
|
||||
APP_EVENT_SUBSCRIBE(MODULE, module_state_event);
|
||||
APP_EVENT_SUBSCRIBE_EARLY(MODULE, power_down_event);
|
||||
APP_EVENT_SUBSCRIBE(MODULE, wake_up_event);
|
||||
Reference in New Issue
Block a user