mirror of https://github.com/esphome/esphome.git
73 lines
2.5 KiB
C++
73 lines
2.5 KiB
C++
#include <vector>
|
|
#include "modbus_number.h"
|
|
#include "esphome/core/log.h"
|
|
|
|
namespace esphome {
|
|
namespace modbus_controller {
|
|
|
|
static const char *const TAG = "modbus.number";
|
|
|
|
void ModbusNumber::parse_and_publish(const std::vector<uint8_t> &data) {
|
|
float result = payload_to_float(data, *this);
|
|
|
|
// Is there a lambda registered
|
|
// call it with the pre converted value and the raw data array
|
|
if (this->transform_func_.has_value()) {
|
|
// the lambda can parse the response itself
|
|
auto val = (*this->transform_func_)(this, result, data);
|
|
if (val.has_value()) {
|
|
ESP_LOGV(TAG, "Value overwritten by lambda");
|
|
result = val.value();
|
|
}
|
|
}
|
|
ESP_LOGD(TAG, "Number new state : %.02f", result);
|
|
// this->sensor_->raw_state = result;
|
|
this->publish_state(result);
|
|
}
|
|
|
|
void ModbusNumber::control(float value) {
|
|
std::vector<uint16_t> data;
|
|
// Is there are lambda configured?
|
|
if (this->write_transform_func_.has_value()) {
|
|
// data is passed by reference
|
|
// the lambda can fill the empty vector directly
|
|
// in that case the return value is ignored
|
|
auto val = (*this->write_transform_func_)(this, value, data);
|
|
if (val.has_value()) {
|
|
ESP_LOGV(TAG, "Value overwritten by lambda");
|
|
value = val.value();
|
|
} else {
|
|
ESP_LOGV(TAG, "Communication handled by lambda - exiting control");
|
|
return;
|
|
}
|
|
} else {
|
|
value = multiply_by_ * value;
|
|
}
|
|
|
|
// lambda didn't set payload
|
|
if (data.empty()) {
|
|
data = float_to_payload(value, this->sensor_value_type);
|
|
}
|
|
|
|
ESP_LOGD(TAG,
|
|
"Updating register: connected Sensor=%s start address=0x%X register count=%d new value=%.02f (val=%.02f)",
|
|
this->get_name().c_str(), this->start_address, this->register_count, value, value);
|
|
|
|
// Create and send the write command
|
|
auto write_cmd = ModbusCommandItem::create_write_multiple_command(parent_, this->start_address + this->offset,
|
|
this->register_count, data);
|
|
|
|
// publish new value
|
|
write_cmd.on_data_func = [this, write_cmd, value](ModbusRegisterType register_type, uint16_t start_address,
|
|
const std::vector<uint8_t> &data) {
|
|
// gets called when the write command is ack'd from the device
|
|
parent_->on_write_register_response(write_cmd.register_type, start_address, data);
|
|
this->publish_state(value);
|
|
};
|
|
parent_->queue_command(write_cmd);
|
|
}
|
|
void ModbusNumber::dump_config() { LOG_NUMBER(TAG, "Modbus Number", this); }
|
|
|
|
} // namespace modbus_controller
|
|
} // namespace esphome
|