OilfieldToolsNet/opm-common
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python numpy array conversion function in namespace conv ert.

Browse Source
This commit is contained in:
Steinar Foss
2019-10-25 11:06:12 +02:00
parent ce3781775f
commit 02cc292f38
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/cxx/converters.hpp
View File

@@ -21,9 +21,10 @@ std::string str( const T& t ) {
return stream.str();
}
namespace convert {
template <class T>
std::vector<T> numpy_array_to_vector(py::array_t<T>& input) {
std::vector<T> vector(py::array_t<T>& input) {
T * input_ptr = (T *) input.request().ptr;
std::vector<T> output(input.size());
@@ -35,7 +36,7 @@ std::vector<T> numpy_array_to_vector(py::array_t<T>& input) {
template <class T>
py::array_t<T> vector_to_numpy_array(const std::vector<T>& input) {
py::array_t<T> numpy_array(const std::vector<T>& input) {
auto output = py::array_t<T>(input.size());
T * py_array_ptr = (T*)output.request().ptr;
@@ -45,4 +46,6 @@ py::array_t<T> vector_to_numpy_array(const std::vector<T>& input) {
return output;
}
}
#endif //SUNBEAM_CONVERTERS_HPP