A cleaner interface for bindings.

1 files changed, 24 insertions, 0 deletions

diff --git a/pyhelpers/vector_list_conversion.hpp b/pyhelpers/vector_list_conversion.hpp
new file mode 100644
index 0000000000..44ecfa24d3
--- /dev/null
+++ b/pyhelpers/vector_list_conversion.hpp
@@ -0,0 +1,24 @@
+#pragma once
+
+#include "std/vector.hpp"
+
+#include <boost/python.hpp>
+
+namespace
+{
+using namespace boost::python;
+
+template <typename T>
+inline vector<T> python_list_to_std_vector(boost::python::object const & iterable)
+{
+  return vector<T>(boost::python::stl_input_iterator<T>(iterable),
+                   boost::python::stl_input_iterator<T>());
+}
+
+template <typename T>
+boost::python::list std_vector_to_python_list(vector<T> const & v)
+{
+  boost::python::object get_iter = boost::python::iterator<vector<T>>();
+  return boost::python::list(get_iter(v));
+}
+}  // namespace