mirror of
https://github.com/esphome/esphome.git
synced 2024-12-04 20:48:21 +01:00
fc01a70b65
* Hbridge Christmas light component Can be used for Christmas lights that use 2 wires to run 2 different strings of lights using a hbridge driver. * Add Test NOTE: I am unable to test this via the docker image * Update hbridge_light_output.h * Update hbridge_light_output.h * Update hbridge_light_output.h * Update light.py * Fixed duty as white value bug fixed * lint changes * Name case change * thanks lint
76 lines
2.4 KiB
C++
76 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/components/output/float_output.h"
|
|
#include "esphome/components/light/light_output.h"
|
|
#include "esphome/core/log.h"
|
|
|
|
namespace esphome {
|
|
namespace hbridge {
|
|
|
|
// Using PollingComponent as the updates are more consistent and reduces flickering
|
|
class HBridgeLightOutput : public PollingComponent, public light::LightOutput {
|
|
public:
|
|
HBridgeLightOutput() : PollingComponent(1) {}
|
|
|
|
void set_pina_pin(output::FloatOutput *pina_pin) { pina_pin_ = pina_pin; }
|
|
void set_pinb_pin(output::FloatOutput *pinb_pin) { pinb_pin_ = pinb_pin; }
|
|
|
|
light::LightTraits get_traits() override {
|
|
auto traits = light::LightTraits();
|
|
traits.set_supports_brightness(true); // Dimming
|
|
traits.set_supports_rgb(false);
|
|
traits.set_supports_rgb_white_value(true); // hbridge color
|
|
traits.set_supports_color_temperature(false);
|
|
return traits;
|
|
}
|
|
|
|
void setup() override { this->forward_direction_ = false; }
|
|
|
|
void update() override {
|
|
// This method runs around 60 times per second
|
|
// We cannot do the PWM ourselves so we are reliant on the hardware PWM
|
|
if (!this->forward_direction_) { // First LED Direction
|
|
this->pinb_pin_->set_level(this->duty_off_);
|
|
this->pina_pin_->set_level(this->pina_duty_);
|
|
this->forward_direction_ = true;
|
|
} else { // Second LED Direction
|
|
this->pina_pin_->set_level(this->duty_off_);
|
|
this->pinb_pin_->set_level(this->pinb_duty_);
|
|
this->forward_direction_ = false;
|
|
}
|
|
}
|
|
|
|
float get_setup_priority() const override { return setup_priority::HARDWARE; }
|
|
|
|
void write_state(light::LightState *state) override {
|
|
float bright;
|
|
state->current_values_as_brightness(&bright);
|
|
|
|
state->set_gamma_correct(0);
|
|
float red, green, blue, white;
|
|
state->current_values_as_rgbw(&red, &green, &blue, &white);
|
|
|
|
if ((white / bright) > 0.55) {
|
|
this->pina_duty_ = (bright * (1 - (white / bright)));
|
|
this->pinb_duty_ = bright;
|
|
} else if (white < 0.45) {
|
|
this->pina_duty_ = bright;
|
|
this->pinb_duty_ = white;
|
|
} else {
|
|
this->pina_duty_ = bright;
|
|
this->pinb_duty_ = bright;
|
|
}
|
|
}
|
|
|
|
protected:
|
|
output::FloatOutput *pina_pin_;
|
|
output::FloatOutput *pinb_pin_;
|
|
float pina_duty_ = 0;
|
|
float pinb_duty_ = 0;
|
|
float duty_off_ = 0;
|
|
bool forward_direction_ = false;
|
|
};
|
|
|
|
} // namespace hbridge
|
|
} // namespace esphome
|