o 2jT.@sddlZddlmmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZddlmZmZmZdd lmZmZdd lmZdd lmZejejd d dejej dejej!ej"dej"fddZ#ej$ej%GdddeZ&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_fmtsreturncCs`ttjt|j}d}||dkr'|||jkr||S|d7}||dks|jr-|jStj S)Nr) cythoncastpointerAVCodecPrivateDataopaquehardware_pix_fmtallow_software_fallback sw_pix_fmtlibAV_PIX_FMT_NONE)rr private_datair"P/var/www/html/whisper/venv/lib/python3.10/site-packages/av/video/codeccontext.py_get_hw_formats   r$c@seZdZejdejejdejejej de dBfddZ ejde dBde fd d Zejde fd d Zejd e defddZejd e fddZeddZejdefddZeddZejdejfddZeddZejdejfddZeddZejdejfd dZed!d"Zejd#d"Zed$d%Zejd&d%Zed'd(Zejd)d(Zed*d+Zejdejfd,d+Zed-d.Z e jd/d.Z ed0d1Z!ed2d3Z"ed4d5Z#e#jdejfd6d5Z#ed7d8Z$ed9d:Z%ed;d<Z&e&jd=d<Z&ed>d?Z'e'jd@d?Z'edAdBZ(e(jdCdBZ(edDdEZ)e)jdFdEZ)edGdHZ*e*jdIdHZ*edJdKZ+e+jdLdKZ+edMdNZ,e,jdOdNZ,dS)PVideoCodecContextptrcodechwaccelNcCst|||||durOz5||j|_t|jj|j_|jj jj |j_ t |j_ |jj jj |j _|jj|j _t|j |j_WdStyNd|_YdSwdSN)r_initcreater' hwaccel_ctxr av_buffer_refr& hw_device_ctxconfigpix_fmtr$ get_format _private_datarr(rraddressrNotImplementedError)selfr&r'r(r"r"r#r*+s    zVideoCodecContext._initinputrcCs|dus|s dgS|}|jj|jks |j|jjks |j|jjkr9|js't|_|jj||jj|jj|j|jjd}|jj t j krF|jj |j_ |gS)N)threads) formatr0widthr&height reformatterrreformat thread_countptsrAV_NOPTS_VALUE frame_num)r5r6vframer"r"r#_prepare_frames_for_encodeLs$  z,VideoCodecContext._prepare_frames_for_encodecCstSr))r r5r"r"r#_alloc_next_framegsz#VideoCodecContext._alloc_next_frameframepacketcCst||||}|dSr))r_setup_decoded_frame_init_user_attributes)r5rErFrAr"r"r#rGks z&VideoCodecContext._setup_decoded_framecCsn|jdur|S|jj|jjjjkr|S|jjr"|jjtt |_ |S| }t t |j|jd|j|_|SNr)r,r&r8r/r0 is_hw_owned device_idrrr _device_idrDrrav_hwframe_transfer_datar>)r5rEframe_swr"r"r#_transfer_hwframeqs z#VideoCodecContext._transfer_hwframecCs"tttj|jj|jj|jjSr)) r rrr AVPixelFormatr&r0r9r:rCr"r"r#r8s zVideoCodecContext.formatr8cCs"|j|j_|j|j_|j|j_dSr))r0r&r9r:)r5r8r"r"r#r8s  cC|jtjurdS|jjSrI)r&rNULLr9rCr"r"r#r9 zVideoCodecContext.widthvaluecC ||j_dSr))r&r9r5rTr"r"r#r9 cCrQrI)r&rrRr:rCr"r"r#r:rSzVideoCodecContext.heightcCrUr))r&r:rVr"r"r#r:rWcC|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_samplerCr"r"r#rY z'VideoCodecContext.bits_per_coded_samplecC|jrtd||j_dS)NzNot supported for encoders) is_encoder ValueErrorr&rYrVr"r"r#rYs cCs&tttj|jj}tt|jS)zE The pixel format's name. :type: str | None ) rav_pix_fmt_desc_getrrrPr&r0strname)r5descr"r"r#r0szVideoCodecContext.pix_fmtcCst||j_dSr))r r&r0rVr"r"r#r0scCtt|jjS)zZ The frame rate, in frames per second. :type: fractions.Fraction )rrr3r& frameraterCr"r"r#rcszVideoCodecContext.frameratecCt|t|jjdSr))rrr3r&rcrVr"r"r#rccCs|jS)z#Another name for :attr:`framerate`.rcrCr"r"r#rateszVideoCodecContext.ratecCs ||_dSr)rfrVr"r"r#rgs cCs|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_sizerCr"r"r#rlszVideoCodecContext.gop_sizecCr[)NrhrirVr"r"r#rls cCrbr))rrr3r&sample_aspect_ratiorCr"r"r#rmsz%VideoCodecContext.sample_aspect_ratiocCrdr))rrr3r&rmrVr"r"r#rmrecCsLtt|jt|j|jj|jjj|jj |jjjdt t|S)Ni) r av_reducerr3numdenr&r9rmr:r)r5darr"r"r#display_aspect_ratios  z&VideoCodecContext.display_aspect_ratiocCs t|jjS)z :type: bool )boolr& has_b_framesrCr"r"r#rts zVideoCodecContext.has_b_framescCrX)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&rtrCr"r"r# reorder_depth szVideoCodecContext.reorder_depthcCrUr)rurVr"r"r#rvrWcCrXz :type: int )r& coded_widthrCr"r"r#rxzVideoCodecContext.coded_widthcCrXrw)r& coded_heightrCr"r"r#rz ryzVideoCodecContext.coded_heightcCrX)z Describes the signal range of the colorspace. Wraps :ffmpeg:`AVFrame.color_range`. :type: int r& color_rangerCr"r"r#r|'rZzVideoCodecContext.color_rangecCrUr)r{rVr"r"r#r|2rWcCrX)z Describes the RGB/XYZ matrix of the colorspace. Wraps :ffmpeg:`AVFrame.color_primaries`. :type: int r&color_primariesrCr"r"r#r~6rZz!VideoCodecContext.color_primariescCrUr)r}rVr"r"r#r~ArWcCrX)z Describes the linearization function (a.k.a. transformation characteristics) of the colorspace. Wraps :ffmpeg:`AVFrame.color_trc`. :type: int r& color_trcrCr"r"r#rErZzVideoCodecContext.color_trccCrUr)rrVr"r"r#rPrWcCrX)z Describes the YUV/RGB transformation matrix of the colorspace. Wraps :ffmpeg:`AVFrame.colorspace`. :type: int r& colorspacerCr"r"r#rTrZzVideoCodecContext.colorspacecCrUr)rrVr"r"r#r_rWcCrX)zd The maximum run of consecutive B frames when encoding a video. :type: int r& max_b_framesrCr"r"r#rcszVideoCodecContext.max_b_framescCrUr)rrVr"r"r#rlrWcCrX)z The minimum quantiser value of an encoded stream. Wraps :ffmpeg:`AVCodecContext.qmin`. :type: int r&qminrCr"r"r#rprZzVideoCodecContext.qmincCrUr)rrVr"r"r#r{rWcCrX)z The maximum quantiser value of an encoded stream. Wraps :ffmpeg:`AVCodecContext.qmax`. :type: int r&qmaxrCr"r"r#rrZzVideoCodecContext.qmaxcCrUr)rrVr"r"r#rrW)-__name__ __module__ __qualname__rcfuncrrAVCodecContextconstAVCodecrr*rlistrBrDrrGrOpropertyr8setterr r9uintr:rYintr0rcrgrlrmrrrtrvrxrzr|r~rrrrrr"r"r"r#r%(s                                      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.utilsrrcython.cimports.av.video.formatr r r cython.cimports.av.video.framer r $cython.cimports.av.video.reformatterrcython.cimports.libc.stdintrr exceptvalrrrrPr$finalcclassr%r"r"r"r#s0