o 2j @sddlmZddlZddlmmZddlmZej de dej fddZ ej ej dd d ejejd ej d edBdejfd dZejejGdddZdS))IteratorN)StreamtypereturncCsT|dkrtjS|dkrtjS|dkrtjS|dkrtjS|dkr#tjStd|)Nvideoaudiosubtitle attachmentdatazInvalid stream type: )libAVMEDIA_TYPE_VIDEOAVMEDIA_TYPE_AUDIOAVMEDIA_TYPE_SUBTITLEAVMEDIA_TYPE_ATTACHMENTAVMEDIA_TYPE_DATA ValueErrorrrO/var/www/html/whisper/venv/lib/python3.10/site-packages/av/container/streams.py_get_media_type_enumsrF)checkfmtctxenumtyperelatedcCs@|durt||ddtjd}|St||d|jjtjd}|S)Nr)r av_find_best_streamcythonNULLptrindex)rrr stream_indexrrr_get_best_stream_indexs r!c@seZdZdZddZejdedejfddZ dd Z d d Z d d Z e ddZe ddZe ddZe ddZe ddZdeefddZddZd!dedBdefdd ZdS)"StreamContainerz A tuple-like container of :class:`Stream`. :: # There are a few ways to pulling out streams. first = container.streams[0] video = container.streams.video[0] audio = container.streams.get(audio=(0, 1)) cCs g|_dSN)_streamsselfrrr __cinit__< zStreamContainer.__cinit__streamrcCs&|jjt|jks J|j|dSr#)rrlenr$append)r&r)rrr add_stream?szStreamContainer.add_streamcC t|jSr#)r*r$r%rrr__len__Er(zStreamContainer.__len__cCr-r#)iterr$r%rrr__iter__Hr(zStreamContainer.__iter__cCs"t|tr ||dS||S)Nr) isinstanceintget)r&rrrr __getitem__Ks  zStreamContainer.__getitem__cCtdd|jDS)Ncs|] }|jdkr|VqdS)rNr.0srrr Sz(StreamContainer.video..tupler$r%rrrrQzStreamContainer.videocCr5)Ncsr6)rNrr7rrrr:Wr;z(StreamContainer.audio..r<r%rrrrUr>zStreamContainer.audiocCr5)Ncsr6)rNrr7rrrr:[r;z,StreamContainer.subtitles..r<r%rrr subtitlesYr>zStreamContainer.subtitlescCr5)Ncsr6)r Nrr7rrrr:_r;z'StreamContainer.data..r<r%rrrr ]r>zStreamContainer.datacCr5)Ncsr6)r Nrr7rrrr:cr;z.StreamContainer.attachments..r<r%rrr attachmentsar>zStreamContainer.attachmentsccs|durdSt|tr|VdSt|tr|j|VdSt|ttfr4|D] }||EdHq'dSt|trd|D]$\}}|dkrH|jnt ||}t|ttfsW|g}|D]}||VqYq=dSt dt |)NstreamszArgument must be Stream or int.) r1rr2r$r=list _get_streamsdictitemsgetattr TypeErrorr)r&xitemtype_indicesrAirrrrCes*     zStreamContainer._get_streamscOs4t||}|r||||p|jddS)a+get(streams=None, video=None, audio=None, subtitles=None, data=None) Get a selection of :class:`.Stream` as a ``list``. Positional arguments may be ``int`` (which is an index into the streams), or ``list`` or ``tuple`` of those:: # Get the first channel. streams.get(0) # Get the first two audio channels. streams.get(audio=(0, 1)) Keyword arguments (or dicts as positional arguments) as interpreted as ``(stream_type, index_value_or_set)`` pairs:: # Get the first video channel. streams.get(video=0) # or streams.get({'video': 0}) :class:`.Stream` objects are passed through untouched. If nothing is selected, then all streams are returned. N)rBrCextendr$)r&argskwargs selectionrrrr3zszStreamContainer.getNrrcCsRt|jdkr dStt|jd}|jj}t|t||}|dkr$dS|j|S)abest(type: Literal["video", "audio", "subtitle", "attachment", "data"], /, related: Stream | None) Finds the "best" stream in the file. Wraps :ffmpeg:`av_find_best_stream`. Example:: stream = container.streams.best("video") :param type: The type of stream to find :param related: A related stream to use as a reference (optional) :return: The best stream of the specified type :rtype: Stream | None rN) r*r$rcastr containerrr!r)r&rr first_streamrRr rrrbests   zStreamContainer.bestr#)__name__ __module__ __qualname____doc__r'rcfuncrvoidr,r.r0r4propertyrrr?r r@rrCr3strrTrrrrr"+s*     !r")typingrrcython.cimports.libavcimportslibavr cython.cimports.av.streamrrYr\ AVMediaTyper exceptvalpointerAVFormatContextr2r!finalcclassr"rrrrs(