:jT.ddlZddlmcmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZddlmZmZmZdd lmZmZdd lmZdd lmZej:ej<d dej>ej@dej>ejBejDdejDfdZ#ejHejJGddeZ&y)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}||dk7r%|||j k(r||S|dz }||dk7r%|j r |jStjS)Nr) cythoncastpointerAVCodecPrivateDataopaquehardware_pix_fmtallow_software_fallback sw_pix_fmtlibAV_PIX_FMT_NONE)rr private_datais j/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/av/video/codeccontext.py_get_hw_formatr%s8>{{)*CJJ8LA 1+  A;,77 7A;  Q 1+  '>>DGDWDWceZdZejdej ejdej ejejde dzfdZ ejde dzde fdZejde fd Zejd e d efd Zejd e fd ZedZej*defdZedZej*dej0fdZedZej*dej0fdZedZej*dej6fdZedZej*dZedZej*dZedZej*dZedZej*dej6fdZed Z e j*d!Z ed"Z!ed#Z"ed$Z#e#j*dej6fd%Z#ed&Z$ed'Z%ed(Z&e&j*d)Z&ed*Z'e'j*d+Z'ed,Z(e(j*d-Z(ed.Z)e)j*d/Z)ed0Z*e*j*d1Z*ed2Z+e+j*d3Z+ed4Z,e,j*d5Z,y)6VideoCodecContextptrcodechwaccelNctj|||||4 |j|j|_t j |jj|j_|jjjj|j_ t|j_ |jjjj|j_|jj |j_t#j$|j|j_yy#t($r d|_YywxYwN)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  4eW5   (#*>>$**#= ),):):4;K;K;O;O)P&#'#3#3#:#:#>#>#F#F &4#$$++//77""3LL88"":#)..1C1C"D ' ($(  (sD2EE$#E$inputrc||sdgS|}|jj|jk7sF|j|jjk7s#|j|jjk7r|j st |_|j j||jj|jj|j|jj}|jjtjk(r%|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_encodeLs =6M" MM ! !T\\ 1||txx~~-}}/###3#5 %%.. -- /F ::>>S// /!XX//FJJNxr&ctSr-)rr9s r$_alloc_next_framez#VideoCodecContext._alloc_next_framegs  ""r&framepacketcVtj||||}|jyr-)r_setup_decoded_frame_init_user_attributes)r9rKrLrFs r$rNz&VideoCodecContext._setup_decoded_frameks%))$v>"$$&r&c|j|S|jj|jjjjk7r|S|jj r5|jj tjt|_ |S|j}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_hwframeqs    #L 99  t//66::BB BL    ' '8<8H8H8R8RFKK E * 5L002#..x||UYYJKyy r&cttjtj|j j |j j|j jSr-) r rrr AVPixelFormatr)r4r>r?rIs r$r=zVideoCodecContext.formatsC KK))488+;+; < HHNN HHOO  r&r=c|j|j_|j|j_|j|j_yr-)r4r)r>r?)r9r=s r$r=zVideoCodecContext.formats0!>> --r&ch|jtjury|jjSrQ)r)rNULLr>rIs r$r>zVideoCodecContext.widths# 88v{{ "xx~~r&valuec&||j_yr-)r)r>r9r]s r$r>zVideoCodecContext.widths r&ch|jtjury|jjSrQ)r)rr\r?rIs r$r?zVideoCodecContext.heights# 88v{{ "xxr&c&||j_yr-)r)r?r_s r$r?zVideoCodecContext.heights r&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_samplesxx---r&cT|jr td||j_y)NzNot supported for encoders) is_encoder ValueErrorr)rcr_s r$rcz'VideoCodecContext.bits_per_coded_samples! ??9: :).&r&ctjtjtj|j j }tjt|jS)zE The pixel format's name. :type: str | None ) r av_pix_fmt_desc_getrrrYr)r4strname)r9descs r$r4zVideoCodecContext.pix_fmtsI  # #FKK0A0A488CSCS$T U {{3 **r&c8t||j_yr-)r r)r4r_s r$r4zVideoCodecContext.pix_fmts&u-r&cfttj|jjS)zZ The frame rate, in frames per second. :type: fractions.Fraction )rrr7r) frameraterIs r$rnzVideoCodecContext.framerates"&fnnTXX5G5G&HIIr&cjt|tj|jjyr-)r rr7r)rnr_s r$rnzVideoCodecContext.framerateseV^^DHH,>,>?@r&c|jS)z#Another name for :attr:`framerate`.rnrIs r$ratezVideoCodecContext.rates~~r&c||_yr-rqr_s r$rrzVideoCodecContext.rates r&c\|jr td|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' ??FG Gxx   r&cT|jr td||j_y)Nrurvr_s r$ryzVideoCodecContext.gop_sizes! ??FG G!r&cfttj|jjSr-)rrr7r)sample_aspect_ratiorIs r$r|z%VideoCodecContext.sample_aspect_ratios %fnnTXX5Q5Q&RSSr&cjt|tj|jjyr-)r rr7r)r|r_s r$r|z%VideoCodecContext.sample_aspect_ratioseV^^DHH,H,HIJr&ctjtjjtj|j |j j|j jjz|j j|j jj zdttj|S)Ni) r av_reducerr7numdenr)r>r|r?r)r9dars r$display_aspect_ratioz&VideoCodecContext.display_aspect_ratios NN377 # NN377 # HHNNTXX99== = HHOOdhh::>> >   &fnnS&9::r&c@t|jjS)z :type: bool )boolr) has_b_framesrIs r$rzVideoCodecContext.has_b_framess DHH))**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 sxx$$$r&c&||j_yr-rr_s r$rzVideoCodecContext.reorder_depth %r&c.|jjSz :type: int )r) coded_widthrIs r$rzVideoCodecContext.coded_widths xx###r&c.|jjSr)r) coded_heightrIs r$rzVideoCodecContext.coded_height s xx$$$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'sxx###r&c&||j_yr-rr_s r$rzVideoCodecContext.color_range2s$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_primaries6sxx'''r&c&||j_yr-rr_s r$rz!VideoCodecContext.color_primariesAs#( 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_trcEsxx!!!r&c&||j_yr-rr_s r$rzVideoCodecContext.color_trcPs"r&c.|jjS)z Describes the YUV/RGB transformation matrix of the colorspace. Wraps :ffmpeg:`AVFrame.colorspace`. :type: int r) colorspacerIs r$rzVideoCodecContext.colorspaceTsxx"""r&c&||j_yr-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_framescsxx$$$r&c&||j_yr-rr_s r$rzVideoCodecContext.max_b_frameslrr&c.|jjS)z The minimum quantiser value of an encoded stream. Wraps :ffmpeg:`AVCodecContext.qmin`. :type: int r)qminrIs r$rzVideoCodecContext.qminpxx}}r&c&||j_yr-rr_s r$rzVideoCodecContext.qmin{  r&c.|jjS)z The maximum quantiser value of an encoded stream. Wraps :ffmpeg:`AVCodecContext.qmax`. :type: int r)qmaxrIs r$rzVideoCodecContext.qmaxrr&c&||j_yr-rr_s r$rzVideoCodecContext.qmaxrr&)-__name__ __module__ __qualname__rcfuncrr AVCodecContextconstAVCodecrr.rlistrGrJrrNrWpropertyr=setterr r>uintr?rcintr4rnrrryr|rrrrrrrrrrrrr&r$r(r((s \\( ^^C.. /(~~fll3;;78(4 ((@ \\ 4 \\#5## \\'%'''  \\u$   ]]([((   \\6;;  ]] FKK  ..!!/6::/"/  + + ^^..JJAA [[!!__"fjj"" TTK K ; ;++ %%&6::&&$$ %% $$%%(())""####$$%%&& [[ [[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%finalcclassr(rr&r$rs ##94.*,JVVHA/ ** +nnV\\#*;*;<= .b bbr&