shapeUtilities.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/eigen.h"
 
#include <memory>
 
#include "ndarray/pybind11.h"
 
#include "lsst/afw/table/BaseRecord.h"
#include "lsst/meas/base/ShapeUtilities.h"
 
namespace py = pybind11;
using namespace pybind11::literals;
 
namespace lsst {
namespace meas {
namespace base {
 
namespace {
 
using PyShapeResult = py::class_<ShapeResult, std::shared_ptr<ShapeResult>>;
using PyShapeResultKey = py::class_<ShapeResultKey, std::shared_ptr<ShapeResultKey>>;
 
void declareShapeResult(py::module &mod) {
    PyShapeResult cls(mod, "ShapeResult");
 
    cls.def(py::init<>());
    cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ShapeCov const &>(), "xx"_a, "yy"_a, "xy"_a,
            "matrix"_a);
    cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ErrElement, ErrElement, ErrElement>(), "xx"_a,
            "yy"_a, "xy"_a, "xxErr"_a, "yyErr"_a, "xyErr"_a);
 
    cls.def("getShape", &ShapeResult::getShape);
    cls.def("getQuadrupole", &ShapeResult::getQuadrupole);
    cls.def("setShape", &ShapeResult::setShape, "shape"_a);
    cls.def("getShapeErr", &ShapeResult::getShapeErr);
    cls.def("setShapeErr", (void (ShapeResult::*)(ShapeCov const &)) & ShapeResult::setShapeErr, "matrix"_a);
    cls.def("setShapeErr",
            (void (ShapeResult::*)(ErrElement, ErrElement, ErrElement)) & ShapeResult::setShapeErr,
            "xxErr"_a, "yyErr"_a, "xyErr"_a);
 
    cls.def_readwrite("xx", &ShapeResult::xx);
    cls.def_readwrite("yy", &ShapeResult::yy);
    cls.def_readwrite("xy", &ShapeResult::xy);
    cls.def_readwrite("xxErr", &ShapeResult::xxErr);
    cls.def_readwrite("yyErr", &ShapeResult::yyErr);
    cls.def_readwrite("xyErr", &ShapeResult::xyErr);
    cls.def_readwrite("xx_yy_Cov", &ShapeResult::xx_yy_Cov);
    cls.def_readwrite("xx_xy_Cov", &ShapeResult::xx_xy_Cov);
    cls.def_readwrite("yy_xy_Cov", &ShapeResult::yy_xy_Cov);
}
 
void declareShapeResultKey(py::module &mod) {
    PyShapeResultKey cls(mod, "ShapeResultKey");
 
    cls.def_static("addFields", &ShapeResultKey::addFields, "schema"_a, "name"_a, "doc"_a, "uncertainty"_a,
                   "coordType"_a = afw::table::CoordinateType::PIXEL);
 
    cls.def(py::init<>());
    cls.def(py::init<afw::table::QuadrupoleKey const &,
                     afw::table::CovarianceMatrixKey<ErrElement, 3> const &>(),
            "shape"_a, "shapeErr"_a);
    cls.def(py::init<afw::table::SubSchema const &>(), "subSchema"_a);
 
    cls.def("__eq__", &ShapeResultKey::operator==, py::is_operator());
    cls.def("__ne__", &ShapeResultKey::operator!=, py::is_operator());
 
    cls.def("get", &ShapeResultKey::get, "record"_a);
    cls.def("set", &ShapeResultKey::set, "record"_a, "value"_a);
    cls.def("isValid", &ShapeResultKey::isValid);
    cls.def("getShape", &ShapeResultKey::getShape);
    cls.def("getShapeErr", &ShapeResultKey::getShapeErr);
    cls.def("getIxx", &ShapeResultKey::getIxx);
    cls.def("getIyy", &ShapeResultKey::getIyy);
    cls.def("getIxy", &ShapeResultKey::getIxy);
}
 
}  // namespace
 
PYBIND11_MODULE(shapeUtilities, mod) {
    py::module::import("lsst.afw.table");
 
    declareShapeResult(mod);
    declareShapeResultKey(mod);
 
    mod.def("makeShapeTransformMatrix", &makeShapeTransformMatrix, "xform"_a);
}
 
}  // namespace base
}  // namespace meas
}  // namespace lsst