Wie kann ich mit ctypes null an eine externe Bibliothek übergeben, wobei ein Argument mit ctypeslib.ndpointer deklariert wird?

Question

von einem anderen hier answer inspiriert, habe ich eine ctypes Funktion, die ich ctypeslib.ndpointer mit nenne:

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

Die externe Funktion ist wie so erklärt:

void foo(cmplx_float *array, int *length) 

Mein Problem ist, dass ich anrufen möchten die Funktion zweimal. Das erste Mal möchte ich nullptr an das array Argument übergeben, damit ich die erforderliche Länge herausfinden kann. Dann das zweite Mal, werde ich in einer Reihe übergeben.

So tue ich so wie folgt aus:

lib.foo(None, length) 

Dies schlägt mit dem folgenden Fehler:

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

Ist es möglich für mich nullptr zu passieren?

Answer 1

Wenn (wie ich), die Sie mit mehreren verschiedenen Array-Typen zu tun hat, eine etwas schönere Abhilfe ndpointer selbst zu wickeln ist:

auf `ctypes`

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

Nach dem hervorragenden Rat von eryksun in einem Kommentar ich Unterklasse den Typ von ctypeslib.ndpointer zurückgegeben und implementiert eine überschriebene from_param. Dies ist ein wenig schwierig, da es mit Klassen-Factory-Techniken durchgeführt werden muss, da die von ctypeslib.ndpointer zurückgegebene Klasse von einer Fabrik hergestellt wird.

Der Code, den ich am Ende mit:

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

Vielen Dank für seine (wie immer) kompetente Beratung eryksun.