:j` ddlmZmZddlZddlmZddlmZm Z ddl m Z m Z ddl mZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZej8eeZej>d ej@ejBdej@ejDejFde$de%fdZ&GddeZ'GddeZ(GddeZ)ejTGddZ%y))FlagIntEnumN)libav) ByteSource bytesource)Codec wrap_codec) Dictionary) err_check)Packet)avrational_to_fraction to_avrational)EAGAIN)uint8_t)memcpyc_ctxc_codechwaccelreturncj|jtjk(rddlm}|t }no|jtj k(rddlm}|t }n?|jtjk(rddl m }|t }ntt }|j||||S)z8Build an bv.CodecContext for an existing AVCodecContext.r)VideoCodecContext)AudioCodecContext)SubtitleCodecContext) codec_typelibAVMEDIA_TYPE_VIDEOav.video.codeccontextr_cinit_sentinelAVMEDIA_TYPE_AUDIOav.audio.codeccontextrAVMEDIA_TYPE_SUBTITLEav.subtitles.codeccontextr CodecContext_init)rrrrpy_ctxrrs e/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/av/codec/context.pywrap_codec_contextr's 3111;"?3   S33 3;"?3   S66 6B%o6o. LL) MceZdZUdZej Zded<ejZ ded<ejej zZ ded<y) ThreadTyperz"Decode more than one frame at onceFRAMEz3Decode more than one part of a single frame at onceSLICEz*Decode using both FRAME and SLICE methods.AUTON) __name__ __module__ __qualname__NONErFF_THREAD_FRAMEr+__annotations__FF_THREAD_SLICEr,r-r(r&r*r*/sJ D252E2EE /ECFCVCVE @V c111 6r(r*ceZdZejZej ZejZ ejZ ejZ ejZej Zej$Zej(Zej,Zej0Zej4Zej8Zej<Zej@Z!ejDZ#ejHZ%ejLZ'ejPZ)ejTZ+y)FlagsN),r.r/r0rAV_CODEC_FLAG_UNALIGNED unalignedAV_CODEC_FLAG_QSCALEqscaleAV_CODEC_FLAG_4MVfour_mvAV_CODEC_FLAG_OUTPUT_CORRUPToutput_corruptAV_CODEC_FLAG_QPELqpelAV_CODEC_FLAG_RECON_FRAME recon_frameAV_CODEC_FLAG_COPY_OPAQUE copy_opaqueAV_CODEC_FLAG_FRAME_DURATIONframe_durationAV_CODEC_FLAG_PASS1pass1AV_CODEC_FLAG_PASS2pass2AV_CODEC_FLAG_LOOP_FILTER loop_filterAV_CODEC_FLAG_GRAYgrayAV_CODEC_FLAG_PSNRpsnrAV_CODEC_FLAG_INTERLACED_DCTinterlaced_dctAV_CODEC_FLAG_LOW_DELAY low_delayAV_CODEC_FLAG_GLOBAL_HEADER global_headerAV_CODEC_FLAG_BITEXACTbitexactAV_CODEC_FLAG_AC_PREDac_predAV_CODEC_FLAG_INTERLACED_ME interlaced_meAV_CODEC_FLAG_CLOSED_GOP closed_gopr5r(r&r7r78s++I  % %F##G55N  ! !D//K//K55N  # #E  # #E//K  ! !D  ! !D55N++I33M))H''G33M--Jr(r7ceZdZejZej ZejZ ejZ ejZ ejZej Zej$Zej(Zy)Flags2N)r.r/r0rAV_CODEC_FLAG2_FASTfastAV_CODEC_FLAG2_NO_OUTPUT no_outputAV_CODEC_FLAG2_LOCAL_HEADER local_headerAV_CODEC_FLAG2_CHUNKSchunksAV_CODEC_FLAG2_IGNORE_CROP ignore_cropAV_CODEC_FLAG2_SHOW_ALLshow_allAV_CODEC_FLAG2_EXPORT_MVS export_mvsAV_CODEC_FLAG2_SKIP_MANUAL skip_manualAV_CODEC_FLAG2_RO_FLUSH_NOOP ro_flush_noopr5r(r&raraOse  " "D,,I22L  & &F00K**H..J00K44Mr(racVeZdZedDdZdEdZejdeje jdejeje jde fdZedZej"d ej$fd Zed Zej"d Zed Zej"dZedZej"d ej$fdZedZej"dZedZedZedZej4dFdej6fdZdZdZdEdZedZ de!dzfdZ"ejde!dzde#fd Z$ejde!fd!Z%ejd"Z&ejde!fd#Z'ejd$Z(ejde!fd%Z)ej4dEde!dzfd&Z*dEde!dzfd'Z+ejd(e,fd)Z-ej4dEd(e,dzfd*Z.ej4d+Z/ejde!d(e,dzfd,Z0ed-Z1ed.Z2ed/Z3ed0Z4e4j"d1Z4ed2Z5e5j"d3Z5ed4Z6e6j"d5Z6eejndd6Z8e8j"d ej$fd7Z8ed8Z9e9j"d ej$fd9Z9ed:Z:ed;Z;e;j"d ej$fd<Z;ed=ZZe>j"dBZ>edCZ?y)Gr#Nct|tr|n t||}tj|j}t ||j|SN) isinstancerravcodec_alloc_context3ptrr')codecmodercy_codecrs r&createzCodecContext.create]sE#-eU#;%udAS474N4N LL5 "%w??r(cT|tur tdi|_d|_d|_y)NzCannot instantiate CodecContextF)r RuntimeErroroptions stream_indexis_open)selfsentinelargskwargss r& __cinit__zCodecContext.__cinit__es, ? *@A A  r(ryrzrcN||_|jjr&|r$|jj|k7r tdt|tj k7r|n|jj|_||_d|j_d|j_y)Nz,Wrapping CodecContext with mismatched codec.r) ryrzrr cythonNULLr thread_count thread_type)rryrzrs r&r$zCodecContext._initmsu 88>>e%(?MN N&++)=488>>R  !"#r(c.|jjS)zU Get and set the flags bitmask of CodecContext. :rtype: int ryflagsrs r&rzCodecContext.flags~xx~~r(valuec&||j_yrvrrrs r&rzCodecContext.flagss r(cbt|jjtjzS)z9 Use fixed qscale. :rtype: bool )boolryrrr:rs r&r;zCodecContext.qscales"DHHNNS%=%==>>r(c|r.|jxjtjzc_y|jxjtjzc_yrv)ryrrr:rs r&r;zCodecContext.qscales8  HHNNc66 6N HHNNs777 7Nr(cbt|jjtjzSrv)rryrrrDrs r&rEzCodecContext.copy_opaques DHHNNS%B%BBCCr(c|r.|jxjtjzc_y|jxjtjzc_yrv)ryrrrDrs r&rEzCodecContext.copy_opaques8  HHNNc;; ;N HHNNs<<< > 488%%vzz6== A&,mmDHH #r(c.|jjSrv)ryrrs r&rzCodecContext.extradata_sizesxx&&&r(c|jtjurytj|jj SNF)ryrrrav_codec_is_encoderrzrs r& is_encoderzCodecContext.is_encoder. 88v{{ "&&txx~~66r(c|jtjurytj|jj Sr)ryrrrav_codec_is_decoderrzrs r& is_decoderzCodecContext.is_decoderrr(Tstrictc |jr|r tdyt}|j|jxsi|j j jsC|jr6|jdk(r|j jjxsd|j j _|j jjxstj|j j _ n|jdk(rKd|j j _|j j|j j _ nDd|j j _tj|j j _ ttj |j |j"j t%j&|j d|j"j(d|jdd|_t+||_y) NzCodecContext is already open.videoaudiozavcodec_open2("z", )T)r ValueErrorr updaterry time_basenumrtype frameratedenr AV_TIME_BASE sample_rater avcodec_open2rzrrnamedict)rrrs r&openzCodecContext.opensz << !@AA (lt||)r*xx!!%%$//yyG#)-););)?)?)D1""&)-););)?)?)S3CSCS""&g%)*""&)-)=)=""&)*""&),)9)9""&   dhh w{{8S Tdjjoo.c$,,q A  G} r(cP|jrntjtj|jj tj tj|j|jr tj|jyyrv) ryrrrrravcodec_free_contextparserav_parser_closers r& __dealloc__zCodecContext.__dealloc__sa 88 LL(:(:; <  $ $V^^DHH%= > ;;    , r(c |jxsd}|jxsd}d|jjd|d|dt |dd S) Nzz z)rr __class__r.id)r_typers r&__repr__zCodecContext.__repr__sP 'Zyy'Kdnn--.awavVBtHQ->?DK;; >$**//1B!CDD' dC060B&** /5/Afmmq  //KKHHNN8,NN8,&&&&   h  )vzz(;"& !% !% ;;''*.++*>*>FJJ';;((A-JJ$$(;(;;$v& x G x Gk  s A4JJc|jduS)zf Returns ``True`` if this codec context is hardware accelerated, ``False`` otherwise. N) hwaccel_ctxrs r& is_hwaccelzCodecContext.is_hwaccelfs t++r(framec#:Ktj5tj|j| |jntj }dddt d|j}|r||j}|ryy#1swY@xYww)Nzavcodec_send_frame())rrravcodec_send_frameryrr _recv_packetrrresrs r&_send_frame_and_recvz!CodecContext._send_frame_and_recvms\\ ((u'8%))fkkC  #-.""$L&&(F  s"B=BSCOO3 #01&&u-#D L A As +B88Cc|Srvr5rs r&rzCodecContext._transfer_hwframes r(ct}tj5tj|j |j }dddt k(s|tjk(ryt|d|s|Sy#1swY8xYw)Nzavcodec_receive_packet()) r rrravcodec_receive_packetryrrr )rrrs r&rzCodecContext._recv_packetsv\\ C,,TXXvzzBC C 6'>SCOO3 #12M C Cs +A==Bc,|jjtjtjfvr t d|j d|j|}|D]*}||j|jj,|S)Nz/Encoding is only supported for audio and video.Fr) ryrrrrrrr _rebase_timer)rrframess r&*_prepare_and_time_rebase_frames_for_encodez7CodecContext._prepare_and_time_rebase_frames_for_encodes 88  s'='=s?U?U&V V%&WX X  007 7E ""488#5#56 7 r(cg}|j|D]:}|j|D]$}|j||j|&<|S)zAEncode a list of :class:`.Packet` from the given :class:`.Frame`.)r r_setup_encoded_packetrrs r&encodezCodecContext.encodes^DDUK #E33E: #**62 6" # # r(c#K|j|D]-}|j|D]}|j||/ywrv)r rrrrrs r& encode_lazyzCodecContext.encode_lazysMDDUK E33E: **62   sAArcN|jj|j_yrv)ryr)rrs r&rz"CodecContext._setup_encoded_packets $xx11 r(c|jjs td|jdtj 5t j|j| |jntj}dddtdg}|j}|r6|j|||j||j}|r6|S#1swYaxYw)aDecode a list of :class:`.Frame` from the given :class:`.Packet`. If the packet is None, the buffers will be flushed. This is useful if you do not want the library to automatically re-order frames for you (if they are encoded with a codec that has B-frames). .. warning:: This method is **not thread-safe**. Calling :meth:`decode` concurrently from multiple threads on the same :class:`CodecContext` will corrupt internal FFmpeg state and likely cause a crash (segfault). FFmpeg 8.1 enforces this more strictly than earlier releases. If you need to decode from multiple threads, give each thread its own :class:`CodecContext`. zcannot decode unknown codecFr Nzavcodec_send_packet()) rzryrrrrravcodec_send_packetrr r_setup_decoded_framer)rrroutrs r&decodezCodecContext.decodes"zz~~:; ;  \\ ))(:&** C  #./  "  % %eV 4 JJu $$&E   s =C!!C*c|jr9tj5tj|j dddyy#1swYyxYw)zReset the internal codec state and discard all internal buffers. Should be called before you start decoding from a new position e.g. when seeking or when switching to a different stream. N)rrrravcodec_flush_buffersryrs r& flush_bufferszCodecContext.flush_bufferssC << 4))$((3 4 4  4 4s AAc@||jj|_yyrv)ryr _time_basers r&rz!CodecContext._setup_decoded_frames!  %zz33E  r(c.|jjSrv)rzrrs r&rzCodecContext.namezzr(c.|jjSrv)rzrrs r&rzCodecContext.typer r(cg}|jjr6|jjr |jjjs|S|jj}d}|j|jt j k7rX|j|j|j|dz }|j|jt j k7rX|S)zX List the available profiles for this stream. :type: list[str] rr) ryrzdescprofilesprofilerAV_PROFILE_UNKNOWNrr)rretr#is r&r$zCodecContext.profilessxx~~TZZ__DJJOO,>?@r(cr|jjjdddjdS)NlittleF byteordersignedasciiencoding)ry codec_tagto_bytesrrs r&r8zCodecContext.codec_tagXs:xx!!**1*OVVW  r(ct|trFt|dk(r8tj |j ddd|j _ytd)Nr0r5r6r1Fr2z)Codec tag should be a 4 character string.) rwstrlenint from_bytesrryr8rrs r&r8zCodecContext.codec_tag^sP eS !c%jAo!$ g .(5"0"DHH HI Ir(cP|jjtjzS)zGlobal quality for codecs which cannot change it per frame. Stored internally in lambda units; this property converts to/from QP units using ``FF_QP2LAMBDA``. Wraps :ffmpeg:`AVCodecContext.global_quality`. )ryglobal_qualityr FF_QP2LAMBDArs r&r@zCodecContext.global_qualitygsxx&&#*:*:::r(cH|tjz|j_yrv)rrAryr@rs r&r@zCodecContext.global_qualityts"'#*:*:":r(cd|jjdkDr|jjSdSrrybit_raters r&rEzCodecContext.bit_ratexs($(HH$5$5$9txx  CtCr(c&||j_yrvrDrs r&rEzCodecContext.bit_rate|s!r(cb|jjdkDr|jjSyr)ry rc_max_raters r& max_bit_ratezCodecContext.max_bit_rates' 88  ! #88'' 'r(c0|jjyrvrybit_rate_tolerancers r&rLzCodecContext.bit_rate_tolerances ##r(c&||j_yrvrKrs r&rLzCodecContext.bit_rate_tolerances&+#r(c.|jjS)zfHow many threads to use; 0 means auto. Wraps :ffmpeg:`AVCodecContext.thread_count`. )ryrrs r&rzCodecContext.thread_countsxx$$$r(cT|jr td||j_y)Nz/Cannot change thread_count after codec is open.)rrryrrs r&rzCodecContext.thread_counts! <<PQ Q %r(c@t|jjS)z[One of :class:`.ThreadType`. Wraps :ffmpeg:`AVCodecContext.thread_type`. )r*ryrrs r&rzCodecContext.thread_types$((..//r(c|jr tdt|tur||j_yt|t ur#t|j|j_y|j|j_y)Nz.Cannot change thread_type after codec is open.) rrrr=ryrr;r*rrs r&rzCodecContext.thread_types` <<OP P ;# #(DHH %[C #-e#4#:#:DHH #(;;DHH r(cL|jj}|tjk(ry|tjk(ry|tj k(ry|tj k(ry|tjk(ry|tjk(ry|tjk(ry|S)aReturns one of the following str literals: "NONE" Discard nothing "DEFAULT" Discard useless packets like 0 size packets in AVI "NONREF" Discard all non reference "BIDIR" Discard all bidirectional frames "NONINTRA" Discard all non intra frames "NONKEY Discard all frames except keyframes "ALL" Discard all Wraps :ffmpeg:`AVCodecContext.skip_frame`. r1DEFAULTNONREFBIDIRNONINTRANONKEYALL) ry skip_framerAVDISCARD_NONEAVDISCARD_DEFAULTAVDISCARD_NONREFAVDISCARD_BIDIRAVDISCARD_NONINTRAAVDISCARD_NONKEY AVDISCARD_ALLrs r&rYzCodecContext.skip_frames## C&& & C)) ) C(( ( C'' ' C** * C(( ( C%% %r(c|dk(r tj|j_y|dk(r tj|j_y|dk(r tj |j_y|dk(r tj |j_y|dk(r tj|j_y|dk(r tj|j_y|dk(r tj|j_ytd) Nr1rSrTrUrVrWrXzInvalid skip_frame type) rrZryrYr[r\r]r^r_r`rrs r&rYzCodecContext.skip_frames F?"%"4"4DHH  i "%"7"7DHH  h "%"6"6DHH  g "%"5"5DHH  j "%"8"8DHH  h "%"6"6DHH  e^"%"3"3DHH 67 7r(c.|jjS)zECodec delay. Wraps :ffmpeg:`AVCodecContext.delay`. )rydelayrs r&rczCodecContext.delayrr()NNrv)T)@r.r/r0 staticmethodr}rrcfuncrrAVCodecContextconstAVCodecHWAccelr$propertyrsetterr=r;rErrrrrccallbintrrrrrFramerlistrrrrrr rrr rrrrrrr$r%rr8 cdivisionr@rErIrLrrrYrcr5r(r&r#r#[s@@ \\$ ^^C.. /$~~fll3;;78$ $$  \\6::?? ]]88 DD==  ]] FJJ    44"''77 77  \\%6;;%%6-R Zx,, )%$, ) \\  \\F5FF \\( \\u \\   \\  \\EDL  \\2F22 \\"Vd]""H \\ 4 4 \\4%4$44(   ^^  JJ AA   JJVd ; ;;FJJ;;DD__"fjj"" $$, ,,%%&&**&& 00//<88$r(r#)+enumrrrcython.cimportsrrcython.cimports.av.bufferrrcython.cimports.av.codec.codecrr cython.cimports.av.dictionaryr cython.cimports.av.errorr cython.cimports.av.packetr cython.cimports.av.utilsr rcython.cimports.libc.errnorcython.cimports.libc.stdintrcython.cimports.libc.stringrdeclareobjectrrerrfrgrhrir#r'r*r7racclassr5r(r&rs (<<4.,J-/. &..2 >>#,, - ^^FLL5 6 8.G.. 5W 5O O O r(