o
    YÚÁi¯  ã                   @   s€   d dl mZ ddlZe ej¡jd ZdaG dd„ dƒadd„ Z	G dd	„ d	e
ƒZdd
d„Zdd„ Zdd„ Zddd„Zdd„ ZdS )é   )Ú_ccallback_cé    Nc                   @   s   e Zd ZdS )ÚCDataN)Ú__name__Ú
__module__Ú__qualname__© r   r   ú</tmp/pip-target-1s0edx8b/lib/python/scipy/_lib/_ccallback.pyr   	   s    r   c                  C   sB   t d urd S zdd l} |  ¡ a t jaW d S  ty    da Y d S w )Nr   F)ÚffiÚcffiÚFFIr   ÚImportError)r   r   r   r	   Ú_import_cffi   s   
ÿr   c                   @   sn   e Zd ZdZdZddd„Zdd„ Zedd	„ ƒZed
d„ ƒZ	edd„ ƒZ
dd„ Zeddd„ƒZeddd„ƒZdS )ÚLowLevelCallablea¤  
    Low-level callback function.

    Some functions in SciPy take as arguments callback functions, which
    can either be python callables or low-level compiled functions. Using
    compiled callback functions can improve performance somewhat by
    avoiding wrapping data in Python objects.

    Such low-level functions in SciPy are wrapped in `LowLevelCallable`
    objects, which can be constructed from function pointers obtained from
    ctypes, cffi, Cython, or contained in Python `PyCapsule` objects.

    .. seealso::

       Functions accepting low-level callables:

       `scipy.integrate.quad`, `scipy.ndimage.generic_filter`,
       `scipy.ndimage.generic_filter1d`, `scipy.ndimage.geometric_transform`

       Usage examples:

       :ref:`ndimage-ccallbacks`, :ref:`quad-callbacks`

    Parameters
    ----------
    function : {PyCapsule, ctypes function pointer, cffi function pointer}
        Low-level callback function.
    user_data : {PyCapsule, ctypes void pointer, cffi void pointer}
        User data to pass on to the callback function.
    signature : str, optional
        Signature of the function. If omitted, determined from *function*,
        if possible.

    Attributes
    ----------
    function
        Callback function given.
    user_data
        User data given.
    signature
        Signature of the function.

    Methods
    -------
    from_cython
        Class method for constructing callables from Cython C-exported
        functions.

    Notes
    -----
    The argument ``function`` can be one of:

    - PyCapsule, whose name contains the C function signature
    - ctypes function pointer
    - cffi function pointer

    The signature of the low-level callback must match one of those expected
    by the routine it is passed to.

    If constructing low-level functions from a PyCapsule, the name of the
    capsule must be the corresponding signature, in the format::

        return_type (arg1_type, arg2_type, ...)

    For example::

        "void (double)"
        "double (double, int *, void *)"

    The context of a PyCapsule passed in as ``function`` is used as ``user_data``,
    if an explicit value for ``user_data`` was not given.

    r   Nc                 C   s    |   |||¡}t | |||f¡S ©N)Ú_parse_callbackÚtupleÚ__new__)ÚclsÚfunctionÚ	user_dataÚ	signatureÚitemr   r   r	   r   h   s   zLowLevelCallable.__new__c                 C   s   d| j ›d| j›dS )NzLowLevelCallable(ú, ú))r   r   ©Úselfr   r   r	   Ú__repr__n   s   zLowLevelCallable.__repr__c                 C   ó   t  | d¡S )Nr   ©r   Ú__getitem__r   r   r   r	   r   q   ó   zLowLevelCallable.functionc                 C   r   )Né   r   r   r   r   r	   r   u   r!   zLowLevelCallable.user_datac                 C   s   t  t | d¡¡S )Nr   )r   Úget_capsule_signaturer   r    r   r   r   r	   r   y   s   zLowLevelCallable.signaturec                 C   s   t ƒ ‚r   )Ú
ValueError)r   Úidxr   r   r	   r    }   s   zLowLevelCallable.__getitem__c              
   C   sn   z|j | }W n) ty } zd}t|ƒ|‚d}~w ty0 } zd|›d}t|ƒ|‚d}~ww | |||ƒS )a  
        Create a low-level callback function from an exported Cython function.

        Parameters
        ----------
        module : module
            Cython module where the exported function resides
        name : str
            Name of the exported function
        user_data : {PyCapsule, ctypes void pointer, cffi void pointer}, optional
            User data to pass on to the callback function.
        signature : str, optional
            Signature of the function. If omitted, determined from *function*.

        z?Given module is not a Cython module with __pyx_capi__ attributeNzNo function z$ found in __pyx_capi__ of the module)Ú__pyx_capi__ÚAttributeErrorr$   ÚKeyError)r   ÚmoduleÚnamer   r   r   ÚeÚmessager   r   r	   Úfrom_cython€   s   
€
€þzLowLevelCallable.from_cythonc                 C   sÈ   t ƒ  t|tƒrt |d¡}n&t|tƒrt||ƒ\}}nt|tƒr)t||ƒ\}}nt	 
|¡r1|}ntdƒ‚t|tjƒr@t|ƒ}nt|tƒrJt|ƒ}n|d u rQd}nt	 
|¡rY|}ntdƒ‚t	 |||¡S )Nr   zMGiven input is not a callable or a low-level callable (pycapsule/ctypes/cffi)zNGiven user data is not a valid low-level void* pointer (pycapsule/ctypes/cffi))r   Ú
isinstancer   r   r    Ú
PyCFuncPtrÚ_get_ctypes_funcr   Ú_get_cffi_funcr   Úcheck_capsuler$   ÚctypesÚc_void_pÚ_get_ctypes_dataÚ_get_cffi_dataÚget_raw_capsule)r   Úobjr   r   ÚfuncÚcontextr   r   r	   r   ›   s(   







z LowLevelCallable._parse_callback)NN)r   r   r   Ú__doc__Ú	__slots__r   r   Úpropertyr   r   r   r    Úclassmethodr-   r   r   r   r   r	   r      s     K



r   c                 C   sp   t  | t j¡j}|d u r4t| jƒd }t| jƒD ]\}}|dkr'|t|ƒ7 }q|dt|ƒ 7 }q|d7 }||fS )Nz (r   r   r   )r3   Úcastr4   ÚvalueÚ_typename_from_ctypesÚrestypeÚ	enumerateÚargtypes)r9   r   Úfunc_ptrÚjÚargr   r   r	   r0   ¾   s   r0   c                 C   s~   | d u rdS | t ju rdS | j}d}| d¡r&|d7 }|dd … }| d¡s| d¡r1|dd … }|dkr=|d	d
|  7 }|S )NÚvoidzvoid *r   ÚLP_r   é   Úc_r"   ú Ú*)r3   r4   r   Ú
startswith)r   r*   Úpointer_levelr   r   r	   rA   Ï   s   


þ
rA   c                 C   s   t  | t j¡jS r   )r3   r?   r4   r@   ©Údatar   r   r	   r5   å   s   r5   c                 C   s4   t  d| ¡}|d u rt  t  | ¡¡ dd¡}||fS )NÚ	uintptr_tz(*)rL   )r
   r?   ÚgetctypeÚtypeofÚreplace)r9   r   rE   r   r   r	   r1   î   s   r1   c                 C   s   t  d| ¡S )NrR   )r
   r?   rP   r   r   r	   r6   ù   r!   r6   r   )Ú r   r3   Ú	CFUNCTYPEr4   Ú	__bases__r/   r
   r   r   r   r   r0   rA   r5   r1   r6   r   r   r   r	   Ú<module>   s     
%
	