AzIntEngine: Fixes
This commit is contained in:
@@ -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()){}
|
||||
|
||||
@@ -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);
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
};
|
||||
Reference in New Issue
Block a user