mx/Jungfraujoch
2
1
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 54 Wiki Activity

DiffractionExperiment: Remove ROI-mask function

Browse Source
This commit is contained in:
leonarski_f
2023-11-06 16:51:34 +01:00
parent b2743072e6
commit d6c1b19599
14 changed files with 168 additions and 606 deletions
Download Patch File Download Diff File Expand all files Collapse all files
common/FrameTransformation.cpp
+52 -31
View File
@@ -8,9 +8,15 @@
#include "RawToConvertedGeometry.h"
#include "JFJochException.h"
template <class T> void FillVector(std::vector<char> &v, T fill_value) {
auto ptr = (T *) v.data();
for (int i = 0; i < v.size() / sizeof(T); i++)
ptr[i] = fill_value;
}
FrameTransformation::FrameTransformation(const DiffractionExperiment &in_experiment) :
experiment(in_experiment),
pixel_depth(experiment.GetPixelDepth()),
pixel_depth(experiment.GetPixelDepth()), pixel_signed(experiment.IsPixelSigned()),
compressor(in_experiment.GetCompressionAlgorithmEnum()) {
precompression_buffer.resize(experiment.GetPixelsNum() * pixel_depth);
@@ -18,13 +24,15 @@ FrameTransformation::FrameTransformation(const DiffractionExperiment &in_experim
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;
if (pixel_signed)
FillVector<int16_t>(precompression_buffer, INT16_MIN);
else
FillVector<uint16_t>(precompression_buffer, UINT16_MAX);
} else if (pixel_depth == 4) {
if (pixel_signed)
FillVector<int32_t>(precompression_buffer, INT32_MIN);
else
FillVector<uint32_t>(precompression_buffer, UINT32_MAX);
}
}
}
@@ -35,33 +43,46 @@ size_t FrameTransformation::SaveCompressedImage(void *output) {
}
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));
if (experiment.GetDetectorMode() != DetectorMode::Conversion) {
size_t mod_size = RAW_MODULE_SIZE * pixel_depth;
memcpy(precompression_buffer.data() + module_number_abs * mod_size, input, mod_size);
} else {
if (pixel_depth == 2) {
if (pixel_signed)
TransferModuleAdjustMultipixels((int16_t *) precompression_buffer.data(), (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));
else
TransferModuleAdjustMultipixels((uint16_t *) precompression_buffer.data(), (uint16_t *) input,
experiment.GetModuleSlowDirectionStep(module_number_abs),
static_cast<uint16_t>(0),
static_cast<uint16_t>(UINT16_MAX - 1),
experiment.GetModuleFastDirectionStep(module_number_abs),
experiment.GetPixel0OfModule(module_number_abs));
} else if (pixel_depth == 4) {
if (pixel_signed)
TransferModuleAdjustMultipixels((int32_t *) precompression_buffer.data(), (int32_t *) input,
experiment.GetModuleSlowDirectionStep(module_number_abs),
static_cast<int32_t>(INT32_MIN),
static_cast<int32_t>(INT32_MAX),
experiment.GetModuleFastDirectionStep(module_number_abs),
experiment.GetPixel0OfModule(module_number_abs));
else
TransferModuleAdjustMultipixels((uint32_t *) precompression_buffer.data(), (uint32_t *) input,
experiment.GetModuleSlowDirectionStep(module_number_abs),
static_cast<uint32_t>(0),
static_cast<uint32_t>(UINT32_MAX - 1),
experiment.GetModuleFastDirectionStep(module_number_abs),
experiment.GetPixel0OfModule(module_number_abs));
}
}
}
const void *FrameTransformation::GetPreviewImage() const {
return precompression_buffer.data();
}