AzIntEngine: Fixes

This commit is contained in:
2026-06-02 20:05:38 +02:00
parent 57fc38195e
commit 4e4b0a9b33
3 changed files with 32 additions and 25 deletions
+1
View File
@@ -7,5 +7,6 @@ AzIntEngine::AzIntEngine(const AzimuthalIntegrationMapping &integration)
: integration(integration),
npixel(integration.GetPixelToBin().size()),
azint_sum(integration.GetBinNumber(), 0.0f),
azint_sum2(integration.GetBinNumber(), 0.0f),
azint_count(integration.GetBinNumber(), 0.0f),
azint_bins(integration.GetBinNumber()){}
+1 -25
View File
@@ -7,29 +7,5 @@ AzIntEngineCPU::AzIntEngineCPU(const AzimuthalIntegrationMapping &integration)
: AzIntEngine(integration) {}
void AzIntEngineCPU::Run(const ImagePreprocessorBuffer &image, AzimuthalIntegrationProfile &profile){
for (int i = 0; i < azint_count.size(); i++) {
azint_sum[i] = 0.0f;
azint_sum2[i] = 0.0f;
azint_count[i] = 0;
}
if (image.size() != npixel)
throw std::runtime_error("ImageSpotFinder::AzimIntegration: Mismatch in size");
const uint16_t *pixel_to_bin = integration.GetPixelToBin().data();
const float *corrections = integration.Corrections().data();
for (int i = 0; i < image.size(); i++) {
const uint16_t bin = pixel_to_bin[i];
if (bin < azint_bins) {
float val = static_cast<float>(image[i]) * corrections[i];
azint_sum[bin] += val;
azint_sum2[bin] += val * val;
++azint_count[bin];
}
}
profile.Clear(integration);
profile.Add(azint_sum, azint_sum2, azint_count);
RunAzint(image, profile);
}
+30
View File
@@ -7,6 +7,36 @@
class AzIntEngineCPU : public AzIntEngine {
public:
// image is anything that can be referenced with operator[]
template <class T>
void RunAzint(const T &image, AzimuthalIntegrationProfile &profile) {
for (int i = 0; i < azint_count.size(); i++) {
azint_sum[i] = 0.0f;
azint_sum2[i] = 0.0f;
azint_count[i] = 0;
}
if (image.size() != npixel)
throw std::runtime_error("ImageSpotFinder::AzimIntegration: Mismatch in size");
const uint16_t *pixel_to_bin = integration.GetPixelToBin().data();
const float *corrections = integration.Corrections().data();
for (int i = 0; i < image.size(); i++) {
const float val = static_cast<float>(image[i]) * corrections[i];
const float val_sq = val * val;
const uint16_t bin = pixel_to_bin[i];
if (bin < azint_bins) {
azint_sum[bin] += val;
azint_sum2[bin] += val_sq;
++azint_count[bin];
}
}
profile.Clear(integration);
profile.Add(azint_sum, azint_sum2, azint_count);
}
AzIntEngineCPU(const AzimuthalIntegrationMapping& integration);
void Run(const ImagePreprocessorBuffer &image, AzimuthalIntegrationProfile &profile) override;
};