rgbled/application.cpp

/*
 * productionMode.cpp
 *
 *  Created on: Apr 05, 2019
 *      Author: wn
 */



#include "defines.h"


#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <stdbool.h>
#include <mqttHandling.h>


#ifdef WS2811
#define FASTLED_ESP8266_NODEMCU_PIN_ORDER
#include <FastLED.h>
#endif

#ifdef PL9823
#include <Adafruit_NeoPixel.h>
#endif


#include "configuration.h"


#ifdef WS2811
CRGB leds[NUM_OF_LEDs];
#endif

#ifdef PL9823
typedef struct {
	uint8_t r;
	uint8_t g;
	uint8_t b;
} CRGB;


Adafruit_NeoPixel pixels(NUM_OF_LEDs, PIXEL_PIN, NEO_RGB + NEO_KHZ400);
#endif

bool show = false;

typedef enum {
	E_FC_IDLE,
	E_FC_ABORT,
	E_FC_FADE_ON,
	E_FC_FADE_OFF,
} e_colorFlowCmds;

e_colorFlowCmds colorFlowCmd = E_FC_IDLE;

typedef enum {
	E_FE_IDLE,
	E_FE_INCREASE_BRIGHTNESS,
	E_FE_DECREASE_BRIGHTNESS,
} e_colorFlowStates;



static void colorFlowEngine() {
	e_colorFlowStates colorFlowState = E_FE_IDLE;

	if (colorFlowCmd == E_FC_ABORT) {
		colorFlowState = E_FE_IDLE;
		colorFlowCmd = E_FC_IDLE;
	}

	switch (colorFlowState) {
	case E_FE_IDLE:
		switch (colorFlowCmd) {
		case E_FC_FADE_ON:
			colorFlowState = E_FE_INCREASE_BRIGHTNESS;
			break;
		case E_FC_FADE_OFF:
			colorFlowState = E_FE_DECREASE_BRIGHTNESS;
			break;
		case E_FC_IDLE:
		case E_FC_ABORT:
		default:
			break;
		}
		colorFlowCmd = E_FC_IDLE;
		break;
	case E_FE_INCREASE_BRIGHTNESS:
		// fade up
		show = true;
		break;
	case E_FE_DECREASE_BRIGHTNESS:
		// fade down
		show = true;
		break;
	}
}



static CRGB evaluationColorWord(char* cmd) {
	uint8_t red = 0, green = 0, blue = 0;
	if ((! strcmp(cmd, "on")) || (! strcmp(cmd, "white"))) {
		red = 255;
		green = 255;
		blue = 255;
	} else if (! strcmp(cmd, "off")) {
		red = 0;
		green = 0;
		blue = 0;
	} else if (! strcmp(cmd, "warmwhite")) {
		red = 0;
		green = 0;
		blue = 0;
	} else if (! strcmp(cmd, "red")) {
		red = 255;
		green = 0;
		blue = 0;
	} else if (! strcmp(cmd, "green")) {
		red = 0;
		green = 255;
		blue = 0;
	} else if (! strcmp(cmd, "blue")) {
		red = 0;
		green = 0;
		blue = 255;
	} else if (! strcmp(cmd, "purple")) {
		red = 255;
		green = 0;
		blue = 255;
	} else if (! strcmp(cmd, "yellow")) {
		red = 255;
		green = 255;
		blue = 0;
	}

	CRGB res;
	res.r = red;
	res.g = green;
	res.b = blue;
	return res;
}



void subscribeApplication() {
	mqttClient.subscribe(configBlock.mqttTopicColorCommand);
	mqttClient.subscribe(configBlock.mqttTopicCommand);
	mqttClient.subscribe(configBlock.mqttTopicFlowCommand);
}



static void setColor(int8_t ledNumber, uint8_t red, uint8_t green, uint8_t blue) {
	if (ledNumber == -1) {
		for (uint8_t i = 0; i < NUM_OF_LEDs; i++) {
#ifdef WS2811
			leds[i].r = red;
			leds[i].g = green;
			leds[i].b = blue;
#endif
#ifdef PL9823
			pixels.setPixelColor(i, pixels.Color(red, green, blue));
#endif
		}
		show = true;
	} else {
		if (ledNumber >= 0 && ledNumber < NUM_OF_LEDs) {

#ifdef WS2811
			leds[ledNumber].r = red;
			leds[ledNumber].g = green;
			leds[ledNumber].b = blue;
#endif
#ifdef PL9823
			pixels.setPixelColor(ledNumber, pixels.Color(red, green, blue));
#endif
			show = true;
		}
	}
}



void callbackApplication(char *topic, uint8_t tokenCnt, char **tokens) {
	if (! strcmp(topic, configBlock.mqttTopicFlowCommand)) {
		switch (tokenCnt) {
		case 1:
			if (! strcmp(tokens[0], "up")) {
				colorFlowCmd = E_FC_FADE_ON;
			} else if (! strcmp(tokens[0], "down")) {
				colorFlowCmd = E_FC_FADE_OFF;
			}
			break;
		}
	} else if (! strcmp(topic, configBlock.mqttTopicCommand)) {
		int32_t n, b;
		CRGB pseudoColors;
		switch (tokenCnt) {
		case 1:
			// set brightness
			b = strtol(tokens[0], NULL, 10);
			setColor(-1, b, b, b);
			break;
		case 2:
			// set brightness for one LED
			n = strtol(tokens[0], NULL, 10);
			b = strtol(tokens[1], NULL, 10);
			setColor(n, b, b, b);
			break;
		}
		colorFlowCmd = E_FC_ABORT;
	} else if (! strcmp(topic, configBlock.mqttTopicColorCommand)) {
		int32_t n, red, green, blue;
		CRGB colors;
		switch (tokenCnt) {
		case 1:
			// on, off, color word for all LEDs
			colors = evaluationColorWord(tokens[0]);
			setColor(-1, colors.r, colors.g, colors.b);
			break;
		case 2:
			// token0 = LED number, token1 = color
			n = strtol(tokens[0], NULL, 10);
			colors = evaluationColorWord(tokens[1]);
			setColor(n, colors.r, colors.g, colors.b);
			break;
		case 3:
			// token0 = red, token1 = green, token2 = blue
			red = strtol(tokens[0], NULL, 10);
			green = strtol(tokens[1], NULL, 10);
			blue = strtol(tokens[2], NULL, 10);
			setColor(-1, red, green, blue);
			break;
		case 4:
			// token0 = LED number, token1 = red, token2 = green, token3 = blue
			n = strtol(tokens[0], NULL, 10);
			red = strtol(tokens[1], NULL, 10);
			green = strtol(tokens[2], NULL, 10);
			blue = strtol(tokens[3], NULL, 10);
			setColor(n, red, green, blue);
			break;
		}
		colorFlowCmd = E_FC_ABORT;
	}
}


void setupApplication() {
	mqttSetup();

#ifdef WS2811
	FastLED.addLeds<NEOPIXEL, PIXEL_PIN>(leds, NUM_OF_LEDs);
#endif
#ifdef PL9823
	pixels.begin();

	for (uint8_t i = 0; i < NUM_OF_LEDs; i++) {
		pixels.setPixelColor(i, pixels.Color(0,0,0));
	}
	pixels.show();
#endif
}


void loopApplication() {
	mqttLoop();

	colorFlowEngine();

	if (show) {
		show = false;
#ifdef WS2811
		FastLED.show();
#endif
#ifdef PL9823
		pixels.show();
#endif
	}
}