From 36a0a240f98bed24dae9c1d75d86a482f502018b Mon Sep 17 00:00:00 2001
From: cschwinne <dev.aircoookie@gmail.com>
Date: Tue, 1 Dec 2020 00:33:47 +0100
Subject: [PATCH] First progress in busmnger

---
 wled00/bus_manager.h | 135 +++++++++++++++++++++++++++++++++++++++++++
 wled00/const.h       |  10 ++++
 2 files changed, 145 insertions(+)
 create mode 100644 wled00/bus_manager.h

diff --git a/wled00/bus_manager.h b/wled00/bus_manager.h
new file mode 100644
index 000000000..feac1dda4
--- /dev/null
+++ b/wled00/bus_manager.h
@@ -0,0 +1,135 @@
+#ifndef BusManager_h
+#define BusManager_h
+
+/*
+ * Class for addressing various light types
+ */
+
+#include "wled.h"
+
+class BusManager {
+  public:
+  BusManager() {
+
+  };
+  
+  int add(uint8_t busType, uint8_t* pins, uint16_t len = 1) {
+    if (numBusses >= WLED_MAX_BUSSES) return -1;
+    if (IS_DIGITAL(busType)) {
+      busses[numBusses] = new BusDigital(busType, pins, len);
+    } else {
+      busses[numBusses] = new BusPwm(busType, pins);
+    }
+    numBusses++;
+    return numBusses -1;
+  }
+
+  void removeAll() {
+    for (uint8_t i = 0; i < numBusses; i++) delete busses[i];
+    numBusses = 0;
+  }
+  //void remove(uint8_t id);
+
+  void show() {
+    for (uint8_t i = 0; i < numBusses; i++) {
+      busses[i]->show();
+    }
+  }
+
+  void setPixelColor(uint16_t pix, uint32_t c) {
+    for (uint8_t i = 0; i < numBusses; i++) {
+      Bus* b = busses[i];
+      uint16_t bstart = b->getStart();
+      if (pix < bstart) continue;
+      busses[i]->setPixelColor(pix - bstart, c);
+    }
+  }
+
+
+  private:
+  uint8_t numBusses = 0;
+  Bus* busses[WLED_MAX_BUSSES];
+};
+
+//parent class of BusDigital and BusPwm
+class Bus {
+  public:
+  Bus(uint8_t type) {
+    _type = type;
+  };
+  
+  virtual void show() {}
+
+  virtual void setPixelColor(uint16_t pix, uint32_t c) {};
+
+  virtual void setBrightness(uint8_t b) {};
+
+  virtual uint32_t getPixelColor(uint16_t pix) { return 0; };
+
+  virtual ~Bus() { //throw the bus under the bus
+    
+  }
+
+  uint16_t getStart() {
+    return _start;
+  }
+
+  void setStart(uint16_t start) {
+    _start = start;
+  }
+
+  virtual uint8_t getColorOrder() {
+    return COL_ORDER_RGB;
+  }
+
+  virtual void setColorOrder() {}
+
+  uint8_t getType() {
+    return _type;
+  }
+
+  protected:
+  uint8_t _type = TYPE_NONE;
+  uint16_t _start;
+};
+
+class BusDigital : public Bus {
+  public:
+  BusDigital(uint8_t type, uint8_t* pins, uint16_t len) : Bus(type) {
+    if (!IS_DIGITAL(type)) return;
+    _pins[0] = pins[0];
+    if (IS_2PIN(type)) _pins[1] = pins[1];
+    _len = len;
+  };
+
+  uint8_t getColorOrder() {
+    return _colorOrder;
+  }
+
+  void setColorOrder(uint8_t colorOrder) {
+    if (colorOrder > 5) return;
+    _colorOrder = colorOrder;
+  }
+
+  private: 
+  uint8_t _colorOrder = COL_ORDER_GRB;
+  uint8_t _pins[2];
+  uint16_t _len;
+};
+
+class BusPwm : public Bus {
+  public:
+  BusPwm(uint8_t type, uint8_t* pins) : Bus(type) {
+    if (!IS_ANALOG(type)) return;
+    uint8_t numPins = NUM_PWM_PINS(type);
+    if (numPins == 0) numPins = 1;
+    for (uint8_t i = 0; i < numPins; i++) {
+      _pins[i] = pins[i];
+    }
+  };
+  private: 
+  uint8_t _pins[5];
+  uint8_t _data[5];
+};
+
+#endif
diff --git a/wled00/const.h b/wled00/const.h
index 421747a48..f69d3ecd6 100644
--- a/wled00/const.h
+++ b/wled00/const.h
@@ -13,6 +13,12 @@
 //increase if you need more
 #define WLED_MAX_USERMODS 4
 
+#ifdef ESP8266
+#define WLED_MAX_BUSSES 2
+#else
+#define WLED_MAX_BUSSES 8
+#endif
+
 //Usermod IDs
 #define USERMOD_ID_RESERVED       0            //Unused. Might indicate no usermod present
 #define USERMOD_ID_UNSPECIFIED    1            //Default value for a general user mod that does not specify a custom ID
@@ -103,6 +109,10 @@
 #define TYPE_P9813               53
 #define TYPE_TM1814              54
 
+#define IS_DIGITAL(t) (t & 0x10) //digital are 16-31 and 48-63
+#define IS_ANALOG(t)  (t > 39 && t < 46)
+#define NUM_PWM_PINS(t) (t - 40) //for analog PWM 40-45 only
+#define IS_2PIN(t)      (t > 47)
 
 //Color orders
 #define COL_ORDER_GRB             0           //GRB(w),defaut