Files
Jungfraujoch/common/DiffractionGeometry.h
2025-10-20 20:43:44 +02:00

66 lines
2.8 KiB
C++

// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only
#ifndef JUNGFRAUJOCH_DIFFRACTIONGEOMETRY_H
#define JUNGFRAUJOCH_DIFFRACTIONGEOMETRY_H
#include "JFJochException.h"
#include "Coord.h"
class DiffractionGeometry {
float beam_x_pxl = 0.0;
float beam_y_pxl = 0.0;
float det_distance_mm = 100.0;
float pixel_size_mm = 0.075;
float wavelength_A = 1.0;
float poni_rot_1 = 0.0f;
float poni_rot_2 = 0.0f;
float poni_rot_3 = 0.0f;
RotMatrix poni_rot;
void UpdatePoniRotMatrix();
public:
DiffractionGeometry &BeamX_pxl(float input);
DiffractionGeometry &BeamY_pxl(float input);
DiffractionGeometry &DetectorDistance_mm(float input);
DiffractionGeometry &PixelSize_mm(float input);
DiffractionGeometry &Wavelength_A(float input);
DiffractionGeometry &PoniRot1_rad(float input);
DiffractionGeometry &PoniRot2_rad(float input);
DiffractionGeometry &PoniRot3_rad(float input);
[[nodiscard]] float GetBeamX_pxl() const;
[[nodiscard]] float GetBeamY_pxl() const;
[[nodiscard]] float GetDetectorDistance_mm() const;
[[nodiscard]] float GetPixelSize_mm() const;
[[nodiscard]] float GetWavelength_A() const;
[[nodiscard]] Coord GetScatteringVector() const;
[[nodiscard]] float GetPoniRot1_rad() const;
[[nodiscard]] float GetPoniRot2_rad() const;
[[nodiscard]] float GetPoniRot3_rad() const;
[[nodiscard]] std::pair<float, float> GetDirectBeam_pxl() const;
[[nodiscard]] Coord LabCoord(float x, float y) const;
[[nodiscard]] Coord DetectorToRecip(float x, float y) const;
[[nodiscard]] std::pair<float, float> RecipToDector(const Coord &recip) const;
[[nodiscard]] float TwoTheta_rad(float x, float y) const;
[[nodiscard]] float Phi_rad(float x, float y) const;
[[nodiscard]] float PxlToRes(float x, float y) const;
[[nodiscard]] float PxlToQ(float x, float y) const;
[[nodiscard]] float PxlToRes(float dist_pxl) const;
[[nodiscard]] float ResToPxl(float d_A) const;
[[nodiscard]] Coord ResToPxl(float d_A, float phi) const;
[[nodiscard]] float DistFromEwaldSphere(const Coord& recip) const;
[[nodiscard]] float CalcAzIntSolidAngleCorr(float q) const;
[[nodiscard]] float CalcAzIntSolidAngleCorr(float x, float y) const;
[[nodiscard]] float CalcAzIntPolarizationCorr(float x, float y, float coeff) const;
[[nodiscard]] std::pair<float, float> ResPhiToPxl(float d_A, float phi_rad) const;
[[nodiscard]] Coord ProjectToEwaldSphere(const Coord &p0) const;
// eq. 18 in https://journals.iucr.org/d/issues/2014/08/00/dz5332/index.html
[[nodiscard]] float AngleFromEwaldSphere_deg(const Coord &p0) const;
[[nodiscard]] const RotMatrix& GetPoniRotMatrix() const;
};
#endif //JUNGFRAUJOCH_DIFFRACTIONGEOMETRY_H