j1ddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZmZdd lmZmZejd efd ZejejGd deZdS)N) CodecContextwrap_codec_context)StreamContainer) Dictionary) err_check)Packet)Stream wrap_stream)avdict_to_dict)int64_tuint8_t)freemallocselfc t|_tj5|jdzrAt tj|j|xjdzc_ddddS#1swxYwYdS)N) rstreamscythonnogil_myflaglibavformat_close_inputaddressptrrs w/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/av/container/input.py close_inputr s"$$DL   > > LLB LL sA A99A=A=c eZdZdZdZedZedZedZedZ edZ dZ d Z d Z d d d d d ddejdejded zdejdejf dZejdZd S)InputContainerc tj}|jj}|jr|dkrt d|js|jrm|dkrft|j}tjtj tj tj t|tj tjz}t|D]}tj||<|t!|jkru|jrn|}||j|ttj|||jdttj|||jd||j|tj5t|j|}dddn #1swxYwY|d|||rSt|D]4}ttj||5t9|d} t;|_t|jjD]}|jj|} t| j j!} | rst"| } t5t#| | j | j$| _%tM| | |j'} | j(rd} nd} |j)tU|| | |j'r|j'j+s| stYdt[|jj.|j/|j0|_1dS)Nrzstream_options were provided, but this format does not expose its streams before avformat_find_stream_info (e.g. MPEG). Per-stream options cannot be applied.FTzAHardware accelerated decode requested but no stream is compatible)2rNULLr nb_streamsstream_options ValueErroroptionsrcastpointerr AVDictionaryrsizeofp_voidrangelencopyupdate av_dict_copyr set_timeout open_timeout start_timeoutravformat_find_stream_infor av_dict_freerrravcodec_find_decodercodecparcodec_idavcodec_alloc_context3avcodec_parameters_to_context time_base pkt_timebaserhwaccel is_hwaccel add_streamr allow_software_fallback RuntimeErrorr metadatametadata_encodingmetadata_errors _metadata)rargskwargs c_optionsnb_streams_before base_dicti stream_dictret at_least_one_accelerated_contextstreamcodec codec_contextpy_codec_contexts r __cinit__zInputContainer.__cinit__s GMk *.)<   #4#9#98  L UD/ U5F5J5J"4<00I v~c.>?@(6=+G+GGHHI,-- U U%{ ! s4.////D4G/"+.."2"2K&&t':1'=>>>$$V^IaL%A%A;?TUVVVV$$V^IaL%A%A9=RSTTTT *+++  \ E E//)DDC E E E E E E E E E E E E E E E  s  ,-- ? ?   ! !=!=>>>> OOO+0(&(( tx*++ Q QAX%a(F,,V_-EFFE ( # : :5 A A 55mV_UU.4-= *#5!5$,$$ $.<7;4$( L # #Kf>N$O$O P P P P L L8 5  S ( H t5t7K  s+!HHHc$t|dSNrrs r __dealloc__zInputContainer.__dealloc__lDcz||jjtjkr |jjSdSrU) _assert_openr start_timerAV_NOPTS_VALUErs rr\zInputContainer.start_timeos:  8 #"4 4 48& & 5 4rYcz||jjtjkr |jjSdS)a`Start time of the stream in real world time, in microseconds since the Unix epoch, or ``None`` if unknown. Only a few demuxers know this; RTSP computes it from RTCP sender reports, which is what makes it useful for aligning analytics with a camera's clock. Wraps :ffmpeg:`AVFormatContext.start_time_realtime`. N)r[rstart_time_realtimerr]rs rr_z"InputContainer.start_time_realtimeus<  8 '3+= = =8/ / > =rYcz||jjtjkr |jjSdSrU)r[rdurationrr]rs rrazInputContainer.durations:  8  2 2 28$ $ 3 2rYcB||jjSrU)r[rbit_raters rrczInputContainer.bit_rates x  rYcr|t|jjSrU)r[r avio_sizerpbrs rsizezInputContainer.sizes* }}TX[)))rYc$t|dSrUrVrs rclosezInputContainer.closerXrYc /JK||jj|i|}|jjdkrdSt jt jtt|jjt j tz}|t j krt| |j t|jjD]}d||<|D]1}|j}||jjkrt#d|dd||<2t j5t&}dddn #1swxYwY|t j krtd t j5t&|dddn #1swxYwY |t j5t&|j|}dddn #1swxYwY||n#t2$rYnwxYw||jr|jt7|jkrt9} t j5t&| j|dddn #1swxYwY|j| jj| _| jjj| j_| Vvt|jjD]I}||r?t9} |j|| _| jjj| j_| VJ | dtA||t j kr.t&!t j"|dSdS#| dtA||t j kr-t&!t j"|wwxYw)a0demux(streams=None, video=None, audio=None, subtitles=None, data=None) Yields a series of :class:`.Packet` from the given set of :class:`.Stream`:: for packet in container.demux(): # Do something with `packet`, often: for frame in packet.decode(): # Do something with `frame`. .. seealso:: :meth:`.StreamContainer.get` for the interpretation of the arguments. .. note:: The last packets are dummy packets that when decoded will flush the buffers. rNz stream index z out of rangezCould not allocate packet)#r[rgetrr#rr'r(r rr*r" MemoryErrorr1 read_timeoutr,indexr%rrav_packet_allocav_packet_unrefr3 av_read_framerEOFError stream_indexr-rav_packet_move_ref_streamr;rav_packet_freer) rrFrGrinclude_streamrKrO read_packetrMpackets rdemuxzInputContainer.demuxs   0 0$ A& A A 8 ! # # F28+ N7 # 48&w)?)?? @ @3 3  V[ ( (--  *+++7 @48.// & &$%q!!! & &L+++$%EQ%E%E%EFFF$%q!! 4 4!1133  4 4 4 4 4 4 4 4 4 4 4 4 4 4 4fk))!"=>>> %\55'' 444555555555555555&&(((GG!//+FFGGGGGGGGGGGGGGGNN3''''E"+":;% #/#dl2C2CCC"(#\LL226:{KKKLLLLLLLLLLLLLLL)-fj6M)N/5~/A/K ,$ 5 %:48.// ! !!!$!#XXF%)\!_FN+1>+=+GFJ( LLL  !   T " " "  fk))""6>+#>#>?????*)   T " " "  fk))""6>+#>#>????*sA!N?0E N?EN?E/N? F4( N?4F88N?;F8<N? H. !H  H. HH.HH.-N?. H;8N?:H;;AN?!J/# N?/J33N?6J37B#N??A#P"c/K||j|i|D]}|Ed{VdS)a`decode(streams=None, video=None, audio=None, subtitles=None, data=None) Yields a series of :class:`.Frame` from the given set of streams:: for frame in container.decode(): # Do something with `frame`. .. seealso:: :meth:`.StreamContainer.get` for the interpretation of the arguments. N)r[r{decode)rrFrGrzs rr}zInputContainer.decodesg  dj$1&11 ' 'F}} & & & & & & & & ' 'rYTFN)backward any_framerOunsupported_frame_offsetunsupported_byte_offsetr~rrOrrc |t|tstdt ||}d}|r|t jz}|r|t jz}|r|t jz}|r|t j z}|r|j nd} tj 5t |j| ||} dddn #1swxYwYt| |dS)a)seek(offset, *, backward=True, any_frame=False, stream=None) Seek to a (key)frame nearest to the given timestamp. :param int offset: Time to seek to, expressed in``stream.time_base`` if ``stream`` is given, otherwise in :data:`av.time_base`. :param bool backward: If there is not a (key)frame at the given offset, look backwards for it. :param bool any_frame: Seek to any frame, not just a keyframe. :param Stream stream: The stream who's ``time_base`` the ``offset`` is in. :param bool unsupported_frame_offset: ``offset`` is a frame index instead of a time; not supported by any known format. :param bool unsupported_byte_offset: ``offset`` is a byte location in the file; not supported by any known format. After seeking, packets that you demux should correspond (roughly) to the position you requested. In most cases, the defaults of ``backwards = True`` and ``any_frame = False`` are the best course of action, followed by you demuxing/decoding to the position that you want. This is because to properly decode video frames you need to start from the previous keyframe. .. seealso:: :ffmpeg:`avformat_seek_file` for discussion of the flags. z+Container.seek only accepts integer offset.rN)r[ isinstanceint TypeErrortyperAVSEEK_FLAG_BACKWARDAVSEEK_FLAG_ANYAVSEEK_FLAG_FRAMEAVSEEK_FLAG_BYTErorr av_seek_framerr flush_buffers) roffsetr~rrOrrc_offsetflagsrtrMs rseekzInputContainer.seeks`J &#&& YI4PV<<XX X"  . S- -E  ) S( (E $ + S* *E " * S) )E39#A6<>>+ > ; >  >#)+>"(>>>>@ \ . .\ . . .rYr )r cython.cimports.av.codec.contextrr$cython.cimports.av.container.streamsrcython.cimports.av.dictionaryrcython.cimports.av.errorrcython.cimports.av.packetrcython.cimports.av.streamr r cython.cimports.av.utilsr cython.cimports.libc.stdintr r cython.cimports.libc.stdlibrrrr rfinalcclass ContainerrrYrrsI MMMMMMMM@@@@@@444444......,,,,,,999999993333338888888844444444nr.r.r.r.r.Yr.r.r.r.r.rY