MatchSrcToCatalogue.cc

Go to the documentation of this file.

// -*- LSST-C++ -*-
 
/*
 * 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/>.
 */
 
#include "lsst/meas/astrom/sip/MatchSrcToCatalogue.h"
#include "lsst/geom/Angle.h"
#include "lsst/afw/geom/SkyWcs.h"
 
namespace lsst {
namespace meas {
namespace astrom {
namespace sip {

MatchSrcToCatalogue::MatchSrcToCatalogue(afw::table::SimpleCatalog const& catSet,
                                         afw::table::SourceCatalog const& imgSet,
                                         std::shared_ptr<afw::geom::SkyWcs const> wcs, geom::Angle dist) {
    setImgSrcSet(imgSet);
    setCatSrcSet(catSet);
    setDist(dist);
    setWcs(wcs);
}

void MatchSrcToCatalogue::setDist(geom::Angle dist) {
    if (dist <= 0) {
        throw LSST_EXCEPT(pex::exceptions::InvalidParameterError, "Distance must be > 0");
    }
    _dist = dist;
}

void MatchSrcToCatalogue::setWcs(std::shared_ptr<afw::geom::SkyWcs const> wcs) { _wcs = wcs; }

void MatchSrcToCatalogue::setImgSrcSet(afw::table::SourceCatalog const& srcSet) { _imgSet = srcSet; }
 
void MatchSrcToCatalogue::setCatSrcSet(afw::table::SimpleCatalog const& srcSet) { _catSet = srcSet; }
 
void MatchSrcToCatalogue::findMatches() {
    if (!_imgSet.getTable()->getCentroidSlot().isValid()) {
        throw LSST_EXCEPT(pex::exceptions::LogicError,
                          "SourceTable passed to MatchSrcToCatalogue does not have its centroid slot set.");
    }
 
    for (afw::table::SourceCatalog::const_iterator i = _imgSet.begin(); i != _imgSet.end(); ++i) {
        i->updateCoord(*_wcs);
    }
 
    _match = afw::table::matchRaDec(_catSet, _imgSet, _dist);
 
    _removeOneToMany();
    _removeManyToOne();
}

void MatchSrcToCatalogue::_removeOneToMany() {
    std::size_t size = _match.size();
    for (std::size_t i = 0; i < size; ++i) {
        for (std::size_t j = i + 1; j < size; ++j) {
            // If the same Source appears twice keep the one with the smaller separation from its match
            if (_match[i].first == _match[j].first) {
                // Keep the one with the shorter match distance, and disgard the other
                if (_match[i].distance < _match[j].distance) {
                    _match.erase(_match.begin() + j);
                    size--;
                    j--;
                } else {
                    _match.erase(_match.begin() + i);
                    size--;
                    i--;
                    break;
                }
            }
        }
    }
}

void MatchSrcToCatalogue::_removeManyToOne() {
    std::size_t size = _match.size();
    for (std::size_t i = 0; i < size; ++i) {
        for (std::size_t j = i + 1; j < size; ++j) {
            // If the same Source appears twice
            if (_match[i].second == _match[j].second) {
                // Keep the one with the shorter match distance, and disgard the other
                if (_match[i].distance < _match[j].distance) {
                    _match.erase(_match.begin() + j);
                    size--;
                    j--;
                } else {
                    _match.erase(_match.begin() + i);
                    size--;
                    i--;
                    break;
                }
            }
        }
    }
}
 
afw::table::ReferenceMatchVector MatchSrcToCatalogue::getMatches() {
    findMatches();
    return _match;
}
 
}  // namespace sip
}  // namespace astrom
}  // namespace meas
}  // namespace lsst