forked from ayufan/esphome-components
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Change to `E131AddressableLightEffect` - Update log messages - Add MONO mode of operation
- Loading branch information
Showing
7 changed files
with
182 additions
and
120 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,106 @@ | ||
#include "e131.h" | ||
#include "e131_addressable_light_effect.h" | ||
#include "esphome/core/log.h" | ||
|
||
#define MAX_DATA_SIZE (sizeof(e131_packet_t::property_values)-1) | ||
|
||
#define TAG "e131_light_effect" | ||
|
||
namespace esphome { | ||
namespace e131 { | ||
|
||
E131AddressableLightEffect::E131AddressableLightEffect(const std::string &name) | ||
: AddressableLightEffect(name) { | ||
} | ||
|
||
int E131AddressableLightEffect::get_data_per_universe() const { | ||
return get_lights_per_universe() * channels_; | ||
} | ||
|
||
int E131AddressableLightEffect::get_lights_per_universe() const { | ||
return MAX_DATA_SIZE / channels_; | ||
} | ||
|
||
int E131AddressableLightEffect::get_first_universe() const { | ||
return first_universe_; | ||
} | ||
|
||
int E131AddressableLightEffect::get_last_universe() const { | ||
return first_universe_ + get_universe_count() - 1; | ||
} | ||
|
||
int E131AddressableLightEffect::get_universe_count() const { | ||
// Round up to lights_per_universe | ||
auto lights = get_lights_per_universe(); | ||
return (get_addressable_()->size() + lights - 1) / lights; | ||
} | ||
|
||
void E131AddressableLightEffect::start() { | ||
AddressableLightEffect::start(); | ||
|
||
if (this->e131_) { | ||
this->e131_->add_effect(this); | ||
} | ||
} | ||
|
||
void E131AddressableLightEffect::stop() { | ||
if (this->e131_) { | ||
this->e131_->remove_effect(this); | ||
} | ||
|
||
AddressableLightEffect::stop(); | ||
} | ||
|
||
void E131AddressableLightEffect::apply(esphome::light::AddressableLight &it, const esphome::light::ESPColor ¤t_color) { | ||
// ignore, it is run by `::update()` | ||
} | ||
|
||
bool E131AddressableLightEffect::process(int universe, const e131_packet_t &packet) { | ||
auto it = get_addressable_(); | ||
|
||
// check if this is our universe and data are valid | ||
if (universe < first_universe_ || universe > get_last_universe()) | ||
return false; | ||
|
||
int output_offset = (universe - first_universe_) * get_lights_per_universe(); | ||
int output_end = std::min( | ||
output_offset + get_lights_per_universe(), | ||
it->size()); | ||
auto input_data = packet.property_values + 1; | ||
|
||
ESP_LOGV(TAG, "Applying data for '%s' on %d universe, for %d-%d.", | ||
get_name().c_str(), | ||
universe, | ||
output_offset, output_end); | ||
|
||
switch (channels_) { | ||
case E131_MONO: | ||
for ( ; output_offset < output_end; output_offset++, input_data++) { | ||
auto output = (*it)[output_offset]; | ||
output.set(esphome::light::ESPColor( | ||
input_data[0], input_data[0], input_data[0], input_data[0])); | ||
} | ||
break; | ||
|
||
case E131_RGB: | ||
for ( ; output_offset < output_end; output_offset++, input_data += 3) { | ||
auto output = (*it)[output_offset]; | ||
output.set(esphome::light::ESPColor( | ||
input_data[0], input_data[1], input_data[2], (input_data[0] + input_data[1] + input_data[2]) / 3)); | ||
} | ||
break; | ||
|
||
case E131_RGBW: | ||
for ( ; output_offset < output_end; output_offset++, input_data += 4) { | ||
auto output = (*it)[output_offset]; | ||
output.set(esphome::light::ESPColor( | ||
input_data[0], input_data[1], input_data[2], input_data[3])); | ||
} | ||
break; | ||
} | ||
|
||
return true; | ||
} | ||
|
||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.