o 2j/@sddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZmZdd lmZmZejd efd d ZejejGdddeZdS)N) CodecContextwrap_codec_context)StreamContainer) Dictionary) err_check)Packet)Stream wrap_stream)avdict_to_dict)int64_tuint8_t)freemallocselfcCsnt|_tj&|jd@r%tt|j|jdM_WddSWddS1s0wYdS)N) rstreamscythonnogil_myflaglibavformat_close_inputaddressptrrrM/var/www/html/whisper/venv/lib/python3.10/site-packages/av/container/input.py close_input s "rc @seZdZddZddZeddZeddZed d Zed d Z d dZ ddZ ddZ ddddddde jde jdedBde jde jf ddZe jddZdS) InputContainercOs<tj}|jj}|jr|dkrtd|js|jrs|dkrst|j}ttj tj t j t |t tj}t|D]8}tj||<|t|jkre|jre|}||j|t t|||jdq:t t|||jdq:||j|tjt |j|}Wdn1swY|d|||rt|D] }t t||qt|d} t|_t|jjD]<}|jj|} t | j j!} | rt "| } tt #| | j | j$| _%t&| | |j'} | j(rd} nd} |j)t*|| | q|j'r|j'j+s| st,dt-|jj.|j/|j0|_.dS)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)1rNULLr nb_streamsstream_options ValueErroroptionsrcastpointerr AVDictionaryrsizeofp_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_optionsnb_streams_before base_dicti stream_dictret at_least_one_accelerated_contextstreamcodec codec_contextpy_codec_contextrrr __cinit__sz            zInputContainer.__cinit__cC t|dSNrrrrr __dealloc__l zInputContainer.__dealloc__cC"||jjtjkr|jjSdSrQ) _assert_openr start_timerAV_NOPTS_VALUErrrrrWozInputContainer.start_timecCrUrQ)rVrdurationrrXrrrrrZurYzInputContainer.durationcCs||jjSrQ)rVrbit_raterrrrr[{szInputContainer.bit_ratecCs|t|jjSrQ)rVr avio_sizerpbrrrrsizeszInputContainer.sizecCrPrQrRrrrrcloserTzInputContainer.closec os||jj|i|}|jjdkrdSttjtt |jjt t}|tj kr0t | |jz t|jjD]}d||<q>|D]}|j}||jjkrZtd|dd||<qGtj t}Wdn1sqwY|tj krt d tj t|Wdn1swYz%|tjt|j|}Wdn1swY||Wn tyYn@w||jr|jt|jkrt} tjt| j|Wdn1swY|j| jj| _| jjj| j_| Vqt|jjD]}||r&t} |j|| _| jjj| j_| Vq W| dt ||tj krBt!t"|dSdS| dt ||tj kr\t!t"|ww)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)#rVrgetrr rr$r%r rr'r MemoryErrorr. read_timeoutr)indexr"rrav_packet_allocav_packet_unrefr0 av_read_framerEOFError stream_indexr*rav_packet_move_ref_streamr8r av_packet_freer) rrBrCrinclude_streamrGrK read_packetrIpacketrrrdemuxs                 zInputContainer.demuxcos8||j|i|D] }|D]}|Vqq dS)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)rVrpdecode)rrBrCroframerrrrqs  zInputContainer.decodeTFN)backward any_framerKunsupported_frame_offsetunsupported_byte_offsetrsrtrKrurvc Cs|t|tstdt||}d}|r|tjO}|r"|tjO}|r)|tjO}|r0|tj O}|r5|j nd} t j t |j| ||} Wdn1sNwYt| |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)rV isinstanceint TypeErrortyperAVSEEK_FLAG_BACKWARDAVSEEK_FLAG_ANYAVSEEK_FLAG_FRAMEAVSEEK_FLAG_BYTErdrr av_seek_framerr flush_buffers) roffsetrsrtrKrurvc_offsetflagsrirIrrrseeks&%      zInputContainer.seekcCs*||jD] }|j}|r|qdSrQ)rVrrMr)rrKrMrrrr4s zInputContainer.flush_buffers)__name__ __module__ __qualname__rOrSpropertyrWrZr[r^r_rprqrbintrrcfuncrrrrrrs@R    [ @r)r cython.cimports.av.codec.contextrr$cython.cimports.av.container.streamsrcython.cimports.av.dictionaryrcython.cimports.av.errorrcython.cimports.av.packetrcython.cimports.av.streamrr cython.cimports.av.utilsr cython.cimports.libc.stdintr r cython.cimports.libc.stdlibr rrrrfinalcclass Containerrrrrs