m il &ddlZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZeZhd Zejd efd ZGd d eZejdejfdZejdedejdejdejfdZejdedejfdZej ddedejdefdZejdedejfdZej Gdde!ZdS) N)IntEnum) err_checkget_display_rotation) check_ndarray) get_pix_fmtget_video_format VideoPlane)uint8_t>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_rggb16lereturnc*ttS)zGet a mostly uninitialized VideoFrame. You MUST call VideoFrame._init(...) or VideoFrame._init_user_attributes() before exposing to the user. ) VideoFrame_cinit_bypass_sentinelN/var/www/html/auto_sub_bot/venv/lib/python3.11/site-packages/av/video/frame.pyalloc_video_framerhgs , - --rfc~eZdZejZejZejZ ej Z ej Z ejZejZejZdS) PictureTypeN)__name__ __module__ __qualname__libAV_PICTURE_TYPE_NONENONEAV_PICTURE_TYPE_IIAV_PICTURE_TYPE_PPAV_PICTURE_TYPE_BBAV_PICTURE_TYPE_SSAV_PICTURE_TYPE_SISIAV_PICTURE_TYPE_SPSPAV_PICTURE_TYPE_BIBIrerfrgrjrjrsN  #D A A A A B B BBBrfrj big_endiancVtjdk|kr|S|S)Nbig)sys byteorderbyteswap)arrayrs rgbyteswap_arrayr}s) :--~~ Lrfplanebytes_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)Nr)widthabs line_sizecythondeclareintheightrange) img_bytesrrrri_bufi_posi_strideo_bufo_poso_stride start_rowend_rowsteprowijs rgcopy_bytes_to_planersQ'0EE#kO;HEE!%/22H%~fj99IwL1$  ,Y..  h  N1h88  //A+0(1,>B,E%!)a-((  /4EEH) __class__rkptsrnamerridrs rg__repr__zVideoFrame.__repr__su =4>* = =$( = =T[=M = =z = = K = =/1$xx < = = = rfczd}tjjjD])}jjj|jdz}||kr|}*jjdkrd}d}||kr/jj|r|dz }||krjj|tfdt|DS)z: A tuple of :class:`.VideoPlane` objects. rrrc0g|]}t|Srer ).0rrs rg z%VideoFrame.planes..s#FFFajq))FFFrf) rrr nb_componentscomprr extended_datatuple)rmax_plane_countrcount plane_counts` rgplaneszVideoFrame.planess'(t{455 ( (AKO(+1A5E&&"' ; v % %O"# O++0F{0S+ 1 KO++0F{0S+FFFF53E3EFFFGGGrfc|jjS)zWidth of the image, in pixels.)rrrs rgrzVideoFrame.widthsx~rfc|jjS)zHeight of the image, in pixels.)rrrs rgrzVideoFrame.heightsxrfc 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]. rrs rgrotationzVideoFrame.rotations$D)))rfcNt|jjtjzS)z+Is this frame an interlaced or progressive?)boolrflagsrnAV_FRAME_FLAG_INTERLACEDrs rginterlaced_framezVideoFrame.interlaced_frame'sDHNS%AABBBrfc|jjS)zReturns an integer that corresponds to the PictureType enum. Wraps :ffmpeg:`AVFrame.pict_type` :type: int r pict_typers rgrzVideoFrame.pict_type-sx!!rfc||j_dSrrrvalues rgrzVideoFrame.pict_type7s"rfc|jjS)zKColorspace of frame. Wraps :ffmpeg:`AVFrame.colorspace`. r colorspacers rgrzVideoFrame.colorspace;sx""rfc||j_dSrrrs rgrzVideoFrame.colorspaceDs#rfc|jjS)zMColor range of frame. Wraps :ffmpeg:`AVFrame.color_range`. r color_rangers rgrzVideoFrame.color_rangeHsx##rfc||j_dSrrrs rgrzVideoFrame.color_rangeQs$rfcb|jst|_|jj|g|Ri|S)areformat(width=None, height=None, format=None, src_colorspace=None, dst_colorspace=None, interpolation=None) Create a new :class:`VideoFrame` with the given width/height/format/colorspace. .. seealso:: :meth:`.VideoReformatter.reformat` for arguments. ) reformatterVideoReformatterreformat)rargskwargss rgrzVideoFrame.reformatUsC 2/11D (t(???????rfc |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' rrre)r)rrs rgto_rgbzVideoFrame.to_rgbas!t}66G6v666rffilepathcJ|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.mjpegpngr<rr)openwupdate1)options)pix_fmt)rN) endswithrav.container.corer add_streamrrmuxencoder)rr is_jpgencoderrroutput output_streams rgsavezVideoFrame.saveos   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)ImagerrRGBrawrre) PILrrrrrr bytearray frombytesbytes) rrrrrrrrro_sizers rgto_imagezVideoFrame.to_images )DMCCCFCCJ1M*/ "'/ "'+/ % x 7$V,,fnn.3EEHdd?dd@didAddBddCddDd dEddFddGd,dHd,dIddJddKddLd dMddNd dOd dPdQdRdQdSdSdTdTd ddddd dUdV jjdW\BfdXjD}t|dYkr |d}n| |dZ}t|jj d[}|j ddYkr| dSjjd\r_|dddddf}|dddddf|dddddf<|dddddYf|dddddf<||dddddYf<|s%jjd]vr||dd}|Sjjd^vry|t%jdt%jdYt%jdgd_jSjjd`krt%jdddajj}t%jdYddajjdz}t%jdddajjdz} ||ddYZ} || ddYZ} |r||| | gdZS||| | gdZSjjdbkrt%jdjj} |jdYdcddd_dY|j} | | fSjjdekra|t%jdt%jdYdgd_jSt5dfjjdg)iaIGet 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. rN>r#r%r<rTrz.the width has to be even for this pixel formatz/the height has to be even for this pixel formatr )rrrPrrrQrRrSrZ)ruint16rYr\r[r^r]r`r_r)r rr)r,rrrr&)r,r'rr>r?r@rArBrCrDrErN)r*float32rOrr(r)r*r+r,r-r.r/rrrFrGrr0r1r2r3r4r5r6r7rrr rHrIrr!r"rrr8r9rJ)r.float16rK)rr/) r/)rr) rLrMr:r;r$rUrVrWrXr=r%)NNcpg|]2}t|jjd3S)r)rrrr)rplanrframeitemsizes rgrz)VideoFrame.to_ndarray..sRT8U33;;L%+rrfraxisbegbrr$r=r#r<rrTr,ri4>i4rz'Conversion to numpy array with format `` is not yet supportedre)rrrrrrgetrlen concatenaterrrsqueeze startswithcopymoveaxishstackrrrepeatstackrastyperrr)r channel_lastrrlayersrbufferyuvu_fullv_fullimagepaletterr4r5s @@@rg to_ndarrayzVideoFrame.to_ndarrays,*DM33F33 <  Q Q Q;?a'''@('' .?P @ .AP B LCP D  EP P P F  GP H  IP J  KP L  MP N  OP P  QP R  SP T }UP V }WP X YP Z [P \ L]P ^  _P `  aP b  cP d  eP f  gP P P h  iP j  kP l  mP n \oP p }qP r }sP t uP v wP x \yP z }{P | }}P ~ LP @ LAP B  CP D  EP F GP H IP P J)(''#(())$#_P P P ` #el . .a %b  "L F 6{{aq vA66"5%,*;*D*DT*J*JKKE{1~""}}Q'''| ++E22 (qqq!!!Qw,,..!&qqq!!!QwaaaAg!&qqq!!!QwaaaAg!'aaaAg 1EL$59P$P$P E1a00L <  7 7 799 a11 a11 a11 gb%+&&  ' <  - -U\!_a::BB ekAU\!_a::BB ekQ.AU\!_a::BB ekQ.A YYq!!Y,,FYYq!!Y,,F =xxFF 3!x<<<88Q/a888 8 <  & & a1199%, TTE el1ot44Qbh  '> ! <  & &99 a11 a!44 gb%+&&  ' _el6G _ _ _   rfc~|jdkr|d}t||jdddS)z6 Update content from a ``PIL.Image``. r!rr N)modeconvertrr)rimgs rg set_imagezVideoFrame.set_imagees@ 8u  ++e$$CCQ33333rfc~t|jd|jdd}|||S)z9 Construct a frame from a ``PIL.Image``. rrr)rcsizerY)rXr4s rg from_imagezVideoFrame.from_imagens8 'sx{CHQKII  rfrcdddl}|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. rNrrrrr r)r rz.provided array does not have C_CONTIGUOUS rowsrrr!r"r,)r-rr:r;r/)r1r*r rrrr*)r*rr&r'r8r9)r.rrJrKr0rLrM)rr*>rrrrrPrQrRrS>rr r0r1r2r3r4r5r6r7rYrZr[r\r]r^r_r`>rHrI>r> rrr(r)r*r+r,r-r.r/>rFrG>r>r>r?r@rArBrCrDrE>rNrO>rr#r<r-r;)Conversion from numpy array with format `r>r.)rrrr rr)rrr) rrrrstridesrrCascontiguousarrayrErcrd_image_fill_pointers_numpy)rrrrr linesizesr4s rgfrom_numpy_bufferzVideoFrame.from_numpy_bufferxs 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 rfc|jj}tjtjt |}t |}ttj |j ||_ ||j _ ||j _||j _t!|D]\} } | |j j| <t|j jtjtj|j j |j j||j j} | rt)| |dSr)ctypesrrrpointerr rrnrrrrrrrr enumeraterav_image_fill_pointersrrr) rrLrrrirc_datac_ptrrrrrs rgrhz%VideoFrame._image_fill_pointers_numpyUs*# FN73V<<v&& V^DL11222!" $Y// , ,KAx#+DH a (( HM K)48? ; ; HO  H       cNNN ""$$$$$rfc 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|dQkrct1||jr|j|jkrt9dR|r)|jdAdBkr||dAd}n |jddBkst9dS|jdEd\} }|dAzdks | dAzdkrt9dTt|| |}t|d|jddA|dEdddddAf}|dAdddddAf}t||jdEdAt||jdAdA|S|dUkrvt|dIdBt ||jddAzdkt ||jdEdAzdkt ||jdAdAknw|dVvr2t|dIdBt ||jdAdBknA|dWvr2t|dIdBt ||jdAdOkn |dXvrt|dYdBt ||jdAdBkt|jdE|jd|}tt||dD|jddZ|S|d[vrt|d\dBt ||jdAdBkt|jdE|jd|}tt||dD|jdd]|S|d^vrt|dYdBt ||jdAdOkt|jdE|jd|}tt||dD|jdd_|S|d`vrt|dadBt ||jdAdOkt|jdE|jd|}tt||dD|jdd_|S|dbvrt|d\dBt ||jdAdOkt|jdE|jd|}tt||dD|jddc|S|ddkrt|dIdAt ||jddBzdkt ||jdEdAzdkt|jdE|jddAzdBz|}|j|jz}|dP} t| d||jddEt| |d|jdEdA|St9de|dft|jdE|jd|}t||jd|jdAkrdEn |jdA|S)ga 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. rNrY)rrr,rZrP)rrrr[r\rQr]r^rRr_r`rSrr)r*rrr>)r*rr,r?r@rArBrCrDrErN)r*r*r/rOr)r rrr()r rr,r)r*r+r,r-r.r/rrrF)r r*r/rGrr0r1r2r3r4r5r6r7rrr rH)rr*r/rI)rr$rUrVrWrXr=)NNNrr r:r8rr9r6rr)r*r=r<r;r*rrTzArray must be uint16 typez>Array must have shape (3, height, width) or (height, width, 3)zWidth and height must be evenr%r^rar_r,r-r`r/r1rbr.rcr0rdrrrer>)rr?ndimrrErrrrrcrrrCrArrrrrrIrrrr isinstancendarrayrr,rrD)rrrJrchannelsr5rr4rrSu_startv_startflatrrrNrOuv_starts rg 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&QQEkEL0G'kQ&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    rfrrc t|||}|dkrt||jdd||nM|dvr6t||jd|drdnd||nt d|d |S) Nrrr*> rPrQrRrSrYrZr[r\r]r^r_r`8rrzFormat 'z' is not supported.)rcrrrNotImplementedError)rrrrrrr4s rg from_byteszVideoFrame.from_bytescs5&&11 V   5<?A           Q__S))0q     &&L&L&L&LMM M rfN)rrr#)F)rr)rF)rFF)'rkrlrmrrcfuncrnruintrrrrpropertyrrrrrrsetterrrrr ccallobjectrr(rTrY staticmethodr\rjrhr}r&rrrerfrgrcrcs,,,, \%C-%fk%6;%%%\%* \  \    HHXH(XX**X*CCXC ""X"#####X#$$$$$X$%%% @ @ @ 7 7 7 \3V333\36   rs  ......EEEEEE222222IIIIIIII555555//////UUUp.:....     '   fk # #[#[ # ; ####LHjH6;HHHH FM%% %(. %@C%%%%$JvJ6;JJJJ }}}}}}}}}}rf