Jak mogę przekazać wartość null do biblioteki zewnętrznej, używając ctypes, z argumentem zadeklarowanym przy pomocy ctypeslib.ndpointer?

Question

Zainspirowany innym answer tutaj, mam funkcję ctypes że wzywam użyciu ctypeslib.ndpointer:

lib.foo.argtypes = [ctypeslib.ndpointer(np.complex64, ndim=1, flags='C'), POINTER(c_int)] 

Funkcja zewnętrzna jest zadeklarowany tak:

void foo(cmplx_float *array, int *length) 

Mój problem jest, że chcę zadzwonić funkcja dwa razy. Po raz pierwszy chcę przekazać nullptr do argumentu array, dzięki czemu mogę znaleźć wymaganą długość. Za drugim razem przejdę w tablicy numpy.

więc zrobić tak:

lib.foo(None, length) 

ten nie powiedzie się z powodu następującego błędu:

ctypes.ArgumentError: argument 1: : argument must be an ndarray

Czy jest możliwe dla mnie przechodzą nullptr?

Answer 1

Jeśli (jak ja) masz do czynienia z wieloma różnymi typami macierzy, nieco ładniejszy obejście jest owinąć ndpointer się:

import numpy as np 
from numpy.ctypeslib import ndpointer 

def wrapped_ndptr(*args, **kwargs): 
    base = ndpointer(*args, **kwargs) 
    def from_param(cls, obj): 
    if obj is None: 
     return obj 
    return base.from_param(obj) 
    return type(base.__name__, (base,), {'from_param': classmethod(from_param)}) 

ComplexArrayType = wrapped_ndptr(dtype=np.complex128, ndim=1, flags='C_CONTIGUOUS') 
DoubleArrayType = wrapped_ndptr(dtype=np.float64, ndim=1, flags='C_CONTIGUOUS') 

Answer 2

Po doskonałej porady z eryksun w komentarzu podklasowałem typ zwrócony przez ctypeslib.ndpointer i zaimplementowałem nadpisany from_param. Jest to trochę skomplikowane, ponieważ musi być wykonane przy użyciu technik klasowych, ponieważ klasa zwrócona przez ctypeslib.ndpointer jest wytwarzana przez fabrykę.

Kod że skończyło się z:

_ComplexArrayTypeBase = numpy.ctypeslib.ndpointer(dtype=numpy.complex128, ndim=1, 
    flags='C_CONTIGUOUS') 

def _from_param(cls, obj): 
    if obj is None: 
     return obj 
    return _ComplexArrayTypeBase.from_param(obj) 

ComplexArrayType = type(
    'ComplexArrayType', 
    (_ComplexArrayTypeBase,), 
    {'from_param': classmethod(_from_param)} 
) 

Wiele dzięki eryksun za (jak zawsze) porad ekspertów.