Come posso passare null a una libreria esterna, usando i ctypes, con un argomento dichiarato con ctypeslib.ndpointer?

Question

Ispirato da un altro answer qui, ho una funzione ctypes che io chiamo utilizzando ctypeslib.ndpointer:

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

La funzione esterna è dichiarata in questo modo:

void foo(cmplx_float *array, int *length) 

mio problema è che io voglio chiamare la funzione due volte. La prima volta voglio passare nullptr all'argomento array, in modo che possa trovare la lunghezza richiesta. Poi la seconda volta, passerò in una serie numpy.

Così lo faccio in questo modo:

lib.foo(None, length) 

Questo non riesce con il seguente errore:

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

E 'possibile per me di passare nullptr?

Answer 1

Se (come me) avete a che fare con più tipi di array diversi, una soluzione un po 'più bello è quello di avvolgere ndpointer stessa:

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

Seguendo l'eccellente consiglio di eryksun in un commento ho suddiviso il tipo restituito da ctypeslib.ndpointer e implementato un from_param sottoposto a override. Questo è un po 'complicato in quanto deve essere fatto usando tecniche di classe factory, perché la classe restituita da ctypeslib.ndpointer è fatta da una fabbrica.

Il codice che ho finito con:

_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)} 
) 

Molte grazie per eryksun per la sua (come sempre) la consulenza di esperti.