diff --git a/src/WriterManager.cpp b/src/WriterManager.cpp index 53ca9f6..628c3ac 100644 --- a/src/WriterManager.cpp +++ b/src/WriterManager.cpp @@ -3,12 +3,15 @@ using namespace std; -WriterManager::WriterManager(uint64_t n_images): +WriterManager::WriterManager(uint64_t n_images, std::string dataset_name): n_images(n_images), running_flag(true), n_received_frames(0), n_written_frames(0) { #ifdef DEBUG_OUTPUT cout << "[WriterManager::WriterManager] Writer manager for n_images " << n_images << endl; + cout << "[WriterManager::WriterManager] Using dataset_name " << dataset_name << endl; #endif + + parameters.insert({"dataset_name", dataset_name}); } void WriterManager::stop() @@ -40,11 +43,9 @@ map WriterManager::get_statistics() return result; } -map& WriterManager::get_parameters() +map WriterManager::get_parameters() { - // TODO: Implement this. - map result = {}; - return result; + return parameters; } void WriterManager::set_parameters(map& new_parameters) diff --git a/src/WriterManager.hpp b/src/WriterManager.hpp index b4c8fd8..6f51e49 100644 --- a/src/WriterManager.hpp +++ b/src/WriterManager.hpp @@ -4,6 +4,7 @@ #include #include #include +#include "config.hpp" class WriterManager { @@ -16,13 +17,12 @@ class WriterManager std::atomic_int n_written_frames; public: - WriterManager(uint64_t n_images=0); + WriterManager(uint64_t n_images=0, std::string dataset_name=config::dataset_name); void stop(); std::string get_status(); std::map get_statistics(); - std::map get_paramters(); + std::map get_parameters(); void set_parameters(std::map &new_parameters); - std::map& get_parameters(); bool is_running(); void received_frame(size_t frame_index); void written_frame(size_t frame_index); diff --git a/src/rest_interface.cpp b/src/rest_interface.cpp index 10ff541..2e9c7f7 100644 --- a/src/rest_interface.cpp +++ b/src/rest_interface.cpp @@ -62,9 +62,9 @@ void start_rest_api(WriterManager& writer_manager, uint16_t port) if (req.method == "GET"_method) { - // for (const auto& item : writer_manager.get_paramters()) { - // result[item.first] = item.second; - // } + for (const auto& item : writer_manager.get_parameters()) { + result[item.first] = item.second; + } return result; } else {