leonarski_f
38c2826c09
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 13m46s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 15m19s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 15m49s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 16m44s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 17m5s
Build Packages / build:rpm (rocky8) (push) Successful in 17m17s
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 18m15s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 11m44s
Build Packages / build:rpm (rocky9) (push) Successful in 13m29s
Build Packages / Generate python client (push) Successful in 35s
Build Packages / Create release (push) Has been skipped
Build Packages / Build documentation (push) Successful in 53s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 12m37s
Build Packages / XDS test (neggia plugin) (push) Successful in 10m19s
Build Packages / XDS test (durin plugin) (push) Successful in 11m41s
Build Packages / XDS test (JFJoch plugin) (push) Successful in 11m34s
Build Packages / DIALS test (push) Successful in 14m15s
Build Packages / Unit tests (push) Successful in 55m15s
64 lines
2.1 KiB
C++
64 lines
2.1 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCHHDF5IMAGEREADER_H
|
|
#define JFJOCHHDF5IMAGEREADER_H
|
|
|
|
#include "JFJochReader.h"
|
|
#include "../writer/HDF5Objects.h"
|
|
|
|
class JFJochHDF5Reader : public JFJochReader {
|
|
FileWriterFormat format = FileWriterFormat::NoFile;
|
|
HDF5DataSetLayout data_layout = HDF5DataSetLayout::CONTIGUOUS;
|
|
|
|
std::shared_ptr<HDF5ReadOnlyFile> master_file;
|
|
|
|
std::vector<std::string> legacy_format_files;
|
|
std::vector<HDF5VirtualDatasetMapping> vds_data_mappings;
|
|
std::string master_file_directory;
|
|
std::string master_filename;
|
|
|
|
size_t images_per_file = 1;
|
|
size_t number_of_images = 0;
|
|
|
|
bool LoadImage_i(std::shared_ptr<JFJochReaderDataset> &dataset,
|
|
DataMessage& message,
|
|
std::vector<uint8_t> &buffer,
|
|
int64_t image_number,
|
|
bool update_dataset) override;
|
|
|
|
CompressedImage LoadImageDataset(std::vector<uint8_t> &tmp, HDF5Object &file, hsize_t number);
|
|
|
|
template <class T>
|
|
void ReadVector(std::vector<T> &v,
|
|
HDF5Object &file,
|
|
const std::string &dataset_name,
|
|
size_t image0,
|
|
size_t nimages);
|
|
|
|
std::pair<std::shared_ptr<HDF5ReadOnlyFile>, uint32_t> GetImageLocation(int64_t image_number);
|
|
std::optional<GoniometerAxis> ReadAxis(HDF5Object *file, const std::string &name);
|
|
public:
|
|
~JFJochHDF5Reader() override = default;
|
|
|
|
void ReadFile(const std::string& filename);
|
|
|
|
uint64_t GetNumberOfImages() const override;
|
|
|
|
void Close() override;
|
|
|
|
std::vector<HDF5DataSourceMessage> GetHDF5DataSource(
|
|
uint64_t first_image = 0,
|
|
std::optional<uint64_t> image_count = {}
|
|
) const;
|
|
|
|
std::vector<std::vector<Reflection>> ReadReflections(size_t start_image = 0, std::optional<size_t> end_image = {}) const;
|
|
|
|
CompressedImage ReadCalibration(std::vector<uint8_t> &tmp, const std::string &name) const;
|
|
|
|
std::shared_ptr<JFJochReaderRawImage> GetRawImage(int64_t image_number) override;
|
|
};
|
|
|
|
|
|
#endif //JFJOCHHDF5IMAGEREADER_H
|