python/test/numpy/ufunc.cpp at python313 · boostorg/python · GitHub

36 lines (29 loc) · 965 Bytes

// Copyright Jim Bosch & Ankit Daftery 2010-2012.

// Copyright Stefan Seefeld 2016.

// Distributed under the Boost Software License, Version 1.0.

// (See accompanying file LICENSE_1_0.txt or copy at

// http://www.boost.org/LICENSE_1_0.txt)

#include <boost/python/numpy.hpp>

namespace p = boost::python;

namespace np = boost::python::numpy;

struct UnaryCallable

{

typedef double argument_type;

typedef double result_type;

double operator()(double r) const { return r * 2;}

};

struct BinaryCallable

{

typedef double first_argument_type;

typedef double second_argument_type;

typedef double result_type;

double operator()(double a, double b) const { return a * 2 + b * 3;}

};

BOOST_PYTHON_MODULE(ufunc_ext)

{

np::initialize();

p::class_<UnaryCallable>("UnaryCallable")

.def("__call__", np::unary_ufunc<UnaryCallable>::make());

p::class_< BinaryCallable>("BinaryCallable")

.def("__call__", np::binary_ufunc<BinaryCallable>::make());

}

Expand file tree