mirror of
https://github.com/esphome/esphome.git
synced 2026-03-04 11:48:21 -07:00
ac0d921413
* Socket refactor and SSL * esp-idf temp * Fixes * Echo component and noise * Add noise API transport support * Updates * ESP-IDF * Complete * Fixes * Fixes * Versions update * New i2c APIs * Complete i2c refactor * SPI migration * Revert ESP Preferences migration, too complex for now * OTA support * Remove echo again * Remove ssl again * GPIOFlags updates * Rename esphal and ICACHE_RAM_ATTR * Make ESP32 arduino compilable again * Fix GPIO flags * Complete pin registry refactor and fixes * Fixes to make test1 compile * Remove sdkconfig file * Ignore sdkconfig file * Fixes in reviewing * Make test2 compile * Make test4 compile * Make test5 compile * Run clang-format * Fix lint errors * Use esp-idf APIs instead of btStart * Another round of fixes * Start implementing ESP8266 * Make test3 compile * Guard esp8266 code * Lint * Reformat * Fixes * Fixes v2 * more fixes * ESP-IDF tidy target * Convert ARDUINO_ARCH_ESPxx * Update WiFiSignalSensor * Update time ifdefs * OTA needs millis from hal * RestartSwitch needs delay from hal * ESP-IDF Uart * Fix OTA blank password * Allow setting sdkconfig * Fix idf partitions and allow setting sdkconfig from yaml * Re-add read/write compat APIs and fix esp8266 uart * Fix esp8266 store log strings in flash * Fix ESP32 arduino preferences not initialized * Update ifdefs * Change how sdkconfig change is detected * Add checks to ci-custom and fix them * Run clang-format * Add esp-idf clang-tidy target and fix errors * Fixes from clang-tidy idf round 2 * Fixes from compiling tests with esp-idf * Run clang-format * Switch test5.yaml to esp-idf * Implement ESP8266 Preferences * Lint * Re-do PIO package version selection a bit * Fix arduinoespressif32 package version * Fix unit tests * Lint * Lint fixes * Fix readv/writev not defined * Fix graphing component * Re-add all old options from core/config.py Co-authored-by: Jesse Hills <3060199+jesserockz@users.noreply.github.com>
111 lines
2.6 KiB
C++
111 lines
2.6 KiB
C++
#ifdef USE_ARDUINO
|
|
|
|
#include "e131.h"
|
|
#include "e131_addressable_light_effect.h"
|
|
#include "esphome/core/log.h"
|
|
|
|
#ifdef USE_ESP32
|
|
#include <WiFi.h>
|
|
#endif
|
|
|
|
#ifdef USE_ESP8266
|
|
#include <ESP8266WiFi.h>
|
|
#include <WiFiUdp.h>
|
|
#endif
|
|
|
|
namespace esphome {
|
|
namespace e131 {
|
|
|
|
static const char *const TAG = "e131";
|
|
static const int PORT = 5568;
|
|
|
|
E131Component::E131Component() {}
|
|
|
|
E131Component::~E131Component() {
|
|
if (udp_) {
|
|
udp_->stop();
|
|
}
|
|
}
|
|
|
|
void E131Component::setup() {
|
|
udp_ = make_unique<WiFiUDP>();
|
|
|
|
if (!udp_->begin(PORT)) {
|
|
ESP_LOGE(TAG, "Cannot bind E131 to %d.", PORT);
|
|
mark_failed();
|
|
return;
|
|
}
|
|
|
|
join_igmp_groups_();
|
|
}
|
|
|
|
void E131Component::loop() {
|
|
std::vector<uint8_t> payload;
|
|
E131Packet packet;
|
|
int universe = 0;
|
|
|
|
while (uint16_t packet_size = udp_->parsePacket()) {
|
|
payload.resize(packet_size);
|
|
|
|
if (!udp_->read(&payload[0], payload.size())) {
|
|
continue;
|
|
}
|
|
|
|
if (!packet_(payload, universe, packet)) {
|
|
ESP_LOGV(TAG, "Invalid packet received of size %zu.", payload.size());
|
|
continue;
|
|
}
|
|
|
|
if (!process_(universe, packet)) {
|
|
ESP_LOGV(TAG, "Ignored packet for %d universe of size %d.", universe, packet.count);
|
|
}
|
|
}
|
|
}
|
|
|
|
void E131Component::add_effect(E131AddressableLightEffect *light_effect) {
|
|
if (light_effects_.count(light_effect)) {
|
|
return;
|
|
}
|
|
|
|
ESP_LOGD(TAG, "Registering '%s' for universes %d-%d.", light_effect->get_name().c_str(),
|
|
light_effect->get_first_universe(), light_effect->get_last_universe());
|
|
|
|
light_effects_.insert(light_effect);
|
|
|
|
for (auto universe = light_effect->get_first_universe(); universe <= light_effect->get_last_universe(); ++universe) {
|
|
join_(universe);
|
|
}
|
|
}
|
|
|
|
void E131Component::remove_effect(E131AddressableLightEffect *light_effect) {
|
|
if (!light_effects_.count(light_effect)) {
|
|
return;
|
|
}
|
|
|
|
ESP_LOGD(TAG, "Unregistering '%s' for universes %d-%d.", light_effect->get_name().c_str(),
|
|
light_effect->get_first_universe(), light_effect->get_last_universe());
|
|
|
|
light_effects_.erase(light_effect);
|
|
|
|
for (auto universe = light_effect->get_first_universe(); universe <= light_effect->get_last_universe(); ++universe) {
|
|
leave_(universe);
|
|
}
|
|
}
|
|
|
|
bool E131Component::process_(int universe, const E131Packet &packet) {
|
|
bool handled = false;
|
|
|
|
ESP_LOGV(TAG, "Received E1.31 packet for %d universe, with %d bytes", universe, packet.count);
|
|
|
|
for (auto light_effect : light_effects_) {
|
|
handled = light_effect->process_(universe, packet) || handled;
|
|
}
|
|
|
|
return handled;
|
|
}
|
|
|
|
} // namespace e131
|
|
} // namespace esphome
|
|
|
|
#endif // USE_ARDUINO
|