o ڷiv@sfdZddlZddlmZddlmZddlmZmZmZm Z m Z m Z ddl Z ddlmZddlmZdd lmZgd Zgd Zed Z  d%d e edee jdede eeeee jfeeffddZGdddeZdeefddZeZdeeefdedefddZd e edefddZ GdddeZ!de defd d!Z"d"ee defd#d$Z#dS)&zBase module for calling SoX N)Path)CalledProcessError)UnionListOptionalTupleIterableAny)Literal)NO_SOX)logger) BbcaDetsr) zsigned-integerzunsigned-integerzfloating-pointza-lawzu-lawz oki-adpcmz ima-adpcmzms-adpcmz gsm-full-rateTargs src_arraydecode_out_with_utfreturnc CsTdd|D}|ddkr|ddnd|d<z^tdd||durHtj|tjtjd}|\}}|r?| d }| d }|j }n+t |t j rotj|tjtjtjd }||jjd d \}}| d }|j }ntd |||fWSty}z td|WYd}~dSd}~wty}z td|WYd}~dSd}~ww)aPass an argument list to SoX. Parameters ---------- args : iterable Argument list for SoX. The first item can, but does not need to, be 'sox'. src_array : np.ndarray, or None If src_array is not None, then we make sure it's a numpy array and pass it into stdin. decode_out_with_utf : bool, default=True Whether or not sox is outputting a bytestring that should be decoded with utf-8. Returns ------- status : bool True on success. out : str, np.ndarray, or None Returns a np.ndarray if src_array was an np.ndarray. Returns the stdout produced by sox if src_array is None. Otherwise, returns None if there's an error. err : str, or None Returns stderr as a string. cSg|]}t|qSstr.0xrr>/home/ubuntu/vllm_env/lib/python3.10/site-packages/sox/core.py 9zsox..rsox Executing: %s Nstdoutstderrutf-8)stdinr)r*F)orderz src_array must be an np.ndarray!zOSError: SoX failed! %s TypeError: %s)r NN)lowerinsertr infojoin subprocessPopenPIPE communicatedecode returncode isinstancenpndarrayTtobytes TypeErrorOSErrorerror)rrrprocess_handleouterrstatus error_msgrrr"r%sH       r%c@eZdZdZddZdS)SoxErrorz@Exception to be raised when SoX exits with non-zero status. cOtj|g|Ri|dSN Exception__init__selfrkwargsrrr"rMkzSoxError.__init__N__name__ __module__ __qualname____doc__rMrrrr"rHg rHcsptrgStddgtturtdddfddttDd}|d d d }|S) z Calls SoX help for a lists of audio formats available with the current install of SoX. Returns ------- formats : list List of audio file extensions that SoX can process. r%z-hzUTF-8)encoding csg|] }d|vr|qS)zAUDIO FILE FORMATS:r)r isorr"r#sz&_get_valid_formats..rr'N)r r4 check_outputtypersplitrangelen)idxformatsrr[r"_get_valid_formatsos    refilepathargumentc Cst|}|tvrtd|dddg}|d|||z tj|tjd}WntyH}zt d|j t d|j d }~ww| d }t|d S) a  Base call to SoXI. Parameters ---------- filepath : path-like (str or pathlib.Path) Path to audio file. argument : str Argument to pass to SoXI. Returns ------- shell_output : str Command line output of SoXI zInvalid argument 'z ' to SoXIr%z--i-)r*zSoXI error message: zSoXI failed with exit code Nr+z )r SOXI_ARGS ValueErrorappendr4r^r6rr r2output SoxiErrorr9r8strip)rfrgr shell_outputcperrr"soxis$   rqc Csdd|D}|ddkr|ddnd|d<z2tdd|tj|tjtjd}|}|j dur=t|j |dkrDWd Std |Wd St ye}z t d |WYd}~d Sd}~wt y}}z t d |WYd}~d Sd}~ww)zPass an argument list to play. Parameters ---------- args : iterable Argument list for play. The first item can, but does not need to, be 'play'. Returns ------- status : bool True on success. cSrrrrrrr"r#r$zplay..rplayr&r'r(NTz Play returned with error code %sFzOSError: Play failed! %sr/) r0r1r r2r3r4r5r6waitr*r@rAr?)rrBrErFrrr"rrs2    rrc@rG)rmzAException to be raised when SoXI exits with non-zero status. cOrIrJrKrNrrr"rMrQzSoxiError.__init__NrRrrrr"rmrWrmvarcCs6zt|WdStyYdStyYdSw)zCheck if variable is a numeric value. Parameters ---------- var : object Returns ------- is_number : bool True if var is numeric, False otherwise. TF)floatrjr?)rtrrr" is_numbers   rvlist_of_thingscCstt|dkS)zCheck if a list contains identical elements. Parameters ---------- list_of_things : list list of objects Returns ------- all_equal : bool True if all list elements are the same. r )rbset)rwrrr" all_equals ry)NT)$rVr4pathlibrrtypingrrrrrr numpyr;typing_extensionsr r logr ri ENCODING_VALS EncodingValuerr<boolr%rLrHre VALID_FORMATSrqrrrmrvryrrrr"s>        M'-