\j4ddlZddlmcmZddlmZddlmZm Z m Z ddl m Z ddl mZmZddlmZddlmZmZmZddlmZmZej2ej4Gd d eZej:ej<ej>d d ej@e!dejDfdZ#ej:ej>d de$dejDfdZ%y)N) err_check)VideoFormatComponent get_pix_fmtget_video_format) VideoFrame)PyBUF_WRITABLEPyBuffer_FillInfo) Py_buffer)PyCapsule_GetPointerPyCapsule_IsValid PyCapsule_New)freemallocceZdZdedej fdZejdejfdZ e dZ ejdejfdZ dejed ej fd Zd Zd d ded zfdZy ) VideoPlaneframeindexcf|j}|jjrtjtj t j|jjj}t|j|jj|jj}|dk(r%|jdk(rd|_ d|_ d|_yt|jj D]X}|jj"|j$|k(s*t'||}|j|_ |j|_ nt)d|d|t+|j,jj.|j0|jz|_y)Npal8izcould not find plane z of )formatptr hw_frames_ctxcythoncastpointerlibAVHWFramesContextdatar sw_formatwidthheightname buffer_sizerange nb_componentscompplaner RuntimeErrorabsrlinesizer)selfrrfmt frames_ctxi components A/root/.hermes/venv/lib/python3.12/site-packages/av/video/plane.py __cinit__zVideoPlane.__cinit__sNll 99 " "@F s445uyy7N7N7S7SAJ#$$eiioouyy7G7GC A:#((f,DJDK&D  sww,,- KAww||A$$-0a8 &__ '..   K!6ugT#IJ J tzz~~66tzzBCdkkQreturnc|jS)N)r%r-s r2 _buffer_sizezVideoPlane._buffer_size6sr4c\|jjj|jS)zN Bytes per horizontal line in this plane. :type: int )rrr,rr7s r2 line_sizezVideoPlane.line_size:s!zz~~&&tzz22r4cF|jjjryy)NFT)rrrr7s r2_buffer_writablezVideoPlane._buffer_writableCs ::>> ' 'r4viewflagsc|jjjr td|tzr|j s t d|j}|jjj|j}|dkr9|j}tjtj||dz |zz}t||||jd|y)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  > !$*?*?*A56 6!--/ $  7 7 C q=!%F++fmmS1VaZ94LLC$c4+<+<+>5Ir4c<|jjjrutjt j |jjjtdk7r tdt|jjfStdfS)Ncuda4DLPack export is only implemented for CUDA hw framesr) rrrrrr AVPixelFormatrrNotImplementedErrorkCuda _device_idkCPUr7s r2__dlpack_device__zVideoPlane.__dlpack_device__Zsx ::>> ' '{{3,,djjnn.C.CD I*J4::001 1ayr4N)streamrNc |jjjdtjk(r t d|jjj rtjtj|jjjtdk7r td| tdtjtjtj|jjj j}|j }t"}|jj$}nJtjtj|jjj}t&}d}|j(}|dkr tdtd}td}td } tj*tj,\} } } tj*t.\} }}tj*t.\}}}||k(rd } d } |j0dk(rHd } |jjj2} |jjj4}|}d }nl|j0d k(rRd } |jjj2d z} |jjj4d z}d }|}d }d }n t7d||k(s|| k(rd } d} || zr t7d|j0dk(rKd } |jjj2} |jjj4}|| z}d }n|j0d k(rUd } |jjj2d z} |jjj4d z}d }|| z}d }d }n$t7d|t&k7r tdtj8|}|tjk(r td|j:tj<tj>ztj@zzr tdd}d}tC|jDD]}|jF|jH|j0k7r*|dk(r|jF|jJ}nItjtj,|jF|jJ|k7r td|d z }|dk(rt7d|j0d||zr td||z} | d k7r| d k7r td| d k(r(|j:tjLzr td| d z} || zr t7d|d k(r"d } |j2} |j4}|| z}d }n%d } |j2} |j4}|}|| z}|}d }tjN}|tjk(r tQdtStjT||jjtjtjt.tW| tjXt.z}tjtjt.tW| tjXt.z}|tjk(s|tjk(ro|tjk7r t[||tjk7r t[|tj\tj^|tQd| d k(r| |d<||d <||d<||d <n| |d<||d <||d <||d<||d <||d <tjtjtj`tWd tjXtj`z}|tjk(rIt[|t[|tj\tj^|tQdtjtj`||d<tjtj`||d <tjtj`||d <tjtjtbtWtjXtb}|tjk(rTt[|t[|t[|tj\tj^|tQdtjtj`|j|j0|jd_||jd_3||jd_4| |jd_5tmd | d |jd_7||jd_8||jd_9d|jd_:tjtj`||_;tx|_= t}tjtj`|dt~}|S#t$r ty|wxYw)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__es ::>>  a FKK /T  ::>> ' '{{3,,djjnn.C.CD I*J!)R AG s445tzz~~7S7S7X7XAJ ))FK --I[[!2!2DJJNN4I4IJFKINN q=%F 7#Y'Y'%~~fjj9dH^^G, Bw/ S# T>HDzzQZZ^^**ZZ^^))qZZ^^**a/ZZ^^))Q. !?@@ v 6!1HD8# !CDDzzQZZ^^**ZZ^^))8+qZZ^^**a/ZZ^^))Q.8+ !DEE D %&OP P''/ v{{")*RSSzz--))*++, *$ &'J !E4--. 99Q<%%3A:!%1!2!2J[[TYYq\->->?:M-Q  z 6$**5G!HIIE!)E"U*H1}Q)S1}c.D.D!D)La NN7 #VD6==3I,I%J ++ NN7 #VD6==3I,I%J  FKK 7fkk#9 #U &++%W   fnnY7 8/0 0 19E!HE!HGAJGAJE!HE!HE!HGAJGAJGAJkk NN6== )6!fmmFMM6R2R+S  &++  K M   fnnY7 8/0 0V]]I6AV]]E2AV]]G4A++ NN? +VFMM/4R-S  fkk ! I K M   fnnY7 8/0 0!'V]]INN4:: \\ fmm  33 \\&++ J&..";JFJJJ" 26kC$Jkr4rF)checkrr5c|tjk(rytjtjtj|j }|tjk7rtjtjt j|d}tjtjt|d}tjtjt|d}|tjk7r(t jtj||tjk7r t||tjk7r t|t|t|y)NrrrQ) rrYrrrirtrAVFramer\rgrhr)rrrrqrrs r2ruruSs &++ ++fnnV]]3W5H5H IC fkkKKs{{ ;SVD  FNN73SV<++fnnW5s1v>  #   fnnY7 8 FKK  K fkk ! M S Mr4rct|dr[tjtjtt |d}|tj k7r|j|yyy)NrW)r rrrrjr rYrv)rrs r2rwrwlsV+.++ NN? + + 6  fkk ! 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 ##. 6E4 5zzzz  ^^O , [[,%%6;;%%r4