forked from xtensor-stack/xtensor-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyvectorize.hpp
More file actions
55 lines (45 loc) · 1.73 KB
/
Copy pathpyvectorize.hpp
File metadata and controls
55 lines (45 loc) · 1.73 KB
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
/***************************************************************************
* Copyright (c) 2016, Johan Mabille and Sylvain Corlay *
* *
* Distributed under the terms of the BSD 3-Clause License. *
* *
* The full license is in the file LICENSE, distributed with this software. *
****************************************************************************/
#ifndef PY_VECTORIZE_HPP
#define PY_VECTORIZE_HPP
#include "pyarray.hpp"
#include "xtensor/xvectorize.hpp"
namespace xt
{
template <class Func, class R, class... Args>
struct pyvectorizer
{
xvectorizer<Func, R> m_vectorizer;
template <class F>
pyvectorizer(F&& func)
: m_vectorizer(std::forward<F>(func))
{
}
pybind11::object operator()(pyarray<Args, pybind_array::c_style | pybind_array::forcecast>... args)
{
pyarray<R> res = m_vectorizer(args...);
return res;
}
};
template <class R, class... Args>
inline pyvectorizer<R(*)(Args...), R, Args...> pyvectorize(R(*f) (Args...))
{
return pyvectorizer<R(*) (Args...), R, Args...>(f);
}
template <class F, class R, class... Args>
inline pyvectorizer<F, R, Args...> pyvectorize(F&& f, R(*) (Args...))
{
return pyvectorizer<F, R, Args...>(std::forward<F>(f));
}
template <class F>
inline auto pyvectorize(F&& f) -> decltype(pyvectorize(std::forward<F>(f), (detail::get_function_type<F>*)nullptr))
{
return pyvectorize(std::forward<F>(f), (detail::get_function_type<F>*)nullptr);
}
}
#endif