mirror of
https://github.com/Threnklyn/esphome-dev.git
synced 2026-05-21 21:43:26 +02:00
Andrew Zaborowski
a62b6548d2
Apparently play()/stop() etc. are not meant to be called directly by users of the class and if they're called directly that would not give the expected result for the classes that have an empty play(). Make all methods except play_complex, stop_comples and is_running protected. While there also make RemoteTransmitterActionBase::encode protected.
87 lines
2.4 KiB
C++
87 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/preferences.h"
|
|
#include "esphome/core/automation.h"
|
|
#include "esphome/components/sensor/sensor.h"
|
|
|
|
namespace esphome {
|
|
namespace integration {
|
|
|
|
enum IntegrationSensorTime {
|
|
INTEGRATION_SENSOR_TIME_MILLISECOND = 0,
|
|
INTEGRATION_SENSOR_TIME_SECOND,
|
|
INTEGRATION_SENSOR_TIME_MINUTE,
|
|
INTEGRATION_SENSOR_TIME_HOUR,
|
|
INTEGRATION_SENSOR_TIME_DAY,
|
|
};
|
|
|
|
enum IntegrationMethod {
|
|
INTEGRATION_METHOD_TRAPEZOID = 0,
|
|
INTEGRATION_METHOD_LEFT,
|
|
INTEGRATION_METHOD_RIGHT,
|
|
};
|
|
|
|
class IntegrationSensor : public sensor::Sensor, public Component {
|
|
public:
|
|
void setup() override;
|
|
void dump_config() override;
|
|
float get_setup_priority() const override { return setup_priority::DATA; }
|
|
void set_sensor(Sensor *sensor) { sensor_ = sensor; }
|
|
void set_time(IntegrationSensorTime time) { time_ = time; }
|
|
void set_method(IntegrationMethod method) { method_ = method; }
|
|
void set_restore(bool restore) { restore_ = restore; }
|
|
void reset() { this->publish_and_save_(0.0f); }
|
|
|
|
protected:
|
|
void process_sensor_value_(float value);
|
|
float get_time_factor_() {
|
|
switch (this->time_) {
|
|
case INTEGRATION_SENSOR_TIME_MILLISECOND:
|
|
return 1.0f;
|
|
case INTEGRATION_SENSOR_TIME_SECOND:
|
|
return 1.0f / 1000.0f;
|
|
case INTEGRATION_SENSOR_TIME_MINUTE:
|
|
return 1.0f / 60000.0f;
|
|
case INTEGRATION_SENSOR_TIME_HOUR:
|
|
return 1.0f / 3600000.0f;
|
|
case INTEGRATION_SENSOR_TIME_DAY:
|
|
return 1.0f / 86400000.0f;
|
|
default:
|
|
return 0.0f;
|
|
}
|
|
}
|
|
void publish_and_save_(double result) {
|
|
this->result_ = result;
|
|
this->publish_state(result);
|
|
float result_f = result;
|
|
this->rtc_.save(&result_f);
|
|
}
|
|
std::string unit_of_measurement() override;
|
|
std::string icon() override { return this->sensor_->get_icon(); }
|
|
int8_t accuracy_decimals() override { return this->sensor_->get_accuracy_decimals() + 2; }
|
|
|
|
sensor::Sensor *sensor_;
|
|
IntegrationSensorTime time_;
|
|
IntegrationMethod method_;
|
|
bool restore_;
|
|
ESPPreferenceObject rtc_;
|
|
|
|
uint32_t last_update_;
|
|
double result_{0.0f};
|
|
float last_value_{0.0f};
|
|
};
|
|
|
|
template<typename... Ts> class ResetAction : public Action<Ts...> {
|
|
public:
|
|
explicit ResetAction(IntegrationSensor *parent) : parent_(parent) {}
|
|
|
|
protected:
|
|
void play_(Ts... x) override { this->parent_->reset(); }
|
|
|
|
IntegrationSensor *parent_;
|
|
};
|
|
|
|
} // namespace integration
|
|
} // namespace esphome
|