Bind a serial eval_func (called do)

Author: robertodr

ompybind11/ompybind11.cpp

@@ -38,7 +38,9 @@ double calc_pi(int n) {
 };
 
 PYBIND11_MODULE(ompybind11, m) {
-  m.def("eval_func", &eval_func), py::arg("rank"), py::arg("function"),
+  m.def("eval_func", &eval_func, py::arg("rank"), py::arg("function"),
         py::arg("x"))
+      .def("do", [](std::function<double(double)> f, double x) { return f(x); },
+           py::arg("function"), py::arg("x"))
       .def("calc_pi", &calc_pi);
 }

ompybind11/test.py

@@ -2,19 +2,22 @@
 import sys
 
 sys.path.append(os.getenv('OMPYBIND11_MODULE_PATH'))
-import ompybind11 # isort:skip
+import ompybind11  # isort:skip
 
+done = ompybind11.do(function=(lambda x: x + 40.0), x=2.0)
+print('ompybind11.do(function=(lambda x: x + 40.0), x=2.0) returned {}'.format(done))
 
 def f(x):
-    return x*x
+    return x * x
+
 
 x = 2.0
 
 pi = ompybind11.calc_pi(100)
 print('pi is {}'.format(pi))
 
-foo = ompybind11.eval_func(0, f, x)
+foo = ompybind11.eval_func(rank=0, function=f, x=x)
 print('Output of eval_func {}\n'.format(foo))
 
-bar = ompybind11.eval_func(1, f, x)
+bar = ompybind11.eval_func(rank=1, function=f, x=x)
 print('Output of eval_func {}\n'.format(bar))