Optimize BufferH5Writer.cpp

This commit is contained in:
2020-05-04 13:56:42 +02:00
parent 261d4751e5
commit e2278860a2
2 changed files with 87 additions and 190 deletions
+21 -29
View File
@@ -6,57 +6,49 @@
#include <string>
#include <H5Cpp.h>
#include <memory>
#include "jungfrau.hpp"
#include <unordered_map>
class BufferH5Writer {
const uint16_t CHUNKING_FACTOR = 1;
const hsize_t meta_disk_dims[2] = {
core_buffer::FILE_MOD,
ModuleFrame_N_FIELDS
};
const hsize_t data_disk_dims[3] = {
core_buffer::FILE_MOD,
core_buffer::MODULE_Y_SIZE,
core_buffer::MODULE_X_SIZE
};
const size_t n_frames_per_file_;
const uint16_t y_frame_size_;
const uint16_t x_frame_size_;
const size_t frame_bytes_size_;
const std::string device_name_;
const std::string root_folder_;
const std::string latest_filename_;
const std::string current_filename_;
const std::string LATEST_filename_;
const std::string CURRENT_filename_;
std::string output_filename_;
H5::H5File h5_file_;
std::string current_output_filename_;
H5::H5File current_output_file_;
H5::DataSet current_image_dataset_;
H5::DataSet current_metadata_dataset_;
uint64_t current_pulse_id_;
size_t current_frame_index_;
std::unordered_map<std::string, char*> buffers_;
std::unordered_map<std::string, H5::DataSet> datasets_;
std::unordered_map<std::string, H5::PredType> scalar_metadata_;
size_t current_file_index_;
void create_file(const std::string& filename);
void write_scalar_metadata(
const std::string& name,
const void* value,
const H5::DataType data_type);
public:
BufferH5Writer(
const size_t n_frames_per_file,
const uint16_t y_frame_size,
const uint16_t x_frame_size,
const std::string& device_name,
const std::string& root_folder);
virtual ~BufferH5Writer();
template <class T> void add_scalar_metadata(
const std::string& metadata_name);
void set_pulse_id(const uint64_t pulse_id);
void write_data(const char* buffer);
template <class T> void write_scalar_metadata(
const std::string& name,
const void* value);
void write(const ModuleFrame* metadata, const char* data);
void close_file();
};