hiệu quả chuyển đổi chuỗi (hoặc tuple) để ctypes mảng

Question

đang

Tôi đã có mà phải mất một hình ảnh PIL và chuyển đổi nó vào một mảng ctypes đến ngất xỉu để một hàm C:

w_px, h_px = img.size 
pixels = struct.unpack('%dI'%(w_px*h_px), img.convert('RGBA').tostring()) 
pixels_array = (ctypes.c_int * len(pixels))(*pixels) 

Nhưng tôi đối phó với hình ảnh lớn và giải nén nhiều mục vào đối số chức năng có vẻ chậm đáng kể. Điều đơn giản nhất tôi có thể làm để tăng tốc hợp lý là gì?

Tôi chỉ chuyển đổi thành bộ tuple làm bước trung gian, vì vậy nếu không cần thiết, tất cả đều tốt hơn.

Answer 1

Trước tiên, bạn có thể xây dựng một mảng chưa được khởi tạo:

pixarray = (ctypes.c_int * (w_px * h_px))() 

và sau đó sao chép nội dung của hình ảnh vào nó:

# dylib in MacOSX, cdll.wincrt in Win, libc.so.? in Unix, ... 
clib = ctypes.CDLL('libc.dylib') 

_ = clib.memcpy(pixarray, im.tostring(), w_px * h_px * 4) 

Giá trị trả về của memcpy là một địa chỉ bạn không quan tâm, vì vậy tôi "nuốt" nó bằng cách gán cho nó để đặt tên "dấu gạch dưới đơn" (mà theo quy ước có nghĩa là "Tôi không quan tâm đến cái này" ;-).

Sửa: như @Mu tâm chỉ ra trong một chú thích, đoạn thứ hai một cách hữu ích có thể được đơn giản hóa để sử dụng ctypes.memmove mà không cần phải đi nền tảng phụ thuộc để chồn ra clib: chỉ cần làm

_ = ctypes.memmove(pixarray, im.tostring(), w_px * h_px * 4)