LED strip support

Author: sctanf

src/system/led.c

@@ -16,6 +16,13 @@ K_THREAD_DEFINE(led_thread_id, 512, led_thread, NULL, NULL, NULL, 6, 0, 0);
 
 #define ZEPHYR_USER_NODE DT_PATH(zephyr_user)
 
+#if CONFIG_LED_STRIP
+#define LED_STRIP_EXISTS true
+#include <zephyr/drivers/led_strip.h>
+#define STRIP_NODE DT_ALIAS(led_strip)
+static const struct device *const strip = DEVICE_DT_GET(STRIP_NODE);
+#endif
+
 #if DT_NODE_HAS_PROP(ZEPHYR_USER_NODE, led_gpios)
 #define LED_EXISTS true
 static const struct gpio_dt_spec led = GPIO_DT_SPEC_GET(ZEPHYR_USER_NODE, led_gpios);
@@ -30,8 +37,9 @@ static const struct gpio_dt_spec led0 = GPIO_DT_SPEC_GET(DT_ALIAS(led0), gpios);
 #endif
 #endif
 #ifndef LED_EXISTS
+#ifndef LED_STRIP_EXISTS
 #warning "LED GPIO does not exist"
-//static const struct gpio_dt_spec led = {0};
+#endif
 #endif
 #if DT_NODE_EXISTS(DT_ALIAS(led1))
 #define LED1_EXISTS true
@@ -50,8 +58,10 @@ static const struct gpio_dt_spec led3 = GPIO_DT_SPEC_GET(DT_ALIAS(led3), gpios);
 #define PWM_LED_EXISTS true
 static const struct pwm_dt_spec pwm_led = PWM_DT_SPEC_GET(DT_ALIAS(pwm_led0));
 #else
+#ifndef LED_STRIP_EXISTS
 #warning "PWM LED node does not exist"
 #endif
+#endif
 #if DT_NODE_EXISTS(DT_ALIAS(pwm_led1))
 #define PWM_LED1_EXISTS true
 static const struct pwm_dt_spec pwm_led1 = PWM_DT_SPEC_GET(DT_ALIAS(pwm_led1));
@@ -64,15 +74,17 @@ static const struct pwm_dt_spec pwm_led2 = PWM_DT_SPEC_GET(DT_ALIAS(pwm_led2));
 static enum sys_led_pattern current_led_pattern;
 static int current_priority;
 
-#if LED_EXISTS
+#if LED_EXISTS || LED_STRIP_EXISTS
 static enum sys_led_pattern led_patterns[SYS_LED_PATTERN_DEPTH] = {[0 ... (SYS_LED_PATTERN_DEPTH - 1)] = SYS_LED_PATTERN_OFF};
 static int led_pattern_state;
 
 static int led_pin_init(void)
 {
 	LOG_DBG("led_pin_init");
+#if LED_EXISTS
 	gpio_pin_configure_dt(&led, GPIO_OUTPUT);
 	gpio_pin_set_dt(&led, 0);
+#endif
 #if LED0_EXISTS
 	gpio_pin_configure_dt(&led0, GPIO_OUTPUT);
 	gpio_pin_set_dt(&led0, 0);
@@ -97,7 +109,9 @@ SYS_INIT(led_pin_init, APPLICATION, CONFIG_APPLICATION_INIT_PRIORITY);
 static void led_pin_reset(void)
 {
 	LOG_DBG("led_pin_reset");
+#if LED_EXISTS
 	gpio_pin_configure_dt(&led, GPIO_DISCONNECTED);
+#endif
 #if LED0_EXISTS
 	gpio_pin_configure_dt(&led0, GPIO_DISCONNECTED);
 #endif
@@ -124,11 +138,13 @@ static void led_resume(void)
 	led_pin_init();
 }
 
+#if LED_STRIP_EXISTS
+#define LED_RGB_COLOR
+#else
 #ifdef CONFIG_LED_RGB_COLOR
 #define LED_RGB_COLOR
 #define LED_RG_COLOR
 #endif
-
 #if PWM_LED_EXISTS && PWM_LED1_EXISTS && PWM_LED2_EXISTS
 #define LED_TRI_COLOR
 #else
@@ -141,41 +157,42 @@ static void led_resume(void)
 #undef LED_DUAL_COLOR
 #endif
 #endif
+#endif
 
 #ifdef LED_RGB_COLOR
-static int led_pwm_period[5][3] = {
+static const int led_pwm_period[5][3] = {
 	{CONFIG_LED_DEFAULT_COLOR_R, CONFIG_LED_DEFAULT_COLOR_G, CONFIG_LED_DEFAULT_COLOR_B}, // Default
 	{0, 10000, 0}, // Success
 	{10000, 0, 0}, // Error
 	{8000, 2000, 0}, // Charging
 	{0, 0, 10000}, // Pairing
 };
 #elif defined(LED_TRI_COLOR)
-static int led_pwm_period[5][3] = {
+static const int led_pwm_period[5][3] = {
 	{0, 0, 10000}, // Default
 	{0, 10000, 0}, // Success
 	{10000, 0, 0}, // Error
 	{6000, 4000, 0}, // Charging
 	{0, 0, 10000}, // Pairing
 };
 #elif defined(LED_RG_COLOR)
-static int led_pwm_period[5][2] = {
+static const int led_pwm_period[5][2] = {
 	{CONFIG_LED_DEFAULT_COLOR_R, CONFIG_LED_DEFAULT_COLOR_G}, // Default
 	{0, 10000}, // Success
 	{10000, 0}, // Error
 	{8000, 2000}, // Charging
 	{4000, 6000}, // Pairing
 };
 #elif defined(LED_DUAL_COLOR)
-static int led_pwm_period[5][2] = {
+static const int led_pwm_period[5][2] = {
 	{0, 10000}, // Default
 	{0, 10000}, // Success
 	{10000, 0}, // Error
 	{6000, 4000}, // Charging
 	{0, 10000}, // Pairing
 };
 #else
-static int led_pwm_period[5][1] = {
+static const int led_pwm_period[5][1] = {
 	{10000}, // Default
 	{10000}, // Success
 	{10000}, // Error
@@ -197,7 +214,14 @@ static void led_pin_set(enum sys_led_color color, int brightness_pptt, int value
 		value_pptt = 0;
 	else if (value_pptt > 10000)
 		value_pptt = 10000;
-#if PWM_LED_EXISTS
+#if LED_STRIP_EXISTS
+	static struct led_rgb pixel[1];
+	value_pptt = value_pptt * brightness_pptt / 10000;
+	pixel[0].r = 255 * (led_pwm_period[color][0] * value_pptt / 10000) / 10000;
+	pixel[0].g = 255 * (led_pwm_period[color][1] * value_pptt / 10000) / 10000;
+	pixel[0].b = 255 * (led_pwm_period[color][2] * value_pptt / 10000) / 10000;
+	led_strip_update_rgb(strip, pixel, 1);
+#elif PWM_LED_EXISTS
 	value_pptt = value_pptt * brightness_pptt / 10000;
 	// only supporting color if PWM is supported
 	pwm_set_pulse_dt(&pwm_led, pwm_led.period / 10000 * (led_pwm_period[color][0] * value_pptt / 10000));
@@ -217,7 +241,7 @@ void set_led(enum sys_led_pattern led_pattern, int priority)
 {
 	LOG_DBG("set_led: current_led_pattern %d, current_priority %d", current_led_pattern, current_priority);
 	LOG_DBG("set_led: pattern %d, priority %d", led_pattern, priority);
-#if LED_EXISTS
+#if LED_EXISTS || LED_STRIP_EXISTS
 	if (led_pattern <= SYS_LED_PATTERN_OFF && k_current_get() == led_thread_id)
 		led_patterns[current_priority] = led_pattern;
 	else
@@ -238,29 +262,36 @@ void set_led(enum sys_led_pattern led_pattern, int priority)
 	{
 		led_suspend();
 		k_thread_suspend(led_thread_id);
+		LOG_DBG("set_led: suspended led_thread_id");
 	}
 	else if (k_current_get() != led_thread_id) // do not suspend if called from thread
 	{
 		k_thread_suspend(led_thread_id);
+		LOG_DBG("set_led: suspended led_thread_id");
 		led_resume();
 		k_thread_resume(led_thread_id);
+		k_wakeup(led_thread_id);
+		LOG_DBG("set_led: resumed led_thread_id");
 	}
 	else
 	{
 		led_resume();
 		k_thread_resume(led_thread_id);
+		k_wakeup(led_thread_id);
+		LOG_DBG("set_led: resumed led_thread_id");
 	}
 #endif
 }
 
 static void led_thread(void)
 {
-#if !LED_EXISTS
+#if !LED_EXISTS && !LED_STRIP_EXISTS
 	LOG_WRN("LED GPIO does not exist");
 	return;
 #else
 	while (1)
 	{
+		LOG_DBG("led_thread: current_led_pattern %d", current_led_pattern);
 		switch (current_led_pattern)
 		{
 		case SYS_LED_PATTERN_ON:
@@ -373,6 +404,7 @@ static void led_thread(void)
 			break;
 
 		default:
+			LOG_DBG("led_thread: suspending led_thread_id");
 			k_thread_suspend(led_thread_id);
 		}
 	}