jddlZddlmZddlZddlmcmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZddlmZmZmZmZdd lmZdd lmZmZmZdd lmZm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'ej(ej)ej*ddej+ej,dej-fdZ.ej/ej0GddZ1ej(de2de2dej+efdZ3ej(ej)ej*ddej4dej+e&dej-fdZ5ej(ej)ej*ddej4dej+e&dej-fdZ6ej7e2e2Z8hdZ9ej7e:e;ed&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 Z>ej(de?fdmZ@GdndoeZAej7ejBejCdpkZDej(ejEdqejBfdrZFej(dsedtej=duejBdvejBfdwZGej(dsedtej=fdxZHej(ejE d!dsedtej=d{e;fd|ZIej(d}e2d~ejBfdZJej/ej0GddeKZ?dS)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)AVCUDADeviceContextCUstream)int64_tuint8_t uintptr_tF)check device_ctxreturnctjtjtj|j}tjtj|tj|_dSN) cythoncastpointerlib AVBufferRef user_opaqueav_buffer_unrefaddressNULL)r owner_refs s/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/av/video/frame.py_cuda_device_ctx_freer*sS 28s')?22Iy11222#[Jc eZdZdZ ddejdejdejd efd Zd Z ej d ej e j fd Zej de jdejdejd ej e j fdZdS) CudaContextaA reusable FFmpeg CUDA context for frames imported with DLPack. :param int device_id: CUDA device ordinal. :param bool primary_ctx: Retain the device's primary CUDA context. :param bool current_ctx: Wrap the CUDA context current on the calling thread when this object is first used. This is mutually exclusive with ``primary_ctx``. :param int cuda_stream: Optional raw ``CUstream`` pointer for FFmpeg CUDA operations, including NVENC input and output. The caller owns the stream and must keep it alive as long as this context can be used. Nonzero stream pointers are currently supported only for logical CUDA device 0. To use another physical GPU, expose it as logical device 0 via ``CUDA_VISIBLE_DEVICES``. rTFN device_id primary_ctx current_ctx cuda_streamc||_||_||_d|_tj|_i|_|r|rtd|Xt|tstd|dkrtdt j t||_|jr|jdkrtddSdS)Nrz2primary_ctx and current_ctx are mutually exclusivez&cuda_stream must be an integer or Nonez cuda_stream must be non-negativezxcuda_stream currently requires device_id 0; expose the desired physical GPU as logical device 0 via CUDA_VISIBLE_DEVICES)r.r/r0r1rr' _device_ref _frames_cache ValueError isinstanceint TypeErrorr r)selfr.r/r0r1s r) __cinit__zCudaContext.__cinit__6s#&&!;  S; SQRR R  "k3// J HIIIQ !CDDD%{9kBBD    ! 3 3L    3 3r+c|jD]t}tjtjt jtjtj|}t jtj |u|j |j }|tj kr9t jtj |tj |_ dSdSr) r4valuesrr r!r"r#size_tr%r&clearr3r')r9vrefs r) __dealloc__zCudaContext.__dealloc__Qs#**,, 5 5A+s/ FM1--C  s 3 3 4 4 4 4   """ &+    s 3 3 4 4 4%{D     r+rc |j}|tjkr|Stj}|j}|}d|jrdndi}|jrd|d<t|}ttj tj |tj ||j d|js ||_|Stjtj }|tjkr5tjtj |t#d tjtjtj|j}tjtjt,|j}tjtjtj|j} tjtjt,| j} |j| _tjt2|j| _t6| _tjtj|| _tj}ttj|n[#t@$rNtjtj |tjtj |wxYw||_|S)Nr/10r0rzav_hwdevice_ctx_alloc() failed)!r3rr'r.encoder/r0rrr"av_hwdevice_ctx_creater&AV_HWDEVICE_TYPE_CUDAptrr1av_hwdevice_ctx_allocr% MemoryErrorr r!AVHWDeviceContextdatarhwctxcuda_ctxrstreamr*freep_voidr$av_hwdevice_ctx_init Exception) r9 device_refr( device_bytesc_device options_dictoptionsowner_device_ctxowner_cuda_ctxrrNs r)_get_device_refzCudaContext._get_device_refas6:6F  $ $ K .*1133 ".%d.>'GssCH   .*-L '(66  &y)))        (D  .s/HII  $ $  y 9 9 : : :>?? ? FLks45y~GG CI+235E5KCCNAG s45zAAJ=CK23Z5E==H!/ 7H $k(D4DEEHO3JO%+[ %J%JJ " I c.z:: ; ; ; ;     z : : ; ; ;  y 9 9 : : :   &s$EI88AKsw_fmtwidthheightct|t|t|f}|j|}|tjtjt jtjtj|}t j |}|tj krtd|S| }t j |} | tj krtd tjtjt j| j} t!d| _|| _t|| _t|| _t+t j| n5#t.$r(t jtj| wxYwt j | }|tj kr5t jtj| tdtjtj| |j|<|S)Nzav_buffer_ref() failedzav_hwframe_ctx_alloc() failedcuda)r7r4getrr r!r"r#r= av_buffer_refr'rJr[av_hwframe_ctx_allocAVHWFramesContextrLr format sw_formatr]r^rav_hwframe_ctx_initrSr%r&) r9r\r]r^keycached cached_refout_refrT frames_ref frames_ctxs r)get_frames_ctxzCudaContext.get_frames_ctxs6{{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++2G)rTFN)__name__ __module__ __qualname____doc__rr7bintobjectr:rAcfuncr!r"r#r[ AVPixelFormatrnr+r)r-r-$s   "!"#'#(" :[[   6+++  \8!@888\8t \0!0z0 0  ( 000\000r+r-objrOct|dr-||n||}n|}t|dstt dt jt jtt|d}|t j krtdt|ddkrtd|S) N __dlpack__)rOsdltensorz>expected a DLPack capsule or an object implementing __dlpack__z"PyCapsule_GetPointer returned NULLs used_dltensorrzPyCapsule_SetName failed)hasattrrzrrr8rr r!r rr'r5r RuntimeError)rxrOcapsulemanageds r)_consume_dlpackrs sL!!&,n#.."""#..PV.:W:W Wk 2 2   L   k'Wk22G&+=>>>"233q885666 Nr+opaquerLctjtjt|}|tjkr||dSdSr)rr r!r r'deleter)rrLr~s r)_dlpack_avbuffer_freersP06{'00G&+     r+c|tjkrNtj5ttjt |ddddS#1swxYwYdSdSr)rr'gilrr rt)rrLs r)_numpy_avbuffer_freers 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>Uabgrargbbgr8bgragbrpgraynv12pal8rgb8rgbabgr24gbrapgray8rgb24bgr48bebgr48legbrp9begbrp9legray9begray9lergb48bergb48leyuv420pyuv444pyuyv422bgra64bebgra64legbrp10begbrp10legbrp12begbrp12legbrp14begbrp14legbrp16begbrp16legray10begray10legray12begray12legray14begray14legray16begray16lergba64bergba64lergbf32bergbf32leyuvj420pyuvj444p 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 yuv420p10le yuv422p10le yuv444p16be yuv444p16le yuva444p16be yuva444p16lebayer_bggr16bebayer_bggr16lebayer_gbrg16bebayer_gbrg16lebayer_grbg16bebayer_grbg16lebayer_rggb16bebayer_rggb16ler)uint8rr)rrrrr)uint16rrrrrrrr)rr)rrrrr)rrrrrrrrrrrr)rfloat32rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr)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_sentinelrwr+r)alloc_video_framers , - --r+c~eZdZejZejZejZ ej Z ej Z ejZejZejZdS) PictureTypeN)rorprqr"AV_PICTURE_TYPE_NONENONEAV_PICTURE_TYPE_IIAV_PICTURE_TYPE_PPAV_PICTURE_TYPE_BBAV_PICTURE_TYPE_SSAV_PICTURE_TYPE_SISIAV_PICTURE_TYPE_SPSPAV_PICTURE_TYPE_BIBIrwr+r)rrsN  #D A A A A B B BBBr+rbig big_endiancDt|kr|S|Sr)_is_big_endianbyteswap)arrayrs r)byteswap_arrayrs$##~~ 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_sizerdeclarer7r^range) 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-(( /4EEH1}},"o5##2gabbk22zz 5    ::e9UG: L LLr+rokc:|std|jddS)NzUnexpected numpy array shape ``)r5r0)rr2s r)check_ndarray_shaper5.s3 JH%+HHHIIIJJr+ceZdZd3dZejdejdejdejfdZ ejdZ d Z e d Z d Ze d Ze d Ze dZe dZe dZe dZejdZe dZejdZe dZejdZe dZejdZe dZejdZdZdZdefdZdZejd d4d"Z d#Z!e"d$Z#e"d5d&Z$d'Z%e"ejd d6d(Z&e" d7d*e'de(de(fd+Z)e" d8de*de(de(d.e(d-zd/e+d-zd0e+d-zf d1Z,e"d2Z-d-S)9rrrch|turdSt|}||||dSr)rr _init)r9r]r^rec_formats r)r:zVideoFrame.__cinit__7s< * * * F&1&&9&9 8UF+++++r+rer]r^c d}tj5||j_||j_||j_|r|rt j|jd}dddn #1swxYwY|rt|| dS)Nrr) rnogilrHr]r^rer"av_frame_get_bufferr_init_user_attributes)r9rer]r^ress r)r8zVideoFrame._init>s \ < <"DHN$DHO$DHO < <-dh;; < < < < < < < < < < < < < < <   cNNN ""$$$$$sAAA"%A"cttjtj|jj|jj|jj|_dSr) r rr r"rvrHrer]r^r9s r)r=z VideoFrame._init_user_attributesQs;& K)48? ; ; HN HO   r+c8tj|jdSr)r"av_frame_unrefrHr@s r)rAzVideoFrame.__dealloc__Ys 48$$$$$r+c|jjsdStjtjt j|jjj}t|j |jj |jj S)z For a hardware frame (e.g. ``format.name == "cuda"``), the underlying software pixel format (``nv12``, ``yuv444p``, ``p010le``, ...). ``None`` for a regular software frame. :type: VideoFormat | None N) rH hw_frames_ctxrr r!r"rdrLr rfr]r^)r9rms r)rfzVideoFrame.sw_format\s]x% 4) __class__roptsrenamer]r^idr@s r)__repr__zVideoFrame.__repr__lsu =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. rrrrc0g|]}t|Srw)r ).0rr9s r) z%VideoFrame.planes..s#FFFajq))FFFr+)rerHrDrr r!r"rdrLr rfr]r^r nb_componentscomprrK extended_datar tuple)r9fmtrmmax_plane_countrcount plane_counts` r)planeszVideoFrame.planesrsOk 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.)rHr]r@s r)r]zVideoFrame.widthsx~r+c|jjS)zHeight of the image, in pixels.)rHr^r@s r)r^zVideoFrame.heightsxr+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.rotations$D)))r+cNt|jjtjzS)z+Is this frame an interlaced or progressive?)boolrHflagsr"AV_FRAME_FLAG_INTERLACEDr@s r)interlaced_framezVideoFrame.interlaced_framesDHNS%AABBBr+c|jjS)zReturns an integer that corresponds to the PictureType enum. Wraps :ffmpeg:`AVFrame.pict_type` :type: int rH pict_typer@s r)rfzVideoFrame.pict_typesx!!r+c||j_dSrrer9values r)rfzVideoFrame.pict_type"r+c|jjS)zKColorspace of frame. Wraps :ffmpeg:`AVFrame.colorspace`. rH colorspacer@s r)rmzVideoFrame.colorspacesx""r+c||j_dSrrlrhs r)rmzVideoFrame.colorspaces#r+c|jjS)zMColor range of frame. Wraps :ffmpeg:`AVFrame.color_range`. rH color_ranger@s r)rqzVideoFrame.color_rangesx##r+c||j_dSrrprhs r)rqzVideoFrame.color_ranges$r+c|jjS)zWTransfer characteristic of frame. Wraps :ffmpeg:`AVFrame.color_trc`. rH color_trcr@s r)ruzVideoFrame.color_trcsx!!r+c||j_dSrrtrhs r)ruzVideoFrame.color_trcrjr+c|jjS)zUColor primaries of frame. Wraps :ffmpeg:`AVFrame.color_primaries`. rHcolor_primariesr@s r)ryzVideoFrame.color_primariessx''r+c||j_dSrrxrhs r)ryzVideoFrame.color_primariess#(   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)r9argskwargss r)r~zVideoFrame.reformatsC 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' rerrw)r~)r9rs r)to_rgbzVideoFrame.to_rgbs!t}66G6v666r+filepathc t|}|drd}n<|ds|drd}ntd|rdnd}|rd nd }d d lm}||d ddi5}|||d|D} |j| _|j| _| | | || | dddddS#1swxYwYdS)aoSave a VideoFrame as a JPG or PNG. :param filepath: str | Path :param \**options: Encoder options, e.g. ``pred="none"`` or ``compression_level=1`` for PNG, ``qscale=2`` for JPG. Values are coerced to ``str``. The PNG defaults favor file size over speed; ``pred="none"`` is roughly 3x faster and 2x larger. z.pngFz.jpgz.jpegTz"filepath must end with png or jpg.mjpegpngrrr)openwupdaterC)rXc4i|]\}}|t|Srw)str)rPkr?s r) z#VideoFrame.save..$s$???tq!CFF???r+)pix_fmtrX)reN) rendswithr5av.container.corer add_streamitemsr]r^muxrEr~) r9rrXrKis_jpgencoderrroutput output_streams r)savezVideoFrame.savesMM ==  CFF ]]6 " " CdmmG&<&< CFFABB B"(3wwe%+8zz****** T(C(C 9 9 9 3V"--??w}}???.M #'*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 3sB1EE Ec ddlm}|jd ddi|jd}|}|j}t |}d}|jdz}|j|z} t| } |dkr |jdz |znd} || kr1|| | |z| |||z<|dkr| |z} n| |z } ||z }|| k1| 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)ImagererrrRGBrawrw) PILrr~rZr r r]r^ bytearray frombytesbytes) r9rrrrr rrro_sizerrs r)to_imagezVideoFrame.to_image,s3 )DMCCCFCCJ1M*/ % "%i.. "'+/ % x 7$V,,AJA  q 0H<ryuv422prrz$width has to be even for this formatz%height has to be even for this formatrrrr raxisri4>i4rz'Conversion to numpy array with format `` is not yet supportedrw)%rHrDrr r!r"rdrLdictr rfr]r^rKlenr~r5r&rerZ_np_pix_fmt_dtypesr1empty startswithr enumeraterrmoveaxishstackreshaperepeatstack frombufferastypeviewr)r9 channel_lastrrmr'r, format_namerZr*r! num_planesr plane_indicesrp_idxis_10bityur?u_fullv_fullimagepalettes r) to_ndarrayzVideoFrame.to_ndarrayQs. 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 K K K9?Q&!+++-S+++9#a'1,,,.U,,,$/=$@H$+qq!H (5HHgEy!##uy'71'<'<xxEIO 4ExBBB99 He<<DDRHH He<<DDRHH He<<DDRHH 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``. rrrN)modeconvertr rZ)r9imgs r) set_imagezVideoFrame.set_images@ 8u  ++e$$CCQ33333r+c~t|jd|jdd}|||S)z9 Construct a frame from a ``PIL.Image``. rrr)rsizer)rr's r) from_imagezVideoFrame.from_images8 'sx{CHQKII  r+rcdddl}|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. rNrrrrrr)rrz.provided array does not have C_CONTIGUOUS rowsrrrrr)rrrrr)rrrrrrr)rrrrrr)rrrrrrr)rr>rrrrrrrr>rrrrrrrrrrrrrrrrrr>rr>r> rrrrrrrrrr>rr>r>rrrrrrrr>rr>rrrrrr)Conversion from numpy array with format `rr.)rrrrr r)rrr) r&r0rr5r%r5rascontiguousarrayrrr_image_fill_pointers_numpy)rrer]r,r^ linesizesr's r)from_numpy_bufferzVideoFrame.from_numpy_buffers 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)ctypesrLrr r!rr r"rBrHrer]r^rr(rav_image_fill_pointersrvrtrav_buffer_createrrQbufr'rrJr=) r9r#r]r^rrec_datac_ptrr9rr(requiredpy_bufs r)rz%VideoFrame._image_fill_pointers_numpys"%+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|dMvr4t|dIdAt ||jddBzdkt ||jdEdAzdkt|jdE|jddAzdBz|}|jdks |jdkr|S|j|jz} dN| zdOz} |dP} t| d| |jddEt| | | |jdEdEt| | d|jdAdE|S|dQkr4t|dRdAt ||jddBzdkt ||jdEdAzdkt|jdE|jddAzdBz|}|jdks |jdkr|S|j|jz} dN| zdOz} |dP} t| d| |jddAt| | | |jdEdAt| | d|jdAdA|S|dSkr1t|dIdAt ||jddOzdkt ||jdEdAzdkt|jdE|jddAz|}|jdks |jdkr|S|j|jz} | | dAzz} |dP} t| d| |jddEt| | | |jdEdEt| | d|jdAdE|S|dTkrct1||jr|j|jkrt9dU|r)|jdAdBkr||dAd}n |jddBkst9dV|jdEd\} }|dAzdks | dAzdkrt9dWt|| |}t|d|jddA|dEdddddAf}|dAdddddAf}t||jdEdAt||jdAdA|S|dXkrvt|dIdBt ||jddAzdkt ||jdEdAzdkt ||jdAdAknw|dYvr2t|dIdBt ||jdAdBknA|dZvr2t|dIdBt ||jdAdOkn |d[vrt|dRdBt ||jdAdBkt|jdE|jd|}tt||dD|jdd\|S|d]vrt|d^dBt ||jdAdBkt|jdE|jd|}tt||dD|jdd_|S|d`vrt|dRdBt ||jdAdOkt|jdE|jd|}tt||dD|jdda|S|dbvrt|dcdBt ||jdAdOkt|jdE|jd|}tt||dD|jdda|S|ddvrt|d^dBt ||jdAdOkt|jdE|jd|}tt||dD|jdde|S|dfkrt|dIdAt ||jddBzdkt ||jdEdAzdkt|jdE|jddAzdBz|}|j|jz}|dP} t| d||jddEt| |d|jdEdA|St9dg|dht|jdE|jd|}t||jd|jdAkrdEn |jdA|S)ia 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)rrrrrrrrrrrrrr)rrrr)rrrrrrrrrrr)rrrrr)rrrr)rrrrrrrrrrrrr)rrrrrrrrrrrrrrrrr)rrrr)rrrrrrr)NNNrrrrrrrrr)rrrrrr rrrrzArray must be uint16 typez>Array must have shape (3, height, width) or (height, width, 3)zWidth and height must be evenrrrrrrrrrrrrrrrrr)r&randimrrr5r0rrrrerKr concatenaterr rZrrrrr]r^rr6r+r!rr5copy)rrerr,r/r*r!r'rru_startv_startflatr^r]rr?uv_starts r) from_ndarrayzVideoFrame.from_ndarrays$ ;% .;% .;%  ?;%  . ;% . ;% ? ;%  .;%  .;%  ?;%  .;%  .;%  ?;%  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&QQE{a5<1#4#4 kEL0G'kQ&G==$$D QwYa! D D D WW_ 5u|A J J J WXX Q C C CL } $ $ %1 - - - u{1~'9Q'> ? ? ? u{1~'9Q'> ? ? ?u{1~ A0Bq/H&QQE{a5<1#4#4 kEL0G'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~/BFKKE{a5<1#4#4 kEL0G1 ,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+rrc t|||}|dkrt||jdd||nM|dvr6t||jd|drdnd||nt d|d |S) Nrrr> rrrrrrrrrrrr8rrzFormat 'z' is not supported.)rrrZrNotImplementedError)rr]r^rerrr's r) from_byteszVideoFrame.from_bytess5&&11 V   5<?A           Q__S))0q     &&L&L&L&LMM M r+rNr.r/r0c t|ttfs|f}t|} td} td} td} | | | | fvr t|| |||||St |dkrtd| | krdnd} | dkrdnd}tj }tj }d} t|d |}t|d|}|j j }|j j }||krtd |ttfvrtd |j j}|j j}||krtd |tkr||}n+||krtd n|dvrtdd }|tkr|d ks|d krtd|j jjdks*|j jj| ks|j jjdkrt+d|j jjdks*|j jj| ks|j jjdkrt+d|j jdkrtdtjt0|j jd }tjt0|j jd}|d kr;|d kr5tjt4|}tjt4|}n6|d ks|d krtd||ks||krtd|dzs|dzrtd|j jtj krP|j jddkrtdtjt0|j jd }ntjt0|}tjt4||z}tjt4||z}|j j}|dz}|dkrtjt4|j jd }tjt4|j jd}||ks||krtd|j jtj krQ|j jddkrtdtjt0|j jd }notjt0|}nS|dkr=tjt4|j jd }tjt4|j jd} tjt4|j jd}!||ks| |dzks|!dkrtd|j jtj krf|j jddks|j jddkrtdtjt0|j jd }n*tjt0|}ntdtjt4||z}"tjt4|"|dzz}#t9}||j_||j_|tkrP|%tA||| n|tC|}$nt|t@st+dt5|jt5|krtd |4tC|j"tC|krtd!|4tC|j#tC|krtd"tjt@|}$|$$| ||}%td#|j_%|%|j_&||_'|$|_(n | |j_%tjtj)tT|j j+tjtj,|j j-z}&tjtj)tT|j j+tjtj,|j j-z}'t]j/|&|t`tjtj1|d |jj2d <|jj2d tj krtgd$|&|jj+d <||jj4d <tj }t]j/|'|#t`tjtj1|d |jj2d<|jj2dtj krtgd%|'|jj+d<|"|jj4d<tj }|5|S#tl$rg|t]j7|j|tj kr|8||tj kr|8|wxYw)&Nsnv12sp010lesp016lerzHfrom_dlpack currently supports 2-plane formats only (nv12/p010le/p016le)rrrrz,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_idNr#device_id must be 0 for CPU tensors+CPU DLPack tensors must have device_id == 0zunexpected dtype for plane 0zunexpected dtype for plane 1zplane 0 must be 2D (H, W)+either specify both width/height or neither)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)r.r/r0z"cuda_context must be a CudaContextzAcuda_context.device_id does not match the DLPack tensor device_idz3cuda_context.primary_ctx does not match primary_ctxz3cuda_context.current_ctx does not match current_ctxr`z#av_buffer_create failed for plane 0z#av_buffer_create failed for plane 1)9r6rUlistr r_from_dlpack_planarrr5rr'r dl_tensor device_typerr rr.r!codebitslanesr8rr rr0r7r%rrHr]r^r-r`r/r0rnrerD _device_id _cuda_ctxr!rrLr= byte_offsetr"rrrQrrJr(r=rSrBr)(rZrer]r^rOr.r/ cuda_contextr0r\rp010lep016le expected_bitsr*m0m1r' dev_type0 dev_type1dev0dev1y_hy_w y_pitch_elems y_linesizey_sizeuv_ndim uv_h_expecteduv_huv_wuv_pitch_elemsuv_w2uv_c uv_linesizeuv_sizectxrly_ptruv_ptrs( r) from_dlpackzVideoFrame.from_dlpacks &5$-00 YF$/$7$77##Y''Y'' $/ / /11vvy  v;;!  Z $t^^ %**11.4k.4k ~  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!! '%"+/:/B OO $($5$5 CC&lK@@N'(LMMM<122c)nnDD(_#.4$044k**4+4+)Q#.4$044k**4+4+)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  sfi44A1k%c H tj|}|tjkrt d||jtjtjztjzzrt d|dd}t|j D]M} tj tj |j | jdz|kr|j | jdz}Ndg|z} t|D]q} d} t|j D]?} tj tj |j | j| kr | | | <| dz } @| dkrt d|drt||kr$t!|d|d t||j dj} | dkr| d krt d | d kr#|jtjzrt d | d z}|dvrt!d|j}|j}d}tj} t+}||j_t|D]} t1|| |}|jjt6krt d|jjdkrt!d|jjjdks*|jjj|ks|jjj dkrtCd| |jj"d krt!d| dtj tF|jj$d}tj tF|jj$d}| dkr|dkr;|dkr5tj t|}tj t|}n6|dks|dkrt!d||ks||krt!d|r|d zs|r|d zrt!d|||j_%||j_&| | }|dks|d ko|p|}|r| |z n|}|r| |z n|}||ks||krt!d| d|d|d|jj'tjkrT|jj'ddkrt!d| dtj tF|jj'd}ntj tF|}tj t|| z}tj t||z}tj tj(tR|jj*tj tj+|jj,z}tj-||t\tj tj/|d|jj0| <|jj0| tjkrtcd| ||jj*| <||jj2| <tj}|3|S#th$rB|tj5|j|tjkr|6|wxYw)Nzunknown pixel format zCfrom_dlpack does not support bitstream, palette, or Bayer formats ()rrzMfrom_dlpack supports nv12/p010le/p016le and planar single-component formats; z is not supportedz requires z plane(s), got rz=only 8- and 16-bit components are supported for DLPack importz6big-endian formats are not supported for DLPack importrrrz5only CPU DLPack tensors are supported for this formatrzunexpected dtype for plane zplane z must be 2D (H, W)rrzwidth/height must be even for z must have shape (z, z) must be contiguous in the last dimensionz"av_buffer_create failed for plane )7r"av_pix_fmt_desc_getrr'rraAV_PIX_FMT_FLAG_BITSTREAMAV_PIX_FMT_FLAG_PALAV_PIX_FMT_FLAG_BAYERrrRr r7rSrrr5rAV_PIX_FMT_FLAG_BE log2_chroma_w log2_chroma_hrrHrerrrr r.r!rrrr8rrr0r]r^r%r!rrLr=rrrrQrrJr(r=rSrBr)rZr\rer]r^rOr.desc nb_planesr comp_of_planeprXr*r log2_wlog2_hr'mphpwcomp_idx is_chromaexp_wexp_h pitch_elemsr(rrHs r)rzVideoFrame._from_dlpack_planars  #F + +  6;  %&Hf&H&HII I :  )% &' (   &("(((  !" t)** 3 3A{6:ty|'9::Q>JJ IaL.2 i y!!  AE4-..  ;vz49Q<+=>>!CC'(M!$QJEzz)M17MMM v;;) # #NNyNNVNN  $y|0 q==X]]%O  q==TZ#*@@=%H %-qL I % %BCC C!/!/ -3[Q %''E%EI 9%%C C #F1Iv66;*d22-O;(A--$%RSSSK%*a//{(->>{(.!33#$E!$E$EFFF;#q(($%Ca%C%C%CDDD[!+*;A*>??[!+*;A*>??66zzfkk & C 4 4!'S"!5!5!v{{()VWWWuf ()TUUUV519V&VVaZV()T&)T)TUUU&+EIO'-EI$(+%];h!mS&BRF 3<GUFv-..%4=IVG.//6;;"++$%Ta%T%T5%T%TE%T%T%TUUU;&&+55{*1-22(QQQQQ#)+gq{7J17M"N"NKK"(+gu"="=K!;sK(,BCC{35(899kN7+Q[-=K q{/FGGH$'#74fk&-QR6S6SUV$$ a 9=#v{22%&N1&N&NOOO$' q!(0 "1%K  ' ' ) ) )L    "59---FK !   s%P/YA Z!)rrr)F)rr)rF)rFF)rrrNNNNN).rorprqr:rrur"rvuintr8r=rApropertyrfrMrZr]r^r^rcrfsetterrmrqruryr~rrtrr cdivisionrr staticmethodrrrrrr7rrr`r!rrwr+r)rr4s,,,, \%C-%fk%6;%%%\%$ \  \ %%% W WX W   HHXH>XX**X*CCXC ""X"#####X#$$$$$X$%%%""X"###((X())) @ @ @ 7 7 7"3V"3"3"3"3H# # # JVdr r r r h444\ZZZ\Zx5%5%5%nVd}}}\}~  $$$$$$$\$L $#'#'bbbb b : bD[bD[bbb\bHRR\RRRr+r)Lsysenumrrcython.cimports.libavcimportslibavr"cython.cimports.av.dictionaryrcython.cimports.av.errorr$cython.cimports.av.sidedata.sidedatarcython.cimports.av.utilsrcython.cimports.av.video.formatr r cython.cimports.av.video.planer r r rcython.cimports.cpython.excr!cython.cimports.cpython.pycapsulerrrcython.cimports.cpython.refrrcython.cimports.hwcontext_cudarrcython.cimports.libc.stdintrrrrur; exceptvalr!rKvoidr*finalcclassr-rtrrQrrr rsupported_np_pix_fmtsrrrUr9rrrrrs byteorderrinlinerrr r1r5Framerwr+r)rVs   #########444444......EEEEEE222222IIIIIIIISSSSSSSSSSSS333333 =<<<<<<<HHHHHHHHCCCCCCCCCC)s45) [))))ggggggggTFN?4S8! M! . !! [!!!!3 M3 . !3 [3333(99VVVt$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@CMMMMDJvJ6;JJJJ WWWWWWWWWWr+