d2c7afeef0
* Add PZEM004T Support * Don't flush as much * Update pzem004t.cpp * Add generalized modbus * Add PZEMAC * Add PZEMDC * Fix file modes * Lint * Fix * Fix * Add check_uart_settings
52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/components/uart/uart.h"
|
|
|
|
namespace esphome {
|
|
namespace modbus {
|
|
|
|
class ModbusDevice;
|
|
|
|
class Modbus : public uart::UARTDevice, public Component {
|
|
public:
|
|
Modbus() = default;
|
|
|
|
void loop() override;
|
|
|
|
void dump_config() override;
|
|
|
|
void register_device(ModbusDevice *device) { this->devices_.push_back(device); }
|
|
|
|
float get_setup_priority() const override;
|
|
|
|
void send(uint8_t address, uint8_t function, uint16_t start_address, uint16_t register_count);
|
|
|
|
protected:
|
|
bool parse_modbus_byte_(uint8_t byte);
|
|
|
|
std::vector<uint8_t> rx_buffer_;
|
|
uint32_t last_modbus_byte_{0};
|
|
std::vector<ModbusDevice *> devices_;
|
|
};
|
|
|
|
class ModbusDevice {
|
|
public:
|
|
void set_parent(Modbus *parent) { parent_ = parent; }
|
|
void set_address(uint8_t address) { address_ = address; }
|
|
virtual void on_modbus_data(const std::vector<uint8_t> &data) = 0;
|
|
|
|
void send(uint8_t function, uint16_t start_address, uint16_t register_count) {
|
|
this->parent_->send(this->address_, function, start_address, register_count);
|
|
}
|
|
|
|
protected:
|
|
friend Modbus;
|
|
|
|
Modbus *parent_;
|
|
uint8_t address_;
|
|
};
|
|
|
|
} // namespace modbus
|
|
} // namespace esphome
|