[RPi] a test of driver for WS281X RGB LED (NeoPixel) ring in use by Standalone Edition [skip ci]

Author: lyusupov

software/firmware/source/SoftRF/Makefile

@@ -32,6 +32,7 @@ ifeq ($(LGPIO), yes)
 endif
 
 BRIDGE        ?= no
+NEOPIXEL      ?= no
 
 CC            = gcc
 CXX           = g++
@@ -108,6 +109,7 @@ LGPIO_PATH    = $(LIB_PATH)/lg-0.1
 RADIOLIB_PATH = $(LIB_PATH)/RadioLib/src
 BRIDGE_PATH   = $(LIB_PATH)/Bridge/src
 BMP280_PATH   = $(LIB_PATH)/Adafruit_BMP280_Library
+WS281X_PATH   = $(LIB_PATH)/rpi_ws281x-1.0.0
 
 ifeq ($(BASICMAC), yes)
 RADIO_PATH    = $(BASICMAC_PATH)
@@ -124,7 +126,7 @@ INCLUDE       = -I$(PRODAT_PATH) -I$(PRORAD_PATH)  -I$(PLATFORM_PATH) \
                 -I$(GFX_PATH)    -I$(U8G2_AF_PATH) -I$(EPD2_PATH)     \
                 -I$(MODES_PATH)  -I$(APRS_PATH)    -I$(RADIOLIB_PATH) \
                 -I$(LGPIO_PATH)  -I$(BRIDGE_PATH)  -I$(BMP280_PATH)   \
-                -I$(U8G2_PATH)
+                -I$(U8G2_PATH)   -I$(WS281X_PATH)
 
 SRC_CPPS      := $(SRC_PATH)/TrafficHelper.cpp \
                  $(SRC_PATH)/Library.cpp
@@ -284,6 +286,13 @@ ifeq ($(BRIDGE), yes)
   CFLAGS      += -DUSE_BRIDGE
 endif
 
+ifeq ($(NEOPIXEL), yes)
+  OBJS        += $(WS281X_PATH)/mailbox.o $(WS281X_PATH)/ws2811.o \
+                 $(WS281X_PATH)/pwm.o $(WS281X_PATH)/pcm.o \
+                 $(WS281X_PATH)/dma.o $(WS281X_PATH)/rpihw.o
+  CFLAGS      += -DUSE_NEOPIXEL
+endif
+
 ifeq ($(NOMAVLINK), no)
   OBJS        += $(MAVLINK_PATH)/mavlink.o
 else

software/firmware/source/SoftRF/src/platform/RPi.cpp

@@ -360,6 +360,187 @@ void notFound(Request &req, Response &res) {
 }
 #endif /* USE_BRIDGE */
 
+#if defined(USE_NEOPIXEL)
+#include "clk.h"
+#include "gpio.h"
+#include "dma.h"
+#include "pwm.h"
+
+#include "ws2811.h"
+
+#define ARRAY_SIZE(stuff)       (sizeof(stuff) / sizeof(stuff[0]))
+
+#define TARGET_FREQ             WS2811_TARGET_FREQ
+#define GPIO_PIN                18
+#define DMA                     10
+//#define STRIP_TYPE            WS2811_STRIP_RGB		// WS2812/SK6812RGB integrated chip+leds
+#define STRIP_TYPE              WS2811_STRIP_GBR		// WS2812/SK6812RGB integrated chip+leds
+//#define STRIP_TYPE            SK6812_STRIP_RGBW		// SK6812RGBW (NOT SK6812RGB)
+
+#define WIDTH                   8
+#define HEIGHT                  8
+#define LED_COUNT               (WIDTH * HEIGHT)
+
+static int width = WIDTH;
+static int height = HEIGHT;
+static int led_count = LED_COUNT;
+
+ws2811_t ledstring =
+{
+    .freq = TARGET_FREQ,
+    .dmanum = DMA,
+    .channel =
+    {
+        [0] =
+        {
+            .gpionum = GPIO_PIN,
+            .invert = 0,
+            .count = LED_COUNT,
+            .strip_type = STRIP_TYPE,
+            .brightness = 255,
+        },
+        [1] =
+        {
+            .gpionum = 0,
+            .invert = 0,
+            .count = 0,
+            .brightness = 0,
+        },
+    },
+};
+
+ws2811_led_t *matrix;
+
+void matrix_render(void)
+{
+    int x, y;
+
+    for (x = 0; x < width; x++)
+    {
+        for (y = 0; y < height; y++)
+        {
+            ledstring.channel[0].leds[(y * width) + x] = matrix[y * width + x];
+        }
+    }
+}
+
+void matrix_raise(void)
+{
+    int x, y;
+
+    for (y = 0; y < (height - 1); y++)
+    {
+        for (x = 0; x < width; x++)
+        {
+            // This is for the 8x8 Pimoroni Unicorn-HAT where the LEDS in subsequent
+            // rows are arranged in opposite directions
+            matrix[y * width + x] = matrix[(y + 1)*width + width - x - 1];
+        }
+    }
+}
+
+void matrix_clear(void)
+{
+    int x, y;
+
+    for (y = 0; y < (height ); y++)
+    {
+        for (x = 0; x < width; x++)
+        {
+            matrix[y * width + x] = 0;
+        }
+    }
+}
+
+int dotspos[] = { 0, 1, 2, 3, 4, 5, 6, 7 };
+
+ws2811_led_t dotcolors[] =
+{
+    0x00200000,  // red
+    0x00201000,  // orange
+    0x00202000,  // yellow
+    0x00002000,  // green
+    0x00002020,  // lightblue
+    0x00000020,  // blue
+    0x00100010,  // purple
+    0x00200010,  // pink
+};
+
+ws2811_led_t dotcolors_rgbw[] =
+{
+    0x00200000,  // red
+    0x10200000,  // red + W
+    0x00002000,  // green
+    0x10002000,  // green + W
+    0x00000020,  // blue
+    0x10000020,  // blue + W
+    0x00101010,  // white
+    0x10101010,  // white + W
+
+};
+
+void matrix_bottom(void)
+{
+    int i;
+
+    for (i = 0; i < (int)(ARRAY_SIZE(dotspos)); i++)
+    {
+        dotspos[i]++;
+        if (dotspos[i] > (width - 1))
+        {
+            dotspos[i] = 0;
+        }
+
+        if (ledstring.channel[0].strip_type == SK6812_STRIP_RGBW) {
+            matrix[dotspos[i] + (height - 1) * width] = dotcolors_rgbw[i];
+        } else {
+            matrix[dotspos[i] + (height - 1) * width] = dotcolors[i];
+        }
+    }
+}
+
+int ws2811_init(void)
+{
+    ws2811_return_t ret;
+
+    matrix = (ws2811_led_t *) malloc(sizeof(ws2811_led_t) * width * height);
+
+    if ((ret = ws2811_init(&ledstring)) != WS2811_SUCCESS)
+    {
+        fprintf(stderr, "ws2811_init failed: %s\n", ws2811_get_return_t_str(ret));
+        return ret;
+    }
+
+    return ret;
+}
+
+void ws2811_test(void)
+{
+    ws2811_return_t ret;
+
+    for (int i=0; i < 75; i++)
+    {
+        matrix_raise();
+        matrix_bottom();
+        matrix_render();
+
+        if ((ret = ws2811_render(&ledstring)) != WS2811_SUCCESS)
+        {
+            fprintf(stderr, "ws2811_render failed: %s\n", ws2811_get_return_t_str(ret));
+            break;
+        }
+
+        // 15 frames /sec
+        usleep(1000000 / 15);
+    }
+
+	matrix_clear();
+	matrix_render();
+	ws2811_render(&ledstring);
+}
+
+#endif /* USE_NEOPIXEL */
+
 //-------------------------------------------------------------------------
 //
 // The MIT License (MIT)
@@ -1446,6 +1627,13 @@ int main()
 //  hw_info.gnss = GNSS_setup();
 
   Traffic_setup();
+
+#if defined(USE_NEOPIXEL)
+  // LED_setup();
+
+  ws2811_init();
+#endif /* USE_NEOPIXEL */
+
   NMEA_setup();
 
   Traffic_TCP_Server.setup(JSON_SRV_TCP_PORT);
@@ -1486,6 +1674,13 @@ int main()
   Serial.println((unsigned long) RELAY_SRC_PORT);
 #endif /* USE_BRIDGE */
 
+#if defined(USE_NEOPIXEL)
+  // LED_test();
+
+  ws2811_test();
+  ws2811_fini(&ledstring);
+#endif /* USE_NEOPIXEL */
+
   Sound_setup();
   SoC->Sound_test(reset_info.reason);