Передать список питонов в C / C ++ как 'float *' или 'int *' - PullRequest
Новая
       25

Передать список питонов в C / C ++ как 'float *' или 'int *'

0 голосов
/ 22 февраля 2019

C / C ++ 

void func(float* xyz1, float* xyz2,int n){
     //do something
     for(int i=0; i<n;i++){
        printf("%f %f\n",xyz1[i],xyz2[i]);
     }
}

Python 

import numpy as np
n = 1000000
xyz1 = np.random.random((n,)).tolist()
xyz2 = np.random.random((n,)).tolist()

#pass above array to the C/C++ code for further processing.
func(xyz1,xyz2,n) # <-- the call to the c/c++ code

Я видел примеры, которые вызывают код C ++ с использованием более высокоуровневых структур данных, таких как C ++ * array.Однако я просто хочу передать его, используя базовые типы данных, такие как int и float *

Любой простой способ сделать это, скажем, PyBind11 или встроенные в Python типы C?

1 Ответ

0 голосов
/ 22 февраля 2019

Вы можете достичь этого с помощью ctypes.Сначала создайте общий объект с C Api. ctypes не поддерживает C ++, но только C. Это означает, что вы можете использовать c ++ в своем исходном коде, но вы должны предоставить интерфейс C без функций языка C ++, таких как перегрузка функций или искажение имен.Для этого определения функций помечены extern "C".

Затем загрузите общий объект в python.Установите типы аргументов и типы результатов.Наконец, вы можете вызвать свою функцию.Вот пример: 

import ctypes
import numpy as np

n = 1000000
xyz1 = np.random.random((n,)).tolist()
xyz2 = np.random.random((n,)).tolist()

#pass above array to the C/C++ code for further processing.
Func = ctypes.CDLL("path/to/libFunc.so")
Func.func.argtypes = [ctypes.POINTER(ctypes.c_double), ctypes.POINTER(ctypes.c_double), ctypes.c_int]
res = Func.func(xyz1,xyz2,n) # <-- the call to the c/c++ code

Я не тестировал его, поэтому, возможно, есть некоторые ошибки, но я думаю, что идея должна быть ясной.Вот пример, который я протестировал:

CMakeLists.txt: 

cmake_minimum_required (VERSION 3.5.1) 
project (DotProduct) 
set(CMAKE_CXX_STANDARD 14) 
set(CMAKE_BUILD_TYPE Debug) 

add_library(DotProduct SHARED src/DotProduct.cpp src/DotProduct.h)

DotProduct.h: 

extern "C" double dotProduct(double* l, double* r, unsigned int len);

DotProduct.cpp: 

#include "DotProduct.h" 

double dotProduct(double *l, double *r, unsigned int len) { 
    double sum(0); 

    while (len--) { 
        sum += l[len] * r[len]; 
    } 
    return sum; 
}

main.py: 

import ctypes

def dot_product(v1, v2):    
    l = len(v1)    
    if l != len(v2):        
        return 0    
    vec1 = (ctypes.c_double * l)(*v1)    
    vec2 = (ctypes.c_double * l)(*v2)    
    Dot_Product = ctypes.CDLL("build/lib/libDotProduct.so")      
    Dot_Product.dotProduct.argtypes = [ctypes.POINTER(ctypes.c_double), ctypes.POINTER(ctypes.c_double), ctypes.c_uint]    
    Dot_Product.dotProduct.restype = ctypes.c_double
    return Dot_Product.dotProduct(vec1, vec2, l)

vec1 = [2, 2]
vec2 = [2, 3]
print("{} * {} = {}".format(vec1, vec2, dot_product(vec1, vec2)))
...