d jF.ddlZddlmcmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZddlmZmZmZdd lmZmZdd lmZdd lmZejejd dejej dejej!ej"dej"fdZ#ej$GddeZ%dS)N) CodecContext)HWAccel) err_check)Frame)Packet)avrational_to_fraction to_avrational) VideoFormat get_pix_fmtget_video_format) VideoFramealloc_video_frame)VideoReformatter)int64_tF)checkctxpix_fmtsreturnctjtjt|j}d}||dkr*|||jkr||S|dz }||dk*|jr|jn tj S)Nr) cythoncastpointerAVCodecPrivateDataopaquehardware_pix_fmtallow_software_fallback sw_pix_fmtlibAV_PIX_FMT_NONE)rr private_datais P/home/agentuser/manim-venv/lib/python3.11/site-packages/av/video/codeccontext.py_get_hw_formatr%s8>{)*CJ88LA 1+   A;,7 7 7A;  Q 1+   '>WCDWceZdZejdejejdejejej de dzfdZ ejde dzde fdZejde fd Zejd e d efd Zejd e fd ZedZejdefdZedZejdejfdZedZejdejfdZedZejdejfdZedZejdZedZejdZedZejdZedZejdejfdZed Z e jd!Z ed"Z!ed#Z"ed$Z#e#jdejfd%Z#ed&Z$ed'Z%ed(Z&e&jd)Z&ed*Z'e'jd+Z'ed,Z(e(jd-Z(ed.Z)e)jd/Z)ed0Z*e*jd1Z*ed2Z+e+jd3Z+ed4Z,e,jd5Z,dS)6VideoCodecContextptrcodechwaccelNctj||||| ||j|_t j|jj|j_|jj jj |j_ t|j_ |jj jj |j _|jj|j _t#j|j |j_dS#t($r d|_YdSwxYwdSN)r_initcreater* hwaccel_ctxr av_buffer_refr) hw_device_ctxconfigpix_fmtr% get_format _private_datarr+rraddressrNotImplementedError)selfr)r*r+s r$r.zVideoCodecContext._init*s 4eW555   (#*>>$*#=#= ),):4;K;O)P)P&#'#3#:#>#F &4#$+/7"3L8":#).1C"D"D& ( ( ($(     (  sCC..DDinputrc||sdgS|}|jj|jks*|j|jjks|j|jjkr\|jst |_|j||jj|jj|j|jj}|jj tj kr|jj |j_ |gS)N)threads) formatr4widthr)height reformatterrreformat thread_countptsr AV_NOPTS_VALUE frame_num)r9r:vframes r$_prepare_frames_for_encodez,VideoCodecContext._prepare_frames_for_encodeKs ==6M" M !T\ 1 1|tx~--}//# 6#3#5#5 %.. - /F :>S/ / /!X/FJNxr&ctSr-)rr9s r$_alloc_next_framez#VideoCodecContext._alloc_next_framefs """r&framepacketc^tj||||}|dSr-)r_setup_decoded_frame_init_user_attributes)r9rKrLrFs r$rNz&VideoCodecContext._setup_decoded_framejs4)$v>>>"$$&&&&&r&cl|j|S|jj|jjjjkr|S|jjr+|jjtjt|_ |S| }ttj|j|jd|j|_|SNr)r0r)r=r3r4 is_hw_owned device_idrrr _device_idrJrr av_hwframe_transfer_datarC)r9rKframe_sws r$_transfer_hwframez#VideoCodecContext._transfer_hwframeps   #L 9 t/6:B B BL   ' 8<8H8RFK E * * 5L0022#.x|UYJJKKKy r&cttjtj|jj|jj|jjSr-) r rrr AVPixelFormatr)r4r>r?rIs r$r=zVideoCodecContext.formats9 K)48+; < < HN HO   r&r=cl|j|j_|j|j_|j|j_dSr-)r4r)r>r?)r9r=s r$r=zVideoCodecContext.formats(!> -r&cD|jtjurdS|jjSrQ)r)rNULLr>rIs r$r>zVideoCodecContext.widths 8v{ " "1x~r&valuec||j_dSr-)r)r>r9r]s r$r>zVideoCodecContext.widthsr&cD|jtjurdS|jjSrQ)r)rr\r?rIs r$r?zVideoCodecContext.heights 8v{ " "1xr&c||j_dSr-)r)r?r_s r$r?zVideoCodecContext.heightsr&c|jjS)z The number of bits per sample in the codedwords. It's mandatory for this to be set for some formats to decode properly. Wraps :ffmpeg:`AVCodecContext.bits_per_coded_sample`. :type: int )r)bits_per_coded_samplerIs r$rcz'VideoCodecContext.bits_per_coded_samplesx--r&cJ|jrtd||j_dS)NzNot supported for encoders) is_encoder ValueErrorr)rcr_s r$rcz'VideoCodecContext.bits_per_coded_samples+ ? ;9:: :).&&&r&ctjtjtj|jj}tjt|jS)zE The pixel format's name. :type: str | None ) r av_pix_fmt_desc_getrrrYr)r4strname)r9descs r$r4zVideoCodecContext.pix_fmtsA  #FK0A48CS$T$T U U {3 ***r&c8t||j_dSr-)r r)r4r_s r$r4zVideoCodecContext.pix_fmts&u--r&cXttj|jjS)zZ The frame rate, in frames per second. :type: fractions.Fraction )rrr7r) frameraterIs r$rnzVideoCodecContext.framerates!&fnTX5G&H&HIIIr&c^t|tj|jjdSr-)r rr7r)rnr_s r$rnzVideoCodecContext.framerates'eV^DH,>??@@@@@r&c|jS)z#Another name for :attr:`framerate`.rnrIs r$ratezVideoCodecContext.rates ~r&c||_dSr-rqr_s r$rrzVideoCodecContext.rates r&cF|jrtd|jjS)zj Sets the number of frames between keyframes. Used only for encoding. :type: int %Cannot access 'gop_size' as a decoder is_decoder RuntimeErrorr)gop_sizerIs r$ryzVideoCodecContext.gop_sizes) ? HFGG Gx  r&cJ|jrtd||j_dS)Nrurvr_s r$ryzVideoCodecContext.gop_sizes, ? HFGG G!r&cXttj|jjSr-)rrr7r)sample_aspect_ratiorIs r$r|z%VideoCodecContext.sample_aspect_ratios%fnTX5Q&R&RSSSr&c^t|tj|jjdSr-)r rr7r)r|r_s r$r|z%VideoCodecContext.sample_aspect_ratios'eV^DH,HIIJJJJJr&c@tjtj|jtj|j|jj|jjjz|jj |jjjzdttj|S)Ni) r av_reducerr7numdenr)r>r|r?r)r9dars r$display_aspect_ratioz&VideoCodecContext.display_aspect_ratios{ N37 # # N37 # # HNTX9= = HOdh:> >     &fnS&9&9:::r&c4t|jjS)z :type: bool )boolr) has_b_framesrIs r$rzVideoCodecContext.has_b_framess DH)***r&c|jjS)aRaw ``has_b_frames`` value from FFmpeg (int, not bool). After :meth:`flush_buffers`, FFmpeg may reset the internal reorder heuristic. Set this to the known reorder depth *after* seeking to avoid dropped hierarchical B-frames. r)rrIs r$ reorder_depthzVideoCodecContext.reorder_depth sx$$r&c||j_dSr-rr_s r$rzVideoCodecContext.reorder_depth %r&c|jjSz :type: int )r) coded_widthrIs r$rzVideoCodecContext.coded_widths x##r&c|jjSr)r) coded_heightrIs r$rzVideoCodecContext.coded_heights x$$r&c|jjS)z Describes the signal range of the colorspace. Wraps :ffmpeg:`AVFrame.color_range`. :type: int r) color_rangerIs r$rzVideoCodecContext.color_range&sx##r&c||j_dSr-rr_s r$rzVideoCodecContext.color_range1s$r&c|jjS)z Describes the RGB/XYZ matrix of the colorspace. Wraps :ffmpeg:`AVFrame.color_primaries`. :type: int r)color_primariesrIs r$rz!VideoCodecContext.color_primaries5sx''r&c||j_dSr-rr_s r$rz!VideoCodecContext.color_primaries@s#(   r&c|jjS)z Describes the linearization function (a.k.a. transformation characteristics) of the colorspace. Wraps :ffmpeg:`AVFrame.color_trc`. :type: int r) color_trcrIs r$rzVideoCodecContext.color_trcDsx!!r&c||j_dSr-rr_s r$rzVideoCodecContext.color_trcOs"r&c|jjS)z Describes the YUV/RGB transformation matrix of the colorspace. Wraps :ffmpeg:`AVFrame.colorspace`. :type: int r) colorspacerIs r$rzVideoCodecContext.colorspaceSsx""r&c||j_dSr-rr_s r$rzVideoCodecContext.colorspace^s#r&c|jjS)zd The maximum run of consecutive B frames when encoding a video. :type: int r) max_b_framesrIs r$rzVideoCodecContext.max_b_framesbsx$$r&c||j_dSr-rr_s r$rzVideoCodecContext.max_b_frameskrr&c|jjS)z The minimum quantiser value of an encoded stream. Wraps :ffmpeg:`AVCodecContext.qmin`. :type: int r)qminrIs r$rzVideoCodecContext.qminox}r&c||j_dSr-rr_s r$rzVideoCodecContext.qminz r&c|jjS)z The maximum quantiser value of an encoded stream. Wraps :ffmpeg:`AVCodecContext.qmax`. :type: int r)qmaxrIs r$rzVideoCodecContext.qmax~rr&c||j_dSr-rr_s r$rzVideoCodecContext.qmaxrr&)-__name__ __module__ __qualname__rcfuncrr AVCodecContextconstAVCodecrr.rlistrGrJrrNrWpropertyr=setterr r>uintr?rcintr4rnrrryr|rrrrrrrrrrrrr&r$r(r((s \( ^C. /(~fl3;78(4 (((\(@ \ \4 \#5###\# \'%''''\'  \u\$  X  ]([(((]( X  \6;\X  ] FK   ] ..X.!/6:///"!/  + +X + ^..^.JJXJAAAX [[!!X!_"fj"""_" TTXTKK K ; ;X ;++X+ %%X%&6:&&&&$$X$ %%X% $$X$%%%((X()))""X"#####X#$$$%%X%&&&X [[X [[r&r()&rcython.cimports.libavcimportslibavr cython.cimports.av.codec.contextr cython.cimports.av.codec.hwaccelrcython.cimports.av.errorrcython.cimports.av.framercython.cimports.av.packetrcython.cimports.av.utilsrr cython.cimports.av.video.formatr r r cython.cimports.av.video.framer r$cython.cimports.av.video.reformatterrcython.cimports.libc.stdintrr exceptvalrrrrYr%cclassr(rr&r$rs #########999999444444......******,,,,,,JJJJJJJJVVVVVVVVVVHHHHHHHHAAAAAA////// * +nV\#*;<= .bbbbb bbbbbr&