RecenterImage.h

Go to the documentation of this file.

/*
 * @file SEFramework/Image/WarpImage.h
 * @date 11/09/18
 * @author Alejandro Alvarez Ayllon
 */
 
#ifndef _SEFRAMEWORK_IMAGE_WARPIMAGE_H
#define _SEFRAMEWORK_IMAGE_WARPIMAGE_H
 
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ImageAccessor.h"
 
namespace SourceXtractor {

template<typename T>
class RecenterImage : public Image<T> {
protected:
  RecenterImage(std::shared_ptr<const Image<T>> img, const PixelCoordinate &new_center) : m_img{img},
                                                                                          m_center{new_center} {
  }
 
public:
  template<typename... Args>
  static std::shared_ptr<RecenterImage<T>> create(Args &&... args) {
    return std::shared_ptr<RecenterImage<T>>(new RecenterImage{std::forward<Args>(args)...});
  }
 
  std::string getRepr() const override {
    return "RecenterImage(" + m_img->getRepr() + ")";
  }
 
  int getWidth() const override {
    return m_img->getWidth();
  }
 
  int getHeight() const override {
    return m_img->getHeight();
  }
 
  std::shared_ptr<ImageChunk<T>> getChunk(int x, int y, int width, int height) const override{
    ImageAccessor<T> accessor(m_img);
    auto chunk = UniversalImageChunk<T>::create(width, height);
    auto img_w = accessor.getWidth();
    auto img_h = accessor.getHeight();
    for (int iy = 0; iy < height; ++iy) {
      for (int ix = 0; ix < width; ++ix) {
        chunk->at(ix, iy) = accessor.getValue(
          (ix + x + m_center.m_x) % img_w,
          (iy + y + m_center.m_y) % img_h
        );
      }
    }
    return chunk;
  }
 
private:
  std::shared_ptr<const Image<T>> m_img;
  PixelCoordinate m_center;
};
 
} // end SourceXtractor
 
#endif // _SEFRAMEWORK_IMAGE_WARPIMAGE_H