o 헦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 ssinfowaveformrrX/home/ubuntu/transcripts/venv/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$aformatatrimrrr _get_load_filter$s  r(rztorchaudio.io.StreamReaderfilterchannels_firstcCsH|jdd|d||d}|durtd|j}|r"|jS|S)Nr") filter_descrzFailed to decode audio.)add_audio_streamprocess_all_packets 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   r6dtypec Cs\tjdtjdtjdtjdtjdtjdi}||}|dur,td|dt | d |S) Nu8s16s32s64fltdblzNo format found for dtype z; dtype must be one of .) torchuint8int16int32int64float32float64get ValueErrorlistkeys)r7dtype_to_formatr rrr _get_sample_format_s rKcCstjdkrdSdS)Nlittlelebe)sys byteorderrrrr _native_endiannessns rQencodingrcCs |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>N @zInvalid bits_per_sample z for WAV encoding.pcm_s16rSpcm_u8pcm_sPCM_SrTz4For WAV signed PCM, 8-bit encoding is not supported.PCM_U)NrSz7For WAV unsigned PCM, only 8-bit encoding is supported.PCM_FrV)rVrWpcm_fz?For WAV float PCM, only 32- and 64-bit encodings are supported.ULAW pcm_mulawz5For WAV PCM mu-law, only 8-bit encoding is supported.ALAWpcm_alawz4For WAV PCM A-law, only 8-bit encoding is supported.z WAV encoding z is not supported.)rGrQ)rRr endiannessrrr _get_encoder_for_wavusD rdcCs0|dus|dkr dS|dkrdStd|d)NrTr9rUr:z8FLAC only supports bits_per_sample values of 16 and 24 (z specified).)rG)bpsrrr _get_flac_sample_fmts rfextrecs~fdd}|ds|drd}t||}d}n!|dr$d}d}d}n}d}d}|dr2t|}|dr:t|}|||fS)Ncs|kp duo |kSNr)specrgr rr _typesz_parse_save_args.._typewavambr3r4flac)rdrf)rgr rRrerkmuxerencoder sample_fmtrrjr _parse_save_argss"  rrurir 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.r>rr")r r )rr rpencoder_format codec_configr)r5r%ospathnormpathstrsplitlowerrrr0rr StreamWriterr,rrKr7openwrite_audio_chunk)rsr rr*r rRrr rtrgtokensrorpenc_fmtrrrr save_audios.    "rcCs6dD] }||vr |Sq|dkrdS|dkrdS|S)N)r[r\r] PCM_MULAWr_PCM_ALAWrar)rRdstrrr _map_encodingsrcCs.td|}rt|dS|dvrdS|S)NzPCM_\w(\d+)\w*r)rrrS)researchrgroup)rRrmrrr _get_bits_per_samples rc@seZdZeddedeededefddZ e  dded ed ede de deedede e j effddZe  ddede j dede deedeedeededeeejjeefdd fddZededeede fddZededeede fddZd S) FFmpegBackendr rsr r r cCs,t|||}t|j|j|_t|j|_|Srh)r!rrRrr)rsr r metadatarrr infos  zFFmpegBackend.inforr"TNr#r normalizer*cCst||||||Srh)r6)rsr#rrr*r r rrr loads zFFmpegBackend.loadr rrRrrtc 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 CodecConfigtyperGr) rsr rr*r rRrr rtrrr save+s   zFFmpegBackend.savecCdSNTrrsr rrr can_decodeHzFFmpegBackend.can_decodecCrrrrrrr can_encodeLrzFFmpegBackend.can_encoder rr"TTNr TNNNr N)__name__ __module__ __qualname__ staticmethod InputTyperr{rrrboolrr?Tensorrrrrrfloatrrrrrrr rsv"      rr)rr"Trrr)$rxrrOtypingrrrrr?rbackendrcommonrr{PathLikerrr!rr(rrr6r7rKrQrdrfrrrrrrrrrrrr s       & ?  &