68 lines
2.5 KiB
C++
68 lines
2.5 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#include <cstring>
|
|
|
|
#include <bitshuffle/bitshuffle.h>
|
|
|
|
#include "FrameTransformation.h"
|
|
#include "RawToConvertedGeometry.h"
|
|
#include "JFJochException.h"
|
|
|
|
FrameTransformation::FrameTransformation(const DiffractionExperiment &in_experiment) :
|
|
experiment(in_experiment),
|
|
pixel_depth(experiment.GetPixelDepth()),
|
|
compressor(in_experiment.GetCompressionAlgorithmEnum()) {
|
|
|
|
precompression_buffer.resize(experiment.GetPixelsNum() * pixel_depth);
|
|
|
|
if (experiment.GetApplyPixelMaskInFPGA()) {
|
|
// Mask gaps
|
|
if (pixel_depth == 2) {
|
|
auto ptr = (int16_t *) precompression_buffer.data();
|
|
for (int i = 0; i < experiment.GetPixelsNum(); i++)
|
|
ptr[i] = INT16_MIN;
|
|
} else {
|
|
auto ptr = (uint32_t *) precompression_buffer.data();
|
|
for (int i = 0; i < experiment.GetPixelsNum(); i++)
|
|
ptr[i] = INT32_MIN;
|
|
}
|
|
}
|
|
}
|
|
|
|
size_t FrameTransformation::SaveCompressedImage(void *output) {
|
|
return compressor.Compress((char *) output, precompression_buffer.data(),
|
|
experiment.GetPixelsNum(), pixel_depth);
|
|
}
|
|
|
|
void FrameTransformation::ProcessModule(const int16_t *input, uint16_t module_number, int data_stream) {
|
|
|
|
size_t module_number_abs = experiment.GetFirstModuleOfDataStream(data_stream) + module_number;
|
|
|
|
auto output = (int16_t *) precompression_buffer.data();
|
|
|
|
if (experiment.GetDetectorMode() != DetectorMode::Conversion)
|
|
memcpy(output + RAW_MODULE_SIZE * module_number_abs,
|
|
input,
|
|
RAW_MODULE_SIZE * experiment.GetPixelDepth());
|
|
else
|
|
TransferModuleAdjustMultipixels(output, (int16_t *) input,
|
|
experiment.GetModuleSlowDirectionStep(module_number_abs),
|
|
static_cast<int16_t>(INT16_MIN),
|
|
static_cast<int16_t>(INT16_MAX),
|
|
experiment.GetModuleFastDirectionStep(module_number_abs),
|
|
experiment.GetPixel0OfModule(module_number_abs));
|
|
}
|
|
|
|
void FrameTransformation::ApplyROI(const ROIFilter &filter) {
|
|
if (pixel_depth == 2)
|
|
filter.Apply((int16_t *) precompression_buffer.data(), static_cast<int16_t>(INT16_MIN));
|
|
else
|
|
filter.Apply((int32_t *) precompression_buffer.data(), static_cast<int32_t>(INT32_MIN));
|
|
}
|
|
|
|
const void *FrameTransformation::GetPreviewImage() const {
|
|
return precompression_buffer.data();
|
|
}
|
|
|
|
|