42 lines
1.8 KiB
C++
42 lines
1.8 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_IMAGESPOTFINDER_H
|
|
#define JFJOCH_IMAGESPOTFINDER_H
|
|
|
|
|
|
#include <cstddef>
|
|
#include <vector>
|
|
|
|
#include "SpotFindingSettings.h"
|
|
#include "../common/AzimuthalIntegration.h"
|
|
#include "../common/DiffractionSpot.h"
|
|
|
|
// This is "slow" spot finder for image-based analysis
|
|
// To complement "fast" PSI detector module based spot finder
|
|
// This one is expected to be used in cases, where images are already assembled
|
|
// and it aims for 100 ms execution
|
|
|
|
class ImageSpotFinder {
|
|
size_t width;
|
|
size_t height;
|
|
const std::vector<float> &resolution_map;
|
|
int NBX = 15;
|
|
template <class T>
|
|
std::vector<DiffractionSpot> Run(const T *input,
|
|
const T special_value,
|
|
const SpotFindingSettings &settings) const;
|
|
public:
|
|
explicit ImageSpotFinder(const AzimuthalIntegration &integration);
|
|
ImageSpotFinder(const std::vector<float> &resolution_map, size_t width, size_t height);
|
|
void nbx(int input);
|
|
std::vector<DiffractionSpot> Run(const int32_t *input, const SpotFindingSettings &settings) const;
|
|
std::vector<DiffractionSpot> Run(const int16_t *input, const SpotFindingSettings &settings) const;
|
|
std::vector<DiffractionSpot> Run(const int8_t *input, const SpotFindingSettings &settings) const;
|
|
std::vector<DiffractionSpot> Run(const uint32_t *input, const SpotFindingSettings &settings) const;
|
|
std::vector<DiffractionSpot> Run(const uint16_t *input, const SpotFindingSettings &settings) const;
|
|
std::vector<DiffractionSpot> Run(const uint8_t *input, const SpotFindingSettings &settings) const;
|
|
};
|
|
|
|
#endif //JFJOCH_IMAGESPOTFINDER_H
|