LSST Applications  21.0.0+04719a4bac,21.0.0-1-ga51b5d4+f5e6047307,21.0.0-11-g2b59f77+a9c1acf22d,21.0.0-11-ga42c5b2+86977b0b17,21.0.0-12-gf4ce030+76814010d2,21.0.0-13-g1721dae+760e7a6536,21.0.0-13-g3a573fe+768d78a30a,21.0.0-15-g5a7caf0+f21cbc5713,21.0.0-16-g0fb55c1+b60e2d390c,21.0.0-19-g4cded4ca+71a93a33c0,21.0.0-2-g103fe59+bb20972958,21.0.0-2-g45278ab+04719a4bac,21.0.0-2-g5242d73+3ad5d60fb1,21.0.0-2-g7f82c8f+8babb168e8,21.0.0-2-g8f08a60+06509c8b61,21.0.0-2-g8faa9b5+616205b9df,21.0.0-2-ga326454+8babb168e8,21.0.0-2-gde069b7+5e4aea9c2f,21.0.0-2-gecfae73+1d3a86e577,21.0.0-2-gfc62afb+3ad5d60fb1,21.0.0-25-g1d57be3cd+e73869a214,21.0.0-3-g357aad2+ed88757d29,21.0.0-3-g4a4ce7f+3ad5d60fb1,21.0.0-3-g4be5c26+3ad5d60fb1,21.0.0-3-g65f322c+e0b24896a3,21.0.0-3-g7d9da8d+616205b9df,21.0.0-3-ge02ed75+a9c1acf22d,21.0.0-4-g591bb35+a9c1acf22d,21.0.0-4-g65b4814+b60e2d390c,21.0.0-4-gccdca77+0de219a2bc,21.0.0-4-ge8a399c+6c55c39e83,21.0.0-5-gd00fb1e+05fce91b99,21.0.0-6-gc675373+3ad5d60fb1,21.0.0-64-g1122c245+4fb2b8f86e,21.0.0-7-g04766d7+cd19d05db2,21.0.0-7-gdf92d54+04719a4bac,21.0.0-8-g5674e7b+d1bd76f71f,master-gac4afde19b+a9c1acf22d,w.2021.13
LSST Data Management Base Package
fitsChan.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  *
4  * This product includes software developed by the
5  * LSST Project (http://www.lsst.org/).
6  * See the COPYRIGHT file
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 LSST License Statement and
19  * the GNU General Public License along with this program. If not,
20  * see <https://www.lsstcorp.org/LegalNotices/>.
21  */
22 #include <complex>
23 #include <memory>
24 
25 #include <pybind11/pybind11.h>
26 #include <pybind11/complex.h>
27 #include <pybind11/stl.h>
28 
29 #include "astshim/Channel.h"
30 #include "astshim/FitsChan.h"
31 
32 namespace py = pybind11;
33 using namespace pybind11::literals;
34 
35 namespace ast {
36 namespace {
37 
38 template <typename T>
39 void wrapFoundValue(py::module &mod, std::string const &suffix) {
40  py::class_<FoundValue<T>> cls(mod, ("FoundValue" + suffix).c_str());
41 
42  cls.def(py::init<bool, T const &>(), "found"_a, "value"_a);
43 
44  cls.def_readwrite("found", &FoundValue<T>::found);
45  cls.def_readwrite("value", &FoundValue<T>::value);
46 }
47 
48 PYBIND11_MODULE(fitsChan, mod) {
49  py::module::import("astshim.channel");
50 
51  py::enum_<FitsKeyState>(mod, "FitsKeyState")
52  .value("ABSENT", FitsKeyState::ABSENT)
53  .value("NOVALUE", FitsKeyState::NOVALUE)
54  .value("PRESENT", FitsKeyState::PRESENT)
55  .export_values();
56 
57  py::enum_<CardType>(mod, "CardType")
58  .value("NOTYPE", CardType::NOTYPE)
59  .value("COMMENT", CardType::COMMENT)
60  .value("INT", CardType::INT)
61  .value("FLOAT", CardType::FLOAT)
62  .value("STRING", CardType::STRING)
63  .value("COMPLEXF", CardType::COMPLEXF)
64  .value("COMPLEXI", CardType::COMPLEXI)
65  .value("LOGICAL", CardType::LOGICAL)
66  .value("CONTINUE", CardType::CONTINUE)
67  .value("UNDEF", CardType::UNDEF)
68  .export_values();
69 
70  // Wrap FoundValue struct for returning various types
71  wrapFoundValue<std::string>(mod, "S");
72  wrapFoundValue<std::complex<double>>(mod, "CF");
73  wrapFoundValue<double>(mod, "F");
74  wrapFoundValue<int>(mod, "I");
75  wrapFoundValue<bool>(mod, "L");
76 
77  // Wrap FitsChan
78  py::class_<FitsChan, std::shared_ptr<FitsChan>, Channel> cls(mod, "FitsChan");
79 
80  cls.def(py::init<Stream &, std::string const &>(), "stream"_a, "options"_a = "");
81 
82  cls.def_property("carLin", &FitsChan::getCarLin, &FitsChan::setCarLin);
83  cls.def_property("cdMatrix", &FitsChan::getCDMatrix, &FitsChan::setCDMatrix);
84  cls.def_property("clean", &FitsChan::getClean, &FitsChan::setClean);
85  cls.def_property("defB1950", &FitsChan::getDefB1950, &FitsChan::setDefB1950);
86  cls.def_property("encoding", &FitsChan::getEncoding, &FitsChan::setEncoding);
87  cls.def_property("fitsAxisOrder", &FitsChan::getFitsAxisOrder, &FitsChan::setFitsAxisOrder);
88  cls.def_property("fitsDigits", &FitsChan::getFitsDigits, &FitsChan::setFitsDigits);
89  cls.def_property_readonly("nCard", &FitsChan::getNCard);
90  cls.def_property_readonly("nKey", &FitsChan::getNKey);
91  cls.def_property("iwc", &FitsChan::getIwc, &FitsChan::setIwc);
92  cls.def_property("sipOK", &FitsChan::getSipOK, &FitsChan::setSipOK);
93  cls.def_property("sipReplace", &FitsChan::getSipReplace, &FitsChan::setSipReplace);
94  cls.def_property("tabOK", &FitsChan::getTabOK, &FitsChan::setTabOK);
95  cls.def_property("polyTan", &FitsChan::getPolyTan, &FitsChan::setPolyTan);
96  cls.def_property("warnings", &FitsChan::getWarnings, &FitsChan::setWarnings);
97  cls.def_property("fitsTol", &FitsChan::getFitsTol, &FitsChan::setFitsTol);
98 
99  cls.def("delFits", &FitsChan::delFits);
100  cls.def("emptyFits", &FitsChan::emptyFits);
101  cls.def("findFits", &FitsChan::findFits, "name"_a, "inc"_a);
102  cls.def("getFitsCF", &FitsChan::getFitsCF, "name"_a = "", "defval"_a = std::complex<double>(0, 0));
103  cls.def("getFitsCN", &FitsChan::getFitsCN, "name"_a = "", "defval"_a = "");
104  cls.def("getFitsF", &FitsChan::getFitsF, "name"_a = "", "defval"_a = 0);
105  cls.def("getFitsI", &FitsChan::getFitsI, "name"_a = "", "defval"_a = 0);
106  cls.def("getFitsL", &FitsChan::getFitsL, "name"_a = "", "defval"_a = false);
107  cls.def("getFitsS", &FitsChan::getFitsS, "name"_a = "", "defval"_a = "");
108  cls.def("getAllCardNames", &FitsChan::getAllCardNames);
109  cls.def("getAllWarnings", &FitsChan::getAllWarnings);
110  cls.def("getCard", &FitsChan::getCard);
111  cls.def("getCardComm", &FitsChan::getCardComm);
112  cls.def("getCardName", &FitsChan::getCardName);
113  cls.def("getCardType", &FitsChan::getCardType);
114  cls.def("getTables", &FitsChan::getTables);
115  cls.def("purgeWcs", &FitsChan::purgeWcs);
116  cls.def("putCards", &FitsChan::putCards, "cards"_a);
117  cls.def("putFits", &FitsChan::putFits, "card"_a, "overwrite"_a);
118  cls.def("readFits", &FitsChan::readFits);
119  cls.def("retainFits", &FitsChan::retainFits);
120  cls.def("setFitsCF", &FitsChan::setFitsCF, "name"_a, "value"_a, "comment"_a = "", "overwrite"_a = false);
121  cls.def("setFitsCM", &FitsChan::setFitsCM, "comment"_a = "", "overwrite"_a = false);
122  cls.def("setFitsCN", &FitsChan::setFitsCN, "name"_a, "value"_a, "comment"_a = "", "overwrite"_a = false);
123  cls.def("setFitsF", &FitsChan::setFitsF, "name"_a, "value"_a, "comment"_a = "", "overwrite"_a = false);
124  cls.def("setFitsI", &FitsChan::setFitsI, "name"_a, "value"_a, "comment"_a = "", "overwrite"_a = false);
125  cls.def("setFitsL", &FitsChan::setFitsL, "name"_a, "value"_a, "comment"_a = "", "overwrite"_a = false);
126  cls.def("setFitsS", &FitsChan::setFitsS, "name"_a, "value"_a, "comment"_a = "", "overwrite"_a = false);
127  cls.def("setFitsU", &FitsChan::setFitsU, "name"_a, "comment"_a = "", "overwrite"_a = false);
128  cls.def("showFits", &FitsChan::showFits);
129  cls.def("testFits", &FitsChan::testFits, "name"_a = "");
130  cls.def("writeFits", &FitsChan::writeFits);
131  cls.def("clearCard", &FitsChan::clearCard);
132  cls.def("setCard", &FitsChan::setCard, "i"_a);
133 }
134 
135 } // namespace
136 } // namespace ast
AST wrapper classes and functions.
PYBIND11_MODULE(_cameraGeom, mod)
Definition: _cameraGeom.cc:38
def writeFits(filename, stamp_ims, metadata, type_name, write_mask, write_variance)
Definition: stamps.py:40