3 h V@sdZddlZddlZddlZddlZddlmZdddgZGdddeZ Gdd d eZ e j e j hZ d d Zdd dZddZddZdddZddZddZdS)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. N)IntEnumWavFileWarningreadwritec@s eZdZdS)rN)__name__ __module__ __qualname__r r 1/tmp/pip-build-riy7u7_k/scipy/scipy/io/wavfile.pyrsc@s|eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"d Z#d!Z$d"Z%d#Z&d$Z'd%Z(d&Z)d'Z*d(Z+d)Z,d*Z-d+Z.d,Z/d-Z0d.Z1d/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5Z8d6Z9d7Z:d8Z;d9ZdZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZHdFZIdGZJdHZKdIZLdJZMdKZNdLZOdMZPdNZQdOZRdPZSdQZTdRZUdSZVdTZWdUZXdVZYdWZZdXZ[dYZ\dZZ]d[Z^d\Z_d]Z`d^Zad_Zbd`ZcdaZddbZedcZfddZgdeZhdfZidgZjdhZkdiZldjZmdkZndlZodmZpdnZqdoZrdpZsdqZtdrZudsZvdtZwduZxdvZydwZzdxZ{dyZ|dzZ}d{Z~d|Zd}Zd~ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd S( WAVE_FORMATz WAVE form wFormatTag IDs Complete list is in mmreg.h in Windows 10 SDK. ALAC and OPUS are the newest additions, in v10.0.14393 2016-07 r  !"#$%&'(0123456789:;<=@ABCDEPRSUY`abcdefgipqrstuvwxyz{iiii i!i#i%i0i1i2i3i4i5i6i@iAiBiPiQiUi`iaibicidipiqirisitiuiviwixiiiiiiiii i0i@iAiIiPiQi`ipiqirisiiiiiPiQiiiiPiiiiiiiiiiiiiiiiiiiiiii i i iiiqiyii iii i i3iCAiBiCBiLCiLViVWiOgiPgiQgiogipgiqgialipiOpimpiasibsicsi!zi"ziiiiiiiiii i i i i iiiiiiiiiiiiiiiiiii i!i"i#i$iiiN(rrr__doc__UNKNOWNPCMZADPCM IEEE_FLOATZVSELPZIBM_CVSDZALAWZMULAWZDTSZDRMZ WMAVOICE9Z WMAVOICE10Z OKI_ADPCMZ DVI_ADPCMZ IMA_ADPCMZMEDIASPACE_ADPCMZ SIERRA_ADPCMZ G723_ADPCMZDIGISTDZDIGIFIXZDIALOGIC_OKI_ADPCMZMEDIAVISION_ADPCMZCU_CODECZ HP_DYN_VOICEZ YAMAHA_ADPCMZSONARCZDSPGROUP_TRUESPEECHZECHOSC1ZAUDIOFILE_AF36ZAPTXZAUDIOFILE_AF10Z PROSODY_1612ZLRCZ DOLBY_AC2ZGSM610ZMSNAUDIOZ ANTEX_ADPCMEZCONTROL_RES_VQLPCZDIGIREALZ DIGIADPCMZCONTROL_RES_CR10Z NMS_VBXADPCMZ CS_IMAADPCMZECHOSC3ZROCKWELL_ADPCMZROCKWELL_DIGITALKZXEBECZ G721_ADPCMZ G728_CELPZMSG723Z INTEL_G723_1Z INTEL_G729Z SHARP_G726ZMPEGZRT24ZPACZ MPEGLAYER3Z LUCENT_G723ZCIRRUSZESPCMZVOXWAREZ CANOPUS_ATRACZ G726_ADPCMZ G722_ADPCMZDSATZ DSAT_DISPLAYZVOXWARE_BYTE_ALIGNEDZ VOXWARE_AC8Z VOXWARE_AC10Z VOXWARE_AC16Z VOXWARE_AC20Z VOXWARE_RT24Z VOXWARE_RT29ZVOXWARE_RT29HWZ VOXWARE_VR12Z VOXWARE_VR18Z VOXWARE_TQ40Z VOXWARE_SC3Z VOXWARE_SC3_1Z SOFTSOUNDZ VOXWARE_TQ60ZMSRT24ZG729AZMVI_MVI2ZDF_G726Z DF_GSM610ZISIAUDIOZONLIVEZMULTITUDE_FT_SX20ZINFOCOM_ITS_G721_ADPCMZ CONVEDIA_G729Z CONGRUENCYZSBC24ZDOLBY_AC3_SPDIFZMEDIASONIC_G723Z PROSODY_8KBPSZ ZYXEL_ADPCMZ PHILIPS_LPCBBZPACKEDZMALDEN_PHONYTALKZRACAL_RECORDER_GSMZRACAL_RECORDER_G720_AZRACAL_RECORDER_G723_1ZRACAL_RECORDER_TETRA_ACELPZNEC_AACZRAW_AAC1ZRHETOREX_ADPCMZIRATZ VIVO_G723Z VIVO_SIRENZ PHILIPS_CELPZPHILIPS_GRUNDIGZ DIGITAL_G723ZSANYO_LD_ADPCMZSIPROLAB_ACEPLNETZSIPROLAB_ACELP4800ZSIPROLAB_ACELP8V3Z SIPROLAB_G729ZSIPROLAB_G729AZSIPROLAB_KELVINZ VOICEAGE_AMRZ G726ADPCMZDICTAPHONE_CELP68ZDICTAPHONE_CELP54ZQUALCOMM_PUREVOICEZQUALCOMM_HALFRATEZTUBGSMZMSAUDIO1ZWMAUDIO2ZWMAUDIO3ZWMAUDIO_LOSSLESSZWMASPDIFZUNISYS_NAP_ADPCMZUNISYS_NAP_ULAWZUNISYS_NAP_ALAWZUNISYS_NAP_16KZSYCOM_ACM_SYC008ZSYCOM_ACM_SYC701_G726LZSYCOM_ACM_SYC701_CELP54ZSYCOM_ACM_SYC701_CELP68ZKNOWLEDGE_ADVENTURE_ADPCMZFRAUNHOFER_IIS_MPEG2_AACZDTS_DSZCREATIVE_ADPCMZCREATIVE_FASTSPEECH8ZCREATIVE_FASTSPEECH10Z UHER_ADPCMZULEAD_DV_AUDIOZULEAD_DV_AUDIO_1Z QUARTERDECKZILINK_VCZ RAW_SPORTZESST_AC3ZGENERIC_PASSTHRUZIPI_HSXZ IPI_RPELPZCS2ZSONY_SCXZSONY_SCYZ SONY_ATRAC3ZSONY_SPCZ TELUM_AUDIOZTELUM_IA_AUDIOZNORCOM_VOICE_SYSTEMS_ADPCMZ FM_TOWNS_SNDZMICRONASZMICRONAS_CELP833Z BTV_DIGITALZINTEL_MUSIC_CODERZ INDEO_AUDIOZ QDESIGN_MUSICZ ON2_VP7_AUDIOZ ON2_VP6_AUDIOZ VME_VMPCMZTPCZLIGHTWAVE_LOSSLESSZOLIGSMZOLIADPCMZOLICELPZOLISBCZOLIOPRZLH_CODECZ LH_CODEC_CELPZ LH_CODEC_SBC8ZLH_CODEC_SBC12ZLH_CODEC_SBC16ZNORRISZ ISIAUDIO_2ZSOUNDSPACE_MUSICOMPRESSZ MPEG_ADTS_AACZ MPEG_RAW_AACZ MPEG_LOASZNOKIA_MPEG_ADTS_AACZNOKIA_MPEG_RAW_AACZVODAFONE_MPEG_ADTS_AACZVODAFONE_MPEG_RAW_AACZ MPEG_HEAACZVOXWARE_RT24_SPEECHZSONICFOUNDRY_LOSSLESSZINNINGS_TELECOM_ADPCMZLUCENT_SX8300PZLUCENT_SX5363SZCUSEEMEZNTCSOFT_ALF2CM_ACMZDVMZDTS2ZMAKEAVISZDIVIO_MPEG4_AACZNOKIA_ADAPTIVE_MULTIRATEZ DIVIO_G726Z LEAD_SPEECHZ LEAD_VORBISZ WAVPACK_AUDIOZOGG_VORBIS_MODE_1ZOGG_VORBIS_MODE_2ZOGG_VORBIS_MODE_3ZOGG_VORBIS_MODE_1_PLUSZOGG_VORBIS_MODE_2_PLUSZOGG_VORBIS_MODE_3_PLUSZALACZ _3COM_NBXZOPUSZFAAD_AACZAMR_NBZAMR_WBZAMR_WPZ GSM_AMR_CBRZGSM_AMR_VBR_SIDZCOMVERSE_INFOSYS_G723_1ZCOMVERSE_INFOSYS_AVQSBCZCOMVERSE_INFOSYS_SBCZ SYMBOL_G729_AZVOICEAGE_AMR_WBZINGENIENT_G726Z MPEG4_AACZ ENCORE_G726Z ZOLL_ASAOZ SPEEX_VOICEZ VIANIX_MASCZWM9_SPECTRUM_ANALYZERZWMF_SPECTRUM_ANAYZERZGSM_610ZGSM_620ZGSM_660ZGSM_690ZGSM_ADAPTIVE_MULTIRATE_WBZ POLYCOM_G722Z POLYCOM_G728ZPOLYCOM_G729_AZ POLYCOM_SIRENZGLOBAL_IP_ILBCZRADIOTIME_TIME_SHIFT_RADIOZNICE_ACAZ NICE_ADPCMZ VOCORD_G721Z VOCORD_G726Z VOCORD_G722_1Z VOCORD_G728Z VOCORD_G729Z VOCORD_G729_AZ VOCORD_G723_1Z VOCORD_LBCZ NICE_G728ZFRACE_TELECOM_G729ZCODIANZFLAC EXTENSIBLEZ DEVELOPMENTr r r r r sr cCs|r d}nd}tj|d|jdd}}d}|dkr@tdtj|d|jd}|d7}|\}}}} } } |tjko~|dkr tj|d |jd d} |d 7}| d kr|jd } |d 7}| dd}|rd }nd }|j|r tj|d|ddd}ntd|tkrhyt|j}Wntk rB|d}YnXtd|ddj ddtD||kr|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 ><Irrrz.Binary structure of wave file is not compliantZHHIIHHr Hrs 8qs 8qNz#06xzUnknown wave file format: z. Supported formats: z, css|] }|jVqdS)N)name).0xr r r usz"_read_fmt_chunk..rrrr) structunpackr ValueErrorr ryendswithKNOWN_WAVE_FORMATSr~join)fid is_big_endianfmtsizeres bytes_read format_tagchannelsfsbytes_per_second block_align bit_depthZext_chunk_sizeZextensible_chunk_dataZraw_guidtailZ format_namer r r _read_fmt_chunk8sD        rFc Cs|r d}nd}tj||jdd}|d}|dkr:d} n2|rDd} nd} |tjkr`| d |7} n | d |7} |stj|j|| d } n0|j} tj|| d | ||fd } |j | ||dkr| j d|} | S)Nz>IzIzIz File format z2 not understood. Only 'RIFF' and 'RIFX' supported.rrsWAVEz"Not a WAV file. RIFF form type is .)rrreprrr)rZstr1rr file_sizeZstr2r r r _read_riff_chunks  rc Cst|dr|}d}n t|d}zt|\}}d}d}d}d}tj} xf|j|kr|jd} | s|rtjdj |j|t dd Pqt d n@t | dkrd t | } |r|rtj| d t dd nt | | d krd}t||} | dd\} }} | d}|dkrt dj |qH| dkr4t||qH| dkrdd}|sPt dt|| ||||}qH| dkrzt||qH| d krt||qHtjdt dd t||qHWWdt|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 >>> from scipy.io import wavfile >>> import scipy.io Get the filename for an example .wav file from the tests/data directory. >>> data_dir = pjoin(dirname(scipy.io.__file__), 'tests', 'data') >>> wav_fname = pjoin(data_dir, 'test-44100Hz-2ch-32bit-float-be.wav') Load the .wav file contents. >>> samplerate, data = 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 rrzQReached EOF prematurely; finished at {:d} bytes, expected {:d} bytes from header.r ) stacklevelzUnexpected end of file.zIncomplete chunk ID: z, ignoring it.sfmt Trrr"r9rDrYz4Unsupported bit depth: the wav file has {}-bit data.sfactsdatazNo fmt chunk before datasLISTJUNKFakez-Chunk (non-data) not understood, skipping it.Nr>r"r9rDrYrr>rr)hasattropenrr rwrrwarningswarnformatrrlenrrrrcloser)filenamerrrrZfmt_chunk_receivedZdata_chunk_receivedrrrZchunk_idmsgZ fmt_chunkrrr r r rslQ                           cCs4t|dr|}n t|d}|}z|jj}|dkpL|dkpL|dkoL|jjdks\td|jd}|d 7}|d 7}|d 7}|d 7}|dkrtj}ntj}|j dkrd}n |j d}|jjd } || d |} || d } t j d|||| | | } |dkp|dks| d7} |t j dt | 7}|| 7}|dkp4|dksX|d7}|t j dd|j d7}t |ddd|jdkrtd|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 rz s*  I #