Add python bindings for traffic

1 files changed, 29 insertions, 0 deletions

diff --git a/pyhelpers/pair.hpp b/pyhelpers/pair.hpp
new file mode 100644
index 0000000000..54155aed4c
--- /dev/null
+++ b/pyhelpers/pair.hpp
@@ -0,0 +1,29 @@
+#pragma once
+
+#include "std/utility.hpp"
+
+#include <boost/python.hpp>
+#include <boost/python/suite/indexing/vector_indexing_suite.hpp>
+
+namespace
+{
+using namespace boost::python;
+
+// Converts a std::pair instance to a Python tuple.
+template <typename T1, typename T2>
+struct pair_to_tuple
+{
+  static PyObject * convert(pair<T1, T2> const & p)
+  {
+    return incref(make_tuple(p.first, p.second).ptr());
+  }
+  
+  static PyTypeObject const * get_pytype() { return &PyTuple_Type; }
+};
+
+template <typename T1, typename T2>
+struct pair_to_python_converter
+{
+  pair_to_python_converter() { to_python_converter<pair<T1, T2>, pair_to_tuple<T1, T2>, true>(); }
+};
+}  // namespace