mirror of
https://github.com/BotChain-Robots/firmware.git
synced 2026-07-08 17:47:21 +02:00
Config manager fixes
This commit is contained in:
@@ -2,101 +2,182 @@
|
|||||||
// Created by Johnathon Slightham on 2025-06-12.
|
// Created by Johnathon Slightham on 2025-06-12.
|
||||||
//
|
//
|
||||||
|
|
||||||
|
|
||||||
|
#include <mutex>
|
||||||
|
#include <esp_log.h>
|
||||||
|
|
||||||
#include "nvs_flash.h"
|
#include "nvs_flash.h"
|
||||||
#include "constants/config.h"
|
#include "constants/config.h"
|
||||||
|
|
||||||
#include "ConfigManager.h"
|
#include "ConfigManager.h"
|
||||||
|
|
||||||
|
static auto TAG = "ConfigManager";
|
||||||
|
|
||||||
void ConfigManager::init_config() {
|
void ConfigManager::init_config() {
|
||||||
nvs_flash_init();
|
std::unique_lock lock(rw_lock);
|
||||||
|
if (this->initialized)
|
||||||
|
return;
|
||||||
|
this->initialized = true;
|
||||||
|
lock.unlock();
|
||||||
|
|
||||||
nvs_handle config_handle;
|
esp_err_t ret = nvs_flash_init();
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READWRITE, &config_handle);
|
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
|
||||||
|
// NVS partition was truncated and needs to be erased, retry nvs_flash_init after erasing
|
||||||
uint16_t id;
|
ESP_ERROR_CHECK(nvs_flash_erase());
|
||||||
if (ESP_ERR_NVS_NOT_FOUND == nvs_get_u16(config_handle, MODULE_ID_KEY, &id)) {
|
ret = nvs_flash_init();
|
||||||
nvs_set_u16(config_handle, MODULE_ID_KEY, DEFAULT_MODULE_ID);
|
|
||||||
}
|
}
|
||||||
|
ESP_ERROR_CHECK(ret);
|
||||||
uint16_t type;
|
|
||||||
if (ESP_ERR_NVS_NOT_FOUND == nvs_get_u16(config_handle, MODULE_TYPE_KEY, &type)) {
|
|
||||||
nvs_set_u16(config_handle, MODULE_TYPE_KEY, DEFAULT_MODULE_TYPE);
|
|
||||||
}
|
|
||||||
|
|
||||||
nvs_close(config_handle);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// todo: we should probably cache some of these things
|
uint16_t ConfigManager::get_module_id() const {
|
||||||
uint16_t ConfigManager::get_module_id() {
|
uint16_t i = DEFAULT_MODULE_ID;
|
||||||
nvs_handle config_handle;
|
get_uint16(MODULE_ID_KEY, &i);
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READONLY, &config_handle);
|
return i;
|
||||||
uint16_t id;
|
|
||||||
nvs_get_u16(config_handle, MODULE_ID_KEY, &id);
|
|
||||||
nvs_close(config_handle);
|
|
||||||
return id;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ModuleType ConfigManager::get_module_type() {
|
ModuleType ConfigManager::get_module_type() const {
|
||||||
nvs_handle config_handle;
|
uint16_t type = DEFAULT_MODULE_TYPE;
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READONLY, &config_handle);
|
get_uint16(MODULE_TYPE_KEY, &type);
|
||||||
uint16_t type;
|
|
||||||
nvs_get_u16(config_handle, MODULE_TYPE_KEY, &type);
|
|
||||||
nvs_close(config_handle);
|
|
||||||
return static_cast<ModuleType>(type);
|
return static_cast<ModuleType>(type);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string ConfigManager::get_wifi_ssid() {
|
std::string ConfigManager::get_wifi_ssid() const {
|
||||||
return get_string(WIFI_SSID_KEY);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string ConfigManager::get_wifi_password() {
|
|
||||||
return get_string(WIFI_PASSWORD_KEY);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string ConfigManager::get_string(const char *key) {
|
|
||||||
nvs_handle config_handle;
|
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READONLY, &config_handle);
|
|
||||||
|
|
||||||
size_t required_size; // get size of the string
|
|
||||||
if (ESP_ERR_NVS_NOT_FOUND == nvs_get_str(config_handle, key, nullptr, &required_size)) {
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string str;
|
std::string str;
|
||||||
str.resize(required_size);
|
get_string(WIFI_SSID_KEY, str);
|
||||||
if (ESP_ERR_NVS_NOT_FOUND == nvs_get_str(config_handle, key, str.data(), &required_size)) {
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
nvs_close(config_handle);
|
|
||||||
return str;
|
return str;
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConfigManager::set_module_id(const uint16_t id) {
|
std::string ConfigManager::get_wifi_password() const {
|
||||||
|
std::string str;
|
||||||
|
get_string(WIFI_PASSWORD_KEY, str);
|
||||||
|
return str;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t ConfigManager::get_string(const char *key, std::string& out) const {
|
||||||
|
std::shared_lock lock(rw_lock);
|
||||||
|
esp_err_t ret = ESP_OK;
|
||||||
|
if (const auto str = get_config_from_cache<std::string>(key)) {
|
||||||
|
out = *str;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
ESP_LOGD(TAG, "get_string cache miss");
|
||||||
|
|
||||||
nvs_handle config_handle;
|
nvs_handle config_handle;
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READWRITE, &config_handle);
|
if (ret = nvs_open(NVS_FLASH_NAMESPACE, NVS_READONLY, &config_handle); ret != ESP_OK) {
|
||||||
nvs_set_u16(config_handle, MODULE_ID_KEY, id);
|
ESP_LOGE(TAG, "get_string Failed to open NVS");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t required_size; // get size of the string
|
||||||
|
if (ret = nvs_get_str(config_handle, key, nullptr, &required_size); ret != ESP_OK) {
|
||||||
|
ESP_LOGE(TAG, "get_string failed to get string size");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
out.resize(required_size);
|
||||||
|
if (ret = nvs_get_str(config_handle, key, out.data(), &required_size); ret != ESP_OK) {
|
||||||
|
ESP_LOGE(TAG, "get_string failed to get string");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
nvs_close(config_handle);
|
nvs_close(config_handle);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t ConfigManager::get_uint16(const char *key, uint16_t *out) const {
|
||||||
|
esp_err_t ret = ESP_OK;
|
||||||
|
std::shared_lock lock(rw_lock);
|
||||||
|
if (const auto i = get_config_from_cache<uint16_t>(key)) {
|
||||||
|
*out = i.value();
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
ESP_LOGD(TAG, "get_uint16 cache miss");
|
||||||
|
|
||||||
|
nvs_handle config_handle;
|
||||||
|
if (ret = nvs_open(NVS_FLASH_NAMESPACE, NVS_READONLY, &config_handle); ret != ESP_OK) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ret = nvs_get_u16(config_handle, key, out); ret != ESP_OK) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
nvs_close(config_handle);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
std::optional<T> ConfigManager::get_config_from_cache(const std::string& key) const {
|
||||||
|
// Not thread safe
|
||||||
|
if (const auto it = cache.find(key); it != cache.end()) {
|
||||||
|
if (auto val = std::get_if<T>(&it->second))
|
||||||
|
return *val;
|
||||||
|
}
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ConfigManager::set_module_id(const uint16_t id) {
|
||||||
|
set<nvs_set_u16, uint16_t>(MODULE_ID_KEY, id);
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConfigManager::set_module_type(const ModuleType type) {
|
void ConfigManager::set_module_type(const ModuleType type) {
|
||||||
nvs_handle config_handle;
|
set<nvs_set_u16, uint16_t>(MODULE_TYPE_KEY, type);
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READWRITE, &config_handle);
|
|
||||||
nvs_set_u16(config_handle, MODULE_TYPE_KEY, type);
|
|
||||||
nvs_close(config_handle);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConfigManager::set_wifi_ssid(const char* ssid) {
|
void ConfigManager::set_wifi_ssid(const char* ssid) {
|
||||||
nvs_handle config_handle;
|
set<nvs_set_cpp_str, std::string>(WIFI_SSID_KEY, ssid);
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READWRITE, &config_handle);
|
|
||||||
nvs_set_str(config_handle, WIFI_SSID_KEY, ssid);
|
|
||||||
nvs_close(config_handle);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConfigManager::set_wifi_password(const char* password) {
|
void ConfigManager::set_wifi_password(const char* password) {
|
||||||
nvs_handle config_handle;
|
set<nvs_set_cpp_str, std::string>(WIFI_PASSWORD_KEY, password);
|
||||||
nvs_open(NVS_FLASH_NAMESPACE, NVS_READWRITE, &config_handle);
|
|
||||||
nvs_set_str(config_handle, WIFI_PASSWORD_KEY, password);
|
|
||||||
nvs_close(config_handle);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Func - the esp write function to call (ie. nvs_set_u16)
|
||||||
|
// T - the type of the value in the key value pair
|
||||||
|
template<auto Func, typename T>
|
||||||
|
esp_err_t ConfigManager::set(const char* key, const T value) {
|
||||||
|
std::unique_lock lock(rw_lock);
|
||||||
|
|
||||||
|
if (write_to_cache(key, value)) {
|
||||||
|
return ESP_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t ret = ESP_OK;
|
||||||
|
nvs_handle config_handle;
|
||||||
|
ret = nvs_open(NVS_FLASH_NAMESPACE, NVS_READWRITE, &config_handle);
|
||||||
|
if (ESP_OK != ret) {
|
||||||
|
ESP_LOGE(TAG, "set error opening NVS handle");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = Func(config_handle, key, value);
|
||||||
|
if (ESP_OK != ret) {
|
||||||
|
ESP_LOGE(TAG, "set error writing to NVS key %s", key);
|
||||||
|
}
|
||||||
|
nvs_close(config_handle);
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns true when we don't need to write through the cache
|
||||||
|
template<typename T>
|
||||||
|
bool ConfigManager::write_to_cache(const std::string& key, const T value) {
|
||||||
|
const auto it = cache.find(key);
|
||||||
|
if (it != cache.end()) {
|
||||||
|
if (const auto existing = std::get_if<T>(&it->second)) {
|
||||||
|
if (*existing == value)
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cache[key] = value;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t ConfigManager::nvs_set_cpp_str(const nvs_handle_t handle, const std::string& key, const std::string& str) {
|
||||||
|
return nvs_set_str(handle, key.c_str(), str.c_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t ConfigManager::nvs_get_cpp_str(const nvs_handle_t handle, const std::string& key, std::string& out_str, size_t *length) {
|
||||||
|
return nvs_get_str(handle, key.c_str(), out_str.data(), length);
|
||||||
|
}
|
||||||
|
|||||||
@@ -5,26 +5,60 @@
|
|||||||
#ifndef CONFIGMANAGER_H
|
#ifndef CONFIGMANAGER_H
|
||||||
#define CONFIGMANAGER_H
|
#define CONFIGMANAGER_H
|
||||||
|
|
||||||
#include "constants/config.h"
|
#include <variant>
|
||||||
|
#include <unordered_map>
|
||||||
|
#include <shared_mutex>
|
||||||
|
#include <esp_check.h>
|
||||||
|
#include <nvs.h>
|
||||||
|
|
||||||
#include "flatbuffers_generated/RobotModule_generated.h"
|
#include "flatbuffers_generated/RobotModule_generated.h"
|
||||||
|
|
||||||
|
// Singleton to r/w config from the ESP32 nvs (thread safe and cached)
|
||||||
|
|
||||||
class ConfigManager {
|
class ConfigManager {
|
||||||
public:
|
public:
|
||||||
static void init_config();
|
ConfigManager(const ConfigManager&) = delete;
|
||||||
|
void operator=(ConfigManager &) = delete;
|
||||||
|
|
||||||
static uint16_t get_module_id();
|
static ConfigManager& get_instance() { // Thread safe as of C++11
|
||||||
static ModuleType get_module_type();
|
static ConfigManager instance;
|
||||||
static std::string get_wifi_ssid();
|
instance.init_config();
|
||||||
static std::string get_wifi_password();
|
return instance;
|
||||||
|
}
|
||||||
|
|
||||||
static void set_module_id(uint16_t id);
|
uint16_t get_module_id() const;
|
||||||
static void set_module_type(ModuleType type);
|
ModuleType get_module_type() const;
|
||||||
static void set_wifi_ssid(const char* ssid);
|
std::string get_wifi_ssid() const;
|
||||||
static void set_wifi_password(const char* password);
|
std::string get_wifi_password() const;
|
||||||
|
|
||||||
|
void set_module_id(uint16_t id);
|
||||||
|
void set_module_type(ModuleType type);
|
||||||
|
void set_wifi_ssid(const char* ssid);
|
||||||
|
void set_wifi_password(const char* password);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
static std::string get_string(const char* key);
|
ConfigManager() = default;
|
||||||
|
esp_err_t get_string(const char *key, std::string &out) const;
|
||||||
|
esp_err_t get_uint16(const char *key, uint16_t *out) const;
|
||||||
|
|
||||||
|
|
||||||
|
static esp_err_t nvs_set_cpp_str(nvs_handle_t handle, const std::string& key, const std::string& str);
|
||||||
|
static esp_err_t nvs_get_cpp_str(nvs_handle_t handle, const std::string& key, std::string& out_str, size_t *length);
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
std::optional<T> get_config_from_cache(const std::string& key) const;
|
||||||
|
|
||||||
|
template<auto Func, typename T>
|
||||||
|
esp_err_t set(const char* key, T value);
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
bool write_to_cache(const std::string& key, T value);
|
||||||
|
|
||||||
|
void init_config();
|
||||||
|
|
||||||
|
std::unordered_map<std::string, std::variant<uint16_t, int8_t, std::string>> cache;
|
||||||
|
mutable std::shared_mutex rw_lock;
|
||||||
|
bool initialized = false;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif //CONFIGMANAGER_H
|
#endif //CONFIGMANAGER_H
|
||||||
|
|||||||
@@ -13,7 +13,7 @@
|
|||||||
template <typename T>
|
template <typename T>
|
||||||
class PtrQueue {
|
class PtrQueue {
|
||||||
public:
|
public:
|
||||||
explicit PtrQueue(UBaseType_t queueLength)
|
explicit PtrQueue(const UBaseType_t queueLength)
|
||||||
: queue(xQueueCreate(queueLength, sizeof(T*))) {}
|
: queue(xQueueCreate(queueLength, sizeof(T*))) {}
|
||||||
|
|
||||||
~PtrQueue() {
|
~PtrQueue() {
|
||||||
|
|||||||
@@ -1,4 +1,6 @@
|
|||||||
idf_component_register(SRCS "WifiManager.cpp" "mDNSDiscoveryService.cpp" "TCPServer.cpp" "CommunicationRouter.cpp" "MessagingInterface.cpp" "OrientationDetection.cpp"
|
idf_component_register(SRCS "WifiManager.cpp" "mDNSDiscoveryService.cpp" "TCPServer.cpp" "CommunicationRouter.cpp" "MessagingInterface.cpp" "OrientationDetection.cpp"
|
||||||
PRIV_REQUIRES driver esp_event nvs_flash esp_netif esp_wifi espressif__mdns constants config flatbuffers dataLink rmt
|
PRIV_REQUIRES driver esp_event nvs_flash esp_netif espressif__mdns constants config flatbuffers dataLink rmt
|
||||||
REQUIRES ptrQueue
|
REQUIRES ptrQueue esp_wifi
|
||||||
INCLUDE_DIRS "include")
|
INCLUDE_DIRS "include")
|
||||||
|
|
||||||
|
target_compile_options(${COMPONENT_LIB} PRIVATE -fexceptions)
|
||||||
|
|||||||
@@ -15,14 +15,14 @@ MessagingInterface::~MessagingInterface() {
|
|||||||
vQueueDelete(m_mpi_rx_queue);
|
vQueueDelete(m_mpi_rx_queue);
|
||||||
vSemaphoreDelete(m_map_semaphore);
|
vSemaphoreDelete(m_map_semaphore);
|
||||||
|
|
||||||
for (const auto [_tag, queue] : m_tag_to_queue) {
|
for (const auto queue: m_tag_to_queue | std::views::values) {
|
||||||
vQueueDelete(queue);
|
vQueueDelete(queue);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
int MessagingInterface::send(char* buffer, int size, int destination, int tag, bool durable) {
|
int MessagingInterface::send(char* buffer, const int size, const int destination, const int tag, const bool durable) {
|
||||||
Flatbuffers::MPIMessageBuilder builder;
|
Flatbuffers::MPIMessageBuilder builder;
|
||||||
const auto [mpi_buffer, mpi_size] = builder.build_mpi_message(Messaging::MessageType_PTP, ConfigManager::get_module_id(), destination, sequence_number++, durable, tag, std::vector<uint8_t>(buffer, buffer + size));
|
const auto [mpi_buffer, mpi_size] = builder.build_mpi_message(Messaging::MessageType_PTP, m_config_manager.get_module_id(), destination, m_sequence_number++, durable, tag, std::vector<uint8_t>(buffer, buffer + size));
|
||||||
|
|
||||||
m_router->send_msg(static_cast<char *>(mpi_buffer), mpi_size);
|
m_router->send_msg(static_cast<char *>(mpi_buffer), mpi_size);
|
||||||
return 0;
|
return 0;
|
||||||
|
|||||||
@@ -10,7 +10,8 @@
|
|||||||
#include <bits/ostream.tcc>
|
#include <bits/ostream.tcc>
|
||||||
|
|
||||||
void OrientationDetection::init() {
|
void OrientationDetection::init() {
|
||||||
for (int i = 0; i < MODULE_TO_NUM_CHANNELS_MAP[ConfigManager::get_module_type()]; i++) {
|
const auto& config_manager = ConfigManager::get_instance();
|
||||||
|
for (int i = 0; i < MODULE_TO_NUM_CHANNELS_MAP[config_manager.get_module_type()]; i++) {
|
||||||
setup_gpio(static_cast<gpio_num_t>(CHANNEL_TO_0_DEG_MAP[i]));
|
setup_gpio(static_cast<gpio_num_t>(CHANNEL_TO_0_DEG_MAP[i]));
|
||||||
setup_gpio(static_cast<gpio_num_t>(CHANNEL_TO_90_DEG_MAP[i]));
|
setup_gpio(static_cast<gpio_num_t>(CHANNEL_TO_90_DEG_MAP[i]));
|
||||||
setup_gpio(static_cast<gpio_num_t>(CHANNEL_TO_180_DEG_MAP[i]));
|
setup_gpio(static_cast<gpio_num_t>(CHANNEL_TO_180_DEG_MAP[i]));
|
||||||
|
|||||||
@@ -1,30 +1,9 @@
|
|||||||
#include "WifiManager.h"
|
|
||||||
|
|
||||||
#include <ConfigManager.h>
|
|
||||||
#include <esp_netif.h>
|
|
||||||
#include <esp_event.h>
|
|
||||||
#include <freertos/semphr.h>
|
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
#include <mDNSDiscoveryService.h>
|
|
||||||
|
|
||||||
#include "freertos/FreeRTOS.h"
|
#include "WifiManager.h"
|
||||||
#include "freertos/semphr.h"
|
#include "ConfigManager.h"
|
||||||
#include "esp_wifi.h"
|
|
||||||
#include "constants/wifi.h"
|
#include "constants/wifi.h"
|
||||||
|
#include "mDNSDiscoveryService.h"
|
||||||
WifiManager::WifiManager() {
|
|
||||||
esp_netif_init();
|
|
||||||
esp_wifi_set_storage(WIFI_STORAGE_RAM);
|
|
||||||
esp_event_loop_create_default();
|
|
||||||
|
|
||||||
this->m_mutex = xSemaphoreCreateMutex();
|
|
||||||
this->m_state = wifi_state::disconnected;
|
|
||||||
this->m_attempts = 0;
|
|
||||||
this->m_task = nullptr;
|
|
||||||
this->m_netif = nullptr;
|
|
||||||
|
|
||||||
xTaskCreate(reinterpret_cast<TaskFunction_t>(s_manage), "wifi_task", 3096, this, 5, &m_task);
|
|
||||||
}
|
|
||||||
|
|
||||||
WifiManager::~WifiManager() {
|
WifiManager::~WifiManager() {
|
||||||
this->handle_disconnect();
|
this->handle_disconnect();
|
||||||
@@ -103,8 +82,8 @@ int WifiManager::init_connection() {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
std::string ssid = ConfigManager::get_wifi_ssid();
|
std::string ssid = m_config_manager.get_wifi_ssid();
|
||||||
std::string pass = ConfigManager::get_wifi_password();
|
std::string pass = m_config_manager.get_wifi_password();
|
||||||
std::strncpy(reinterpret_cast<char *>(wifi_configuration.sta.ssid), ssid.c_str(), 32);
|
std::strncpy(reinterpret_cast<char *>(wifi_configuration.sta.ssid), ssid.c_str(), 32);
|
||||||
std::strncpy(reinterpret_cast<char *>(wifi_configuration.sta.password), pass.c_str(), 64);
|
std::strncpy(reinterpret_cast<char *>(wifi_configuration.sta.password), pass.c_str(), 64);
|
||||||
wifi_configuration.sta.ssid[31] = '\0';
|
wifi_configuration.sta.ssid[31] = '\0';
|
||||||
|
|||||||
@@ -36,17 +36,18 @@ public:
|
|||||||
CommunicationRouter(const std::function<void(char*, int)> &rx_callback, std::unique_ptr<WifiManager>&& pc_connection)
|
CommunicationRouter(const std::function<void(char*, int)> &rx_callback, std::unique_ptr<WifiManager>&& pc_connection)
|
||||||
: m_tcp_rx_queue(std::make_shared<PtrQueue<std::vector<uint8_t>>>(10)),
|
: m_tcp_rx_queue(std::make_shared<PtrQueue<std::vector<uint8_t>>>(10)),
|
||||||
m_rx_callback(rx_callback),
|
m_rx_callback(rx_callback),
|
||||||
|
m_config_manager(ConfigManager::get_instance()),
|
||||||
m_tcp_server(std::make_unique<TCPServer>(TCP_PORT, m_tcp_rx_queue)),
|
m_tcp_server(std::make_unique<TCPServer>(TCP_PORT, m_tcp_rx_queue)),
|
||||||
m_data_link_manager(std::make_unique<DataLinkManager>(ConfigManager::get_module_id(), MODULE_TO_NUM_CHANNELS_MAP[ConfigManager::get_module_type()])),
|
m_data_link_manager(std::make_unique<DataLinkManager>(m_config_manager.get_module_id(), MODULE_TO_NUM_CHANNELS_MAP[m_config_manager.get_module_type()])),
|
||||||
m_pc_connection(std::move(pc_connection)),
|
m_pc_connection(std::move(pc_connection)),
|
||||||
m_module_id(ConfigManager::get_module_id()),
|
m_module_id(m_config_manager.get_module_id()),
|
||||||
m_last_leader_updated(std::chrono::system_clock::now()){
|
m_last_leader_updated(std::chrono::system_clock::now()){
|
||||||
OrientationDetection::init();
|
OrientationDetection::init();
|
||||||
update_leader();
|
update_leader();
|
||||||
|
|
||||||
xTaskCreate(router_thread, "communication_router", 4096, this, 3, &this->m_router_thread);
|
xTaskCreate(router_thread, "communication_router", 4096, this, 3, &this->m_router_thread);
|
||||||
|
|
||||||
const auto num_channels = MODULE_TO_NUM_CHANNELS_MAP[ConfigManager::get_module_type()];
|
const auto num_channels = MODULE_TO_NUM_CHANNELS_MAP[m_config_manager.get_module_type()];
|
||||||
this->m_link_layer_threads.resize(num_channels);
|
this->m_link_layer_threads.resize(num_channels);
|
||||||
for (int i = 0; i < num_channels; i++) {
|
for (int i = 0; i < num_channels; i++) {
|
||||||
auto *params = new link_layer_thread_params(this, i);
|
auto *params = new link_layer_thread_params(this, i);
|
||||||
@@ -73,6 +74,7 @@ public:
|
|||||||
std::function<void(char*, int)> m_rx_callback;
|
std::function<void(char*, int)> m_rx_callback;
|
||||||
private:
|
private:
|
||||||
TaskHandle_t m_router_thread = nullptr;
|
TaskHandle_t m_router_thread = nullptr;
|
||||||
|
ConfigManager &m_config_manager;
|
||||||
std::vector<TaskHandle_t> m_link_layer_threads;
|
std::vector<TaskHandle_t> m_link_layer_threads;
|
||||||
std::unique_ptr<TCPServer> m_tcp_server; // todo: dependency injection
|
std::unique_ptr<TCPServer> m_tcp_server; // todo: dependency injection
|
||||||
std::unique_ptr<DataLinkManager> m_data_link_manager;
|
std::unique_ptr<DataLinkManager> m_data_link_manager;
|
||||||
|
|||||||
@@ -14,8 +14,9 @@
|
|||||||
class MessagingInterface {
|
class MessagingInterface {
|
||||||
public:
|
public:
|
||||||
explicit MessagingInterface(std::unique_ptr<WifiManager>&& pc_connection)
|
explicit MessagingInterface(std::unique_ptr<WifiManager>&& pc_connection)
|
||||||
: m_mpi_rx_queue(xQueueCreate(MAX_RX_BUFFER_SIZE, RX_QUEUE_SIZE)),
|
: m_config_manager(ConfigManager::get_instance()),
|
||||||
m_router(std::make_unique<CommunicationRouter>([this](char* buffer, const int size) { handleRecv(buffer, size); }, std::move(pc_connection))),
|
m_mpi_rx_queue(xQueueCreate(MAX_RX_BUFFER_SIZE, RX_QUEUE_SIZE)),
|
||||||
|
m_router(std::make_unique<CommunicationRouter>([this](const char* buffer, const int size) { handleRecv(buffer, size); }, std::move(pc_connection))),
|
||||||
m_map_semaphore(xSemaphoreCreateMutex()) {};
|
m_map_semaphore(xSemaphoreCreateMutex()) {};
|
||||||
|
|
||||||
~MessagingInterface();
|
~MessagingInterface();
|
||||||
@@ -31,7 +32,8 @@ private:
|
|||||||
|
|
||||||
void checkOrInsertTag(uint8_t tag);
|
void checkOrInsertTag(uint8_t tag);
|
||||||
|
|
||||||
uint16_t sequence_number = 0;
|
ConfigManager& m_config_manager;
|
||||||
|
uint16_t m_sequence_number = 0;
|
||||||
QueueHandle_t m_mpi_rx_queue; // todo: maybe move this down classes more
|
QueueHandle_t m_mpi_rx_queue; // todo: maybe move this down classes more
|
||||||
std::unique_ptr<CommunicationRouter> m_router;
|
std::unique_ptr<CommunicationRouter> m_router;
|
||||||
SemaphoreHandle_t m_map_semaphore;
|
SemaphoreHandle_t m_map_semaphore;
|
||||||
|
|||||||
@@ -2,17 +2,27 @@
|
|||||||
#define NETWORKMANAGER_H
|
#define NETWORKMANAGER_H
|
||||||
|
|
||||||
#include <esp_netif_types.h>
|
#include <esp_netif_types.h>
|
||||||
|
#include <esp_event.h>
|
||||||
|
#include <esp_netif.h>
|
||||||
|
#include <esp_wifi.h>
|
||||||
|
#include <freertos/FreeRTOS.h>
|
||||||
|
#include <freertos/semphr.h>
|
||||||
|
|
||||||
#include "esp_event.h"
|
#include "ConfigManager.h"
|
||||||
|
|
||||||
#include "freertos/FreeRTOS.h"
|
|
||||||
#include "freertos/semphr.h"
|
|
||||||
|
|
||||||
#include "IWifiManager.h"
|
#include "IWifiManager.h"
|
||||||
|
|
||||||
class WifiManager final : IWifiManager {
|
class WifiManager final : IWifiManager {
|
||||||
public:
|
public:
|
||||||
WifiManager();
|
WifiManager() : m_config_manager(ConfigManager::get_instance()),
|
||||||
|
m_mutex(xSemaphoreCreateMutex()),
|
||||||
|
m_state(wifi_state::disconnected) {
|
||||||
|
esp_netif_init();
|
||||||
|
esp_wifi_set_storage(WIFI_STORAGE_RAM);
|
||||||
|
esp_event_loop_create_default();
|
||||||
|
// todo: move all task metadata to a constants/config file
|
||||||
|
xTaskCreate(reinterpret_cast<TaskFunction_t>(s_manage), "wifi_task", 3096, this, 5, &m_task);
|
||||||
|
}
|
||||||
|
|
||||||
~WifiManager() override;
|
~WifiManager() override;
|
||||||
int connect() override;
|
int connect() override;
|
||||||
int disconnect() override;
|
int disconnect() override;
|
||||||
@@ -31,11 +41,12 @@ private:
|
|||||||
|
|
||||||
static void wifi_event_handler(void *event_handler_arg, esp_event_base_t event_base, int32_t event_id, void *event_data);
|
static void wifi_event_handler(void *event_handler_arg, esp_event_base_t event_base, int32_t event_id, void *event_data);
|
||||||
|
|
||||||
|
ConfigManager& m_config_manager;
|
||||||
SemaphoreHandle_t m_mutex;
|
SemaphoreHandle_t m_mutex;
|
||||||
wifi_state m_state;
|
wifi_state m_state;
|
||||||
TaskHandle_t m_task;
|
TaskHandle_t m_task = nullptr;
|
||||||
unsigned int m_attempts;
|
unsigned int m_attempts = 0;
|
||||||
esp_netif_t *m_netif;
|
esp_netif_t *m_netif = nullptr;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -18,15 +18,18 @@
|
|||||||
// todo: clean this up (strange to be a constructor) also need to add more details, need to add to routing table
|
// todo: clean this up (strange to be a constructor) also need to add more details, need to add to routing table
|
||||||
void mDNSDiscoveryService::setup() {
|
void mDNSDiscoveryService::setup() {
|
||||||
mdns_init();
|
mdns_init();
|
||||||
mdns_hostname_set(std::format("botchain-{}", ConfigManager::get_module_id()).c_str());
|
|
||||||
mdns_instance_name_set(std::format("BotChain Robot Module {}", ConfigManager::get_module_id()).c_str());
|
const auto& config_manager = ConfigManager::get_instance();
|
||||||
|
|
||||||
|
mdns_hostname_set(std::format("botchain-{}", config_manager.get_module_id()).c_str());
|
||||||
|
mdns_instance_name_set(std::format("BotChain Robot Module {}", config_manager.get_module_id()).c_str());
|
||||||
|
|
||||||
mdns_service_add(nullptr, "_robotcontrol", "_tcp", TCP_PORT, nullptr, 0);
|
mdns_service_add(nullptr, "_robotcontrol", "_tcp", TCP_PORT, nullptr, 0);
|
||||||
mdns_service_instance_name_set("_robotcontrol", "_tcp", "Robot Control TCP Server");
|
mdns_service_instance_name_set("_robotcontrol", "_tcp", "Robot Control TCP Server");
|
||||||
|
|
||||||
mdns_txt_item_t service_txt_data[3] = {
|
mdns_txt_item_t service_txt_data[3] = {
|
||||||
{"module_id",std::to_string(ConfigManager::get_module_id()).c_str()},
|
{"module_id",std::to_string(config_manager.get_module_id()).c_str()},
|
||||||
{"module_type",std::to_string(ConfigManager::get_module_type()).c_str()},
|
{"module_type",std::to_string(config_manager.get_module_type()).c_str()},
|
||||||
{"connected_modules",""},
|
{"connected_modules",""},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -36,6 +39,8 @@ void mDNSDiscoveryService::setup() {
|
|||||||
void mDNSDiscoveryService::set_connected_boards(const std::vector<int>& boards) {
|
void mDNSDiscoveryService::set_connected_boards(const std::vector<int>& boards) {
|
||||||
std::stringstream ss;
|
std::stringstream ss;
|
||||||
|
|
||||||
|
const auto& config_manager = ConfigManager::get_instance();
|
||||||
|
|
||||||
for (int i = 0; i < boards.size(); i++) {
|
for (int i = 0; i < boards.size(); i++) {
|
||||||
ss << std::to_string(boards[i]);
|
ss << std::to_string(boards[i]);
|
||||||
if (i < boards.size() - 1) {
|
if (i < boards.size() - 1) {
|
||||||
@@ -44,8 +49,8 @@ void mDNSDiscoveryService::set_connected_boards(const std::vector<int>& boards)
|
|||||||
}
|
}
|
||||||
|
|
||||||
mdns_txt_item_t service_txt_data[3] = {
|
mdns_txt_item_t service_txt_data[3] = {
|
||||||
{"module_id",std::to_string(ConfigManager::get_module_id()).c_str()},
|
{"module_id",std::to_string(config_manager.get_module_id()).c_str()},
|
||||||
{"module_type",std::to_string(ConfigManager::get_module_type()).c_str()},
|
{"module_type",std::to_string(config_manager.get_module_type()).c_str()},
|
||||||
{"connected_modules",ss.str().c_str()},
|
{"connected_modules",ss.str().c_str()},
|
||||||
};
|
};
|
||||||
mdns_service_txt_set("_robotcontrol", "_tcp", service_txt_data, 3);
|
mdns_service_txt_set("_robotcontrol", "_tcp", service_txt_data, 3);
|
||||||
|
|||||||
@@ -17,7 +17,7 @@
|
|||||||
#define METADATA_PERIOD_MS 1000
|
#define METADATA_PERIOD_MS 1000
|
||||||
|
|
||||||
[[noreturn]] void LoopManager::control_loop() const {
|
[[noreturn]] void LoopManager::control_loop() const {
|
||||||
const auto actuator = ActuatorFactory::create_actuator(ConfigManager::get_module_type());
|
const auto actuator = ActuatorFactory::create_actuator(m_config_manager.get_module_type());
|
||||||
|
|
||||||
uint8_t buffer[512];
|
uint8_t buffer[512];
|
||||||
while (true) {
|
while (true) {
|
||||||
@@ -38,7 +38,7 @@
|
|||||||
casted_orientations.emplace_back(orientation);
|
casted_orientations.emplace_back(orientation);
|
||||||
}
|
}
|
||||||
|
|
||||||
const auto [data, size] = topology_message_builder->build_topology_message(ConfigManager::get_module_id(), ConfigManager::get_module_type(), module_ids, casted_orientations);
|
const auto [data, size] = topology_message_builder->build_topology_message(that->m_config_manager.get_module_id(), that->m_config_manager.get_module_type(), module_ids, casted_orientations);
|
||||||
that->m_messaging_interface->send(static_cast<char *>(data), size, PC_ADDR, TOPOLOGY_CMD_TAG, true);
|
that->m_messaging_interface->send(static_cast<char *>(data), size, PC_ADDR, TOPOLOGY_CMD_TAG, true);
|
||||||
vTaskDelay(METADATA_PERIOD_MS / portTICK_PERIOD_MS);
|
vTaskDelay(METADATA_PERIOD_MS / portTICK_PERIOD_MS);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,11 +14,14 @@
|
|||||||
|
|
||||||
class LoopManager {
|
class LoopManager {
|
||||||
public:
|
public:
|
||||||
LoopManager() : m_messaging_interface(std::make_unique<MessagingInterface>(std::make_unique<WifiManager>())) {}
|
LoopManager() : m_config_manager(ConfigManager::get_instance()),
|
||||||
|
m_messaging_interface(std::make_unique<MessagingInterface>(std::make_unique<WifiManager>())) {}
|
||||||
[[noreturn]] void control_loop() const;
|
[[noreturn]] void control_loop() const;
|
||||||
[[noreturn]] static void metadata_tx_loop(char * args);
|
[[noreturn]] static void metadata_tx_loop(char * args);
|
||||||
|
[[noreturn]] static void metadata_rx_loop(char * args);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
ConfigManager& m_config_manager;
|
||||||
std::unique_ptr<MessagingInterface> m_messaging_interface;
|
std::unique_ptr<MessagingInterface> m_messaging_interface;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|||||||
@@ -1,25 +1,25 @@
|
|||||||
#include <cstdio>
|
#include <cstdio>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
|
||||||
#include "sdkconfig.h"
|
|
||||||
#include "freertos/FreeRTOS.h"
|
#include "freertos/FreeRTOS.h"
|
||||||
|
#include "sdkconfig.h"
|
||||||
|
|
||||||
#include "WifiManager.h"
|
|
||||||
#include "mDNSDiscoveryService.h"
|
|
||||||
#include "TCPServer.h"
|
|
||||||
#include "ConfigManager.h"
|
#include "ConfigManager.h"
|
||||||
#include "LoopManager.h"
|
#include "LoopManager.h"
|
||||||
|
#include "TCPServer.h"
|
||||||
|
#include "WifiManager.h"
|
||||||
#include "esp_log.h"
|
#include "esp_log.h"
|
||||||
|
#include "mDNSDiscoveryService.h"
|
||||||
|
|
||||||
extern "C" [[noreturn]] void app_main(void) {
|
extern "C" [[noreturn]] void app_main(void) {
|
||||||
ESP_LOGI("MEM", "Free internal RAM: %d", heap_caps_get_free_size(MALLOC_CAP_8BIT));
|
ESP_LOGI("MEM", "Free internal RAM: %d",
|
||||||
|
heap_caps_get_free_size(MALLOC_CAP_8BIT));
|
||||||
ESP_LOGI("MEM", "Free PSRAM: %d", heap_caps_get_free_size(MALLOC_CAP_SPIRAM));
|
ESP_LOGI("MEM", "Free PSRAM: %d", heap_caps_get_free_size(MALLOC_CAP_SPIRAM));
|
||||||
|
|
||||||
ConfigManager::init_config();
|
auto& config_manager = ConfigManager::get_instance(); // NOLINT - here for easily adding temporary config
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
const auto loop_manager = std::make_unique<LoopManager>();
|
const auto loop_manager = std::make_unique<LoopManager>();
|
||||||
xTaskCreate(reinterpret_cast<TaskFunction_t>(LoopManager::metadata_tx_loop), "metadata_tx", 3096, loop_manager.get(), 3, nullptr);
|
xTaskCreate(reinterpret_cast<TaskFunction_t>(LoopManager::metadata_tx_loop),
|
||||||
|
"metadata_tx", 3096, loop_manager.get(), 3, nullptr);
|
||||||
loop_manager->control_loop();
|
loop_manager->control_loop();
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user