o U)i,@sddlZddlZddlZddlmZmZmZmZddlZddl Z ddl m Z ddl m Z eeeejfZ d6dedeed ed e fd d Z d7dededed eefddZ  d8dddeeded ejfddZ    d9dedededededeed ed eejeffddZdejd efddZd efdd Zd!ed"ed efd#d$Zd%d&Zd'eedeed!eed(eefd)d*Z      d:d+edejd,ededeed!eed"eed ed-ee jjd dfd.d/Z d!ed efd0d1Z!d!ed"ed efd2d3Z"Gd4d5d5e Z#dS);N)BinaryIOOptionalTupleUnion)Backend) AudioMetaDatasrcformat buffer_sizereturncCsbtj||d|}||j}|jdkrt|}|d}n|j}tt |j ||j |j |j S)Nrr) torchaudioio StreamReaderget_src_stream_infodefault_audio_stream num_frames _load_audiosizerint sample_rate num_channelsbits_per_samplecodecupper)r r r ssinfowaveformrrf/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/torchaudio/_backend/ffmpeg.py info_audios   r!T frame_offsetrconvertcCs|dkr td||dks|dkrtd||dkr&|dkr&|s&dSd}|dkr4|dkr4|r4|S|dkrAd|||}nd|}|sJ|Sd||S) Nrz>Invalid argument: frame_offset must be non-negative. Found: {}r"zDInvalid argument: num_frames must be -1 or greater than 0. Found: {}zaformat=sample_fmts=fltpz#atrim=start_sample={}:end_sample={}zatrim=start_sample={}z{},{}) RuntimeErrorr )r#rr$ZaformatZatrimrrr _get_load_filter$s  r&rztorchaudio.io.StreamReaderfilterchannels_firstcCsH|jdd|d||d}|durtd|j}|r"|jS|S)Nr")Z filter_descrzFailed to decode audio.)add_audio_streamZprocess_all_packetsZ pop_chunksr%_elemT)rr'r(chunkrrrr r?s rc CsZt|dr |dkr d}tj||d|}t||jj}t|||} t || |} | |fS)Nreadvorbisogg) hasattrrrrrrrrr&r) r r#rr$r(r r rrr'rrrr load_audioMs   r1dtypec Cs\tjdtjdtjdtjdtjdtjdi}||}|dur,td|dt | d |S) Nu8s16s32Zs64ZfltZdblzNo format found for dtype z; dtype must be one of .) torchZuint8Zint16Zint32Zint64Zfloat32Zfloat64get ValueErrorlistkeys)r2Zdtype_to_formatr rrr _get_sample_format_s r<cCstjdkrdSdS)Nlittlelebe)sys byteorderrrrr _native_endiannessns rBencodingrcCs |dvr td|dt}|s%|sd|S|dkrdSd||S|dkr<|s-d }|dkr5td d||S|d krJ|d vrFdStd |dkra|sRd}|dvr]d||Std|dkro|d vrkdStd|dkr}|d vrydStdtd|d)N> @NzInvalid bits_per_sample z for WAV encoding.Zpcm_s16rFZpcm_u8Zpcm_sPCM_SrGz4For WAV signed PCM, 8-bit encoding is not supported.PCM_U)NrFz7For WAV unsigned PCM, only 8-bit encoding is supported.PCM_FrD)rDrEZpcm_fz?For WAV float PCM, only 32- and 64-bit encodings are supported.ULAWZ pcm_mulawz5For WAV PCM mu-law, only 8-bit encoding is supported.ALAWZpcm_alawz4For WAV PCM A-law, only 8-bit encoding is supported.z WAV encoding z is not supported.)r9rB)rCrZ endiannessrrr _get_encoder_for_wavusD rNcCs0|dus|dkr dS|dkrdStd|d)NrGr4rHr5z8FLAC only supports bits_per_sample values of 16 and 24 (z specified).)r9)bpsrrr _get_flac_sample_fmts rPextrOcs~fdd}|ds|drd}t||}d}n!|dr$d}d}d}n}d}d}|dr2t|}|dr:t|}|||fS)Ncs|kp duo |kSNr)specrQr rr _typesz_parse_save_args.._typeZwavZambr.r/Zflac)rNrP)rQr rCrOrUmuxerencoderZ sample_fmtrrTr _parse_save_argss"  rXurir compressionc Csd} t|dr|durtdntj|}t|ddd} r)| d} t| |||\} } } |r8|j }t j j || |d}|j ||dt|j| | |d||d|WddS1shwYdS) Nwritez0'format' is required when saving to file object.r6rr")r r )rr rWZencoder_formatZ codec_configr)r0r%ospathnormpathstrsplitlowerrXr+rr StreamWriterr)rr<r2openZwrite_audio_chunk)rYr rr(r rCrr rZrQtokensrVrWZenc_fmtrrrr save_audios.    "recCs6dD] }||vr |Sq|dkrdS|dkrdS|S)N)rIrJrK PCM_MULAWrLPCM_ALAWrMr)rCdstrrr _map_encodingsricCs.td|}rt|dS|dvrdS|S)NzPCM_\w(\d+)\w*r)rgrfrF)researchrgroup)rCrmrrr _get_bits_per_samples rnc@seZdZeddedeededefddZ e  dded ed ede de deedede e j effddZe  ddede j dede deedeedeededeeejjeefdd fddZededeede fddZededeede fddZd S) FFmpegBackendr rYr r r cCs,t|||}t|j|j|_t|j|_|SrR)r!rnrCrri)rYr r metadatarrr infos  zFFmpegBackend.inforr"TNr#r normalizer(cCst||||||SrR)r1)rYr#rrrr(r r rrr loads zFFmpegBackend.loadr rrCrrZc CsFt|tjjtdfstddt|t||||||||| dS)NzLFFmpeg backend expects non-`None` value for argument `compression` to be of z=type `torchaudio.io.CodecConfig`, but received value of type ) isinstancerr CodecConfigtyper9re) rYr rr(r rCrr rZrrr save+s   zFFmpegBackend.savecCdSNTrrYr rrr can_decodeHzFFmpegBackend.can_decodecCrxryrrzrrr can_encodeLr|zFFmpegBackend.can_encoder rr"TTNr TNNNr N)__name__ __module__ __qualname__ staticmethod InputTyperr_rrrqboolrr7Tensorrsrrrrufloatrwr{r}rrrr rosv"      ror~)rr"Tryrr)$r\rjr@typingrrrrr7rbackendrcommonrr_PathLikerrr!rr&rrr1r2r<rBrNrPrXrrurerirnrorrrr s       & ?  &