ó \ K]c@`s¢dZddlmZmZmZddlmZddlZddlm Z ddl Z ddl mZddl m Z ddlmZd d lmZyddlZejƒZWnek rÏdZnXedk ZiZiZeƒZd „Zd „Zd „Zdad„Z e!d„Z"d„Z#e!d„Z$ej%ƒZ&e&j'dej(fd„ƒYƒZ)e&j*dej+fd„ƒYƒZ,d„Z-d„Z.dS(su Support for CFFI. Allows checking whether objects are CFFI functions and obtaining the pointer and numba signature. i(tprint_functiontdivisiontabsolute_import(tBuiltinFunctionTypeN(tpartial(ttypes(t numpy_support(t TypingErrori(t templatescC`s9y |tkpt|tjƒSWntk r4tSXdS(N(t_ffi_instancest isinstancetcffitFFIt TypeErrortFalse(tobj((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pytis_ffi_instances  cC`sKytj|ƒjdkSWn*tk rFy|tkSWqGtSXnXdS(s(Check whether the obj is a CFFI functiontfunctionN(tffittypeoftkindR t_ool_func_typesR(R((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyt is_cffi_func(s cC`s*|tkrt|Sttjd|ƒƒS(sY Get a pointer to the underlying function for a CFFI function as an integer. t uintptr_t(t _ool_func_ptrtintRtcast(t cffi_func((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyt get_pointer2s cC`sátdkrÝitjtjdƒ6tjtjdƒ6tjtjdƒ6tjtjdƒ6tj tjdƒ6tj tjdƒ6tj tjdƒ6tj tjdƒ6tj tjd ƒ6tjtjd ƒ6tjtjd ƒ6tjtjd ƒ6tj tjd ƒ6tjtjdƒ6tj tjdƒ6tjtjdƒ6tj tjdƒ6tj tjdƒ6tjtjdƒ6tjtjdƒ6tjtjdƒ6tjtjdƒ6tjtjdƒ6tjtjdƒ6antS(sc Lazily compute type map, as calling ffi.typeof() involves costly parsing of C code... tbooltchartshortRtlongs long longs unsigned charsunsigned shorts unsigned ints unsigned longsunsigned long longtint8_ttuint8_ttint16_ttuint16_ttint32_ttuint32_ttint64_ttuint64_ttfloattdoubletssize_ttsize_ttvoidN(t_cached_type_maptNoneRtbooleanRRRRtintctlong_tlonglongtuchartushorttuintctulongt ulonglongtfloat_R*tintptuintpR-(((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyt _type_map>s6 c C`ssttd|ƒ}t|ddƒ}|dkr?tdƒ‚n0|dkr¤|jrctdƒ‚n||jƒ}g|jD]}||ƒ^q|}tj||ŒS|dkrå|j }|j d krÏt j St j ||ƒƒSnŠ|d kr"||j ƒ}|j} t jd |d | fƒS|d kr>|r>t|ƒStƒj|ƒ} | dkrkt|ƒ‚n| SdS(sì 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. tuse_record_dtypeRttunionsNo support for CFFI unionRs vararg function is not supportedtpointerR-tarraytdtypetshapetstructN(Rtmap_typetgetattrR tellipsistresulttargsRt signaturetitemRRtvoidptrtCPointertlengtht NestedArraytmap_struct_to_record_dtypeR<tgetR/( t cffi_typeR=tprimed_map_typeRtrestypetargtargtypestpointeeRBtnelemRH((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyREbs2    "      cC`s\igd6gd6gd6tj|ƒd6}t}x |jD]\}}|jdkrtd}t|j|ƒƒ‚n|jdkr¡d}t|j|ƒƒ‚n|jdkrÎd }t|j|ƒƒ‚nt j t |j d tƒƒ}|dj |ƒ|dj |ƒ|dj |jƒ||j|jdkM}q;Wt jtj|d |ƒƒS( s2Convert a cffi type into a NumPy Record dtype tnamestformatstoffsetstitemsizeiÿÿÿÿs.field {!r} has bitshift, this is not supportedis+field {!r} has flags, this is not supporteds-field {!r} has bitsize, this is not supportedR=talign(RtsizeoftTruetfieldstbitshiftt ValueErrortformattflagstbitsizeRtas_dtypeREttypetappendtoffsett alignmentt from_dtypetnpRB(RRR`t is_alignedtktvtmsgRB((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyRP‹s.cC`sjtj|ƒptj|ƒ}t|ddƒdkrEtdƒ‚nt|d|ƒ}tj|dt ƒS(sB Return a Numba type for the given CFFI function pointer. RR>RDs!No support for CFFI struct valuesR=R( RRQRRRFR RERtExternalFunctionPointerR(RR=RRtsig((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pytmake_function_type¬s tFFI_from_buffercB`seZdZd„ZRS(sffi.from_buffercC`sÈ|st|ƒdkrdS|\}t|tjƒsMtd|fƒ‚n|jdkrrtd|fƒ‚n|jdkr¦|jdkr¦td|fƒ‚ntj|jƒ}t j ||ƒS(Nis.from_buffer() expected a buffer object, got %stCtFs<from_buffer() unsupported on non-contiguous buffers (got %s)sJfrom_buffer() only supports multidimensional arrays with C layout (got %s)(RuRv( tlenR RtBufferRtlayouttndimRMRBRRJ(tselfRItkwstarytptr((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pytgeneric½s (t__name__t __module__tkeyR(((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyRt¹st FFIAttributecB`seZejZd„ZRS(cC`stjttjƒS(N(Rt BoundFunctionRtR(R{R((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pytresolve_from_bufferÑs(R€RRRR‚R…(((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pyRƒÍs cC`s¢x›t|jƒD]Š}t|j|ƒ}t|tƒrŠ|jj|ƒt|<|jj|j|j ƒ}t |jj d|ƒƒt |RR R Rt ImportErrorR/t SUPPORTEDRRtsetR RRRR.R<RRERPRstRegistrytregistrytregistertAbstractTemplateRtt register_attrtAttributeTemplateRƒRR(((s6lib/python2.7/site-packages/numba/typing/cffi_utils.pytsB         $ ) !