ImageMode.h

Go to the documentation of this file.

 
#ifndef SOURCEXTRACTORPLUSPLUS_IMAGEMODE_H
#define SOURCEXTRACTORPLUSPLUS_IMAGEMODE_H
 
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/VectorImage.h"
 
namespace SourceXtractor {

template<typename T>
class ImageMode {
public:

  ImageMode(const std::shared_ptr<Image<T>>& image, const std::shared_ptr<Image<T>>& variance,
            int cell_w, int cell_h,
            T invalid_value, T kappa1 = 2, T kappa2 = 5, T kappa3 = 3,
            T rtol = 1e-4, size_t max_iter = 100);

  virtual ~ImageMode() = default;

  std::shared_ptr<VectorImage<T>> getModeImage() const;

  std::shared_ptr<VectorImage<T>> getSigmaImage() const;

  std::shared_ptr<VectorImage<T>> getVarianceModeImage() const;

  std::shared_ptr<VectorImage<T>> getVarianceSigmaImage() const;
 
private:
  std::shared_ptr<const Image<T>> m_image;
  std::shared_ptr<VectorImage<T>> m_mode, m_sigma;
  std::shared_ptr<VectorImage<T>> m_var_mode, m_var_sigma;
  int m_cell_w, m_cell_h;
  T m_invalid, m_kappa1, m_kappa2, m_kappa3, m_rtol;
  size_t m_max_iter;
 
  std::tuple<T, T> getBackGuess(const std::vector<T> &data) const;
  void processCell(const Image<T>& img, int x, int y, VectorImage<T>& out_mode, VectorImage<T>& out_sigma) const;
};
 
extern template
class ImageMode<SeFloat>;
 
} // end of namespace SourceXtractor
 
#endif //SOURCEXTRACTORPLUSPLUS_IMAGEMODE_H