LSSTApplications  17.0+11,17.0+34,17.0+56,17.0+57,17.0+59,17.0+7,17.0-1-g377950a+33,17.0.1-1-g114240f+2,17.0.1-1-g4d4fbc4+28,17.0.1-1-g55520dc+49,17.0.1-1-g5f4ed7e+52,17.0.1-1-g6dd7d69+17,17.0.1-1-g8de6c91+11,17.0.1-1-gb9095d2+7,17.0.1-1-ge9fec5e+5,17.0.1-1-gf4e0155+55,17.0.1-1-gfc65f5f+50,17.0.1-1-gfc6fb1f+20,17.0.1-10-g87f9f3f+1,17.0.1-11-ge9de802+16,17.0.1-16-ga14f7d5c+4,17.0.1-17-gc79d625+1,17.0.1-17-gdae4c4a+8,17.0.1-2-g26618f5+29,17.0.1-2-g54f2ebc+9,17.0.1-2-gf403422+1,17.0.1-20-g2ca2f74+6,17.0.1-23-gf3eadeb7+1,17.0.1-3-g7e86b59+39,17.0.1-3-gb5ca14a,17.0.1-3-gd08d533+40,17.0.1-30-g596af8797,17.0.1-4-g59d126d+4,17.0.1-4-gc69c472+5,17.0.1-6-g5afd9b9+4,17.0.1-7-g35889ee+1,17.0.1-7-gc7c8782+18,17.0.1-9-gc4bbfb2+3,w.2019.22
LSSTDataManagementBasePackage
wcsUtils.cc
Go to the documentation of this file.
1 // -*- lsst-c++ -*-
2 
3 /*
4  * LSST Data Management System
5  * Copyright 2018 LSST Corporation.
6  *
7  * This product includes software developed by the
8  * LSST Project (http://www.lsst.org/).
9  *
10  * This program is free software: you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation, either version 3 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the LSST License Statement and
21  * the GNU General Public License along with this program. If not,
22  * see <http://www.lsstcorp.org/LegalNotices/>.
23  */
24 
25 #include <memory>
26 #include <vector>
27 
28 #include "lsst/pex/exceptions.h"
29 #include "lsst/afw/table/fwd.h"
31 #include "lsst/afw/table/Simple.h"
32 #include "lsst/afw/table/Source.h"
34 
35 namespace lsst {
36 namespace afw {
37 namespace table {
38 namespace {
39 
40 // Get the schema from a record
41 Schema getSchema(BaseRecord const &record) { return record.getSchema(); }
42 
43 // Get the schema from a shared pointer to a record
44 Schema getSchema(std::shared_ptr<const BaseRecord> const record) { return record->getSchema(); }
45 
46 // Get a value from a record
47 template <typename Key>
48 inline typename Key::Value getValue(BaseRecord const &record, Key const &key) {
49  return record.get(key);
50 }
51 
52 // Get a value from a shared pointer to a record
53 template <typename Key>
54 inline typename Key::Value getValue(std::shared_ptr<const BaseRecord> const record, Key const &key) {
55  return record->get(key);
56 }
57 
58 // Set a value in a record
59 template <typename Key>
60 inline void setValue(SimpleRecord &record, Key const &key, typename Key::Value const &value) {
61  record.set(key, value);
62 }
63 
64 // Set a value in a shared pointer to a record
65 template <typename Key>
66 inline void setValue(std::shared_ptr<SimpleRecord> record, Key const &key, typename Key::Value const &value) {
67  record->set(key, value);
68 }
69 
70 } // namespace
71 
72 template <typename ReferenceCollection>
73 void updateRefCentroids(geom::SkyWcs const &wcs, ReferenceCollection &refList) {
74  if (refList.empty()) {
75  return;
76  }
77  auto const schema = getSchema(refList[0]);
78  CoordKey const coordKey(schema["coord"]);
79  Point2DKey const centroidKey(schema["centroid"]);
80  Key<Flag> const hasCentroidKey(schema["hasCentroid"]);
82  skyList.reserve(refList.size());
83  for (auto const &record : refList) {
84  skyList.emplace_back(getValue(record, coordKey));
85  }
86  std::vector<lsst::geom::Point2D> const pixelList = wcs.skyToPixel(skyList);
87  auto pixelPos = pixelList.cbegin();
88  for (auto &refObj : refList) {
89  setValue(refObj, centroidKey, *pixelPos);
90  setValue(refObj, hasCentroidKey, true);
91  ++pixelPos;
92  }
93 }
94 
95 template <typename SourceCollection>
96 void updateSourceCoords(geom::SkyWcs const &wcs, SourceCollection &sourceList) {
97  if (sourceList.empty()) {
98  return;
99  }
100  auto const schema = getSchema(sourceList[0]);
101  Point2DKey const centroidKey(schema["slot_Centroid"]);
102  CoordKey const coordKey(schema["coord"]);
104  pixelList.reserve(sourceList.size());
105  for (auto const &source : sourceList) {
106  pixelList.emplace_back(getValue(source, centroidKey));
107  }
108  std::vector<lsst::geom::SpherePoint> const skyList = wcs.pixelToSky(pixelList);
109  auto skyCoord = skyList.cbegin();
110  for (auto &source : sourceList) {
111  setValue(source, coordKey, *skyCoord);
112  ++skyCoord;
113  }
114 }
115 
118 template void updateRefCentroids(geom::SkyWcs const &, SimpleCatalog &);
119 
121 template void updateSourceCoords(geom::SkyWcs const &, SourceCatalog &);
123 
124 } // namespace table
125 } // namespace afw
126 } // namespace lsst
A 2-dimensional celestial WCS that transform pixels to ICRS RA/Dec, using the LSST standard for pixel...
Definition: SkyWcs.h:117
lsst::geom::Point2D skyToPixel(lsst::geom::SpherePoint const &sky) const
Compute pixel position(s) from sky position(s)
Definition: SkyWcs.h:349
lsst::geom::SpherePoint pixelToSky(lsst::geom::Point2D const &pixel) const
Compute sky position(s) from pixel position(s)
Definition: SkyWcs.h:334
void updateRefCentroids(geom::SkyWcs const &wcs, ReferenceCollection &refList)
Update centroids in a collection of reference objects.
Definition: wcsUtils.cc:73
table::Key< table::Array< std::uint8_t > > wcs
Definition: SkyWcs.cc:71
void updateSourceCoords(geom::SkyWcs const &wcs, SourceCollection &sourceList)
Update sky coordinates in a collection of source objects.
Definition: wcsUtils.cc:96
A base class for image defects.
Custom catalog class for record/table subclasses that are guaranteed to have an ID, and should generally be sorted by that ID.
Definition: fwd.h:63
table::Schema schema
Definition: Camera.cc:161
const char * source()
Source function that allows astChannel to source from a Stream.
Definition: Stream.h:224
T get(T... args)
STL class.
T Value
the type returned by BaseRecord::get
Definition: FieldBase.h:44
T cbegin(T... args)
Key< U > key
Definition: Schema.cc:281
Key specialization for Flag.
Definition: Flag.h:94
T reserve(T... args)
A FunctorKey used to get or set celestial coordinates from a pair of lsst::geom::Angle keys...
Definition: aggregates.h:210
T emplace_back(T... args)