Merged i2c-helper in Main
This commit is contained in:
101
vacum-control/vacum-control-firmware/src/base.cpp
Normal file
101
vacum-control/vacum-control-firmware/src/base.cpp
Normal file
@@ -0,0 +1,101 @@
|
||||
#include "vacum.hpp"
|
||||
#include "pinConfigNode.hpp"
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "semphr.h"
|
||||
|
||||
#include "pico/stdlib.h"
|
||||
|
||||
void vInterruptionTaskFn(void *pvParameters) {
|
||||
VacumControl *vacumControl = (VacumControl *)pvParameters;
|
||||
vacumControl->interruptionTaskFn();
|
||||
}
|
||||
|
||||
void vButtonTaskFn(void *pvParameters) {
|
||||
VacumControl *vacumControl = (VacumControl *)pvParameters;
|
||||
vacumControl->buttonTaskFn();
|
||||
}
|
||||
|
||||
VacumControl::VacumControl(uint coreMask) {
|
||||
this->coreMask = coreMask;
|
||||
initVacumPins();
|
||||
state = false;
|
||||
vacumMutex = xSemaphoreCreateMutex();
|
||||
interruptionSemaphore = xSemaphoreCreateBinary();
|
||||
xTaskCreateAffinitySet(vInterruptionTaskFn, "Interruption Task", 1024, this, 3, coreMask, &interruptionTask);
|
||||
xTaskCreateAffinitySet(vButtonTaskFn, "Button Task", 1024, this, 2, coreMask, &buttonTask);
|
||||
}
|
||||
|
||||
void VacumControl::initVacumPins() {
|
||||
gpio_init(LED1_PIN);
|
||||
gpio_set_dir(LED1_PIN, GPIO_OUT);
|
||||
gpio_put(LED1_PIN, 0);
|
||||
|
||||
gpio_init(LED2_PIN);
|
||||
gpio_set_dir(LED2_PIN, GPIO_OUT);
|
||||
gpio_put(LED2_PIN, 0);
|
||||
|
||||
gpio_init(RELAY_PIN);
|
||||
gpio_set_dir(RELAY_PIN, GPIO_OUT);
|
||||
gpio_put(RELAY_PIN, 0);
|
||||
|
||||
gpio_init(BUTTON_PIN);
|
||||
gpio_set_dir(BUTTON_PIN, GPIO_IN);
|
||||
gpio_pull_up(BUTTON_PIN);
|
||||
}
|
||||
|
||||
void VacumControl::setState(bool on) {
|
||||
xSemaphoreTake(vacumMutex, portMAX_DELAY);
|
||||
if(on) {
|
||||
gpio_put(RELAY_PIN, 1);
|
||||
gpio_put(LED1_PIN, 1);
|
||||
state = true;
|
||||
xSemaphoreGive(interruptionSemaphore);
|
||||
} else {
|
||||
gpio_put(RELAY_PIN, 0);
|
||||
gpio_put(LED1_PIN, 0);
|
||||
state = false;
|
||||
}
|
||||
xSemaphoreGive(vacumMutex);
|
||||
}
|
||||
|
||||
void VacumControl::interruptionTaskFn() {
|
||||
while(true) {
|
||||
xSemaphoreTake(interruptionSemaphore, portMAX_DELAY);
|
||||
if(state) {
|
||||
while(state) {
|
||||
vTaskDelay(INTERRUPTION_DELAY / portTICK_PERIOD_MS);
|
||||
if(state) {
|
||||
gpio_put(RELAY_PIN, 0);
|
||||
gpio_put(LED1_PIN, 0);
|
||||
} else
|
||||
break;
|
||||
|
||||
vTaskDelay(INTERRUPTION_DURATION / portTICK_PERIOD_MS);
|
||||
if(state) {
|
||||
gpio_put(RELAY_PIN, 1);
|
||||
gpio_put(LED1_PIN, 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void VacumControl::buttonTaskFn() {
|
||||
bool buttonLastState = gpio_get(BUTTON_PIN);
|
||||
|
||||
while(true) {
|
||||
bool buttonState = gpio_get(BUTTON_PIN);
|
||||
|
||||
if(!buttonState && buttonLastState) {
|
||||
setState(true);
|
||||
vTaskDelay(250 / portTICK_PERIOD_MS);
|
||||
} else if(buttonState && !buttonLastState) {
|
||||
setState(false);
|
||||
}
|
||||
|
||||
buttonLastState = buttonState;
|
||||
vTaskDelay(250 / portTICK_PERIOD_MS);
|
||||
}
|
||||
}
|
||||
119
vacum-control/vacum-control-firmware/src/main.cpp
Normal file
119
vacum-control/vacum-control-firmware/src/main.cpp
Normal file
@@ -0,0 +1,119 @@
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include "pico/stdlib.h"
|
||||
|
||||
#include "vacum.hpp"
|
||||
|
||||
#include "FreeRTOSConfig.h"
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "queue.h"
|
||||
|
||||
#include "hardware/i2c.h"
|
||||
|
||||
#include "pinConfigNode.hpp"
|
||||
#include "node_interface_hardware.hpp"
|
||||
#include "node_interface.hpp"
|
||||
#include "protocol.hpp"
|
||||
|
||||
struct AppData {
|
||||
xQueueHandle txQueue;
|
||||
xQueueHandle rxQueue;
|
||||
VacumControl * vacumControl;
|
||||
GobotRPC_NI * gobotrpc_ni;
|
||||
};
|
||||
|
||||
AppData g_appData;
|
||||
|
||||
void onRxPackage(void * args, char *data, uint16_t len, GobotRPCTypes type, GobotRPCNumber number) {
|
||||
AppData * appData = (AppData *) args;
|
||||
char txBuffer[32];
|
||||
|
||||
switch (number) {
|
||||
case GobotRPCNumber::VACUM: {
|
||||
GobotRPCPackage_Req_Vacum * pkg = (GobotRPCPackage_Req_Vacum *) data;
|
||||
appData->vacumControl->setState(pkg->enable);
|
||||
g_appData.gobotrpc_ni->sendPackage(txBuffer, 0, GobotRPCTypes::RESPONSE, GobotRPCNumber::VACUM);
|
||||
break;
|
||||
}
|
||||
|
||||
case GobotRPCNumber::GET_INFO: {
|
||||
InfoData info = appData->gobotrpc_ni->getInfo();
|
||||
txBuffer[0] = (info.addr >> 24) & 0xFF;
|
||||
txBuffer[1] = (info.addr >> 16) & 0xFF;
|
||||
txBuffer[2] = (info.addr >> 8) & 0xFF;
|
||||
txBuffer[3] = info.addr & 0xFF;
|
||||
txBuffer[4] = info.type;
|
||||
|
||||
g_appData.gobotrpc_ni->sendPackage(txBuffer, 5, GobotRPCTypes::RESPONSE, GobotRPCNumber::GET_INFO);
|
||||
break;
|
||||
}
|
||||
|
||||
case GobotRPCNumber::RESET: {
|
||||
softwareReset();
|
||||
break;
|
||||
}
|
||||
|
||||
default: {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void core1_main(void *pvParameters) {
|
||||
AppData * appData = (AppData *) pvParameters;
|
||||
GobotRPC_NI_Hardware_RP2040_I2C gobotrpc_ni_hardware(
|
||||
appData->txQueue, appData->rxQueue, CORE_MASK_VACUM,
|
||||
i2c0, I2C_ADDR, GORPC_SDA_PIN, GORPC_SCL_PIN, GORPC_INT_PIN
|
||||
);
|
||||
|
||||
while (true) {
|
||||
gpio_put(LED2_PIN, 0);
|
||||
vTaskDelay(500 / portTICK_PERIOD_MS);
|
||||
gpio_put(LED2_PIN, 1);
|
||||
vTaskDelay(500 / portTICK_PERIOD_MS);
|
||||
}
|
||||
}
|
||||
|
||||
TaskHandle_t xCore0TaskHandle = NULL;
|
||||
TaskHandle_t xCore1TaskHandle = NULL;
|
||||
|
||||
int main() {
|
||||
stdio_init_all();
|
||||
printf("Hello, world!\n");
|
||||
|
||||
gpio_init(LED1_PIN);
|
||||
gpio_set_dir(LED1_PIN, GPIO_OUT);
|
||||
gpio_put(LED1_PIN, 1);
|
||||
|
||||
gpio_init(LED2_PIN);
|
||||
gpio_set_dir(LED2_PIN, GPIO_OUT);
|
||||
gpio_put(LED2_PIN, 1);
|
||||
|
||||
g_appData.txQueue = xQueueCreate(16, sizeof(GobotRPC_NI_Package_Transport));
|
||||
g_appData.rxQueue = xQueueCreate(16, sizeof(GobotRPC_NI_Package_Transport));
|
||||
|
||||
VacumControl vacumControl = VacumControl(CORE_MASK_VACUM);
|
||||
vacumControl.setState(false);
|
||||
g_appData.vacumControl = &vacumControl;
|
||||
|
||||
InfoData info = {.addr=I2C_ADDR, .type=NODE_TYPE_VACUM};
|
||||
|
||||
GobotRPC_NI gobotrpc_ni(CORE_MASK_VACUM, g_appData.txQueue, g_appData.rxQueue, info);
|
||||
gobotrpc_ni.registerOnPackageRxCallback(onRxPackage, &g_appData);
|
||||
|
||||
g_appData.gobotrpc_ni = &gobotrpc_ni;
|
||||
|
||||
xTaskCreateAffinitySet(
|
||||
core1_main, "Core 1 Main", 2048, &g_appData, 1, CORE_MASK_VACUM, &xCore1TaskHandle
|
||||
);
|
||||
|
||||
char txBuffer[32];
|
||||
g_appData.gobotrpc_ni->sendPackage(txBuffer, 0, GobotRPCTypes::RESPONSE, GobotRPCNumber::RESET);
|
||||
|
||||
vTaskStartScheduler();
|
||||
|
||||
while (true) {
|
||||
tight_loop_contents();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user