o 8ήc @s*dZddlmZddlZddlmZddlZddlmZddl m Z ddl m Z ddl mZz ddlZeZWn eyCdZYnweduZiZiZeZd d Zd d Zd dZdaddZd ddZddZd ddZe Z!e!j"Gddde j#Z$e!j%Gddde j&Z'ddZ(ddZ)dS)!zu Support for CFFI. Allows checking whether objects are CFFI functions and obtaining the pointer and numba signature. )BuiltinFunctionTypeN)partial)types) TypingError) templates) numpy_supportcCs,z |tvp t|tjWStyYdSw)NF)_ffi_instances isinstancecffiFFI TypeErrorobjrC/tmp/pip-target-vg8gfxp4/lib/python/numba/core/typing/cffi_utils.pyis_ffi_instances  rc Cs@z t|jdkWStyz|tvWYSYYdSw)z(Check whether the obj is a CFFI functionfunctionF)ffitypeofkindr _ool_func_typesr rrr is_cffi_func's rcCs |tvrt|Sttd|S)zY Get a pointer to the underlying function for a CFFI function as an integer. uintptr_t) _ool_func_ptrintrcast) cffi_funcrrr get_pointer1srcCsVtduritdtjtdtjtdtjtdtjtdtjtdtj tdtj td tj td tj td tj td tjtd tjtdtj tdtjtdtj tdtjtdtj tdtj tdtjtdtjtdtjtdtjtdtjtdtjiatS)zc Lazily compute type map, as calling ffi.typeof() involves costly parsing of C code... Nboolcharshortrlongz long longz unsigned charzunsigned shortz unsigned intz unsigned longzunsigned long longint8_tuint8_tint16_tuint16_tint32_tuint32_tint64_tuint64_tfloatdoublessize_tsize_tvoid)_cached_type_maprrrbooleanrr intclong_longlongucharushortuintculong ulonglongfloat_r+intpuintpr.rrrr _type_map=sZ                        r<Fc stt|dt|dd}|dkrtd|dkr7|jrtd|j}fdd |jD}tj|g|RS|d krM|j }|j d krFt j St |S|d kra|j }|j}t j||fd S|dkrk|rkt|St|}|duryt||S)z Map CFFI type to numba type. Parameters ---------- cffi_type: The CFFI type to be converted. use_record_dtype: bool (default: False) When True, struct types are mapped to a NumPy Record dtype. use_record_dtyperunionzNo support for CFFI unionrz vararg function is not supportedcsg|]}|qSrr).0argprimed_map_typerr uszmap_type..pointerr.array)dtypeshapestructN)rmap_typegetattrr ellipsisresultargsr signatureitemrrvoidptrCPointerlength NestedArraymap_struct_to_record_dtyper<get) cffi_typer>rrestypeargtypespointeerHnelemrNrrCrrKas2      rKcCsgggt|d}d}|jD]X\}}|jdkr!d}t|||jdkr/d}t|||jdkr=d}t||t t |j dd}|d  ||d  ||d  |j ||j |jdkM}qttj||d S) z2Convert a cffi type into a NumPy Record dtype )namesformatsoffsetsitemsizeTz.field {!r} has bitshift, this is not supportedrz+field {!r} has flags, this is not supportedz-field {!r} has bitsize, this is not supportedr=r]r^r_)align)rsizeoffieldsbitshift ValueErrorformatflagsbitsizeras_dtyperKtypeappendoffset alignment from_dtypenprH)rXrd is_alignedkvmsgrHrrrrVs0    rVcCsFt|p t|}t|dddkrtdt||d}tj|t dS)zB Return a Numba type for the given CFFI function pointer. rr?rJz!No support for CFFI struct valuesr=)r) rrWrrrLr rKrExternalFunctionPointerr)rr>rXsigrrrmake_function_types  rwc@seZdZdZddZdS)FFI_from_bufferzffi.from_buffercCs|st|dkr dS|\}t|tjstd|f|jdvr&td|f|jdkr7|jdkr7td|ft|j}t ||S)Nz.from_buffer() expected a buffer object, got %s)CFzsD            $) !