将 python StringIO 传递给 C++使用 swig 的字符串流

Question

我有一些生成 StringIO 变量的 python 代码。我想使用 stringstream 参数将此变量传递给 C++ 函数（假设 C++ stringstream 是与 python StringIO 最接近的匹配）。有没有一种简单的方法使用 swig 来进行这种翻译？

我在 swig 模块中看到一个 std_sstream.i，其中包含一些 stringstream 代码，但我在 swig 文档中找不到任何对它的引用，也找不到它在网络搜索中使用的任何示例。

为了让它更清晰一些，我将在下面添加一些非常简单的代码。 这是 test.cpp，它是固定的，我无法更改：

#include <iostream>
#include <sstream>

#include "test.h"

using namespace std;

void StTest(stringstream &ss)
{
    cout << ss << endl;
}

这是 test.h，它也是固定的：

#include <sstream>

using namespace std;

void StTest(stringstream &ss);

的 python 代码，它调用 C++ StTest 函数（我们称之为 test_test.py）：

#!/usr/bin/env python

import StringIO
import test

ss = StringIO.StringIO()
ss.write("Hello")

# Maybe some thing here to convert ss from StringIO to stringstream

test.StTest(ss)

这是我可以编辑 我正在寻找可以在上面的注释行中从 StringIO 转换为 stringstream 的东西。

这是我对 test.i 文件的初步挖掘，以便 swig 做它的事情：

/* test.i */
%module test

%{
#include "test.h"
%}

%include "std_sstream.i"
%include "test.h"

我在 test.i 中包含了“std_sstream.i”，因为我认为这是应该使用的。我需要向该文件添加一些内容才能让它执行某些操作吗？

目前运行test_test.py时的错误报告为：

TypeError: in method 'StTest', argument 1 of type 'stringstream &'

Answer 1

为什么不直接使用通过调用获得的字符串来调用 PyArg_ParseTuple
StringIO.getvalue() ？

有这样的东西吗？

const char* cstring;
if (!PyArg_ParseTuple(args,"s", &cstring)) //Note the &
        return NULL;
std::string stdstring = cstring;
std::stringstream buffer(stdstring);

另请参阅 cStringIO 的 C API 此处 。

在满足附加要求之后，我尝试在固定约束内使用 SWIG 找到解决方案。

第一个可能的解决方案是内联或扩展解决方案（扩展 StTest 以获取字符串，然后让扩展创建字符串流并调用 StTest），但由于提供的示例只有一个方法并且没有类，因此这不满足约束。

最终的界面文件如下所示。

/* test.i */
%module test

%{
#include "test.h"
#include <sstream>
#include <iostream>
%}
using namespace std;
%include "std_sstream.i"
%include "test.h"

%include <typemaps.i>

%typemap(in) std::stringstream & (char *buf = 0, size_t size = 0, int alloc = 0, std::stringstream stream)
{
  if (SWIG_AsCharPtrAndSize($input, &buf, &size, &alloc) != SWIG_OK)
  {
    %argument_fail(SWIG_TypeError, "(TYPEMAP, SIZE)", $symname, $argnum);
  }

  stream.write(buf, size - 1);

  $1 = &stream;
}

请注意，SWIG 忽略了上面接口文件中的类型映射，不知道为什么。它设法在 Python 中创建 stringstream 对象，但它们不会传递给您的 StTest，从而给出相同的类型错误，因为对象的类型不是 stringstream & 的类型。 （映射的 swig 类型也与 std_sstream.i 生成的类型不匹配，可能是因为它不包含引用类型映射）。

接下来是我建议的覆盖 SWIG 包装器的方法的实现，最终的包装函数如下。 （不会包含整个包装文件，因为它超过 18000 行长）

SWIGINTERN PyObject *_wrap_StTest(PyObject *SWIGUNUSEDPARM(self), PyObject *args) {
  PyObject *resultobj = 0;
  void *argp1 = 0 ;
  int res1 = 0 ;
  PyObject * obj0 = 0 ;
  int argc;
  PyObject *argv[3];
  int ii;
  char *cstring;
  std::stringstream testbuffer;
  if (!PyTuple_Check(args)) SWIG_fail;
  argc = args ? (int)PyObject_Length(args) : 0;
  for (ii = 0; (ii < 2) && (ii < argc); ii++) {
    argv[ii] = PyTuple_GET_ITEM(args,ii);
  }
  if (argc == 1) {

  if (!PyArg_ParseTuple(args,"s",&cstring)) 
SWIG_fail;
  std::string teststring = cstring;
  testbuffer.str(cstring);
  std::cout << "the string in the stream: " << testbuffer.str() << std::endl;
  }  

  StTest(testbuffer);
  resultobj = SWIG_Py_Void();
  return resultobj;
fail:
  return NULL;
}   

流行中的字符串可以正确打印，但是将字符串流传递给 StTest 函数会打印字符串流在内存中的位置。这是预期的，因为字符串流不像以下可接受的答案那样绑定堆栈溢出问题。

我假设 StTest(testbuffer); 可以替换为使用 boost 绑定的版本，并且它可以按要求工作，但已经能够验证。

有用的参考文献包括

• PyTutorial98

• 使用 SWIG 在 C++ 中进行 Python 扩展

• < p>SWIG1.3a5 Python、Perl5(、Tcl) 类型映射常见问题解答