285 lines
14 KiB
C++
285 lines
14 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#include <catch2/catch.hpp>
|
|
#include <iostream>
|
|
#include <bitshuffle/bitshuffle.h>
|
|
|
|
#include "../common/FrameTransformation.h"
|
|
#include "../common/RawToConvertedGeometry.h"
|
|
#include "../compression/JFJochDecompress.h"
|
|
|
|
using namespace std::literals::chrono_literals;
|
|
|
|
TEST_CASE("Bshuf_SSE", "[bitshuffle]") {
|
|
REQUIRE (bshuf_using_SSE2() == 1);
|
|
}
|
|
|
|
TEST_CASE("FrameTransformation_Raw_NoCompression" ,"") {
|
|
const uint16_t nmodules = 4;
|
|
const uint16_t ndatastreams = 2;
|
|
DiffractionExperiment experiment(DetectorGeometry(ndatastreams * nmodules, 1));
|
|
experiment.DataStreams(ndatastreams);
|
|
|
|
experiment.Mode(DetectorMode::Raw);
|
|
experiment.Compression(JFJochProtoBuf::NO_COMPRESSION);
|
|
|
|
FrameTransformation transformation(experiment);
|
|
|
|
std::mt19937 g1(1587);
|
|
std::uniform_int_distribution<int16_t> dist;
|
|
|
|
std::vector<int16_t> input_0(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_0[i] = dist(g1);
|
|
|
|
std::vector<int16_t> input_1(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_1[i] = dist(g1);
|
|
|
|
std::vector<int16_t> output(experiment.GetPixelsNum());
|
|
|
|
for (int i = 0; i < nmodules; i++) {
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_0.data() + i * RAW_MODULE_SIZE, i, 0));
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_1.data() + i * RAW_MODULE_SIZE, i, 1));
|
|
}
|
|
|
|
REQUIRE(transformation.SaveCompressedImage(output.data()) == experiment.GetPixelDepth() * experiment.GetPixelsNum());
|
|
|
|
uint32_t diff_0 = 0;
|
|
uint32_t diff_1 = 0;
|
|
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++) {
|
|
if (input_0[i] != output[i]) diff_0++;
|
|
if (input_1[i] != output[i + nmodules*RAW_MODULE_SIZE]) diff_1++;
|
|
}
|
|
REQUIRE(diff_0 == 0);
|
|
REQUIRE(diff_1 == 0);
|
|
}
|
|
|
|
TEST_CASE("FrameTransformation_Converted_NoCompression" ,"") {
|
|
const uint16_t nmodules = 4;
|
|
const uint16_t ndatastreams = 2;
|
|
DiffractionExperiment experiment(DetectorGeometry(ndatastreams * nmodules, 2));
|
|
experiment.DataStreams(ndatastreams);
|
|
experiment.Mode(DetectorMode::Conversion).Compression(JFJochProtoBuf::NO_COMPRESSION);
|
|
|
|
FrameTransformation transformation(experiment);
|
|
|
|
std::mt19937 g1(1687);
|
|
std::uniform_int_distribution<int16_t> dist;
|
|
|
|
std::vector<int16_t> input_0(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_0[i] = dist(g1);
|
|
|
|
std::vector<int16_t> input_1(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_1[i] = dist(g1);
|
|
|
|
std::vector<int16_t> output(experiment.GetPixelsNum());
|
|
|
|
for (int i = 0; i < nmodules; i++) {
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_0.data() + i * RAW_MODULE_SIZE, i, 0));
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_1.data() + i * RAW_MODULE_SIZE, i, 1));
|
|
}
|
|
|
|
REQUIRE(transformation.SaveCompressedImage(output.data()) == experiment.GetPixelDepth() * experiment.GetPixelsNum());
|
|
|
|
REQUIRE(input_0[511*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 0]);
|
|
REQUIRE(input_0[511*1024+256]/2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 258]);
|
|
REQUIRE(input_0[256*1024+256]/4 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 2 * 1030 * 255 + 257]);
|
|
|
|
REQUIRE(input_0[311*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_0[311*1024+256] / 2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 258]);
|
|
|
|
REQUIRE(input_0[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030]);
|
|
REQUIRE(input_0[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030 + 800 + 6]);
|
|
|
|
REQUIRE(input_1[511*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 0]);
|
|
REQUIRE(input_1[511*1024+256] /2 == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 258]);
|
|
|
|
REQUIRE(input_1[(311+2*512)*1024] == output[200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_1[(311+2*512)*1024+512] / 2 == output[200 * 1030 * 2 + 256*2+3]);
|
|
|
|
REQUIRE(input_1[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030]);
|
|
REQUIRE(input_1[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030 + 800 + 6]);
|
|
}
|
|
|
|
TEST_CASE("FrameTransformation_Converted_bshuf_lz4" ,"") {
|
|
const uint16_t nmodules = 4;
|
|
const uint16_t ndatastreams = 2;
|
|
DiffractionExperiment experiment(DetectorGeometry(ndatastreams * nmodules, 2));
|
|
experiment.DataStreams(ndatastreams);
|
|
|
|
experiment.Mode(DetectorMode::Conversion).Compression(JFJochProtoBuf::BSHUF_LZ4);
|
|
|
|
FrameTransformation transformation(experiment);
|
|
|
|
// Predictable random number generator
|
|
std::mt19937 g1(23433);
|
|
std::uniform_int_distribution<int16_t> distribution;
|
|
|
|
std::vector<int16_t> input_0(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_0[i] = distribution(g1);
|
|
|
|
std::vector<int16_t> input_1(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_1[i] = distribution(g1);
|
|
|
|
std::vector<char> output_compressed(experiment.GetMaxCompressedSize());
|
|
|
|
for (int i = 0; i < nmodules; i++) {
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_0.data() + i * RAW_MODULE_SIZE, i, 0));
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_1.data() + i * RAW_MODULE_SIZE, i, 1));
|
|
}
|
|
|
|
size_t compressed_size;
|
|
REQUIRE_NOTHROW(compressed_size = transformation.SaveCompressedImage(output_compressed.data()));
|
|
output_compressed.resize(compressed_size);
|
|
|
|
REQUIRE(bshuf_read_uint64_BE(output_compressed.data()) == experiment.GetPixelsNum() * experiment.GetPixelDepth());
|
|
REQUIRE(bshuf_read_uint32_BE(output_compressed.data()+8) == JFJochBitShuffleCompressor::DefaultBlockSize * experiment.GetPixelDepth());
|
|
|
|
std::vector<int16_t> output;
|
|
REQUIRE_NOTHROW(JFJochDecompress(output, experiment.GetCompressionAlgorithmEnum(), output_compressed,
|
|
experiment.GetPixelsNum()));
|
|
|
|
REQUIRE(input_0[511*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 0]);
|
|
REQUIRE(input_0[511*1024+256]/2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 258]);
|
|
REQUIRE(input_0[256*1024+256]/4 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 2 * 1030 * 255 + 257]);
|
|
|
|
REQUIRE(input_0[311*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_0[311*1024+256] / 2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 258]);
|
|
|
|
REQUIRE(input_0[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030]);
|
|
REQUIRE(input_0[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030 + 800 + 6]);
|
|
|
|
REQUIRE(input_1[511*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 0]);
|
|
REQUIRE(input_1[511*1024+256] /2 == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 258]);
|
|
|
|
REQUIRE(input_1[(311+2*512)*1024] == output[200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_1[(311+2*512)*1024+512] / 2 == output[200 * 1030 * 2 + 256*2+3]);
|
|
|
|
REQUIRE(input_1[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030]);
|
|
REQUIRE(input_1[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030 + 800 + 6]);
|
|
}
|
|
|
|
TEST_CASE("FrameTransformation_Converted_bshuf_zstd" ,"") {
|
|
const uint16_t nmodules = 4;
|
|
const uint16_t ndatastreams = 2;
|
|
DiffractionExperiment experiment(DetectorGeometry(ndatastreams * nmodules, 2));
|
|
experiment.DataStreams(ndatastreams);
|
|
|
|
experiment.Mode(DetectorMode::Conversion).Compression(JFJochProtoBuf::BSHUF_ZSTD);
|
|
|
|
FrameTransformation transformation(experiment);
|
|
|
|
std::mt19937 g1(1987);
|
|
std::uniform_int_distribution<int16_t> dist;
|
|
|
|
std::vector<int16_t> input_0(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_0[i] = dist(g1);
|
|
|
|
std::vector<int16_t> input_1(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_1[i] = dist(g1);
|
|
|
|
std::vector<char> output_compressed(experiment.GetMaxCompressedSize());
|
|
|
|
for (int i = 0; i < nmodules; i++) {
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_0.data() + i * RAW_MODULE_SIZE, i, 0));
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_1.data() + i * RAW_MODULE_SIZE, i, 1));
|
|
}
|
|
|
|
size_t compressed_size;
|
|
REQUIRE_NOTHROW(compressed_size = transformation.SaveCompressedImage(output_compressed.data()));
|
|
|
|
REQUIRE(bshuf_read_uint64_BE(output_compressed.data()) == experiment.GetPixelsNum() * experiment.GetPixelDepth());
|
|
REQUIRE(bshuf_read_uint32_BE(output_compressed.data()+8) == JFJochBitShuffleCompressor::DefaultBlockSize * experiment.GetPixelDepth());
|
|
|
|
output_compressed.resize(compressed_size);
|
|
std::vector<int16_t> output;
|
|
REQUIRE_NOTHROW(JFJochDecompress(output, experiment.GetCompressionAlgorithmEnum(), output_compressed,
|
|
experiment.GetPixelsNum()));
|
|
|
|
REQUIRE(input_0[511*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 0]);
|
|
REQUIRE(input_0[511*1024+256]/2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 258]);
|
|
REQUIRE(input_0[256*1024+256]/4 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 2 * 1030 * 255 + 257]);
|
|
|
|
REQUIRE(input_0[311*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_0[311*1024+256] / 2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 258]);
|
|
|
|
REQUIRE(input_0[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030]);
|
|
REQUIRE(input_0[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030 + 800 + 6]);
|
|
|
|
REQUIRE(input_1[511*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 0]);
|
|
REQUIRE(input_1[511*1024+256] /2 == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 258]);
|
|
|
|
REQUIRE(input_1[(311+2*512)*1024] == output[200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_1[(311+2*512)*1024+512] / 2 == output[200 * 1030 * 2 + 256*2+3]);
|
|
|
|
REQUIRE(input_1[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030]);
|
|
REQUIRE(input_1[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030 + 800 + 6]);
|
|
}
|
|
|
|
TEST_CASE("FrameTransformation_Converted_bshuf_zstd_rle" ,"") {
|
|
const uint16_t nmodules = 4;
|
|
const uint16_t ndatastreams = 2;
|
|
DiffractionExperiment experiment(DetectorGeometry(ndatastreams * nmodules, 2));
|
|
experiment.DataStreams(ndatastreams);
|
|
|
|
experiment.Mode(DetectorMode::Conversion).Compression(JFJochProtoBuf::BSHUF_ZSTD_RLE);
|
|
|
|
FrameTransformation transformation(experiment);
|
|
|
|
std::mt19937 g1(1987);
|
|
std::uniform_int_distribution<int16_t> dist;
|
|
|
|
std::vector<int16_t> input_0(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_0[i] = dist(g1);
|
|
|
|
std::vector<int16_t> input_1(nmodules*RAW_MODULE_SIZE);
|
|
for (int i = 0; i < nmodules*RAW_MODULE_SIZE; i++)
|
|
input_1[i] = dist(g1);
|
|
|
|
std::vector<char> output_compressed(experiment.GetMaxCompressedSize());
|
|
|
|
for (int i = 0; i < nmodules; i++) {
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_0.data() + i * RAW_MODULE_SIZE, i, 0));
|
|
REQUIRE_NOTHROW(transformation.ProcessModule(input_1.data() + i * RAW_MODULE_SIZE, i, 1));
|
|
}
|
|
|
|
size_t compressed_size;
|
|
REQUIRE_NOTHROW(compressed_size = transformation.SaveCompressedImage(output_compressed.data()));
|
|
|
|
REQUIRE(bshuf_read_uint64_BE(output_compressed.data()) == experiment.GetPixelsNum() * experiment.GetPixelDepth());
|
|
REQUIRE(bshuf_read_uint32_BE(output_compressed.data()+8) == JFJochBitShuffleCompressor::DefaultBlockSize * experiment.GetPixelDepth());
|
|
|
|
output_compressed.resize(compressed_size);
|
|
std::vector<int16_t> output;
|
|
REQUIRE_NOTHROW(JFJochDecompress(output, experiment.GetCompressionAlgorithmEnum(), output_compressed,
|
|
experiment.GetPixelsNum()));
|
|
|
|
REQUIRE(input_0[511*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 0]);
|
|
REQUIRE(input_0[511*1024+256]/2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 258]);
|
|
REQUIRE(input_0[256*1024+256]/4 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 2 * 1030 * 255 + 257]);
|
|
|
|
REQUIRE(input_0[311*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_0[311*1024+256] / 2 == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 200 * 1030 * 2 + 258]);
|
|
|
|
REQUIRE(input_0[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030]);
|
|
REQUIRE(input_0[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (2 * nmodules - 2) + 1030 + 800 + 6]);
|
|
|
|
REQUIRE(input_1[511*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 0]);
|
|
REQUIRE(input_1[511*1024+256] /2 == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 258]);
|
|
|
|
REQUIRE(input_1[(311+2*512)*1024] == output[200 * 1030 * 2 + 0]);
|
|
REQUIRE(input_1[(311+2*512)*1024+512] / 2 == output[200 * 1030 * 2 + 256*2+3]);
|
|
|
|
REQUIRE(input_1[(511+512)*1024] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030]);
|
|
REQUIRE(input_1[(511+512)*1024 + 800] == output[CONVERTED_MODULE_SIZE * (nmodules - 2) + 1030 + 800 + 6]);
|
|
}
|