o 2jv@s\ddlZddlmZddlZddlmmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZddlmZmZmZmZdd lmZdd lmZmZmZdd lmZm Z dd l!m"Z"m#Z#ej$ej%Gd ddZ&ej'de(de(dej)efddZ*ej'ej+ej,dddej-dej)e#dej.fddZ/ej'ej+ej,dddej-dej)e#dej.fddZ0e1e(e(Z2hdZ3e1e4e5e6ej7e5ffiddddd d!d"d!d#d!d$d!d%d&d'd&d(d&d)d&d*d&d+d&d,d&d-d&d.d/d0d1d2d1id3d!d4dd5d6d7d6d8d!d9d&d:d&d;d&dd&d?d&d@d&dAdBdCdBdDd!dEd&idFd&dGd&dHd&dId&dJd&dKd&dLd&dMd&dNd&dOdBdPdBdQd!dRd&dSd&dTd&dUd&dVd&idWd&dXd&dYd&dZd!d[d&d\d&d]dBd^dBd_d/d`d1dad1dbd!dcdddd6ded6dfdgdhdgdididjdjd!d&d&d&d&d!dkdl Z8ej'de9fdmdnZ:GdodpdpeZ;e1ejej'ej?drejt|t|t|f}|j|}|dur4ttjtjttj|}t |}|tj kr2t d|S| }t |} | tj krFt dz&ttjtj| j} td| _|| _t|| _t|| _tt| Wnty|tt| wt | }|tj krtt| t dttj| |j|<|S)Nzav_buffer_ref() failedzav_hwframe_ctx_alloc() failedcuda)intrgetrr#r$r%r&r' av_buffer_refr MemoryErrorr9av_hwframe_ctx_allocAVHWFramesContextdatarformat sw_formatr;r<rav_hwframe_ctx_init Exceptionr(r)) rr:r;r<keycached cached_refout_refr5 frames_ref frames_ctxrrr get_frames_ctxGsH            zCudaContext.get_frames_ctxN)rT)__name__ __module__ __qualname__rr>bintr!r-cfuncr$r%r&r9 AVPixelFormatrOrrrr rs robjstreamr.cCst|dr|dur |n|j|d}n|}t|ds"ttdttjtt |d}|tj kr7t dt |ddkrBt d|S) N __dlpack__)rWsdltensorz>expected a DLPack capsule or an object implementing __dlpack__z"PyCapsule_GetPointer returned NULLs used_dltensorrzPyCapsule_SetName failed)hasattrrXrr TypeErrorrr#r$r rr ValueErrorr RuntimeError)rVrWcapsulemanagedrrr _consume_dlpack{s"   r_F)checkopaquerDcCs.ttjt|}|tjkr||dSdSr)rr#r$r rdeleter)rarDr^rrr _dlpack_avbuffer_frees   rccCsJ|tjkr#tjttt|WddS1swYdSdSr)rrgilrr#object)rarDrrr _numpy_avbuffer_frees "rf>Tabgrargbbgr8bgragbrpgraynv12pal8rgb8rgbabgr24gbrapgray8rgb24bgr48bebgr48legbrp9begbrp9legray9begray9lergb48bergb48leyuv420pyuv444pyuyv422bgra64bebgra64legbrp10begbrp10legbrp12begbrp12legbrp14begbrp14legbrp16begbrp16legray10begray10legray12begray12legray14begray14legray16begray16lergba64bergba64lergbf32bergbf32leyuvj420pyuvj444p gbrap10be gbrap10le gbrap12be gbrap12le gbrap14be gbrap14le gbrap16be gbrap16le gbrpf32be gbrpf32le grayf32be grayf32le rgbaf16be rgbaf16le rgbaf32be rgbaf32le gbrapf32be gbrapf32le bayer_bggr8 bayer_gbrg8 bayer_grbg8 bayer_rggb8 yuv422p10le yuv444p16be yuv444p16le yuva444p16be yuva444p16lebayer_bggr16bebayer_bggr16lebayer_gbrg16bebayer_gbrg16lebayer_grbg16bebayer_grbg16lebayer_rggb16bebayer_rggb16lerg)uint8rhr)rrrrr)uint16rrrrrrrrq)rru)rrvrirjr)rrrrrrrrrrrrr)rfloat32rrkrrrrrrrrrwrxrrrlrrrrrrrrrsryrzrrrtr{r|rorprrr)rfloat16r)r) r)rr) rrrrr~rrrrrrcCsttS)zGet a mostly uninitialized VideoFrame. You MUST call VideoFrame._init(...) or VideoFrame._init_user_attributes() before exposing to the user. ) VideoFrame_cinit_bypass_sentinelrrrr alloc_video_framedrc@s<eZdZejZejZejZ ej Z ej Z ejZejZejZdS) PictureTypeN)rPrQrRr%AV_PICTURE_TYPE_NONENONEAV_PICTURE_TYPE_IIAV_PICTURE_TYPE_PPAV_PICTURE_TYPE_BBAV_PICTURE_TYPE_SSAV_PICTURE_TYPE_SISIAV_PICTURE_TYPE_SPSPAV_PICTURE_TYPE_BIBIrrrr ros rbig big_endiancCst|kr|S|Sr)_is_big_endianbyteswap)arrayrrrr byteswap_array}srplanebytes_per_pixelflip_horizontal flip_verticalcCs|}d}|j|}|}d} t|j} ttj\} } } |r'|jd} d} d} nd} |j} d} t| | | D]=}||}|r_td||D]}t|D]}|||||||| ||<qHqBn|||||| | |<| | 7} q4dS)Nrr)r;abs line_sizerdeclarer>r<range) img_bytesrrrri_bufi_posi_strideo_bufo_poso_stride start_rowend_rowsteprowijrrr copy_bytes_to_planes6     rcCs|}t|||dddS)NF)tobytesr)rrrimgbytesrrr copy_array_to_planesrrrdtypec Csddl}||}|jjj|j}t|}|j}||}|dkr+|j|j f} ||f} n |j|j |f} |||f} |dkrW|jd|} | g| ddR} |j | ||| | dS|j | ||| dS)z Return the useful part of the VideoPlane as a strided array. We are simply creating a view that discards any padding which was added for alignment. rNr)rbufferoffsetstrides)rrr) numpyrframer4linesizeindexritemsizer<r;ndarray) rrrnp dtype_objrtotal_line_sizerchannelsshaperrrrr useful_arrays$     r rokcCs|s td|jddS)NzUnexpected numpy array shape ``)r[r )rr rrr check_ndarray_shapesrc @seZdZdKddZejdejdejdejfdd Z ejd d Z d d Z ddZ e ddZe ddZe ddZe ddZe ddZe ddZejddZe ddZejddZe d d!Zejd"d!Ze d#d$Zejd%d$Ze d&d'Zejd(d'Zd)d*Zd+d,Zejd-efd.d/Zd0d1ZdLd3d4Zd5d6Z e!d7d8Z"e!dMd:d;Z#dd?Z%e! @ 2 2dOdAe&de'de'fdBdCZ(e! D   E E F EdPde)de'de'dGe'dEBdHe*f dIdJZ+dES)Qrrr}cCs&|turdSt|}||||dSr)rr_init)rr;r<rEc_formatrrr r!szVideoFrame.__cinit__rEr;r<cCspd}tj||j_||j_||j_|r|rt|jd}Wdn1s'wY|r2t|| dS)Nrr) rnogilr4r;r<rEr%av_frame_get_bufferr_init_user_attributes)rrEr;r<resrrr rs  zVideoFrame._initcCs(tttj|jj|jj|jj|_dSr) r rr#r%rUr4rEr;r<rrrr rs  z VideoFrame._init_user_attributescCst|jdSr)r%av_frame_unrefr4rrrr r-szVideoFrame.__dealloc__c Cs>d|jjd|jd|jjd|jd|jdt|dd S)Nz) __class__rPptsrEnamer;r<idrrrr __repr__szVideoFrame.__repr__csj}jjrttjtjjjj}t |j jj jj }d}t |jjD]}|jj|jd}||kr8|}q'|jdkr@d}d}||krZjj|rZ|d7}||krZjj|sL|dkrdtdfStfddt |DS)z: A tuple of :class:`.VideoPlane` objects. rrrnrcsg|]}t|qSr)r ).0rrrr *sz%VideoFrame.planes..)rEr4 hw_frames_ctxrr#r$r%rCrDr rFr;r<r nb_componentscomprr extended_datar tuple)rfmtrNmax_plane_countrcount plane_countrrr planes s.  zVideoFrame.planescC|jjS)zWidth of the image, in pixels.)r4r;rrrr r;,zVideoFrame.widthcCr+)zHeight of the image, in pixels.)r4r<rrrr r<1r,zVideoFrame.heightcCst|S)zThe rotation component of the `DISPLAYMATRIX` transformation matrix. Returns: int: The angle (in degrees) by which the transformation rotates the frame counterclockwise. The angle will be in range [-180, 180]. rrrrr rotation6rzVideoFrame.rotationcCst|jjtj@S)z+Is this frame an interlaced or progressive?)boolr4flagsr%AV_FRAME_FLAG_INTERLACEDrrrr interlaced_frame@szVideoFrame.interlaced_framecCr+)zReturns an integer that corresponds to the PictureType enum. Wraps :ffmpeg:`AVFrame.pict_type` :type: int r4 pict_typerrrr r3FrzVideoFrame.pict_typecC ||j_dSrr2rvaluerrr r3P cCr+)zKColorspace of frame. Wraps :ffmpeg:`AVFrame.colorspace`. r4 colorspacerrrr r9TzVideoFrame.colorspacecCr4rr8r5rrr r9]r7cCr+)zMColor range of frame. Wraps :ffmpeg:`AVFrame.color_range`. r4 color_rangerrrr r<ar:zVideoFrame.color_rangecCr4rr;r5rrr r<jr7cCr+)zWTransfer characteristic of frame. Wraps :ffmpeg:`AVFrame.color_trc`. r4 color_trcrrrr r>nr:zVideoFrame.color_trccCr4rr=r5rrr r>wr7cCr+)zUColor primaries of frame. Wraps :ffmpeg:`AVFrame.color_primaries`. r4color_primariesrrrr r@{r:zVideoFrame.color_primariescCr4rr?r5rrr r@r7cOs(|jst|_|jj|g|Ri|S)a%reformat(width=None, height=None, format=None, src_colorspace=None, dst_colorspace=None, interpolation=None, threads=None) Create a new :class:`VideoFrame` with the given width/height/format/colorspace. .. seealso:: :meth:`.VideoReformatter.reformat` for arguments. ) reformatterVideoReformatterreformat)rargskwargsrrr rCszVideoFrame.reformatcKs|jdddi|S)a Get an RGB version of this frame. Any ``**kwargs`` are passed to :meth:`.VideoReformatter.reformat`. >>> frame = VideoFrame(1920, 1080) >>> frame.format.name 'yuv420p' >>> frame.to_rgb().format.name 'rgb24' rErtNr)rC)rrErrr to_rgbs zVideoFrame.to_rgbfilepathcCs|drd}n|ds|drd}ntd|rdnd}|r#d nd }d d lm}||d ddid,}|j||d}|j|_|j|_|||j |d||dWddS1scwYdS)zPSave a VideoFrame as a JPG or PNG. :param filepath: str | Path z.pngFz.jpgz.jpegTz"filepath must end with png or jpg.mjpegpngrrtr)openwupdater/)r8)pix_fmt)rEN) endswithr[av.container.corerJ add_streamr;r<muxr1rC)rrGis_jpgencoderrMrJoutput output_streamrrr saves    "zVideoFrame.savec Ksddlm}|jd ddi|jd}|}|j}t|}d}|jd}|j|} t| } |dkr7|jd|nd} || kr`|| | || |||<|dkrT| |8} n| |7} ||7}|| ks=| d|j|jft | ddddS) zGet an RGB ``PIL.Image`` of this frame. Any ``**kwargs`` are passed to :meth:`.VideoReformatter.reformat`. .. note:: PIL or Pillow must be installed. r)ImagerErtrrRGBrawNr) PILrWrCr*rrr;r< bytearray frombytesbytes) rrErWrrrrrro_sizerrrrr to_images(    zVideoFrame.to_imageFcKs||jjr'd|vr'ttjtj|jjj}t|}t |j |jj |jj j |d<t|dkr5|jdi|n|}|jjr?tdddl}|jj }|j}|tvr|dkrj|jj ddks^Jd|jj ddksjJdt|\}} t|} | d krt|d|| } n=|j|jj |jj | f| d } |d rddd gtd | R} nt| } t| D]\} }t|||| | dddd| f<qt| |d } |s|dvr|| dd} | S|dvr|jj ddksJd|jj ddksJd|t|ddt|d dt|ddgd|jj S|dkrs|jj ddks'Jd|jj ddks4Jdt|ddd}t|d dd}t|ddd}|j|dd d}|j|dd d}|ri|j |||gddS|j |||gddS|dkrt|d}|!|d d"ddd #|j$}||fS|dkr|t|ddt|d ddgd|jj Std|d)aIGet a numpy array of this frame. Any ``**kwargs`` are passed to :meth:`.VideoReformatter.reformat`. The array returned is generally of dimension (height, width, channels). :param bool channel_last: If True, the shape of array will be (height, width, channels) rather than (channels, height, width) for the "yuv444p" and "yuvj444p" formats. .. note:: Numpy must be installed. .. note:: For formats which return an array of ``uint16``, ``float16`` or ``float32``, the samples will be in the system's native byte order. .. note:: For ``pal8``, an ``(image, palette)`` tuple will be returned, with the palette being in ARGB (PyAV will swap bytes if needed). .. note:: For ``gbrp`` formats, channels are flipped to RGB order. rErz2Cannot convert a hardware frame to numpy directly.Nrrz width has to be even for yuyv422z!height has to be even for yuyv422r)rgbrrber~r>r}yuv422prz$width has to be even for this formatz%height has to be even for this formatrrraxisrni4>i4rmz'Conversion to numpy array with format `` is not yet supportedr)%r4r!rr#r$r%rCrDdictr rFr;r<rlenrCr[rrEr*_np_pix_fmt_dtypesr empty startswithr enumeraterrNmoveaxishstackreshaperepeatstack frombufferastypeviewr)r channel_lastrErNrr format_namer*rr num_planesr plane_indicesrp_idxyur+u_fullv_fullimagepaletterrr to_ndarrays   $         zVideoFrame.to_ndarraycCs*|jdkr |d}t||jdddS)z6 Update content from a ``PIL.Image``. rXrrN)modeconvertrr*)rimgrrr set_imageRs  zVideoFrame.set_imagecCs&t|jd|jdd}|||S)z9 Construct a frame from a ``PIL.Image``. rrrt)rsizer)rrrrr from_image[s zVideoFrame.from_imagertcCs ddl}|jd}|s|jd}|dvr9t|ddt||jddk|jdddkr1td |jdf}n|d vrbt|d dt||jddk|jddd krZtd |jdf}n|d vrt|ddt||jddk|jdddkrtd |jdf}n|dvrt|ddt||jddk|jdddkrtd |jdf}n|dvrt|d dt||jddk|jdddkrtd |jdf}nw|dvrt|ddt||jddk|jdddkrtd |jdf}nM|dvr2t|ddt||jddk|jdddkr*td |jdf}n"|dvrQt|dd|jddkrItd |jdf}n|dvrpt|d d|jddkrhtd |jdf}n|dvrt|dd|jddkrtd |jdf}n|dvrt|ddt||jddk|jdddkrtd |jdd|jdd|jddf}n|dvrt|d dt||jddk|jddd krtd |jdd|jdd|jddf}nS|dvr:t|ddt||jddk|jdddkr$td |jdd|jdd|jddf}n|dvrxt|ddt||jddk|jdddkr]td |jdd|jdd|jdd|jddf}n|d vrt|d dt||jddk|jdddkrtd |jdd|jdd|jdd|jddf}n|d!vrt|ddt||jddk|jdddkrtd |jdd|jdd|jdd|jddf}n`|d"vrLt|ddt||jdddkt||jdddk|d#d}|jddkr)td |d$vrA|jd|jdd|jddf}n|jd|jdf}ntd%|d&|d'rk|||d(gd)fd*d}n|d+r|||d(gd,fd*d}t t }| ||||||S)-a Construct a frame from a numpy buffer. :param int width: optional width of actual image, if different from the array width. .. note:: For formats which expect an array of ``uint16``, ``float16`` or ``float32``, the samples must be in the system's native byte order. .. note:: for ``gbrp`` formats, channels are assumed to be given in RGB order. .. note:: For formats where width of the array is not the same as the width of the image, for example with yuv420p images the UV rows at the bottom have padding bytes in the middle of the row as well as at the end. To cope with these, callers need to be able to pass the actual width. rNrrqrtrrr)rrz.provided array does not have C_CONTIGUOUS rowsrurvr{r|r)rrrrr)rrrgrhrjrpr)rrrrrr)rrrrrrr)rr>rirlrorsrrrr>ryrzrrrrrrrrrrrrrrrr>rr>rk> rwrxrrrrrrrr>rr>rr>rrrrrrrr>rr>rmr}rrr}r)Conversion from numpy array with format `rhrr.)rrrrrrk)rrr) rr rrrr[rmascontiguousarrayrorr_image_fill_pointers_numpy)rrEr;rr< linesizesrrrr from_numpy_bufferes                                                          "  zVideoFrame.from_numpy_bufferc Cs|jj}ttjt|}t|}t|j ||j _ ||j _ ||j _ t |D] \} } | |j j| <q'tt|j jttj|j j |j j ||j j} tt|} t| t|| tttj| d|j jd<|j jdtjkrwt| td|dS)Nrzav_buffer_create failed)ctypesrDrr#r$rrr%rr4rEr;r<rnrrav_image_fill_pointersrUrerav_buffer_createrfp_voidbufrrrAr) rrr;r<rrEc_datac_ptrrrrrequiredpy_bufrrr rBs>     z%VideoFrame._image_fill_pointers_numpyc Cs ddl}iddddddddd dd dd dd dd dddddddddddddddddiddddddddddddddd d!d"d#d$d#d%d#d&d#d'd#d(d#d)d#d*d#d+d#id,d#d-d.d/d.d0dd1dd2dd3dd4dd5dd6dd7dd8dd9dd:dd;ddd=dd!d#d#ddd!d?|d@\}}}|durI|jdAkr|dddddf}t||dB|s|dCvr||ddA}t||jdA|kt||dD}t |jdE|jd|}|j j dFr-|j |dddddEdBf|ddddddEf|dddddBdfgdAdG}t|D]}t|dddd|f|j||q1|S|dHkr|\}} t|dIdAt| dIdAt| | jdJkt |jdE|jd|}t||jddE|jdE| dKdL|S|dMvrt|dIdAt||jddBdkt||jdEdAdkt |jdE|jddAdB|}|j|j} dN| dO} |dP} t| d| |jddEt| | | |jdEdEt| | d|jdAdE|S|dQkret|dIdAt||jddOdkt||jdEdAdkt |jdE|jddA|}|j|j} | | dA} |dP} t| d| |jddEt| | | |jdEdEt| | d|jdAdE|S|dRkrt||jrx|j|jkr|tdS|r|jdAdBkr||dAd}n |jddBkstdT|jdEd\} }|dAdks| dAdkrtdUt || |}t|d|jddA|dEdddddAf}|dAdddddAf}t||jdEdAt||jdAdA|S|dVkr&t|dIdBt||jddAdkt||jdEdAdkt||jdAdAkn|dWvr=t|dIdBt||jdAdBkns|dXvrTt|dIdBt||jdAdOkn\|dYvrt|dZdBt||jdAdBkt |jdE|jd|}tt||dD|jdd[|S|d\vrt|d]dBt||jdAdBkt |jdE|jd|}tt||dD|jdd^|S|d_vrt|dZdBt||jdAdOkt |jdE|jd|}tt||dD|jdd`|S|davrt|dbdBt||jdAdOkt |jdE|jd|}tt||dD|jdd`|S|dcvrNt|d]dBt||jdAdOkt |jdE|jd|}tt||dD|jddd|S|dekrt|dIdAt||jddBdkt||jdEdAdkt |jdE|jddAdB|}|j|j}|dP} t| d||jddEt| |d|jdEdA|Stdf|dgt |jdE|jd|}t||jd|jdAkrdE|S|jdA|S)ha Construct a frame from a numpy array. :param bool channel_last: If False (default), the shape for the yuv444p and yuvj444p is given by (channels, height, width) rather than (height, width, channels). .. note:: For formats which expect an array of ``uint16``, ``float16`` or ``float32``, the samples must be in the system's native byte order. .. note:: for ``pal8``, an ``(image, palette)`` pair must be passed. `palette` must have shape (256, 4) and is given in ARGB format (PyAV will swap bytes if needed). .. note:: for ``gbrp`` formats, channels are assumed to be given in RGB order. rNr)rrrrr)rrrrrrrrrrrrrirr)rrrr)rrrrrrrrrrr)rrrrrk)rrrr)rrrrrrrrrrrwrxr)rrrrrlrrrrrrrrrsryrzr)rrrr)ror~rrrrr)NNNrrrbrarr`rdrnr)rrgrfrrrrcrzArray must be uint16 typez>Array must have shape (3, height, width) or (height, width, 3)zWidth and height must be evenrrrrrrrrrrrrrrrrmrrh)rr?ndimrrorr rrNrrErrm concatenaterrr*rLrvrurr;r<rq isinstancerrrr[copy)rrErwrr rrrrru_startv_startflatr<r;r}r+uv_startrrr from_ndarrayys      !"#$%&'()*+,-./01234; <    (                                       zVideoFrame.from_ndarrayrprcCsnt|||}|dkrt||jdd|||S|dvr/t||jd|dr(dnd|||Std|d ) Nrprr> rrrrrrrrrrrr8rrzFormat 'z' is not supported.)rrr*rNNotImplementedError)rr;r<rErrrrrr from_bytesfs  zVideoFrame.from_bytesrmNTrrc' Cst|ttfs |f}t|dkrtdt|}td} td} td} || | | fvr/td|| kr5dnd} | dkr=d nd} tj}tj}d}zt |d |}t |d |}|j j }|j j }||krgtd |t t fvrqtd |j j}|j j}||krtd |t kr|dur|}n||krtdn |dvrtdd }|t kr|d ks|d krtd|j jjd ks|j jj| ks|j jjd krtd|j jjd ks|j jj| ks|j jjd krtd|j jdkrtdtt|j jd }tt|j jd }|d kr|d krtt|}tt|}n|d ks |d kr$td||ks.||kr2td|ds<|dr@td|j jtjkr`|j jd d krUtdtt|j jd }ntt|}tt|| }tt||}|j j}|d}|dkrtt|j jd }tt|j jd }||ks||krtd|j jtjkr|j jd d krtdtt|j jd }nstt|}nl|dkr4tt|j jd }tt|j jd }tt|j jd} ||ks||dks| dkrtd|j jtjkr-|j jdd ks|j jd dkr"tdtt|j jd }n tt|}ntdtt|| }!tt|!|d}"t}||j_||j_|t kr|durft||d}#n,t|tsptd t|jt|kr~td!t|j t|krtd"tt|}#|#!|||}$td#|j_"|$|j_#||_$|#|_%n||j_"ttj&t'|j j(ttj)|j j*}%ttj&t'|j j(ttj)|j j*}&t+,|%|t-ttj.|d |jj/d <|jj/d tjkrt0d$|%|jj(d <||jj1d <tj}t+,|&|"t-ttj.|d |jj/d <|jj/d tjkr%t0d%|&|jj(d <|!|jj1d <tj}|2|WSt3yd|durMt+4|j|tjkrX|5||tjkrc|5|w)&NrzHfrom_dlpack currently supports 2-plane formats only (nv12/p010le/p016le)snv12sp010lesp016lez.from_dlpack supports nv12, p010le, p016le onlyrrrrz,plane tensors must have the same device_typez.only CPU and CUDA DLPack tensors are supportedz-plane tensors must be on the same CUDA devicez4device_id does not match the DLPack tensor device_id)Nrz#device_id must be 0 for CPU tensorsz+CPU DLPack tensors must have device_id == 0zunexpected dtype for plane 0zunexpected dtype for plane 1zplane 0 must be 2D (H, W)z+either specify both width/height or neitherz)plane 0 shape does not match width/heightz0width/height must be even for nv12/p010le/p016lez0plane 0 must be contiguous in the last dimensionz*plane 1 must have shape (H/2, W) for 2D UVz0plane 1 must be contiguous in the last dimensionrz/plane 1 must have shape (H/2, W/2, 2) for 3D UVz-unexpected UV plane strides for (H/2, W/2, 2)zplane 1 must be 2D or 3D)rrz"cuda_context must be a CudaContextzAcuda_context.device_id does not match the DLPack tensor device_idz3cuda_context.primary_ctx does not match primary_ctxr=z#av_buffer_create failed for plane 0z#av_buffer_create failed for plane 1)6rr%listrjr[rrrrr_ dl_tensor device_typer r rrcodebitslanesrZrr#rr r>rrr4r;r<rr.rrOrEr! _device_id _cuda_ctxr$rrDr' byte_offsetr%rrcrrrArrrHrrb)'r*rEr;r<rWrr cuda_contextr:rmp010lep016le expected_bitsrm0m1r dev_type0 dev_type1dev0dev1y_hy_w y_pitch_elems y_linesizey_sizeuv_ndim uv_h_expecteduv_huv_wuv_pitch_elemsuv_w2uv_c uv_linesizeuv_sizectxrMy_ptruv_ptrrrr from_dlpacksN        "$                  zVideoFrame.from_dlpack)rrr})F)rtr)rtF)rpFF)rmrrNNTN),rPrQrRr!rrTr%rUuintrrr-rpropertyr*r;r<r-r1r3setterr9r<r>r@rCrFccallrerVr_rr staticmethodrrrrr]r>rstrr.rrrrr rs                   %o  ]7 m &r)Fsysenumrrcython.cimports.libavcimportslibavr%cython.cimports.av.dictionaryrcython.cimports.av.errorr$cython.cimports.av.sidedata.sidedatarcython.cimports.av.utilsrcython.cimports.av.video.formatrr cython.cimports.av.video.planer r r r cython.cimports.cpython.excr!cython.cimports.cpython.pycapsulerrrcython.cimports.cpython.refrrcython.cimports.libc.stdintrrfinalcclassrrTrer$r_r exceptvalrvoidrcrfrrsupported_np_pix_fmtsrirr%rrkrrrrS byteorderrinlinerrrr rFramerrrr s      c    Y      !"#$%&'()*+,-./0123456789:;<=>?@ABCDEV   ("