Associations.h

Go to the documentation of this file.

// -*- LSST-C++ -*-
#ifndef LSST_JOINTCAL_ASSOCIATIONS_H
#define LSST_JOINTCAL_ASSOCIATIONS_H

#include <string>
#include <iostream>
#include <list>

#include "lsst/afw/table/Source.h"
#include "lsst/afw/image/TanWcs.h"
#include "lsst/afw/image/Calib.h"
#include "lsst/afw/image/VisitInfo.h"
#include "lsst/daf/base/PropertySet.h"
#include "lsst/afw/geom/Box.h"

#include "lsst/jointcal/RefStar.h"
#include "lsst/jointcal/FittedStar.h"
#include "lsst/jointcal/CcdImage.h"
#include "lsst/jointcal/Point.h"
#include "lsst/jointcal/JointcalControl.h"

#include "lsst/afw/table/SortedCatalog.h"

namespace lsst {
namespace jointcal {

class Associations {
public:
    CcdImageList ccdImageList;      // the catalog handlers
    RefStarList refStarList;        // e.g. GAIA or SDSS reference stars
    FittedStarList fittedStarList;  // stars that are going to be fitted

    Point _commonTangentPoint;

public:
    // These are strictly speaking not needed anymore (after DM-4043),
    // but keeping them seems cleaner then exposing the lists themselves.
    size_t refStarListSize() { return refStarList.size(); }
    size_t fittedStarListSize() { return fittedStarList.size(); }

    Associations();

    void setCommonTangentPoint(lsst::afw::geom::Point2D const &commonTangentPoint);

    Point getCommonTangentPoint() const { return _commonTangentPoint; }

    void addImage(lsst::afw::table::SortedCatalogT<lsst::afw::table::SourceRecord> &catalog,
                  std::shared_ptr<lsst::afw::image::TanWcs> wcs,
                  std::shared_ptr<lsst::afw::image::VisitInfo> visitInfo, lsst::afw::geom::Box2I const &bbox,
                  std::string const &filter, std::shared_ptr<lsst::afw::image::Calib> calib, int visit,
                  int ccd, std::shared_ptr<lsst::jointcal::JointcalControl> control);

    void associateCatalogs(const double matchCutInArcsec = 0, const bool useFittedList = false,
                           const bool enlargeFittedList = true);

    void collectRefStars(lsst::afw::table::SortedCatalogT<lsst::afw::table::SimpleRecord> &refCat,
                         afw::geom::Angle matchCut, std::string const &fluxField);

    void deprojectFittedStars();

/* If Color is "g-i", then the color is assigned from columns "g" and "i" of the colored catalog. */
#ifdef TODO
    void setFittedStarColors(std::string const &dicStarListName, std::string const &color,
                             double matchCutArcSec);
#endif

    void selectFittedStars(int minMeasurements);

    const CcdImageList &getCcdImageList() const { return ccdImageList; }

    unsigned NBands() const { return 1; }

    // Return the bounding box in (ra, dec) coordinates containing the whole catalog
    const lsst::afw::geom::Box2D getRaDecBBox();

    int nCcdImagesValidForFit() const {
        return std::count_if(
                ccdImageList.begin(), ccdImageList.end(),
                [](std::shared_ptr<CcdImage> const &item) { return item->getCatalogForFit().size() > 0; });
    }

private:
    void associateRefStars(double matchCutInArcsec, const Gtransfo *gtransfo);

    void assignMags();
};

}  // namespace jointcal
}  // namespace lsst
#endif  // LSST_JOINTCAL_ASSOCIATIONS_H