ubuntu下将c++转换为so文件并利用python调用

python调用.so

LINUX下C++生成.so文件及编译生成可执行文件的过程

[linux][原创]C++ so库的编译python调用

python 打包成 so | python 调用cpp | python 调用C++简单例子

C++转so文件

这里以简单的加法为例，代码如下，add传入两个int的参数，完成加法。主函数main调用add，return结果。这里需要注意的点在于，不能忘记将函数添加到extern C中。原因是python 的ctype可以调用C而无法调用c++，加上extern "C"后，会指示编译器这部分代码按C语言（而不是C++）的方式进行编译。

#include <iostream>
using namespace std;

extern "C"{

   double add(int, int);

}
double add(int x1, int x2)
{
    return x1+x2;
}
int main()
{
  int a = 1;
  int b =2 ;
  int c;
  c = add(a,b);
  return c;
}

之后打开窗口，输入以下命令，即将上述cpp文件转换为了so文件，注意so文件的名称必须以lib开头。

g++ add.cpp -fpic -shared -o libadd.so

python调用so文件

import ctypes  
ll = ctypes.cdll.LoadLibrary   

lib = ll("./libadd.so") 
input1 = 100
input2 = 220
result1 = lib.add(input1,input2)
result2 = lib.main()
print(result1,result2)
print '***finish***' 

python调用.so多个C++文件转so文件可以看到最后结果为（320，3）

如果c++中add函数的输入为double，则python调用so文件时需要将输入input1 input2，以及add的输出都转换为double的形式，代码如下：

lib.add.restype = ctypes.c_double
result1 = lib.add(ctypes.c_double(input1),ctypes.c_double(input2))

多个C++文件转so文件

下面将多个cpp文件转换为so文件，下面是一个例子：

其中convert.cpp是需要调用的函数，如下，主要的两个函数是xy2lalong和laton2xy，两个函数的返回都是double型的数组。

#include <cstring>
#include <iostream>
#include "AffineModel.h"
#include "RPCGeoModel.h"

extern "C"{
    double * xy2lalong(double, double, char*);
    double * latlon2xy(double, double, char*);
}

double * xy2lalong(double x, double y,char* rpcpath)
{
    string m_prpc=rpcpath;
    //printf("%s\n",m_prpc);
    CRPCGeoModel m_prpc1;
    m_prpc1.InitRPCModel(m_prpc);
    double *latlon =new double[2];
    m_prpc1.GetLatLonByAffine(x,y, 0,latlon[0], latlon[1]); //像素坐标转经纬度
    //printf("you input %f and %f\n", latlon[0], latlon[1]);
    return latlon;
}

double * latlon2xy(double lat ,double lon,char* rpcpath)
{
    string m_prpc=rpcpath;
    //printf("%s\n",m_prpc);
    CRPCGeoModel m_prpc1;
    m_prpc1.InitRPCModel(m_prpc);
    double *xy =new double[2];
    m_prpc1.GetXY(lat, lon, 0, xy[0], xy[1]); //经纬度转像素坐标
    return xy;
}

对于多个cpp转程so文件，命令如下:

g++ -std=gnu++0x convert.cpp AffineModel.cpp CommonFunc.cpp RPCGeoModel.cpp -fPIC -shared -o libconvert.so

即将cpp文件名称都输入至终端,然后加上-fPIC -shared -o 表示生成共享库,最后加上 libXXX.so。注意一定要加上-std=gnu++0x，否则会出现如下错误。

error: ‘nullptr’ was not declared in this scope
  FILE *fp= nullptr;

之后用python调用，如下。由于c++中函数的输出是一个double的数组，那么在python的调用中也需要将函数的输出转换为相同的类型（lib.function.restype = ctypes.POINTER(ctypes.c_double)），即double的数组。

input1 = 100.0
input2 = 100.0
lib.xy2lalong.restype = ctypes.POINTER(ctypes.c_double)
a = lib.xy2lalong(ctypes.c_double(input1),ctypes.c_double(input2), strs)
print(a[0],a[1])

input1 = 9.1949132495
input2 = 118.038103295
lib.latlon2xy.restype = ctypes.POINTER(ctypes.c_double)
b = lib.latlon2xy(ctypes.c_double(input1),ctypes.c_double(input2),strs)
print(b[0],b[1])
print '***finish***'