Passer un ByteArray de Python à la fonction C

Question

Je voudrais passer une variable ByteArray de mon programme Python à ma DLL écrite en C afin d'accélérer un traitement spécifique qui est trop lent en Python. J'ai traversé le Web, essayé des Ctypes avec des combinaisons de byref, cast, memoryviews, addressof, mais rien ne fonctionne. Existe-t-il un moyen simple d'y parvenir sans copier mon ByteArray dans quelque chose d'autre qui passera? Voici ce que je suis en train de faire:

/* My C DLL */ 
__declspec(dllexport) bool FastProc(char *P, int L) 
{ 
    /* Do some complex processing on the char buffer */ 
    ; 
    return true; 
} 

# My Python program 
from ctypes import * 
def main(argv): 
    MyData = ByteArray([1,2,3,4,5,6]) 
    dll = CDLL('CHELPER.dll') 
    dll.FastProc.argtypes = (c_char_p, c_int) 
    dll.FastProc.restype = c_bool 

    Result = dll.FastProc(MyData, len(MyData)) 
    print(Result) 

Mais je reçois une erreur de type lors du passage du premier paramètre (MyData) à la fonction C.

Y at-il une solution qui ne nécessite pas trop de frais généraux et qui gaspillerait les avantages de ma fonction C?

Olivier

Answer 1

Je suppose que ByteArray est censé être bytearray. Nous pouvons utiliser create_string_buffer pour créer un tampon de caractères mutable qui est un tableau ctypes de c_char. Mais create_string_bufferpas accepter un bytearray, nous devons lui passer un objet bytes pour l'initialiser; heureusement, le casting entre bytes et bytearray est rapide et efficace.

Je n'ai pas votre DLL, donc pour tester que le tableau se comporte correctement, j'utiliserai la fonction libc.strfry pour mélanger ses caractères.

from ctypes import CDLL, create_string_buffer 

libc = CDLL("libc.so.6") 

# Some test data, NUL-terminated so we can safely pass it to a str function. 
mydata = bytearray([65, 66, 67, 68, 69, 70, 0]) 
print(mydata) 

# Convert the Python bytearray to a C array of char 
p = create_string_buffer(bytes(mydata), len(mydata)) 

#Shuffle the bytes before the NUL terminator byte, in-place. 
libc.strfry(p) 

# Convert the modified C array back to a Python bytearray 
newdata = bytearray(p.raw) 
print(newdata) 

sortie typique

bytearray(b'ABCDEF\x00') 
bytearray(b'BFDACE\x00')