mirror of
https://github.com/esphome/esphome.git
synced 2024-12-02 19:54:14 +01:00
67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/components/i2c/i2c.h"
|
|
#include "esphome/components/sensor/sensor.h"
|
|
|
|
namespace esphome {
|
|
namespace ade7953 {
|
|
|
|
class ADE7953 : public i2c::I2CDevice, public PollingComponent {
|
|
public:
|
|
void set_voltage_sensor(sensor::Sensor *voltage_sensor) { voltage_sensor_ = voltage_sensor; }
|
|
void set_current_a_sensor(sensor::Sensor *current_a_sensor) { current_a_sensor_ = current_a_sensor; }
|
|
void set_current_b_sensor(sensor::Sensor *current_b_sensor) { current_b_sensor_ = current_b_sensor; }
|
|
void set_active_power_a_sensor(sensor::Sensor *active_power_a_sensor) {
|
|
active_power_a_sensor_ = active_power_a_sensor;
|
|
}
|
|
void set_active_power_b_sensor(sensor::Sensor *active_power_b_sensor) {
|
|
active_power_b_sensor_ = active_power_b_sensor;
|
|
}
|
|
|
|
void setup() override {
|
|
this->set_timeout(100, [this]() {
|
|
this->ade_write_<uint8_t>(0x0010, 0x04);
|
|
this->ade_write_<uint8_t>(0x00FE, 0xAD);
|
|
this->ade_write_<uint16_t>(0x0120, 0x0030);
|
|
this->is_setup_ = true;
|
|
});
|
|
}
|
|
|
|
void dump_config() override;
|
|
|
|
void update() override;
|
|
|
|
protected:
|
|
template<typename T> bool ade_write_(uint16_t reg, T value) {
|
|
std::vector<uint8_t> data;
|
|
data.push_back(reg >> 8);
|
|
data.push_back(reg >> 0);
|
|
for (int i = sizeof(T) - 1; i >= 0; i--)
|
|
data.push_back(value >> (i * 8));
|
|
return this->write_bytes_raw(data);
|
|
}
|
|
template<typename T> optional<T> ade_read_(uint16_t reg) {
|
|
uint8_t hi = reg >> 8;
|
|
uint8_t lo = reg >> 0;
|
|
if (!this->write_bytes_raw({hi, lo}))
|
|
return {};
|
|
auto ret = this->read_bytes_raw<sizeof(T)>();
|
|
if (!ret.has_value())
|
|
return {};
|
|
T result = 0;
|
|
for (int i = 0, j = sizeof(T) - 1; i < sizeof(T); i++, j--)
|
|
result |= T((*ret)[i]) << (j * 8);
|
|
return result;
|
|
}
|
|
|
|
bool is_setup_{false};
|
|
sensor::Sensor *voltage_sensor_{nullptr};
|
|
sensor::Sensor *current_a_sensor_{nullptr};
|
|
sensor::Sensor *current_b_sensor_{nullptr};
|
|
sensor::Sensor *active_power_a_sensor_{nullptr};
|
|
sensor::Sensor *active_power_b_sensor_{nullptr};
|
|
};
|
|
|
|
} // namespace ade7953
|
|
} // namespace esphome
|