// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#pragma once

#include <cstdint>
#include <vector>

#include "../../common/DiffractionSpot.h"

#include "../image_preprocessing/ImagePreprocessorBuffer.h"

class ImageSpotFinder {
protected:
     const int32_t width, height;
     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 ImagePreprocessorBuffer &image, const SpotFindingSettings &settings, const std::vector<bool> &res_mask);
public:
     constexpr static int32_t MIN_VALID_PIXELS = 100;
     constexpr static int NBX = 15;

     virtual ~ImageSpotFinder() = default;
     virtual std::vector<DiffractionSpot> Run(const ImagePreprocessorBuffer &image, const SpotFindingSettings &settings, const std::vector<bool> &res_mask) = 0;
};