141 lines
4.1 KiB
C++
141 lines
4.1 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include <catch2/catch_all.hpp>
|
|
|
|
#include "../common/ImageBuffer.h"
|
|
|
|
TEST_CASE("ImageBuffer_Location size") {
|
|
ImageBuffer buf(2*1024*1024);
|
|
REQUIRE_THROWS(buf.StartMeasurement(4 * 1024 * 1024));
|
|
REQUIRE_NOTHROW(buf.StartMeasurement(1024 * 1024));
|
|
REQUIRE(buf.GetStatus().total_slots == 2);
|
|
REQUIRE(buf.GetStatus().available_slots == 2);
|
|
}
|
|
|
|
TEST_CASE("ImageBuffer") {
|
|
DiffractionExperiment experiment(DetectorGeometry(18,3,8,36));
|
|
|
|
ImageBuffer buf_ctrl(4*experiment.GetImageBufferLocationSize());
|
|
buf_ctrl.StartMeasurement(experiment);
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 4);
|
|
|
|
ZeroCopyReturnValue *ret0 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret0 != nullptr);
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 3);
|
|
|
|
ZeroCopyReturnValue *ret1 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret1 != nullptr);
|
|
|
|
ZeroCopyReturnValue *ret2 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret2 != nullptr);
|
|
|
|
ZeroCopyReturnValue *ret3 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret3 != nullptr);
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 0);
|
|
|
|
ZeroCopyReturnValue *ret4 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret4 == nullptr);
|
|
|
|
ret0->release();
|
|
ret1->release();
|
|
ret2->release();
|
|
ret3->release();
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 4);
|
|
|
|
REQUIRE(buf_ctrl.CheckIfBufferReturned(std::chrono::microseconds(1)));
|
|
}
|
|
|
|
TEST_CASE("ImageBuffer_Preview_Access") {
|
|
DiffractionExperiment experiment(DetectorGeometry(18,3,8,36));
|
|
|
|
ImageBuffer buf_ctrl(4*experiment.GetImageBufferLocationSize());
|
|
buf_ctrl.StartMeasurement(experiment);
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 4);
|
|
|
|
ZeroCopyReturnValue *ret0 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret0 != nullptr);
|
|
ret0->SetImageNumber(0);
|
|
ret0->release();
|
|
|
|
ZeroCopyReturnValue *ret1 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret1 != nullptr);
|
|
ret1->SetImageNumber(1);
|
|
ret1->release();
|
|
|
|
ZeroCopyReturnValue *ret2 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret2 != nullptr);
|
|
ret2->SetImageNumber(2);
|
|
ret2->SetImageSize(3456);
|
|
ret2->release();
|
|
|
|
ZeroCopyReturnValue *ret3 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret3 != nullptr);
|
|
ret3->SetImageNumber(15);
|
|
ret3->release();
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 4);
|
|
|
|
auto v = buf_ctrl.GetStatus().images_in_the_buffer;
|
|
REQUIRE(v.size() == 4);
|
|
REQUIRE(v[0] == 0);
|
|
REQUIRE(v[3] == 15);
|
|
|
|
std::vector<uint8_t> image_2, image_2_2, image_12;
|
|
REQUIRE(buf_ctrl.GetImage(image_2, 2));
|
|
REQUIRE(image_2.size() == 3456);
|
|
|
|
REQUIRE(!buf_ctrl.GetImage(image_12, 12));
|
|
|
|
REQUIRE(buf_ctrl.Finalize(std::chrono::microseconds(1)));
|
|
|
|
// After finalize was run, there should be no preview
|
|
REQUIRE(!buf_ctrl.GetImage(image_2_2, 2));
|
|
}
|
|
|
|
TEST_CASE("ImageBuffer_Restart") {
|
|
DiffractionExperiment experiment(DetectorGeometry(18,3,8,36));
|
|
|
|
ImageBuffer buf_ctrl(4*experiment.GetImageBufferLocationSize());
|
|
buf_ctrl.StartMeasurement(experiment);
|
|
|
|
ZeroCopyReturnValue *ret0 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret0 != nullptr);
|
|
ret0->SetImageNumber(0);
|
|
ret0->release();
|
|
|
|
REQUIRE(buf_ctrl.GetStatus().images_in_the_buffer.size() == 1);
|
|
|
|
// Restart measurement
|
|
REQUIRE_NOTHROW(buf_ctrl.StartMeasurement(experiment));
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 4);
|
|
REQUIRE(buf_ctrl.GetStatus().images_in_the_buffer.empty());
|
|
|
|
|
|
ret0 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret0 != nullptr);
|
|
ret0->SetImageNumber(0);
|
|
ret0->release();
|
|
|
|
REQUIRE_NOTHROW(buf_ctrl.StartMeasurement(experiment));
|
|
}
|
|
|
|
TEST_CASE("ImageBuffer_BuffersNotReturned") {
|
|
DiffractionExperiment experiment(DetectorGeometry(18,3,8,36));
|
|
|
|
ImageBuffer buf_ctrl(4*experiment.GetImageBufferLocationSize());
|
|
buf_ctrl.StartMeasurement(experiment);
|
|
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 4);
|
|
|
|
ZeroCopyReturnValue *ret0 = buf_ctrl.GetImageSlot();
|
|
REQUIRE(ret0 != nullptr);
|
|
REQUIRE(buf_ctrl.GetAvailSlots() == 3);
|
|
REQUIRE(!buf_ctrl.CheckIfBufferReturned(std::chrono::microseconds(1)));
|
|
}
|