utils.h

Go to the documentation of this file.

#ifndef LSST_MEAS_EXTENSIONS_ASTROMETRYNET_UTILS_H
#define LSST_MEAS_EXTENSIONS_ASTROMETRYNET_UTILS_H

#include <string>
#include <vector>

#include "lsst/afw/table/Source.h"
#include "lsst/afw/coord.h"
#include "lsst/afw/geom.h"

namespace lsst {
namespace meas {
namespace extensions {
namespace astrometryNet {
namespace detail {

    struct MagColInfo {
        std::string filterName; 
        std::string magCol;     
        std::string magErrCol;  

        bool hasErr() const {
            return !magErrCol.empty();
        }
    };


struct IndexManager {
    index_t* index;
    IndexManager(index_t* ind) : index(ind) {}
    ~IndexManager() {
        // Change once astrometry.net-0.40+ is in...
        /*
          if (index_close_fds(ind)) {
          throw LSST_EXCEPT(lsst::pex::exceptions::IoError,
          "Failed to index_close_fds() an astrometry_net_data file");
          }
        */
        if (index->quads && index->quads->fb && index->quads->fb->fid) {
            _close(index->quads->fb->fid);
        }
        if (index->codekd && index->codekd->tree && index->codekd->tree->io) {
            _close(index->codekd->tree->io);
        }
        if (index->starkd && index->starkd->tree && index->starkd->tree->io) {
            _close(index->starkd->tree->io);
        }
    }
    void _close(FILE* & fid) {
        if (fid) {
            if (fclose(fid)) {
                std::cerr << "Error closing an astrometry_net_data quadfile" << std::endl;
            }
            fid = NULL;
        }
    }
    void _close(void* io) {
        _close(reinterpret_cast<kdtree_fits_t*>(io)->fid);
    }
};

lsst::afw::table::SimpleCatalog
getCatalogImpl(
    std::vector<index_t*> inds,
    lsst::afw::coord::Coord const &ctrCoord,
    lsst::afw::geom::Angle const &radius,
    const char* idCol,
    std::vector<MagColInfo> const& magColInfoList,
    const char* starGalCol,
    const char* varCol,
    bool uniqueIds=true);

}}}}}  // lsst::meas::extensions::astrometryNet::detail

#endif // LSST_MEAS_EXTENSIONS_ASTROMETRYNET_UTILS_H