SortedCatalog.h

Go to the documentation of this file.

// -*- lsst-c++ -*-
/* 
 * LSST Data Management System
 * Copyright 2008, 2009, 2010, 2011 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/>.
 */
#ifndef AFW_TABLE_SortedCatalog_h_INCLUDED
#define AFW_TABLE_SortedCatalog_h_INCLUDED

#include "lsst/afw/table/fwd.h"
#include "lsst/afw/table/Catalog.h"

namespace lsst { namespace afw { namespace table {

#ifndef SWIG

template <typename RecordT>
class SortedCatalogT : public CatalogT<RecordT> {
    typedef CatalogT<RecordT> Base;
public:

    typedef RecordT Record;
    typedef typename Record::Table Table;

    typedef typename Base::iterator iterator;
    typedef typename Base::const_iterator const_iterator;

    using Base::isSorted;
    using Base::sort;
    using Base::find;

    bool isSorted() const { return this->isSorted(Table::getIdKey()); }

    void sort() { this->sort(Table::getIdKey()); }


    iterator find(RecordId id) { return this->find(id, Table::getIdKey()); }
    const_iterator find(RecordId id) const { return this->find(id, Table::getIdKey()); }

    explicit SortedCatalogT(PTR(Table) const & table = PTR(Table)()) : Base(table) {}

    explicit SortedCatalogT(Schema const & schema) : Base(schema) {}

    template <typename InputIterator>
    SortedCatalogT(PTR(Table) const & table, InputIterator first, InputIterator last, bool deep=false) :
        Base(table, first, last, deep)
    {}

    template <typename OtherRecordT>
    SortedCatalogT(SortedCatalogT<OtherRecordT> const & other) : Base(other) {}

    static SortedCatalogT readFits(std::string const & filename, int hdu=0, int flags=0) {
        return io::FitsReader::apply<SortedCatalogT>(filename, hdu, flags);
    }

    static SortedCatalogT readFits(fits::MemFileManager & manager, int hdu=0, int flags=0) {
        return io::FitsReader::apply<SortedCatalogT>(manager, hdu, flags);
    }

    static SortedCatalogT readFits(fits::Fits & fitsfile, int flags=0) {
        return io::FitsReader::apply<SortedCatalogT>(fitsfile, flags);
    }

    SortedCatalogT<RecordT> subset(ndarray::Array<bool const,1> const & mask) const {
        return SortedCatalogT(Base::subset(mask));
    }

    SortedCatalogT subset(std::ptrdiff_t startd, std::ptrdiff_t stopd, std::ptrdiff_t step) const {
        return SortedCatalogT(Base::subset(startd, stopd, step));
    }

protected:
    explicit SortedCatalogT(Base const & other) : Base(other) {}
};

#endif // !SWIG

}}} // namespace lsst::afw::table

#endif // !AFW_TABLE_SortedCatalog_h_INCLUDED