63 lines
2.3 KiB
C++
63 lines
2.3 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JUNGFRAUJOCH_AZIMUTHALINTEGRATIONMAPPING_H
|
|
#define JUNGFRAUJOCH_AZIMUTHALINTEGRATIONMAPPING_H
|
|
|
|
#include <optional>
|
|
#include "DiffractionExperiment.h"
|
|
#include "PixelMask.h"
|
|
|
|
class AzimuthalIntegration {
|
|
protected:
|
|
const AzimuthalIntegrationSettings settings;
|
|
const float wavelength;
|
|
const size_t width, height;
|
|
|
|
std::vector<float> bin_to_q;
|
|
std::vector<float> bin_to_2theta;
|
|
std::vector<float> bin_to_d;
|
|
std::vector<float> bin_to_phi;
|
|
std::vector<uint16_t> pixel_to_bin;
|
|
|
|
std::vector<float> pixel_resolution;
|
|
std::vector<float> corrections;
|
|
|
|
std::optional<float> polarization_correction;
|
|
|
|
void UpdateMaxBinNumber();
|
|
AzimuthalIntegration(const AzimuthalIntegrationSettings& settings,
|
|
size_t width, size_t height, float wavelength);
|
|
|
|
void SetupRawGeom(const DiffractionExperiment& experiment,
|
|
const std::vector<uint32_t> &mask);
|
|
|
|
void SetupConvGeom(const DiffractionGeometry &geom,
|
|
const std::vector<uint32_t> &mask);
|
|
|
|
void SetupPixel(const DiffractionGeometry &geom, const std::vector<uint32_t> &mask,
|
|
uint32_t pxl, uint32_t col, uint32_t row);
|
|
public:
|
|
|
|
AzimuthalIntegration(const DiffractionExperiment& experiment,
|
|
const PixelMask& mask);
|
|
|
|
[[nodiscard]] uint16_t GetBinNumber() const;
|
|
[[nodiscard]] const std::vector<uint16_t>&GetPixelToBin() const;
|
|
[[nodiscard]] const std::vector<float> &GetBinToQ() const;
|
|
[[nodiscard]] const std::vector<float> &GetBinToD() const;
|
|
[[nodiscard]] const std::vector<float> &GetBinToTwoTheta() const;
|
|
[[nodiscard]] const std::vector<float> &GetBinToPhi() const;
|
|
[[nodiscard]] uint16_t QToBin(float q) const;
|
|
[[nodiscard]] const std::vector<float> &Corrections() const;
|
|
[[nodiscard]] const std::vector<float> &Resolution() const;
|
|
[[nodiscard]] size_t GetWidth() const;
|
|
[[nodiscard]] size_t GetHeight() const;
|
|
[[nodiscard]] const AzimuthalIntegrationSettings& Settings() const;
|
|
[[nodiscard]] int32_t GetAzimuthalBinCount() const;
|
|
[[nodiscard]] int32_t GetQBinCount() const;
|
|
};
|
|
|
|
|
|
#endif //JUNGFRAUJOCH_AZIMUTHALINTEGRATIONMAPPING_H
|