B ²}ú]O8ã@s¾dZddlmZmZmZddlZddlZddlZddlZdddgZ Gdd„de ƒZ dZ d Z d Ze e fZd d „Zddd„Zdd„Zdd„Zddd„Zdd„Zejdd kr²dd„Zndd„ZdS)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. é)ÚdivisionÚprint_functionÚabsolute_importNÚWavFileWarningÚreadÚwritec@s eZdZdS)rN)Ú__name__Ú __module__Ú __qualname__©r r ú\/home/fristb/BIRCH/lib-linux-x86_64/python/lib64/python3.7/site-packages/scipy/io/wavfile.pyrsééiþÿcCsH|r d}nd}t |d| d¡¡d}}d}|dkr@tdƒ‚t |d| d¡¡}|d7}|\}}}} } } |tkr |d kr t |d | d ¡¡d} |d 7}| d kr| d ¡} |d 7}| d d …}|rÖd}nd}| |¡r t |d|dd…¡d}ntdƒ‚|tkrtdƒ‚||kr6| ||¡||||| | | 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 ú>ú<ÚIéréz.Binary structure of wave file is not compliantZHHIIHHéÚHééés €ª8›qs €ª8›qNzUnknown wave file format)ÚstructÚunpackrÚ ValueErrorÚWAVE_FORMAT_EXTENSIBLEÚendswithÚKNOWN_WAVE_FORMATS)ÚfidÚ is_big_endianÚfmtÚsizeÚresZ bytes_readÚ format_tagÚchannelsÚfsÚbytes_per_secondÚ block_alignÚ bit_depthZext_chunk_sizeZextensible_chunk_dataZraw_guidÚtailr r r Ú_read_fmt_chunk's:       r+Fc CsÌ|r d}nd}t || d¡¡d}|d}|dkr:d} n0|rDd} nd} |tkr^| d |7} n | d |7} |s„tj| |¡| d } n0| ¡} tj|| d | ||fd } | | |¡|dkrÈ|   d|¡} | S)Nz>IzcCs<|r d}nd}| d¡}|r8t ||¡d}| |d¡dS)Nz>IzIz!File format {}... not understood.rr,sWAVEzNot a WAV file.)rrÚformatÚreprrr)rZstr1r r!Ú file_sizeZstr2r r r Ú_read_riff_chunks   rCc Csªt|dƒr|}d}n t|dƒ}z^t|ƒ\}}d}d}d}d}t} x6| ¡|krz| d¡} | s|r†tjd | ¡|¡t dd Pq¤t d ƒ‚nt | ƒdkr¤t d ƒ‚| d krìd }t ||ƒ} | dd…\} }} | d}|dkrêt d |¡ƒ‚qF| dkrt ||ƒqF| dkr2d }|st dƒ‚t|| ||||ƒ} qF| dkrHt ||ƒqF| dkr^t ||ƒqFtjdt dd t ||ƒqFWWdt|dƒs–| ¡n | d¡X| | 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() rFÚrbr r,rzQReached EOF prematurely; finished at {:d} bytes, expected {:d} bytes from header.r)Ú stacklevelzUnexpected end of file.zIncomplete wav chunk.sfmt Tr)r,ré é@é`é€z4Unsupported bit depth: the wav file has {}-bit data.sfactsdatazNo fmt chunk before datasLIST)sJUNKsFakez-Chunk (non-data) not understood, skipping it.Nr)ÚhasattrÚopenrCr4r7rÚwarningsÚwarnr@rrÚlenr+r?r>Úcloser9)Úfilenamer;rrBr Zfmt_chunk_receivedZdata_chunk_receivedr%r)r$Zchunk_idZ fmt_chunkr&r<r r r r´sdP                       c Cs2t|dƒr|}n t|dƒ}|}zê|jj}|dks\|dks\|dkrN|jjdks\td|jƒ‚d}|d 7}|d 7}|d 7}|d 7}|dkrŽt}nt}|jdkr¢d}n |j d}|jjd } || d |} || d } t   d|||| | | ¡} |dks|dks| d7} |t   dt | ƒ¡7}|| 7}|dksV|dksV|d7}|t   dd|j d¡7}t |ƒddd |j dkr~tdƒ‚| |¡| d¡| t   d|j ¡¡|jjdksÎ|jjdkrÖtjdkrÖ| ¡}t||ƒ| ¡} | d¡| t   d| d ¡¡Wdt|dƒs"| ¡n | d¡XdS)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) rÚwbÚiÚfÚur zUnsupported data type '%s'ósRIFFssWAVEsfmt r,zr?rCrrÚ version_inforar r r r Ú s.C #