python调用c/c++编译出的so库(没有使用swig)

东方耀

python调用c/c++编译出的so库(没有使用swig)

1. #include <iostream>
   
2. #include <string>
   
3. #include <cstdlib>
   
4. #include <vector>
   
5. #include <stdio.h>
   
6. 
   
7. // 将cpp编译成so
   
8. // g++ -shared -Wl,-soname,test -o test.so -fPIC test.cpp
   
9. 
   
10. 
    
11. class Test{
    
12.     private:
    
13.         double _calculate(int a, double b);
    
14.     public:
    
15.         double calculate(int a, double b, char c[], int * d, double * e, char ** f);
    
16. };
    
17. 
    
18. double Test::_calculate(int a, double b){
    
19.     double res = a+b;
    
20.     std::cout<<"res: "<<res<<std::endl;
    
21.     return res;
    
22. }
    
23. 
    
24. double Test::calculate(int a, double b, char c[], int * d, double * e, char ** f){
    
25.     std::cout<<"a: "<<a<<std::endl;
    
26.     std::cout<<"b: "<<b<<std::endl;
    
27.     std::cout<<"c: "<<c<<std::endl;
    
28.     std::cout<<"d: "<<d[0] << ","<<d[1]<<std::endl;
    
29.     std::cout<<"e: "<<e[0] << ","<<e[1]<<std::endl;
    
30.     std::cout<<"f: "<<f[0] << ","<<f[1]<<std::endl;
    
31.     return this->_calculate(a, b);
    
32. }
    
33. 
    
34. 
    
35. // 封装C接口  这样lib才能访问得到 这是暴露接口给python使用
    
36. extern "C"{
    
37.     // 创建对象
    
38.     Test* test_new(){
    
39.         // 谁调这个 函数test_new()？
    
40.         return new Test;
    
41.     }
    
42.     double my_calculate(Test* t, int a, double b, char c[], int * d, double * e, char ** f){
    
43.         return t->calculate(a, b,c,d,e,f);
    
44.     }
    
45. }
    

复制代码

1. # -*- coding: utf-8 -*-
   
2. __author__ = u'东方耀 微信：dfy_88888'
   
3. __date__ = '2021/12/6 下午1:36'
   
4. __product__ = 'PyCharm'
   
5. __filename__ = 'test'
   
6. 
   
7. import ctypes
   
8. 
   
9. # python调用c/c++编译出的so库(没有使用swig)
   
10. 
    
11. # 指定动态链接库
    
12. lib = ctypes.cdll.LoadLibrary('./test.so')
    
13. # 需要指定返回值的类型，默认是int
    
14. lib.my_calculate.restype = ctypes.c_double
    
15. 
    
16. 
    
17. class Test(object):
    
18.     def __init__(self):
    
19.         # 动态链接对象
    
20.         self.obj = lib.test_new()
    
21. 
    
22.     def calculate(self, a, b, c, d, e, f):
    
23.         res = lib.my_calculate(self.obj, a, b, c, d, e, f)
    
24.         return res
    
25. 
    
26. 
    
27. # 将python类型转换成c类型，支持int, float,string的变量和数组的转换
    
28. def convert_type(input):
    
29.     # python调用c++ py的类型需要转换为c++可以识别的类型啊
    
30.     ctypes_map = {int: ctypes.c_int,
    
31.                   float: ctypes.c_double,
    
32.                   str: ctypes.c_char_p
    
33.                   }
    
34.     input_type = type(input)
    
35.     if input_type is list:
    
36.         length = len(input)
    
37.         if length == 0:
    
38.             print("convert type failed...input is " + input)
    
39.             return None
    
40.         else:
    
41.             arr = (ctypes_map[type(input[0])] * length)()
    
42.             for i in range(length):
    
43.                 arr[i] = bytes(input[i], encoding="utf-8") if (type(input[0]) is str) else input[i]
    
44.             return arr
    
45.     else:
    
46.         if input_type in ctypes_map:
    
47.             return ctypes_map[input_type](bytes(input, encoding="utf-8") if type(input) is str else input)
    
48.         else:
    
49.             print("convert type failed...input is " + input)
    
50.             return None
    
51. 
    
52. 
    
53. if __name__ == '__main__':
    
54.     t = Test()
    
55.     A1 = 123
    
56.     A2 = 0.789
    
57.     A3 = "C789"
    
58.     A4 = [456, 789]
    
59.     A5 = [0.123, 0.456]
    
60.     A6 = ["A123", "B456"]
    
61.     print(t.calculate(convert_type(A1), convert_type(A2), convert_type(A3), convert_type(A4), convert_type(A5),
    
62.                       convert_type(A6)))
    

复制代码