3 6^O8@sdZddlmZmZmZddlZddlZddlZddlZdddgZ Gddde Z dZ d Z d Ze e fZd d ZdddZddZddZdddZddZejdd krddZnddZdS)z Module to read / write wav files using numpy arrays Functions --------- `read`: Return the sample rate (in samples/sec) and data from a WAV file. `write`: Write a numpy array as a WAV file. )divisionprint_functionabsolute_importNWavFileWarningreadwritec@s eZdZdS)rN)__name__ __module__ __qualname__r r 1/tmp/pip-build-vw4w4j08/scipy/scipy/io/wavfile.pyrsicCsF|r d}nd}tj|d|jdd}}d}|dkr@tdtj|d|jd}|d7}|\}}}} } } |tko||dkr tj|d |jd d} |d 7}| d kr|jd } |d 7}| dd}|rd }nd }|j|r tj|d|ddd}ntd|tkrtd||kr4|j||||||| | | fS)a Returns ------- size : int size of format subchunk in bytes (minus 8 for "fmt " and itself) format_tag : int PCM, float, or compressed format channels : int number of channels fs : int sampling frequency in samples per second bytes_per_second : int overall byte rate for the file block_align : int bytes per sample, including all channels bit_depth : int bits per sample ><Irz.Binary structure of wave file is not compliantZHHIIHHHs 8qs 8qNzUnknown wave file formatrr)structunpackr ValueErrorWAVE_FORMAT_EXTENSIBLEendswithKNOWN_WAVE_FORMATS)fid is_big_endianfmtsizeres bytes_read format_tagchannelsfsbytes_per_second block_align bit_depthZext_chunk_sizeZextensible_chunk_dataZraw_guidtailr r r _read_fmt_chunk's:       r,Fc Cs|r d}nd}tj||jdd}|d}|dkr:d} n0|rDd} nd} |tkr^| d |7} n | d |7} |stj|j|| d } n0|j} tj|| d | ||fd } |j| ||dkr| j d|} | S)Nz>IzIzIz!File format {}... not understood.rr-sWAVEzNot a WAV file.)rrformatreprrr)rZstr1r r! file_sizeZstr2r r r _read_riff_chunks   rAc Cst|dr|}d}n t|d}z^t|\}}d}d}d}d}t} x6|j|krz|jd} | s|rtjdj|j|t dd Pqt d nt | dkrt d | d krd }t ||} | dd\} }} | d}|dkrt dj|qF| dkrt ||qF| dkr2d }|st dt|| ||||} qF| dkrHt ||qF| dkr^t ||qFtjdt dd t ||qFWWdt|ds|jn |jdX| | fS) a Open a WAV file Return the sample rate (in samples/sec) and data from a WAV file. Parameters ---------- filename : string or open file handle Input wav file. mmap : bool, optional Whether to read data as memory-mapped. Only to be used on real files (Default: False). .. versionadded:: 0.12.0 Returns ------- rate : int Sample rate of wav file. data : numpy array Data read from wav file. Data-type is determined from the file; see Notes. Notes ----- This function cannot read wav files with 24-bit data. Common data types: [1]_ ===================== =========== =========== ============= WAV format Min Max NumPy dtype ===================== =========== =========== ============= 32-bit floating-point -1.0 +1.0 float32 32-bit PCM -2147483648 +2147483647 int32 16-bit PCM -32768 +32767 int16 8-bit PCM 0 255 uint8 ===================== =========== =========== ============= Note that 8-bit PCM is unsigned. References ---------- .. [1] IBM Corporation and Microsoft Corporation, "Multimedia Programming Interface and Data Specifications 1.0", section "Data Format of the Samples", August 1991 http://www.tactilemedia.com/info/MCI_Control_Info.html Examples -------- >>> from os.path import dirname, join as pjoin >>> import scipy.io as sio Get the filename for an example .wav file from the tests/data directory. >>> data_dir = pjoin(dirname(sio.__file__), 'tests', 'data') >>> wav_fname = pjoin(data_dir, 'test-44100Hz-2ch-32bit-float-be.wav') Load the .wav file contents. >>> samplerate, data = sio.wavfile.read(wav_fname) >>> print(f"number of channels = {data.shape[1]}") number of channels = 2 >>> length = data.shape[0] / samplerate >>> print(f"length = {length}s") length = 0.01s Plot the waveform. >>> import matplotlib.pyplot as plt >>> import numpy as np >>> time = np.linspace(0., length, data.shape[0]) >>> plt.plot(time, data[:, 0], label="Left channel") >>> plt.plot(time, data[:, 1], label="Right channel") >>> plt.legend() >>> plt.xlabel("Time [s]") >>> plt.ylabel("Amplitude") >>> plt.show() rFrbr r-rzQReached EOF prematurely; finished at {:d} bytes, expected {:d} bytes from header.r) stacklevelzUnexpected end of file.zIncomplete wav chunk.sfmt Trr @`z4Unsupported bit depth: the wav file has {}-bit data.sfactsdatazNo fmt chunk before datasLISTJUNKFakez-Chunk (non-data) not understood, skipping it.Nr)r-rrDrErFrG)rHrI)hasattropenrAr5r7rwarningswarnr>rrlenr,r=r<closer8)filenamer9rr@r Zfmt_chunk_receivedZdata_chunk_receivedr&r*r%Zchunk_idZ fmt_chunkr'r:r r r rsdP                       cCs0t|dr|}n t|d}|}z|jj}|dkpL|dkpL|dkoL|jjdks\td|jd}|d 7}|d 7}|d 7}|d 7}|dkrt}nt}|jdkrd}n |j d}|jjd } || d |} || d } t j d|||| | | } |dkp|dks| d7} |t j dt | 7}|| 7}|dkp0|dksT|d7}|t j dd|j d7}t |ddd|j dkr|td|j||jd|jt j d|j |jjdks|jjdkrtjdkr|j}t|||j} |jd|jt j d| d Wdt|ds |jn |jdXdS)a Write a numpy array as a WAV file. Parameters ---------- filename : string or open file handle Output wav file. rate : int The sample rate (in samples/sec). data : ndarray A 1-D or 2-D numpy array of either integer or float data-type. Notes ----- * Writes a simple uncompressed WAV file. * To write multiple-channels, use a 2-D array of shape (Nsamples, Nchannels). * The bits-per-sample and PCM/float will be determined by the data-type. Common data types: [1]_ ===================== =========== =========== ============= WAV format Min Max NumPy dtype ===================== =========== =========== ============= 32-bit floating-point -1.0 +1.0 float32 32-bit PCM -2147483648 +2147483647 int32 16-bit PCM -32768 +32767 int16 8-bit PCM 0 255 uint8 ===================== =========== =========== ============= Note that 8-bit PCM is unsigned. References ---------- .. [1] IBM Corporation and Microsoft Corporation, "Multimedia Programming Interface and Data Specifications 1.0", section "Data Format of the Samples", August 1991 http://www.tactilemedia.com/info/MCI_Control_Info.html Examples -------- Create a 100Hz sine wave, sampled at 44100Hz. Write to 16-bit PCM, Mono. >>> from scipy.io.wavfile import write >>> samplerate = 44100; fs = 100 >>> t = np.linspace(0., 1., samplerate) >>> amplitude = np.iinfo(np.int16).max >>> data = amplitude * np.sin(2. * np.pi * fs * t) >>> write("example.wav", samplerate, data) rwbifur zUnsupported data type '%s'sRIFFssWAVEsfmt r-z s.C #