U
    âÇ-e  ã                   @  s@   d dl mZ d dlZd dlZG dd„ dejƒZG dd„ dƒZdS )é    )ÚannotationsNc                   @  s0   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
S )ÚDlpackDeviceTypez3Integer enum for device type codes matching DLPack.é   é   é   é   é   é   é	   é
   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚCPUÚCUDAZ
CPU_PINNEDZOPENCLZVULKANZMETALZVPIZROCM© r   r   ú[/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/pyarrow/interchange/buffer.pyr      s   r   c                   @  sl   e Zd ZdZdddddœdd„Zed	d
œdd„ƒZed	d
œdd„ƒZdd„ Zdd
œdd„Z	dd
œdd„Z
dS )Ú_PyArrowBufferaY  
    Data in the buffer is guaranteed to be contiguous in memory.

    Note that there is no dtype attribute present, a buffer can be thought of
    as simply a block of memory. However, if the column that the buffer is
    attached to has a dtype that's supported by DLPack and ``__dlpack__`` is
    implemented, then that dtype information will be contained in the return
    value from ``__dlpack__``.

    This distinction is useful to support both data exchange via DLPack on a
    buffer and (b) dtypes like variable-length strings which do not have a
    fixed number of bytes per element.
    Tz	pa.BufferÚboolÚNone)ÚxÚ
allow_copyÚreturnc                 C  s
   || _ dS )z)
        Handle PyArrow Buffers.
        N)Ú_x)Úselfr   r   r   r   r   Ú__init__4   s    z_PyArrowBuffer.__init__Úint)r   c                 C  s   | j jS )z'
        Buffer size in bytes.
        )r   Úsize©r   r   r   r   Úbufsize:   s    z_PyArrowBuffer.bufsizec                 C  s   | j jS )z?
        Pointer to start of the buffer as an integer.
        )r   Úaddressr   r   r   r   ÚptrA   s    z_PyArrowBuffer.ptrc                 C  s   t dƒ‚dS )a}  
        Produce DLPack capsule (see array API standard).

        Raises:
            - TypeError : if the buffer contains unsupported dtypes.
            - NotImplementedError : if DLPack support is not implemented

        Useful to have to connect to array libraries. Support optional because
        it's not completely trivial to implement for a Python-only library.
        Ú
__dlpack__N)ÚNotImplementedErrorr   r   r   r   r#   H   s    z_PyArrowBuffer.__dlpack__z#tuple[DlpackDeviceType, int | None]c                 C  s   | j jrtjdfS tdƒ‚dS )zÆ
        Device type and device ID for where the data in the buffer resides.
        Uses device type codes matching DLPack.
        Note: must be implemented even if ``__dlpack__`` is not.
        NÚ__dlpack_device__)r   Zis_cpur   r   r$   r   r   r   r   r%   U   s    
z _PyArrowBuffer.__dlpack_device__Ústrc                 C  s&   dt | j| j|  ¡ d jdœƒ d S )NzPyArrowBuffer(r   )r    r"   Zdeviceú))r&   r    r"   r%   Únamer   r   r   r   Ú__repr__`   s    ýÿÿøÿz_PyArrowBuffer.__repr__N)T)r   r   r   r   r   Úpropertyr    r"   r#   r%   r)   r   r   r   r   r   %   s   r   )Ú
__future__r   ÚenumZpyarrowÚpaÚIntEnumr   r   r   r   r   r   Ú<module>   s   