ó ¢áSc@`s¶dZddlmZmZmZddddddd gZdd lZdd lZdd lm Z m Z m Z m Z mZdd lmZmZydd lZWnek r¼d ZnXed krd „ZeZeZeZeZddlmZeZnEdd ljj Z!e!j"ƒZ[!ej#Zd„Ze eddƒZd„Z$ddddddgZ%d„Z&defd„ƒYZ'iZ(d d d d d„Z)ed k r²eZ*iZ+d„Z,e*j-e*j.e*j/e*j0e*j1fdfe*j2e*j3e*j4e*j5e*j6fdfe*j7e*j8fdfgZ9xEe9D]=\Z:Z;x.e:D]&Z<e,e<de;e*j=e<ƒfƒqGWq4We>e*j/d ƒZ?d!„Z@d"„ZAd d#„Zd$„Znd S(%s% ============================ ``ctypes`` Utility Functions ============================ See Also --------- load_library : Load a C library. ndpointer : Array restype/argtype with verification. as_ctypes : Create a ctypes array from an ndarray. as_array : Create an ndarray from a ctypes array. References ---------- .. [1] "SciPy Cookbook: ctypes", http://www.scipy.org/Cookbook/Ctypes Examples -------- Load the C library: >>> _lib = np.ctypeslib.load_library('libmystuff', '.') #doctest: +SKIP Our result type, an ndarray that must be of type double, be 1-dimensional and is C-contiguous in memory: >>> array_1d_double = np.ctypeslib.ndpointer( ... dtype=np.double, ... ndim=1, flags='CONTIGUOUS') #doctest: +SKIP Our C-function typically takes an array and updates its values in-place. For example:: void foo_func(double* x, int length) { int i; for (i = 0; i < length; i++) { x[i] = i*i; } } We wrap it using: >>> _lib.foo_func.restype = None #doctest: +SKIP >>> _lib.foo_func.argtypes = [array_1d_double, c_int] #doctest: +SKIP Then, we're ready to call ``foo_func``: >>> out = np.empty(15, dtype=np.double) >>> _lib.foo_func(out, len(out)) #doctest: +SKIP i(tdivisiontabsolute_importtprint_functiont load_libraryt ndpointerttesttctypes_load_librarytc_intpt as_ctypestas_arrayN(tintegertndarraytdtypet deprecatetarray(t _flagdicttflagsobjcO`stdƒ‚dS(s± Dummy object that raises an ImportError if ctypes is not available. Raises ------ ImportError If ctypes is not available. sctypes is not available.N(t ImportError(targstkwds((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt_dummyCs (tintpc C`s[tjdkr+ddl}|jdƒntjj|ƒd}|s¢ddlm}|ƒ}||g}|dt ƒ}||ks«|j d||ƒq«n |g}tjj |ƒ}tjj |ƒsätjj |ƒ}n|}x^|D]V} tjj|| ƒ} tjj| ƒrñytj| SWqGtk rC‚qGXqñqñWtdƒ‚dS(Ns1.0.1isAAll features of ctypes interface may not work with ctypes < 1.0.1i(tget_shared_lib_extensiont is_python_extsno file with expected extension(tctypest __version__twarningstwarntostpathtsplitexttnumpy.distutils.misc_utilRtTruetinserttabspathtisdirtdirnametjointexiststcdlltOSError( tlibnamet loader_pathRtextRtso_extt libname_exttso_ext2tlibdirtlntlibpath((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyR[s0       cC`s)d}x|D]}|t|7}q W|S(Ni(R(tflaglisttnumtval((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt_num_fromflags…s t C_CONTIGUOUSt F_CONTIGUOUStALIGNEDt WRITEABLEtOWNDATAt UPDATEIFCOPYcC`s?g}x2tD]*}t|}||@r |j|ƒq q W|S(N(t _flagnamesRtappend(R3trestkeytvalue((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt_flags_fromnums    t_ndptrcB`s/eZd„Zed„ƒZed„ƒZRS(cC`s t|ƒS(sªThis method is called when this class is used as the .restype asttribute for a shared-library function. It constructs a numpy array from a void pointer.(R(tself((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt_check_retval_˜scC`sUi|jjd6|d6dd6|jd6dd6|jjddd 6|jtfd 6S( Ntdescrt__reftstridestshapeitversioniittypestrtdata(t_dtype_REtNonet_shape_R@tFalse(RC((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt__array_interface__žs cC`st|tƒstdƒ‚n|jdk rU|j|jkrUtd|jƒ‚n|jdk rŒ|j|jkrŒtd|jƒ‚n|jdk rÉ|j |jkrÉtdt |jƒƒ‚n|j dk r|j j |j @|j krtdt|j ƒƒ‚n|jS(Nsargument must be an ndarraysarray must have data type %ssarray must have %d dimension(s)sarray must have shape %ssarray must have flags %s(t isinstanceR t TypeErrorRLRMR t_ndim_tndimRNRHtstrt_flags_tflagsR3RAR(tclstobj((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt from_param©s (t__name__t __module__RDtpropertyRPt classmethodRZ(((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyRB–s  c C`sq|d k rt|ƒ}nd }|d k rüt|tƒrN|jdƒ}nQt|ttfƒrx|}t|ƒ}n't|tƒrŸ|j }t|ƒ}n|d krüy)g|D]}|j ƒj ƒ^qµ}Wnt dƒ‚nXt |ƒ}qünyt||||fSWntk r'nX|d kr=d}n'|jr[tt|ƒƒ}n |j}|d k r|d|7}n|d k ry#g|D]}t|ƒ^q—}Wn)t k rÛt|ƒg}|f}nXt|ƒ}|ddj|ƒ7}n|d k r(|ddj|ƒ7}ng}td|tfi|d6|d 6|d 6|d 6ƒ}|t|<|S( sC Array-checking restype/argtypes. An ndpointer instance is used to describe an ndarray in restypes and argtypes specifications. This approach is more flexible than using, for example, ``POINTER(c_double)``, since several restrictions can be specified, which are verified upon calling the ctypes function. These include data type, number of dimensions, shape and flags. If a given array does not satisfy the specified restrictions, a ``TypeError`` is raised. Parameters ---------- dtype : data-type, optional Array data-type. ndim : int, optional Number of array dimensions. shape : tuple of ints, optional Array shape. flags : str or tuple of str Array flags; may be one or more of: - C_CONTIGUOUS / C / CONTIGUOUS - F_CONTIGUOUS / F / FORTRAN - OWNDATA / O - WRITEABLE / W - ALIGNED / A - UPDATEIFCOPY / U Returns ------- klass : ndpointer type object A type object, which is an ``_ndtpr`` instance containing dtype, ndim, shape and flags information. Raises ------ TypeError If a given array does not satisfy the specified restrictions. Examples -------- >>> clib.somefunc.argtypes = [np.ctypeslib.ndpointer(dtype=np.float64, ... ndim=1, ... flags='C_CONTIGUOUS')] ... #doctest: +SKIP >>> clib.somefunc(np.array([1, 2, 3], dtype=np.float64)) ... #doctest: +SKIP t,sinvalid flags specificationtanys_%ddt_txs ndpointer_%sRLRNRSRVN(RMt_dtypeRQRUtsplittintR RARR3tstriptupperRRR5t_pointer_type_cachetKeyErrortnamestidttupleR%ttypeRB( R RTRHRWR3Rbtnametstrshapetklass((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyRÀs^4    )       #       c`s^y |jWntk rnXdSt|ƒj‰|tˆ<‡fd†}t|ƒ|_dS(s„Given a ctypes simple type, construct and attach an __array_interface__ property to it if it does not yet have one. Nc`sMidˆfgd6|d6dd6d d6dd6ˆd6tj|ƒtfd 6S( NtRERFRGRHiRIRJRK((RMtctt addressofRO(RC(RJ(sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyRP:s(RPtAttributeErrorRcRUt _typecodesR](t simple_typeR RP((RJsI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt prep_simple/s    titutfs%c%dic`sµy |jWntk rnXdSg‰|}x/t|ƒtkr`ˆj|jƒ|j}q2Wtˆƒ‰|ƒj}|d‰|d‰‡‡‡fd†}t|ƒ|_dS(sƒGiven a ctypes array type, construct and attach an __array_interface__ property to it if it does not yet have one. NRERJc`sDiˆd6|d6dd6ˆd6dd6ˆd6tj|ƒtfd6S( NRERFRGRHiRIRJRK(RMRrRsRO(RC(RERHRJ(sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyRPhs ( RPRtRmt _ARRAY_TYPER=t_length_t_type_RlR](t array_typetobtaiRP((RERHRJsI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt prep_arrayVs         cC`sy |jWntk rnXdS|j}tt|ƒƒ}idd6|jd6tj|ƒtfd6|d6}||_dS(sGiven a ctypes pointer object, construct and attach an __array_interface__ property to it if it does not yet have one. NiRIRJRKRH( RPRttcontentsRcRmRURrRsRO(t pointer_objRHR‚R tinter((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt prep_pointerts      cC`sdt|ƒ}y |jWn:tk rSt|dƒrFt||ƒqTt|ƒnXt|dtƒS(sCreate a numpy array from a ctypes array or a ctypes POINTER. The numpy array shares the memory with the ctypes object. The size parameter must be given if converting from a ctypes POINTER. The size parameter is ignored if converting from a ctypes array R‚tcopy(RmRPRtthasattrR…RRRO(RYRHttp((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyR Šs   cC`s¼|j}|dr"tdƒ‚n|ddkrAtdƒ‚n|d\}}|rftdƒ‚nt|d}x)|d d d d …D]}||}qŒW|j|ƒ}||_|S( s‚Create and return a ctypes object from a numpy array. Actually anything that exposes the __array_interface__ is accepted.RGsstrided arrays not supportedRIis,only __array_interface__ version 3 supportedRKsreadonly arrays unsupportedRJRHNiÿÿÿÿ(RPRRRut from_addresst__keep(RYR€taddrtreadonlyRˆtdimtresult((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyRšs   (Bt__doc__t __future__RRRt__all__tsysRtnumpyR R R RcR Rtnumpy.core.multiarrayRRRRRMRRRRR RRtobjectt _ndptr_basetnumpy.core._internaltcoret _internaltnict_getintp_ctypetc_void_pR5R<RARBRhRRrRuRwtc_bytetc_shorttc_inttc_longt c_longlongtc_ubytetc_ushorttc_uinttc_ulongt c_ulonglongtc_floattc_doublet simple_typesttypestcodeRˆtsizeofRmR{RR…(((sI/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ctypeslib.pyt3s^ (        '      )f  '' (