interpolate.cc File Reference

#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include "ndarray/pybind11.h"
#include "lsst/afw/math/Interpolate.h"

Go to the source code of this file.

Functions
	PYBIND11_MODULE (interpolate, mod)

Function Documentation

PYBIND11_MODULE()

PYBIND11_MODULE	(	interpolate	,
		mod
	)

Definition at line 36 of file interpolate.cc.

                                  {
    py::class_<Interpolate, std::shared_ptr<Interpolate>> clsInterpolate(mod, "Interpolate");
    py::enum_<Interpolate::Style>(clsInterpolate, "Style")
            .value("UNKNOWN", Interpolate::Style::UNKNOWN)
            .value("CONSTANT", Interpolate::Style::CONSTANT)
            .value("LINEAR", Interpolate::Style::LINEAR)
            .value("NATURAL_SPLINE", Interpolate::Style::NATURAL_SPLINE)
            .value("CUBIC_SPLINE", Interpolate::Style::CUBIC_SPLINE)
            .value("CUBIC_SPLINE_PERIODIC", Interpolate::Style::CUBIC_SPLINE_PERIODIC)
            .value("AKIMA_SPLINE", Interpolate::Style::AKIMA_SPLINE)
            .value("AKIMA_SPLINE_PERIODIC", Interpolate::Style::AKIMA_SPLINE_PERIODIC)
            .value("NUM_STYLES", Interpolate::Style::NUM_STYLES)
            .export_values();

    clsInterpolate.def("interpolate", [](Interpolate &t, double const x) {
        /*
        We use a lambda function here because interpolate (with a double) is a virtual function and therefor
        cannot be wrapped directly.
        */
        return t.interpolate(x);
    });

    clsInterpolate.def("interpolate",
                       (std::vector<double> (Interpolate::*)(std::vector<double> const &) const) &
                               Interpolate::interpolate);
    clsInterpolate.def(
            "interpolate",
            (ndarray::Array<double, 1> (Interpolate::*)(ndarray::Array<double const, 1> const &) const) &
                    Interpolate::interpolate);

    mod.def("makeInterpolate",
            (std::shared_ptr<Interpolate>(*)(std::vector<double> const &, std::vector<double> const &,
                                             Interpolate::Style const))makeInterpolate,
            "x"_a, "y"_a, "style"_a = Interpolate::AKIMA_SPLINE);
    mod.def("makeInterpolate", (std::shared_ptr<Interpolate>(*)(ndarray::Array<double const, 1> const &,
                                                                ndarray::Array<double const, 1> const &y,
                                                                Interpolate::Style const))makeInterpolate,
            "x"_a, "y"_a, "style"_a = Interpolate::AKIMA_SPLINE);

    mod.def("stringToInterpStyle", stringToInterpStyle, "style"_a);
    mod.def("lookupMaxInterpStyle", lookupMaxInterpStyle, "n"_a);
    mod.def("lookupMinInterpPoints", lookupMinInterpPoints, "style"_a);
}