Externalize check for frame destination

This commit is contained in:
2018-12-03 13:46:33 +01:00
parent 4be0a8935b
commit 2a20f13db9
2 changed files with 17 additions and 6 deletions
+16 -6
View File
@@ -243,19 +243,29 @@ bool H5Writer::is_file_open() const
return (file.getId() != -1);
}
inline bool H5Writer::is_data_for_current_file(const size_t data_index)
{
if (frames_per_file) {
hsize_t frame_chunk = (data_index / frames_per_file) + 1;
// This frames does not go into this file.
if (frame_chunk != current_frame_chunk) {
return false;
}
}
return true;
}
hsize_t H5Writer::prepare_storage_for_data(const string& dataset_name, const size_t data_index, const std::vector<size_t>& data_shape,
const string& data_type, const string& endianness)
{
hsize_t relative_data_index = data_index;
// Check if we have to create a new file.
if (frames_per_file) {
if (!is_data_for_current_file(data_index)) {
hsize_t frame_chunk = (data_index / frames_per_file) + 1;
// This frames does not go into this file.
if (frame_chunk != current_frame_chunk) {
create_file(frame_chunk);
}
create_file(frame_chunk);
// Make the data_index relative to this chunk (file).
relative_data_index = data_index - ((frame_chunk - 1) * frames_per_file);
+1
View File
@@ -41,6 +41,7 @@ class H5Writer
virtual void write_data(const std::string& dataset_name, const size_t data_index, const char* data, const std::vector<size_t>& data_shape,
const size_t data_bytes_size, const std::string& data_type, const std::string& endianness);
virtual H5::H5File& get_h5_file();
virtual bool is_data_for_current_file(const size_t data_index);
};
class DummyH5Writer : public H5Writer