netgen/libsrc/general/ngpython.hpp

74 lines
1.6 KiB
C++
Raw Normal View History

2014-10-08 20:09:03 +06:00
#ifdef NG_PYTHON
2016-11-04 16:14:52 +05:00
#include <pybind11/pybind11.h>
#include <pybind11/operators.h>
#include <pybind11/numpy.h>
2019-02-09 00:24:54 +05:00
#include <pybind11/stl.h>
2016-11-04 16:14:52 +05:00
namespace py = pybind11;
2014-10-08 21:48:01 +06:00
#include <iostream>
2016-11-04 16:14:52 +05:00
#include <sstream>
2014-10-08 20:09:03 +06:00
template <typename T>
py::array MoveToNumpy(std::vector<T>& vec)
{
auto newvec = new std::vector<T>();
std::swap(*newvec, vec);
auto capsule = py::capsule(newvec, [](void *v) { delete reinterpret_cast<std::vector<T>*>(v); });
return py::array(newvec->size(), newvec->data(), capsule);
}
namespace PYBIND11_NAMESPACE {
2016-11-04 16:14:52 +05:00
template<typename T>
bool CheckCast( py::handle obj ) {
try{
obj.cast<T>();
return true;
}
catch (py::cast_error &e) {
return false;
}
}
template <typename T>
struct extract
2014-10-08 20:09:03 +06:00
{
2016-11-04 16:14:52 +05:00
py::handle obj;
extract( py::handle aobj ) : obj(aobj) {}
2014-10-08 20:09:03 +06:00
2016-11-04 16:14:52 +05:00
bool check() { return CheckCast<T>(obj); }
T operator()() { return obj.cast<T>(); }
};
}
struct NGDummyArgument {};
inline void NOOP_Deleter(void *) { ; }
namespace netgen
{
2014-10-08 21:48:01 +06:00
2014-10-08 20:09:03 +06:00
//////////////////////////////////////////////////////////////////////
// Lambda to function pointer conversion
template <typename Function>
struct function_traits
: public function_traits<decltype(&Function::operator())> {};
template <typename ClassType, typename ReturnType, typename... Args>
struct function_traits<ReturnType(ClassType::*)(Args...) const> {
typedef ReturnType (*pointer)(Args...);
typedef ReturnType return_type;
};
template <typename Function>
typename function_traits<Function>::pointer
FunctionPointer (const Function& lambda) {
return static_cast<typename function_traits<Function>::pointer>(lambda);
}
2019-01-03 19:54:50 +05:00
} // namespace netgen
2014-10-08 20:09:03 +06:00
#endif