#pragma once #include "wled.h" #ifndef MULTI_RELAY_MAX_RELAYS #define MULTI_RELAY_MAX_RELAYS 4 #else #if MULTI_RELAY_MAX_RELAYS>8 #undef MULTI_RELAY_MAX_RELAYS #define MULTI_RELAY_MAX_RELAYS 8 #warning Maximum relays set to 8 #endif #endif #ifndef MULTI_RELAY_PINS #define MULTI_RELAY_PINS -1 #define MULTI_RELAY_ENABLED false #else #define MULTI_RELAY_ENABLED true #endif #define WLED_DEBOUNCE_THRESHOLD 50 //only consider button input of at least 50ms as valid (debouncing) #define ON true #define OFF false #ifndef USERMOD_USE_PCF8574 #undef USE_PCF8574 #define USE_PCF8574 false #else #undef USE_PCF8574 #define USE_PCF8574 true #endif #ifndef PCF8574_ADDRESS #define PCF8574_ADDRESS 0x20 // some may start at 0x38 #endif /* * This usermod handles multiple relay outputs. * These outputs complement built-in relay output in a way that the activation can be delayed. * They can also activate/deactivate in reverse logic independently. * * Written and maintained by @blazoncek */ typedef struct relay_t { int8_t pin; struct { // reduces memory footprint bool active : 1; // is the relay waiting to be switched bool invert : 1; // does On mean 1 or 0 bool state : 1; // 1 relay is On, 0 relay is Off bool external : 1; // is the relay externally controlled int8_t button : 4; // which button triggers relay }; uint16_t delay; // amount of ms to wait after it is activated } Relay; class MultiRelay : public Usermod { private: // array of relays Relay _relay[MULTI_RELAY_MAX_RELAYS]; uint32_t _switchTimerStart; // switch timer start time bool _oldMode; // old brightness bool enabled; // usermod enabled bool initDone; // status of initialisation bool usePcf8574; uint8_t addrPcf8574; bool HAautodiscovery; uint16_t periodicBroadcastSec; unsigned long lastBroadcast; // strings to reduce flash memory usage (used more than twice) static const char _name[]; static const char _enabled[]; static const char _relay_str[]; static const char _delay_str[]; static const char _activeHigh[]; static const char _external[]; static const char _button[]; static const char _broadcast[]; static const char _HAautodiscovery[]; static const char _pcf8574[]; static const char _pcfAddress[]; void handleOffTimer(); void InitHtmlAPIHandle(); int getValue(String data, char separator, int index); uint8_t getActiveRelayCount(); byte IOexpanderWrite(byte address, byte _data); byte IOexpanderRead(int address); void publishMqtt(int relay); #ifndef WLED_DISABLE_MQTT void publishHomeAssistantAutodiscovery(); #endif public: /** * constructor */ MultiRelay(); /** * desctructor */ //~MultiRelay() {} /** * Enable/Disable the usermod */ inline void enable(bool enable) { enabled = enable; } /** * Get usermod enabled/disabled state */ inline bool isEnabled() { return enabled; } /** * getId() allows you to optionally give your V2 usermod an unique ID (please define it in const.h!). * This could be used in the future for the system to determine whether your usermod is installed. */ inline uint16_t getId() { return USERMOD_ID_MULTI_RELAY; } /** * switch relay on/off */ void switchRelay(uint8_t relay, bool mode); /** * toggle relay */ inline void toggleRelay(uint8_t relay) { switchRelay(relay, !_relay[relay].state); } /** * setup() is called once at boot. WiFi is not yet connected at this point. * You can use it to initialize variables, sensors or similar. */ void setup(); /** * connected() is called every time the WiFi is (re)connected * Use it to initialize network interfaces */ inline void connected() { InitHtmlAPIHandle(); } /** * loop() is called continuously. Here you can check for events, read sensors, etc. */ void loop(); #ifndef WLED_DISABLE_MQTT bool onMqttMessage(char* topic, char* payload); void onMqttConnect(bool sessionPresent); #endif /** * handleButton() can be used to override default button behaviour. Returning true * will prevent button working in a default way. * Replicating button.cpp */ bool handleButton(uint8_t b); /** * addToJsonInfo() can be used to add custom entries to the /json/info part of the JSON API. */ void addToJsonInfo(JsonObject &root); /** * addToJsonState() can be used to add custom entries to the /json/state part of the JSON API (state object). * Values in the state object may be modified by connected clients */ void addToJsonState(JsonObject &root); /** * readFromJsonState() can be used to receive data clients send to the /json/state part of the JSON API (state object). * Values in the state object may be modified by connected clients */ void readFromJsonState(JsonObject &root); /** * provide the changeable values */ void addToConfig(JsonObject &root); void appendConfigData(); /** * restore the changeable values * readFromConfig() is called before setup() to populate properties from values stored in cfg.json * * The function should return true if configuration was successfully loaded or false if there was no configuration. */ bool readFromConfig(JsonObject &root); }; // class implementetion void MultiRelay::publishMqtt(int relay) { #ifndef WLED_DISABLE_MQTT //Check if MQTT Connected, otherwise it will crash the 8266 if (WLED_MQTT_CONNECTED){ char subuf[64]; sprintf_P(subuf, PSTR("%s/relay/%d"), mqttDeviceTopic, relay); mqtt->publish(subuf, 0, false, _relay[relay].state ? "on" : "off"); } #endif } /** * switch off the strip if the delay has elapsed */ void MultiRelay::handleOffTimer() { unsigned long now = millis(); bool activeRelays = false; for (int i=0; i 0 && now - _switchTimerStart > (_relay[i].delay*1000)) { if (!_relay[i].external) switchRelay(i, !offMode); _relay[i].active = false; } else if (periodicBroadcastSec && now - lastBroadcast > (periodicBroadcastSec*1000)) { if (_relay[i].pin>=0) publishMqtt(i); } activeRelays = activeRelays || _relay[i].active; } if (!activeRelays) _switchTimerStart = 0; if (periodicBroadcastSec && now - lastBroadcast > (periodicBroadcastSec*1000)) lastBroadcast = now; } /** * HTTP API handler * borrowed from: * https://github.com/gsieben/WLED/blob/master/usermods/GeoGab-Relays/usermod_GeoGab.h */ #define GEOGABVERSION "0.1.3" void MultiRelay::InitHtmlAPIHandle() { // https://github.com/me-no-dev/ESPAsyncWebServer DEBUG_PRINTLN(F("Relays: Initialize HTML API")); server.on("/relays", HTTP_GET, [this](AsyncWebServerRequest *request) { DEBUG_PRINTLN("Relays: HTML API"); String janswer; String error = ""; //int params = request->params(); janswer = F("{\"NoOfRelays\":"); janswer += String(MULTI_RELAY_MAX_RELAYS) + ","; if (getActiveRelayCount()) { // Commands if(request->hasParam("switch")) { /**** Switch ****/ AsyncWebParameter* p = request->getParam("switch"); // Get Values for (int i=0; ivalue(), ',', i); if (value==-1) { error = F("There must be as many arguments as relays"); } else { // Switch if (_relay[i].external) switchRelay(i, (bool)value); } } } else if(request->hasParam("toggle")) { /**** Toggle ****/ AsyncWebParameter* p = request->getParam("toggle"); // Get Values for (int i=0;ivalue(), ',', i); if (value==-1) { error = F("There must be as many arguments as relays"); } else { // Toggle if (value && _relay[i].external) toggleRelay(i); } } } else { error = F("No valid command found"); } } else { error = F("No active relays"); } // Status response char sbuf[16]; for (int i=0; isend(200, "application/json", janswer); }); } int MultiRelay::getValue(String data, char separator, int index) { int found = 0; int strIndex[] = {0, -1}; int maxIndex = data.length()-1; for(int i=0; i<=maxIndex && found<=index; i++){ if(data.charAt(i)==separator || i==maxIndex){ found++; strIndex[0] = strIndex[1]+1; strIndex[1] = (i == maxIndex) ? i+1 : i; } } return found>index ? data.substring(strIndex[0], strIndex[1]).toInt() : -1; } //Write a byte to the IO expander byte MultiRelay::IOexpanderWrite(byte address, byte _data ) { Wire.beginTransmission(address); Wire.write(_data); return Wire.endTransmission(); } //Read a byte from the IO expander byte MultiRelay::IOexpanderRead(int address) { byte _data = 0; Wire.requestFrom(address, 1); if (Wire.available()) { _data = Wire.read(); } return _data; } // public methods MultiRelay::MultiRelay() : _switchTimerStart(0) , enabled(MULTI_RELAY_ENABLED) , initDone(false) , usePcf8574(USE_PCF8574) , addrPcf8574(PCF8574_ADDRESS) , HAautodiscovery(false) , periodicBroadcastSec(60) , lastBroadcast(0) { const int8_t defPins[] = {MULTI_RELAY_PINS}; for (size_t i=0; i=MULTI_RELAY_MAX_RELAYS || _relay[relay].pin<0) return; _relay[relay].state = mode; if (usePcf8574 && _relay[relay].pin >= 100) { // we need to send all ouputs at the same time uint8_t state = 0; for (int i=0; i=0) count++; return count; } //Functions called by WLED #ifndef WLED_DISABLE_MQTT /** * handling of MQTT message * topic only contains stripped topic (part after /wled/MAC) * topic should look like: /relay/X/command; where X is relay number, 0 based */ bool MultiRelay::onMqttMessage(char* topic, char* payload) { if (strlen(topic) > 8 && strncmp_P(topic, PSTR("/relay/"), 7) == 0 && strncmp_P(topic+8, PSTR("/command"), 8) == 0) { uint8_t relay = strtoul(topic+7, NULL, 10); if (relaysubscribe(subuf, 0); if (HAautodiscovery) publishHomeAssistantAutodiscovery(); for (int i=0; i= 0 && _relay[i].external) { StaticJsonDocument<1024> json; sprintf_P(buf, PSTR("%s Switch %d"), serverDescription, i); //max length: 33 + 8 + 3 = 44 json[F("name")] = buf; sprintf_P(buf, PSTR("%s/relay/%d"), mqttDeviceTopic, i); //max length: 33 + 7 + 3 = 43 json["~"] = buf; strcat_P(buf, PSTR("/command")); mqtt->subscribe(buf, 0); json[F("stat_t")] = "~"; json[F("cmd_t")] = F("~/command"); json[F("pl_off")] = "off"; json[F("pl_on")] = "on"; json[F("uniq_id")] = uid; strcpy(buf, mqttDeviceTopic); //max length: 33 + 7 = 40 strcat_P(buf, PSTR("/status")); json[F("avty_t")] = buf; json[F("pl_avail")] = F("online"); json[F("pl_not_avail")] = F("offline"); //TODO: dev payload_size = serializeJson(json, json_str); } else { //Unpublish disabled or internal relays json_str[0] = 0; payload_size = 0; } sprintf_P(buf, PSTR("homeassistant/switch/%s/config"), uid); mqtt->publish(buf, 0, true, json_str, payload_size); } } #endif /** * setup() is called once at boot. WiFi is not yet connected at this point. * You can use it to initialize variables, sensors or similar. */ void MultiRelay::setup() { // pins retrieved from cfg.json (readFromConfig()) prior to running setup() // if we want PCF8574 expander I2C pins need to be valid if (i2c_sda<0 || i2c_scl<0) usePcf8574 = false; uint8_t state = 0; for (int i=0; i= 100) { uint8_t pin = _relay[i].pin - 100; if (!_relay[i].external) _relay[i].state = !offMode; state |= (uint8_t)(_relay[i].invert ? !_relay[i].state : _relay[i].state) << pin; } else if (_relay[i].pin<100 && _relay[i].pin>=0) { if (pinManager.allocatePin(_relay[i].pin,true, PinOwner::UM_MultiRelay)) { if (!_relay[i].external) _relay[i].state = !offMode; switchRelay(i, _relay[i].state); _relay[i].active = false; } else { _relay[i].pin = -1; // allocation failed } } } if (usePcf8574) { IOexpanderWrite(addrPcf8574, state); // init expander (set all outputs) DEBUG_PRINTLN(F("PCF8574(s) inited.")); } _oldMode = offMode; initDone = true; } /** * loop() is called continuously. Here you can check for events, read sensors, etc. */ void MultiRelay::loop() { yield(); if (!enabled || strip.isUpdating()) return; static unsigned long lastUpdate = 0; if (millis() - lastUpdate < 100) return; // update only 10 times/s lastUpdate = millis(); //set relay when LEDs turn on if (_oldMode != offMode) { _oldMode = offMode; _switchTimerStart = millis(); for (int i=0; i=0) && !_relay[i].external) _relay[i].active = true; } } handleOffTimer(); } /** * handleButton() can be used to override default button behaviour. Returning true * will prevent button working in a default way. * Replicating button.cpp */ bool MultiRelay::handleButton(uint8_t b) { yield(); if (!enabled || buttonType[b] == BTN_TYPE_NONE || buttonType[b] == BTN_TYPE_RESERVED || buttonType[b] == BTN_TYPE_PIR_SENSOR || buttonType[b] == BTN_TYPE_ANALOG || buttonType[b] == BTN_TYPE_ANALOG_INVERTED) { return false; } bool handled = false; for (int i=0; i WLED_DEBOUNCE_THRESHOLD) { //fire edge event only after 50ms without change (debounce) for (int i=0; i 600) { //long press //longPressAction(b); //not exposed //handled = false; //use if you want to pass to default behaviour buttonLongPressed[b] = true; } } else if (!isButtonPressed(b) && buttonPressedBefore[b]) { //released long dur = now - buttonPressedTime[b]; if (dur < WLED_DEBOUNCE_THRESHOLD) { buttonPressedBefore[b] = false; return handled; } //too short "press", debounce bool doublePress = buttonWaitTime[b]; //did we have short press before? buttonWaitTime[b] = 0; if (!buttonLongPressed[b]) { //short press // if this is second release within 350ms it is a double press (buttonWaitTime!=0) if (doublePress) { //doublePressAction(b); //not exposed //handled = false; //use if you want to pass to default behaviour } else { buttonWaitTime[b] = now; } } buttonPressedBefore[b] = false; buttonLongPressed[b] = false; } // if 350ms elapsed since last press/release it is a short press if (buttonWaitTime[b] && now - buttonWaitTime[b] > 350 && !buttonPressedBefore[b]) { buttonWaitTime[b] = 0; //shortPressAction(b); //not exposed for (int i=0; i"); uiDomString += F(""); infoArr.add(uiDomString); } } } /** * addToJsonState() can be used to add custom entries to the /json/state part of the JSON API (state object). * Values in the state object may be modified by connected clients */ void MultiRelay::addToJsonState(JsonObject &root) { if (!initDone || !enabled) return; // prevent crash on boot applyPreset() JsonObject multiRelay = root[FPSTR(_name)]; if (multiRelay.isNull()) { multiRelay = root.createNestedObject(FPSTR(_name)); } #if MULTI_RELAY_MAX_RELAYS > 1 JsonArray rel_arr = multiRelay.createNestedArray(F("relays")); for (int i=0; i() && usermod[FPSTR(_relay_str)].as()>=0) { int rly = usermod[FPSTR(_relay_str)].as(); if (usermod["on"].is()) { switchRelay(rly, usermod["on"].as()); } else if (usermod["on"].is() && usermod["on"].as()[0] == 't') { toggleRelay(rly); } } } else if (root[FPSTR(_name)].is()) { JsonArray relays = root[FPSTR(_name)].as(); for (JsonVariant r : relays) { if (r[FPSTR(_relay_str)].is() && r[FPSTR(_relay_str)].as()>=0) { int rly = r[FPSTR(_relay_str)].as(); if (r["on"].is()) { switchRelay(rly, r["on"].as()); } else if (r["on"].is() && r["on"].as()[0] == 't') { toggleRelay(rly); } } } } } /** * provide the changeable values */ void MultiRelay::addToConfig(JsonObject &root) { JsonObject top = root.createNestedObject(FPSTR(_name)); top[FPSTR(_enabled)] = enabled; top[FPSTR(_pcf8574)] = usePcf8574; top[FPSTR(_pcfAddress)] = addrPcf8574; top[FPSTR(_broadcast)] = periodicBroadcastSec; top[FPSTR(_HAautodiscovery)] = HAautodiscovery; for (int i=0; i(not hex!)');")); oappend(SET_F("addInfo('MultiRelay:broadcast-sec',1,'(MQTT message)');")); //oappend(SET_F("addInfo('MultiRelay:relay-0:pin',1,'(use -1 for PCF8574)');")); oappend(SET_F("d.extra.push({'MultiRelay':{pin:[['P0',100],['P1',101],['P2',102],['P3',103],['P4',104],['P5',105],['P6',106],['P7',107]]}});")); } /** * restore the changeable values * readFromConfig() is called before setup() to populate properties from values stored in cfg.json * * The function should return true if configuration was successfully loaded or false if there was no configuration. */ bool MultiRelay::readFromConfig(JsonObject &root) { int8_t oldPin[MULTI_RELAY_MAX_RELAYS]; JsonObject top = root[FPSTR(_name)]; if (top.isNull()) { DEBUG_PRINT(FPSTR(_name)); DEBUG_PRINTLN(F(": No config found. (Using defaults.)")); return false; } //bool configComplete = !top.isNull(); //configComplete &= getJsonValue(top[FPSTR(_enabled)], enabled); enabled = top[FPSTR(_enabled)] | enabled; usePcf8574 = top[FPSTR(_pcf8574)] | usePcf8574; addrPcf8574 = top[FPSTR(_pcfAddress)] | addrPcf8574; // if I2C is not globally initialised just ignore if (i2c_sda<0 || i2c_scl<0) usePcf8574 = false; periodicBroadcastSec = top[FPSTR(_broadcast)] | periodicBroadcastSec; periodicBroadcastSec = min(900,max(0,(int)periodicBroadcastSec)); HAautodiscovery = top[FPSTR(_HAautodiscovery)] | HAautodiscovery; for (int i=0; i=0 && oldPin[i]<100) { pinManager.deallocatePin(oldPin[i], PinOwner::UM_MultiRelay); } // allocate new pins setup(); DEBUG_PRINTLN(F(" config (re)loaded.")); } // use "return !top["newestParameter"].isNull();" when updating Usermod with new features return !top[FPSTR(_pcf8574)].isNull(); } // strings to reduce flash memory usage (used more than twice) const char MultiRelay::_name[] PROGMEM = "MultiRelay"; const char MultiRelay::_enabled[] PROGMEM = "enabled"; const char MultiRelay::_relay_str[] PROGMEM = "relay"; const char MultiRelay::_delay_str[] PROGMEM = "delay-s"; const char MultiRelay::_activeHigh[] PROGMEM = "active-high"; const char MultiRelay::_external[] PROGMEM = "external"; const char MultiRelay::_button[] PROGMEM = "button"; const char MultiRelay::_broadcast[] PROGMEM = "broadcast-sec"; const char MultiRelay::_HAautodiscovery[] PROGMEM = "HA-autodiscovery"; const char MultiRelay::_pcf8574[] PROGMEM = "use-PCF8574"; const char MultiRelay::_pcfAddress[] PROGMEM = "PCF8574-address";