amplifier.cc

Go to the documentation of this file.

/*
 * LSST Data Management System
 * Copyright 2008-2017  AURA/LSST.
 *
 * This product includes software developed by the
 * LSST Project (http://www.lsst.org/).
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the LSST License Statement and
 * the GNU General Public License along with this program.  If not,
 * see <https://www.lsstcorp.org/LegalNotices/>.
 */
 
#include "pybind11/pybind11.h"
#include "pybind11/stl.h"
 
#include "ndarray/pybind11.h"
 
#include "lsst/afw/table/BaseRecord.h"
#include "lsst/afw/cameraGeom/Amplifier.h"
 
namespace py = pybind11;
using namespace pybind11::literals;
 
namespace lsst {
namespace afw {
namespace cameraGeom {
namespace {
 
using PyAmplifier = py::class_<Amplifier, std::shared_ptr<Amplifier>>;
using PyAmplifierBuilder = py::class_<Amplifier::Builder, Amplifier, std::shared_ptr<Amplifier::Builder>>;
 
PyAmplifier declarePyAmplifier(py::module & mod) {
    py::enum_<ReadoutCorner>(mod, "ReadoutCorner")
        .value("LL", ReadoutCorner::LL)
        .value("LR", ReadoutCorner::LR)
        .value("UR", ReadoutCorner::UR)
        .value("UL", ReadoutCorner::UL);
    py::enum_<AssemblyState>(mod, "AssemblyState")
        .value("RAW", AssemblyState::RAW)
        .value("SCIENCE", AssemblyState::SCIENCE);
    PyAmplifier cls(mod, "Amplifier");
    cls.def_static("getRecordSchema", &Amplifier::getRecordSchema);
    cls.def("toRecord", &Amplifier::toRecord);
    cls.def("rebuild", &Amplifier::rebuild);
    cls.def("getName", &Amplifier::getName);
    cls.def("getBBox", &Amplifier::getBBox);
    cls.def("getGain", &Amplifier::getGain);
    cls.def("getReadNoise", &Amplifier::getReadNoise);
    cls.def("getSaturation", &Amplifier::getSaturation);
    cls.def("getSuspectLevel", &Amplifier::getSuspectLevel);
    cls.def("getReadoutCorner", &Amplifier::getReadoutCorner);
    cls.def("getLinearityCoeffs", &Amplifier::getLinearityCoeffs);
    cls.def("getLinearityType", &Amplifier::getLinearityType);
    cls.def("getLinearityThreshold", &Amplifier::getLinearityThreshold);
    cls.def("getLinearityMaximum", &Amplifier::getLinearityMaximum);
    cls.def("getLinearityUnits", &Amplifier::getLinearityUnits);
    cls.def("getRawBBox", &Amplifier::getRawBBox);
    cls.def("getRawDataBBox", &Amplifier::getRawDataBBox);
    cls.def("getRawFlipX", &Amplifier::getRawFlipX);
    cls.def("getRawFlipY", &Amplifier::getRawFlipY);
    cls.def("getRawXYOffset", &Amplifier::getRawXYOffset);
    cls.def("getRawHorizontalOverscanBBox", &Amplifier::getRawHorizontalOverscanBBox);
    cls.def("getRawVerticalOverscanBBox", &Amplifier::getRawVerticalOverscanBBox);
    cls.def("getRawPrescanBBox", &Amplifier::getRawPrescanBBox);
    cls.def("getRawSerialOverscanBBox", &Amplifier::getRawSerialOverscanBBox);
    cls.def("getRawParallelOverscanBBox", &Amplifier::getRawParallelOverscanBBox);
    cls.def("getRawSerialPrescanBBox", &Amplifier::getRawSerialPrescanBBox);
    cls.def("getRawHorizontalPrescanBBox", &Amplifier::getRawHorizontalPrescanBBox);
    return cls;
}
 
void declarePyAmplifierBuilder(PyAmplifier & parent) {
    PyAmplifierBuilder cls(parent, "Builder");
    cls.def_static("fromRecord", &Amplifier::Builder::fromRecord);
    cls.def(py::init());
    cls.def("finish", &Amplifier::Builder::finish);
    cls.def("assign", [](Amplifier::Builder & self, Amplifier const & other) { self = other; });
    cls.def("setName", &Amplifier::Builder::setName, "name"_a);
    cls.def("setBBox", &Amplifier::Builder::setBBox, "bbox"_a);
    cls.def("setGain", &Amplifier::Builder::setGain, "gain"_a);
    cls.def("setReadNoise", &Amplifier::Builder::setReadNoise, "readNoise"_a);
    cls.def("setSaturation", &Amplifier::Builder::setSaturation, "saturation"_a);
    cls.def("setSuspectLevel", &Amplifier::Builder::setSuspectLevel, "suspectLevel"_a);
    cls.def("setReadoutCorner", &Amplifier::Builder::setReadoutCorner, "corner"_a);
    cls.def("setLinearityCoeffs", &Amplifier::Builder::setLinearityCoeffs, "coeffs"_a);
    // Backwards compatibility: accept std::vector (list in Python) in
    // addition to ndarray::Array (np.ndarray)
    cls.def("setLinearityCoeffs",
            [](Amplifier::Builder & self, std::vector<double> const & coeffs) {
                ndarray::Array<double, 1, 1> array = ndarray::allocate(coeffs.size());
                std::copy(coeffs.begin(), coeffs.end(), array.begin());
                self.setLinearityCoeffs(array);
            });
    cls.def("setLinearityType", &Amplifier::Builder::setLinearityType, "type"_a);
    cls.def("setLinearityThreshold", &Amplifier::Builder::setLinearityThreshold, "threshold"_a);
    cls.def("setLinearityMaximum", &Amplifier::Builder::setLinearityMaximum, "maximum"_a);
    cls.def("setLinearityUnits", &Amplifier::Builder::setLinearityUnits, "units"_a);
    cls.def("setRawBBox", &Amplifier::Builder::setRawBBox, "bbox"_a);
    cls.def("setRawDataBBox", &Amplifier::Builder::setRawDataBBox, "bbox"_a);
    cls.def("setRawFlipX", &Amplifier::Builder::setRawFlipX, "rawFlipX"_a);
    cls.def("setRawFlipY", &Amplifier::Builder::setRawFlipY, "rawFlipY"_a);
    cls.def("setRawXYOffset", &Amplifier::Builder::setRawXYOffset, "offset"_a);
    cls.def("setRawHorizontalOverscanBBox", &Amplifier::Builder::setRawHorizontalOverscanBBox, "bbox"_a);
    cls.def("setRawVerticalOverscanBBox", &Amplifier::Builder::setRawVerticalOverscanBBox, "bbox"_a);
    cls.def("setRawPrescanBBox", &Amplifier::Builder::setRawPrescanBBox, "bbox"_a);
    cls.def("setRawSerialOverscanBBox", &Amplifier::Builder::setRawSerialOverscanBBox, "bbox"_a);
    cls.def("setRawParallelOverscanBBox", &Amplifier::Builder::setRawParallelOverscanBBox, "bbox"_a);
    cls.def("setRawSerialPrescanBBox", &Amplifier::Builder::setRawSerialPrescanBBox, "bbox"_a);
    cls.def("setRawHorizontalPrescanBBox", &Amplifier::Builder::setRawHorizontalPrescanBBox, "bbox"_a);
}
 
PYBIND11_MODULE(amplifier, mod) {
    auto cls = declarePyAmplifier(mod);
    declarePyAmplifierBuilder(cls);
}
 
}  // namespace
}  // namespace cameraGeom
}  // namespace afw
}  // namespace lsst