DownSampledImagePsf.h

Go to the documentation of this file.

/*
 * DownSampledImagePsf.h
 *
 *  Created on: Nov 5, 2021
 *      Author: mschefer
 */

#ifndef _SEIMPLEMENTATION_IMAGE_DOWNSAMPLEDIMAGEPSF_H_
#define _SEIMPLEMENTATION_IMAGE_DOWNSAMPLEDIMAGEPSF_H_

#include "ElementsKernel/Exception.h"
#include "AlexandriaKernel/memory_tools.h"
#include "ModelFitting/Image/PsfTraits.h"
#include "SEFramework/Image/VectorImage.h"
#include "SEFramework/Image/ProcessedImage.h"
#include "SEFramework/Convolution/Convolution.h"

#include "SEImplementation/Image/ImagePsf.h"

namespace SourceXtractor {

/*
 * Encapsulates ImagePsf to provide a downscaled version for performance
 */
class DownSampledImagePsf {
public:

  DownSampledImagePsf(double pixel_scale, std::shared_ptr<VectorImage<SeFloat>> image, double down_scaling=1.0);

  virtual ~DownSampledImagePsf() = default;

  double getPixelScale() const;
  std::size_t getSize() const;
  std::shared_ptr<VectorImage<SourceXtractor::SeFloat>> getScaledKernel(SeFloat scale) const;
  void convolve(std::shared_ptr<WriteableImage<float>> image) const;

  // For ConvolutionContext optimization
  std::unique_ptr<DFTConvolution<SeFloat>::ConvolutionContext> prepare(const std::shared_ptr<const Image<SeFloat>>& model_ptr) const;
  void convolve(std::shared_ptr<WriteableImage<float>> image, std::unique_ptr<DFTConvolution<SeFloat>::ConvolutionContext>& context) const;

private:
  double m_down_scaling;
  std::shared_ptr<ImagePsf> m_psf;
};

} // end of SourceXtractor

namespace ModelFitting {

template<>
struct PsfTraits<SourceXtractor::DownSampledImagePsf> {
  using context_t = typename std::unique_ptr<SourceXtractor::ImagePsf::ConvolutionContext>;
  static constexpr bool has_context = true;
};

} // end of ModelFitting

#endif /* _SEIMPLEMENTATION_IMAGE_DOWNSAMPLEDIMAGEPSF_H_ */