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
spatialCellExample.cc
/*
* 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/>.
*/
/*
* This C++ example does the same thing as SpatialCellExample.py. The latter of the python version
* is that you can set display == True and see what's going on
*/
#include <string>
#include "lsst/daf/base.h"
#include "lsst/afw/image.h"
#include "lsst/afw/math.h"
#include "lsst/afw/geom.h"
#include "testSpatialCell.h"
namespace afwDetect = lsst::afw::detection;
namespace afwImage = lsst::afw::image;
namespace afwMath = lsst::afw::math;
namespace afwGeom = lsst::afw::geom;
typedef float PixelT;
std::pair<afwImage::MaskedImage<PixelT>::Ptr, PTR(afwDetect::FootprintSet)> readImage();
/************************************************************************************************************/
/*
* A demonstration of the use of a SpatialCellSet
*/
void SpatialCellSetDemo() {
boost::tie(im, fs) = readImage();
/*
* Create an (empty) SpatialCellSet
*/
afwMath::SpatialCellSet cellSet(
im->getBBox(),
260, 200
);
/*
* Populate the cellSet using the detected object in the FootprintSet
*/
for (afwDetect::FootprintSet::FootprintList::iterator ptr = fs->getFootprints()->begin(),
end = fs->getFootprints()->end(); ptr != end; ++ptr) {
afwGeom::Box2I const bbox = (*ptr)->getBBox();
float const xc = (bbox.getMinX() + bbox.getMaxX())/2.0;
float const yc = (bbox.getMinY() + bbox.getMaxY())/2.0;
ExampleCandidate::Ptr tc(new ExampleCandidate(xc, yc, im, bbox));
cellSet.insertCandidate(tc);
}
/*
* OK, the SpatialCellList is populated. Let's do something with it
*/
ExampleCandidateVisitor visitor;
cellSet.visitCandidates(&visitor);
std::cout << boost::format("There are %d candidates\n") % visitor.getN();
/*
* Now label too-small object as BAD
*/
for (unsigned int i = 0; i != cellSet.getCellList().size(); ++i) {
afwMath::SpatialCell::Ptr cell = cellSet.getCellList()[i];
for (afwMath::SpatialCell::iterator candidate = cell->begin(), candidateEnd = cell->end();
candidate != candidateEnd; ++candidate) {
dynamic_cast<ExampleCandidate *>((*candidate).get())->getBBox();
#if 0
std::cout << boost::format("%d %5.2f %5.2f %d\n")
% i % (*candidate)->getXCenter() % (*candidate)->getYCenter() % (w*h);
#endif
if (box.getArea() < 75) {
(*candidate)->setStatus(afwMath::SpatialCellCandidate::BAD);
}
}
}
/*
* Now count the good and bad candidates
*/
for (unsigned int i = 0; i != cellSet.getCellList().size(); ++i) {
afwMath::SpatialCell::Ptr cell = cellSet.getCellList()[i];
cell->visitCandidates(&visitor);
cell->setIgnoreBad(false); // include BAD in cell.size()
std::cout << boost::format("%s nobj=%d N_good=%d NPix_good=%d\n") %
cell->getLabel() % cell->size() % visitor.getN() % visitor.getNPix();
}
cellSet.setIgnoreBad(true); // don't visit BAD candidates
cellSet.visitCandidates(&visitor);
std::cout << boost::format("There are %d good candidates\n") % visitor.getN();
}
/*
* Read an image and background subtract it
*/
std::pair<afwImage::MaskedImage<PixelT>::Ptr, PTR(afwDetect::FootprintSet)>
readImage() {
try {
std::string dataDir = lsst::utils::getPackageDir("afwdata");
std::string filename = dataDir + "/CFHT/D4/cal-53535-i-797722_1.fits";
afwGeom::Point2I(270, 2530),
);
mi.reset(new afwImage::MaskedImage<PixelT>(filename, md, bbox));
} catch (lsst::pex::exceptions::NotFoundError &e) {
std::cerr << e << std::endl;
exit(1);
}
/*
* Subtract the background. We can't fix those pesky cosmic rays, as that's in a dependent product
* (meas/algorithms)
*/
bctrl.setNxSample(mi->getWidth()/256 + 1);
bctrl.setNySample(mi->getHeight()/256 + 1);
bctrl.getStatisticsControl()->setNumSigmaClip(3.0);
bctrl.getStatisticsControl()->setNumIter(2);
try {
*mi->getImage() -= *afwMath::makeBackground(*im, bctrl)->getImage<PixelT>();
} catch(std::exception &) {
bctrl.setInterpStyle(afwMath::Interpolate::CONSTANT);
*mi->getImage() -= *afwMath::makeBackground(*im, bctrl)->getImage<PixelT>();
}
/*
* Find sources
*/
int npixMin = 5; // we didn't smooth
PTR(afwDetect::FootprintSet) fs(new afwDetect::FootprintSet(*mi, threshold, "DETECTED", npixMin));
int const grow = 1;
bool const isotropic = false;
PTR(afwDetect::FootprintSet) grownFs(new afwDetect::FootprintSet(*fs, grow, isotropic));
grownFs->setMask(mi->getMask(), "DETECTED");
return std::make_pair(mi, grownFs);
}
//-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
/*
* Run the example
*/
int main() {
std::pair<afwImage::MaskedImage<PixelT>::Ptr, PTR(afwDetect::FootprintSet)> data = readImage();
assert (data.first != NULL); // stop compiler complaining about data being unused
SpatialCellSetDemo();
}