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();
        PTR(ImageT) returnMean() {return _mean;}
    private:
        std::shared_ptr<KernelPca<ImageT> > _imagePca;  
        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