SWIG void *参数

Question

我的swig .i文件中有两个结构（实际来自第三方lib），遵循以下格式：

typedef struct MY_STRUCT {
  void*       pParameter;
  unsigned long pLen;
} MY_STRUCT;
%extend MY_STRUCT
{
    MY_STRUCT()
    {
        MY_STRUCT *m= new MY_STRUCT();
        m->pParameter = NULL;
        m->pLen = 0;
        return m;
    }
}

typedef struct ANOTHER_STRUCT {
    char * another;
    unsigned long len;
} ANOTHER_STRUCT;
%extend ANOTHER_STRUCT
{
    ANOTHER_STRUCT()
    {
        ANOTHER_STRUCT *p= new ANOTHER_STRUCT();
        p->another = NULL;
        p->len = 0;
        return p;
    }
    unsigned long __len__()
    {
        return sizeof(ANOTHER_STRUCT);
    }
}

MY_STRUCT中的pParameter是一个void *，因为它可以是char *或指向struct的指针（例如ANOTHER_STRUCT）。使用%typemap(in) void* = char*;处理char *映射很简单，但任何使用struct的尝试都会失败。这是我想在Python中看到的内容：

s = MY_STRUCT()
another = ANOTHER_STRUCT()
s.pParameter = another # this should pass the struct pointer
s.pParameter = "some string" # or a char pointer if this is provided

这可能吗？如果没有，我是否需要声明一些辅助函数来分配指针值？

Answer 1

您可以使用SWIG执行此操作。您最终会编写一个fairly large if来选择如何处理Python输入，具体取决于您要处理的类型数量。以下是您需要的两个类型映射的完整示例：

%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;
    }
}

这里in typemap采用PyObject并弄清楚如何处理它。它还将长度保存到局部变量中。

memberin typemap使用它来设置void*和长度。

您可能需要考虑使pLen不可变（请参阅：%immutable），以便人们无法随意更改该指针。您还需要仔细考虑类似这样的类型映射的内存所有权语义，以避免泄漏或双重释放。