You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
252 lines
6.2 KiB
252 lines
6.2 KiB
/*
|
|
|
|
ESPurna
|
|
RELAY MODULE
|
|
|
|
Copyright (C) 2016 by Xose Pérez <xose dot perez at gmail dot com>
|
|
|
|
*/
|
|
|
|
#include <EEPROM.h>
|
|
#include <ArduinoJson.h>
|
|
#include <vector>
|
|
|
|
std::vector<unsigned char> _relays;
|
|
bool recursive = false;
|
|
#ifdef SONOFF_DUAL
|
|
unsigned char dualRelayStatus = 0;
|
|
#endif
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// RELAY
|
|
// -----------------------------------------------------------------------------
|
|
|
|
void relayMQTT(unsigned char id) {
|
|
if (id >= _relays.size()) return;
|
|
String mqttGetter = getSetting("mqttGetter", MQTT_USE_GETTER);
|
|
char buffer[strlen(MQTT_RELAY_TOPIC) + mqttGetter.length() + 3];
|
|
sprintf(buffer, "%s/%d%s", MQTT_RELAY_TOPIC, id, mqttGetter.c_str());
|
|
mqttSend(buffer, relayStatus(id) ? "1" : "0");
|
|
}
|
|
|
|
void relayMQTT() {
|
|
for (unsigned int i=0; i < _relays.size(); i++) {
|
|
relayMQTT(i);
|
|
}
|
|
}
|
|
|
|
String relayString() {
|
|
DynamicJsonBuffer jsonBuffer;
|
|
JsonObject& root = jsonBuffer.createObject();
|
|
JsonArray& relay = root.createNestedArray("relayStatus");
|
|
for (unsigned char i=0; i<relayCount(); i++) {
|
|
relay.add(relayStatus(i));
|
|
}
|
|
String output;
|
|
root.printTo(output);
|
|
return output;
|
|
}
|
|
|
|
void relayWS() {
|
|
String output = relayString();
|
|
wsSend((char *) output.c_str());
|
|
}
|
|
|
|
bool relayStatus(unsigned char id) {
|
|
#ifdef SONOFF_DUAL
|
|
if (id >= 2) return false;
|
|
return ((dualRelayStatus & (1 << id)) > 0);
|
|
#else
|
|
if (id >= _relays.size()) return false;
|
|
return (digitalRead(_relays[id]) == HIGH);
|
|
#endif
|
|
}
|
|
|
|
bool relayStatus(unsigned char id, bool status, bool report) {
|
|
|
|
if (id >= _relays.size()) return false;
|
|
|
|
bool changed = false;
|
|
|
|
if (relayStatus(id) != status) {
|
|
|
|
DEBUG_MSG("[RELAY] %d => %s\n", id, status ? "ON" : "OFF");
|
|
changed = true;
|
|
|
|
#ifdef SONOFF_DUAL
|
|
|
|
dualRelayStatus ^= (1 << id);
|
|
Serial.flush();
|
|
Serial.write(0xA0);
|
|
Serial.write(0x04);
|
|
Serial.write(dualRelayStatus);
|
|
Serial.write(0xA1);
|
|
Serial.flush();
|
|
|
|
#else
|
|
digitalWrite(_relays[id], status);
|
|
#endif
|
|
|
|
if (!recursive) {
|
|
relaySync(id);
|
|
relaySave();
|
|
}
|
|
|
|
}
|
|
|
|
if (report) relayMQTT(id);
|
|
if (!recursive) relayWS();
|
|
return changed;
|
|
|
|
}
|
|
|
|
bool relayStatus(unsigned char id, bool status) {
|
|
if (id >= _relays.size()) return false;
|
|
return relayStatus(id, status, true);
|
|
}
|
|
|
|
void relaySync(unsigned char id) {
|
|
|
|
if (_relays.size() > 1) {
|
|
|
|
recursive = true;
|
|
|
|
byte relaySync = getSetting("relaySync", RELAY_SYNC).toInt();
|
|
bool status = relayStatus(id);
|
|
|
|
// If RELAY_SYNC_SAME all relays should have the same state
|
|
if (relaySync == RELAY_SYNC_SAME) {
|
|
for (unsigned short i=0; i<_relays.size(); i++) {
|
|
if (i != id) relayStatus(i, status);
|
|
}
|
|
|
|
// If NONE_OR_ONE or ONE and setting ON we should set OFF all the others
|
|
} else if (status) {
|
|
if (relaySync != RELAY_SYNC_ANY) {
|
|
for (unsigned short i=0; i<_relays.size(); i++) {
|
|
if (i != id) relayStatus(i, false);
|
|
}
|
|
}
|
|
|
|
// If ONLY_ONE and setting OFF we should set ON the other one
|
|
} else {
|
|
if (relaySync == RELAY_SYNC_ONE) {
|
|
unsigned char i = (id + 1) % _relays.size();
|
|
relayStatus(i, true);
|
|
}
|
|
}
|
|
|
|
recursive = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
void relaySave() {
|
|
unsigned char bit = 1;
|
|
unsigned char mask = 0;
|
|
for (unsigned int i=0; i < _relays.size(); i++) {
|
|
if (relayStatus(i)) mask += bit;
|
|
bit += bit;
|
|
}
|
|
EEPROM.write(0, mask);
|
|
EEPROM.commit();
|
|
}
|
|
|
|
void relayRetrieve() {
|
|
recursive = true;
|
|
unsigned char bit = 1;
|
|
unsigned char mask = EEPROM.read(0);
|
|
for (unsigned int i=0; i < _relays.size(); i++) {
|
|
relayStatus(i, ((mask & bit) == bit));
|
|
bit += bit;
|
|
}
|
|
recursive = false;
|
|
}
|
|
|
|
void relayToggle(unsigned char id) {
|
|
if (id >= _relays.size()) return;
|
|
relayStatus(id, !relayStatus(id));
|
|
}
|
|
|
|
unsigned char relayCount() {
|
|
return _relays.size();
|
|
}
|
|
|
|
void relayMQTTCallback(unsigned int type, const char * topic, const char * payload) {
|
|
|
|
String mqttSetter = getSetting("mqttSetter", MQTT_USE_SETTER);
|
|
String mqttGetter = getSetting("mqttGetter", MQTT_USE_GETTER);
|
|
bool sameSetGet = mqttGetter.compareTo(mqttSetter) == 0;
|
|
|
|
if (type == MQTT_CONNECT_EVENT) {
|
|
relayMQTT();
|
|
char buffer[strlen(MQTT_RELAY_TOPIC) + mqttSetter.length() + 3];
|
|
sprintf(buffer, "%s/+%s", MQTT_RELAY_TOPIC, mqttSetter.c_str());
|
|
mqttSubscribe(buffer);
|
|
}
|
|
|
|
if (type == MQTT_MESSAGE_EVENT) {
|
|
|
|
// Match topic
|
|
String t = String(topic + mqttTopicRootLength());
|
|
if (!t.startsWith(MQTT_RELAY_TOPIC)) return;
|
|
if (!t.endsWith(mqttSetter)) return;
|
|
|
|
// Get relay ID
|
|
unsigned int relayID = topic[strlen(MQTT_RELAY_TOPIC)+1] - '0';
|
|
if (relayID >= relayCount()) relayID = 0;
|
|
|
|
// Action to perform
|
|
unsigned int value = (char)payload[0] - '0';
|
|
if (value == 2) {
|
|
relayToggle(relayID);
|
|
} else {
|
|
relayStatus(relayID, value > 0, !sameSetGet);
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
void relaySetup() {
|
|
|
|
#ifdef SONOFF_DUAL
|
|
|
|
// Two dummy relays for the dual
|
|
_relays.push_back(0);
|
|
_relays.push_back(0);
|
|
|
|
#else
|
|
|
|
#ifdef RELAY1_PIN
|
|
_relays.push_back(RELAY1_PIN);
|
|
#endif
|
|
#ifdef RELAY2_PIN
|
|
_relays.push_back(RELAY2_PIN);
|
|
#endif
|
|
#ifdef RELAY3_PIN
|
|
_relays.push_back(RELAY3_PIN);
|
|
#endif
|
|
#ifdef RELAY4_PIN
|
|
_relays.push_back(RELAY4_PIN);
|
|
#endif
|
|
|
|
#endif
|
|
|
|
EEPROM.begin(4096);
|
|
byte relayMode = getSetting("relayMode", RELAY_MODE).toInt();
|
|
|
|
for (unsigned int i=0; i < _relays.size(); i++) {
|
|
pinMode(_relays[i], OUTPUT);
|
|
if (relayMode == RELAY_MODE_OFF) relayStatus(i, false);
|
|
if (relayMode == RELAY_MODE_ON) relayStatus(i, true);
|
|
}
|
|
|
|
if (relayMode == RELAY_MODE_SAME) relayRetrieve();
|
|
|
|
mqttRegister(relayMQTTCallback);
|
|
|
|
DEBUG_MSG("[RELAY] Number of relays: %d\n", _relays.size());
|
|
|
|
}
|