KernelPca.h

Go to the documentation of this file.

// -*- lsst-c++ -*-
#ifndef LSST_IP_DIFFIM_KERNELPCA_H
#define LSST_IP_DIFFIM_KERNELPCA_H
 
#include "lsst/afw/image.h"
#include "lsst/afw/math.h"
 
namespace lsst {
namespace ip {
namespace diffim {
namespace detail {
 
    template <typename ImageT>
    class KernelPca : public lsst::afw::image::ImagePca<ImageT> {
        typedef typename lsst::afw::image::ImagePca<ImageT> Super; 
    public:
        typedef typename std::shared_ptr<KernelPca<ImageT> > Ptr;
        using lsst::afw::image::ImagePca<ImageT>::getEigenImages;
        using lsst::afw::image::ImagePca<ImageT>::getEigenValues;
        using lsst::afw::image::ImagePca<ImageT>::addImage;
 
        explicit KernelPca(bool constantWeight=true) : Super(constantWeight) {}
 
        virtual void analyze();
    };
 
    template<typename PixelT>
    class KernelPcaVisitor : public lsst::afw::math::CandidateVisitor {
    public:
        typedef lsst::afw::image::Image<lsst::afw::math::Kernel::Pixel> ImageT;
        typedef std::shared_ptr<KernelPcaVisitor<PixelT> > Ptr;
 
        KernelPcaVisitor(std::shared_ptr<KernelPca<ImageT> > imagePca);
        virtual ~KernelPcaVisitor() {};
 
        lsst::afw::math::KernelList getEigenKernels();
        void processCandidate(lsst::afw::math::SpatialCellCandidate *candidate);
        void subtractMean();
        std::shared_ptr<ImageT> returnMean() {return _mean;}
    private:
        std::shared_ptr<KernelPca<ImageT> > _imagePca;  
        std::shared_ptr<ImageT> _mean;                                
    };
 
    template<typename PixelT>
    std::shared_ptr<KernelPcaVisitor<PixelT> >
    makeKernelPcaVisitor(std::shared_ptr<KernelPca<typename KernelPcaVisitor<PixelT>::ImageT> > imagePca) {
        return std::shared_ptr<KernelPcaVisitor<PixelT>>(new KernelPcaVisitor<PixelT>(imagePca));
    };
 
}}}} // end of namespace lsst::ip::diffim::detail
 
#endif