Add more examples

This commit is contained in:
2026-03-07 15:17:45 -05:00
parent e78106d540
commit 87bb2d74b5
16 changed files with 228 additions and 26 deletions

View File

@@ -1,3 +1,4 @@
//
// Created by Johnathon Slightham on 2026-02-16.
//
@@ -21,6 +22,7 @@ class RemoteManagement {
m_robot_controller(controller) {
}
bool perform_ota();
double ota_progress(); // 0 to 1 representing % progress.
void restart();
private:
@@ -29,6 +31,7 @@ class RemoteManagement {
bool ota_end();
uint16_t m_sequence_num = 0;
uint16_t m_total_packets = 0;
uint8_t m_module_id;
std::ifstream m_file;
std::unique_ptr<Flatbuffers::OTAPacketBuilder> m_builder;

View File

@@ -9,40 +9,51 @@
#include "libcontrol.h"
#include "rpc/RemoteManagement.h"
void progress_monitor_task(std::shared_ptr<RemoteManagement> rm, uint8_t module_id) {
while (true) {
std::this_thread::sleep_for(std::chrono::seconds(5));
std::cout << "Module " << (int)module_id << ": " << rm->ota_progress() * 100 << "% complete\n";
if (rm->ota_progress() >= 1.0) return;
}
}
void task(std::shared_ptr<RobotController> controller, uint8_t module_id, const std::string& filepath) {
auto rm = std::make_shared<RemoteManagement>(module_id, filepath, controller);
std::thread t(progress_monitor_task, rm, module_id);
rm->perform_ota();
t.join();
std::cout << "Done updating " << (int)module_id << "\n";
}
int main() {
// const auto messaging_interface = std::make_unique<MessagingInterface>();
// const auto modules = messaging_interface->find_connected_modules(std::chrono::seconds(3));
// std::cout << "Found " << modules.size() << " modules" << std::endl;
// for (const auto module : modules) {
// std::cout << "Found ID " << (int)module << std::endl;
// }
const auto robot_controller = std::make_shared<RobotController>();
robot_controller->fetchDirectlyConnectedModules(true);
std::this_thread::sleep_for(std::chrono::seconds(5));
// const auto function_tag = 3;
// const auto module = 99;
// std::string msg = "Hello world!";
// std::vector<uint8_t> parameters(msg.begin(), msg.end());
// auto maybe_return_value = messaging_interface->remote_call(function_tag, module, parameters);
std::vector<uint8_t> to_update{};
for (const auto& maybe_module : robot_controller->getModules()) {
if (const auto &module = maybe_module.lock()) {
if (module->get_leader() != module->get_device_id()) { continue; }
std::cout << "Updating module " << (int)module->get_device_id();
to_update.emplace_back(module->get_device_id());
}
}
// if (maybe_return_value) {
// auto return_value = std::move(*maybe_return_value);
// std::cout << "Got return value " << (char *)return_value->data() << std::endl;
// } else {
// std::cout << "Function call time out" << std::endl;
// }
//
if (to_update.size() < 1) {
std::cout << "No modules found to update" << std::endl;
return 0;
}
std::string filename =
"/Users/jslightham/Documents/Classes/capstone/firmware/build/firmware.bin";
const auto rm = std::make_unique<RemoteManagement>(99, filename, robot_controller);
rm->perform_ota();
"/Users/jslightham/Documents/Classes/capstone/firmware/build/firmware.bin";
std::vector<std::thread> threads;
for (int i = 0; i < to_update.size(); i++) {
threads.emplace_back(task, robot_controller, to_update[i], filename);
}
for (auto &t : threads) {
t.join();
}
return 0;
}

View File

@@ -23,6 +23,7 @@ bool RemoteManagement::perform_ota() {
m_file.seekg(0, std::ios::end);
std::streamsize total_size = m_file.tellg();
m_file.seekg(0, std::ios::beg);
m_total_packets = total_size/OTA_CHUNK_SIZE;
// std::cout << "Total number of chunks: " << total_size/OTA_CHUNK_SIZE << std::endl;
while (m_file) {
@@ -96,3 +97,14 @@ bool RemoteManagement::ota_end() {
}
return false;
}
double RemoteManagement::ota_progress() {
if (m_total_packets < 1) {
return 0.0;
}
if (m_sequence_num >= m_total_packets) {
return 1.0;
}
return static_cast<double>(m_sequence_num) / static_cast<double>(m_total_packets);
}