我有两个swig类型图,其中包含一堆重复的代码。我想按如下方式合并代码:
%{
#include "structure_defs.h"
%}
%ignore Cartesian2PyList(const schrodinger::Cartesian&);
PyObject* Cartesian2PyList(const schrodinger::Cartesian& cartesian)
{
PyObject *o;
o = PyList_New(3);
PyObject* item = PyFloat_FromDouble(cartesian.x);
PyList_SetItem(o, 0, item);
item = PyFloat_FromDouble(cartesian.y);
PyList_SetItem(o, 1, item);
item = PyFloat_FromDouble(cartesian.z);
PyList_SetItem(o, 2, item);
return o;
}
%typemap(out) schrodinger::Cartesian
{
$result = Cartesian2PyList($1);
}
%typemap(out) std::vector<schrodinger::Cartesian>
{
PyObject *o;
o = PyList_New($1.size());
for (uint i=0; i<$1.size(); i++) {
PyObject *elem = Cartesian2PyList($1.at(i));
PyList_SetItem(o, i, elem);
}
$result = o;
}
%include "cartesian.h"
但是,这无法编译,因为在编译时无法找到Cartesian2PyList的定义。在多个类型映射中重用代码的最佳方法是什么?
答案 0 :(得分:3)
您可以使用%{ %}
将代码直接传递到生成的.c文件。因此,在包装器中重用代码的最简单方法就是将其放入其中,可能作为静态函数,以便它不会与同一模块中的任何其他内容发生冲突。在您的示例中,这将起作用:
%{
#include "structure_defs.h"
%}
%{
static PyObject* Cartesian2PyList(const schrodinger::Cartesian& cartesian)
{
PyObject *o;
o = PyList_New(3);
PyObject* item = PyFloat_FromDouble(cartesian.x);
PyList_SetItem(o, 0, item);
item = PyFloat_FromDouble(cartesian.y);
PyList_SetItem(o, 1, item);
item = PyFloat_FromDouble(cartesian.z);
PyList_SetItem(o, 2, item);
return o;
}
%}
%typemap(out) schrodinger::Cartesian
{
$result = Cartesian2PyList($1);
}
%typemap(out) std::vector<schrodinger::Cartesian>
{
PyObject *o;
o = PyList_New($1.size());
for (uint i=0; i<$1.size(); i++) {
PyObject *elem = Cartesian2PyList($1.at(i));
PyList_SetItem(o, i, elem);
}
$result = o;
}
%include "cartesian.h"
如果您愿意,可以将两个%{ %}
块合并到一个块中。
我还删除了那里的%ignore
指令,因为%{ %}
中的代码只是输出到生成的模块中而没有被包装,所以这将是多余的。另一方面,如果您确实希望将其包装在生成的代码中,并且可以使用%inline %{ ... %}
,例如:
%inline %{
static PyObject* Cartesian2PyList(const schrodinger::Cartesian& cartesian)
{
//...
}
%}
如果您正在编写更通用的SWIG代码而不仅仅是单个模块,则可以使用更智能的内容,请参阅fragments,%define
和$typemap
。在简单的情况下,虽然只是如上所示只是在模块内部编写代码就足够了。