SubImage.h

Go to the documentation of this file.

#ifndef _SEFRAMEWORK_IMAGE_SUBIMAGE_H_
#define _SEFRAMEWORK_IMAGE_SUBIMAGE_H_

#include <memory>

#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ImageAccessor.h"

namespace SourceXtractor {

template<typename T>
class SubImage : public Image<T> {
protected:
  SubImage(std::shared_ptr<const Image<T>> image, const PixelCoordinate &offset, int width, int height)
    : m_image(image), m_offset(offset), m_width(width), m_height(height) {
    assert(offset.m_x >= 0 && offset.m_y >= 0 && width > 0 && height > 0 &&
        offset.m_x + width <= image->getWidth() && offset.m_y + height <= image->getHeight());
  }

  SubImage(std::shared_ptr<const Image<T>> image, int x, int y, int width, int height)
    : m_image(image), m_offset(x, y), m_width(width), m_height(height) {
    assert(x >= 0 && y >= 0 && width > 0 && height > 0 &&
        x + width <= image->getWidth() && y + height <= image->getHeight());
  }

public:
  virtual ~SubImage() = default;

  template<typename... Args>
  static std::shared_ptr<SubImage<T>> create(Args &&... args) {
    return std::shared_ptr<SubImage<T>>(new SubImage{std::forward<Args>(args)...});
  }

  std::string getRepr() const override {
    return "SubImage(" + m_image->getRepr() + ", " + std::to_string(m_offset.m_x) + ", " + std::to_string(m_offset.m_y) + ", " + std::to_string(m_width) + ", " + std::to_string(m_height) + ")";
  }

  int getWidth() const override {
    return m_width;
  }

  int getHeight() const override {
    return m_height;
  }

  std::shared_ptr<ImageChunk<T>> getChunk(int x, int y, int width, int height) const override {
    return m_image->getChunk(x + m_offset.m_x, y + m_offset.m_y, width, height);
  }

private:
  std::shared_ptr<const Image<T>> m_image;
  PixelCoordinate m_offset;
  int m_width, m_height;
};

} /* namespace SourceXtractor */

#endif /* _SEFRAMEWORK_IMAGE_SUBIMAGE_H_ */