o ڷi*@sdZddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z dd l m Z dd l mZdd lmZd e eefd e efddZd e eefd e efddZd e eefd efddZd e eefd efddZd e eefd e efddZd e eefd efddZd e eefd efddZd e eefd e efddZd e eefd efddZd8d e eefd!ed efd"d#Zd e eefd dfd$d%Zd&ee eefd dfd'd(Z d)e eefd dfd*d+Z!d,e eefd efd-d.Z"d,e eefd eee eefffd/d0Z#d,e eefd eee effd1d2Z$d,e eefd efd3d4Z%d5ed eee effd6d7Z&dS)9z# Audio file info computed by soxi. N)Number)Path)ListDict)OptionalUnion) VALID_FORMATS)sox)soxi)loggerinput_filepathreturncCs2t|t|d}|dkrtd|dSt|S)a Number of bits per sample, or None if not applicable. Parameters ---------- input_filepath : str Path to audio file. Returns ------- bitdepth : int or None Number of bits per sample. Returns None if not applicable. b0zBit depth unavailable for %sN)validate_input_filer r warningintr outputrC/home/ubuntu/vllm_env/lib/python3.10/site-packages/sox/file_info.pybitdepths   rcCspt|t|d}d}|dkrtd|dS|d|vr0d||d}t|dd|St|ddS)aV Bit rate averaged over the whole file. Expressed in bytes per second (bps), or None if not applicable. Parameters ---------- input_filepath : str Path to audio file. Returns ------- bitrate : float or None Bit rate, expressed in bytes per second. Returns None if not applicable. Bz kMGTPEZYrzBit rate unavailable for %sNg@@)rr r rindexfloat)r rgreek_prefixes multiplierrrrbitrate's   rcCt|t|d}t|S)z Show number of channels. Parameters ---------- input_filepath : str Path to audio file. Returns ------- channels : int number of channels c)rr rrrrrchannelsF r"cCr )a Show file comments (annotations) if available. Parameters ---------- input_filepath : str Path to audio file. Returns ------- comments : str File comments from header. If no comments are present, returns an empty string. arr strrrrrcommentsYs r'cCs6t|t|d}t|dkrtd|dSt|S)a! Show duration in seconds, or None if not available. Parameters ---------- input_filepath : str Path to audio file. Returns ------- duration : float or None Duration of audio file in seconds. If unavailable or empty, returns None. DgzDuration unavailable for %sN)rr rr rrrrrdurationms    r)cCr )z Show the name of the audio encoding. Parameters ---------- input_filepath : str Path to audio file. Returns ------- encoding : str audio encoding type er%rrrrencodingr#r+cCr )z Show detected file-type. Parameters ---------- input_filepath : str Path to audio file. Returns ------- file_type : str file format type (ex. 'wav') tr%rrrr file_typer#r-cCs:t|}t|t|d}|dkrtd|dSt|S)a; Show number of samples, or None if unavailable. Parameters ---------- input_filepath : path-like (str or pathlib.Path) Path to audio file. Returns ------- n_samples : int or None total number of samples in audio file. Returns None if empty or unavailable. srz$Number of samples unavailable for %sN)r&rr r rrrrrr num_sampless  r/cCr )z Show sample-rate. Parameters ---------- input_filepath : str Path to audio file. Returns ------- samplerate : float number of samples/second r)rr rrrrr sample_rater#r1MbP? thresholdcCs8t|t|}|d}|tdur||krdSdSdS)a Determine if an input file is silent. Parameters ---------- input_filepath : str The input filepath. threshold : float Threshold for determining silence Returns ------- is_silent : bool True if file is determined silent. z Mean normnanFT)rstatr)r r3stat_dictionary mean_normrrrsilents r8cCs\t|}|std|dt|}|tvr,tddttd|ddSdS)zInput file validation function. Checks that file exists and can be processed by SoX. Parameters ---------- input_filepath : path-like (str or pathlib.Path) The input filepath. zinput_filepath z does not exist.Valid formats: %s $This install of SoX cannot process . files.N) rexistsOSErrorfile_extensionr r infojoinr)r extrrrrs   rinput_filepath_listcCs<t|ts tdt|dkrtd|D]}t|qdS)zInput file list validation function. Checks that object is a list and contains valid filepaths that can be processed by SoX. Parameters ---------- input_filepath_list : list A list of filepaths. z#input_filepath_list must be a list.z/input_filepath_list must have at least 2 files.N) isinstancelist TypeErrorlen ValueErrorr)rCr rrrvalidate_input_file_lists   rJoutput_filepathcCst|}|dkr dSttj|ptttj ttj|tj g}t|r3t d|t |}|t vrMt ddt t d|dtj|r[t d|dSdS) a2Output file validation function. Checks that file can be written, and has a valid file extension. Throws a warning if the path already exists, as it will be overwritten on build. Parameters ---------- output_filepath : path-like (str or pathlib.Path) The output filepath. -nNz$SoX cannot write to output_filepath r9r:r;r<z?output_file: %s already exists and will be overwritten on build)r&boolospathdirnameaccessgetcwdW_OKallr>r?r r r@rArr=)rKnowrite_conditionsrBrrrvalidate_output_files.   rVfilepathcCst|jddS)zGet the extension of a filepath. Parameters ---------- filepath : path-like (str or pathlib.Path) File path. Returns ------- extension : str The file's extension rN)rsuffixlower)rWrrrr?Es r?c Cs:t|t|t|t|t|t|t|t|d}|S)aGet a dictionary of file information Parameters ---------- filepath : str File path. Returns ------- info_dictionary : dict Dictionary of file information. Fields are: * channels * sample_rate * bitdepth * bitrate * duration * num_samples * encoding * silent r"r1rrr)r/r+r8rZ)rWinfo_dictionaryrrrr@Us r@cCst|}t|}|S)zReturns a dictionary of audio statistics. Parameters ---------- filepath : str File path. Returns ------- stat_dictionary : dict Dictionary of audio statistics. ) _stat_call _parse_stat)rW stat_outputr6rrrr5ws r5cCs&t|d|ddg}t|\}}}|S)zCall sox's stat function. Parameters ---------- filepath : str File path. Returns ------- stat_output : str Sox output from stderr. r rLr5)rr )rWargs_r^rrrr\s r\r^c Cst|d}i}|D].}|d}t|dkr7|d}|dd}zt|}Wn ty2d}Ynw|||<q |S)zParse the string output from sox's stat function Parameters ---------- stat_output : str Sox output from stderr. Returns ------- stat_dictionary : dict Dictionary of audio statistics.  :rDrrr:N)splitrHstriprrI)r^lines stat_dictline split_linekeyvalrrrr]s     r])r2)'__doc__rNnumbersrpathlibrtypingrrrrcorer r r logr r&rrrrr"r'r)r+r-r/r1rMr8rrJrVr?r@r5r\r]rrrrs8       **&""