401 lines
17 KiB
C++
401 lines
17 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#include <catch2/catch.hpp>
|
|
#include "../broker/JFJochStateMachine.h"
|
|
#include "../broker/JFJochBroker.h"
|
|
|
|
using namespace std::literals::chrono_literals;
|
|
|
|
TEST_CASE("JFJochStateMachine_States") {
|
|
Logger logger("JFJochStateMachine_States");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
state_machine.AddDetectorSetup(DetectorGeometry(4));
|
|
|
|
JFJochProtoBuf::DatasetSettings setup;
|
|
setup.set_ntrigger(1);
|
|
setup.set_detector_distance_mm(100);
|
|
setup.set_file_prefix("integration_test");
|
|
setup.set_images_per_trigger(5);
|
|
setup.set_photon_energy_kev(12.4);
|
|
setup.set_data_file_count(5);
|
|
|
|
JFJochState state;
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Inactive);
|
|
|
|
REQUIRE_THROWS(state_machine.Start(setup));
|
|
REQUIRE_THROWS(state_machine.Pedestal());
|
|
|
|
REQUIRE_NOTHROW(state_machine.Initialize());
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Idle);
|
|
|
|
REQUIRE_NOTHROW(state_machine.Deactivate());
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Inactive);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_State_Pedestal") {
|
|
Logger logger("JFJochStateMachine_State_Pedestal");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
state_machine.AddDetectorSetup(DetectorGeometry(4));
|
|
|
|
JFJochProtoBuf::DatasetSettings setup;
|
|
setup.set_ntrigger(1);
|
|
setup.set_detector_distance_mm(100);
|
|
setup.set_file_prefix("integration_test");
|
|
setup.set_images_per_trigger(5);
|
|
setup.set_photon_energy_kev(12.4);
|
|
setup.set_data_file_count(5);
|
|
|
|
state_machine.DebugOnly_SetState(JFJochState::Pedestal);
|
|
|
|
JFJochState state;
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Pedestal);
|
|
|
|
REQUIRE_THROWS(state_machine.Start(setup));
|
|
REQUIRE_THROWS(state_machine.Pedestal());
|
|
REQUIRE_THROWS(state_machine.Initialize());
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(500);
|
|
settings.set_storage_cell_count(16);
|
|
REQUIRE_THROWS(state_machine.SetDetectorSettings(settings));
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_State_Measure") {
|
|
Logger logger("JFJochStateMachine_State_Measure");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
state_machine.AddDetectorSetup(DetectorGeometry(4));
|
|
|
|
JFJochProtoBuf::DatasetSettings setup;
|
|
setup.set_ntrigger(1);
|
|
setup.set_detector_distance_mm(100);
|
|
setup.set_file_prefix("integration_test");
|
|
setup.set_images_per_trigger(5);
|
|
setup.set_photon_energy_kev(12.4);
|
|
setup.set_data_file_count(5);
|
|
|
|
state_machine.DebugOnly_SetState(JFJochState::Measuring);
|
|
|
|
JFJochState state;
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Measuring);
|
|
|
|
REQUIRE_THROWS(state_machine.Start(setup));
|
|
REQUIRE_THROWS(state_machine.Pedestal());
|
|
REQUIRE_THROWS(state_machine.Initialize());
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(500);
|
|
settings.set_storage_cell_count(16);
|
|
REQUIRE_THROWS(state_machine.SetDetectorSettings(settings));
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_State_Error") {
|
|
Logger logger("JFJochStateMachine_State_Error");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
state_machine.AddDetectorSetup(DetectorGeometry(4));
|
|
|
|
JFJochProtoBuf::DatasetSettings setup;
|
|
setup.set_ntrigger(1);
|
|
setup.set_detector_distance_mm(100);
|
|
setup.set_file_prefix("integration_test");
|
|
setup.set_images_per_trigger(5);
|
|
setup.set_photon_energy_kev(12.4);
|
|
setup.set_data_file_count(5);
|
|
|
|
state_machine.DebugOnly_SetState(JFJochState::Error);
|
|
|
|
JFJochState state;
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Error);
|
|
|
|
REQUIRE_THROWS(state_machine.Start(setup));
|
|
REQUIRE_THROWS(state_machine.Pedestal());
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(500);
|
|
settings.set_storage_cell_count(16);
|
|
REQUIRE_NOTHROW(state_machine.SetDetectorSettings(settings));
|
|
|
|
REQUIRE_NOTHROW(state_machine.Initialize());
|
|
REQUIRE_NOTHROW(state = state_machine.GetState());
|
|
REQUIRE(state == JFJochState::Idle);
|
|
}
|
|
|
|
|
|
TEST_CASE("JFJochStateMachine_Setup") {
|
|
Logger logger("JJFJochStateMachine_Setup");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
|
|
JFJochProtoBuf::DetectorSettings settings, settings_save;
|
|
settings.set_pedestal_g1_frames(-15);
|
|
settings.set_pedestal_g0_frames(2378);
|
|
REQUIRE_THROWS(state_machine.SetDetectorSettings(settings));
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG0Frames() != 2378);
|
|
|
|
settings.set_pedestal_g2_frames(2800);
|
|
settings.set_pedestal_g1_frames(3000);
|
|
settings.set_pedestal_g0_frames(2378);
|
|
settings.set_storage_cell_count(16);
|
|
settings.set_frame_time_us(600);
|
|
settings.set_count_time_us(247);
|
|
REQUIRE_NOTHROW(state_machine.SetDetectorSettings(settings));
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG0Frames() == 2378);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG1Frames() == 3000);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG2Frames() == 2800);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellNumber() == 16);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameTime() == std::chrono::microseconds(600));
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameCountTime() == std::chrono::microseconds(247));
|
|
|
|
REQUIRE_NOTHROW(settings_save = state_machine.GetDetectorSettings());
|
|
REQUIRE(settings_save.storage_cell_count() == 16);
|
|
REQUIRE(settings_save.count_time_us() == 247);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_NoDetectorSetup") {
|
|
Logger logger("JFJochStateMachine_NoDetectorSetup");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
REQUIRE_THROWS(state_machine.Initialize());
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_AddDetectorSetup") {
|
|
Logger logger("JFJochStateMachine_AddDetectorSetup");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
DetectorSetup setup = DetectorGeometry(4);
|
|
state_machine.AddDetectorSetup(setup);
|
|
REQUIRE_NOTHROW(state_machine.Initialize());
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_AddDetectorSetup_Gain") {
|
|
Logger logger("JFJochStateMachine_AddDetectorSetup_Gain");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
DetectorSetup setup = DetectorGeometry(4);
|
|
setup.LoadGain({"../../tests/test_data/gainMaps_M049.bin",
|
|
"../../tests/test_data/gainMaps_M049.bin",
|
|
"../../tests/test_data/gainMaps_M049.bin",
|
|
"../../tests/test_data/gainMaps_M049.bin"});
|
|
state_machine.AddDetectorSetup(setup);
|
|
REQUIRE_NOTHROW(state_machine.Initialize());
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_StorageCells") {
|
|
Logger logger("JFJochBrokerService_StorageCells");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
state_machine.AddDetectorSetup(DetectorGeometry(4));
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(500);
|
|
settings.set_storage_cell_count(16);
|
|
REQUIRE_NOTHROW(state_machine.SetDetectorSettings(settings));
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellNumber() == 16);
|
|
REQUIRE_NOTHROW(state_machine.Initialize());
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_AddDetectorSetup_Multiple") {
|
|
Logger logger("JFJochBrokerService_StorageCells");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
REQUIRE_NOTHROW(state_machine.AddDetectorSetup(DetectorSetup(DetectorGeometry(4), "Det1", {"mx1", "mx2", "mx3", "mx4"})));
|
|
REQUIRE_NOTHROW(state_machine.AddDetectorSetup(DetectorSetup(DetectorGeometry(2), "Det2", {"mx1", "mx2"})));
|
|
REQUIRE_NOTHROW(state_machine.AddDetectorSetup(DetectorSetup(DetectorGeometry(1), "Det3", {"mx1"})));
|
|
|
|
auto dl = state_machine.GetDetectorsList();
|
|
REQUIRE(dl.detector_size() == 3);
|
|
REQUIRE(dl.detector(0).description() == "Det1");
|
|
REQUIRE(dl.detector(0).nmodules() == 4);
|
|
REQUIRE(dl.detector(1).description() == "Det2");
|
|
REQUIRE(dl.detector(1).nmodules() == 2);
|
|
REQUIRE(dl.detector(2).description() == "Det3");
|
|
REQUIRE(dl.detector(2).nmodules() == 1);
|
|
|
|
REQUIRE_NOTHROW(state_machine.Initialize());
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetModulesNum() == 4);
|
|
REQUIRE(state_machine.GetState() == JFJochState::Idle);
|
|
|
|
REQUIRE_THROWS(state_machine.SelectDetector(7));
|
|
REQUIRE(state_machine.GetState() == JFJochState::Idle);
|
|
|
|
REQUIRE_NOTHROW(state_machine.SelectDetector(2));
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetModulesNum() == 1);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetDetectorDescription() == "Det3");
|
|
REQUIRE(state_machine.GetState() == JFJochState::Inactive);
|
|
}
|
|
|
|
TEST_CASE("JFJochBroker") {
|
|
DiffractionExperiment experiment;
|
|
|
|
JFJochBroker broker(experiment);
|
|
broker.AddDetectorSetup(DetectorGeometry(4));
|
|
|
|
REQUIRE(broker.Initialize(nullptr, nullptr, nullptr).ok());
|
|
|
|
JFJochProtoBuf::DatasetSettings setup;
|
|
setup.set_ntrigger(1);
|
|
setup.set_detector_distance_mm(100);
|
|
setup.set_file_prefix("integration_test");
|
|
setup.set_images_per_trigger(5);
|
|
setup.set_photon_energy_kev(12.4);
|
|
setup.set_data_file_count(5);
|
|
|
|
REQUIRE(broker.Start(nullptr, &setup, nullptr).ok());
|
|
REQUIRE(broker.Stop(nullptr, nullptr, nullptr).ok());
|
|
REQUIRE(broker.Deactivate(nullptr, nullptr, nullptr).ok());
|
|
}
|
|
|
|
TEST_CASE("FJochStateMachine_LoadDatasetSettings", "[DiffractionExperiment]") {
|
|
DiffractionExperiment x;
|
|
x.ImagesPerTrigger(567).BeamY_pxl(324).Compression(CompressionAlgorithm::BSHUF_ZSTD);
|
|
JFJochProtoBuf::DatasetSettings settings;
|
|
settings.set_images_per_trigger(234);
|
|
settings.set_ntrigger(56);
|
|
settings.set_beam_x_pxl(23.4);
|
|
settings.set_beam_y_pxl(123.4);
|
|
settings.set_photon_energy_kev(WVL_1A_IN_KEV);
|
|
settings.set_detector_distance_mm(57.6);
|
|
settings.set_data_file_count(5);
|
|
settings.set_space_group_number(45);
|
|
settings.set_sample_name("lyso1");
|
|
settings.set_summation(36);
|
|
settings.set_fpga_pixel_output(JFJochProtoBuf::INT16);
|
|
REQUIRE_NOTHROW(LoadDatasetSettings(x,settings));
|
|
|
|
REQUIRE(x.GetImageNumPerTrigger() == 234);
|
|
REQUIRE(x.GetBeamX_pxl() == Approx(23.4));
|
|
REQUIRE(x.GetBeamY_pxl() == Approx(123.4));
|
|
REQUIRE(x.GetSpaceGroupNumber() == 45);
|
|
REQUIRE(x.GetCompressionAlgorithm() == CompressionAlgorithm::BSHUF_LZ4);
|
|
REQUIRE(x.GetSampleName() == "lyso1");
|
|
REQUIRE(x.GetDataFileCount() == 5);
|
|
REQUIRE(x.GetDetectorDistance_mm() == Approx(57.6));
|
|
REQUIRE(x.GetSummation() == 36);
|
|
REQUIRE(x.GetFPGAOutputMode() == FPGAPixelOutput::Int16);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_LoadDatasetSettings_Invalid", "[DiffractionExperiment]") {
|
|
DiffractionExperiment x;
|
|
x.ImagesPerTrigger(567).BeamY_pxl(324).Compression(CompressionAlgorithm::BSHUF_ZSTD);
|
|
JFJochProtoBuf::DatasetSettings settings;
|
|
settings.set_images_per_trigger(-1);
|
|
settings.set_ntrigger(56);
|
|
settings.set_beam_x_pxl(23.4);
|
|
settings.set_beam_y_pxl(123.4);
|
|
settings.set_photon_energy_kev(WVL_1A_IN_KEV);
|
|
settings.set_detector_distance_mm(57.6);
|
|
settings.set_data_file_count(5);
|
|
settings.set_space_group_number(45);
|
|
settings.set_sample_name("lyso1");
|
|
|
|
REQUIRE_THROWS(LoadDatasetSettings(x, settings));
|
|
|
|
REQUIRE(x.GetImageNumPerTrigger() == 567);
|
|
REQUIRE(x.GetBeamY_pxl() == Approx(324));
|
|
REQUIRE(x.GetSpaceGroupNumber() == 0);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_SetDetectorSettings", "[DiffractionExperiment]") {
|
|
Logger logger("JFJochStateMachine_SetDetectorSettings");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
|
|
state_machine.NotThreadSafe_Experiment().PedestalG0Frames(456).PedestalG1Frames(1234).PedestalG2Frames(123).StorageCellDelay(2500ns);
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(600);
|
|
settings.set_count_time_us(400);
|
|
settings.set_storage_cell_count(8);
|
|
settings.set_use_internal_packet_generator(true);
|
|
settings.set_collect_raw_data(true);
|
|
settings.set_pedestal_g0_frames(5000);
|
|
settings.set_pedestal_g1_frames(100);
|
|
settings.set_pedestal_g2_frames(150);
|
|
|
|
REQUIRE_NOTHROW(state_machine.SetDetectorSettings(settings));
|
|
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameTime().count() == 600);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameCountTime().count() == 400);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().IsUsingInternalPacketGen());
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellNumber() == 8);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetDetectorMode() == DetectorMode::Raw);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG0Frames() == 5000);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG1Frames() == 100);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG2Frames() == 150);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellDelay().count() == 2500);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_SetDetectorSettings_StorageCellDelay", "[DiffractionExperiment]") {
|
|
Logger logger("JFJochStateMachine_SetDetectorSettings_StorageCellDelay");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
|
|
state_machine.NotThreadSafe_Experiment().PedestalG0Frames(456).PedestalG1Frames(1234).PedestalG2Frames(123).StorageCellDelay(5000ns);
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(600);
|
|
settings.set_count_time_us(400);
|
|
settings.set_storage_cell_count(8);
|
|
settings.set_use_internal_packet_generator(true);
|
|
settings.set_collect_raw_data(true);
|
|
settings.set_storage_cell_delay_ns(7000);
|
|
REQUIRE_NOTHROW(state_machine.SetDetectorSettings(settings));
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellDelay().count() == 7000);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_SetDetectorSettings_invalid", "[DiffractionExperiment]") {
|
|
Logger logger("JFJochStateMachine_SetDetectorSettings_invalid");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
|
|
state_machine.NotThreadSafe_Experiment().PedestalG0Frames(456).PedestalG1Frames(1234).PedestalG2Frames(123)
|
|
.FrameTime(525us).Mode(DetectorMode::Conversion);
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(600);
|
|
settings.set_count_time_us(800);
|
|
settings.set_storage_cell_count(16);
|
|
settings.set_use_internal_packet_generator(true);
|
|
settings.set_collect_raw_data(true);
|
|
settings.set_pedestal_g0_frames(5000);
|
|
settings.set_pedestal_g1_frames(100);
|
|
settings.set_pedestal_g2_frames(150);
|
|
REQUIRE_THROWS(state_machine.SetDetectorSettings(settings));
|
|
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameTime().count() == 525);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameCountTime().count() == 525 - READOUT_TIME_IN_US);
|
|
REQUIRE(!state_machine.NotThreadSafe_Experiment().IsUsingInternalPacketGen());
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellNumber() == 1);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetDetectorMode() == DetectorMode::Conversion);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG0Frames() == 456);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG1Frames() == 1234);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetPedestalG2Frames() == 123);
|
|
}
|
|
|
|
TEST_CASE("JFJochStateMachine_SetDetectorSettings_inferred", "[DiffractionExperiment]") {
|
|
Logger logger("JFJochStateMachine_SetDetectorSettings_inferred");
|
|
JFJochServices services(logger);
|
|
JFJochStateMachine state_machine(services, logger);
|
|
|
|
state_machine.NotThreadSafe_Experiment().PedestalG0Frames(456).PedestalG1Frames(1234).PedestalG2Frames(123);
|
|
|
|
JFJochProtoBuf::DetectorSettings settings;
|
|
settings.set_frame_time_us(600);
|
|
settings.set_storage_cell_count(1);
|
|
REQUIRE_NOTHROW(state_machine.SetDetectorSettings(settings));
|
|
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameTime().count() == 600);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetFrameCountTime().count() == 600 - READOUT_TIME_IN_US);
|
|
REQUIRE(!state_machine.NotThreadSafe_Experiment().IsUsingInternalPacketGen());
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetStorageCellNumber() == 1);
|
|
REQUIRE(state_machine.NotThreadSafe_Experiment().GetDetectorMode() == DetectorMode::Conversion);
|
|
} |