o 䯪g ã@s@ddlmZddlZddlZGdd„dejƒZGdd„dƒZdS)é)Ú annotationsNc@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) ÚDlpackDeviceTypez3Integer enum for device type codes matching DLPack.ééééééé é N) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚCPUÚCUDAÚ CPU_PINNEDÚOPENCLÚVULKANÚMETALÚVPIÚROCM©rrúV/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/pyarrow/interchange/buffer.pyrsrc@sTeZdZdZddd d „Zedd d „ƒZeddd„ƒZdd„Zddd„Z ddd„Z dS)Ú_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. TÚxú pa.BufferÚ allow_copyÚboolÚreturnÚNonecCs ||_dS)z) Handle PyArrow Buffers. N)Ú_x)ÚselfrrrrrÚ__init__4s z_PyArrowBuffer.__init__ÚintcCó|jjS)z' Buffer size in bytes. )r!Úsize©r"rrrÚbufsize:óz_PyArrowBuffer.bufsizecCr%)z? Pointer to start of the buffer as an integer. )r!Úaddressr'rrrÚptrAr)z_PyArrowBuffer.ptrcCstdƒ‚)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__)ÚNotImplementedErrorr'rrrr,Hs z_PyArrowBuffer.__dlpack__ú#tuple[DlpackDeviceType, int | None]cCs|jjr tjdfStdƒ‚)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!Úis_cpurrr-r'rrrr/Us z _PyArrowBuffer.__dlpack_device__ÚstrcCs&dt|j|j| ¡djdœƒdS)NzPyArrowBuffer(r)r(r+Údeviceú))r1r(r+r/Únamer'rrrÚ__repr__`s ýÿÿøÿz_PyArrowBuffer.__repr__N)T)rrrrrr )rr$)rr.)rr1) r r rrr#Úpropertyr(r+r,r/r5rrrrr%s     r)Ú __future__rÚenumÚpyarrowÚpaÚIntEnumrrrrrrÚs