blob: 3a9d43cbbeb96a50cc978af14179e5c06869e526 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
// 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());
}
|
