o پi)@srdZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z ddlmZmZmZmZddlmZmZe dkr>eneZz3e d krHeejd kr`dd lmZmZmZmZdd lmZndd l mZmZmZmZdd l mZWneydd l!mZmZmZmZdd l!mZYnwdd dZ"ddZ#de"ddfddZ$ddZ%GdddeZddZ&dddZ'dS)zRuntime NDArray api)absolute_importN)_LIB check_callc_array string_types _FFI_MODEc_str) DECORDType DECORDContext DECORDArrayDECORDArrayHandle)TypeCodedecord_shape_index_tcythonctypes)r)_set_class_ndarray_reg_extension _make_array _from_dlpack) NDArrayBasecCs@t|tr|d}|tjvrtd|tj|}t||S)a~Construct a DECORD context with given device type and id. Parameters ---------- dev_type: int or str The device type mask or name of the device. dev_id : int, optional The integer device id Returns ------- ctx: DECORDContext The corresponding context. Examples -------- Context can be used to create reflection of context by string representation of the device type. .. code-block:: python assert decord.context("cpu", 1) == decord.cpu(1) assert decord.context("gpu", 0) == decord.gpu(0) assert decord.context("cuda", 0) == decord.gpu(0) rzUnknown device type %s) isinstancersplitr STR2MASK ValueError)dev_typedev_idrG/home/ubuntu/.local/lib/python3.10/site-packages/decord/_ffi/ndarray.pycontexts      r cCsp|}|jds Jt}tt|j}|jtj|_||_d|_ t t |j j |_ |j|_tdd|_||fS)z:Return a DECORDArray representation of a numpy array. C_CONTIGUOUSNrr)flagsr rrshaperdata_asc_void_pdatastridesr npdtypenamendimr ctx)np_datar&arrr#rrr numpyasarrayAs  r/float32c Csntt|}tt|}t}t|}tt ||t|j t|j t|j |j |jt|t|dS)aXCreate an empty array given shape and device Parameters ---------- shape : tuple of int The shape of the array dtype : type or str The data type of the array. ctx : DECORDContext The context of the array Returns ------- arr : decord.nd.NDArray The array decord supported. F)rrrc_intlenr r rrDECORDArrayAlloc type_codebitslanes device_type device_idbyrefr)r#r)r,r+handlerrremptyRs     r;cCst|S)aProduce an array from a DLPack tensor without memory copy. Retrieves the underlying DLPack tensor's pointer to create an array from the data. Removes the original DLPack tensor's destructor as now the array is responsible for destruction. Parameters ---------- dltensor : DLPack tensor Input DLManagedTensor, can only be consumed once. Returns ------- arr: decord.nd.NDArray The array view of the tensor data. )r)dltensorrrr from_dlpacktsr=c@seZdZdZeddZeddZeddZedd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZddZddZdS)rz,A simple Device/CPU Array object in runtime.cs tfddtjjjDS)zShape of this arrayc3s|] }jjj|VqdSN)r:contentsr#).0iselfrr sz$NDArrayBase.shape..)tupleranger:r?r+rBrrBrr#s zNDArrayBase.shapecCst|jjjS)zType of this array)strr:r?r)rBrrrr)szNDArrayBase.dtypecCs |jjjSzcontext of this array)r:r?r,rBrrrr,s zNDArrayBase.ctxcCs|jSrH)r,rBrrrr szNDArrayBase.contextcCst|jtjjSr>)rcastr:r%valuerBrrr__hash__szNDArrayBase.__hash__cCs ||Sr>)same_asrCotherrrr__eq__s zNDArrayBase.__eq__cCs || Sr>)rOrMrrr__ne__ zNDArrayBase.__ne__cCst|tsdS||kS)zCheck object identity equality Parameters ---------- other : object The other object to compare to Returns ------- same : bool Whether other is same as self. F)rrrKrMrrrrLs zNDArrayBase.same_ascCst|tr|jdus|jdurtdt|tr'|j|jur%||dSdSt|tj tj fr7| |dSt dt t|)zSet ndarray valueNz'Array only support set from numpy arrayztype %s not supported)rslicestartstoprrr:copytor(ndarraygenericcopyfrom TypeErrorrGtype)rCin_slicerJrrr __setitem__s     zNDArrayBase.__setitem__c Cst|tr |||St|tjs,z tj||jd}Wntddtt |t |j}|j |j}}|j dkrJ||j f}d|_ t|}|j |krXt d|j |tj||d}|jdsfJ|jtj}t|j|jj}tt|j|||S)a Perform a synchronized copy from the array. Parameters ---------- source_array : array_like The data source we should like to copy from. Returns ------- arr : NDArray Reference to self. r)z!array must be an array_like data,ztype %s is not supportedrz8array shape do not match the shape of NDArray {0} vs {1}r!)rrrUr(rVarrayr)rYrGrZr r#r6rformatascontiguousarrayr"rr$r%c_size_tsizeitemsizerrDECORDArrayCopyFromBytesr:)rC source_arraytr#r)r&nbytesrrrrXs4      zNDArrayBase.copyfromcCs$d|j|j}||7}|S)Nz )r_r#r asnumpy__repr__)rCresrrrriszNDArrayBase.__repr__cCs t|Sr>)rGrhrBrrr__str__rQzNDArrayBase.__str__cCst|j}|j|j}}|jdkr||jf}d|_t|}tj||d}|jds,J|j tj }t |j |jj }tt|j|||S)zConvert this array to numpy array Returns ------- np_arr : numpy.ndarray The corresponding numpy array. rr]r!)r r)r#r6rGr(r;r"rr$r%rarbrcrrDECORDArrayCopyToBytesr:)rCrfr#r)np_arrr&rgrrrrhs   zNDArrayBase.asnumpycCsRt|tr t|j|j|}t|trtt|j |j d|St dt t |)zCopy array to target Parameters ---------- target : NDArray The target array to be copied, must have same shape as this array. NzUnsupported target type %s) rr r;r#r)rrrDECORDArrayCopyFromTor:rrGrZ)rCtargetrrrrU s   zNDArrayBase.copytoN)__name__ __module__ __qualname____doc__propertyr#r)r,r rKrOrPrLr\rXrirkrhrUrrrrrs(    ( rcCstt|t|dS)zFree c++ extension type handle Parameters ---------- handle : ctypes.c_void_p The handle to the extension type. type_code : int The tyoe code N)rrDECORDExtTypeFreerr1)r:r4rrrfree_extension_handles rvcCs&|r |jtjkr tdt|||S)aRegister a extension class to DECORD. After the class is registered, the class will be able to directly pass as Function argument generated by DECORD. Parameters ---------- cls : class The class object to be registered as extension. Note ---- The registered class is requires one property: _decord_handle and a class attribute _decord_tcode. - ```_decord_handle``` returns integer represents the address of the handle. - ```_decord_tcode``` gives integer represents type code of the class. Returns ------- cls : class The class being registered. fcreate : function, optional The creation function to create a class object given handle value. Example ------- The following code registers user defined class MyTensor to be DLTensor compatible. .. code-block:: python @decord.register_extension class MyTensor(object): _decord_tcode = decord.TypeCode.ARRAY_HANDLE def __init__(self): self.handle = _LIB.NewDLTensor() @property def _decord_handle(self): return self.handle.value z>Cannot register create when extension tcode is same as buildin) _decord_tcoder EXT_BEGINrr)clsfcreaterrrregister_extension)s, r{)rr>)(rs __future__rsysrnumpyr(baserrrrrr runtime_ctypesr r r r rr RuntimeError ImportError IMPORT_EXCEPT version_info _cy3.corerrrrr _NDArrayBase _cy2.core_ctypes.ndarrayr r/r;r=rvr{rrrrs<      #"