lsst.ip.isr  13.0-11-gf01aa92+14
 All Classes Namespaces Files Functions Variables Typedefs Macros Groups
applyLookupTable.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 "pybind11/pybind11.h"
23 #include "pybind11/stl.h"
24 
25 #include "numpy/arrayobject.h"
26 #include "ndarray/pybind11.h"
27 
29 
30 namespace py = pybind11;
31 using namespace pybind11::literals;
32 
33 namespace lsst {
34 namespace ip {
35 namespace isr {
36 
37 namespace {
38 
39 template <typename PixelT>
40 static void declareApplyLookupTable(py::module& mod) {
41  mod.def("applyLookupTable", &applyLookupTable<PixelT>, "image"_a, "table"_a, "indOffset"_a);
42 }
43 
44 } // namespace lsst::ip::isr::<anonymous>
45 
47  py::module mod("applyLookupTable");
48 
49  // Need to import numpy for ndarray and eigen conversions
50  if (_import_array() < 0) {
51  PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
52  return nullptr;
53  }
54 
55  declareApplyLookupTable<float>(mod);
56  declareApplyLookupTable<double>(mod);
57 
58  return mod.ptr();
59 }
60 
61 } // isr
62 } // ip
63 } // lsst
PYBIND11_PLUGIN(applyLookupTable)
int applyLookupTable(afw::image::Image< PixelT > &image, ndarray::Array< PixelT, 1, 1 > const &table, PixelT indOffset)
Add the values in a lookup table to an image, e.g.