lsst.ip.diffim  16.0-9-g1de645c+6
BuildSpatialKernelVisitor.cc
Go to the documentation of this file.
1 // -*- lsst-c++ -*-
12 #include <memory>
13 #include "boost/timer.hpp"
14 
15 #include "Eigen/Core"
16 #include "Eigen/Cholesky"
17 #include "Eigen/LU"
18 #include "Eigen/QR"
19 
20 #include "lsst/afw/math.h"
21 #include "lsst/afw/geom.h"
22 #include "lsst/log/Log.h"
23 #include "lsst/pex/policy/Policy.h"
25 
29 
30 namespace afwMath = lsst::afw::math;
31 namespace afwGeom = lsst::afw::geom;
32 namespace pexPolicy = lsst::pex::policy;
33 namespace pexExcept = lsst::pex::exceptions;
34 
35 namespace lsst {
36 namespace ip {
37 namespace diffim {
38 namespace detail {
72  template<typename PixelT>
74  lsst::afw::math::KernelList const& basisList,
75  lsst::afw::geom::Box2I const& regionBBox,
77  ) :
78  afwMath::CandidateVisitor(),
79  _kernelSolution(),
80  _nCandidates(0)
81  {
82  int spatialKernelOrder = policy.getInt("spatialKernelOrder");
83  afwMath::Kernel::SpatialFunctionPtr spatialKernelFunction;
84 
85  int fitForBackground = policy.getBool("fitForBackground");
86  int spatialBgOrder = fitForBackground ? policy.getInt("spatialBgOrder") : 0;
88 
89  std::string spatialModelType = policy.getString("spatialModelType");
90  if (spatialModelType == "chebyshev1") {
91  spatialKernelFunction = afwMath::Kernel::SpatialFunctionPtr(
92  new afwMath::Chebyshev1Function2<double>(spatialKernelOrder, afwGeom::Box2D(regionBBox))
93  );
95  new afwMath::Chebyshev1Function2<double>(spatialBgOrder, afwGeom::Box2D(regionBBox))
96  );
97 
98  }
99  else if (spatialModelType == "polynomial") {
100  spatialKernelFunction = afwMath::Kernel::SpatialFunctionPtr(
101  new afwMath::PolynomialFunction2<double>(spatialKernelOrder)
102  );
104  new afwMath::PolynomialFunction2<double>(spatialBgOrder)
105  );
106  }
107  else {
109  str(boost::format("Invalid type (%s) for spatial models") %
110  spatialModelType));
111  }
112 
113  /* */
114 
115  _kernelSolution = std::shared_ptr<SpatialKernelSolution>(
116  new SpatialKernelSolution(basisList, spatialKernelFunction, background, policy));
117  };
118 
119 
120  template<typename PixelT>
123  ) {
124  KernelCandidate<PixelT> *kCandidate = dynamic_cast<KernelCandidate<PixelT> *>(candidate);
125  if (kCandidate == NULL) {
127  "Failed to cast SpatialCellCandidate to KernelCandidate");
128  }
129  if (!(kCandidate->isInitialized())) {
131  LOGL_DEBUG("TRACE2.ip.diffim.BuildSpatialKernelVisitor.processCandidate",
132  "Cannot process candidate %d, continuing", kCandidate->getId());
133  return;
134  }
135 
136  LOGL_DEBUG("TRACE5.ip.diffim.BuildSpatialKernelVisitor.processCandidate",
137  "Processing candidate %d", kCandidate->getId());
138  _nCandidates += 1;
139 
140  /*
141  Build the spatial kernel from the most recent fit, e.g. if its Pca
142  you want to build a spatial model on the Pca basis, not original
143  basis
144  */
145  _kernelSolution->addConstraint(kCandidate->getXCenter(),
146  kCandidate->getYCenter(),
147  kCandidate->getKernelSolution(
149  )->getM(),
150  kCandidate->getKernelSolution(
152  )->getB());
153 
154  }
155 
156  template<typename PixelT>
158  _kernelSolution->solve();
159  }
160 
161  template<typename PixelT>
164  return _kernelSolution->getSolutionPair();
165  }
166 
167  typedef float PixelT;
168  template class BuildSpatialKernelVisitor<PixelT>;
169 
170 }}}} // end of namespace lsst::ip::diffim::detail
171 
Creates a spatial kernel and background from a list of candidates.
Class stored in SpatialCells for spatial Kernel fitting.
std::shared_ptr< lsst::afw::math::Function2< double > > SpatialFunctionPtr
std::shared_ptr< StaticKernelSolution< PixelT > > getKernelSolution(CandidateSwitch cand) const
Declaration of BuildSpatialKernelVisitor.
Declaration of classes to store the solution for convolution kernels.
STL class.
BuildSpatialKernelVisitor(lsst::afw::math::KernelList const &basisList, lsst::afw::geom::Box2I const &regionBBox, lsst::pex::policy::Policy policy)
bool getBool(const std::string &name) const
std::pair< std::shared_ptr< lsst::afw::math::LinearCombinationKernel >, lsst::afw::math::Kernel::SpatialFunctionPtr > getSolutionPair()
int getInt(const std::string &name) const
const std::string getString(const std::string &name) const
Class used by SpatialModelCell for spatial Kernel fitting.
#define LSST_EXCEPT(type,...)
STL class.
#define LOGL_DEBUG(logger, message...)
void processCandidate(lsst::afw::math::SpatialCellCandidate *candidate)