如何在 SWIG 中处理 void* 参数

在使用 SWIG 将 C++ 代码绑定到 Python 时，遇到一个问题：如何处理 void* 参数。void* 是一个通用的指针类型，可以指向任意类型的数据。在 C++ 中，void* 参数通常用于传递各种类型的对象，而这些对象在 Python 中往往是不同的类型。因此，在 SWIG 中需要一种方法来将 void* 参数映射到相应的 Python 类型。

举个例子：假如在 C++ 代码中有两个结构体 MY_STRUCT 和 ANOTHER_STRUCT，它们都包含一个 void* 成员变量 pParameter。在 Python 中，希望能够将 MY_STRUCT 的 pParameter 成员变量设置为一个字符串或另一个 ANOTHER_STRUCT 对象。

2、解决方案

SWIG 提供了一种方法来处理 void* 参数，即使用 %typemap(in) 和 %typemap(memberin) 修饰符。%typemap(in) 修饰符用于指定如何将 Python 对象转换为 C++ 中的 void* 参数，而 %typemap(memberin) 修饰符用于指定如何将 C++ 中的 void* 成员变量转换为 Python 对象。

以下是一个示例代码，展示了如何使用 %typemap(in) 和 %typemap(memberin) 修饰符来处理 void* 参数：

%module test

%typemap(in) void* pParameter (int res=0, void *other_struct=NULL) %{
  int len;
  res = SWIG_ConvertPtr($input, &other_struct, $descriptor(struct ANOTHER_STRUCT*), 0);
  if (SWIG_IsOK(res)) {
    fprintf(stderr, "struct\n");
    $1 = reinterpret_cast< ANOTHER_STRUCT * >(argp1);
    len = sizeof(ANOTHER_STRUCT);
  }
  else if (PyString_Check($input)) {
    $1 = PyString_AsString($input);
    len = strlen((const char*)$1);
    fprintf(stderr, "string\n");
  }
  //else if (...) {
  //}
  else {
    SWIG_exception_fail(SWIG_TypeError, "some more details, see special typemap variables docs for ideas");
  }
%}

%typemap(memberin) void* pParameter %{
  $1 = $input; //fobar $self
  $self->pLen = len;
%}

%inline %{
typedef struct MY_STRUCT {
  void*       pParameter;
  unsigned long pLen;
} MY_STRUCT;

typedef struct ANOTHER_STRUCT {
} ANOTHER_STRUCT;
%}

%extend MY_STRUCT
{
    MY_STRUCT()
    {
        MY_STRUCT *m= new MY_STRUCT;
        m->pParameter = NULL;
        m->pLen = 0;
        return m;
    }
}

在上面的代码中，%typemap(in) 修饰符用于指定如何将 Python 对象转换为 C++ 中的 void* 参数。它首先尝试将 Python 对象转换为 ANOTHER_STRUCT 对象，如果成功，则将 ANOTHER_STRUCT 对象的地址赋给 void* 参数。如果 Python 对象不是 ANOTHER_STRUCT 对象，则尝试将其转换为字符串，如果成功，则将字符串的地址赋给 void* 参数。

%typemap(memberin) 修饰符用于指定如何将 C++ 中的 void* 成员变量转换为 Python 对象。它首先检查 void* 成员变量是否指向 ANOTHER_STRUCT 对象，如果是，则将 ANOTHER_STRUCT 对象转换为 Python 对象。如果不是，则将 void* 成员变量转换为字符串，然后将字符串转换为 Python 对象。

使用上面的代码，就可以在 Python 中将 MY_STRUCT 的 pParameter 成员变量设置为一个字符串或另一个 ANOTHER_STRUCT 对象。例如：

s = MY_STRUCT()
another = ANOTHER_STRUCT()
s.pParameter = another
s.pParameter = "some string"

这样，就可以在 Python 中处理 void* 参数了。