20240410
This commit is contained in:
48
esphome/components/fastled_base/__init__.py
Normal file
48
esphome/components/fastled_base/__init__.py
Normal file
@@ -0,0 +1,48 @@
|
||||
import esphome.codegen as cg
|
||||
import esphome.config_validation as cv
|
||||
from esphome.components import light
|
||||
from esphome.const import (
|
||||
CONF_OUTPUT_ID,
|
||||
CONF_NUM_LEDS,
|
||||
CONF_RGB_ORDER,
|
||||
CONF_MAX_REFRESH_RATE,
|
||||
)
|
||||
|
||||
CODEOWNERS = ["@OttoWinter"]
|
||||
fastled_base_ns = cg.esphome_ns.namespace("fastled_base")
|
||||
FastLEDLightOutput = fastled_base_ns.class_(
|
||||
"FastLEDLightOutput", light.AddressableLight
|
||||
)
|
||||
|
||||
RGB_ORDERS = [
|
||||
"RGB",
|
||||
"RBG",
|
||||
"GRB",
|
||||
"GBR",
|
||||
"BRG",
|
||||
"BGR",
|
||||
]
|
||||
|
||||
BASE_SCHEMA = light.ADDRESSABLE_LIGHT_SCHEMA.extend(
|
||||
{
|
||||
cv.GenerateID(CONF_OUTPUT_ID): cv.declare_id(FastLEDLightOutput),
|
||||
cv.Required(CONF_NUM_LEDS): cv.positive_not_null_int,
|
||||
cv.Optional(CONF_RGB_ORDER): cv.one_of(*RGB_ORDERS, upper=True),
|
||||
cv.Optional(CONF_MAX_REFRESH_RATE): cv.positive_time_period_microseconds,
|
||||
}
|
||||
).extend(cv.COMPONENT_SCHEMA)
|
||||
|
||||
|
||||
async def new_fastled_light(config):
|
||||
var = cg.new_Pvariable(config[CONF_OUTPUT_ID])
|
||||
await cg.register_component(var, config)
|
||||
|
||||
if CONF_MAX_REFRESH_RATE in config:
|
||||
cg.add(var.set_max_refresh_rate(config[CONF_MAX_REFRESH_RATE]))
|
||||
|
||||
await light.register_light(var, config)
|
||||
# https://github.com/FastLED/FastLED/blob/master/library.json
|
||||
# 3.3.3 has an issue on ESP32 with RMT and fastled_clockless:
|
||||
# https://github.com/esphome/issues/issues/1375
|
||||
cg.add_library("fastled/FastLED", "3.5.0")
|
||||
return var
|
||||
43
esphome/components/fastled_base/fastled_light.cpp
Normal file
43
esphome/components/fastled_base/fastled_light.cpp
Normal file
@@ -0,0 +1,43 @@
|
||||
#ifdef USE_ARDUINO
|
||||
|
||||
#include "fastled_light.h"
|
||||
#include "esphome/core/log.h"
|
||||
|
||||
namespace esphome {
|
||||
namespace fastled_base {
|
||||
|
||||
static const char *const TAG = "fastled";
|
||||
|
||||
void FastLEDLightOutput::setup() {
|
||||
ESP_LOGCONFIG(TAG, "Setting up FastLED light...");
|
||||
this->controller_->init();
|
||||
this->controller_->setLeds(this->leds_, this->num_leds_);
|
||||
this->effect_data_ = new uint8_t[this->num_leds_]; // NOLINT
|
||||
if (!this->max_refresh_rate_.has_value()) {
|
||||
this->set_max_refresh_rate(this->controller_->getMaxRefreshRate());
|
||||
}
|
||||
}
|
||||
void FastLEDLightOutput::dump_config() {
|
||||
ESP_LOGCONFIG(TAG, "FastLED light:");
|
||||
ESP_LOGCONFIG(TAG, " Num LEDs: %u", this->num_leds_);
|
||||
ESP_LOGCONFIG(TAG, " Max refresh rate: %u", *this->max_refresh_rate_);
|
||||
}
|
||||
void FastLEDLightOutput::write_state(light::LightState *state) {
|
||||
// protect from refreshing too often
|
||||
uint32_t now = micros();
|
||||
if (*this->max_refresh_rate_ != 0 && (now - this->last_refresh_) < *this->max_refresh_rate_) {
|
||||
// try again next loop iteration, so that this change won't get lost
|
||||
this->schedule_show();
|
||||
return;
|
||||
}
|
||||
this->last_refresh_ = now;
|
||||
this->mark_shown_();
|
||||
|
||||
ESP_LOGVV(TAG, "Writing RGB values to bus...");
|
||||
this->controller_->showLeds();
|
||||
}
|
||||
|
||||
} // namespace fastled_base
|
||||
} // namespace esphome
|
||||
|
||||
#endif // USE_ARDUINO
|
||||
243
esphome/components/fastled_base/fastled_light.h
Normal file
243
esphome/components/fastled_base/fastled_light.h
Normal file
@@ -0,0 +1,243 @@
|
||||
#pragma once
|
||||
|
||||
#ifdef USE_ARDUINO
|
||||
|
||||
#include "esphome/core/component.h"
|
||||
#include "esphome/core/helpers.h"
|
||||
#include "esphome/components/light/addressable_light.h"
|
||||
|
||||
#define FASTLED_ESP8266_RAW_PIN_ORDER
|
||||
#define FASTLED_ESP32_RAW_PIN_ORDER
|
||||
#define FASTLED_RMT_BUILTIN_DRIVER true
|
||||
|
||||
// Avoid annoying compiler messages
|
||||
#define FASTLED_INTERNAL
|
||||
|
||||
#include "FastLED.h"
|
||||
|
||||
namespace esphome {
|
||||
namespace fastled_base {
|
||||
|
||||
class FastLEDLightOutput : public light::AddressableLight {
|
||||
public:
|
||||
/// Only for custom effects: Get the internal controller.
|
||||
CLEDController *get_controller() const { return this->controller_; }
|
||||
|
||||
inline int32_t size() const override { return this->num_leds_; }
|
||||
|
||||
/// Set a maximum refresh rate in µs as some lights do not like being updated too often.
|
||||
void set_max_refresh_rate(uint32_t interval_us) { this->max_refresh_rate_ = interval_us; }
|
||||
|
||||
/// Add some LEDS, can only be called once.
|
||||
CLEDController &add_leds(CLEDController *controller, int num_leds) {
|
||||
this->controller_ = controller;
|
||||
this->num_leds_ = num_leds;
|
||||
this->leds_ = new CRGB[num_leds]; // NOLINT
|
||||
|
||||
for (int i = 0; i < this->num_leds_; i++)
|
||||
this->leds_[i] = CRGB::Black;
|
||||
|
||||
return *this->controller_;
|
||||
}
|
||||
|
||||
template<ESPIChipsets CHIPSET, uint8_t DATA_PIN, uint8_t CLOCK_PIN, EOrder RGB_ORDER, uint32_t SPI_DATA_RATE>
|
||||
CLEDController &add_leds(int num_leds) {
|
||||
switch (CHIPSET) {
|
||||
case LPD8806: {
|
||||
static LPD8806Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case WS2801: {
|
||||
static WS2801Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case WS2803: {
|
||||
static WS2803Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case SM16716: {
|
||||
static SM16716Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case P9813: {
|
||||
static P9813Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case DOTSTAR:
|
||||
case APA102: {
|
||||
static APA102Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case SK9822: {
|
||||
static SK9822Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER, SPI_DATA_RATE> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template<ESPIChipsets CHIPSET, uint8_t DATA_PIN, uint8_t CLOCK_PIN> CLEDController &add_leds(int num_leds) {
|
||||
switch (CHIPSET) {
|
||||
case LPD8806: {
|
||||
static LPD8806Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case WS2801: {
|
||||
static WS2801Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case WS2803: {
|
||||
static WS2803Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case SM16716: {
|
||||
static SM16716Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case P9813: {
|
||||
static P9813Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case DOTSTAR:
|
||||
case APA102: {
|
||||
static APA102Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case SK9822: {
|
||||
static SK9822Controller<DATA_PIN, CLOCK_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template<ESPIChipsets CHIPSET, uint8_t DATA_PIN, uint8_t CLOCK_PIN, EOrder RGB_ORDER>
|
||||
CLEDController &add_leds(int num_leds) {
|
||||
switch (CHIPSET) {
|
||||
case LPD8806: {
|
||||
static LPD8806Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case WS2801: {
|
||||
static WS2801Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case WS2803: {
|
||||
static WS2803Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case SM16716: {
|
||||
static SM16716Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case P9813: {
|
||||
static P9813Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case DOTSTAR:
|
||||
case APA102: {
|
||||
static APA102Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
case SK9822: {
|
||||
static SK9822Controller<DATA_PIN, CLOCK_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef FASTLED_HAS_CLOCKLESS
|
||||
template<template<uint8_t DATA_PIN, EOrder RGB_ORDER> class CHIPSET, uint8_t DATA_PIN, EOrder RGB_ORDER>
|
||||
CLEDController &add_leds(int num_leds) {
|
||||
static CHIPSET<DATA_PIN, RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
|
||||
template<template<uint8_t DATA_PIN, EOrder RGB_ORDER> class CHIPSET, uint8_t DATA_PIN>
|
||||
CLEDController &add_leds(int num_leds) {
|
||||
static CHIPSET<DATA_PIN, RGB> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
|
||||
template<template<uint8_t DATA_PIN> class CHIPSET, uint8_t DATA_PIN> CLEDController &add_leds(int num_leds) {
|
||||
static CHIPSET<DATA_PIN> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
#endif
|
||||
|
||||
template<template<EOrder RGB_ORDER> class CHIPSET, EOrder RGB_ORDER> CLEDController &add_leds(int num_leds) {
|
||||
static CHIPSET<RGB_ORDER> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
|
||||
template<template<EOrder RGB_ORDER> class CHIPSET> CLEDController &add_leds(int num_leds) {
|
||||
static CHIPSET<RGB> controller;
|
||||
return add_leds(&controller, num_leds);
|
||||
}
|
||||
|
||||
#ifdef FASTLED_HAS_BLOCKLESS
|
||||
template<EBlockChipsets CHIPSET, int NUM_LANES, EOrder RGB_ORDER> CLEDController &add_leds(int num_leds) {
|
||||
switch (CHIPSET) {
|
||||
#ifdef PORTA_FIRST_PIN
|
||||
case WS2811_PORTA:
|
||||
return add_leds(
|
||||
new InlineBlockClocklessController<NUM_LANES, PORTA_FIRST_PIN, NS(320), NS(320), NS(640), RGB_ORDER>(),
|
||||
num_leds);
|
||||
case WS2811_400_PORTA:
|
||||
return add_leds(
|
||||
new InlineBlockClocklessController<NUM_LANES, PORTA_FIRST_PIN, NS(800), NS(800), NS(900), RGB_ORDER>(),
|
||||
num_leds);
|
||||
case WS2813_PORTA:
|
||||
return add_leds(new InlineBlockClocklessController<NUM_LANES, PORTA_FIRST_PIN, NS(320), NS(320), NS(640),
|
||||
RGB_ORDER, 0, false, 300>(),
|
||||
num_leds);
|
||||
case TM1803_PORTA:
|
||||
return add_leds(
|
||||
new InlineBlockClocklessController<NUM_LANES, PORTA_FIRST_PIN, NS(700), NS(1100), NS(700), RGB_ORDER>(),
|
||||
num_leds);
|
||||
case UCS1903_PORTA:
|
||||
return add_leds(
|
||||
new InlineBlockClocklessController<NUM_LANES, PORTA_FIRST_PIN, NS(500), NS(1500), NS(500), RGB_ORDER>(),
|
||||
num_leds);
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
template<EBlockChipsets CHIPSET, int NUM_LANES> CLEDController &add_leds(int num_leds) {
|
||||
return add_leds<CHIPSET, NUM_LANES, GRB>(num_leds);
|
||||
}
|
||||
#endif
|
||||
|
||||
// ========== INTERNAL METHODS ==========
|
||||
// (In most use cases you won't need these)
|
||||
light::LightTraits get_traits() override {
|
||||
auto traits = light::LightTraits();
|
||||
traits.set_supported_color_modes({light::ColorMode::RGB});
|
||||
return traits;
|
||||
}
|
||||
void setup() override;
|
||||
void dump_config() override;
|
||||
void write_state(light::LightState *state) override;
|
||||
float get_setup_priority() const override { return setup_priority::HARDWARE; }
|
||||
|
||||
void clear_effect_data() override {
|
||||
for (int i = 0; i < this->size(); i++)
|
||||
this->effect_data_[i] = 0;
|
||||
}
|
||||
|
||||
protected:
|
||||
light::ESPColorView get_view_internal(int32_t index) const override {
|
||||
return {&this->leds_[index].r, &this->leds_[index].g, &this->leds_[index].b, nullptr,
|
||||
&this->effect_data_[index], &this->correction_};
|
||||
}
|
||||
|
||||
CLEDController *controller_{nullptr};
|
||||
CRGB *leds_{nullptr};
|
||||
uint8_t *effect_data_{nullptr};
|
||||
int num_leds_{0};
|
||||
uint32_t last_refresh_{0};
|
||||
optional<uint32_t> max_refresh_rate_{};
|
||||
};
|
||||
|
||||
} // namespace fastled_base
|
||||
} // namespace esphome
|
||||
|
||||
#endif // USE_ARDUINO
|
||||
Reference in New Issue
Block a user