taskDriven by claude - must be checked

This commit is contained in:
2026-01-03 22:59:36 -06:00
parent 4e4add5287
commit daced162ba
11 changed files with 378 additions and 198 deletions

View File

@@ -7,9 +7,6 @@
#include <mutex> #include <mutex>
#include "bmHTTP.hpp" #include "bmHTTP.hpp"
std::atomic<bool> flag_scan_requested{false};
std::atomic<bool> credsGiven{false};
std::atomic<bool> tokenGiven{false};
std::atomic<bool> isBLEClientConnected{false}; std::atomic<bool> isBLEClientConnected{false};
std::atomic<bool> scanBlock{false}; std::atomic<bool> scanBlock{false};
std::atomic<bool> finalAuth{false}; std::atomic<bool> finalAuth{false};
@@ -121,125 +118,12 @@ void notifyAuthStatus(bool success) {
tmpConfChar->setValue(""); // Clear value after notify tmpConfChar->setValue(""); // Clear value after notify
} }
bool BLEtick(NimBLEAdvertising* pAdvertising) { // BLEtick() removed - replaced by bleSetupTask() in setup.cpp
printf("BleTick\n");
if(flag_scan_requested) {
flag_scan_requested = false;
if (!scanBlock) {
scanBlock = true;
printf("Scanning WiFi...\n");
bmWiFi.scanAndUpdateSSIDList();
}
else printf("Duplicate scan request\n");
}
else if (credsGiven) {
std::string tmpSSID;
std::string tmpUNAME;
std::string tmpPASS;
wifi_auth_mode_t tmpAUTH;
{
std::lock_guard<std::mutex> lock(dataMutex);
tmpSSID = SSID;
tmpUNAME = UNAME;
tmpPASS = PASS;
tmpAUTH = auth;
credsGiven = false;
}
bool wifiConnect;
if (tmpAUTH == WIFI_AUTH_WPA2_ENTERPRISE || tmpAUTH == WIFI_AUTH_WPA3_ENTERPRISE)
wifiConnect = bmWiFi.attemptConnect(tmpSSID.c_str(), tmpUNAME.c_str(), tmpPASS.c_str(), tmpAUTH);
else wifiConnect = bmWiFi.attemptConnect(tmpSSID.c_str(), tmpPASS.c_str(), tmpAUTH);
if (!wifiConnect) {
// notify errored
notifyConnectionStatus(false);
return false;
}
nvs_handle_t WiFiHandle;
esp_err_t err = nvs_open(nvsWiFi, NVS_READWRITE, &WiFiHandle);
if (err != ESP_OK) {
printf("ERROR Saving Credentials\n");
// notify errored
notifyConnectionStatus(false);
return false;
}
else {
err = nvs_set_str(WiFiHandle, ssidTag, tmpSSID.c_str());
if (err == ESP_OK) err = nvs_set_str(WiFiHandle, passTag, tmpPASS.c_str());
if (err == ESP_OK) err = nvs_set_str(WiFiHandle, unameTag, tmpUNAME.c_str());
if (err == ESP_OK) err = nvs_set_u8(WiFiHandle, authTag, (uint8_t)tmpAUTH);
if (err == ESP_OK) nvs_commit(WiFiHandle);
nvs_close(WiFiHandle);
}
if (err == ESP_OK) {
// notify connected
notifyConnectionStatus(true);
}
else {
// notify connected
notifyConnectionStatus(false);
}
}
else if (tokenGiven) {
tokenGiven = false;
if (!bmWiFi.isConnected()) {
printf("ERROR: token given without WiFi connection\n");
notifyAuthStatus(false);
return false;
}
// HTTP request to verify device with token
std::string tmpTOKEN;
{
std::lock_guard<std::mutex> lock(dataMutex);
tmpTOKEN = TOKEN;
}
cJSON *responseRoot;
bool success = httpGET("verify_device", tmpTOKEN, responseRoot);
if (!success) return false;
success = false;
if (responseRoot != NULL) {
cJSON *tokenItem = cJSON_GetObjectItem(responseRoot, "token");
if (cJSON_IsString(tokenItem) && tokenItem->valuestring != NULL) {
printf("New token received: %s\n", tokenItem->valuestring);
// Save token to NVS
nvs_handle_t AuthHandle;
esp_err_t nvs_err = nvs_open(nvsAuth, NVS_READWRITE, &AuthHandle);
if (nvs_err == ESP_OK) {
nvs_err = nvs_set_str(AuthHandle, tokenTag, tokenItem->valuestring);
if (nvs_err == ESP_OK) {
nvs_commit(AuthHandle);
success = true;
webToken = tokenItem->valuestring;
}
else printf("ERROR: could not save webToken to NVS\n");
nvs_close(AuthHandle);
}
else printf("ERROR: Couldn't open NVS for auth token\n");
}
cJSON_Delete(responseRoot);
}
else printf("Failed to parse JSON response\n");
finalAuth = true;
notifyAuthStatus(success);
if (success) NimBLEDevice::deinit(true); // deinitialize BLE
return success;
}
return false;
}
void reset() { void reset() {
esp_wifi_scan_stop(); esp_wifi_scan_stop();
if (!finalAuth) esp_wifi_disconnect(); if (!finalAuth) esp_wifi_disconnect();
scanBlock = false; scanBlock = false;
flag_scan_requested = false;
credsGiven = false;
tokenGiven = false;
} }
void MyServerCallbacks::onConnect(NimBLEServer* pServer, NimBLEConnInfo& connInfo) { void MyServerCallbacks::onConnect(NimBLEServer* pServer, NimBLEConnInfo& connInfo) {
@@ -295,7 +179,6 @@ void MyCharCallbacks::onWrite(NimBLECharacteristic* pChar, NimBLEConnInfo& connI
else error = true; else error = true;
if (error) { if (error) {
printf("ERROR: Invalid Auth mode passed in with JSON.\n"); printf("ERROR: Invalid Auth mode passed in with JSON.\n");
credsGiven = false;
cJSON_Delete(root); cJSON_Delete(root);
return; return;
} }
@@ -314,13 +197,13 @@ void MyCharCallbacks::onWrite(NimBLECharacteristic* pChar, NimBLEConnInfo& connI
SSID = ssid->valuestring; SSID = ssid->valuestring;
PASS = passPresent ? password->valuestring : ""; PASS = passPresent ? password->valuestring : "";
UNAME = unamePresent ? uname->valuestring : ""; UNAME = unamePresent ? uname->valuestring : "";
credsGiven = tempCredsGiven; // update the global flag. // Signal via event group instead of flag
xEventGroupSetBits(g_system_events, EVENT_BLE_CREDS_RECEIVED);
} }
else printf("ERROR: Did not receive necessary credentials.\n"); else printf("ERROR: Did not receive necessary credentials.\n");
cJSON_Delete(root); cJSON_Delete(root);
} else { } else {
printf("Failed to parse JSON\n"); printf("Failed to parse JSON\n");
credsGiven = false;
} }
} }
} }
@@ -329,14 +212,16 @@ void MyCharCallbacks::onWrite(NimBLECharacteristic* pChar, NimBLEConnInfo& connI
printf("Received Token: %s\n", val.c_str()); printf("Received Token: %s\n", val.c_str());
std::lock_guard<std::mutex> lock(dataMutex); std::lock_guard<std::mutex> lock(dataMutex);
TOKEN = val; TOKEN = val;
tokenGiven = true; // Signal via event group instead of flag
xEventGroupSetBits(g_system_events, EVENT_BLE_TOKEN_RECEIVED);
} }
} }
else if (pChar == currentRefreshChar) { else if (pChar == currentRefreshChar) {
if (val == "Start") { if (val == "Start") {
// Refresh characteristic // Refresh characteristic
printf("Refresh Requested\n"); printf("Refresh Requested\n");
flag_scan_requested = true; // Signal via event group instead of flag
xEventGroupSetBits(g_system_events, EVENT_BLE_SCAN_REQUEST);
} }
else if (val == "Done") { else if (val == "Done") {
printf("Data read complete\n"); printf("Data read complete\n");

View File

@@ -23,6 +23,6 @@ class MyCharCallbacks : public NimBLECharacteristicCallbacks {
}; };
NimBLEAdvertising* initBLE(); NimBLEAdvertising* initBLE();
bool BLEtick(NimBLEAdvertising* pAdvertising); // BLEtick removed - now using event-driven bleSetupTask
#endif #endif

View File

@@ -2,6 +2,33 @@
#define DEFINES_H #define DEFINES_H
#include "driver/gpio.h" #include "driver/gpio.h"
#include "driver/ledc.h" #include "driver/ledc.h"
#include "freertos/FreeRTOS.h"
#include "freertos/event_groups.h"
#include "freertos/queue.h"
#include "freertos/semphr.h"
// Task priorities
#define SERVO_TASK_PRIORITY (tskIDLE_PRIORITY + 4) // Highest - real-time control
#define ENCODER_TASK_PRIORITY (tskIDLE_PRIORITY + 3) // High - encoder processing
#define SOCKETIO_TASK_PRIORITY (tskIDLE_PRIORITY + 2) // Medium
#define BLE_TASK_PRIORITY (tskIDLE_PRIORITY + 1) // Low
#define MAIN_TASK_PRIORITY (tskIDLE_PRIORITY + 1) // Low
// Event bits for system events
#define EVENT_WIFI_CONNECTED BIT0
#define EVENT_SOCKETIO_CONNECTED BIT1
#define EVENT_SOCKETIO_DISCONNECTED BIT2
#define EVENT_CLEAR_CALIB BIT3
#define EVENT_SAVE_POSITION BIT4
#define EVENT_BLE_SCAN_REQUEST BIT5
#define EVENT_BLE_CREDS_RECEIVED BIT6
#define EVENT_BLE_TOKEN_RECEIVED BIT7
// Global synchronization primitives
extern EventGroupHandle_t g_system_events;
extern QueueHandle_t g_servo_command_queue;
extern QueueHandle_t g_encoder_event_queue;
extern SemaphoreHandle_t g_calibration_mutex;
#define ccwSpeed 6500 #define ccwSpeed 6500
#define cwSpeed 3300 #define cwSpeed 3300

View File

@@ -3,6 +3,7 @@
#include "esp_log.h" #include "esp_log.h"
#include "soc/gpio_struct.h" #include "soc/gpio_struct.h"
#include "servo.hpp" #include "servo.hpp"
#include "defines.h"
static const char *TAG = "ENCODER"; static const char *TAG = "ENCODER";
@@ -47,18 +48,32 @@ void IRAM_ATTR Encoder::isr_handler(void* arg)
if (encoder->last_count_base > 3) { if (encoder->last_count_base > 3) {
encoder->count += 1; encoder->count += 1;
encoder->last_count_base -= 4; encoder->last_count_base -= 4;
if (calibListen) servoCalibListen();
if (encoder->feedWDog) esp_timer_restart(encoder->watchdog_handle, 500000); // DEFER to task via queue instead of direct function calls
if (encoder->wandListen) servoWandListen(); encoder_event_t event = {
if (encoder->serverListen) servoServerListen(); .count = encoder->count.load(),
.is_top_encoder = (encoder == topEnc)
};
BaseType_t xHigherPriorityTaskWoken = pdFALSE;
if (g_encoder_event_queue != NULL) {
xQueueSendFromISR(g_encoder_event_queue, &event, &xHigherPriorityTaskWoken);
portYIELD_FROM_ISR(xHigherPriorityTaskWoken);
}
} }
else if (encoder->last_count_base < 0) { else if (encoder->last_count_base < 0) {
encoder->count -= 1; encoder->count -= 1;
encoder->last_count_base += 4; encoder->last_count_base += 4;
if (calibListen) servoCalibListen();
if (encoder->feedWDog) esp_timer_restart(encoder->watchdog_handle, 500000); // DEFER to task via queue instead of direct function calls
if (encoder->wandListen) servoWandListen(); encoder_event_t event = {
if (encoder->serverListen) servoServerListen(); .count = encoder->count.load(),
.is_top_encoder = (encoder == topEnc)
};
BaseType_t xHigherPriorityTaskWoken = pdFALSE;
if (g_encoder_event_queue != NULL) {
xQueueSendFromISR(g_encoder_event_queue, &event, &xHigherPriorityTaskWoken);
portYIELD_FROM_ISR(xHigherPriorityTaskWoken);
}
} }
encoder->last_state_a = current_a; encoder->last_state_a = current_a;

View File

@@ -4,6 +4,12 @@
#include <atomic> #include <atomic>
#include "esp_timer.h" #include "esp_timer.h"
// Encoder event structure for queue
typedef struct {
int32_t count;
bool is_top_encoder;
} encoder_event_t;
class Encoder { class Encoder {
public: public:
// Shared between ISR and main code // Shared between ISR and main code

View File

@@ -115,9 +115,11 @@ void initMainLoop() {
} }
void IRAM_ATTR watchdogCallback(void* arg) { void IRAM_ATTR watchdogCallback(void* arg) {
BaseType_t xHigherPriorityTaskWoken = pdFALSE;
if (runningManual || runningServer) { if (runningManual || runningServer) {
// if we're trying to move and our timer ran out, we need to recalibrate // if we're trying to move and our timer ran out, we need to recalibrate
clearCalibFlag = true; xEventGroupSetBitsFromISR(g_system_events, EVENT_CLEAR_CALIB, &xHigherPriorityTaskWoken);
topEnc->pauseWatchdog(); topEnc->pauseWatchdog();
// get ready for recalibration by clearing all these listeners // get ready for recalibration by clearing all these listeners
@@ -129,11 +131,13 @@ void IRAM_ATTR watchdogCallback(void* arg) {
else { else {
// if no movement is running, we're fine // if no movement is running, we're fine
// save current servo-encoder position for reinitialization // save current servo-encoder position for reinitialization
savePosFlag = true; xEventGroupSetBitsFromISR(g_system_events, EVENT_SAVE_POSITION, &xHigherPriorityTaskWoken);
} }
// clear running flags // clear running flags
runningManual = false; runningManual = false;
runningServer = false; runningServer = false;
portYIELD_FROM_ISR(xHigherPriorityTaskWoken);
} }
void servoSavePos() { void servoSavePos() {
@@ -240,4 +244,69 @@ void runToAppPos(uint8_t appPos) {
if (startLess) servoOn(CCW, server); // begin servo movement if (startLess) servoOn(CCW, server); // begin servo movement
else servoOn(CW, server); else servoOn(CW, server);
topEnc->serverListen = true; // start listening for shutoff point topEnc->serverListen = true; // start listening for shutoff point
}
// Servo control task - processes encoder events and servo commands
void servoControlTask(void* arg) {
encoder_event_t enc_event;
servo_cmd_msg_t cmd;
printf("Servo control task started\n");
while (1) {
// Block waiting for encoder events (higher priority)
if (xQueueReceive(g_encoder_event_queue, &enc_event, pdMS_TO_TICKS(10)) == pdTRUE) {
// Process encoder event (work that was done in ISR before)
// Handle calibration listening
if (calibListen) {
int32_t effDiff = (bottomEnc->getCount() - topEnc->getCount()) - baseDiff;
if (effDiff > 1) {
servoOn(CCW, manual);
}
else if (effDiff < -1) {
servoOn(CW, manual);
}
else {
servoOff();
}
}
// Only process top encoder events for watchdog and listeners
if (enc_event.is_top_encoder) {
// Feed watchdog in task context (not ISR)
if (topEnc->feedWDog) {
esp_timer_restart(topEnc->watchdog_handle, 500000);
}
// Check wand listener - now safe in task context
if (topEnc->wandListen) {
servoWandListen();
}
// Check server listener - now safe in task context
if (topEnc->serverListen) {
servoServerListen();
}
}
}
// Check for direct servo commands (lower priority)
if (xQueueReceive(g_servo_command_queue, &cmd, 0) == pdTRUE) {
switch (cmd.command) {
case SERVO_CMD_STOP:
servoOff();
break;
case SERVO_CMD_MOVE_CCW:
servoOn(CCW, cmd.is_manual ? manual : server);
break;
case SERVO_CMD_MOVE_CW:
servoOn(CW, cmd.is_manual ? manual : server);
break;
case SERVO_CMD_MOVE_TO_POSITION:
runToAppPos(cmd.target_position);
break;
}
}
}
} }

View File

@@ -9,6 +9,20 @@
#define server 1 #define server 1
#define manual 0 #define manual 0
// Command structures for servo control
typedef enum {
SERVO_CMD_STOP,
SERVO_CMD_MOVE_CCW,
SERVO_CMD_MOVE_CW,
SERVO_CMD_MOVE_TO_POSITION
} servo_command_t;
typedef struct {
servo_command_t command;
uint8_t target_position; // For MOVE_TO_POSITION
bool is_manual; // vs server-initiated
} servo_cmd_msg_t;
extern std::atomic<bool> calibListen; extern std::atomic<bool> calibListen;
extern std::atomic<bool> clearCalibFlag; extern std::atomic<bool> clearCalibFlag;
extern std::atomic<bool> savePosFlag; extern std::atomic<bool> savePosFlag;
@@ -35,4 +49,7 @@ void servoWandListen();
void servoServerListen(); void servoServerListen();
void runToAppPos(uint8_t appPos); void runToAppPos(uint8_t appPos);
// Servo control task
void servoControlTask(void* arg);
#endif #endif

View File

@@ -5,14 +5,147 @@
#include "defines.h" #include "defines.h"
#include "bmHTTP.hpp" #include "bmHTTP.hpp"
#include "socketIO.hpp" #include "socketIO.hpp"
#include <mutex>
// External declarations from BLE.cpp
extern std::mutex dataMutex;
extern wifi_auth_mode_t auth;
extern std::string SSID;
extern std::string PASS;
extern std::string UNAME;
extern std::string TOKEN;
extern std::atomic<bool> scanBlock;
extern std::atomic<bool> finalAuth;
// External functions from BLE.cpp
extern void notifyConnectionStatus(bool success);
extern void notifyAuthStatus(bool success);
// BLE setup task - event-driven instead of polling
void bleSetupTask(void* arg) {
NimBLEAdvertising* pAdvertising = initBLE();
EventBits_t bits;
printf("BLE setup task started\n");
while (1) {
// Wait for BLE events instead of polling
bits = xEventGroupWaitBits(
g_system_events,
EVENT_BLE_SCAN_REQUEST | EVENT_BLE_CREDS_RECEIVED | EVENT_BLE_TOKEN_RECEIVED,
pdTRUE, // Clear bits on exit
pdFALSE, // Wait for ANY bit (not all)
portMAX_DELAY // Block forever until event
);
if (bits & EVENT_BLE_SCAN_REQUEST) {
if (!scanBlock) {
scanBlock = true;
printf("Scanning WiFi...\n");
bmWiFi.scanAndUpdateSSIDList();
}
else printf("Duplicate scan request\n");
}
if (bits & EVENT_BLE_CREDS_RECEIVED) {
std::string tmpSSID;
std::string tmpUNAME;
std::string tmpPASS;
wifi_auth_mode_t tmpAUTH;
{
std::lock_guard<std::mutex> lock(dataMutex);
tmpSSID = SSID;
tmpUNAME = UNAME;
tmpPASS = PASS;
tmpAUTH = auth;
}
bool wifiConnect;
if (tmpAUTH == WIFI_AUTH_WPA2_ENTERPRISE || tmpAUTH == WIFI_AUTH_WPA3_ENTERPRISE)
wifiConnect = bmWiFi.attemptConnect(tmpSSID, tmpUNAME, tmpPASS, tmpAUTH);
else wifiConnect = bmWiFi.attemptConnect(tmpSSID, tmpPASS, tmpAUTH);
if (!wifiConnect) {
notifyConnectionStatus(false);
printf("Connection failed\n");
continue;
}
nvs_handle_t WiFiHandle;
esp_err_t err = nvs_open(nvsWiFi, NVS_READWRITE, &WiFiHandle);
if (err == ESP_OK) {
esp_err_t saveErr = ESP_OK;
saveErr |= nvs_set_str(WiFiHandle, ssidTag, tmpSSID.c_str());
saveErr |= nvs_set_str(WiFiHandle, passTag, tmpPASS.c_str());
saveErr |= nvs_set_u8(WiFiHandle, authTag, (uint8_t)tmpAUTH);
if (tmpUNAME.length() > 0) {
saveErr |= nvs_set_str(WiFiHandle, unameTag, tmpUNAME.c_str());
}
if (saveErr == ESP_OK) {
nvs_commit(WiFiHandle);
printf("WiFi credentials saved to NVS\n");
notifyConnectionStatus(true);
} else {
printf("Failed to save WiFi credentials\n");
notifyConnectionStatus(false);
}
nvs_close(WiFiHandle);
}
}
if (bits & EVENT_BLE_TOKEN_RECEIVED) {
std::string tmpTOKEN;
{
std::lock_guard<std::mutex> lock(dataMutex);
tmpTOKEN = TOKEN;
}
cJSON* JSONresponse = NULL;
if (httpGET("api/devices/auth", tmpTOKEN, JSONresponse)) {
cJSON *success = cJSON_GetObjectItem(JSONresponse, "success");
bool authSuccess = cJSON_IsTrue(success);
if (authSuccess) {
nvs_handle_t authHandle;
if (nvs_open(nvsAuth, NVS_READWRITE, &authHandle) == ESP_OK) {
if (nvs_set_str(authHandle, tokenTag, tmpTOKEN.c_str()) == ESP_OK) {
nvs_commit(authHandle);
printf("Token saved to NVS\n");
notifyAuthStatus(true);
finalAuth = true;
// Task complete - delete itself
vTaskDelete(NULL);
}
nvs_close(authHandle);
}
} else {
printf("Token authentication failed\n");
notifyAuthStatus(false);
}
cJSON_Delete(JSONresponse);
} else {
printf("HTTP request failed\n");
notifyAuthStatus(false);
}
}
}
}
void initialSetup() { void initialSetup() {
printf("Entered Setup\n"); printf("Entered Setup\n");
NimBLEAdvertising* pAdv = initBLE();
// Create BLE setup task instead of polling loop
while (!BLEtick(pAdv)) { xTaskCreate(bleSetupTask, "ble_setup", 8192, NULL, BLE_TASK_PRIORITY, NULL);
// Wait for BLE setup to complete (finalAuth = true)
while (!finalAuth) {
vTaskDelay(pdMS_TO_TICKS(100)); vTaskDelay(pdMS_TO_TICKS(100));
} }
printf("BLE setup complete\n");
} }
void setupLoop() { void setupLoop() {
@@ -56,29 +189,28 @@ void setupLoop() {
// The server will verify the token during connection handshake // The server will verify the token during connection handshake
webToken = std::string(token); webToken = std::string(token);
printf("Connecting to Socket.IO server with saved token...\n"); printf("Connecting to Socket.IO server with saved token...\n");
statusResolved = false;
initSocketIO(); initSocketIO();
// Wait for device_init message from server with timeout // Wait for connection event with timeout
int timeout_count = 0; EventBits_t bits = xEventGroupWaitBits(
const int MAX_TIMEOUT = 60; // 10 seconds (20 * 500ms) g_system_events,
while (!statusResolved && timeout_count < MAX_TIMEOUT) { EVENT_SOCKETIO_CONNECTED | EVENT_SOCKETIO_DISCONNECTED,
printf("Waiting for device_init message... (%d/%d)\n", timeout_count, MAX_TIMEOUT); pdTRUE, // Clear on exit
vTaskDelay(pdMS_TO_TICKS(500)); pdFALSE, // Wait for ANY bit
timeout_count++; pdMS_TO_TICKS(10000) // 10 second timeout
} );
if (timeout_count >= MAX_TIMEOUT) { if (bits & EVENT_SOCKETIO_CONNECTED) {
printf("Timeout waiting for device_init - connection failed\n");
stopSocketIO();
initSuccess = false;
initialSetup();
} else {
initSuccess = connected; initSuccess = connected;
if (!initSuccess) { if (!initSuccess) {
printf("Device authentication failed - entering setup\n"); printf("Device authentication failed - entering setup\n");
initialSetup(); initialSetup();
} }
} else {
printf("Timeout waiting for connection\n");
stopSocketIO();
initSuccess = false;
initialSetup();
} }
} }
else { else {

View File

@@ -6,11 +6,11 @@
#include "cJSON.h" #include "cJSON.h"
#include "calibration.hpp" #include "calibration.hpp"
#include "servo.hpp" #include "servo.hpp"
#include "defines.h"
static esp_socketio_client_handle_t io_client; static esp_socketio_client_handle_t io_client;
static esp_socketio_packet_handle_t tx_packet = NULL; static esp_socketio_packet_handle_t tx_packet = NULL;
std::atomic<bool> statusResolved{true};
std::atomic<bool> connected{false}; std::atomic<bool> connected{false};
// Event handler for Socket.IO events // Event handler for Socket.IO events
@@ -64,7 +64,7 @@ static void socketio_event_handler(void *handler_args, esp_event_base_t base,
// Mark connection as failed // Mark connection as failed
connected = false; connected = false;
statusResolved = true; xEventGroupSetBits(g_system_events, EVENT_SOCKETIO_DISCONNECTED);
} }
// Handle device_init event // Handle device_init event
else if (strcmp(eventName->valuestring, "device_init") == 0) { else if (strcmp(eventName->valuestring, "device_init") == 0) {
@@ -101,13 +101,13 @@ static void socketio_event_handler(void *handler_args, esp_event_base_t base,
// Now mark as connected // Now mark as connected
connected = true; connected = true;
statusResolved = true; xEventGroupSetBits(g_system_events, EVENT_SOCKETIO_CONNECTED);
} else { } else {
printf("Device authentication failed\n"); printf("Device authentication failed\n");
calib.clearCalibrated(); calib.clearCalibrated();
deleteWiFiAndTokenDetails(); deleteWiFiAndTokenDetails();
connected = false; connected = false;
statusResolved = true; xEventGroupSetBits(g_system_events, EVENT_SOCKETIO_DISCONNECTED);
} }
} }
} }
@@ -124,7 +124,7 @@ static void socketio_event_handler(void *handler_args, esp_event_base_t base,
calib.clearCalibrated(); calib.clearCalibrated();
deleteWiFiAndTokenDetails(); deleteWiFiAndTokenDetails();
connected = false; connected = false;
statusResolved = true; xEventGroupSetBits(g_system_events, EVENT_SOCKETIO_DISCONNECTED);
} }
// Handle calib_start event // Handle calib_start event
@@ -236,9 +236,9 @@ static void socketio_event_handler(void *handler_args, esp_event_base_t base,
} }
} }
// Set flags to indicate connection failure // Signal disconnection via event group
connected = false; connected = false;
statusResolved = true; xEventGroupSetBits(g_system_events, EVENT_SOCKETIO_DISCONNECTED);
break; break;
} }
} }
@@ -247,7 +247,7 @@ static void socketio_event_handler(void *handler_args, esp_event_base_t base,
if (data->websocket_event_id == WEBSOCKET_EVENT_DISCONNECTED) { if (data->websocket_event_id == WEBSOCKET_EVENT_DISCONNECTED) {
printf("WebSocket disconnected\n"); printf("WebSocket disconnected\n");
connected = false; connected = false;
statusResolved = true; xEventGroupSetBits(g_system_events, EVENT_SOCKETIO_DISCONNECTED);
} }
} }
@@ -255,7 +255,6 @@ void initSocketIO() {
// Prepare the Authorization Header (Bearer format) // Prepare the Authorization Header (Bearer format)
std::string authHeader = "Authorization: Bearer " + webToken + "\r\n"; std::string authHeader = "Authorization: Bearer " + webToken + "\r\n";
statusResolved = false;
connected = false; connected = false;
esp_socketio_client_config_t config = {}; esp_socketio_client_config_t config = {};
@@ -277,7 +276,6 @@ void stopSocketIO() {
io_client = NULL; io_client = NULL;
tx_packet = NULL; tx_packet = NULL;
connected = false; connected = false;
statusResolved = false;
} }
} }

View File

@@ -2,7 +2,6 @@
#define SOCKETIO_HPP #define SOCKETIO_HPP
#include <atomic> #include <atomic>
extern std::atomic<bool> statusResolved;
extern std::atomic<bool> connected; extern std::atomic<bool> connected;
// Initialize Socket.IO client and connect to server // Initialize Socket.IO client and connect to server

View File

@@ -9,59 +9,54 @@
#include "encoder.hpp" #include "encoder.hpp"
#include "calibration.hpp" #include "calibration.hpp"
// Global synchronization primitives
EventGroupHandle_t g_system_events = NULL;
QueueHandle_t g_servo_command_queue = NULL;
QueueHandle_t g_encoder_event_queue = NULL;
SemaphoreHandle_t g_calibration_mutex = NULL;
// Global encoder instances // Global encoder instances
Encoder* topEnc = new Encoder(ENCODER_PIN_A, ENCODER_PIN_B); Encoder* topEnc = new Encoder(ENCODER_PIN_A, ENCODER_PIN_B);
Encoder* bottomEnc = new Encoder(InputEnc_PIN_A, InputEnc_PIN_B); Encoder* bottomEnc = new Encoder(InputEnc_PIN_A, InputEnc_PIN_B);
// Global encoder pointers (used by servo.cpp)
// Global calibration instance // Global calibration instance
Calibration calib; Calibration calib;
void mainApp() { void mainTask(void* arg) {
esp_err_t ret = nvs_flash_init(); // change to secure init logic soon!! EventBits_t bits;
// 2. If NVS is full or corrupt (common after flashing new code), erase and retry
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
ESP_ERROR_CHECK(nvs_flash_erase());
ret = nvs_flash_init();
}
ESP_ERROR_CHECK(ret);
bmWiFi.init();
calib.init();
// Initialize encoders printf("Main task started - event-driven mode\n");
topEnc->init();
bottomEnc->init();
servoInit();
setupLoop();
statusResolved = false;
int32_t prevCount = topEnc->getCount();
// Main loop
while (1) { while (1) {
// websocket disconnect/reconnect handling // Block waiting for ANY event (no polling!)
if (statusResolved) { bits = xEventGroupWaitBits(
g_system_events,
EVENT_SOCKETIO_DISCONNECTED | EVENT_CLEAR_CALIB | EVENT_SAVE_POSITION,
pdTRUE, // Clear bits on exit
pdFALSE, // Wait for ANY bit (not all)
portMAX_DELAY // Block forever - no polling!
);
if (bits & EVENT_SOCKETIO_DISCONNECTED) {
if (!connected) { if (!connected) {
printf("Disconnected! Beginning setup loop.\n"); printf("Disconnected! Beginning setup loop.\n");
stopSocketIO(); stopSocketIO();
setupLoop(); setupLoop();
} }
else printf("Reconnected!\n"); else {
statusResolved = false; printf("Reconnected!\n");
}
} }
if (clearCalibFlag) { if (bits & EVENT_CLEAR_CALIB) {
xSemaphoreTake(g_calibration_mutex, portMAX_DELAY);
calib.clearCalibrated(); calib.clearCalibrated();
xSemaphoreGive(g_calibration_mutex);
emitCalibStatus(false); emitCalibStatus(false);
clearCalibFlag = false;
} }
if (savePosFlag) {
if (bits & EVENT_SAVE_POSITION) {
servoSavePos(); servoSavePos();
savePosFlag = false;
// Send position update to server // Send position update to server
uint8_t currentAppPos = calib.convertToAppPos(topEnc->getCount()); uint8_t currentAppPos = calib.convertToAppPos(topEnc->getCount());
@@ -69,7 +64,6 @@ void mainApp() {
printf("Sent pos_hit: position %d\n", currentAppPos); printf("Sent pos_hit: position %d\n", currentAppPos);
} }
vTaskDelay(pdMS_TO_TICKS(100));
} }
} }
@@ -91,6 +85,44 @@ void encoderTest() {
} }
extern "C" void app_main() { extern "C" void app_main() {
mainApp(); // Initialize NVS first
// encoderTest(); esp_err_t ret = nvs_flash_init();
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
ESP_ERROR_CHECK(nvs_flash_erase());
ret = nvs_flash_init();
}
ESP_ERROR_CHECK(ret);
// Create synchronization primitives
g_system_events = xEventGroupCreate();
g_servo_command_queue = xQueueCreate(10, sizeof(servo_cmd_msg_t));
g_encoder_event_queue = xQueueCreate(50, sizeof(encoder_event_t));
g_calibration_mutex = xSemaphoreCreateMutex();
if (g_system_events == NULL || g_servo_command_queue == NULL ||
g_encoder_event_queue == NULL || g_calibration_mutex == NULL) {
printf("ERROR: Failed to create synchronization primitives\n");
return;
}
// Initialize hardware
bmWiFi.init();
calib.init();
// Initialize encoders
topEnc->init();
bottomEnc->init();
servoInit();
// Create servo control task (highest priority - real-time)
xTaskCreate(servoControlTask, "servo_ctrl", 4096, NULL, SERVO_TASK_PRIORITY, NULL);
// Create main task (lower priority)
xTaskCreate(mainTask, "main", 4096, NULL, MAIN_TASK_PRIORITY, NULL);
// Run setup loop (this will handle WiFi/SocketIO connection)
setupLoop();
// app_main returns, but tasks continue running
printf("app_main complete - tasks running\n");
} }