Peak.h

Go to the documentation of this file.

// -*- lsst-c++ -*-
/*
 * LSST Data Management System
 * Copyright 2008-2014 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_DETECTION_Peak_h_INCLUDED
#define AFW_DETECTION_Peak_h_INCLUDED

#include "lsst/afw/table/BaseRecord.h"
#include "lsst/afw/table/BaseTable.h"
#include "lsst/afw/table/IdFactory.h"
#include "lsst/afw/table/Catalog.h"
#include "lsst/afw/table/BaseColumnView.h"

namespace lsst { namespace afw { namespace detection {

class PeakRecord;
class PeakTable;

class PeakRecord : public afw::table::BaseRecord {
public:

    typedef PeakTable Table;
    typedef afw::table::ColumnViewT<PeakRecord> ColumnView;
    typedef afw::table::CatalogT<PeakRecord> Catalog;
    typedef afw::table::CatalogT<PeakRecord const> ConstCatalog;

    CONST_PTR(PeakTable) getTable() const {
        return std::static_pointer_cast<PeakTable const>(afw::table::BaseRecord::getTable());
    }

    afw::table::RecordId getId() const;
    void setId(afw::table::RecordId id);

    int getIx() const;
    int getIy() const;
    void setIx(int ix);
    void setIy(int iy);
    afw::geom::Point2I getI() const { return afw::geom::Point2I(getIx(), getIy()); }
    afw::geom::Point2I getCentroid(bool) const { return getI(); }

    float getFx() const;
    float getFy() const;
    void setFx(float fx);
    void setFy(float fy);
    afw::geom::Point2D getF() const { return afw::geom::Point2D(getFx(), getFy()); }
    afw::geom::Point2D getCentroid() const { return getF(); }

    float getPeakValue() const;
    void setPeakValue(float peakValue);

protected:

    PeakRecord(PTR(PeakTable) const & table);

};

class PeakTable : public afw::table::BaseTable {
public:

    typedef PeakRecord Record;
    typedef afw::table::ColumnViewT<PeakRecord> ColumnView;
    typedef afw::table::CatalogT<Record> Catalog;
    typedef afw::table::CatalogT<Record const> ConstCatalog;

    static PTR(PeakTable) make(afw::table::Schema const & schema, bool forceNew=false);

    static afw::table::Schema makeMinimalSchema() { return getMinimalSchema().schema; }

    static bool checkSchema(afw::table::Schema const & other) {
        return other.contains(getMinimalSchema().schema);
    }

    PTR(afw::table::IdFactory) getIdFactory() { return _idFactory; }

    CONST_PTR(afw::table::IdFactory) getIdFactory() const { return _idFactory; }

    void setIdFactory(PTR(afw::table::IdFactory) f) { _idFactory = f; }


    static afw::table::Key<afw::table::RecordId> getIdKey() { return getMinimalSchema().id; }
    static afw::table::Key<int> getIxKey() { return getMinimalSchema().ix; }
    static afw::table::Key<int> getIyKey() { return getMinimalSchema().iy; }
    static afw::table::Key<float> getFxKey() { return getMinimalSchema().fx; }
    static afw::table::Key<float> getFyKey() { return getMinimalSchema().fy; }
    static afw::table::Key<float> getPeakValueKey() { return getMinimalSchema().peakValue; }

    PTR(PeakTable) clone() const { return std::static_pointer_cast<PeakTable>(_clone()); }

    PTR(PeakRecord) makeRecord() { return std::static_pointer_cast<PeakRecord>(_makeRecord()); }

    PTR(PeakRecord) copyRecord(afw::table::BaseRecord const & other) {
        return std::static_pointer_cast<PeakRecord>(afw::table::BaseTable::copyRecord(other));
    }

    PTR(PeakRecord) copyRecord(
        afw::table::BaseRecord const & other,
        afw::table::SchemaMapper const & mapper
    ) {
        return std::static_pointer_cast<PeakRecord>(afw::table::BaseTable::copyRecord(other, mapper));
    }

protected:

    PeakTable(afw::table::Schema const & schema, PTR(afw::table::IdFactory) const & idFactory);

    PeakTable(PeakTable const & other);

private:

    // Struct that holds the minimal schema and the special keys we've added to it.
    struct MinimalSchema {
        afw::table::Schema schema;
        afw::table::Key<afw::table::RecordId> id;
        afw::table::Key<float> fx;
        afw::table::Key<float> fy;
        afw::table::Key<int> ix;
        afw::table::Key<int> iy;
        afw::table::Key<float> peakValue;

        MinimalSchema();
    };

    // Return the singleton minimal schema.
    static MinimalSchema & getMinimalSchema();

    friend class afw::table::io::FitsWriter;

     // Return a writer object that knows how to save in FITS format.  See also FitsWriter.
    virtual PTR(afw::table::io::FitsWriter) makeFitsWriter(fits::Fits * fitsfile, int flags) const;

    PTR(afw::table::IdFactory) _idFactory;        // generates IDs for new records
};

#ifndef SWIG

std::ostream & operator<<(std::ostream & os, PeakRecord const & record);

inline afw::table::RecordId PeakRecord::getId() const { return get(PeakTable::getIdKey()); }
inline void PeakRecord::setId(afw::table::RecordId id) { set(PeakTable::getIdKey(), id); }

inline int PeakRecord::getIx() const { return get(PeakTable::getIxKey()); }
inline int PeakRecord::getIy() const { return get(PeakTable::getIyKey()); }
inline void PeakRecord::setIx(int ix) { set(PeakTable::getIxKey(), ix); }
inline void PeakRecord::setIy(int iy) { set(PeakTable::getIyKey(), iy); }

inline float PeakRecord::getFx() const { return get(PeakTable::getFxKey()); }
inline float PeakRecord::getFy() const { return get(PeakTable::getFyKey()); }
inline void PeakRecord::setFx(float fx) { set(PeakTable::getFxKey(), fx); }
inline void PeakRecord::setFy(float fy) { set(PeakTable::getFyKey(), fy); }

inline float PeakRecord::getPeakValue() const { return get(PeakTable::getPeakValueKey()); }
inline void PeakRecord::setPeakValue(float peakValue) { set(PeakTable::getPeakValueKey(), peakValue); }

#endif // !SWIG

typedef afw::table::ColumnViewT<PeakRecord> PeakColumnView;
typedef afw::table::CatalogT<PeakRecord> PeakCatalog;
typedef afw::table::CatalogT<PeakRecord const> ConstPeakCatalog;

}}} // namespace lsst::afw::detection

#endif // !AFW_DETECTION_Peak_h_INCLUDED