Add metadata buffer class

This commit is contained in:
2018-11-19 16:33:02 +01:00
parent 393919aca5
commit 1acd36f77f
2 changed files with 62 additions and 0 deletions
+37
View File
@@ -0,0 +1,37 @@
#include "MetadataBuffer.hpp"
using namespace std;
MetadataBuffer::MetadataBuffer(uint64_t n_images, shared_ptr<unordered_map<string, HeaderDataType>> header_values_type) :
n_images(n_images), header_values_type(header_values_type)
{
if (header_values_type) {
for (const auto& header_type : *header_values_type) {
auto& name = header_type.first;
auto& header_data_type = header_type.second;
size_t bytes_size_per_frame = header_data_type.value_shape * header_data_type.value_bytes_size;
size_t buffer_size_bytes = n_images * bytes_size_per_frame;
shared_ptr<char> buffer(new char[buffer_size_bytes], std::default_delete<char[]>());
metadata_buffer.insert({name, buffer});
metadata_length_bytes.insert({name, bytes_size_per_frame});
}
}
}
void MetadataBuffer::add_metadata_to_buffer(string name, uint64_t frame_index, const char* data)
{
size_t bytes_size_per_frame = metadata_length_bytes.at(name);
size_t buffer_offset = frame_index * bytes_size_per_frame;
char* buffer = (metadata_buffer.at(name)).get();
buffer += buffer_offset;
memcpy(buffer, data, bytes_size_per_frame);
}
shared_ptr<char> MetadataBuffer::get_metadata_values(string name)
{
return metadata_buffer.at(name);
}
+25
View File
@@ -0,0 +1,25 @@
#ifndef METADATABUFFER_H
#define METADATABUFFER_H
#include <unordered_map>
#include <string>
#include "ZmqReceiver.hpp"
class MetadataBuffer
{
const uint64_t n_images;
const std::shared_ptr<std::unordered_map<std::string, HeaderDataType>> header_values_type;
protected:
std::unordered_map<std::string, std::shared_ptr<char>> metadata_buffer;
std::unordered_map<std::string, size_t> metadata_length_bytes;
public:
MetadataBuffer(uint64_t n_images, std::shared_ptr<std::unordered_map<std::string, HeaderDataType>> header_values_type);
void add_metadata_to_buffer(std::string name, uint64_t frame_index, const char* data);
std::shared_ptr<char> get_metadata_values(std::string name);
};
#endif