88 lines
2.8 KiB
C++
88 lines
2.8 KiB
C++
#pragma once
|
|
#ifdef ESP32
|
|
#include <HardwareSerial.h>
|
|
typedef HardwareSerial SerialType;
|
|
#define SERIAL_TYPE HardwareSerial
|
|
#elif defined(ESP8266)
|
|
#include <SoftwareSerial.h>
|
|
typedef SoftwareSerial SerialType;
|
|
#define SERIAL_TYPE SoftwareSerial
|
|
#else
|
|
#error "Unsupported platform. Only ESP32 and ESP8266 are supported."
|
|
#endif
|
|
|
|
#include <pb_encode.h>
|
|
#include <pb_decode.h>
|
|
#include "hardware.pb.h"
|
|
|
|
// Define the role of this device
|
|
// Uncomment one of the following defines to set the role
|
|
//#define HARDWARE_SERIAL_ROLE_CONTROL
|
|
#define HARDWARE_SERIAL_ROLE_SENSOR
|
|
|
|
// Based on role, define the message types
|
|
#if defined(HARDWARE_SERIAL_ROLE_CONTROL)
|
|
typedef hardware_SensorToControlMessage IncomingMessage;
|
|
typedef hardware_ControlToSensorMessage OutgoingMessage;
|
|
#define INCOMING_MESSAGE_SIZE hardware_SensorToControlMessage_size
|
|
#define OUTGOING_MESSAGE_SIZE hardware_ControlToSensorMessage_size
|
|
#define INCOMING_MESSAGE_FIELDS hardware_SensorToControlMessage_fields
|
|
#define OUTGOING_MESSAGE_FIELDS hardware_ControlToSensorMessage_fields
|
|
#define INCOMING_MESSAGE_INIT hardware_SensorToControlMessage_init_default
|
|
#define OUTGOING_MESSAGE_INIT hardware_ControlToSensorMessage_init_default
|
|
#elif defined(HARDWARE_SERIAL_ROLE_SENSOR)
|
|
typedef hardware_ControlToSensorMessage IncomingMessage;
|
|
typedef hardware_SensorToControlMessage OutgoingMessage;
|
|
#define INCOMING_MESSAGE_SIZE hardware_ControlToSensorMessage_size
|
|
#define OUTGOING_MESSAGE_SIZE hardware_SensorToControlMessage_size
|
|
#define INCOMING_MESSAGE_FIELDS hardware_ControlToSensorMessage_fields
|
|
#define OUTGOING_MESSAGE_FIELDS hardware_SensorToControlMessage_fields
|
|
#define INCOMING_MESSAGE_INIT hardware_ControlToSensorMessage_init_default
|
|
#define OUTGOING_MESSAGE_INIT hardware_SensorToControlMessage_init_default
|
|
#else
|
|
#error "Must define either HARDWARE_SERIAL_ROLE_CONTROL or HARDWARE_SERIAL_ROLE_SENSOR"
|
|
#endif
|
|
|
|
class ProtoSerial {
|
|
public:
|
|
ProtoSerial();
|
|
~ProtoSerial();
|
|
|
|
// Initialize with RX and TX pins
|
|
void begin(int rxPin, int txPin, long baud = 9600);
|
|
|
|
// Send a message
|
|
bool sendMessage(const OutgoingMessage& message);
|
|
|
|
// Check for incoming messages (non-blocking)
|
|
void update();
|
|
|
|
// Check if a message has been received
|
|
bool hasMessage() const;
|
|
|
|
// Get the received message (only valid if hasMessage() returns true)
|
|
const IncomingMessage& getMessage() const;
|
|
|
|
// Clear the received message
|
|
void clearMessage();
|
|
|
|
private:
|
|
SerialType* serial;
|
|
bool initialized;
|
|
|
|
// Receive state machine
|
|
enum RxState {
|
|
WAITING_FOR_LENGTH,
|
|
READING_PAYLOAD
|
|
};
|
|
RxState currentState;
|
|
uint16_t payloadLength;
|
|
uint8_t payloadBuffer[64]; // Max message size + some buffer
|
|
IncomingMessage receivedMessage;
|
|
bool messageAvailable;
|
|
|
|
// Helper methods
|
|
void processReceivedMessage(uint8_t* buffer, uint16_t length);
|
|
};
|
|
|