j4ddlZddlmcmZddlmZddlmZm Z m Z ddl m Z ddl mZmZddlmZddlmZmZmZddlmZmZejejGd d eZejejejd d ej e!dej"fdZ#ejejd de$dej"fdZ%dS)N) err_check)VideoFormatComponent get_pix_fmtget_video_format) VideoFrame)PyBUF_WRITABLEPyBuffer_FillInfo) Py_buffer)PyCapsule_GetPointerPyCapsule_IsValid PyCapsule_New)freemallocceZdZdedejfdZejdejfdZ e dZ ejdej fdZ dejed ejfd Zd Zd d ded zfdZd S) VideoPlaneframeindexc|j}|jjrhtjtjt j|jjj}t|j |jj |jj }|dkr"|j dkrd|_ d|_ d|_dSt|jjD]G}|jj|j|kr*t'||}|j |_ |j |_ nHt)d|d|t+|jjj|j|j z|_dS)Npal8izcould not find plane z of )formatptr hw_frames_ctxcythoncastpointerlibAVHWFramesContextdatar sw_formatwidthheightname buffer_sizerange nb_componentscompplaner RuntimeErrorabsrlinesizer)selfrrfmt frames_ctxi components s/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/av/video/plane.py __cinit__zVideoPlane.__cinit__sDl 9 " @F s45uy7N7SAAJ#$eiouy7GC A::#(f,,DJDK&D  Fsw,-- K KAw|A$--0a88 &_ '.  . IuII#IIJJ J tz~6tzBCCdkQreturnc|jS)N)r%r-s r2 _buffer_sizezVideoPlane._buffer_size6s r4c:|jjj|jS)zN Bytes per horizontal line in this plane. :type: int )rrr,rr7s r2 line_sizezVideoPlane.line_size:sz~&tz22r4c,|jjjrdSdS)NFT)rrrr7s r2_buffer_writablezVideoPlane._buffer_writableCs :> ' 5tr4viewflagsc|jjjrtd|tzr#|st d|}|jjj|j }|dkr/|j }tj tj ||dz |zz}t||||d|dS)NzyHardware frame planes do not support the Python buffer protocol. Use DLPack (__dlpack__) or download to a software frame.zbuffer is not writablerr)rrr TypeErrorrr< ValueError _buffer_ptrr,rr#rrp_charr r8)r-r=r>rr:r#s r2 __getbuffer__zVideoPlane.__getbuffer__Is :> ' K  > ! 7$*?*?*A*A 7566 6!--// $  7 C q==!%F+fmS11VaZ94LLC$c4+<+<+>+>5IIIIIr4c|jjjratjt j|jjjtdkrtdt|jj fStdfS)Ncuda4DLPack export is only implemented for CUDA hw framesr) rrrrrr AVPixelFormatrrNotImplementedErrorkCuda _device_idkCPUr7s r2__dlpack_device__zVideoPlane.__dlpack_device__Zsx :> ' 2{3,djn.CDD II*J4:01 1ayr4N)streamrNc |jjjdtjkrt d|jjjrtjtj |jjj tdkrtd|tdtjtj tj|jjjj}|j}t"}|jj}n7tjtj |jjj }t&}d}|j}|dkrtdtd}td}td } tjtj\} } } tjt.\} }}tjt.\}}}||krd } d } |jdkr*d } |jjj} |jjj}|}d }n|jd kr4d } |jjjd z} |jjjd z}d }|}d }d }nt7d||ks|| krd } d} || zrt7d|jdkr-d } |jjj} |jjj}|| z}d }n?|jd kr7d } |jjjd z} |jjjd z}d }|| z}d }d }nt7d|t&krtdtj|}|tjkrtd|jtjtjztj zzrtdd}d}tC|j"D]~}|j#|j$|jkr|dkr|j#|j%}nBtjtj|j#|j%|krtd|d z }|dkrt7d|jd||zrtd||z} | d kr| d krtd| d kr#|jtj&zrtd| d z} || zrt7d|d krd } |j} |j}|| z}d }nd } |j} |j}|}|| z}|}d }tj'}|tjkrtQdtStj*||jjtjtj t.tW| tj,t.z}tjtj t.tW| tj,t.z}|tjks|tjkrs|tjkrt[||tjkrt[|tj.tj/|tQd| d kr| |d<||d <||d<||d <n| |d<||d <||d <||d<||d <||d <tjtj tj0tWd tj,tj0z}|tjkrSt[|t[|tj.tj/|tQdtjtj0||d<tjtj0||d <tjtj0||d <tjtj tbtWtj,tb}|tjkrbt[|t[|t[|tj.tj/|tQdtjtj0|j|j|j2_||j2_3||j2_4| |j2_5tmd | d |j2_7||j2_8||j2_9d|j2_:tjtj0||_;tx|_= t}tjtj0|dt~}n#t$rty|wxYw|S)NrzBDLPack export requires a refcounted AVFrame (frame.buf[0] is NULL)rFrGziCUDA stream synchronization is not supported. Pass stream=None and synchronize before calling __dlpack__.z4negative linesize is not supported for DLPack exportsnv12sp010lesp016lerzinvalid plane index for NV12z linesize is not aligned to dtypez!invalid plane index for P010/P016z'unsupported sw_format for DLPack exportz&unknown pixel format for DLPack exportzIbitstream, palette, and Bayer formats are not supported for DLPack exportz7mixed component step is not supported for DLPack exportzplane z has no componentsz/unsupported component packing for DLPack exportz=only 8- and 16-bit components are supported for DLPack exportz6big-endian formats are not supported for DLPack exportzav_frame_alloc() failedzmalloc() failed)codebitslanesdltensor)ArrbufrNULLr@rrrrHrrrIrrr r!rJrKrLr:declareintint64_trr#r"rAav_pix_fmt_desc_getr>AV_PIX_FMT_FLAG_BITSTREAMAV_PIX_FMT_FLAG_PALAV_PIX_FMT_FLAG_BAYERr&r'r(r)stepAV_PIX_FMT_FLAG_BEav_frame_alloc MemoryErrorr av_frame_refrsizeofr av_frame_freeaddressp_voidDLManagedTensor dl_tensor device_type device_idndim DLDataTypedtypeshapestrides byte_offset manager_ctx_dlpack_managed_tensor_deleterdeleterr _dlpack_capsule_destructor Exception)r-rNr/sw_fmtrlrmr:nv12p010lep016lernrUitemsizes0s1s2st0st1st2desc step_bytesncompr0 frame_refrqrrctxmanagedcapsules r2 __dlpack__zVideoPlane.__dlpack__esO :> a FK / /T  :> ' {3,djn.CDD II*J!)R AG s45tz~7S7XAAJ )FK -II[!2DJN4IJJFKIN q==%F 7##Y''Y''%~fj99dH^G,, Bw// S# T>>HDzQZ^*Z^)qZ^*a/Z^)Q. !?@@@ v  6!1!1HD8# E !CDDDzQZ^*Z^)8+qZ^*a/Z^)Q.8+ !DEEE D %&OPP P'// v{"")*RSSSz-)*+,  *$ &'J !E4-..  9Q<%33A::!%1!2JJ[TYq\->??:MM-Q zz !H$*!H!H!HIIIE! )E"U*H1}}Q)S1}}c.D!D})La>> N7 #VD6=3I3I,I%J%J  + N7 #VD6=3I3I,I%J%J   FK  7fk#9#9 ##U &+%%W  fnY77 8 8 8/00 0 199E!HE!HGAJGAJJE!HE!HE!HGAJGAJGAJk N6= )6!fmFM6R6R2R+S+S   &+   KKK MMM  fnY77 8 8 8/00 0V]I66AV]E22AV]G44A+ N? +VFM/4R4R-S-S   fk ! ! III KKK MMM  fnY77 8 8 8/00 0!'V]IN4: \ fm   \ 33X3 \&+\ J&.";JFJJJJJ"   26kkkC$Jkkkkkkr4rF)checkrr5c|tjkrdStjtjtj|j}|tjkrtjtjt j|d}tjtjt|d}tjtjt|d}|tjkr&t j tj ||tjkrt||tjkrt|t|t|dS)NrrrQ) rrYrrrirtrAVFramer\rgrhr)rrrrqrrs r2ruruSs &+ +fnV]3W5H I IC fkKs{ ;SVDD  FN73SV<<+fnW5s1v>>  # #  fnY77 8 8 8 FK   KKK fk ! ! MMM S MMMMMr4rct|dr_tjtjtt |d}|tjkr||dSdSdS)NrW)r rrrrjr rYrv)rrs r2rwrwlsw+..%+ N? + + 6 6   fk ! ! OOG $ $ $ $ $ %% " !r4)&rcython.cimports.libavcimportslibavrcython.cimports.av.errorrcython.cimports.av.video.formatrrrcython.cimports.av.video.framercython.cimports.cpythonrr cython.cimports.cpython.bufferr !cython.cimports.cpython.pycapsuler r r cython.cimports.libc.stdlibrrfinalcclassPlanerrnogil exceptvalrrjvoidruobjectrwrr4r2rs #########...... 655555EEEEEEEE444444 54444444zzzzzzzzz  ^O , [,%%6;%%%%%%r4