From b438293c9cf41f4dcfcad5c3086a7eb7d79b4504 Mon Sep 17 00:00:00 2001
From: "Ralf W. Grosse-Kunstleve" <rwgk@yahoo.com>
Date: Wed, 21 Feb 2001 08:01:54 +0000
Subject: [PATCH] Mapping of std::complex<T> <-> Python complex.

[SVN r9305]
---
 include/boost/python/conversions.hpp | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/include/boost/python/conversions.hpp b/include/boost/python/conversions.hpp
index f7c07327..c72260ac 100644
--- a/include/boost/python/conversions.hpp
+++ b/include/boost/python/conversions.hpp
@@ -16,6 +16,7 @@
 # include <boost/smart_ptr.hpp>
 # include <boost/python/errors.hpp>
 # include <string>
+# include <complex>
 
 BOOST_PYTHON_BEGIN_CONVERSION_NAMESPACE // this is a gcc 2.95.2 bug workaround
 
@@ -134,6 +135,32 @@ PyObject* to_python(const std::string& s);
 std::string from_python(PyObject*, boost::python::type<std::string>);
 std::string from_python(PyObject*, boost::python::type<const std::string&>);
 
+template <class T>
+PyObject* to_python(const std::complex<T>& sc) {
+    Py_complex pcc;
+    pcc.real = sc.real();
+    pcc.imag = sc.imag();
+    return PyComplex_FromCComplex(pcc);
+}
+
+template <class T>
+std::complex<T> from_python(PyObject* p,
+                            boost::python::type<const std::complex<T>&>) {
+    if (! PyComplex_Check(p)) {
+      PyErr_SetString(PyExc_TypeError, "expected a complex number");
+      throw boost::python::argument_error();
+    }
+    return std::complex<T>(
+      static_cast<T>(PyComplex_RealAsDouble(p)),
+      static_cast<T>(PyComplex_ImagAsDouble(p)));
+}
+
+template <class T>
+inline std::complex<T> from_python(PyObject* p,
+                                   boost::python::type<std::complex<T> >) {
+    return from_python(p, boost::python::type<const std::complex<T>&>());
+}
+
 // For when your C++ function really wants to pass/return a PyObject*
 PyObject* to_python(PyObject*);
 PyObject* from_python(PyObject*, boost::python::type<PyObject*>);