blob_spot_detector.h

Go to the documentation of this file.

/*
 * $Id: blob_spot_detector.h 17797 2014-10-21 04:41:57Z predoehl $
 */

#ifndef BLOB_SPOT_DETECTOR_H_INCLUDED
#define BLOB_SPOT_DETECTOR_H_INCLUDED

#include <m_cpp/m_matrix.h>
#include <m_cpp/m_vector.h>
#include <i_cpp/i_image.h>
#include <vector>

namespace kjb
{

class Spot_detector
{
public:
    typedef std::vector<Vector> Centroid_set;
    typedef std::vector<std::pair<int, int> > Pair_vector;

private:
    Matrix m_background, m_thresholds;
    int m_min_brightness, m_min_size, m_max_size;
    double m_similarity;
    mutable Centroid_set spot_centroids;
    mutable std::vector<Pair_vector> spot_coordinates;

    void swap(Spot_detector& sd)
    {
        using std::swap;

        m_background.swap(sd.m_background);
        m_thresholds.swap(sd.m_thresholds);
        swap(m_min_brightness, sd.m_min_brightness);
        swap(m_min_size, sd.m_min_size);
        swap(m_max_size, sd.m_max_size);
        swap(m_similarity, sd.m_similarity);
        spot_centroids.swap(sd.spot_centroids);
        spot_coordinates.swap(sd.spot_coordinates);
    }

public:
    Spot_detector
    (
        const Matrix& background,
        const Matrix& thresholds,
        int min_brightness,
        int min_size,
        int max_size,
        double similarity
    ) :
        m_background(background),
        m_thresholds(thresholds),
        m_min_brightness(min_brightness),
        m_min_size(min_size),
        m_max_size(max_size),
        m_similarity(similarity)
    {}

    Spot_detector
    (
        int num_rows,
        int num_cols,
        double background,
        double  threshold,
        int min_brightness,
        int min_size,
        int max_size,
        double similarity
    )
    {
        Spot_detector sd(
                Matrix(num_rows, num_cols, background),
                Matrix(num_rows, num_cols, threshold),
                min_brightness,
                min_size,
                max_size,
                similarity
            );
        swap(sd);
    }

    const Centroid_set& operator()(const Image&) const;

    const std::vector<Pair_vector>& get_spot_coordinates() const
    {
        return spot_coordinates;
    }

    const Centroid_set& get_spot_centroids() const
    {
        return spot_centroids;
    }
};

} //namespace kjb

#endif /* BLOB_SPOT_DETECTOR_H_INCLUDED */