LSST Applications g042eb84c57+730a74494b,g04e9c324dd+8c5ae1fdc5,g134cb467dc+1f1e3e7524,g199a45376c+0ba108daf9,g1fd858c14a+fa7d31856b,g210f2d0738+f66ac109ec,g262e1987ae+83a3acc0e5,g29ae962dfc+d856a2cb1f,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+a1e0c9f713,g47891489e3+0d594cb711,g4d44eb3520+c57ec8f3ed,g4d7b6aa1c5+f66ac109ec,g53246c7159+8c5ae1fdc5,g56a1a4eaf3+fd7ad03fde,g64539dfbff+f66ac109ec,g67b6fd64d1+0d594cb711,g67fd3c3899+f66ac109ec,g6985122a63+0d594cb711,g74acd417e5+3098891321,g786e29fd12+668abc6043,g81db2e9a8d+98e2ab9f28,g87389fa792+8856018cbb,g89139ef638+0d594cb711,g8d7436a09f+80fda9ce03,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+a8a29bda4b,g99822b682c+e3ec3c61f9,g9d5c6a246b+0d5dac0c3d,ga41d0fce20+9243b26dd2,gbf99507273+8c5ae1fdc5,gd7ef33dd92+0d594cb711,gdab6d2f7ff+3098891321,ge410e46f29+0d594cb711,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.38
LSST Data Management Base Package
Loading...
Searching...
No Matches
_SpherePoint.cc
Go to the documentation of this file.
1/*
2 * Developed for the LSST Data Management System.
3 * This product includes software developed by the LSST Project
4 * (https://www.lsst.org).
5 * See the COPYRIGHT file at the top-level directory of this distribution
6 * for details of code ownership.
7 *
8 * This program is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program. If not, see <https://www.gnu.org/licenses/>.
20 */
21
22#include "pybind11/pybind11.h"
23#include "pybind11/stl.h"
24#include "pybind11/numpy.h"
25
26#include <cmath>
27#include <memory>
28
32#include "lsst/sphgeom/LonLat.h"
33#include "lsst/geom/Angle.h"
34#include "lsst/geom/Point.h"
36
37namespace py = pybind11;
38using namespace py::literals;
39
40namespace lsst {
41namespace geom {
42
43namespace {
44
45double toUnitX(double longitude, double latitude) {
46 return std::cos(longitude) * std::cos(latitude);
47}
48
49double toUnitY(double longitude, double latitude) {
50 return std::sin(longitude) * std::cos(latitude);
51}
52
53double toUnitZ(double longitude, double latitude) {
54 return std::sin(latitude);
55}
56
57} // anonymous
58
59
60using PySpherePoint = py::classh<SpherePoint>;
61
63 wrappers.addSignatureDependency("lsst.sphgeom");
64 wrappers.wrapType(
65 PySpherePoint(wrappers.module, "SpherePoint"),
66 [](auto & mod, auto & cls) mutable {
67 /* Constructors */
68 cls.def(py::init<>());
69 cls.def(py::init<Angle const &, Angle const &>(), "longitude"_a, "latitude"_a);
70 cls.def(py::init<double, double, AngleUnit>(), "longitude"_a, "latitude"_a, "units"_a);
71 cls.def(py::init<sphgeom::Vector3d const &>(), "vector"_a);
72 cls.def(py::init<sphgeom::UnitVector3d const &>(), "unitVector"_a);
73 cls.def(py::init<sphgeom::LonLat const &>(), "lonLat"_a);
74 cls.def(py::init<SpherePoint const &>(), "other"_a);
75 py::implicitly_convertible<SpherePoint, sphgeom::LonLat>();
76 py::implicitly_convertible<sphgeom::LonLat, SpherePoint>();
77
78 /* Operators */
79 cls.def("__getitem__",
80 [](SpherePoint const &self, std::ptrdiff_t i) {
81 return self[cpputils::python::cppIndex(2, i)];
82 });
83 cls.def("__eq__", &SpherePoint::operator==, py::is_operator());
84 cls.def("__ne__", &SpherePoint::operator!=, py::is_operator());
85
86 /* Members */
87 cls.def("getLongitude", &SpherePoint::getLongitude);
88 cls.def("getLatitude", &SpherePoint::getLatitude);
89 cls.def("getRa", &SpherePoint::getRa);
90 cls.def("getDec", &SpherePoint::getDec);
91 cls.def("getVector", &SpherePoint::getVector);
92 cls.def("getPosition", &SpherePoint::getPosition, "units"_a);
93 cls.def("atPole", &SpherePoint::atPole);
94 cls.def("isFinite", &SpherePoint::isFinite);
95 cls.def("bearingTo", &SpherePoint::bearingTo, "other"_a);
96 cls.def("separation", &SpherePoint::separation, "other"_a);
97 cls.def("rotated", &SpherePoint::rotated, "axis"_a, "amount"_a);
98 cls.def("offset", &SpherePoint::offset, "bearing"_a, "amount"_a);
99 cls.def("getTangentPlaneOffset", &SpherePoint::getTangentPlaneOffset, "other"_a);
100 cpputils::python::addOutputOp(cls, "__str__");
101 cls.def("__len__", [](SpherePoint const &) { return 2; });
102 cls.def("__reduce__", [cls](SpherePoint const &self) {
103 return py::make_tuple(cls, py::make_tuple(py::cast(self.getLongitude()),
104 py::cast(self.getLatitude())));
105 });
106
107
108 /* Module level */
109 mod.def("averageSpherePoint", averageSpherePoint);
110
111 // Used only by pure-Python extension toUnitXYZ in _SpherePoint.py.
112 mod.def("_toUnitX", py::vectorize(&toUnitX), "longitude"_a, "latitude"_a);
113 mod.def("_toUnitY", py::vectorize(&toUnitY), "longitude"_a, "latitude"_a);
114 mod.def("_toUnitZ", py::vectorize(&toUnitZ), "longitude"_a, "latitude"_a);
115 }
116 );
117}
118
119} // namespace geom
120} // namespace lsst
This file contains a class representing spherical coordinates.
This file declares a class for representing unit vectors in ℝ³.
This file declares a class for representing vectors in ℝ³.
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition python.h:242
void addSignatureDependency(std::string const &name)
Indicate an external module that provides a type used in function/method signatures.
Definition python.h:357
PyType wrapType(PyType cls, ClassWrapperCallback function, bool setModuleName=true)
Add a type (class or enum) wrapper, deferring method and other attribute definitions until finish() i...
Definition python.h:391
pybind11::module module
The module object passed to the PYBIND11_MODULE block that contains this WrapperCollection.
Definition python.h:448
Point in an unspecified spherical coordinate system.
Definition SpherePoint.h:57
std::pair< Angle, Angle > getTangentPlaneOffset(SpherePoint const &other) const
Get the offset from a tangent plane centered at this point to another point.
SpherePoint offset(Angle const &bearing, Angle const &amount) const
Return a point offset from this one along a great circle.
bool isFinite() const noexcept
true if this point is a well-defined position.
Point2D getPosition(AngleUnit unit) const noexcept
Return longitude, latitude as a Point2D object.
Angle bearingTo(SpherePoint const &other) const
Orientation at this point of the great circle arc to another point.
bool atPole() const noexcept
true if this point is either coordinate pole.
Angle separation(SpherePoint const &other) const noexcept
Angular distance between two points.
sphgeom::UnitVector3d getVector() const noexcept
A unit vector representation of this point.
Angle getLatitude() const noexcept
The latitude of this point.
Angle getLongitude() const noexcept
The longitude of this point.
SpherePoint rotated(SpherePoint const &axis, Angle const &amount) const noexcept
Return a point rotated from this one around an axis.
Angle getDec() const noexcept
Synonym for getLatitude.
Angle getRa() const noexcept
Synonym for getLongitude.
T cos(T... args)
void addOutputOp(PyClass &cls, std::string const &method)
Add __str__ or __repr__ method implemented by operator<<.
Definition python.h:87
SpherePoint averageSpherePoint(std::vector< SpherePoint > const &coords)
Return the average of a list of coordinates.
void wrapSpherePoint(WrapperCollection &wrappers)
py::classh< SpherePoint > PySpherePoint
T sin(T... args)