Fork of the espurna firmware for `mhsw` switches
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.

261 lines
6.7 KiB

6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
  1. /*
  2. LED MODULE
  3. Copyright (C) 2016-2018 by Xose Pérez <xose dot perez at gmail dot com>
  4. */
  5. // -----------------------------------------------------------------------------
  6. // LED
  7. // -----------------------------------------------------------------------------
  8. typedef struct {
  9. unsigned char pin;
  10. bool reverse;
  11. unsigned char mode;
  12. unsigned char relay;
  13. } led_t;
  14. std::vector<led_t> _leds;
  15. bool _led_update = false; // For relay-based modes
  16. // -----------------------------------------------------------------------------
  17. bool _ledStatus(unsigned char id) {
  18. if (id >= _ledCount()) return false;
  19. bool status = digitalRead(_leds[id].pin);
  20. return _leds[id].reverse ? !status : status;
  21. }
  22. bool _ledStatus(unsigned char id, bool status) {
  23. if (id >=_ledCount()) return false;
  24. digitalWrite(_leds[id].pin, _leds[id].reverse ? !status : status);
  25. return status;
  26. }
  27. bool _ledToggle(unsigned char id) {
  28. if (id >= _ledCount()) return false;
  29. return _ledStatus(id, !_ledStatus(id));
  30. }
  31. unsigned char _ledMode(unsigned char id) {
  32. if (id >= _ledCount()) return false;
  33. return _leds[id].mode;
  34. }
  35. void _ledMode(unsigned char id, unsigned char mode) {
  36. if (id >= _ledCount()) return;
  37. _leds[id].mode = mode;
  38. }
  39. void _ledBlink(unsigned char id, unsigned long delayOff, unsigned long delayOn) {
  40. if (id >= _ledCount()) return;
  41. static unsigned long next = millis();
  42. if (next < millis()) {
  43. next += (_ledToggle(id) ? delayOn : delayOff);
  44. }
  45. }
  46. #if WEB_SUPPORT
  47. void _ledWebSocketOnSend(JsonObject& root) {
  48. if (_ledCount() == 0) return;
  49. root["ledVisible"] = 1;
  50. root["ledMode0"] = _ledMode(0);
  51. }
  52. #endif
  53. #if MQTT_SUPPORT
  54. void _ledMQTTCallback(unsigned int type, const char * topic, const char * payload) {
  55. if (type == MQTT_CONNECT_EVENT) {
  56. char buffer[strlen(MQTT_TOPIC_LED) + 3];
  57. snprintf_P(buffer, sizeof(buffer), PSTR("%s/+"), MQTT_TOPIC_LED);
  58. mqttSubscribe(buffer);
  59. }
  60. if (type == MQTT_MESSAGE_EVENT) {
  61. // Match topic
  62. String t = mqttMagnitude((char *) topic);
  63. if (!t.startsWith(MQTT_TOPIC_LED)) return;
  64. // Get led ID
  65. unsigned int ledID = t.substring(strlen(MQTT_TOPIC_LED)+1).toInt();
  66. if (ledID >= _ledCount()) {
  67. DEBUG_MSG_P(PSTR("[LED] Wrong ledID (%d)\n"), ledID);
  68. return;
  69. }
  70. // Check if LED is managed
  71. if (_ledMode(ledID) != LED_MODE_MQTT) return;
  72. // get value
  73. unsigned char value = relayParsePayload(payload);
  74. // Action to perform
  75. if (value == 2) {
  76. _ledToggle(ledID);
  77. } else {
  78. _ledStatus(ledID, value == 1);
  79. }
  80. }
  81. }
  82. #endif
  83. unsigned char _ledCount() {
  84. return _leds.size();
  85. }
  86. void _ledConfigure() {
  87. for (unsigned int i=0; i < _leds.size(); i++) {
  88. _ledMode(i, getSetting("ledMode", i, _ledMode(i)).toInt());
  89. }
  90. _led_update = true;
  91. }
  92. // -----------------------------------------------------------------------------
  93. void ledUpdate(bool value) {
  94. _led_update = value;
  95. }
  96. void ledSetup() {
  97. #ifdef LED1_PIN
  98. _leds.push_back((led_t) { LED1_PIN, LED1_PIN_INVERSE, LED1_MODE, LED1_RELAY });
  99. #endif
  100. #ifdef LED2_PIN
  101. _leds.push_back((led_t) { LED2_PIN, LED2_PIN_INVERSE, LED2_MODE, LED2_RELAY });
  102. #endif
  103. #ifdef LED3_PIN
  104. _leds.push_back((led_t) { LED3_PIN, LED3_PIN_INVERSE, LED3_MODE, LED3_RELAY });
  105. #endif
  106. #ifdef LED4_PIN
  107. _leds.push_back((led_t) { LED4_PIN, LED4_PIN_INVERSE, LED4_MODE, LED4_RELAY });
  108. #endif
  109. #ifdef LED5_PIN
  110. _leds.push_back((led_t) { LED5_PIN, LED5_PIN_INVERSE, LED5_MODE, LED5_RELAY });
  111. #endif
  112. #ifdef LED6_PIN
  113. _leds.push_back((led_t) { LED6_PIN, LED6_PIN_INVERSE, LED6_MODE, LED6_RELAY });
  114. #endif
  115. #ifdef LED7_PIN
  116. _leds.push_back((led_t) { LED7_PIN, LED7_PIN_INVERSE, LED7_MODE, LED7_RELAY });
  117. #endif
  118. #ifdef LED8_PIN
  119. _leds.push_back((led_t) { LED8_PIN, LED8_PIN_INVERSE, LED8_MODE, LED8_RELAY });
  120. #endif
  121. for (unsigned int i=0; i < _leds.size(); i++) {
  122. pinMode(_leds[i].pin, OUTPUT);
  123. _ledStatus(i, false);
  124. }
  125. _ledConfigure();
  126. #if MQTT_SUPPORT
  127. mqttRegister(_ledMQTTCallback);
  128. #endif
  129. #if WEB_SUPPORT
  130. wsOnSendRegister(_ledWebSocketOnSend);
  131. wsOnAfterParseRegister(_ledConfigure);
  132. #endif
  133. DEBUG_MSG_P(PSTR("[LED] Number of leds: %d\n"), _leds.size());
  134. // Register loop
  135. espurnaRegisterLoop(ledLoop);
  136. }
  137. void ledLoop() {
  138. for (unsigned char i=0; i<_leds.size(); i++) {
  139. if (_ledMode(i) == LED_MODE_WIFI) {
  140. if (wifiConnected()) {
  141. if (WiFi.getMode() == WIFI_AP) {
  142. _ledBlink(i, 900, 100);
  143. } else {
  144. _ledBlink(i, 4900, 100);
  145. }
  146. } else {
  147. _ledBlink(i, 500, 500);
  148. }
  149. }
  150. if (_ledMode(i) == LED_MODE_MIXED) {
  151. if (wifiConnected()) {
  152. if (relayStatus(_leds[i].relay-1)) {
  153. if (WiFi.getMode() == WIFI_AP) {
  154. _ledBlink(i, 900, 100);
  155. } else {
  156. _ledBlink(i, 4900, 100);
  157. }
  158. } else {
  159. if (WiFi.getMode() == WIFI_AP) {
  160. _ledBlink(i, 100, 900);
  161. } else {
  162. _ledBlink(i, 100, 4900);
  163. }
  164. }
  165. } else {
  166. _ledBlink(i, 500, 500);
  167. }
  168. }
  169. // Relay-based modes, update only if relays have been updated
  170. if (!_led_update) continue;
  171. if (_ledMode(i) == LED_MODE_FOLLOW) {
  172. _ledStatus(i, relayStatus(_leds[i].relay-1));
  173. }
  174. if (_ledMode(i) == LED_MODE_FOLLOW_INVERSE) {
  175. _ledStatus(i, !relayStatus(_leds[i].relay-1));
  176. }
  177. if (_ledMode(i) == LED_MODE_FINDME) {
  178. bool status = true;
  179. for (unsigned char k=0; k<relayCount(); k++) {
  180. if (relayStatus(k)) {
  181. status = false;
  182. break;
  183. }
  184. }
  185. _ledStatus(i, status);
  186. }
  187. if (_ledMode(i) == LED_MODE_STATUS) {
  188. bool status = false;
  189. for (unsigned char k=0; k<relayCount(); k++) {
  190. if (relayStatus(k)) {
  191. status = true;
  192. break;
  193. }
  194. }
  195. _ledStatus(i, status);
  196. }
  197. if (_ledMode(i) == LED_MODE_ON) {
  198. _ledStatus(i, true);
  199. }
  200. if (_ledMode(i) == LED_MODE_OFF) {
  201. _ledStatus(i, false);
  202. }
  203. }
  204. _led_update = false;
  205. }