4 #include "astrometry/solver.h" 5 #include "astrometry/index.h" 6 #include "astrometry/starkd.h" 7 #include "astrometry/fitsioutils.h" 8 #include "astrometry/fitstable.h" 17 #include "boost/format.hpp" 24 #pragma clang diagnostic push 25 #pragma clang diagnostic ignored "-Wunused-variable" 28 #pragma clang diagnostic pop 35 namespace extensions {
36 namespace astrometryNet {
46 char const *indexName)
48 float *col =
static_cast<float*
>(fitstable_read_column_inds(tag, colName, type, starinds, nstars));
51 str(boost::format(
"Unable to read data for %s from %s") % colName % indexName));
64 char const* isStarCol,
73 size_t const nMag = magColInfoList.
size();
75 for (
auto mc = magColInfoList.
cbegin(); mc != magColInfoList.
cend(); ++mc) {
76 if (mc->filterName.empty()) {
78 "Magnitude names cannot be empty strings.");
82 if (mc->magCol.empty()) {
84 "Magnitude column names cannot be empty strings.");
92 radecdeg2xyzarr(raDeg, decDeg, xyz);
93 double r2 = deg2distsq(radius.
asDegrees());
98 "centroid on some exposure; invalid unless \"hasCentroid\" is true)",
"pixels");
99 auto hasCentroidKey = schema.addField<afwTable::Flag>(
"hasCentroid",
100 "true if centroid field has been set");
107 for (
auto mc = magColInfoList.
cbegin(); mc != magColInfoList.
cend(); ++mc) {
111 schema.addField<
double>(
112 mc->filterName +
"_flux",
113 mc->filterName +
" flux"));
116 schema.addField<
double>(
117 mc->filterName +
"_fluxSigma",
118 mc->filterName +
" flux uncertainty (sigma)"));
124 resolvedKey = schema.addField<afwTable::Flag>(
126 "set if the reference object is resolved");
130 variableKey = schema.addField<afwTable::Flag>(
132 "set if the reference object is variable");
136 "set if the reference object can be used in photometric calibration");
151 index_t* ind = (*pind);
153 double *radecs = NULL;
154 int *starinds = NULL;
156 startree_search_for(ind->starkd, xyz, r2, NULL, &radecs, &starinds, &nstars);
167 bool* stargal = NULL;
169 if (idCol || nMag || isStarCol || isVarCol) {
170 fitstable_t* tag = startree_get_tagalong(ind->starkd);
171 tfits_type flt = fitscolumn_float_type();
172 tfits_type boo = fitscolumn_boolean_type();
173 tfits_type i64 = fitscolumn_i64_type();
177 "astrometry_net_data index file %s does not contain a tag-along table, " 178 "so can't retrieve extra columns. idCol=%s, isStarCol=%s, isVarCol=%s") %
179 ind->indexname % idCol % isStarCol % isVarCol);
180 msg +=
", mag columns=[";
181 for (
unsigned int i=0; i<nMag; i++) {
185 msg +=
" name='" + magColInfoList[i].filterName +
186 "', mag='" + magColInfoList[i].magCol +
187 "', magErr='" + magColInfoList[i].magErrCol +
"'";
189 msg +=
" ]. You may need to edit the $ASTROMETRY_NET_DATA_DIR/andConfig.py file to set idColumn=None, etc.";
194 id =
static_cast<int64_t*
>(fitstable_read_column_inds(tag, idCol, i64, starinds, nstars));
197 str(boost::format(
"Unable to read data for %s from %s") % idCol % ind->indexname));
200 if (
id && uniqueIds) {
210 for (
int i=0; i<nstars; i++) {
212 if (uids.
insert(
id[i]).second) {
216 starinds[nkeep] = starinds[i];
217 radecs[nkeep*2+0] = radecs[i*2+0];
218 radecs[nkeep*2+1] = radecs[i*2+1];
238 for (
auto mc = magColInfoList.
cbegin(); mc != magColInfoList.
cend(); ++mc) {
239 char const* col = mc->magCol.c_str();
240 mag.
push_back(read_column(tag, col, flt, starinds, nstars, ind->indexname));
242 char const* col = mc->magErrCol.c_str();
243 magErr.
push_back(read_column(tag, col, flt, starinds, nstars, ind->indexname));
253 uint8_t* sg =
static_cast<uint8_t*
>(fitstable_read_column_inds(
254 tag, isStarCol, fitscolumn_u8_type(), starinds, nstars));
255 stargal =
static_cast<bool*
>(malloc(nstars));
258 str(boost::format(
"Unable to read data for %s from %s") % isStarCol % ind->indexname));
260 for (
int j=0; j<nstars; j++) {
261 stargal[j] = (sg[j] > 0);
266 var =
static_cast<bool*
>(fitstable_read_column_inds(tag, isVarCol, boo, starinds, nstars));
269 str(boost::format(
"Unable to read data for %s from %s") % isVarCol % ind->indexname));
274 for (
int i=0; i<nstars; i++) {
281 radecs[i * 2 + 0] * afwGeom::degrees,
282 radecs[i * 2 + 1] * afwGeom::degrees
290 src->set(hasCentroidKey,
false);
292 assert(fluxKey.size() == nMag);
294 assert(fluxErrKey.
size() == magErr.
size());
298 for (
auto mc = magColInfoList.
cbegin(); mc != magColInfoList.
cend(); ++mc, ++j) {
306 src->set(fluxKey[j], flux);
311 src->set(fluxErrKey[ej], fluxErr);
315 assert(ej == fluxErrKey.
size());
317 bool photometric =
true;
319 src->set(resolvedKey, !stargal[i]);
320 photometric &= stargal[i];
323 src->set(variableKey, var[i]);
324 photometric &= (!var[i]);
326 src->set(photometricKey, photometric);
330 for (
size_t j=0; j<mag.
size(); ++j) {
333 for (
size_t j=0; j<magErr.
size(); ++j) {
static PointKey addFields(Schema &schema, std::string const &name, std::string const &doc, std::string const &unit)
static Schema makeMinimalSchema()
Angle getLongitude() const noexcept
double fluxErrFromABMagErr(double magErr, double mag)
double fluxFromABMag(double mag)
lsst::afw::table::SimpleCatalog getCatalogImpl(std::vector< index_t *> inds, lsst::afw::geom::SpherePoint 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)
Implementation for index_t::getCatalog method.
Angle getLatitude() const noexcept
#define LSST_EXCEPT(type,...)
constexpr double asDegrees() const noexcept
static std::shared_ptr< SimpleTable > make(Schema const &schema, std::shared_ptr< IdFactory > const &idFactory)