shapeletFunction.cc

Go to the documentation of this file.

/*
 * LSST Data Management System
 *
 * This product includes software developed by the
 * LSST Project (http://www.lsst.org/).
 * See the COPYRIGHT file
 *
 * 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 "ndarray/pybind11.h"

#include "lsst/shapelet/ShapeletFunction.h"

namespace py = pybind11;
using namespace pybind11::literals;

namespace lsst {
namespace shapelet {

PYBIND11_MODULE(shapeletFunction, mod) {
    py::module::import("lsst.geom");
    py::module::import("lsst.afw.geom");
    py::module::import("lsst.afw.image");

    py::class_<ShapeletFunction, std::shared_ptr<ShapeletFunction>> clsShapeletFunction(mod,
                                                                                        "ShapeletFunction");

    clsShapeletFunction.def_readonly_static("FLUX_FACTOR", &ShapeletFunction::FLUX_FACTOR);

    clsShapeletFunction.def(py::init<int, BasisTypeEnum>(), "order"_a, "basisType"_a);
    clsShapeletFunction.def(py::init<int, BasisTypeEnum, ndarray::Array<double, 1, 1> const &>(), "order"_a,
                            "basisType"_a, "coefficients"_a);
    clsShapeletFunction.def(py::init<int, BasisTypeEnum, double, geom::Point2D const &>(), "order"_a,
                            "basisType"_a, "radius"_a, "center"_a = geom::Point2D());
    clsShapeletFunction.def(py::init<int, BasisTypeEnum, double, geom::Point2D const &,
                                     ndarray::Array<double, 1, 1> const &>(),
                            "order"_a, "basisType"_a, "radius"_a, "center"_a, "coefficients"_a);
    clsShapeletFunction.def(py::init<int, BasisTypeEnum, afw::geom::ellipses::Ellipse const &>(), "order"_a,
                            "basisType"_a, "ellipse"_a);
    clsShapeletFunction.def(py::init<int, BasisTypeEnum, afw::geom::ellipses::Ellipse const &,
                                     ndarray::Array<double const, 1, 1> const &>(),
                            "order"_a, "basisType"_a, "ellipse"_a, "coefficients"_a);
    clsShapeletFunction.def(py::init<ShapeletFunction>());

    clsShapeletFunction.def("getOrder", &ShapeletFunction::getOrder);
    clsShapeletFunction.def("getEllipse", (afw::geom::ellipses::Ellipse & (ShapeletFunction::*)()) &
                                                  ShapeletFunction::getEllipse,
                            py::return_value_policy::reference_internal);
    clsShapeletFunction.def("setEllipse", &ShapeletFunction::setEllipse);
    clsShapeletFunction.def("getBasisType", &ShapeletFunction::getBasisType);
    clsShapeletFunction.def("changeBasisType", &ShapeletFunction::changeBasisType);
    clsShapeletFunction.def("normalize", &ShapeletFunction::normalize, "value"_a = 1.0);
    clsShapeletFunction.def("getCoefficients", (ndarray::Array<double, 1, 1> const (ShapeletFunction::*)()) &
                                                       ShapeletFunction::getCoefficients);
    clsShapeletFunction.def("convolve", &ShapeletFunction::convolve);
    clsShapeletFunction.def("evaluate", &ShapeletFunction::evaluate);
    clsShapeletFunction.def("shiftInPlace", &ShapeletFunction::shiftInPlace);
    clsShapeletFunction.def("transformInPlace", &ShapeletFunction::transformInPlace);

    py::class_<ShapeletFunctionEvaluator, std::shared_ptr<ShapeletFunctionEvaluator>>
            clsShapeletFunctionEvaluator(mod, "ShapeletFunctionEvaluator");

    clsShapeletFunctionEvaluator.def(py::init<ShapeletFunction const &>(), "function"_a);

    clsShapeletFunctionEvaluator.def("__call__",
                                     (double (ShapeletFunctionEvaluator::*)(double, double) const) &
                                             ShapeletFunctionEvaluator::operator());
    clsShapeletFunctionEvaluator.def(
            "__call__", (double (ShapeletFunctionEvaluator::*)(geom::Point2D const &) const) &
                                ShapeletFunctionEvaluator::operator());
    clsShapeletFunctionEvaluator.def(
            "__call__", (double (ShapeletFunctionEvaluator::*)(geom::Extent2D const &) const) &
                                ShapeletFunctionEvaluator::operator());
    clsShapeletFunctionEvaluator.def("__call__", (ndarray::Array<double, 1, 1> (ShapeletFunctionEvaluator::*)(
                                                         ndarray::Array<double const, 1> const &,
                                                         ndarray::Array<double const, 1> const &) const) &
                                                         ShapeletFunctionEvaluator::operator());

    clsShapeletFunctionEvaluator.def(
            "addToImage", (void (ShapeletFunctionEvaluator::*)(ndarray::Array<double, 2, 1> const &,
                                                               geom::Point2I const &) const) &
                                  ShapeletFunctionEvaluator::addToImage,
            "array"_a, "xy0"_a = geom::Point2D());
    clsShapeletFunctionEvaluator.def(
            "addToImage", (void (ShapeletFunctionEvaluator::*)(afw::image::Image<double> &) const) &
                                  ShapeletFunctionEvaluator::addToImage,
            "image"_a);
    clsShapeletFunctionEvaluator.def("integrate", &ShapeletFunctionEvaluator::integrate);
    clsShapeletFunctionEvaluator.def("computeMoments", &ShapeletFunctionEvaluator::computeMoments);
    clsShapeletFunctionEvaluator.def("update", &ShapeletFunctionEvaluator::update);
}

}  // shapelet
}  // lsst