1153 lines
30 KiB
C++
1153 lines
30 KiB
C++
|
|
#include <base64.h>
|
|
#include "feetech.h"
|
|
#include "animation.h"
|
|
#include "nodegraph.h"
|
|
|
|
#define DEVICE_NAME "Little Sophia"
|
|
#define FIRMWARE_VERSION "0.0.1"
|
|
|
|
#define HEADER1 0xAA
|
|
#define HEADER2 0x55
|
|
#define MAX_PAYLOAD_SIZE 6000 // 10 KB
|
|
uint8_t payload[MAX_PAYLOAD_SIZE]; // Global or static
|
|
|
|
#define CMD_ID_REQUEST 0x01
|
|
#define CMD_FILE_LIST 0x02
|
|
#define CMD_LOAD_FILE 0x03
|
|
#define CMD_DELETE_FILE 0x04
|
|
#define CMD_SAVE_FILE 0x05
|
|
#define CMD_MESSAGE 0x06
|
|
#define CMD_SET_POSITION 0x07
|
|
#define CMD_PLAY_FILE 0x08
|
|
#define CMD_SCAN_CHANNEL 0x09
|
|
#define CMD_WRITE_DATA 0x10
|
|
#define CMD_START_POSITION_STREAM 0x14
|
|
#define POSITION_STREAM 0x15
|
|
|
|
|
|
// ESP32 S2 PINOUT
|
|
#define CH0_RX_PIN 13
|
|
#define CH0_TX_PIN 12
|
|
|
|
// #define RX_PIN 18 // RO
|
|
// #define TX_PIN 17 // DI
|
|
#define CH1_RX_PIN 11 // RO
|
|
#define CH1_TX_PIN 10 // DI
|
|
#define DE_PIN 7 // Driver Enable
|
|
#define RE_PIN 8 // Receiver Enable
|
|
|
|
Animation anim;
|
|
Animation sweep;
|
|
Animation stagger;
|
|
|
|
bool streamPositions = false;
|
|
unsigned long lastStreamPositions = 0;
|
|
|
|
Feetech* servos[2];
|
|
|
|
uint16_t flipBytes(uint16_t value) {
|
|
return (value >> 8) | (value << 8);
|
|
}
|
|
|
|
|
|
uint8_t ids[NUM_CHANNELS] = { 10, 11, 12, 13, 14 };
|
|
uint16_t pos1[] = { 0, 0, 0, 0, 0 };
|
|
uint16_t pos2[] = { 1023, 1023, 1023, 1023, 1023 };
|
|
|
|
uint8_t idsSTS[NUM_CHANNELS] = { 15, 103 };
|
|
uint16_t pos1STS[] = { 0, 0 };
|
|
uint16_t pos2STS[] = { 1023, 1023 };
|
|
|
|
const uint8_t testPayload[] = {
|
|
// 🔹 Filename block (length = 17)
|
|
17, 0,
|
|
'/', 'p', 'o', 'i', 'n', 't', 'c', 'u', 'r', 'v', 'e', '2', '.', 'a', 'n', 'i', 'm',
|
|
|
|
// 🔹 Header: "ANIM", frameCount=800, version=1, frameRate=50, reserved[8]
|
|
'A', 'N', 'I', 'M',
|
|
32, 3, // frameCount = 800 (little-endian)
|
|
1, // version
|
|
50, // frameRate
|
|
0, 0, 0, 0, 0, 0, 0, 0, // reserved
|
|
|
|
// 🔹 Curve count = 2
|
|
2, 0,
|
|
|
|
// 🔹 Curve Segment 1
|
|
10, // motorID
|
|
0, 0, // startTime
|
|
40, 0, // endTime
|
|
44, 1, // startPointY = 300
|
|
10, 0, // startHandleX
|
|
44, 1, // startHandleY = 300
|
|
30, 0, // endHandleX
|
|
0, 0, // endHandleY
|
|
0, 0, // endPointY
|
|
|
|
// 🔹 Curve Segment 2
|
|
10, // motorID
|
|
40, 0, // startTime
|
|
32, 3, // endTime = 800
|
|
0, 0, // startPointY
|
|
200, 0, // startHandleX
|
|
0, 0, // startHandleY
|
|
88, 2, // endHandleX = 600
|
|
44, 1, // endHandleY = 300
|
|
44, 1 // endPointY = 300
|
|
};
|
|
|
|
const uint16_t testPayloadLength = sizeof(testPayload);
|
|
|
|
|
|
void setup() {
|
|
Serial.begin(1000000);
|
|
Serial.setRxBufferSize(1024);
|
|
for (int i = 0; i < 5; i++) {
|
|
Serial.println(i);
|
|
delay(500);
|
|
}
|
|
|
|
servos[0] = new Feetech(Serial1, DE_PIN, RE_PIN, CH0_TX_PIN, CH0_RX_PIN); // SCS
|
|
servos[0]->setFeetechMode(Feetech::MODE_SCS);
|
|
servos[1] = new Feetech(Serial2, DE_PIN, RE_PIN, CH1_TX_PIN, CH1_RX_PIN); // STS
|
|
servos[1]->setFeetechMode(Feetech::MODE_STS);
|
|
|
|
// pinMode(RX_PIN, OUTPUT);
|
|
// pinMode(TX_PIN, OUTPUT);
|
|
// pinMode(DE_PIN, OUTPUT);
|
|
// while (true){
|
|
// Serial.println(!digitalRead(RX_PIN));
|
|
// digitalWrite(RX_PIN, !digitalRead(RX_PIN));
|
|
// digitalWrite(TX_PIN, !digitalRead(TX_PIN));
|
|
// digitalWrite(DE_PIN, !digitalRead(DE_PIN));
|
|
// delay(2000);
|
|
// }
|
|
|
|
|
|
pos2[3] = flipBytes(pos2[3]);
|
|
|
|
servos[0]->begin();
|
|
servos[1]->begin();
|
|
|
|
if (!FFat.begin(true)) {
|
|
Serial.println("FFat mount failed");
|
|
return;
|
|
}
|
|
|
|
|
|
|
|
// anim.saveToFile("/scurve.anim");
|
|
// Serial.println("SAVED");
|
|
//handleSaveFile(testPayload, testPayloadLength);
|
|
Serial.println("Loading test.anim");
|
|
anim.loadFromFile("/test.anim");
|
|
Serial.println(anim.header.frameCount);
|
|
anim.printCurves();
|
|
// Serial.print("CurveSegment size: ");
|
|
// Serial.println(sizeof(CurveSegment));
|
|
//anim.createBasicSCurve();
|
|
// Serial.println("loading");
|
|
//anim.saveToFile("/pointcurve.anim");
|
|
//playAnimation(anim);
|
|
// Serial.println("DONE");
|
|
// anim.createEaseOutCurve();
|
|
// playAnimation(anim);
|
|
|
|
//anim.printCurves();
|
|
// PrintFileList();
|
|
// anim.loadFromFile("/bob.anim");
|
|
// anim.printKeyframes();
|
|
// playAnimation(anim);
|
|
// ClearFiles();
|
|
// // PrintFileList();
|
|
// sweep.clear();
|
|
// sweep.createSampleSweep(4);
|
|
// sweep.saveToFile("/sweep.anim");
|
|
// stagger.clear();
|
|
// stagger.createStaggeredSweep(4);
|
|
// stagger.saveToFile("/stagger.anim");
|
|
// //delay(9999);
|
|
// sweep.printKeyframes();
|
|
|
|
|
|
// anim.clear();
|
|
// if (!sweep.loadFromFile("/sweep.anim")) {
|
|
// Serial.println("Failed to load animation");
|
|
// return;
|
|
// } else {
|
|
// Serial.println("Loaded sweep anim");
|
|
// }
|
|
|
|
// if (!stagger.loadFromFile("/stagger.anim")) {
|
|
// Serial.println("Failed to load animation");
|
|
// return;
|
|
// } else {
|
|
// Serial.println("Loaded stagger anim");
|
|
// }
|
|
|
|
|
|
// Serial.println(anim.getFrame(0, 0)); // Should show saved value
|
|
|
|
//SetID(12, 16);
|
|
|
|
//Serial.println(servos[1]->setLockSTS(16, 0));
|
|
// Serial.println("Enable torque");
|
|
// Serial.println(servos[0]->enableTorque(13));
|
|
// Serial.println(servos[1]->enableTorque(13));
|
|
// delay(3000);
|
|
// Serial.println("Disable torque");
|
|
// Serial.println(servos[0]->disableTorque(13));
|
|
// Serial.println(servos[1]->disableTorque(13));
|
|
// delay(3000);
|
|
// servos[0]->setMinAngleLimit(13, 100);
|
|
// servos[1]->setMinAngleLimit(13, 900);
|
|
// servos[0]->setMaxAngleLimit(13, 500);
|
|
// servos[1]->setMaxAngleLimit(13, 3900);
|
|
}
|
|
|
|
void SetID(uint8_t oldID, uint8_t newID) {
|
|
Serial.println("Setting Lock to 0");
|
|
Serial.println(servos[0]->setLock(oldID, 0));
|
|
delay(1000);
|
|
Serial.print("Changing ID ");
|
|
Serial.print(oldID);
|
|
Serial.print(" to ");
|
|
Serial.println(newID);
|
|
Serial.println(servos[0]->setID(oldID, newID));
|
|
delay(1000);
|
|
Serial.println("Setting Lock to 1");
|
|
Serial.println(servos[0]->setLock(newID, 1));
|
|
delay(1000);
|
|
}
|
|
|
|
void HandleSerialRequests() {
|
|
if (Serial.available() >= 6) { // 2 headers + 1 command + 2 length + 1 checksum minimum
|
|
if (Serial.read() == HEADER1 && Serial.read() == HEADER2) {
|
|
uint8_t command = Serial.read();
|
|
uint16_t length = (Serial.read() << 8) | Serial.read();
|
|
|
|
if (length > MAX_PAYLOAD_SIZE) {
|
|
Serial.println("Payload too large");
|
|
while (Serial.available()) Serial.read(); // flush junk
|
|
return;
|
|
}
|
|
|
|
unsigned long start = millis();
|
|
while (Serial.available() < length + 1) {
|
|
if (millis() - start > 1000) { // 1 second timeout
|
|
Serial.println("Serial timeout");
|
|
return;
|
|
}
|
|
}
|
|
|
|
|
|
uint8_t checksum = command ^ (length >> 8) ^ (length & 0xFF);
|
|
|
|
for (uint16_t i = 0; i < length; i++) {
|
|
payload[i] = Serial.read();
|
|
checksum ^= payload[i];
|
|
}
|
|
|
|
uint8_t receivedChecksum = Serial.read();
|
|
|
|
if (checksum == receivedChecksum) {
|
|
handleCommand(command, payload, length);
|
|
} else {
|
|
Serial.println("Checksum mismatch");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
void sendMessage(const String& payload, uint8_t command = CMD_MESSAGE) {
|
|
uint16_t length = payload.length();
|
|
|
|
uint8_t checksum = CMD_MESSAGE ^ (length >> 8) ^ (length & 0xFF);
|
|
for (int i = 0; i < length; i++) {
|
|
checksum ^= payload[i];
|
|
}
|
|
|
|
Serial.write(HEADER1);
|
|
Serial.write(HEADER2);
|
|
Serial.write(command);
|
|
Serial.write((length >> 8) & 0xFF);
|
|
Serial.write(length & 0xFF);
|
|
Serial.write((const uint8_t*)payload.c_str(), length);
|
|
Serial.write(checksum);
|
|
}
|
|
|
|
void sendMessage(const uint8_t* payload, uint16_t length, uint8_t command = CMD_MESSAGE) {
|
|
uint8_t checksum = command ^ (length >> 8) ^ (length & 0xFF);
|
|
for (uint16_t i = 0; i < length; i++) {
|
|
checksum ^= payload[i];
|
|
}
|
|
|
|
Serial.write(HEADER1);
|
|
Serial.write(HEADER2);
|
|
Serial.write(command);
|
|
Serial.write((length >> 8) & 0xFF);
|
|
Serial.write(length & 0xFF);
|
|
Serial.write(payload, length);
|
|
Serial.write(checksum);
|
|
}
|
|
|
|
void handleCommand(uint8_t command, const uint8_t* payload, uint16_t length) {
|
|
switch (command) {
|
|
case CMD_ID_REQUEST: // CMD_ID_REQUEST
|
|
handleIdRequest();
|
|
break;
|
|
|
|
case CMD_FILE_LIST: //
|
|
handleFileList();
|
|
break;
|
|
|
|
case CMD_LOAD_FILE: //
|
|
handleLoadFile(payload, length);
|
|
break;
|
|
|
|
case CMD_DELETE_FILE: //
|
|
handleDeleteFile(payload, length);
|
|
break;
|
|
|
|
case CMD_SAVE_FILE:
|
|
handleSaveFile(payload, length);
|
|
break;
|
|
|
|
case CMD_SET_POSITION:
|
|
handleSetPosition(payload, length);
|
|
break;
|
|
|
|
case CMD_PLAY_FILE:
|
|
handlePlayAnimation(payload, length);
|
|
break;
|
|
|
|
case CMD_SCAN_CHANNEL:
|
|
handleScanChannel(payload, length);
|
|
break;
|
|
|
|
case CMD_WRITE_DATA:
|
|
handleDataWrite(payload, length);
|
|
break;
|
|
|
|
case CMD_START_POSITION_STREAM:
|
|
startPositionStream(payload, length);
|
|
break;
|
|
|
|
default:
|
|
Serial.print("Unknown command: ");
|
|
Serial.println(command, HEX);
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
void startPositionStream(const uint8_t* payload, uint16_t length) {
|
|
bool start = payload[0] != 0;
|
|
servos[0]->disableTorque(10);
|
|
servos[0]->waitOnData1Byte(10);
|
|
servos[0]->disableTorque(11);
|
|
servos[0]->waitOnData1Byte(10);
|
|
servos[0]->disableTorque(12);
|
|
servos[0]->waitOnData1Byte(10);
|
|
servos[0]->disableTorque(13);
|
|
servos[0]->waitOnData1Byte(10);
|
|
servos[0]->disableTorque(14);
|
|
servos[0]->waitOnData1Byte(10);
|
|
streamPositions = start;
|
|
}
|
|
|
|
|
|
void handleDataWrite(const uint8_t* payload, uint16_t length) {
|
|
uint8_t channel = payload[0];
|
|
uint8_t id = payload[1];
|
|
uint8_t writeByte = payload[2];
|
|
|
|
if (payload[4] == 1) {
|
|
servos[channel]->write1Byte(id, writeByte, payload[4]);
|
|
uint8_t response = servos[channel]->waitOnData1Byte(10);
|
|
sendMessage(&response, 1, CMD_WRITE_DATA);
|
|
} else {
|
|
servos[channel]->write2Bytes(id, writeByte, payload[4] | (payload[5] << 8));
|
|
uint16_t response = servos[channel]->waitOnData2Bytes(10);
|
|
uint8_t buffer[2];
|
|
buffer[0] = (response >> 8) & 0xFF; // high byte
|
|
buffer[1] = response & 0xFF; // low byte
|
|
sendMessage(buffer, 2, CMD_WRITE_DATA);
|
|
}
|
|
}
|
|
|
|
void handleIdRequest() {
|
|
String payload = String(DEVICE_NAME) + "|" + FIRMWARE_VERSION;
|
|
uint16_t length = payload.length();
|
|
|
|
uint8_t checksum = CMD_ID_REQUEST ^ (length >> 8) ^ (length & 0xFF);
|
|
for (int i = 0; i < length; i++) {
|
|
checksum ^= payload[i];
|
|
}
|
|
|
|
Serial.write(HEADER1);
|
|
Serial.write(HEADER2);
|
|
Serial.write(CMD_ID_REQUEST);
|
|
Serial.write((length >> 8) & 0xFF);
|
|
Serial.write(length & 0xFF);
|
|
Serial.write((const uint8_t*)payload.c_str(), length);
|
|
Serial.write(checksum);
|
|
}
|
|
|
|
void handleFileList() {
|
|
File root = FFat.open("/");
|
|
if (!root || !root.isDirectory()) {
|
|
sendFileListResponse(""); // empty payload
|
|
return;
|
|
}
|
|
|
|
String payload = "";
|
|
File file = root.openNextFile();
|
|
|
|
while (file) {
|
|
if (!file.isDirectory()) {
|
|
payload += String(file.name()) + "\n";
|
|
}
|
|
file = root.openNextFile();
|
|
}
|
|
|
|
sendFileListResponse(payload);
|
|
}
|
|
|
|
void sendFileListResponse(const String& payloadStr) {
|
|
uint16_t length = payloadStr.length();
|
|
if (length > MAX_PAYLOAD_SIZE) {
|
|
Serial.println("File list too large");
|
|
return;
|
|
}
|
|
|
|
uint8_t checksum = CMD_FILE_LIST ^ (length >> 8) ^ (length & 0xFF);
|
|
for (uint16_t i = 0; i < length; i++) {
|
|
checksum ^= payloadStr[i];
|
|
}
|
|
|
|
Serial.write(HEADER1);
|
|
Serial.write(HEADER2);
|
|
Serial.write(CMD_FILE_LIST);
|
|
Serial.write((length >> 8) & 0xFF);
|
|
Serial.write(length & 0xFF);
|
|
Serial.write((const uint8_t*)payloadStr.c_str(), length);
|
|
Serial.write(checksum);
|
|
}
|
|
|
|
void handleLoadFile(const uint8_t* payload, uint16_t length) {
|
|
if (length == 0 || length >= 128) {
|
|
Serial.println("Invalid filename");
|
|
return;
|
|
}
|
|
|
|
char filename[128];
|
|
memcpy(filename, payload, length);
|
|
filename[length] = '\0';
|
|
|
|
File file = FFat.open(filename, "r");
|
|
if (!file || !file.available()) {
|
|
Serial.println("File not found or empty");
|
|
return;
|
|
}
|
|
|
|
size_t fileSize = file.size();
|
|
if (fileSize > 65535) {
|
|
Serial.println("File too large for single transfer");
|
|
file.close();
|
|
return;
|
|
}
|
|
|
|
uint8_t* buffer = (uint8_t*)malloc(fileSize);
|
|
if (!buffer) {
|
|
Serial.println("Memory allocation failed");
|
|
file.close();
|
|
return;
|
|
}
|
|
|
|
size_t bytesRead = file.read(buffer, fileSize);
|
|
file.close();
|
|
|
|
if (bytesRead != fileSize) {
|
|
Serial.println("File read error");
|
|
free(buffer);
|
|
return;
|
|
}
|
|
|
|
// 🔹 Compute checksum
|
|
uint8_t checksum = CMD_LOAD_FILE ^ (fileSize >> 8) ^ (fileSize & 0xFF);
|
|
for (size_t i = 0; i < fileSize; i++) {
|
|
checksum ^= buffer[i];
|
|
}
|
|
|
|
// 🔹 Send packet
|
|
Serial.write(HEADER1);
|
|
Serial.write(HEADER2);
|
|
Serial.write(CMD_LOAD_FILE);
|
|
Serial.write((fileSize >> 8) & 0xFF);
|
|
Serial.write(fileSize & 0xFF);
|
|
Serial.write(buffer, fileSize);
|
|
Serial.write(checksum);
|
|
|
|
free(buffer);
|
|
Serial.println("File sent in one go");
|
|
}
|
|
|
|
void handleDeleteFile(const uint8_t* payload, uint16_t length) {
|
|
sendMessage("Deleting FILE");
|
|
if (length < 1) {
|
|
Serial.println("Payload too short for filename length");
|
|
sendMessage("Payload too short for filename length");
|
|
return;
|
|
}
|
|
|
|
// 🔹 Parse filename
|
|
uint16_t filenameLength = payload[0] | (payload[1] << 8);
|
|
if (length < 2 + filenameLength) {
|
|
Serial.println("Payload too short for filename");
|
|
sendMessage("Payload too short for filename");
|
|
return;
|
|
}
|
|
|
|
|
|
char filename[filenameLength + 1];
|
|
|
|
|
|
memcpy(filename, payload + 2, filenameLength);
|
|
filename[filenameLength] = '\0';
|
|
|
|
deleteFile(FFat, ("/" + String(filename)).c_str());
|
|
|
|
|
|
sendMessage(("File Deleted: " + String(filename)).c_str(), CMD_DELETE_FILE);
|
|
}
|
|
|
|
|
|
void handlePlayAnimation(const uint8_t* payload, uint16_t length) {
|
|
sendMessage("Playing FILE");
|
|
if (length < 1) {
|
|
Serial.println("Payload too short for filename length");
|
|
sendMessage("Payload too short for filename length");
|
|
return;
|
|
}
|
|
|
|
// 🔹 Parse filename
|
|
uint16_t filenameLength = payload[0] | (payload[1] << 8);
|
|
if (length < 2 + filenameLength) {
|
|
Serial.println("Payload too short for filename");
|
|
sendMessage("Payload too short for filename");
|
|
return;
|
|
}
|
|
|
|
|
|
char filename[filenameLength + 1];
|
|
|
|
|
|
memcpy(filename, payload + 2, filenameLength);
|
|
filename[filenameLength] = '\0';
|
|
|
|
//deleteFile(FFat, ("/" + String(filename)).c_str());
|
|
anim.clear();
|
|
anim.loadFromFile(("/" + String(filename)).c_str());
|
|
playAnimation(anim);
|
|
|
|
sendMessage("File Played", CMD_PLAY_FILE);
|
|
}
|
|
|
|
void handleSaveFile(const uint8_t* payload, uint16_t length) {
|
|
bool valid = parseAnimationPayload(payload, length, anim);
|
|
|
|
if (valid) {
|
|
//Serial.println("Animation parsed successfully!");
|
|
//anim.printKeyframes();
|
|
} else {
|
|
//Serial.println("Failed to parse animation.");
|
|
length = 1; // Override length to 1 for fallback response
|
|
payload = nullptr; // We'll send a single 0x00 instead
|
|
}
|
|
|
|
if (length > MAX_PAYLOAD_SIZE) {
|
|
Serial.println("File list too large");
|
|
return;
|
|
}
|
|
|
|
// Calculate checksum
|
|
uint8_t checksum = CMD_SAVE_FILE ^ (length >> 8) ^ (length & 0xFF);
|
|
if (valid) {
|
|
for (uint16_t i = 0; i < length; i++) {
|
|
checksum ^= payload[i];
|
|
}
|
|
} else {
|
|
checksum ^= 0x00;
|
|
}
|
|
|
|
// Send response
|
|
Serial.write(HEADER1);
|
|
Serial.write(HEADER2);
|
|
Serial.write(CMD_SAVE_FILE);
|
|
Serial.write((length >> 8) & 0xFF);
|
|
Serial.write(length & 0xFF);
|
|
|
|
if (valid) {
|
|
for (uint16_t i = 0; i < length; i++) {
|
|
Serial.write(payload[i]);
|
|
}
|
|
} else {
|
|
Serial.write(0x00);
|
|
}
|
|
|
|
Serial.write(checksum);
|
|
}
|
|
|
|
|
|
|
|
bool parseAnimationPayload(const uint8_t* payload, uint16_t length, Animation& animation) {
|
|
sendMessage("SAVING FILE");
|
|
if (length < 1) {
|
|
Serial.println("Payload too short for filename length");
|
|
return false;
|
|
}
|
|
|
|
// 🔹 Parse filename
|
|
uint16_t filenameLength = payload[0] | (payload[1] << 8);
|
|
if (length < 2 + filenameLength + 18) {
|
|
Serial.println("Payload too short for filename and header");
|
|
return false;
|
|
}
|
|
|
|
char filename[filenameLength + 1];
|
|
|
|
|
|
memcpy(filename, payload + 2, filenameLength);
|
|
filename[filenameLength] = '\0';
|
|
|
|
|
|
const uint8_t* ptr = payload + 2 + filenameLength;
|
|
|
|
String msg = "SAVING FILE: ";
|
|
msg += filename;
|
|
sendMessage(msg);
|
|
// 🔹 Parse header
|
|
memcpy(animation.header.magic, ptr, 4);
|
|
if (strncmp(animation.header.magic, "ANIM", 4) != 0) {
|
|
Serial.println("Invalid magic header");
|
|
|
|
sendMessage("invalid magic header");
|
|
return false;
|
|
}
|
|
|
|
animation.header.frameCount = ptr[5] << 8 | ptr[4]; // little-endian ✅
|
|
|
|
animation.header.version = ptr[6];
|
|
animation.header.frameRate = ptr[7];
|
|
memcpy(animation.header.reserved, ptr + 8, 8);
|
|
|
|
uint16_t curveCount = ptr[17] << 8 | ptr[16]; // little-endian ✅
|
|
Serial.print("curveCount: ");
|
|
Serial.println(curveCount);
|
|
|
|
ptr += 18;
|
|
|
|
if (length < (ptr - payload) + curveCount * sizeof(CurveSegment)) {
|
|
Serial.println("Payload too short for curve segments");
|
|
sendMessage("Payload too short");
|
|
return false;
|
|
}
|
|
|
|
// 🔹 Parse curve segments
|
|
animation.clearAllCurves();
|
|
for (uint16_t i = 0; i < curveCount; i++) {
|
|
CurveSegment seg;
|
|
memcpy(&seg, ptr, sizeof(CurveSegment));
|
|
Serial.print("Segment ");
|
|
Serial.print(i);
|
|
Serial.print(": motorID=");
|
|
Serial.print(seg.motorID);
|
|
Serial.print(", startTime=");
|
|
Serial.print(seg.startTime);
|
|
Serial.print(", endTime=");
|
|
Serial.print(seg.endTime);
|
|
Serial.print(", startPointY=");
|
|
Serial.print(seg.startPointY);
|
|
Serial.print(", startHandleX=");
|
|
Serial.print(seg.startHandleX);
|
|
Serial.print(", startHandleY=");
|
|
Serial.print(seg.startHandleY);
|
|
Serial.print(", endHandleX=");
|
|
Serial.print(seg.endHandleX);
|
|
Serial.print(", endHandleY=");
|
|
Serial.print(seg.endHandleY);
|
|
Serial.print(", endPointY=");
|
|
Serial.println(seg.endPointY);
|
|
|
|
animation.addCurveSegment(seg);
|
|
ptr += sizeof(CurveSegment);
|
|
}
|
|
|
|
// ✅ Advance ptr to node graph payload
|
|
//ptr += curveCount * sizeof(CurveSegment);
|
|
|
|
// 🔹 Parse node graph
|
|
uint8_t nodeCount = ptr[0];
|
|
Serial.print("Node count: ");
|
|
Serial.println(nodeCount);
|
|
sendMessage(String("Node count: ") + String(nodeCount));
|
|
sendMessage(String("ptr offset: ") + String(ptr - payload));
|
|
|
|
|
|
uint16_t remainingLength = length - (ptr - payload);
|
|
if (remainingLength > 0) {
|
|
loadNodeGraph(ptr, remainingLength, animation.nodeGraph);
|
|
} else {
|
|
Serial.println("No node graph data found");
|
|
}
|
|
sendMessage(printNodeGraph(animation.nodeGraph));
|
|
|
|
// 🔹 Save using received filename
|
|
animation.saveToFile(("/" + String(filename)).c_str());
|
|
|
|
sendMessage("SAVED");
|
|
return true;
|
|
}
|
|
|
|
|
|
void handleSetPosition(const uint8_t* payload, uint16_t length) {
|
|
if (length % 3 != 0) {
|
|
Serial.println("Invalid sync packet length");
|
|
return;
|
|
}
|
|
|
|
uint8_t count = 0;
|
|
for (int i = 0; i < length; i += 3) {
|
|
uint8_t motorId = payload[i];
|
|
uint16_t position = (payload[i + 2] << 8) | payload[i + 1];
|
|
pos1[count] = position;
|
|
count++;
|
|
// Apply position to motorId
|
|
|
|
//servos.sendWritePos(ids[motorId], position);
|
|
}
|
|
servos[0]->syncWritePos(ids, pos1, NUM_CHANNELS);
|
|
}
|
|
|
|
|
|
// Scans 0-254 and responds with the channel and ID as each successful ping is received
|
|
// Signals end by responding with channel and 255
|
|
void handleScanChannel(const uint8_t* payload, uint16_t length) {
|
|
if (length != 1) {
|
|
sendMessage("Length of scanChannel Request Wrong");
|
|
return;
|
|
}
|
|
|
|
|
|
for (int i = 0; i < 254; i++) {
|
|
servos[payload[0]]->sendPing(i);
|
|
uint8_t val = servos[payload[0]]->waitOnData1Byte(10);
|
|
|
|
if (val != 0) {
|
|
uint8_t response[44]; // Adjusted size to fit all values
|
|
response[0] = payload[0]; // channel
|
|
response[1] = i; // responding address
|
|
|
|
uint16_t model = servos[payload[0]]->getModel(i);
|
|
uint16_t minAngleLimit = servos[payload[0]]->getMinAngleLimit(i);
|
|
uint16_t maxAngleLimit = servos[payload[0]]->getMaxAngleLimit(i);
|
|
uint16_t position = servos[payload[0]]->getPosition(i);
|
|
uint8_t CWDeadZone = servos[payload[0]]->getCWDeadZone(i);
|
|
uint8_t CCWDeadZone = servos[payload[0]]->getCCWDeadZone(i);
|
|
uint16_t offset = servos[payload[0]]->getOffset(i);
|
|
uint8_t mode = servos[payload[0]]->getMode(i);
|
|
uint8_t torqueEnable = servos[payload[0]]->getTorqueEnable(i);
|
|
uint8_t acceleration = servos[payload[0]]->getAcceleration(i);
|
|
uint16_t goalPosition = servos[payload[0]]->getGoalPosition(i);
|
|
uint16_t goalTime = servos[payload[0]]->getGoalTime(i);
|
|
uint16_t goalSpeed = servos[payload[0]]->getGoalSpeed(i);
|
|
uint8_t lock = servos[payload[0]]->getLock(i);
|
|
int16_t speed = servos[payload[0]]->getSpeed(i);
|
|
uint16_t load = servos[payload[0]]->getLoad(i);
|
|
uint8_t temperature = servos[payload[0]]->getTemperature(i);
|
|
uint8_t moving = servos[payload[0]]->getMoving(i);
|
|
uint8_t current = servos[payload[0]]->getCurrent(i);
|
|
uint8_t voltage = servos[payload[0]]->getVoltage(i);
|
|
|
|
// Pack values into response
|
|
response[2] = (model >> 8) & 0xFF;
|
|
response[3] = model & 0xFF;
|
|
|
|
response[4] = (minAngleLimit >> 8) & 0xFF;
|
|
response[5] = minAngleLimit & 0xFF;
|
|
|
|
response[6] = (maxAngleLimit >> 8) & 0xFF;
|
|
response[7] = maxAngleLimit & 0xFF;
|
|
|
|
response[8] = (position >> 8) & 0xFF;
|
|
response[9] = position & 0xFF;
|
|
|
|
response[10] = CWDeadZone;
|
|
response[11] = CCWDeadZone;
|
|
|
|
response[12] = (offset >> 8) & 0xFF;
|
|
response[13] = offset & 0xFF;
|
|
|
|
response[14] = mode;
|
|
response[15] = torqueEnable;
|
|
response[16] = acceleration;
|
|
|
|
response[17] = (goalPosition >> 8) & 0xFF;
|
|
response[18] = goalPosition & 0xFF;
|
|
|
|
response[19] = (goalTime >> 8) & 0xFF;
|
|
response[20] = goalTime & 0xFF;
|
|
|
|
response[21] = (goalSpeed >> 8) & 0xFF;
|
|
response[22] = goalSpeed & 0xFF;
|
|
|
|
response[23] = lock;
|
|
|
|
response[24] = (speed >> 8) & 0xFF;
|
|
response[25] = speed & 0xFF;
|
|
|
|
response[26] = (load >> 8) & 0xFF;
|
|
response[27] = load & 0xFF;
|
|
|
|
response[28] = temperature;
|
|
response[29] = moving;
|
|
response[30] = (current >> 8) & 0xFF;
|
|
response[31] = current & 0xFF;
|
|
|
|
response[32] = voltage;
|
|
|
|
|
|
// You can continue adding more fields here if needed...
|
|
|
|
sendMessage(response, 33, CMD_SCAN_CHANNEL); // updated length
|
|
}
|
|
}
|
|
uint8_t r[2];
|
|
r[0] = payload[0]; // channel
|
|
r[1] = 255; // responding address
|
|
|
|
sendMessage(r, 2, CMD_SCAN_CHANNEL);
|
|
|
|
|
|
// std::vector<uint8_t> successfulAddresses;
|
|
// servos[payload[0]]->pingAll(successfulAddresses);
|
|
// std::vector<uint8_t> response;
|
|
// response.push_back(payload[0]); // channel
|
|
// response.push_back(successfulAddresses.size()); // count
|
|
|
|
// for (uint8_t address : successfulAddresses) {
|
|
// response.push_back(address);
|
|
// }
|
|
|
|
// sendMessage(response.data(), response.size(), CMD_SCAN_CHANNEL);
|
|
}
|
|
|
|
|
|
|
|
bool flip = false;
|
|
unsigned long lastSend = 0;
|
|
void loop() {
|
|
|
|
// for (int i = 0; i < 1023; i++) {
|
|
// servos[0]->sendWritePos(10, i);
|
|
// Serial.println(servos[0]->waitOnData1Byte(10));
|
|
// delay(20);
|
|
// }
|
|
|
|
|
|
HandleSerialRequests();
|
|
// for (int i = 0; i < 500; i+=32) {
|
|
// // Serial.print("WRITE: ");
|
|
// // Serial.print(i);
|
|
// // Serial.print(" ");
|
|
// // Serial.println(servos[1]->setMinAngleLimit(16, i));
|
|
|
|
// Serial.print("READ: ");
|
|
// Serial.println(servos[1]->getMinAngleLimit(16));
|
|
// delay(2000);
|
|
// }
|
|
// servos[0]->sendWritePos(13, 0);
|
|
// servos[1]->sendWritePos(13, 0);
|
|
|
|
// Serial.print(servos[0]->getMinAngleLimit(13));
|
|
// Serial.print("\t");
|
|
// Serial.print(servos[0]->getMaxAngleLimit(13));
|
|
// Serial.print("\t");
|
|
// Serial.print(servos[1]->getMinAngleLimit(13));
|
|
// Serial.print("\t");
|
|
// Serial.println(servos[1]->getMaxAngleLimit(13));
|
|
// delay(1000);
|
|
|
|
// //
|
|
// for (int i = 0; i < 4095; i+=16){
|
|
// Serial.println(i);
|
|
// servos[0]->sendWritePos(13, i);
|
|
// servos[1]->sendWritePos(13, i);
|
|
// delay(20);
|
|
// }
|
|
// delay(1000);
|
|
// for (int i = 4095; i > 0; i-=16){
|
|
// Serial.println(i);
|
|
// servos[0]->sendWritePos(13, i);
|
|
// servos[1]->sendWritePos(13, i);
|
|
// delay(20);
|
|
// }
|
|
// delay(1000);
|
|
// delay(1000);
|
|
// for (int i = 0; i< 100; i++){
|
|
// Serial.println(servos[1]->getGoalSpeed(13));
|
|
// delay(10);
|
|
// }
|
|
// servos[0]->sendWritePos(13, 0);
|
|
// for (int i = 0; i< 100; i++){
|
|
// Serial.println(servos[1]->getGoalSpeed(13));
|
|
// delay(10);
|
|
// }
|
|
if (millis() - lastSend > 2000) {
|
|
|
|
//sendMessageFromESP32(String(millis()));
|
|
//handleIdRequest();
|
|
//PrintFileList();
|
|
lastSend = millis();
|
|
}
|
|
|
|
if (streamPositions) {
|
|
if (millis() - lastStreamPositions > 50) {
|
|
lastStreamPositions = millis();
|
|
SendMotorPositions();
|
|
}
|
|
}
|
|
}
|
|
|
|
void SendMotorPositions() {
|
|
|
|
uint16_t p0 = servos[0]->getPosition(10);
|
|
uint16_t p1 = servos[0]->getPosition(11);
|
|
uint16_t p2 = servos[0]->getPosition(12);
|
|
uint16_t p3 = servos[0]->getPosition(13);
|
|
uint16_t p4 = servos[0]->getPosition(14);
|
|
|
|
uint8_t payload[10];
|
|
|
|
payload[0] = p0 >> 8;
|
|
payload[1] = p0 & 0xFF;
|
|
payload[2] = p1 >> 8;
|
|
payload[3] = p1 & 0xFF;
|
|
payload[4] = p2 >> 8;
|
|
payload[5] = p2 & 0xFF;
|
|
payload[6] = p3 >> 8;
|
|
payload[7] = p3 & 0xFF;
|
|
payload[8] = p4 >> 8;
|
|
payload[9] = p4 & 0xFF;
|
|
|
|
sendMessage(payload, 10, POSITION_STREAM);
|
|
}
|
|
|
|
void PrintFileList() {
|
|
File root = FFat.open("/");
|
|
if (!root || !root.isDirectory()) {
|
|
Serial.println("Failed to open FFat root directory");
|
|
return;
|
|
}
|
|
|
|
Serial.println("Files in FFat:");
|
|
|
|
File file = root.openNextFile();
|
|
while (file) {
|
|
Serial.print(" ");
|
|
Serial.print(file.name());
|
|
Serial.print(" | Size: ");
|
|
Serial.println(file.size());
|
|
file = root.openNextFile();
|
|
}
|
|
|
|
Serial.println("End of file list.");
|
|
}
|
|
|
|
void ClearFiles() {
|
|
File root = FFat.open("/");
|
|
if (!root || !root.isDirectory()) {
|
|
Serial.println("Failed to open FFat root directory");
|
|
return;
|
|
}
|
|
|
|
Serial.println("Files in FFat:");
|
|
|
|
File file = root.openNextFile();
|
|
while (file) {
|
|
String filename = "/" + String(file.name()); // Add leading slash
|
|
Serial.print(" Deleting: ");
|
|
Serial.println(filename);
|
|
file.close(); // Close before deleting
|
|
deleteFile(FFat, filename.c_str()); // Use corrected path
|
|
file = root.openNextFile();
|
|
}
|
|
|
|
Serial.println("FFat cleanup complete.");
|
|
}
|
|
|
|
|
|
|
|
void deleteFile(fs::FS& fs, const char* path) {
|
|
Serial.printf("Deleting file: %s\r\n", path);
|
|
if (fs.remove(path)) {
|
|
Serial.println("- file deleted");
|
|
} else {
|
|
Serial.println("- delete failed");
|
|
}
|
|
}
|
|
|
|
|
|
void playAnimation(Animation& animation) {
|
|
uint16_t durationCS = animation.getFrameCount();
|
|
const uint8_t fps = 48;
|
|
const uint32_t frameIntervalMS = 1000 / fps; // ~20.83 ms
|
|
const uint32_t totalDurationMS = durationCS * 10;
|
|
|
|
uint32_t startTime = millis();
|
|
uint32_t nextFrameTime = startTime;
|
|
|
|
while (millis() - startTime < totalDurationMS) {
|
|
uint32_t currentTime = millis();
|
|
|
|
if (currentTime >= nextFrameTime) {
|
|
uint16_t timeCS = (currentTime - startTime) / 10;
|
|
|
|
//for (uint8_t motorID = 0; motorID < NUM_CHANNELS; motorID++) {
|
|
uint16_t pos = animation.getMotorPosition(0, timeCS);
|
|
pos = pos / 4;
|
|
servos[0]->sendWritePos(10, pos);
|
|
|
|
Serial.println(pos);
|
|
//}
|
|
|
|
nextFrameTime += frameIntervalMS;
|
|
}
|
|
|
|
// Optional: yield or small delay to avoid busy loop
|
|
delay(1);
|
|
}
|
|
|
|
// Optional: reset motors to center
|
|
// for (uint8_t motorID = 0; motorID < NUM_CHANNELS; motorID++) {
|
|
// servos[0]->sendWritePos(motorID, 2048);
|
|
// }
|
|
}
|
|
|
|
|
|
|
|
void playAnimationOLD(Animation& anim) {
|
|
// uint16_t positions[NUM_CHANNELS];
|
|
// const uint16_t frameCount = 400; //anim.getFrameCount();
|
|
// const uint32_t frameDelay = 1000 / FRAMES_PER_SECOND;
|
|
// uint32_t nextFrameTime = millis();
|
|
// Serial.print("Frame Count: ");
|
|
// Serial.println(frameCount);
|
|
|
|
// // Organize keyframes per motor
|
|
// std::vector<Keyframe> motorKeyframes[NUM_CHANNELS];
|
|
// for (const auto& kf : anim.getKeyframes()) {
|
|
// if (kf.motorId < NUM_CHANNELS) {
|
|
// motorKeyframes[kf.motorId].push_back(kf);
|
|
// }
|
|
// }
|
|
|
|
// // Sort keyframes per motor by frame
|
|
// for (int ch = 0; ch < NUM_CHANNELS; ch++) {
|
|
// std::sort(motorKeyframes[ch].begin(), motorKeyframes[ch].end(),
|
|
// [](const Keyframe& a, const Keyframe& b) {
|
|
// return a.frame < b.frame;
|
|
// });
|
|
// }
|
|
|
|
// for (uint16_t frame = 0; frame < frameCount; frame++) {
|
|
// while (millis() < nextFrameTime) {
|
|
// delay(1);
|
|
// }
|
|
|
|
// for (int ch = 0; ch < NUM_CHANNELS; ch++) {
|
|
// const auto& kfs = motorKeyframes[ch];
|
|
// uint16_t value = 512; // default position
|
|
|
|
// // Find surrounding keyframes
|
|
// Keyframe prev = { ch, 0, 512 }, next = { ch, frameCount, 512 };
|
|
// for (size_t i = 0; i < kfs.size(); i++) {
|
|
// if (kfs[i].frame <= frame) {
|
|
// prev = kfs[i];
|
|
// }
|
|
// if (kfs[i].frame > frame) {
|
|
// next = kfs[i];
|
|
// break;
|
|
// }
|
|
// }
|
|
|
|
// // Interpolate
|
|
// if (prev.frame == next.frame) {
|
|
// value = prev.position;
|
|
// } else {
|
|
// float t = float(frame - prev.frame) / (next.frame - prev.frame);
|
|
// value = prev.position + t * (next.position - prev.position);
|
|
// }
|
|
|
|
// positions[ch] = value;
|
|
// }
|
|
// Serial.println(positions[0]);
|
|
// servos[0]->syncWritePos(ids, positions, NUM_CHANNELS);
|
|
// nextFrameTime += frameDelay;
|
|
// }
|
|
}
|
|
|
|
|
|
|
|
void playLayeredAnimation(Animation& base, Animation& overlay) {
|
|
uint16_t basePositions[NUM_CHANNELS];
|
|
uint16_t overlayPositions[NUM_CHANNELS];
|
|
uint16_t finalPositions[NUM_CHANNELS];
|
|
|
|
const uint16_t frameCount = base.getFrameCount();
|
|
const uint32_t frameDelay = 1000 / FRAMES_PER_SECOND;
|
|
uint32_t nextFrameTime = millis();
|
|
|
|
for (uint16_t frame = 0; frame < frameCount; frame++) {
|
|
while (millis() < nextFrameTime) delay(1);
|
|
|
|
base.getFramePositions(frame, basePositions);
|
|
overlay.getFramePositions(frame, overlayPositions);
|
|
|
|
for (uint8_t ch = 0; ch < NUM_CHANNELS; ch++) {
|
|
finalPositions[ch] = (basePositions[ch] + overlayPositions[ch]) / 2;
|
|
}
|
|
|
|
servos[0]->syncWritePos(ids, finalPositions, NUM_CHANNELS);
|
|
nextFrameTime += frameDelay;
|
|
}
|
|
}
|
|
|
|
|
|
void SCSPingAll() {
|
|
std::vector<uint8_t> successfulAddresses;
|
|
servos[0]->pingAll(successfulAddresses);
|
|
|
|
// Now successfulAddresses contains all successful pings
|
|
Serial.println("Successful Addresses:");
|
|
for (uint8_t address : successfulAddresses) {
|
|
Serial.print(address);
|
|
Serial.print(" ");
|
|
}
|
|
Serial.println();
|
|
}
|
|
|
|
void STSPingAll() {
|
|
std::vector<uint8_t> successfulAddresses;
|
|
servos[1]->pingAll(successfulAddresses);
|
|
|
|
// Now successfulAddresses contains all successful pings
|
|
Serial.println("Successful Addresses:");
|
|
for (uint8_t address : successfulAddresses) {
|
|
Serial.print(address);
|
|
Serial.print(" ");
|
|
}
|
|
Serial.println();
|
|
}
|