e131.h

Go to the documentation of this file.

#pragma once
#include "esphome/core/defines.h"
#ifdef USE_NETWORK
#include "esphome/components/socket/socket.h"
#include "esphome/core/component.h"
 
#include <cinttypes>
#include <memory>
#include <vector>
 
namespace esphome::e131 {
 
class E131AddressableLightEffect;
 
enum E131ListenMethod { E131_MULTICAST, E131_UNICAST };
 
const int E131_MAX_PROPERTY_VALUES_COUNT = 513;
 
struct E131Packet {
  uint16_t count;
  uint8_t values[E131_MAX_PROPERTY_VALUES_COUNT];
};
 
struct UniverseConsumer {
  uint16_t universe;
  uint16_t consumers;
};
 
class E131Component final : public esphome::Component {
 public:
  E131Component();
  ~E131Component();
 
  void setup() override;
  void loop() override;
  float get_setup_priority() const override { return setup_priority::AFTER_WIFI; }
 
  void add_effect(E131AddressableLightEffect *light_effect);
  void remove_effect(E131AddressableLightEffect *light_effect);
 
  void set_method(E131ListenMethod listen_method) { this->listen_method_ = listen_method; }
 
 protected:
  bool packet_(const uint8_t *data, size_t len, int &universe, E131Packet &packet);
  bool process_(int universe, const E131Packet &packet);
  void join_igmp_groups_();
  void join_universe_(uint16_t universe);
  UniverseConsumer *find_universe_(int universe);
  void join_(int universe);
  void leave_(int universe);
 
  E131ListenMethod listen_method_{E131_MULTICAST};
  bool igmp_join_pending_{false};
  std::unique_ptr<socket::UDPSocket> socket_;
  std::vector<E131AddressableLightEffect *> light_effects_;
  std::vector<UniverseConsumer> universe_consumers_;
};
 
}  // namespace esphome::e131
 
#endif