o پi@sdZddlmZddlZddlZddlZddlZddl m Z m Z m Z ddl m Z mZmZmZddl mZGdd d e Zdd d Zdd d ZedfddZddZeedS)ztDECORD Runtime NDArray API. decord.ndarray provides a minimum runtime array structure to be used with C++ library. )absolute_importN) DECORDContext DECORDType NDArrayBase)contextempty from_dlpack numpyasarray)_set_class_ndarrayc@seZdZdZddZdS)NDArrayz/Lightweight NDArray class for DECORD framework.cCsttj|jdS)Nr) functoolsreduceoperatormulshape)selfrB/home/ubuntu/.local/lib/python3.10/site-packages/decord/ndarray.py__len__szNDArray.__len__N)__name__ __module__ __qualname____doc__rrrrrr s r cC td|S)zConstruct a CPU device Parameters ---------- dev_id : int, optional The integer device id Returns ------- ctx : DECORDContext The created context rrdev_idrrrcpu rcCr)zConstruct a GPU device Parameters ---------- dev_id : int, optional The integer device id Returns ------- ctx : DECORDContext The created context rrrrrgpu&rr!cCs0t|tjtfs t|}t|j|j||S)aCreate an array from source arr. Parameters ---------- arr : numpy.ndarray The array to be copied from ctx : DECORDContext, optional The device context to create the array Returns ------- ret : NDArray The created array ) isinstance_npndarrayr arrayrrdtypecopyfrom)arrctxrrrr%5s r%cCs"t|\}}t|}t|ddS)zCreate an array that shares the given numpy data. Parameters ---------- np_data : numpy.ndarray The numpy data Returns ------- NDArray The array T)is_view)r ctypespointerr )np_datar(_handlerrrzerocopy_from_numpyIs  r0)r)r __future__r_absr+r rnumpyr# _ffi.ndarrayrrrrrr r r r rr!r%r0rrrrs