HansonServo/HansonServo.ino

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();
}