[英]C++ Array to Numpy with SWIG Problem with TypeError
我正在研究用於在 Python 中調用 C++ 庫的 SWIG。 一個問題是,當我在 C++ 中使用一維數組並想在 Python 中將其作為 Numpy 數組調用時,出現錯誤。
頭文件:example.h
#include <iostream>
using namespace std;
class Example {
public:
void say_hello();
void add(int x, int y, int *result);
int sub(int *x, int *y);
void array_add(int *a, int *b, int *c);
};
C++ 文件:example.cpp
#include "example.h"
void Example::say_hello() {
cout<<"Hello Example."<<endl;
}
void Example::add(int x, int y, int *result) {
*result = x + y;
}
int Example::sub(int *x, int *y) {
return *x-*y;
}
void Example::array_add(int *a, int *b, int *c) {
c[0] = a[0] + b[0];
c[1] = a[1] + b[1];
}
SWIG 接口文件:example.i
%module example
%{
#define SWIG_FILE_WITH_INIT
#include "example.h"
%}
%include "typemaps.i"
%include "numpy.i"
%init %{
import_array();
%}
%apply int *OUTPUT { int *result };
%apply int *INPUT { int *x, int *y};
%apply int *INPLACE_ARRAY1 {int *a, int *b, int *c};
%include "example.h"
安裝文件:setup.py
#!/usr/bin/env python
from distutils.core import setup, Extension
import numpy
import os
example_module = Extension('_example',
sources=['example.cpp', 'example_wrap.cxx',],
)
setup (
name = 'example',
version = '0.1',
author = "Frank Tang",
description = """Simple swig C\+\+/Python example""",
ext_modules = [example_module],
py_modules = ["example"],
)
文件:test_example.py test_example.py
運行“python test_example.py”后,我收到如下錯誤消息。 我使用 macOS。
(virtualenv) bogon:source tangsg$ python test_example.py
Hello Example.
7
3
Traceback (most recent call last):
File "test_example.py", line 18, in <module>
example.Example().array_add(a, b, c)
File "/Users/tangsg/Projects/test/source/example.py", line 115, in
array_add
return _example.Example_array_add(self, a, b, c)
TypeError: in method 'Example_array_add', argument 2 of type 'int *'
(virtualenv) bogon:source tangsg$ ›
您的類型映射和array_add
函數的聲明無效。 NumPy 數組總是有一個大小,這必須傳達給 C++。 關於NumPy 與 SWIG的使用有相當廣泛的文檔。
另外兩件事:
add
和sub
的函數簽名效率低下。 而不是使用笨拙的指針參數,只需按值調用並返回一個值,即
int add(int x, int y) { return x + y; } int sub(int x, int y) { return x - y; }
然后你也可以刪除%apply int *OUTPUT { int *result };
和%apply int *INPUT { int *x, int *y};
從接口文件。
永遠不要using namespace std;
在頭文件中! (為什么“使用命名空間標准;”被認為是不好的做法? )
example.h
#include <algorithm>
#include <iostream>
class Example {
public:
void array_add(int *a, int len_a, int *b, int len_b, int *c, int len_c) {
int const end = std::min(len_a, std::min(len_b, len_c));
for (int i = 0; i < end; ++i) {
c[i] = a[i] + b[i];
}
}
};
example.i
%module example
%{
#define SWIG_FILE_WITH_INIT
#include "example.h"
%}
%include "numpy.i"
%init %{
import_array();
%}
%apply (int *IN_ARRAY1, int DIM1) { (int *a, int len_a), (int *b, int len_b) };
%apply (int *INPLACE_ARRAY1, int DIM1) { (int *c, int len_c) };
%include "example.h"
test.py
import example
import numpy as np
E = example.Example()
a = np.array([1,1], dtype=np.int32)
b = np.array([1,1], dtype=np.int32)
c = np.array([1,1], dtype=np.int32)
E.array_add(a,b,c)
print(c)
示例調用:
$ swig -python -c++ example.i
$ clang++ -Wall -Wextra -Wpedantic -I/usr/include/python2.7 -fPIC -shared example_wrap.cxx -o _example.so -lpython2.7
$ python test.py
[2 2]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.