d j-ddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZmZejd efd ZejGd deZdS)N) CodecContextwrap_codec_context)StreamContainer) Dictionary) err_check)Packet)Stream wrap_stream)avdict_to_dict)int64_t)freemallocselfc t|_tj5|jdzrAt tj|j|xjdzc_ddddS#1swxYwYdS)N) rstreamscythonnogil_myflaglibavformat_close_inputaddressptrrs M/home/agentuser/manim-venv/lib/python3.11/site-packages/av/container/input.py close_inputr s"$$DL   > > LLB LL sA A99A=A=c eZdZdZdZedZedZedZedZ dZ dZ d Z d d d d d d de jde jded zde jde jf dZe jdZd S)InputContainerc tj}|js|jrzt |j}tjtjtjtjt|j j tj tj z}t|j j D]}tj||<|t|jkru|jrn|}||j|ttj|||j dttj|||j d||j|tj5t|j |}dddn #1swxYwY|d|||r]t|j j D]4}ttj||5t7|d}t9|_t|j j D]}|j j|} t| jj } | rst!| } t3t"| | j| j#| _$tK| | |j&} | j'rd}nd} |j(tS|| | |j&r|j&j*s|stWdtY|j j-|j.|j/|_-dS)NrFTzAHardware accelerated decode requested but no stream is compatible)0rNULLoptionsstream_optionsrcastpointerr AVDictionaryrr nb_streamssizeofp_voidrangelencopyupdate av_dict_copyr set_timeout open_timeout start_timeoutravformat_find_stream_infor av_dict_freer rravcodec_find_decodercodecparcodec_idavcodec_alloc_context3avcodec_parameters_to_context time_base pkt_timebaserhwaccel is_hwaccel add_streamr allow_software_fallback RuntimeErrorr metadatametadata_encodingmetadata_errors) rargskwargs c_options base_dicti stream_dictret at_least_one_accelerated_contextstreamcodec codec_contextpy_codec_contexts r __cinit__zInputContainer.__cinit__sGMk  < U4. U"4<00I v~c.>?@tx*V]6=-I-IIJJI48.// 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  48.// ? ?   ! !=!=>>>> 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!G==HHc$t|dSNrrs r __dealloc__zInputContainer.__dealloc__dDcz||jjtjkr |jjSdSrQ) _assert_openr start_timerAV_NOPTS_VALUErs rrXzInputContainer.start_timegs:  8 #"4 4 48& & 5 4rUcz||jjtjkr |jjSdSrQ)rWrdurationrrYrs rr[zInputContainer.durationms:  8  2 2 28$ $ 3 2rUcB||jjSrQ)rWrbit_raters rr]zInputContainer.bit_ratess x  rUcr|t|jjSrQ)rWr avio_sizerpbrs rsizezInputContainer.sizexs* }}TX[)))rUc$t|dSrQrRrs rclosezInputContainer.close}rTrUc /0K||jj|i|}tjtjtjt|jj tj t z}|tj krt| |j t|jj D]}d||<|D]1}|j}||jj krt#d|dd||<2tj5t&}dddn #1swxYwY|tj krtd tj5t&|dddn #1swxYwY |tj5t&|j|}dddn #1swxYwY||n#t2$rYnwxYw||jr|jt7|jkrt9} tj5t&| j|dddn #1swxYwY|j| jj| _| jjj| j_| Vvt|jj D]I}||r?t9} |j|| _| jjj| j_| VJ | dtA||tj kr.t&!tj"|dSdS#| dtA||tj kr-t&!tj"|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. Fz stream index z out of rangeTNzCould not allocate packet)#rWrgetrr$r%bintrrr'r(r! MemoryErrorr/ read_timeoutr*index ValueErrorrrav_packet_allocav_packet_unrefr1 av_read_framerEOFError stream_indexr+rav_packet_move_ref_streamr9r av_packet_freer) rrCrDrinclude_streamrGrK read_packetrIpackets rdemuxzInputContainer.demuxs  0 0$ A& A A6>> %\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!N2#E = N2 E  N2E /N2F' N2'F++N2.F+/N23 H!!H4 H!HH!HH! N2! H.+N2-H..AN25!J" N2"J&&N2)J&*B#N22A#Pc/K||j|i|D]}|D]}|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)rWrvdecode)rrCrDruframes rrxzInputContainer.decodesk  dj$1&11  F      rUTFN)backward any_framerKunsupported_frame_offsetunsupported_byte_offsetrzr{rKr|r}c |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)rW isinstanceint TypeErrortyperAVSEEK_FLAG_BACKWARDAVSEEK_FLAG_ANYAVSEEK_FLAG_FRAMEAVSEEK_FLAG_BYTErirr av_seek_framerr flush_buffers) roffsetrzr{rKr|r}c_offsetflagsrorIs rseekzInputContainer.seeks`J &#&& YI4PV<<XX X"  . S- -E  ) S( (E $ + S* *E " * S) )E39#A6<>>+ > ; >  >#)+>"(>>>>@ \ . .\ . . .rUr)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 cython.cimports.libc.stdlibr rrrrcclass ContainerrrUrrs7 MMMMMMMM@@@@@@444444......,,,,,,99999999333333//////44444444n\.\.\.\.\.Y\.\.\.\.\.rU