如何从Python调用以complex<float>数组指针为参数的C++函数？

我需要从Python中调用一个以complex<float>数组指针为参数的C++函数，该函数定义如下：

extern "C" void myfunction(complex<float> *arr)

已将其编译为mylib.so库。

目前我使用的Python调用代码如下：

complex_array = np.array([1+2j, 2+3j, 3+4j], dtype=np.complex64)

mylibrary_dll = CDLL('./mylib.so')
mylibrary_c = mylibrary_dll.myfunction
mylibrary_c.argtype = [np.ctypeslib.ndpointer(np.complex64,ndim=1,flags='c')]

调用时使用：

mylibrary_c(complex_array.ctypes.data_as(POINTER(c_float)))

我不确定把float*强制转换为std::complex<float>*是否安全，想问问是不是应该改用下面这种方式？

complex_array = np.array([1+2j, 2+3j, 3+4j], dtype=np.complex64)

# 定义复数结构体
class Complex(ctypes.Structure):
    _fields_ = [("real", ctypes.c_float),
                ("imag", ctypes.c_float)]

mylibrary_dll = CDLL('./mylib.so')
mylibrary_c = mylibrary_dll.myfunction
mylibrary_c.argtype = [POINTER(Complex)] # 与上述方式不同

mylibrary_c(complex_array.ctypes.data_as(POINTER(Complex))) # 与上述方式不同

附C++中complex<float>的特化结构：

/// 26.2.3  complex specializations
/// complex<float> specialization
template<>
  struct complex<float>
  {
    typedef float value_type;
    typedef __complex__ float _ComplexT;

    _GLIBCXX_CONSTEXPR complex(_ComplexT __z) : _M_value(__z) { }

    _GLIBCXX_CONSTEXPR complex(float __r = 0.0f, float __i = 0.0f)
#if __cplusplus >= 201103L
    : _M_value{ __r, __i } { }
#else
    {
  __real__ _M_value = __r;
  __imag__ _M_value = __i;
    }
#endif

两种方式的安全性分析 #

从你提供的std::complex<float>特化代码来看，它内部采用GCC的C语言复数类型__complex__ float存储，而C语言的_Complex float、numpy的complex64、你定义的Complex结构体在内存布局上完全兼容：都是连续的两个float，先实部后虚部。所以两种转换方式在当前GCC实现下都是安全的，但各有优劣：

1. 第一种方式（转POINTER(c_float)）

   • 核心逻辑是利用std::complex<float>的内存布局等价于两个连续float的特性，复数数组可直接视为长度翻倍的float数组。
   • 存在小问题：你设置的argtype是np.ctypeslib.ndpointer(np.complex64,...)，但调用时传入的是POINTER(c_float)，类型不匹配，可能触发ctypes的类型检查报错。建议要么移除argtype设置，要么将其改为POINTER(c_float)。

2. 第二种方式（自定义Complex结构体）

   • 这种写法更直观，直接对应复数的内存结构，与std::complex<float>的布局完全对齐，ctypes的类型检查能正常工作，代码可读性也更强。
   • 这是更推荐的写法，因为它明确表达了参数类型，减少了隐式转换带来的认知负担，更符合ctypes的使用规范。

推荐的最终代码 #

import numpy as np
import ctypes
from ctypes import CDLL, POINTER, Structure, c_float

complex_array = np.array([1+2j, 2+3j, 3+4j], dtype=np.complex64)

# 定义与std::complex<float>内存布局一致的结构体
class Complex(Structure):
    _fields_ = [("real", c_float),
                ("imag", c_float)]

# 加载库并配置函数
mylibrary_dll = CDLL('./mylib.so')
mylibrary_c = mylibrary_dll.myfunction
# 明确指定参数类型为Complex指针
mylibrary_c.argtypes = [POINTER(Complex)]

# 转换并调用
mylibrary_c(complex_array.ctypes.data_as(POINTER(Complex)))

关键注意事项 #

• 确保numpy数组的dtype为np.complex64（对应C++的complex<float>），若使用complex128需对应complex<double>。
• 数组必须是C连续布局（numpy默认是C连续，也可通过flags='c'强制确保），避免内存不连续导致的错误。
• 即使更换为MSVC等其他编译器，std::complex<float>的内存布局也会与C语言复数类型兼容（C++标准强制要求），因此该写法具备可移植性。

内容的提问来源于stack exchange，提问作者Laut567