one more step closer, testing now

This commit is contained in:
Daniël Koek 2024-09-25 17:42:59 +01:00
parent 68309a17c7
commit 361903f40c
2 changed files with 95 additions and 91 deletions

View file

@ -1,6 +1,19 @@
#include "ebyte_lora_component.h" #include "ebyte_lora_component.h"
namespace esphome { namespace esphome {
namespace ebyte_lora { namespace ebyte_lora {
void IRAM_ATTR HOT EbyteAuxStore::gpio_intr(EbyteAuxStore *arg) {
const bool can_send = arg->pin.digital_read();
if (can_send == arg->can_send)
return;
arg->can_send = can_send;
const uint32_t now = micros();
if (!can_send)
arg->on_time += now - arg->last_interrupt;
arg->last_interrupt = now;
}
union FuData { union FuData {
uint32_t u32; uint32_t u32;
float f32; float f32;
@ -237,21 +250,29 @@ void EbyteLoraComponent::update() {
this->get_current_config_(); this->get_current_config_();
return; return;
} else { } else {
if (!this->check_config_()) { // if it already set just continue quickly!
ESP_LOGD(TAG, "Config is not right, changing it now"); if (!this->config_checked_) {
this->set_config_(); this->config_checked_ = this->check_config_();
if (!this->config_checked_) {
ESP_LOGD(TAG, "Config is not right, changing it now");
this->set_config_();
}
} }
} }
if (this->get_mode_() != NORMAL) { if (this->config_mode_ != NORMAL) {
ESP_LOGD(TAG, "Mode is not set right"); this->config_mode_ = this->get_mode_();
this->set_mode_(NORMAL);
}
if (this->sent_switch_state_) if (this->config_mode_ != NORMAL) {
this->send_data_(true); ESP_LOGD(TAG, "Mode is not set right");
// we always request repeater info, since nodes will response too that they are around this->set_mode_(NORMAL);
// you can see it more of a health info }
this->request_repeater_info_(); }
this->updated_ = true;
auto now = millis() / 1000;
if (this->last_key_time_ + this->repeater_request_recyle_time_ < now) {
this->resend_repeater_request_ = true;
this->last_key_time_ = now;
}
} }
void EbyteLoraComponent::set_config_() { void EbyteLoraComponent::set_config_() {
uint8_t data[11]; uint8_t data[11];
@ -281,7 +302,6 @@ void EbyteLoraComponent::set_config_() {
data[10] = 0; data[10] = 0;
this->set_mode_(CONFIGURATION); this->set_mode_(CONFIGURATION);
this->write_array(data, sizeof(data)); this->write_array(data, sizeof(data));
this->setup_wait_response_(5000);
} }
void EbyteLoraComponent::setup() { void EbyteLoraComponent::setup() {
#ifdef USE_SENSOR #ifdef USE_SENSOR
@ -308,6 +328,8 @@ void EbyteLoraComponent::setup() {
this->should_send_ |= !this->binary_sensors_.empty(); this->should_send_ |= !this->binary_sensors_.empty();
#endif #endif
this->pin_aux_->setup(); this->pin_aux_->setup();
this->pin_aux_->attach_interrupt(EbyteAuxStore::gpio_intr, &this->store_, gpio::INTERRUPT_ANY_EDGE);
this->store_.can_send = false;
this->pin_m0_->setup(); this->pin_m0_->setup();
this->pin_m1_->setup(); this->pin_m1_->setup();
this->get_current_config_(); this->get_current_config_();
@ -321,7 +343,8 @@ void EbyteLoraComponent::get_current_config_() {
} }
ModeType EbyteLoraComponent::get_mode_() { ModeType EbyteLoraComponent::get_mode_() {
ModeType internal_mode = MODE_INIT; ModeType internal_mode = MODE_INIT;
if (!this->can_send_message_()) { if (!this->can_send) {
ESP_LOGD(TAG, "Can't sent it right now");
return internal_mode; return internal_mode;
} }
@ -343,15 +366,16 @@ ModeType EbyteLoraComponent::get_mode_() {
// ESP_LOGD(TAG, "MODE Conf!"); // ESP_LOGD(TAG, "MODE Conf!");
internal_mode = CONFIGURATION; internal_mode = CONFIGURATION;
} }
if (internal_mode != this->mode_) { if (internal_mode != this->config_mode_) {
ESP_LOGD(TAG, "Modes are not equal, calling the set function!! , checked: %u, expected: %u", internal_mode, ESP_LOGD(TAG, "Modes are not equal, calling the set function!! , checked: %u, expected: %u", internal_mode,
this->mode_); this->config_mode_);
this->set_mode_(internal_mode); this->set_mode_(internal_mode);
} }
return internal_mode; return internal_mode;
} }
void EbyteLoraComponent::set_mode_(ModeType mode) { void EbyteLoraComponent::set_mode_(ModeType mode) {
if (!this->can_send_message_()) { if (!this->can_send) {
ESP_LOGD(TAG, "Can't sent it right now");
return; return;
} }
if (this->pin_m0_ == nullptr || this->pin_m1_ == nullptr) { if (this->pin_m0_ == nullptr || this->pin_m1_ == nullptr) {
@ -387,71 +411,24 @@ void EbyteLoraComponent::set_mode_(ModeType mode) {
break; break;
} }
} }
// wait until aux pin goes back low this->config_mode_ = mode;
this->setup_wait_response_(1000);
this->mode_ = mode;
ESP_LOGD(TAG, "Mode is going to be set"); ESP_LOGD(TAG, "Mode is going to be set");
} }
bool EbyteLoraComponent::can_send_message_() {
// High means no more information is needed
if (this->pin_aux_->digital_read()) {
if (!(this->starting_to_check_ == 0) && !(this->time_out_after_ == 0)) {
this->starting_to_check_ = 0;
this->time_out_after_ = 0;
this->flush();
ESP_LOGD(TAG, "Aux pin is High! Can send again!");
}
return true;
} else {
// it has taken too long to complete, error out!
if ((millis() - this->starting_to_check_) > this->time_out_after_) {
ESP_LOGD(TAG, "Timeout error! Resetting timers");
this->starting_to_check_ = 0;
this->time_out_after_ = 0;
}
return false;
}
}
void EbyteLoraComponent::setup_wait_response_(uint32_t timeout) {
if (!(this->starting_to_check_ == 0) && !(this->time_out_after_ == 0)) {
ESP_LOGD(TAG, "Wait response already set!! %u", timeout);
}
ESP_LOGD(TAG, "Setting a timer for %u", timeout);
this->starting_to_check_ = millis();
this->time_out_after_ = timeout;
}
void EbyteLoraComponent::dump_config() { void EbyteLoraComponent::dump_config() {
ESP_LOGCONFIG(TAG, "Ebyte Lora E220:"); ESP_LOGCONFIG(TAG, "Ebyte Lora E220:");
ESP_LOGCONFIG(TAG, " Network id: %u", this->network_id_); ESP_LOGCONFIG(TAG, " Network id: %u", this->network_id_);
if (!this->repeater_enabled_ && !this->sent_switch_state_) {
ESP_LOGCONFIG(TAG, " Normal mode");
}
if (this->repeater_enabled_) { if (this->repeater_enabled_) {
ESP_LOGCONFIG(TAG, " Repeater mode"); ESP_LOGCONFIG(TAG, " Repeater mode");
} } else {
if (this->sent_switch_state_) { ESP_LOGCONFIG(TAG, " Normal mode");
ESP_LOGCONFIG(TAG, " Remote switch mode");
} }
}; };
void EbyteLoraComponent::process_(uint8_t *buf, const size_t len) { void EbyteLoraComponent::process_(uint8_t *buf, const size_t len) {
uint8_t *start_ptr = buf; uint8_t *start_ptr = buf;
uint8_t byte; uint8_t byte;
const uint8_t *end = buf + len; // -1 cause the last one is always RSSI
const uint8_t *end = buf + len - 1;
FuData rdata{}; FuData rdata{};
// switch (buf[0]) {
// case REQUEST_REPEATER_INFO:
// ESP_LOGD(TAG, "Got request for repeater info from network id %u", buf[1]);
// this->send_repeater_info_();
// break;
// case REPEATER_INFO:
// ESP_LOGD(TAG, "Got some repeater info from network %u ", buf[2]);
// break;
// case SWITCH_INFO:
// if (this->should_send_) {
// this->repeat_message_(buf);
// }
#ifdef USE_SENSOR #ifdef USE_SENSOR
auto &sensors = this->remote_sensors_[network_id_]; auto &sensors = this->remote_sensors_[network_id_];
#endif #endif
@ -459,6 +436,16 @@ void EbyteLoraComponent::process_(uint8_t *buf, const size_t len) {
auto &binary_sensors = this->remote_binary_sensors_[network_id_]; auto &binary_sensors = this->remote_binary_sensors_[network_id_];
#endif #endif
while (buf < end) { while (buf < end) {
if (byte == REQUEST_REPEATER_INFO) {
ESP_LOGD(TAG, "Got request for repeater info from network id %u", buf[1]);
this->send_repeater_info_();
break;
}
if (byte == REPEATER_INFO) {
ESP_LOGD(TAG, "Got some repeater info from network %u setting rssi next", buf[2]);
break;
}
if (byte == PROGRAM_CONF) { if (byte == PROGRAM_CONF) {
ESP_LOGD(TAG, "GOT PROGRAM_CONF"); ESP_LOGD(TAG, "GOT PROGRAM_CONF");
this->setup_conf_(buf); this->setup_conf_(buf);
@ -521,11 +508,21 @@ void EbyteLoraComponent::process_(uint8_t *buf, const size_t len) {
} }
}; };
void EbyteLoraComponent::loop() { void EbyteLoraComponent::loop() {
this->store_.can_send = this->can_send;
if (auto len = this->available()) { if (auto len = this->available()) {
uint8_t buf[len]; uint8_t buf[len];
this->read_array(buf, len); this->read_array(buf, len);
if (this->repeater_enabled_) {
this->repeat_message_(buf);
}
this->process_(buf, len); this->process_(buf, len);
} }
if (this->resend_repeater_request_)
this->request_repeater_info_();
if (this->updated_) {
this->send_data_(true);
}
} }
void EbyteLoraComponent::setup_conf_(uint8_t const *conf) { void EbyteLoraComponent::setup_conf_(uint8_t const *conf) {
ESP_LOGD(TAG, "Config set"); ESP_LOGD(TAG, "Config set");
@ -554,7 +551,8 @@ void EbyteLoraComponent::setup_conf_(uint8_t const *conf) {
this->current_config_.enable_rssi = (conf[8] >> 7) & 0b1; this->current_config_.enable_rssi = (conf[8] >> 7) & 0b1;
} }
void EbyteLoraComponent::send_data_(bool all) { void EbyteLoraComponent::send_data_(bool all) {
if (!this->can_send_message_()) { if (!this->can_send) {
ESP_LOGD(TAG, "Can't sent it right now");
return; return;
} }
std::vector<uint8_t> data; std::vector<uint8_t> data;
@ -593,11 +591,11 @@ void EbyteLoraComponent::send_data_(bool all) {
} }
#endif #endif
this->write_array(data); this->write_array(data);
this->setup_wait_response_(5000);
} }
void EbyteLoraComponent::send_repeater_info_() { void EbyteLoraComponent::send_repeater_info_() {
if (!this->can_send_message_()) { if (!this->can_send) {
ESP_LOGD(TAG, "Can't sent it right now");
return; return;
} }
uint8_t data[3]; uint8_t data[3];
@ -606,10 +604,10 @@ void EbyteLoraComponent::send_repeater_info_() {
data[2] = network_id_; data[2] = network_id_;
ESP_LOGD(TAG, "Telling system if i am a repeater and what my network_id is"); ESP_LOGD(TAG, "Telling system if i am a repeater and what my network_id is");
this->write_array(data, sizeof(data)); this->write_array(data, sizeof(data));
this->setup_wait_response_(5000);
} }
void EbyteLoraComponent::request_repeater_info_() { void EbyteLoraComponent::request_repeater_info_() {
if (!this->can_send_message_()) { if (!this->can_send) {
ESP_LOGD(TAG, "Can't sent it right now");
return; return;
} }
uint8_t data[2]; uint8_t data[2];
@ -617,15 +615,14 @@ void EbyteLoraComponent::request_repeater_info_() {
data[1] = this->network_id_; // for unique id data[1] = this->network_id_; // for unique id
ESP_LOGD(TAG, "Asking for repeater info"); ESP_LOGD(TAG, "Asking for repeater info");
this->write_array(data, sizeof(data)); this->write_array(data, sizeof(data));
this->setup_wait_response_(5000);
} }
void EbyteLoraComponent::repeat_message_(uint8_t *buf) { void EbyteLoraComponent::repeat_message_(uint8_t *buf) {
ESP_LOGD(TAG, "Got some info that i need to repeat for network %u", buf[1]); ESP_LOGD(TAG, "Got some info that i need to repeat for network %u", buf[1]);
if (!this->can_send_message_()) { if (!this->can_send) {
ESP_LOGD(TAG, "Can't sent it right now");
return; return;
} }
this->write_array(buf, sizeof(buf)); this->write_array(buf, sizeof(buf));
this->setup_wait_response_(5000);
} }
} // namespace ebyte_lora } // namespace ebyte_lora

View file

@ -18,6 +18,14 @@
namespace esphome { namespace esphome {
namespace ebyte_lora { namespace ebyte_lora {
/// Store data in a class that doesn't use multiple-inheritance (vtables in flash)
struct EbyteAuxStore {
volatile uint32_t last_interrupt{0};
volatile uint32_t on_time{0};
volatile bool can_send{false};
ISRInternalGPIOPin pin;
static void gpio_intr(EbyteAuxStore *arg);
};
#ifdef USE_SENSOR #ifdef USE_SENSOR
struct Sensor { struct Sensor {
sensor::Sensor *sensor; sensor::Sensor *sensor;
@ -35,8 +43,6 @@ struct BinarySensor {
#endif #endif
static const char *const TAG = "ebyte_lora"; static const char *const TAG = "ebyte_lora";
static const int MAX_SIZE_TX_PACKET = 200;
// the mode the receiver is in // the mode the receiver is in
enum ModeType { NORMAL = 0, WOR_SEND = 1, WOR_RECEIVER = 2, CONFIGURATION = 3, MODE_INIT = 0xFF }; enum ModeType { NORMAL = 0, WOR_SEND = 1, WOR_RECEIVER = 2, CONFIGURATION = 3, MODE_INIT = 0xFF };
// 1 byte, 8 bits in total // 1 byte, 8 bits in total
@ -89,18 +95,19 @@ class EbyteLoraComponent : public PollingComponent, public uart::UARTDevice {
void set_enable_lbt(EnableByte enable) { expected_config_.enable_lbt = enable; } void set_enable_lbt(EnableByte enable) { expected_config_.enable_lbt = enable; }
void set_transmission_mode(TransmissionMode mode) { expected_config_.transmission_mode = mode; } void set_transmission_mode(TransmissionMode mode) { expected_config_.transmission_mode = mode; }
void set_enable_rssi(EnableByte enable) { expected_config_.enable_rssi = enable; } void set_enable_rssi(EnableByte enable) { expected_config_.enable_rssi = enable; }
void set_sent_switch_state(bool enable) { sent_switch_state_ = enable; } // if enabled, will repeat any message it received as long as it isn't its own network id
void set_repeater(bool enable) { repeater_enabled_ = enable; } void set_repeater(bool enable) { repeater_enabled_ = enable; }
// software network id, this will make sure that only items from that network id are being found
void set_network_id(int id) { network_id_ = id; } void set_network_id(int id) { network_id_ = id; }
private: private:
ModeType mode_ = MODE_INIT; ModeType config_mode_ = MODE_INIT;
// set WOR mode // set WOR mode
void set_mode_(ModeType mode); void set_mode_(ModeType mode);
ModeType get_mode_(); ModeType get_mode_();
// checks the aux port to see if it is done setting bool config_checked_ = false;
void setup_wait_response_(uint32_t timeout = 1000); // check if you can sent a message
bool can_send_message_();
bool check_config_(); bool check_config_();
void set_config_(); void set_config_();
void get_current_config_(); void get_current_config_();
@ -110,21 +117,21 @@ class EbyteLoraComponent : public PollingComponent, public uart::UARTDevice {
protected: protected:
bool updated_{}; bool updated_{};
bool resend_repeater_request_{};
uint32_t repeater_request_recyle_time_{};
uint32_t last_key_time_{};
void setup_conf_(uint8_t const *conf); void setup_conf_(uint8_t const *conf);
void process_(uint8_t *buf, size_t len); void process_(uint8_t *buf, size_t len);
void repeat_message_(uint8_t *buf); void repeat_message_(uint8_t *buf);
// set if there is some sensor info available or it is in repeater mode
bool should_send_{}; bool should_send_{};
bool update_needed_ = false; // if set it will function as a repeater only
// if enabled will sent information about itself
bool sent_switch_state_ = false;
// if set it will function as a repeater
bool repeater_enabled_ = false; bool repeater_enabled_ = false;
// used to tell one lora device apart from another // used to tell one lora device apart from another
int network_id_ = 0; int network_id_ = 0;
int rssi_ = 0; int rssi_ = 0;
uint32_t starting_to_check_; EbyteAuxStore store_;
uint32_t time_out_after_; bool can_send = true;
std::string raw_message_;
RegisterConfig current_config_; RegisterConfig current_config_;
RegisterConfig expected_config_; RegisterConfig expected_config_;
#ifdef USE_SENSOR #ifdef USE_SENSOR