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, bool normalize_psf = true);
 
  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;
  bool m_normalize_psf { true };
};
 
} // 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_ */