59 lines
2.0 KiB
C++
59 lines
2.0 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_JFJOCHREADERIMAGE_H
|
|
#define JFJOCH_JFJOCHREADERIMAGE_H
|
|
|
|
#include <vector>
|
|
#include <unordered_set>
|
|
#include <map>
|
|
#include <mutex>
|
|
|
|
#include "JFJochReaderDataset.h"
|
|
#include "../common/CrystalLattice.h"
|
|
|
|
constexpr static int32_t GAP_PXL_VALUE = INT32_MIN + 1;
|
|
constexpr static int32_t ERROR_PXL_VALUE = INT32_MIN;
|
|
constexpr static int32_t SATURATED_PXL_VALUE = INT32_MAX;
|
|
|
|
class JFJochReaderImage {
|
|
std::shared_ptr<const JFJochReaderDataset> dataset;
|
|
|
|
std::vector<int32_t> image; // Image in the reader must be 32-bit signed, uncompressed
|
|
DataMessage message;
|
|
|
|
std::unordered_set<int64_t> saturated_pixel;
|
|
std::unordered_set<int64_t> error_pixel;
|
|
std::map<int32_t, int32_t> valid_pixel;
|
|
std::optional<ROIMessage> roi;
|
|
mutable std::mutex roi_mutex;
|
|
|
|
template <class T>
|
|
void ProcessInputImage(const void* image, size_t npixel, int64_t sat_value, int64_t special_value);
|
|
void ProcessInputImage(const CompressedImage& image);
|
|
public:
|
|
JFJochReaderImage(const DataMessage &msg, const std::shared_ptr<const JFJochReaderDataset> &dataset);
|
|
JFJochReaderImage(const JFJochReaderImage &other);
|
|
|
|
const DataMessage &ImageData() const;
|
|
DataMessage &ImageData();
|
|
|
|
const std::vector<int32_t> &Image() const;
|
|
const std::unordered_set<int64_t> &SaturatedPixels() const;
|
|
const std::unordered_set<int64_t> &ErrorPixels() const;
|
|
const std::map<int32_t, int32_t> &ValidPixels() const;
|
|
const JFJochReaderDataset &Dataset() const;
|
|
|
|
const std::optional<ROIMessage> &GetROI() const;
|
|
void AddImage(const JFJochReaderImage& other);
|
|
std::vector<float> GetAzInt1D() const;
|
|
std::vector<float> GetAzInt1D_BinToQ() const;
|
|
|
|
// Functions that change the content
|
|
void CalcROI(const ROIElement *input);
|
|
|
|
std::shared_ptr<JFJochReaderDataset> CreateMutableDataset();
|
|
};
|
|
|
|
#endif //JFJOCH_JFJOCHREADERIMAGE_H
|