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
_amplifier.cc
Go to the documentation of this file.
1 /*
2  * This file is part of afw.
3  *
4  * Developed for the LSST Data Management System.
5  * This product includes software developed by the LSST Project
6  * (https://www.lsst.org).
7  * See the COPYRIGHT file at the top-level directory of this distribution
8  * for details of code ownership.
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 GNU General Public License
21  * along with this program. If not, see <https://www.gnu.org/licenses/>.
22  */
23 
24 #include "pybind11/pybind11.h"
25 #include <lsst/utils/python.h>
26 #include "pybind11/stl.h"
27 
28 #include "ndarray/pybind11.h"
29 
32 
33 namespace py = pybind11;
34 using namespace pybind11::literals;
35 
36 namespace lsst {
37 namespace afw {
38 namespace cameraGeom {
39 
40 using PyAmplifier = py::class_<Amplifier, std::shared_ptr<Amplifier>>;
41 using PyAmplifierBuilder = py::class_<Amplifier::Builder, Amplifier, std::shared_ptr<Amplifier::Builder>>;
42 
44  wrappers.addInheritanceDependency("lsst.afw.table");
45  wrappers.wrapType(py::enum_<ReadoutCorner>(wrappers.module, "ReadoutCorner"), [](auto &mod, auto &enm) {
46  enm.value("LL", ReadoutCorner::LL);
47  enm.value("LR", ReadoutCorner::LR);
48  enm.value("UR", ReadoutCorner::UR);
49  enm.value("UL", ReadoutCorner::UL);
50  });
51  wrappers.wrapType(py::enum_<AssemblyState>(wrappers.module, "AssemblyState"), [](auto &mod, auto &enm) {
52  enm.value("RAW", AssemblyState::RAW);
53  enm.value("SCIENCE", AssemblyState::SCIENCE);
54  });
55  auto amplifier = wrappers.wrapType(PyAmplifier(wrappers.module, "Amplifier"), [](auto &mod, auto &cls) {
56  cls.def_static("getRecordSchema", &Amplifier::getRecordSchema);
57  cls.def("toRecord", &Amplifier::toRecord);
58  cls.def("rebuild", &Amplifier::rebuild);
59  cls.def("getName", &Amplifier::getName);
60  cls.def("getBBox", &Amplifier::getBBox);
61  cls.def("getGain", &Amplifier::getGain);
62  cls.def("getReadNoise", &Amplifier::getReadNoise);
63  cls.def("getSaturation", &Amplifier::getSaturation);
64  cls.def("getSuspectLevel", &Amplifier::getSuspectLevel);
65  cls.def("getReadoutCorner", &Amplifier::getReadoutCorner);
66  cls.def("getLinearityCoeffs", &Amplifier::getLinearityCoeffs);
67  cls.def("getLinearityType", &Amplifier::getLinearityType);
68  cls.def("getLinearityThreshold", &Amplifier::getLinearityThreshold);
69  cls.def("getLinearityMaximum", &Amplifier::getLinearityMaximum);
70  cls.def("getLinearityUnits", &Amplifier::getLinearityUnits);
71  cls.def("getRawBBox", &Amplifier::getRawBBox);
72  cls.def("getRawDataBBox", &Amplifier::getRawDataBBox);
73  cls.def("getRawFlipX", &Amplifier::getRawFlipX);
74  cls.def("getRawFlipY", &Amplifier::getRawFlipY);
75  cls.def("getRawXYOffset", &Amplifier::getRawXYOffset);
76  cls.def("getRawHorizontalOverscanBBox", &Amplifier::getRawHorizontalOverscanBBox);
77  cls.def("getRawVerticalOverscanBBox", &Amplifier::getRawVerticalOverscanBBox);
78  cls.def("getRawPrescanBBox", &Amplifier::getRawPrescanBBox);
79  cls.def("getRawSerialOverscanBBox", &Amplifier::getRawSerialOverscanBBox);
80  cls.def("getRawParallelOverscanBBox", &Amplifier::getRawParallelOverscanBBox);
81  cls.def("getRawSerialPrescanBBox", &Amplifier::getRawSerialPrescanBBox);
82  cls.def("getRawHorizontalPrescanBBox", &Amplifier::getRawHorizontalPrescanBBox);
83  });
84  wrappers.wrapType(PyAmplifierBuilder(amplifier, "Builder"), [](auto &mod, auto &cls) {
85  cls.def_static("fromRecord", &Amplifier::Builder::fromRecord);
86  cls.def(py::init());
87  cls.def("finish", &Amplifier::Builder::finish);
88  cls.def("assign", [](Amplifier::Builder &self, Amplifier const &other) { self = other; });
89  cls.def("setName", &Amplifier::Builder::setName, "name"_a);
90  cls.def("setBBox", &Amplifier::Builder::setBBox, "bbox"_a);
91  cls.def("setGain", &Amplifier::Builder::setGain, "gain"_a);
92  cls.def("setReadNoise", &Amplifier::Builder::setReadNoise, "readNoise"_a);
93  cls.def("setSaturation", &Amplifier::Builder::setSaturation, "saturation"_a);
94  cls.def("setSuspectLevel", &Amplifier::Builder::setSuspectLevel, "suspectLevel"_a);
95  cls.def("setReadoutCorner", &Amplifier::Builder::setReadoutCorner, "corner"_a);
96  cls.def("setLinearityCoeffs", &Amplifier::Builder::setLinearityCoeffs, "coeffs"_a);
97  // Backwards compatibility: accept std::vector (list in Python) in
98  // addition to ndarray::Array (np.ndarray)
99  cls.def("setLinearityCoeffs", [](Amplifier::Builder &self, std::vector<double> const &coeffs) {
100  ndarray::Array<double, 1, 1> array = ndarray::allocate(coeffs.size());
101  std::copy(coeffs.begin(), coeffs.end(), array.begin());
102  self.setLinearityCoeffs(array);
103  });
104  cls.def("setLinearityType", &Amplifier::Builder::setLinearityType, "type"_a);
105  cls.def("setLinearityThreshold", &Amplifier::Builder::setLinearityThreshold, "threshold"_a);
106  cls.def("setLinearityMaximum", &Amplifier::Builder::setLinearityMaximum, "maximum"_a);
107  cls.def("setLinearityUnits", &Amplifier::Builder::setLinearityUnits, "units"_a);
108  cls.def("setRawBBox", &Amplifier::Builder::setRawBBox, "bbox"_a);
109  cls.def("setRawDataBBox", &Amplifier::Builder::setRawDataBBox, "bbox"_a);
110  cls.def("setRawFlipX", &Amplifier::Builder::setRawFlipX, "rawFlipX"_a);
111  cls.def("setRawFlipY", &Amplifier::Builder::setRawFlipY, "rawFlipY"_a);
112  cls.def("setRawXYOffset", &Amplifier::Builder::setRawXYOffset, "offset"_a);
113  cls.def("setRawHorizontalOverscanBBox", &Amplifier::Builder::setRawHorizontalOverscanBBox, "bbox"_a);
114  cls.def("setRawVerticalOverscanBBox", &Amplifier::Builder::setRawVerticalOverscanBBox, "bbox"_a);
115  cls.def("setRawPrescanBBox", &Amplifier::Builder::setRawPrescanBBox, "bbox"_a);
116  cls.def("setRawSerialOverscanBBox", &Amplifier::Builder::setRawSerialOverscanBBox, "bbox"_a);
117  cls.def("setRawParallelOverscanBBox", &Amplifier::Builder::setRawParallelOverscanBBox, "bbox"_a);
118  cls.def("setRawSerialPrescanBBox", &Amplifier::Builder::setRawSerialPrescanBBox, "bbox"_a);
119  cls.def("setRawHorizontalPrescanBBox", &Amplifier::Builder::setRawHorizontalPrescanBBox, "bbox"_a);
120  });
121 }
122 
123 } // namespace cameraGeom
124 } // namespace afw
125 } // namespace lsst
ItemVariant const * other
Definition: Schema.cc:56
T begin(T... args)
A mutable Amplifier subclass class that can be used to incrementally construct or modify Amplifiers.
Definition: Amplifier.h:276
Geometry and electronic information about raw amplifier images.
Definition: Amplifier.h:86
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition: python.h:242
pybind11::module module
The module object passed to the PYBIND11_MODULE block that contains this WrapperCollection.
Definition: python.h:448
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
void addInheritanceDependency(std::string const &name)
Indicate an external module that provides a base class for a subsequent addType call.
Definition: python.h:343
T copy(T... args)
T end(T... args)
py::class_< Amplifier::Builder, Amplifier, std::shared_ptr< Amplifier::Builder > > PyAmplifierBuilder
Definition: _amplifier.cc:41
void wrapAmplifier(lsst::utils::python::WrapperCollection &wrappers)
Definition: _amplifier.cc:43
py::class_< Amplifier, std::shared_ptr< Amplifier > > PyAmplifier
Definition: _amplifier.cc:40
def init()
Definition: tests.py:59
A base class for image defects.
T size(T... args)