d jddlZddlmZddlZddlmcmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZddlmZmZmZmZdd lmZdd lmZmZmZdd lmZm Z dd l!m"Z"m#Z#ej$Gd dZ%ej&de'de'dej(efdZ)ej&ej*ej+ddej,dej(e#dej-fdZ.ej&ej*ej+ddej,dej(e#dej-fdZ/ej0e'e'Z1hdZ2ej0e3e4e5ej6e4ffiddddddddd dd!dd"d#d$d#d%d#d&d#d'd#d(d#d)d#d*d#d+d,d-d.d/d.id0dd1dd2d3d4d3d5dd6d#d7d#d8d#d9d#d:d#d;d#dd?d@d?dAddBd#idCd#dDd#dEd#dFd#dGd#dHd#dId#dJd#dKd#dLd?dMd?dNddOd#dPd#dQd#dRd#dSd#idTd#dUd#dVd#dWddXd#dYd#dZd?d[d?d\d,d]d.d^d.d_dd`ddad3dbd3dcdddedddfdfdgdgdd#d#d#d#ddhdi Z7ej&de8fdjZ9GdkdleZ:ej0ej;ej<dmkZ=ej&ej>dnej;fdoZ?ej&dpedqej6drej;dsej;fdtZ@ej&dpedqej6fduZAej&ej> ddpedqej6dxe4fdyZBej&dze'd{ej;fd|ZCej$Gd}d~eDZ8dS)N)IntEnum) Dictionary) err_checkget_display_rotation) check_ndarray) get_pix_fmtget_video_format)DLManagedTensor VideoPlanekCPUkCuda) PyErr_Clear)PyCapsule_GetPointerPyCapsule_IsValidPyCapsule_SetName) Py_DECREF Py_INCREF)int64_tuint8_tc eZdZddejdejfdZdZejdej e j fdZ ejd e j d ejd ejdej e j fd Zd S) CudaContextrT device_id primary_ctxcR||_||_tj|_i|_dSN)rrcythonNULL _device_ref _frames_cache)selfrrs I/home/agentuser/manim-venv/lib/python3.11/site-packages/av/video/frame.py __cinit__zCudaContext.__cinit__s)"&!;c|jD]t}tjtjt jtjtj|}t jtj |u|j |j }|tj kr9t jtj |tj |_ dSdSr) r valuesrcastpointerlib AVBufferRefsize_tav_buffer_unrefaddressclearrr)r!vrefs r" __dealloc__zCudaContext.__dealloc__s#**,, 5 5A+s/ FM1--C  s 3 3 4 4 4 4   """ &+    s 3 3 4 4 4%{D     r$returnc V|j}|tjkr|Stj}|j}|}t d|jrdndi}ttj tj |tj ||j d||_|S)Nr10r) rrrrencoderrrr)av_hwdevice_ctx_creater-AV_HWDEVICE_TYPE_CUDAptr)r! device_ref device_bytesc_deviceoptionss r"_get_device_refzCudaContext._get_device_ref.s6:6F  $ $ [ .*1133 ".( 4#3av_hwframe_ctx_allocAVHWFramesContextdatar format sw_formatr@rArav_hwframe_ctx_init Exceptionr,r-) r!r?r@rAkeycached cached_refout_refr: frames_ref frames_ctxs r"get_frames_ctxzCudaContext.get_frames_ctxFs6{{CJJF 4#'',,  s/ FM622J' 33G&+%%!":;;;N))++ 696N 7 7   $ $=>> > @F s45zAAJ!,G 4 4J #)J "5zzJ  #F J  c-j99 : : : :     z : : ; ; ;  #J// fk ! !  z : : ; ; ;677 7"(+fmZ"H"H3s BF++2GN)rT)__name__ __module__ __qualname__rrDbintr#r1cfuncr(r)r*r> AVPixelFormatrUr$r"rrs  6:      +++  \!@\. \0!0z0 0  ( 000\000r$robjstreamr2ct|dr-||n||}n|}t|dstt dt jt jtt|d}|t j krtdt|ddkrtd|S) N __dlpack__)r^sdltensorz>expected a DLPack capsule or an object implementing __dlpack__z"PyCapsule_GetPointer returned NULLs used_dltensorrzPyCapsule_SetName failed)hasattrr`rr TypeErrorrr'r(r rr ValueErrorr RuntimeError)r]r^capsulemanageds r"_consume_dlpackrgzs sL!!&,n#.."""#..PV.:W:W Wk 2 2   L   k'Wk22G&+=>>>"233q885666 Nr$F)checkopaquerJctjtjt|}|tjkr||dSdSr)rr'r(r rdeleter)rirJrfs r"_dlpack_avbuffer_freerlsP06{'00G&+     r$c|tjkrNtj5ttjt |ddddS#1swxYwYdSdSr)rrgilrr'object)rirJs r"_numpy_avbuffer_freerps Z 3 3 fk&&11 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s(AAA>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_rggb16lerq)uint8rrr)rrrrr)uint16rrrrrrrr{)rr)rrrsrtr)rrr|rrrrrrrrr)rfloat32rrurrrrrrrrrrrrrvrrrrrrrrr}rrrrr~rrryrzrrr)rfloat16r)r) r)rr) rrrrrrrrrrrc*ttS)zGet a mostly uninitialized VideoFrame. You MUST call VideoFrame._init(...) or VideoFrame._init_user_attributes() before exposing to the user. ) VideoFrame_cinit_bypass_sentinelr\r$r"alloc_video_framercs , - --r$c~eZdZejZejZejZ ej Z ej Z ejZejZejZdS) PictureTypeN)rVrWrXr)AV_PICTURE_TYPE_NONENONEAV_PICTURE_TYPE_IIAV_PICTURE_TYPE_PPAV_PICTURE_TYPE_BBAV_PICTURE_TYPE_SSAV_PICTURE_TYPE_SISIAV_PICTURE_TYPE_SPSPAV_PICTURE_TYPE_BIBIr\r$r"rrnsN  #D A A A A B B BBBr$rbig big_endiancDt|kr|S|Sr)_is_big_endianbyteswap)arrayrs r"byteswap_arrayr|s$##~~ Lr$planebytes_per_pixelflip_horizontal flip_verticalc|}d}|j|z}|}d} t|j} tjtj\} } } |r|jdz } d} d} n d} |j} d} t| | | D]g}||z}|rDtd||D]1}t|D]}|||z|z |z |z|| |z|z< 2n||||z|| | |z<| | z } hdS)Nrr)r@abs line_sizerdeclarerDrArange) img_bytesrrrri_bufi_posi_strideo_bufo_poso_stride start_rowend_rowsteprowijs r"copy_bytes_to_planersQ'0EE#kO;HEE!%/22H%~fj99IwL1$  ,Y..  h  N1h88  //A+0(1,>B,E%!)a-(( /4EEH ::e9UG: L LLr$rokc:|std|jddS)NzUnexpected numpy array shape ``)rcr)rrs r"check_ndarray_shapers3 JH%+HHHIIIJJr$c eZdZd0dZejdejdejdejfdZ ejdZ d Z d Z e d Ze d Ze d Ze dZe dZe dZejdZe dZejdZe dZejdZe dZejdZe dZejdZdZdZejdefdZdZd1d Zd!Z e!d"Z"e!d2d$Z#d%Z$e!d3d&Z%e! d4d(e&de'de'fd)Z(e! d5de)de'de'd-e'd+zd.e*f d/Z+d+S)6rrrch|turdSt|}||||dSr)rr _init)r!r@rArKc_formats r"r#zVideoFrame.__cinit__s< * * * F&1&&9&9 8UF+++++r$rKr@rAc d}tj5||j_||j_||j_|r|rt j|jd}dddn #1swxYwY|rt|| dS)Nrr) rnogilr9r@rArKr)av_frame_get_bufferr_init_user_attributes)r!rKr@rAress r"r zVideoFrame._inits \ < <"DHN$DHO$DHO < <-dh;; < < < < < < < < < < < < < < <   cNNN ""$$$$$sAAA"%A"cttjtj|jj|jj|jj|_dSr) r rr'r)r[r9rKr@rAr!s r"r%z VideoFrame._init_user_attributess;& K)48? ; ; HN HO   r$c8tj|jdSr)r)av_frame_unrefr9r(s r"r1zVideoFrame.__dealloc__s 48$$$$$r$cd|jjd|jd|jjd|jd|jdt|dd S)Nz) __class__rVptsrKnamer@rAidr(s r"__repr__zVideoFrame.__repr__su =4>* = =$( = =T[=M = =z = = K = =/1$xx < = = = r$cj}jjrhtjtjt jjjj}t|j jj jj }d}t|jjD]$}|jj|jdz}||kr|}%|jdkrd}d}||kr/jj|r|dz }||krjj||dkrt'dfSt)fdt|DS)z: A tuple of :class:`.VideoPlane` objects. rrrxrc0g|]}t|Sr\)r ).0rr!s r" z%VideoFrame.planes.. s#FFFajq))FFFr$)rKr9 hw_frames_ctxrr'r(r)rIrJr rLr@rAr nb_componentscomprr1 extended_datar tuple)r!fmtrTmax_plane_countrcount plane_counts` r"planeszVideoFrame.planessOk 8 ! @F s45tx7M7RAAJ#$dhndhoC'(sw,-- ( (AGLO)A-E&&"' 8v  O"# O++0F{0S+ 1 KO++0F{0S+ !  tQ'') )FFFF53E3EFFFGGGr$c|jjS)zWidth of the image, in pixels.)r9r@r(s r"r@zVideoFrame.width"sx~r$c|jjS)zHeight of the image, in pixels.)r9rAr(s r"rAzVideoFrame.height'sxr$c t|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]. rr(s r"rotationzVideoFrame.rotation,s$D)))r$cNt|jjtjzS)z+Is this frame an interlaced or progressive?)boolr9flagsr)AV_FRAME_FLAG_INTERLACEDr(s r"interlaced_framezVideoFrame.interlaced_frame6sDHNS%AABBBr$c|jjS)zReturns an integer that corresponds to the PictureType enum. Wraps :ffmpeg:`AVFrame.pict_type` :type: int r9 pict_typer(s r"rMzVideoFrame.pict_type<sx!!r$c||j_dSrrLr!values r"rMzVideoFrame.pict_typeF"r$c|jjS)zKColorspace of frame. Wraps :ffmpeg:`AVFrame.colorspace`. r9 colorspacer(s r"rTzVideoFrame.colorspaceJsx""r$c||j_dSrrSrOs r"rTzVideoFrame.colorspaceSs#r$c|jjS)zMColor range of frame. Wraps :ffmpeg:`AVFrame.color_range`. r9 color_ranger(s r"rXzVideoFrame.color_rangeWsx##r$c||j_dSrrWrOs r"rXzVideoFrame.color_range`s$r$c|jjS)zWTransfer characteristic of frame. Wraps :ffmpeg:`AVFrame.color_trc`. r9 color_trcr(s r"r\zVideoFrame.color_trcdsx!!r$c||j_dSrr[rOs r"r\zVideoFrame.color_trcmrQr$c|jjS)zUColor primaries of frame. Wraps :ffmpeg:`AVFrame.color_primaries`. r9color_primariesr(s r"r`zVideoFrame.color_primariesqsx''r$c||j_dSrr_rOs r"r`zVideoFrame.color_primarieszs#(   r$cb|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)r!argskwargss r"rezVideoFrame.reformat~sC 2/11D (t(???????r$c |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' rKr~r\)re)r!rgs r"to_rgbzVideoFrame.to_rgbs!t}66G6v666r$filepathcJ|drd}n<|ds|drd}ntd|rdnd}|rd nd }d d lm}||d ddi5}|||}|j|_|j|_||| |||dddddS#1swxYwYdS)zPSave a VideoFrame as a JPG or PNG. :param filepath: str | Path z.pngFz.jpgz.jpegTz"filepath must end with png or jpg.mjpegpngrr~r)openwupdater4)r=)pix_fmt)rKN) endswithrcav.container.corern add_streamr@rAmuxr6re)r!rjis_jpgencoderrqrnoutput output_streams r"savezVideoFrame.saves   V $ $ CFF   v & & C(*;*;G*D*D CFFABB B"(3wwe%+8zz****** T(C(C 9 9 9 3V"--gw-GGM"&*M #';M JJ}++DMMM,I,IJJ K K K JJ}++D11 2 2 2  3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s7BDDDc `ddlm}|jd ddi|jd}|}d}|j}d}|jdz}|j|z} t| } || kr%||||z| |||z<||z }||z }|| k%|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)ImagerKr~rRGBrawrr\) PILr|rerArr@rA bytearray frombytesbytes) r!rgr|rrrrrro_sizers r"to_imagezVideoFrame.to_images )DMCCCFCCJ1M*/ "'/ "'+/ % x 7$V,,fnn.3EEHryuv422prz$width has to be even for this formatz%height has to be even for this formatrrraxisrxi4>i4rwz'Conversion to numpy array with format `` is not yet supportedr\)%r9r8rr'r(r)rIrJdictr rLr@rAr1lenrercrrKrA_np_pix_fmt_dtypesrempty startswithr enumeraterrrmoveaxishstackreshaperepeatstack frombufferastypeviewr)r! channel_lastrgrTrr format_namerArr  num_planesr plane_indicesrp_idxyur/u_fullv_fullimagepalettes r" to_ndarrayzVideoFrame.to_ndarrays`, 8 ! hf&<&<@F s45tx7M7RAAJ&\\F/$dhndho   8 8;6{{QMDM33F333D 9 " SQRR Rl' ). , , ,i''y*a///1S///y'!+q0002U0000=OHe(+F JQ$VAY%@@Y%uy C5!))%006%&1$DuQ /C/C$D$DMM$)*$5$5M )- 8 8RRHAu%1&-5%Q%QE!!!QQQ'NN"5+*>*>t*D*DEEE 1K3J$J$J E1a00L < < <9?Q&!+++-S+++9#a'1,,,.U,,,99 ++33B77 ++33B77 ++33B77 gb%)/**  + - ' '9?Q&!+++-S+++9#a'1,,,.U,,,VAY844AVAY844AVAY844AYYq!!Y,,FYYq!!Y,,F =xxFF 3!x<<<88Q/a888 8 &  ++E fQi..Qbh  '> ! & 99 ++33B77 A..66r:: gb%)/**  + Yk Y Y Y   r$c~|jdkr|d}t||jdddS)z6 Update content from a ``PIL.Image``. r}rrN)modeconvertr rA)r!imgs r" set_imagezVideoFrame.set_imageAs@ 8u  ++e$$CCQ33333r$c~t|jd|jdd}|||S)z9 Construct a frame from a ``PIL.Image``. rrr~)rsizer)rrs r" from_imagezVideoFrame.from_imageJs8 'sx{CHQKII  r$r~cdddl}|jd}|s |jd}|dvrbt|ddt||jddk|jdddkrt d |jdf}ni|d vrbt|d dt||jddk|jddd krt d |jdf}n|d vrbt|ddt||jddk|jdddkrt d |jdf}n|dvrbt|ddt||jddk|jdddkrt d |jdf}n7|dvrbt|d dt||jddk|jdddkrt d |jdf}n|dvrbt|ddt||jddk|jdddkrt d |jdf}nk|dvrbt|ddt||jddk|jdddkrt d |jdf}n|dvrAt|dd|jddkrt d |jdf}n|dvrAt|d d|jddkrt d |jdf}n{|dvrAt|dd|jddkrt d |jdf}n6|dvrt|ddt||jddk|jdddkrt d |jddz|jddz|jddzf}n|dvrt|d dt||jddk|jddd krt d |jddz|jddz|jddzf}n(|dvrt|ddt||jddk|jdddkrt d |jddz|jddz|jddzf}n|dvrt|ddt||jddk|jdddkrt d |jddz|jddz|jddz|jddzf}n |d vrt|d dt||jddk|jdddkrt d |jddz|jddz|jddz|jddzf}nu|d!vrt|ddt||jddk|jdddkrt d |jddz|jddz|jddz|jddzf}n|d"vrt|ddt||jddzdkt||jddzdk|d#zdz}|jddkrt d |d$vr-|jd|jddz|jddzf}n.|jd|jdf}nt d%|d&|d'r5|||d(gd)fd*d}nI|d+r4|||d(gd,fd*d}tt}| ||||||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. rNrr{r~rrr)rrz.provided array does not have C_CONTIGUOUS rowsrrrrr)rrrrr)rrrqrrrtrzr)rrrrrr)rrrrrrr)rr>rsrvryr}rrrr>rrrrrrrrrrrrrrrrrr>rr>ru> rrrrrrrrrr>rr>r|>rrrrrrrr>rr>rwrrrrr)Conversion from numpy array with format `rr|.)rrrrrru)rrr) rrrrr rcrascontiguousarrayrrr_image_fill_pointers_numpy)rrKr@rrA linesizesrs r"from_numpy_bufferzVideoFrame.from_numpy_bufferTs Q #KNE ' ' ' %! , , , u{1~': ; ; ;}QRR F** !QRRRq)+II C C C %1 - - - u{1~': ; ; ;}QRR F** !QRRRq)+II / / / %A . . . u{1~': ; ; ;}QRR G++ !QRRRq)+II 7 7 7 %! , , , u{1~': ; ; ;}QRR F** !QRRRq)+II G G G %1 - - - u{1~': ; ; ;}QRR F** !QRRRq)+II 1 1 1 %A . . . u{1~': ; ; ;}QRR F** !QRRRq)+II 1 1 1 %A . . . u{1~': ; ; ;}QRR G++ !QRRRq)+II      %! , , ,}Q1$$ !QRRRq)+II    ( %1 - - -}Q1$$ !QRRRq)+II 1 1 1 %A . . .}Q1$$ !QRRRq)+II x   %! , , , u{1~': ; ; ;}QRR F** !QRRR a A% a A% a A%II      %1 - - - u{1~': ; ; ;}QRR F** !QRRR a A% a A% a A%II 1 1 1 %A . . . u{1~': ; ; ;}QRR G++ !QRRR a A% a A% a A%II y %! , , , u{1~': ; ; ;}QRR F** !QRRR a A% a A% a A% a A% II      %1 - - - u{1~': ; ; ;}QRR F** !QRRR a A% a A% a A% a A% II 3 3 3 %A . . . u{1~': ; ; ;}QRR G++ !QRRR a A% a A% a A% a A% II 6 6 6 %! , , , u{1~'9Q'> ? ? ? u{1~'9Q'> ? ? ?q[1_F}Q1$$ !QRRR000M!$M!$)M!$) #]1-u}Q/?@ ZFZZZ    W % % T((U3 ;L5MrST)U)UVVEE   v & & T((U3 >5JBPQ)R)RSSE122 ((vy&QQQ r$c |jj}tjtjt |}t |}tj|j ||j _ ||j _ ||j _ t|D]\} } | |j j| <ttj|j jtjtj|j j |j j ||j j} tjt$|} t'| tj|| t*tjtj| d|j jd<|j jdtjkrt3| t5d|dS)Nrzav_buffer_create failed)ctypesrJrr'r(rr r)r*r9rKr@rArrrav_image_fill_pointersr[rorav_buffer_createrpp_voidbufrrrGr%) r!r r@rArrKc_datac_ptrr!rrrequiredpy_bufs r"rz%VideoFrame._image_fill_pointers_numpy1s"%+M$6)/V^G5Lf)U)U&1&&9&9 48$$$# $Y// , ,KAx#+DH a   &  C-tx??!     VV,,&.   K v . .    Q 8<?fk ) ) f   788 8 ""$$$$$r$c 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@\}}}|T|jdAkr|dddddf}t||dB|s|dCvr||ddA}t ||jdA|kt||dD}t|jdE|jd|}|j j dFrG| |dddddEdBf|ddddddEf|dddddBdfgdAG}t|D]+}t|dddd|f|j||,|S|dHkr|\}} t|dIdAt| dIdAt | | jdJkt|jdE|jd|}t||jddE|jdE| dKdL|S|dMvrt|dIdAt ||jddBzdkt ||jdEdAzdkt|jdE|jddAzdBz|}|j|jz} dN| zdOz} |dP} t| d| |jddEt| | | |jdEdEt| | d|jdAdE|S|dQkrt|dIdAt ||jddOzdkt ||jdEdAzdkt|jdE|jddAz|}|j|jz} | | dAzz} |dP} t| d| |jddEt| | | |jdEdEt| | d|jdAdE|S|dRkrct1||jr|j|jkrt9dS|r)|jdAdBkr||dAd}n |jddBkst9dT|jdEd\} }|dAzdks | dAzdkrt9dUt|| |}t|d|jddA|dEdddddAf}|dAdddddAf}t||jdEdAt||jdAdA|S|dVkrvt|dIdBt ||jddAzdkt ||jdEdAzdkt ||jdAdAknw|dWvr2t|dIdBt ||jdAdBknA|dXvr2t|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|dcvrt|d]dBt ||jdAdOkt|jdE|jd|}tt||dD|jddd|S|dekrt|dIdAt ||jddBzdkt ||jdEdAzdkt|jdE|jddAzdBz|}|j|jz}|dP} t| d||jddEt| |d|jdEdA|St9df|dgt|jdE|jd|}t||jd|jdAkrdEn |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)rrrrrrrrrrrrrsr|)rrrr)rrrrrrrrrrr)rrrrru)rrrr)rrrrrrrrrrrrr)rrrrrvrrrrrrrrr}rrr)rrrr)ryrrrrrr)NNNrrrrrrrrxr)rrrrrrrrzArray must be uint16 typez>Array must have shape (3, height, width) or (height, width, 3)zWidth and height must be evenrrrrrrrrrrrrrrrrwrr)rrEndimrrrrrrrrrKr1r concatenaterr rArprrrr@rAr isinstancerr rrccopy)rrKrrrrr rrru_startv_startflatrAr@rr/uv_starts r" from_ndarrayzVideoFrame.from_ndarrayhsQ" ;% .;% .;%  ?;%  . ;% . ;% ? ;%  .;%  .;%  ?;%  .;%  .;%  ?;%  O;%  _;%  );% )!;% " )#;% ;% $ )%;% & )';% ( ));% * )+;% , )-;% . +/;% 0 +1;% 2 O3;% 4 (5;% 6 (7;% 8 (9;% : (;;% < (=;% > (?;% @ (A;% B (C;% D 'E;% ;% ;% F 'G;% H *I;% J *K;% L OM;% N (O;% P (Q;% R (S;% T (U;% V (W;% X (Y;% Z ([;% \ (];% ^ __;% ` 'a;% b 'c;% d *e;% f *g;% ;% h$&++,,'u;% ;% ;% v #f( ) )w "(Ex  zQaaaDj) % * * * 1F.E$E$E E1a00 u{1~'A B B B"5&//$*?*?@@Eu{1~u{1~vFFE| ++E22 aaaAaCi(aaaAaCi(aaaABBh  '8__ O O#E!!!QQQ'NELOXNNNNL V  "NE7 %! , , , '7A . . . ()B C C Cu{1~u{1~vFFE u|A : : : LO " "7<<#6#6#=#=d#C#C#K#K#M#M N N NL . . . %! , , , u{1~'9Q'> ? ? ? u{1~'9Q'> ? ? ?u{1~ A0Bq/H&QQEkEL0G'kQ&G==$$D QwYa! D D D WW_ 5u|A J J J WXX Q C C CL y %! , , , u{1~'9Q'> ? ? ? u{1~'9Q'> ? ? ?u{1~u{1~/BFKKEkEL0G1 ,G==$$D QwYa! D D D WW_ 5u|A J J J WXX Q C C CL } $ $eRZ00 >EK294L4L !<===  A! 3 3 E1a00k!n)) T"KOMFEqyA~~!q !@AAAuff55E a%,q/1 = = =aCCaCi %%''AaCCaCi %%''A 5<?A 6 6 6 5<?A 6 6 6L y %! , , , u{1~'9Q'> ? ? ? u{1~'9Q'> ? ? ? u{1~': ; ; ; ; ) ) ) %! , , , u{1~': ; ; ; ; 7 7 7 %! , , , u{1~': ; ; ; ; C C C %1 - - - u{1~': ; ; ;u{1~u{1~vFFE ufood&;&;< ? ? ? u{1~'9Q'> ? ? ?u{1~ A0Bq/H&QQE{U\1H==$$D YhYa! D D D XYYa! D D DLZFZZZ 5;q>5;q>6BB 5<?qAAek!n    r$rzrc t|||}|dkrt||jdd||nM|dvr6t||jd|drdnd||nt d|d |S) Nrzrr> rrrrrrrrrrrr8rrzFormat 'z' is not supported.)rrrArrNotImplementedError)rr@rArKrrrs r" from_byteszVideoFrame.from_bytesUs5&&11 V   5<?A           Q__S))0q     &&L&L&L&LMM M r$rwNTrrc t|ttfs|f}t|dkrt dt |}t d} t d} t d} || | | fvrt d|| krdnd} | dkrd nd} tj}tj}d} t|d |}t|d |}|j j }|j j }||krt d |ttfvrt d |j j}|j j}||krt d |tkr||}n+||krt dn|dvrt dd }|tkr|d ks|d krt d|j jjd ks*|j jj| ks|j jjd krt'd|j jjd ks*|j jj| ks|j jjd krt'd|j jdkrt dtjt,|j jd }tjt,|j jd }|d kr;|d kr5tjt0|}tjt0|}n6|d ks|d krt d||ks||krt d|dzs|dzrt d|j jtjkrP|j jd d krt dtjt,|j jd }ntjt,|}tjt0|| z}tjt0||z}|j j}|dz}|dkrtjt0|j jd }tjt0|j jd }||ks||krt d|j jtjkrQ|j jd d krt dtjt,|j jd }notjt,|}nS|dkr=tjt0|j jd }tjt0|j jd }tjt0|j jd} ||ks||dzks| dkrt d|j jtjkrf|j jdd ks|j jd dkrt dtjt,|j jd }n*tjt,|}nt dtjt0|| z}!tjt0|!|dzz}"t5}||j_||j_|tkr|t=||}#nt|t<st'd t1|jt1|krt d!t?|j t?|krt d"tjt<|}#|#!|||}$t d#|j_"|$|j_#||_$|#|_%n ||j_"tjtj&tN|j j(tjtj)|j j*z}%tjtj&tN|j j(tjtj)|j j*z}&tWj,|%|tZtjtj.|d |jj/d <|jj/d tjkrtad$|%|jj(d <||jj1d <tj}tWj,|&|"tZtjtj.|d |jj/d <|jj/d tjkrtad%|&|jj(d <|!|jj1d <tj}|2|S#tf$rg|tWj4|j|tjkr|5||tjkr|5|wxYw)&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_ctxrCz#av_buffer_create failed for plane 0z#av_buffer_create failed for plane 1)6rr<listrrcr rrrrg dl_tensor device_typerr rr codebitslanesrbrr'rrrDr rr9r@rArrGrrUrKr8 _device_id _cuda_ctxr(rrJr+ byte_offsetr)rrlrrrGrr%rNr*rk)'rArKr@rAr^rr cuda_contextr?rwp010lep016le 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_sizectxrSy_ptruv_ptrs' r" from_dlpackzVideoFrame.from_dlpack|s &5$-00 YF v;;!  Z %0$7$77##Y''Y'' $/ / /%&VWW W#t^^ %**11.4k.4k p  F33B F33B 0I 0II%% !OPPP --)D<)D<)Dt|| !PQQQE!!$ $II$&&$N' I--$%JKKK D  daii4199 !NOOO "'1,,<%*m;;<%+q00 >??? "'1,,<%*m;;<%+q00 >???| A%% !<===+gr|'9!'<==C+gr|'9!'<==Czzfkk C--S#..!v{{ !NOOO%<<3&==$%PQQQqy UFQJ U !STTT|#v{22<'*a//$%WXXX & GR\5I!5L M M & GU ; ; S-(*BCCJ[j6&9::Fl'G"aKM!||{3 (:1(=>>{3 (:1(=>>=((DEMM$%QRRR<'6;66|+A.!33(N&,[",:Nq:Q%R%RNN%+[$%?%?NNA{3 (:1(=>> C);A)>??{3 (:1(=>>=((Eeqj,A,ATQYY$%VWWW<'6;66|+A.!33r|7KA7NRS7S7S(K&,[",:Nq:Q%R%RNN%+[%%@%@NN !;<<< +c>H+DEEKk#{fk'BCCG%''E#EIO%EI E!! '% {SSSCC%lK@@N'(LMMM<122c)nnDD(_L455k9J9JJJ(Q!+k<@@C //vFF #.w#7#7  *4 '#, "%#)  Kw'): FM2<+CDDEE[w'): FM2<+CDDEF #3v4fk&-QS6T6TVW  EIM! y}Q6;..!"GHHH %EIN1 $.EI q !B"3% FM2..   EIM! y}Q6;..!"GHHH &EIN1 $/EI q !B  ' ' ) ) )L    "59---V[   2V[   2  sehA1j )rrr)F)r~r)r~F)rzFF)rwrrNNTN),rVrWrXr#rrZr)r[uintr r%r1r3propertyrAr@rArErJrMsetterrTrXr\r`rericcallrorzrrr staticmethodrrrrrrDrstrrGrr\r$r"rrs ,,,, \%C-%fk%6;%%%\%$ \  \ %%%   HHXH>XX**X*CCXC ""X"#####X#$$$$$X$%%%""X"###((X())) @ @ @ 7 7 7 \3V333\36   rs!   #########444444......EEEEEE222222IIIIIIIISSSSSSSSSSSS333333 =<<<<<<<88888888````````FFN?4S8! M! . !! [!!!!3 M3 . !3 [3333(99UUUr$V^eFK$% %&P P P |P | P | P | P -P -P -P -P -P -P -P -P P =!P" =#PP$  %P&  'P( M)P* M+P, -P. ]/P0 ]1P2 ]3P4 ]5P6 ]7P8 ]9P: ];P< ]=P> n?P@ nAPB  CPD MEPPPF MGPH MIPJ MKPL MMPN MOPP MQPR MSPT =UPV =WPX ^YPZ ^[P\  ]P^ M_P` MaPb McPd MePf MgPPPh MiPj MkPl MmPn oPp =qPr =sPt ^uPv ^wPx yPz ={P| =}P~  P@  APB MCPD MEPF ^GPH ^IPPJ%$##$$%% _PPPSSl.:....     '     S]e-CDDfk % %[%[ % ; %%%%PHjH6;HHHH FMMM M(. M@CMMMM4JvJ6;JJJJ xxxxxxxxxxr$