FunctionalImage.h

Go to the documentation of this file.

/*
 * @file SEFramework/Image/FunctionalImage.h
 * @date 27/03/19
 * @author Alejandro Alvarez Ayllon
 */
 
#ifndef _SEFRAMEWORK_IMAGE_FUNCTIONALIMAGE_H
#define _SEFRAMEWORK_IMAGE_FUNCTIONALIMAGE_H
 
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ImageChunk.h"
 
namespace SourceXtractor {

template<typename T, typename I = T>
class FunctionalImage : public Image<T> {
public:
  using FunctorType = std::function<T(int x, int y, I v)>;
 
protected:
  FunctionalImage(std::shared_ptr<const Image<I>> img, FunctorType functor)
    : m_img{std::move(img)}, m_functor{functor} {
  }
 
public:
  virtual ~FunctionalImage() = default;
 
  template<typename ...Args>
  static std::shared_ptr<FunctionalImage<T, I>> create(Args&& ... args) {
    return std::shared_ptr<FunctionalImage<T, I>>(new FunctionalImage{std::forward<Args>(args)...});
  }
 
  std::string getRepr() const final {
    return "FunctionalImage<" + std::string(m_functor.target_type().name()) + ">";
  }
 
  int getWidth() const final {
    return m_img->getWidth();
  }
 
  int getHeight() const final {
    return m_img->getHeight();
  }
 
  std::shared_ptr<ImageChunk<T>> getChunk(int x, int y, int width, int height) const final {
    auto in_chunk = m_img->getChunk(x, y, width, height);
    auto chunk = UniversalImageChunk<T>::create(width, height);
    for (int iy = 0; iy < height; ++iy) {
      for (int ix = 0; ix < width; ++ix) {
        chunk->at(ix, iy) = m_functor(ix + x, iy + y, in_chunk->getValue(ix, iy));
      }
    }
    return chunk;
  }
 
private:
  std::shared_ptr<const Image<I>> m_img;
  FunctorType m_functor;
};
 
} // end SourceXtractor
 
#endif // _SEFRAMEWORK_IMAGE_FUNCTIONALIMAGE_H