diff --git a/core-buffer/include/WriterH5Writer.hpp b/core-buffer/include/WriterH5Writer.hpp index 7b06848..82ad2e3 100644 --- a/core-buffer/include/WriterH5Writer.hpp +++ b/core-buffer/include/WriterH5Writer.hpp @@ -6,8 +6,8 @@ #include #include "buffer_config.hpp" -// Size of compression header in bytes. -const int BSHUF_LZ4_HEADER_BYTES = 12; +//// Size of compression header in bytes. +//const int BSHUF_LZ4_HEADER_BYTES = 12; struct ImageMetadata { @@ -21,6 +21,8 @@ struct ImageMetadata class WriterH5Writer { const size_t n_frames_; + const size_t n_modules_; + const size_t image_cache_n_images_; size_t current_write_index_; H5::H5File file_; @@ -33,10 +35,10 @@ class WriterH5Writer { public: - WriterH5Writer( - const std::string& output_file, - const size_t n_frames, - const size_t n_modules); + WriterH5Writer(const std::string& output_file, + const size_t n_frames, + const size_t n_modules, + const size_t image_cache_n_images); ~WriterH5Writer(); void write(const ImageMetadata* metadata, const char* data); void close_file(); diff --git a/core-buffer/src/WriterH5Writer.cpp b/core-buffer/src/WriterH5Writer.cpp index 1d8f62f..9814688 100644 --- a/core-buffer/src/WriterH5Writer.cpp +++ b/core-buffer/src/WriterH5Writer.cpp @@ -2,11 +2,11 @@ #include -extern "C" -{ - #include "H5DOpublic.h" - #include -} +//extern "C" +//{ +// #include "H5DOpublic.h" +// #include +//} using namespace std; using namespace core_buffer; @@ -14,8 +14,11 @@ using namespace core_buffer; WriterH5Writer::WriterH5Writer( const string& output_file, const size_t n_frames, - const size_t n_modules) : + const size_t n_modules, + const size_t image_cache_n_images) : n_frames_(n_frames), + n_modules_(n_modules), + image_cache_n_images_(image_cache_n_images), current_write_index_(0) { @@ -107,18 +110,26 @@ void WriterH5Writer::close_file() void WriterH5Writer::write(const ImageMetadata* metadata, const char* data) { - hsize_t image_offset[] = {current_write_index_, 0, 0}; + hsize_t b_i_dims[3] = { + image_cache_n_images_, + MODULE_Y_SIZE*n_modules_, + MODULE_X_SIZE}; + H5::DataSpace b_i_space(3, b_i_dims); - if(H5DOwrite_chunk( - image_dataset_.getId(), - H5P_DEFAULT, - BSHUF_H5FILTER, - image_offset, - metadata->data_n_bytes, - data)) - { - throw runtime_error("Cannot write image dataset."); - } + hsize_t f_i_dims[3] = {n_frames_, + MODULE_Y_SIZE * n_modules_, + MODULE_X_SIZE}; + H5::DataSpace f_i_space(3, f_i_dims); + hsize_t i_count[] = {image_cache_n_images_, + MODULE_Y_SIZE, + MODULE_X_SIZE}; + + hsize_t i_start[] = {current_write_index_, 0, 0}; + f_i_space.selectHyperslab(H5S_SELECT_SET, i_count, i_start); + + image_dataset_.write( + data, H5::PredType::NATIVE_UINT16, + b_i_space, f_i_space); hsize_t b_m_dims[1] = {1}; H5::DataSpace b_m_space (1, b_m_dims);