143 lines
4.3 KiB
C++
143 lines
4.3 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include <catch2/catch_all.hpp>
|
|
#include "../receiver/JFJochReceiverPlots.h"
|
|
|
|
TEST_CASE("JFJochReceiverPlots") {
|
|
DiffractionExperiment experiment;
|
|
experiment.Goniometer(GoniometerAxis("omega", 15, 2, Coord(0,1,0), {}));
|
|
|
|
PixelMask mask(experiment);
|
|
AzimuthalIntegration integration(experiment, mask);
|
|
|
|
JFJochReceiverPlots plots;
|
|
plots.Setup(experiment, integration);
|
|
DataMessage msg_0 {
|
|
.number = 0,
|
|
.error_pixel_count = 11
|
|
};
|
|
|
|
DataMessage msg_1 {
|
|
.number = 2,
|
|
.error_pixel_count = 12
|
|
};
|
|
plots.Add(msg_0, AzimuthalIntegrationProfile(integration));
|
|
plots.Add(msg_1, AzimuthalIntegrationProfile(integration));
|
|
|
|
auto p = plots.GetPlots(PlotRequest{
|
|
.type = PlotType::ErrorPixels,
|
|
.binning = 0,
|
|
.experimental_coord = false
|
|
});
|
|
|
|
REQUIRE(p.GetUnits() == MultiLinePlotUnits::ImageNumber);
|
|
REQUIRE(p.GetPlots().size() == 1);
|
|
REQUIRE(p.GetPlots()[0].x.size() == 2);
|
|
REQUIRE(p.GetPlots()[0].y.size() == 2);
|
|
REQUIRE(p.GetPlots()[0].z.empty());
|
|
|
|
REQUIRE(p.GetPlots()[0].x[0] == 0);
|
|
REQUIRE(p.GetPlots()[0].x[1] == 2);
|
|
REQUIRE(p.GetPlots()[0].y[0] == 11);
|
|
REQUIRE(p.GetPlots()[0].y[1] == 12);
|
|
}
|
|
|
|
TEST_CASE("JFJochReceiverPlots_Angle") {
|
|
DiffractionExperiment experiment;
|
|
experiment.Goniometer(GoniometerAxis("omega", 15, 2, Coord(0,1,0), {}));
|
|
|
|
PixelMask mask(experiment);
|
|
AzimuthalIntegration integration(experiment, mask);
|
|
|
|
JFJochReceiverPlots plots;
|
|
plots.Setup(experiment, integration);
|
|
DataMessage msg_0 {
|
|
.number = 0,
|
|
.error_pixel_count = 11
|
|
};
|
|
|
|
DataMessage msg_1 {
|
|
.number = 2,
|
|
.error_pixel_count = 12
|
|
};
|
|
plots.Add(msg_0, AzimuthalIntegrationProfile(integration));
|
|
plots.Add(msg_1, AzimuthalIntegrationProfile(integration));
|
|
|
|
auto p = plots.GetPlots(PlotRequest{
|
|
.type = PlotType::ErrorPixels,
|
|
.binning = 0,
|
|
.experimental_coord = true
|
|
});
|
|
|
|
REQUIRE(p.GetUnits() == MultiLinePlotUnits::Angle_deg);
|
|
REQUIRE(p.GetPlots().size() == 1);
|
|
REQUIRE(p.GetPlots()[0].x.size() == 2);
|
|
REQUIRE(p.GetPlots()[0].y.size() == 2);
|
|
REQUIRE(p.GetPlots()[0].z.empty());
|
|
|
|
REQUIRE(p.GetPlots()[0].x[0] == 15.0);
|
|
REQUIRE(p.GetPlots()[0].x[1] == 15.0 + 2.0 * 2);
|
|
REQUIRE(p.GetPlots()[0].y[0] == 11);
|
|
REQUIRE(p.GetPlots()[0].y[1] == 12);
|
|
}
|
|
|
|
TEST_CASE("JFJochReceiverPlots_Grid") {
|
|
DatasetSettings dataset;
|
|
dataset.GridScan(GridScanSettings(5, 10, 20, false, false));
|
|
dataset.ImagesPerTrigger(12);
|
|
|
|
DiffractionExperiment experiment;
|
|
experiment.ImportDatasetSettings(dataset);
|
|
|
|
PixelMask mask(experiment);
|
|
AzimuthalIntegration integration(experiment, mask);
|
|
|
|
JFJochReceiverPlots plots;
|
|
plots.Setup(experiment, integration);
|
|
DataMessage msg_0 {
|
|
.number = 0,
|
|
.error_pixel_count = 11
|
|
};
|
|
|
|
DataMessage msg_1 {
|
|
.number = 2,
|
|
.error_pixel_count = 12
|
|
};
|
|
|
|
DataMessage msg_3 {
|
|
.number = 11,
|
|
.error_pixel_count = 8
|
|
};
|
|
|
|
plots.Add(msg_0, AzimuthalIntegrationProfile(integration));
|
|
plots.Add(msg_1, AzimuthalIntegrationProfile(integration));
|
|
plots.Add(msg_3, AzimuthalIntegrationProfile(integration));
|
|
|
|
auto p = plots.GetPlots(PlotRequest{
|
|
.type = PlotType::ErrorPixels,
|
|
.binning = 0,
|
|
.experimental_coord = true
|
|
});
|
|
|
|
REQUIRE(p.GetUnits() == MultiLinePlotUnits::Grid_um);
|
|
REQUIRE(p.GetPlots().size() == 1);
|
|
REQUIRE(p.GetPlots()[0].x.size() == 3);
|
|
REQUIRE(p.GetPlots()[0].y.size() == 3);
|
|
REQUIRE(p.GetPlots()[0].z.size() == 3);
|
|
|
|
REQUIRE(p.GetPlots()[0].x[0] == 0.5 * 10.0);
|
|
REQUIRE(p.GetPlots()[0].x[1] == 2.5 * 10.0);
|
|
REQUIRE(p.GetPlots()[0].x[2] == 1.5 * 10.0);
|
|
|
|
REQUIRE(p.GetPlots()[0].y[0] == 0.5 * 20.0);
|
|
REQUIRE(p.GetPlots()[0].y[1] == 0.5 * 20.0);
|
|
REQUIRE(p.GetPlots()[0].y[2] == 2.5 * 20.0);
|
|
|
|
REQUIRE(p.GetPlots()[0].z[0] == 11);
|
|
REQUIRE(p.GetPlots()[0].z[1] == 12);
|
|
REQUIRE(p.GetPlots()[0].z[2] == 8);
|
|
|
|
REQUIRE(p.GetSizeX() == 10.0 * 5);
|
|
REQUIRE(p.GetSizeY() == 20.0 * 3);
|
|
} |