ThresholdedImage.h

Go to the documentation of this file.

/*
 * ThresholdedImage.h
 *
 *  Created on: Jan 10, 2018
 *      Author: mschefer
 */
 
#ifndef _SEFRAMEWORK_IMAGE_THRESHOLDEDIMAGE_H_
#define _SEFRAMEWORK_IMAGE_THRESHOLDEDIMAGE_H_
 
#include <memory>
 
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ImageChunk.h"
#include "SEFramework/Image/ConstantImage.h"
 
namespace SourceXtractor {

template <typename T>
class ThresholdedImage : public Image<T> {
 
protected:
 
  ThresholdedImage(std::shared_ptr<const Image<T>> image, std::shared_ptr<const Image<T>> variance_map, T threshold_multiplier)
      : m_image(image), m_variance_map(variance_map), m_threshold_multiplier(threshold_multiplier) {
    assert(m_image->getWidth() == m_variance_map->getWidth());
    assert(m_image->getHeight() == m_variance_map->getHeight());
  };
 
public:

  virtual ~ThresholdedImage() = default;
 
  static std::shared_ptr<ThresholdedImage<T>> create(
      std::shared_ptr<const Image<T>> image, std::shared_ptr<const Image<T>> variance_map, T threshold_multiplier) {
    return std::shared_ptr<ThresholdedImage<T>>(new ThresholdedImage<T>(image, variance_map, threshold_multiplier));
  }
 
  std::string getRepr() const override {
    return "ThresholdedImage(" + m_image->getRepr() + ")";
  }
 
  int getWidth() const override {
    return m_image->getWidth();
  }
 
  int getHeight() const override {
    return m_image->getHeight();
  }
 
  std::shared_ptr<ImageChunk<T>> getChunk(int x, int y, int width, int height) const override{
    auto img_chunk = m_image->getChunk(x, y, width, height);
    auto var_chunk = m_variance_map->getChunk(x, y, width, height);
    auto chunk = UniversalImageChunk<T>::create(std::move(*img_chunk));
    for (int iy = 0; iy < height; ++iy) {
      for (int ix = 0; ix < width; ++ix) {
        chunk->at(ix, iy) -= sqrt(var_chunk->getValue(ix, iy)) * m_threshold_multiplier;
      }
    }
    return chunk;
  }
 
private:
  std::shared_ptr<const Image<T>> m_image, m_variance_map;
  T m_threshold_multiplier;
 
}; /* End of ThresholdedImage class */
 
} /* namespace SourceXtractor */
 
#endif /* _SEFRAMEWORK_IMAGE_THRESHOLDEDIMAGE_H_ */