LSSTApplications  10.0-2-g4f67435,11.0.rc2+1,11.0.rc2+12,11.0.rc2+3,11.0.rc2+4,11.0.rc2+5,11.0.rc2+6,11.0.rc2+7,11.0.rc2+8
LSSTDataManagementBasePackage
testSpatialCell.h
/*
* LSST Data Management System
* Copyright 2008, 2009, 2010 LSST Corporation.
*
* This product includes software developed by the
* LSST Project (http://www.lsst.org/).
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the LSST License Statement and
* the GNU General Public License along with this program. If not,
* see <http://www.lsstcorp.org/LegalNotices/>.
*/
#if !defined(TESTSPATIALCELL_H)
#define TESTSPATIALCELL_H
#include "boost/shared_ptr.hpp"
#include "lsst/afw/math.h"
#include "lsst/afw/geom.h"
/************************************************************************************************************/
/*
* Test class for SpatialCellImageCandidate
*/
class ExampleCandidate : public lsst::afw::math::SpatialCellMaskedImageCandidate<float> {
public:
typedef boost::shared_ptr<ExampleCandidate> Ptr;
typedef float PixelT;
ExampleCandidate(float const xCenter, float const yCenter,
MaskedImageT::ConstPtr parent, lsst::afw::geom::Box2I bbox);
lsst::afw::geom::Box2I getBBox() const { return _bbox; }
double getCandidateRating() const;
MaskedImageT::ConstPtr getMaskedImage() const;
private:
ExampleCandidate::MaskedImageT::ConstPtr _parent;
};
/************************************************************************************************************/
/*
* Class to pass around to all our ExampleCandidates. All this one does is count acceptable candidates
*/
class ExampleCandidateVisitor : public lsst::afw::math::CandidateVisitor {
public:
ExampleCandidateVisitor() : lsst::afw::math::CandidateVisitor(), _n(0), _npix(0) {}
// Called by SpatialCellSet::visitCandidates before visiting any Candidates
void reset() { _n = _npix = 0; }
// Called by SpatialCellSet::visitCandidates for each Candidate
++_n;
lsst::afw::geom::Box2I box = dynamic_cast<ExampleCandidate *>(candidate)->getBBox();
_npix += box.getArea();
}
int getN() const { return _n; }
int getNPix() const { return _npix; }
private:
int _n; // number of ExampleCandidates
int _npix; // number of pixels in ExampleCandidates's bounding boxes
};
#endif