34 lines
1.0 KiB
C++
34 lines
1.0 KiB
C++
// SPDX-FileCopyrightText: 2025 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 <cstdint>
|
|
#include <vector>
|
|
|
|
#include "../../common/DiffractionSpot.h"
|
|
|
|
class ImageSpotFinder {
|
|
protected:
|
|
const int32_t width, height;
|
|
std::vector<int32_t> input_buffer;
|
|
std::vector<uint32_t> output_buffer;
|
|
|
|
ImageSpotFinder(int32_t width, int32_t height);
|
|
size_t OutputSize() const;
|
|
size_t OutputByteSize() const;
|
|
std::vector<DiffractionSpot> ExtractSpots(const SpotFindingSettings &settings, const std::vector<bool> &res_mask);
|
|
public:
|
|
constexpr static int32_t MIN_VALID_PIXELS = 100;
|
|
constexpr static int NBX = 15;
|
|
|
|
std::vector<int32_t> &GetInputBuffer();
|
|
|
|
virtual ~ImageSpotFinder() = default;
|
|
virtual std::vector<DiffractionSpot> Run(const SpotFindingSettings &settings, const std::vector<bool> &res_mask) = 0;
|
|
};
|
|
|
|
|
|
#endif //JFJOCH_IMAGESPOTFINDER_H
|