add python complex interface

solvcon · Jan 2, 2025 · c2525fe · c2525fe
1 parent 8e0586c
commit c2525fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/cpp/modmesh/mathtype/pymod/mathtype_pymod.hpp b/cpp/modmesh/mathtype/pymod/mathtype_pymod.hpp
@@ -27,7 +27,6 @@
  */
 
 #include <pybind11/pybind11.h>
-#include <pybind11/stl.h>
 
 #include <modmesh/python/common.hpp>
 

diff --git a/cpp/modmesh/mathtype/pymod/wrap_Complex.cpp b/cpp/modmesh/mathtype/pymod/wrap_Complex.cpp
@@ -74,7 +74,10 @@ class MODMESH_PYTHON_WRAPPER_VISIBILITY WrapComplex
             .def(py::self -= py::self) // NOLINT(misc-redundant-expression)
             .def_property_readonly("real", &wrapped_type::real)
             .def_property_readonly("imag", &wrapped_type::imag)
-            .def("norm", &wrapped_type::norm);
+            .def("norm", &wrapped_type::norm)
+            .def("__complex__",
+                 [](wrapped_type const & self)
+                 { return std::complex<T>(self.real(), self.imag()); });
     }
 
 }; /* end class WrapComplex */
-Original file line number
+Diff line change
@@ Expand Up / @@ -27,7 +27,6 @@ @@
      */
     #include <pybind11/pybind11.h>
-    #include <pybind11/stl.h>
     #include <modmesh/python/common.hpp>
@@ Expand Down @@