33 lines
776 B
C++
33 lines
776 B
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include "LossyFilter.h"
|
|
|
|
LossyFilter::LossyFilter(float in_p)
|
|
: p(in_p) {}
|
|
|
|
|
|
bool LossyFilter::RollDice() {
|
|
std::unique_lock ul(random_m);
|
|
|
|
if (distr(mt) < p)
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
bool LossyFilter::ApplyFilter(DataMessage &message) {
|
|
if (p == 1.0)
|
|
return true;
|
|
else {
|
|
if (message.indexing_result.value_or(false) || ((p > 0.0) && RollDice())) {
|
|
message.number = image_number++;
|
|
return true;
|
|
} else
|
|
return false;
|
|
}
|
|
}
|
|
|
|
LossyFilter::LossyFilter(const DiffractionExperiment &x)
|
|
: p (x.GetLossyCompressionSerialMX()){}
|