README.md

@@ -1,41 +0,0 @@
-# ESPleaf
-
-## Layout description
-
-Layout is defined in `LAYOUT` in `config.h`.
-
-LEDs within a node are supposed to be installed counter-clockwise. The edge where the cables enter the node is node 0. To its right (counter-clockwise, remember) is corner 0. Then edge 1, corner 1, edge 2, corner 2 before coming back to edge 0.
-
-Thereby, the next panel can either be installed along edge 1 or edge 2. (But the cable has to go corner 0, corner 1, corner 2 first.)
-
-First panel is ignored in `LAYOUT`. First entry in `LAYOUT` is the edge through which the cable leaves Node 0 to Node 1 - which can be either Edge 1 or Edge 2.
-
-On the next panel, the edge the cable is coming through is called Edge 0 again. And so on.
-
-So, `LAYOUT` countains a list of edges the cable takes to leave to the next node.
-
-## Commands
-
-Command can be sent via MQTT to `MQTT_TOPIC_COMMANDS`. Syntax is `key=value` or `key=value&key2=value2...`.
-
-Valid keys are:
-* `mode` sets a new mode. Valid modes are:
-  * `corners`
-  * `nodes`
-  * `first_corner`
-  * `first_node`
-  * `flash`
-  * `static`
-  * `off`
-* `brightness` sets the overall brightness of the lamp. Values are from 0 to 255 inclusive.
-* `duration` if sets, `mode` and `brightness` are set for this amount of seconds only.
-* `color` sets the color for `static` and `flash` modes. A predefined color name is expected.
-* `speedup` sets a speedup for faster animations. Default value 1, possible values are 1 to 255. 0 will lead to an exception.
-
-## Startup sequence
-
-During startup:
-* 1 green corner: FastLED is initialized. Layout is now being analyzed.
-* 2 green corners: Layout is done. WiFi is being connected.
-* 3 green corners: WiFi connection established. Connecting to MQTT server.
-* Everything green (quarter of a second): Initialization done.

include/config.sample.h

@@ -21,15 +21,10 @@
 #define SPEEDUP 1
 
 #define MAX_MILLIAMPS 1000
-// Maximum color difference for the random effects.
-// This changes the hue value +/- this value. Use a maximum value of 127, otherwise strange things might happen.
-#define COLOR_DIFFERENCE 25
 
 #define WIFI_SSID "..."
 #define WIFI_PASS "..."
 
-#define OTA_HOSTNAME "..."
-
 #define MQTT_CLIENT_ID "espleaf"
 #define MQTT_USER "..."
 #define MQTT_PASS "..."

include/node.h

@@ -3,28 +3,16 @@
 #include "config.h"
 #include "edge.h"
 #include "corner.h"
-#include "prototypes.h"
-
-// Delta-X data. Per direction of a triangle is a triple with Delta-X values for each direction.
-static const int8_t dx[][3] = {{0, 1, -1}, {1, 0, -1}, {1, -1, 0}, {0, -1, 1}, {-1, 0, 1}, {-1, 1, 0}};
-static const int8_t dy[][3] = {{-1, 0, 0}, {0, 1, 0}, {0, 0, -1}, {1, 0, 0}, {0, -1, 0}, {0, 0, 1}};
-
-typedef struct {
-	int8_t x;
-	int8_t y;
-} Coords;
 
 class Node {
-	public:
+	private:
 		uint16_t _number;
-		Coords coords;
-		uint8_t direction;
+	public:
 		Node* neighbours[CORNERS_PER_PART];
 		Edge* edges[CORNERS_PER_PART];
 		Corner* _corners[CORNERS_PER_PART];
-		Node(uint16_t number, Coords c, uint8_t direction);
+		Node(uint16_t number);
 		Node* create_neighbour(uint8_t edge);
-		Coords coords_at_direction(uint8_t edge);
 
 		void blend_to(CRGB color, uint16_t effect_id=0, uint8_t effect_speed=0);
 		void set_color(CRGB color);

include/prototypes.h

@@ -7,7 +7,6 @@
 #include "edge.h"
 #include "corner.h"
 
-class Node;
 
 extern std::vector<Node*> nodes;
 extern std::list<Edge*> edges;
@@ -24,12 +23,10 @@ enum AnimationMode {
 	AM_FLASH,
 	AM_OFF,
 	AM_ERROR,
-	AM_NONE,
-	AM_STATIC
+	AM_NONE
 };
 
 extern AnimationMode mode;
 extern AnimationMode temp_mode;
 extern unsigned long temp_mode_until;
 extern int return_to_brightness;
-extern uint8_t speedup;

platformio.ini

@@ -12,12 +12,10 @@
 platform = espressif8266
 board = d1_mini
 framework = arduino
-;upload_port = /dev/cu.wchusbserial*
-upload_protocol = espota
-upload_port = espleaf-prod.local
+upload_port = /dev/cu.wchusbserial*
 upload_speed = 921600
 monitor_port = /dev/cu.wchusbserial*
 monitor_speed = 74880
 monitor_filters = default, time, send_on_enter, esp8266_exception_decoder
 lib_deps = fastled/FastLED @ 3.4.0
-    PubSubClient @ 2.8
+    PubSubClient

src/corner.cpp

@@ -44,13 +44,13 @@ bool Corner::reached_level(uint16_t level) {
 }
 
 void Corner::blend_to(CRGB target, uint16_t eid, uint8_t effect_speed) {
-	//LOGln("blendTo called. Corner: %p, target: %d,%d,%d, eid: %d, 'old' effect_id: %d, speed: %d", this, target.r, target.g, target.b, eid, effect_id, effect_speed);
+	LOGln("blendTo called. Corner: %p, target: %d,%d,%d, eid: %d, 'old' effect_id: %d, speed: %d", this, target.r, target.g, target.b, eid, effect_id, effect_speed);
 	if (eid==0) {
 		this->effect_id = random16();
-		//LOGln("Set effect_id to %d", this->effect_id);
+		LOGln("Set effect_id to %d", this->effect_id);
 	} else {
 		if (this->effect_id == eid) {
-			//LOGln("'Old' effect. Doing nothing.");
+			LOGln("'Old' effect. Doing nothing.");
 			return;
 		}
 		this->effect_id = eid;

src/main.cpp

@@ -1,6 +1,5 @@
 #include <Arduino.h>
 #include <ESP8266WiFi.h>
-#include <ArduinoOTA.h>
 #include "my_fastled.h"
 #include "config.h"
 #include "tools.h"
@@ -21,7 +20,6 @@ AnimationMode mode = AM_CORNERS;
 AnimationMode temp_mode;
 unsigned long temp_mode_until;
 int return_to_brightness = -1;
-uint8_t speedup = SPEEDUP;
 
 unsigned long last_loop = 0;
 
@@ -34,7 +32,7 @@ bool looping;
 void setup_layout() {
 	LOGln("Setting up layout...");
 	uint8_t layout[] = LAYOUT;
-	Node* current_node = new Node(0, {0, 0}, 0);
+	Node* current_node = new Node(0);
 	nodes.push_back(current_node);
 
 	for(uint16_t i=0; i<NODE_COUNT-1; i++) {
@@ -55,45 +53,6 @@ void setup_layout() {
 		}
 	}
 
-	for (Node* n1 : nodes) {
-		LOGln("Looking for neighbours of node #%d @ %d,%d", n1->_number, n1->coords.x, n1->coords.y);
-		for(int edge=0; edge<CORNERS_PER_PART; edge++) {
-			Coords c = n1->coords_at_direction(edge);
-			LOGln("  Chcking edge %d @ %d,%d...", edge, c.x, c.y);
-
-			Node* found = nullptr;
-			for(Node* n2 : nodes) {
-				if (n2 == n1) {
-					continue;
-				}
-				if (n2->coords.x == c.x && n2->coords.y == c.y) {
-					found = n2;
-					break;
-				}
-			}
-			if (found != nullptr) {
-				LOGln("    Found node #%d", found->_number);
-				uint8_t inverse_dir = (n1->direction + 2*edge + 3) % 6;
-				int8_t e = (inverse_dir - found->direction) % 6;
-				if (e < 0) e+=6;
-				e = e / 2;
-				//LOGln("      inverse_dir: %d, edge: %d", inverse_dir, edge);
-				int8_t e1 = (edge - 1) % CORNERS_PER_PART;
-				if (e1<0) e1+=CORNERS_PER_PART;
-				int8_t e2 = (e - 1) % CORNERS_PER_PART;
-				if (e2<0) e2+=CORNERS_PER_PART;
-
-				LOGln("      Mapping Corner #%d,%d with #%d,%d", n1->_number, edge, found->_number, e2);
-				n1->_corners[edge]->_short_neighbours.push_back(found->_corners[e2]);
-				
-				LOGln("      Mapping Corner #%d,%d with #%d,%d", n1->_number, e1, found->_number, e);
-				n1->_corners[e1]->_short_neighbours.push_back(found->_corners[e]);
-			} else {
-				LOGln("    No match.");
-			}
-		}
-	}
-
 	for(Corner* corner: corners) {
 		LOGln("Corner %p:", corner);
 		for(auto c: corner->_long_neighbours) {
@@ -117,61 +76,41 @@ void setup_fastled() {
 	FastLED.addLeds<WS2812B, 5, GRB>(leds, LEDS_PER_CORNER * CORNERS_PER_PART * NODE_COUNT).setCorrection(TypicalLEDStrip);
 	LOGln("LEDs:    %3d", LED_COUNT);
 	FastLED.setBrightness(255);
-	//FastLED.setDither(DISABLE_DITHER);
+	FastLED.setDither(DISABLE_DITHER);
 	FastLED.setMaxPowerInVoltsAndMilliamps(5, MAX_MILLIAMPS);
 	set_all_leds(CRGB::Black);
 }
 
-void show_all() {
-	for(Node* node : nodes) {
-		node->draw();
+void setup_rng() {
+	LOGln("Starting WiFi scan for RNG initialization...");
+	WiFi.mode(WIFI_STA);
+	WiFi.disconnect();
+	uint16_t seed = 0;
+	int n = WiFi.scanNetworks();
+	LOGln("%d WiFi networks found", n);
+	for(int i=0; i<n; i++) {
+		LOGln("  %s, %d dB", WiFi.SSID(i).c_str(), WiFi.RSSI(i));
+		seed = (seed << 2) | (WiFi.RSSI(i) & 0x03);
 	}
-	FastLED.show();
-}
-
-void show_status(uint8_t status, CRGB color=CRGB::Green) {
-	for (int i=0; i<status; i++) {
-		if (i<corners.size()) {
-			corners[i]->set_color(color);
-		}
-	}
-	for (int i=status; i<corners.size(); i++) {
-		corners[i]->set_color(CRGB::Black);
-	}
-	show_all();
+	LOGln("WiFi scan done. Generated seed is 0x%04x", seed);
+	random16_set_seed(seed);
 }
 
 void setup() {
 	Serial.begin(74880);
 	LOGln("ESPleaf starting.");
 
-	
-	setup_fastled();
-	show_status(1);
-
 	setup_layout();
-	show_status(2);
-
+	setup_fastled();
 	#ifdef TEST_MODE
 		LOGln("TEST_MODE is active!");
 	#else
+		setup_rng();
+
 		wifi_setup();
-
-		show_status(3);
-
 		mqtt_setup();
-		ArduinoOTA.setHostname(OTA_HOSTNAME);
-		ArduinoOTA.onProgress([&](unsigned int progress, unsigned int total){
-			uint8_t count = progress * corners.size() / total;
-			show_status(count, CRGB::Blue);
-		});
-		ArduinoOTA.onEnd([](){ show_status(0); });
-		ArduinoOTA.begin();
-
-		show_status(255);
-		delay(250);
-		show_status(0);
 	#endif
+	
 }
 
 void loop() {
@@ -190,9 +129,7 @@ void loop() {
 	#else
 		// Normal mode
 		mqtt_loop();
-		ArduinoOTA.handle();
-
-		if (speedup > 0 && (millis() - last_loop > (20 / speedup) || last_loop > millis())) {
+		EVERY_N_MILLISECONDS(20 / SPEEDUP) {
 			looping = false;
 
 			AnimationMode active_mode = mode;
@@ -217,37 +154,44 @@ void loop() {
 						corner->infect(300, 600);
 					}
 					looping |= !corner->is_finished();
+					corner->draw();
 				}
 
 				if (random8(128)==0) {
-					uint16_t corner = (active_mode == AM_FIRST_CORNER) ? 0 : random16(corners.size());
-					CHSV color = rgb2hsv_approximate(corners[corner]->color);
-					corners[corner]->blend_to(CHSV(color.h - COLOR_DIFFERENCE + random8(2*COLOR_DIFFERENCE), 255, 255));
+					if (active_mode == AM_FIRST_CORNER) {
+						corners[0]->blend_to(CHSV(random8(), 255, 255));
+					} else {
+						corners[random16(corners.size())]->blend_to(CHSV(random8(), 255, 255));
+					}
 				}
 			} else if (active_mode == AM_NODES || active_mode == AM_FIRST_NODE) {
 				for(Node* node : nodes) {
 					node->step();
 					node->infect(512);
+					node->draw();
 				}
 
 				if (random8(128)==0) {
-					uint16_t corner = (active_mode == AM_FIRST_NODE) ? 0 : random8(nodes.size());
-					CHSV color = rgb2hsv_approximate(corners[corner]->color);
-					nodes[corner]->blend_to(CHSV(color.h - COLOR_DIFFERENCE + random8(2*COLOR_DIFFERENCE), 255, 255));
+					if (active_mode == AM_FIRST_NODE) {
+						nodes[0]->blend_to(CHSV(random8(), 255, 255));
+					} else {
+						nodes[random8(nodes.size())]->blend_to(CHSV(random8(), 255, 255));
+					}
 				}
 			} else if (active_mode == AM_FLASH) {
 				for (Node* node : nodes) {
 					node->step();
 					node->infect(512);
+					node->draw();
 				}
 
 				if (millis() / 1000 > last_loop / 1000) {
 					nodes[0]->blend_to(((millis() / 1000) % 2 == 0) ? CRGB::Black : color, 0, 64);
 				}
 
-			} else if (active_mode == AM_OFF || active_mode == AM_STATIC) {
+			} else if (active_mode == AM_OFF) {
 				for(Node* node : nodes) {
-					node->set_color(active_mode == AM_OFF ? CRGB::Black : color);
+					node->set_color(CRGB::Black);
 				}
 			} else { // This includes AM_ERROR
 				for(Node* node : nodes) {
@@ -260,5 +204,5 @@ void loop() {
 		}
 	#endif
 
-	show_all();
+	FastLED.show();
 }

src/mqtt.cpp

@@ -24,24 +24,16 @@ void callback(char* topic, byte* pl, unsigned int length) {
     uint16_t duration = 0;
     AnimationMode new_mode = AM_NONE;
     String current_part;
-    LOGln("Received command %s", payload.c_str());
     while (payload.length() > 0) {
         int offset = payload.indexOf("&");
         if (offset != -1) {
             current_part = payload.substring(0, offset);
             payload = payload.substring(offset + 1);
-        } else {
-            current_part = payload;
-            payload = "";
         }
         offset = current_part.indexOf("=");
-        if (offset==-1) {
-            LOGln("Parameter without '=' detected: %s", current_part.c_str());
-            continue;
-        }
+        if (offset==-1) continue;
         String key = current_part.substring(0, offset);
         String value = current_part.substring(offset+1);
-        LOGln("  Processing key %s with value %s", key.c_str(), value.c_str());
 
         if (key.equals("mode")) {
             if (value.equals("nodes")) { new_mode = AM_NODES; }
@@ -50,7 +42,6 @@ void callback(char* topic, byte* pl, unsigned int length) {
             else if (value.equals("first_corner")) { new_mode = AM_FIRST_CORNER; }
             else if (value.equals("off")) { new_mode = AM_OFF; }
             else if (value.equals("flash")) { new_mode = AM_FLASH; }
-            else if (value.equals("static")) { new_mode = AM_STATIC; }
             else { LOGln("Unknown mode '%s'.", value.c_str()); }
         } else if (key.equals("duration")) {
             duration = value.toInt();
@@ -63,17 +54,11 @@ void callback(char* topic, byte* pl, unsigned int length) {
             if (value.equals("red")) { color = CRGB::Red; }
             else if (value.equals("green")) { color = CRGB::Green; }
             else if (value.equals("blue")) { color = CRGB::Blue; }
-            else if (value.equals("pink")) { color = CRGB::Pink; }
-            else if (value.equals("yellow")) { color = CRGB::Yellow; }
-            else if (value.equals("orange")) { color = CRGB::Orange; }
             else { LOGln("Unknown color name %s.", value.c_str());}
-        } else if (key.equals("speedup")) {
-            speedup = value.toInt();
         } else {
             LOGln("Unknown key '%s'. (For reference: Value is '%s'.)", key.c_str(), value.c_str());
         }
     }
-    LOGln("Finished processing the command.");
 
     if (new_mode != AM_NONE) {
         if (duration > 0) {

src/node.cpp

@@ -2,12 +2,8 @@
 #include "prototypes.h"
 #include "tools.h"
 
-Node::Node(uint16_t number, Coords c, uint8_t _dir) {
+Node::Node(uint16_t number) {
 	_number = number;
-	coords = c;
-	direction = _dir;
-
-	LOGln("Created Node #%d at coordinates %d,%d with direction %d.", _number, coords.x, coords.y, direction);
 	for(int i=0; i<CORNERS_PER_PART; i++) {
 		edges[i] = new Edge();
 	}
@@ -30,25 +26,38 @@ Node::Node(uint16_t number, Coords c, uint8_t _dir) {
 	last_corner->_long_neighbours.push_back(_corners[0]);
 }
 
-Coords Node::coords_at_direction(uint8_t edge) {
-	return {(int8_t)(coords.x + dx[direction][edge]), (int8_t)(coords.y + dy[direction][edge])};
-}
-
 Node* Node::create_neighbour(uint8_t edge) {
-	Coords new_c = coords_at_direction(edge);
-	int8_t new_dir = (edge==1) ? ((direction - 1) % 6) : ((direction + 1) % 6);
-	if (new_dir < 0) new_dir+=6;
-	Node* node = new Node(_number + 1, new_c, new_dir);
+	Node* node = new Node(_number + 1);
 	node->neighbours[0] = this;
 	neighbours[edge] = node;
 
 	node->edges[0]->neighbour = this->edges[edge];
 	this->edges[edge]->neighbour = node->edges[0];
 
+	node->_corners[0]->_short_neighbours.push_back(_corners[(edge-1) % CORNERS_PER_PART]);
+	node->_corners[CORNERS_PER_PART - 1]->_short_neighbours.push_back(_corners[edge]);
+	_corners[(edge-1) % CORNERS_PER_PART]->_short_neighbours.push_back(node->_corners[0]);
+	_corners[edge]->_short_neighbours.push_back(node->_corners[CORNERS_PER_PART - 1]);
+
+	/*
+	delete node->edges[0];
+	node->edges[0] = this->edges[edge];
+	
+	Corner* c = this->corners[(edge-1) % CORNERS_PER_PART];
+	c->merge_leds(node->corners[0]);
+	delete node->corners[0];
+	node->corners[0] = c;
+
+	c = this->corners[edge];
+	c->merge_leds(node->corners[CORNERS_PER_PART-1]);
+	delete node->corners[CORNERS_PER_PART-1];
+	node->corners[CORNERS_PER_PART-1] = c;
+	*/
 	return node;
 }
 
 void Node::blend_to(CRGB color, uint16_t effect_id, uint8_t effect_speed) {
+	LOGln("Node::blend_to called. this:%p", this);
 	if (effect_speed == 0) effect_speed = random8(2)+1;
 	if (effect_id == 0) effect_id = random16();
 	for(Corner* corner : this->_corners) {

src/wifi.cpp

@@ -10,5 +10,4 @@ void wifi_setup() {
         delay(300);
     }
     LOGln(" Connected as %s", WiFi.localIP().toString().c_str());
-    random16_add_entropy(micros());
 }