C++中的SWIG运行结果与Python不同。

2024-10-02 22:28:26 发布

您现在位置:Python中文网/ 问答频道 /正文

void BK::BKTree::recursiveSearchNew(BK::Node *node, std::vector<std::pair<std::string, int> > *r, string w,
                                size_t t) {
size_t curDist = levenshteinDistance(node->word, w);
size_t minDist = curDist - t;
size_t maxDist = curDist + t;

if (curDist <= t) {
    std::pair<string, int> p;
    p = std::make_pair(node->word, curDist);
    r->push_back(p);
}
Node* child = node->leftChild;
if (!child) return;

while (child)
{
    if (inRange(child->distance, minDist, maxDist))
        recursiveSearchNew(child, r, w, t);

    child = child->rightSibling;
}}

这个递归函数是由swig编译的,结果向量只返回最后一个元素,但是c++返回正常的结果。这是文件。i:

%module bk
%include <std_pair.i> 
%include <std_vector.i> 
%include <std_string.i> 
%template() std::pair<std::string,int>; 
%template(PairVector) std::vector<std::pair<std::string,int> >; 
%template(StringVector) std::vector<std::string>;

%{ 
 #include "BKTree.h" 
%} 
 %include "BKTree.h" 

Tags: nodechildsizestringifincludetemplatebk
1条回答
网友
1楼 · 发布于 2024-10-02 22:28:26

我觉得你很好。为你的问题创建一个可复制的例子。下面是我的示例,演示了模板的工作情况(SWIG 3.0.12、VS2015、Python 3.8):

%module bk

%{

#include <vector>
#include <string>
#include <utility>

void test(std::vector<std::pair<std::string,int>> *r)
{
    auto p = std::make_pair(std::string{"hello"},5);
    r->push_back(p);
    p = std::make_pair("goodbye",7);
    r->push_back(p);
}

%}

%include <std_pair.i>
%include <std_vector.i>
%include <std_string.i>
%template() std::pair<std::string,int>;
%template(PairVector) std::vector<std::pair<std::string,int> >;

void test(std::vector<std::pair<std::string,int>> *r);

输出:

>>> import bk
>>> v=bk.PairVector()
>>> bk.test(v)
>>> list(v)
[('hello', 5), ('goodbye', 7)]

相关问题 更多 >