From 0ea8c7c2ff805a914ff5dbc25f7d12183e690c3c Mon Sep 17 00:00:00 2001
From: PaulStoffregen <paul@pjrc.com>
Date: Wed, 17 Nov 2021 10:33:37 -0800
Subject: Fix OctoWS2811 driver on Teensy 4.x

---
 src/platforms/arm/mxrt1062/fastled_arm_mxrt1062.h  |  2 +-
 src/platforms/arm/mxrt1062/octows2811_controller.h | 64 ++++++++++++++++++++++
 2 files changed, 65 insertions(+), 1 deletion(-)
 create mode 100644 src/platforms/arm/mxrt1062/octows2811_controller.h

diff --git a/src/platforms/arm/mxrt1062/fastled_arm_mxrt1062.h b/src/platforms/arm/mxrt1062/fastled_arm_mxrt1062.h
index 5098af33..0cd53602 100644
--- a/src/platforms/arm/mxrt1062/fastled_arm_mxrt1062.h
+++ b/src/platforms/arm/mxrt1062/fastled_arm_mxrt1062.h
@@ -3,7 +3,7 @@
 
 #include "fastpin_arm_mxrt1062.h"
 #include "fastspi_arm_mxrt1062.h"
-#include "../k20/octows2811_controller.h"
+#include "octows2811_controller.h"
 #include "../k20/ws2812serial_controller.h"
 #include "../k20/smartmatrix_t3.h"
 #include "clockless_arm_mxrt1062.h"
diff --git a/src/platforms/arm/mxrt1062/octows2811_controller.h b/src/platforms/arm/mxrt1062/octows2811_controller.h
new file mode 100644
index 00000000..d0000a40
--- /dev/null
+++ b/src/platforms/arm/mxrt1062/octows2811_controller.h
@@ -0,0 +1,64 @@
+#ifndef __INC_OCTOWS2811_CONTROLLER_H
+#define __INC_OCTOWS2811_CONTROLLER_H
+
+#ifdef USE_OCTOWS2811
+
+// #include "OctoWS2811.h"
+
+FASTLED_NAMESPACE_BEGIN
+
+template<EOrder RGB_ORDER = GRB, uint8_t CHIP = WS2811_800kHz>
+class COctoWS2811Controller : public CPixelLEDController<RGB_ORDER, 8, 0xFF> {
+  OctoWS2811  *pocto;
+  uint8_t *drawbuffer,*framebuffer;
+
+  void _init(int nLeds) {
+    if(pocto == NULL) {
+      drawbuffer = (uint8_t*)malloc(nLeds * 8 * 3);
+      framebuffer = (uint8_t*)malloc(nLeds * 8 * 3);
+
+      // byte ordering is handled in show by the pixel controller
+      int config = WS2811_RGB;
+      config |= CHIP;
+
+      pocto = new OctoWS2811(nLeds, framebuffer, drawbuffer, config);
+
+      pocto->begin();
+    }
+  }
+public:
+  COctoWS2811Controller() { pocto = NULL; }
+  virtual int size() { return CLEDController::size() * 8; }
+
+  virtual void init() { /* do nothing yet */ }
+
+  virtual void showPixels(PixelController<RGB_ORDER, 8, 0xFF> &pixels) {
+    uint32_t size = pixels.size();
+    uint32_t sizeTimes8 = 8U * size;
+    _init(size);
+
+    uint32_t index = 0;
+    while (pixels.has(1)) {
+      for (uint32_t i = 0; i < 8; i++) {
+        uint8_t r = pixels.loadAndScale0(i);
+        uint8_t g = pixels.loadAndScale1(i);
+        uint8_t b = pixels.loadAndScale2(i);
+        pocto->setPixel(index, r, g, b);
+        index += size;
+      }
+      index -= sizeTimes8;
+      index++;
+      pixels.stepDithering();
+      pixels.advanceData();
+    }
+
+    pocto->show();
+  }
+
+};
+
+FASTLED_NAMESPACE_END
+
+#endif
+
+#endif
-- 
cgit v1.2.3