led-controller/src/main.cpp

276 lines
5.8 KiB
C++

#include <Arduino.h>
#include <ESP8266WiFi.h>
#include <PubSubClient.h>
#define REDPIN1 5
#define GREENPIN1 4
#define BLUEPIN1 13
#define REDPIN2 12
#define GREENPIN2 14
#define BLUEPIN2 16
#define MQTT_TOPIC_START "led1"
#define MQTT_USER ""
#define MQTT_PASS ""
#define MQTT_BROKER "mqtt.example.com"
#define MQTT_CLIENT_ID "LED-Schrank-1"
const char* SSID = "WIFISSID";
const char* PSK = "WIDIPASS";
WiFiClient espClient;
PubSubClient client(espClient);
struct LEDStripe {
String MOD = "off"; //Which mode is used "off", "fade", "on"
String OLD_MOD = "off";
int SPEED = 3000;
int RED = 0;
int GREEN = 0;
int BLUE = 0;
int LOOP = 0;
int LOOPCOUNT = 0;
int MOD_STEP = 0;
};
int Sync = 0; //Sync LED2 on the value of LED1
LEDStripe led1;
LEDStripe led2;
void connectWiFi() {
digitalWrite(LED_BUILTIN, LOW);
Serial.println("Start WiFi Connection");
WiFi.begin(SSID, PSK);
while (WiFi.status() != WL_CONNECTED) {
delay(500);
Serial.print(".");
}
digitalWrite(LED_BUILTIN, HIGH);
Serial.println("");
Serial.println("WiFi connected");
Serial.print("IP address: ");
Serial.println(WiFi.localIP());
}
//BLink
void blink_short() {
digitalWrite(LED_BUILTIN, LOW);
delay(500);
digitalWrite(LED_BUILTIN, HIGH);
delay(500);
}
void blink_long() {
digitalWrite(LED_BUILTIN, LOW);
delay(1500);
digitalWrite(LED_BUILTIN, HIGH);
delay(500);
}
void blink(int status) {
if(status == 1) {
blink_short();
blink_short();
blink_short();
blink_short();
}
}
char* getTopicName(String topic) {
String topicName = String(MQTT_TOPIC_START)+String(topic);
int str_len = topicName.length() + 1;
char cc[str_len];
topicName.toCharArray(cc, str_len);
return cc;
}
void connectMQTT() {
digitalWrite(LED_BUILTIN, LOW);
Serial.print("Reconnecting...");
if (!client.connect(MQTT_CLIENT_ID, MQTT_USER, MQTT_PASS, getTopicName("/status"), 1, true, "0")) {
Serial.print("failed, rc=");
Serial.print(client.state());
Serial.println(" retrying in 5 seconds");
digitalWrite(LED_BUILTIN, HIGH);
delay(2000);
blink(1);
delay(3000);
}
digitalWrite(LED_BUILTIN, HIGH);
client.subscribe(getTopicName("/#"));
}
//Actions
void pulse_white_three_times() {
LEDStripe tmpLed();
}
//Modes
//LED Fade
void fade(LEDStripe led) {
if(led.MOD_STEP == 0) {
led.RED ++;
led.GREEN = 0;
led.BLUE = 255;
if(led.RED >= 255) {
led.MOD_STEP++;
}
}
if(led.MOD_STEP == 1) {
led.RED = 255;
led.GREEN = 0;
led.BLUE --;
if(led.BLUE <= 0) {
led.MOD_STEP++;
}
}
if(led.MOD_STEP == 2) {
led.RED = 255;
led.GREEN ++;
led.BLUE = 0;
if(led.GREEN >= 255) {
led.MOD_STEP++;
}
}
if(led.MOD_STEP == 3) {
led.RED --;
led.GREEN = 255;
led.BLUE = 0;
if(led.RED <= 0) {
led.MOD_STEP++;
}
}
if(led.MOD_STEP == 4) {
led.RED = 0;
led.GREEN = 255;
led.BLUE ++;
if(led.BLUE >= 255) {
led.MOD_STEP++;
}
}
if(led.MOD_STEP == 5) {
led.RED = 0;
led.GREEN --;
led.BLUE = 255;
if(led.GREEN <= 0) {
led.MOD_STEP = 0;
}
}
}
void writeToLedStrips(LEDStripe l1, LEDStripe l2) {
analogWrite(REDPIN1, l1.RED);
analogWrite(REDPIN2, l2.RED);
analogWrite(GREENPIN1, l1.GREEN);
analogWrite(GREENPIN2, l2.GREEN);
analogWrite(BLUEPIN1, l1.BLUE);
analogWrite(BLUEPIN2, l2.BLUE);
Serial.println("LED1: "+String(l1.RED)+"|"+String(l1.GREEN)+"|"+String(l1.BLUE));
Serial.println("LED2: "+String(l2.RED)+"|"+String(l2.GREEN)+"|"+String(l2.BLUE));
}
void callback(char* topic, byte* payload, unsigned int length) {
String message = "";
for (int i=0;i<int(length);i++) {
message = message + String((char)payload[i]);
}
if(String(topic) == String(getTopicName("/1/mode"))) {
led1.MOD = message;
}
if(String(topic) == String(getTopicName("/2/mode"))) {
led2.MOD = message;
}
if(String(topic) == String(getTopicName("/1/speed"))) {
led1.SPEED = message.toInt();
}
if(String(topic) == String(getTopicName("/2/speed"))) {
led2.SPEED = message.toInt();
}
if(String(topic) == String(getTopicName("/1/color"))) {
led1.RED = message.substring(0,3).toInt();
led1.GREEN = message.substring(3,6).toInt();
led1.BLUE = message.substring(6,9).toInt();
}
if(String(topic) == String(getTopicName("/2/color"))) {
led2.RED = message.substring(0,3).toInt();
led2.GREEN = message.substring(3,6).toInt();
led2.BLUE = message.substring(6,9).toInt();
}
if(String(topic) == String(getTopicName("/sync"))) {
Sync = message.toInt();
}
Serial.print("Message arrived [");
Serial.print(topic);
Serial.print("] ");
Serial.print(message);
Serial.println();
}
void prepareMQTT() {
digitalWrite(LED_BUILTIN, LOW);
Serial.println("Prepare MQTT Client");
client.setServer(MQTT_BROKER, 1883);
client.setCallback(callback);
digitalWrite(LED_BUILTIN, HIGH);
}
void setup() {
Serial.begin(115200);
pinMode(LED_BUILTIN, OUTPUT);
Serial.println();
Serial.println("Start LED-Controller");
connectWiFi();
prepareMQTT();
connectMQTT();
//init LED Strips
pinMode(REDPIN1, OUTPUT);
pinMode(REDPIN2, OUTPUT);
pinMode(GREENPIN1, OUTPUT);
pinMode(GREENPIN2, OUTPUT);
pinMode(BLUEPIN1, OUTPUT);
pinMode(BLUEPIN2, OUTPUT);
//turn led Strips off
analogWrite(REDPIN1, 0);
analogWrite(REDPIN2, 0);
analogWrite(GREENPIN1, 0);
analogWrite(GREENPIN2, 0);
analogWrite(BLUEPIN1, 0);
analogWrite(BLUEPIN2, 0);
}
void loop() {
if (!client.connected()) {
connectMQTT();
}
client.loop();
if(led1.MOD == "fade") {
fade(led1);
}
if(Sync == 1) {
led2.RED = led1.RED;
led2.GREEN = led1.GREEN;
led2.BLUE = led1.BLUE;
} else {
if(led2.MOD == "fade") {
fade(led2);
}
}
//Send calculated colors to led-strips
writeToLedStrips(led1, led2);
}