o wTho@sdZddlZddlZddlZddlZddlZddlmZgdZ Gddde Z GdddeZ e j e jhZd d Zd d Z dddZddZddZddZdddZddZddZdS)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)IntEnum)WavFileWarningreadwritec@s eZdZdS)rN)__name__ __module__ __qualname__r r Q/home/air/segue/gemini/back/venv/lib/python3.10/site-packages/scipy/io/wavfile.pyrsrc@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__UNKNOWNPCMADPCM IEEE_FLOATVSELPIBM_CVSDALAWMULAWDTSDRM WMAVOICE9 WMAVOICE10 OKI_ADPCM DVI_ADPCM IMA_ADPCMMEDIASPACE_ADPCM SIERRA_ADPCM G723_ADPCMDIGISTDDIGIFIXDIALOGIC_OKI_ADPCMMEDIAVISION_ADPCMCU_CODEC HP_DYN_VOICE YAMAHA_ADPCMSONARCDSPGROUP_TRUESPEECHECHOSC1AUDIOFILE_AF36APTXAUDIOFILE_AF10 PROSODY_1612LRC DOLBY_AC2GSM610MSNAUDIO ANTEX_ADPCMECONTROL_RES_VQLPCDIGIREAL DIGIADPCMCONTROL_RES_CR10 NMS_VBXADPCM CS_IMAADPCMECHOSC3ROCKWELL_ADPCMROCKWELL_DIGITALKXEBEC G721_ADPCM G728_CELPMSG723 INTEL_G723_1 INTEL_G729 SHARP_G726MPEGRT24PAC MPEGLAYER3 LUCENT_G723CIRRUSESPCMVOXWARE CANOPUS_ATRAC G726_ADPCM G722_ADPCMDSAT DSAT_DISPLAYVOXWARE_BYTE_ALIGNED VOXWARE_AC8 VOXWARE_AC10 VOXWARE_AC16 VOXWARE_AC20 VOXWARE_RT24 VOXWARE_RT29VOXWARE_RT29HW VOXWARE_VR12 VOXWARE_VR18 VOXWARE_TQ40 VOXWARE_SC3 VOXWARE_SC3_1 SOFTSOUND VOXWARE_TQ60MSRT24G729AMVI_MVI2DF_G726 DF_GSM610ISIAUDIOONLIVEMULTITUDE_FT_SX20INFOCOM_ITS_G721_ADPCM CONVEDIA_G729 CONGRUENCYSBC24DOLBY_AC3_SPDIFMEDIASONIC_G723 PROSODY_8KBPS ZYXEL_ADPCM PHILIPS_LPCBBPACKEDMALDEN_PHONYTALKRACAL_RECORDER_GSMRACAL_RECORDER_G720_ARACAL_RECORDER_G723_1RACAL_RECORDER_TETRA_ACELPNEC_AACRAW_AAC1RHETOREX_ADPCMIRAT VIVO_G723 VIVO_SIREN PHILIPS_CELPPHILIPS_GRUNDIG DIGITAL_G723SANYO_LD_ADPCMSIPROLAB_ACEPLNETSIPROLAB_ACELP4800SIPROLAB_ACELP8V3 SIPROLAB_G729SIPROLAB_G729ASIPROLAB_KELVIN VOICEAGE_AMR G726ADPCMDICTAPHONE_CELP68DICTAPHONE_CELP54QUALCOMM_PUREVOICEQUALCOMM_HALFRATETUBGSMMSAUDIO1WMAUDIO2WMAUDIO3WMAUDIO_LOSSLESSWMASPDIFUNISYS_NAP_ADPCMUNISYS_NAP_ULAWUNISYS_NAP_ALAWUNISYS_NAP_16KSYCOM_ACM_SYC008SYCOM_ACM_SYC701_G726LSYCOM_ACM_SYC701_CELP54SYCOM_ACM_SYC701_CELP68KNOWLEDGE_ADVENTURE_ADPCMFRAUNHOFER_IIS_MPEG2_AACDTS_DSCREATIVE_ADPCMCREATIVE_FASTSPEECH8CREATIVE_FASTSPEECH10 UHER_ADPCMULEAD_DV_AUDIOULEAD_DV_AUDIO_1 QUARTERDECKILINK_VC RAW_SPORTESST_AC3GENERIC_PASSTHRUIPI_HSX IPI_RPELPCS2SONY_SCXSONY_SCY SONY_ATRAC3SONY_SPC TELUM_AUDIOTELUM_IA_AUDIONORCOM_VOICE_SYSTEMS_ADPCM FM_TOWNS_SNDMICRONASMICRONAS_CELP833 BTV_DIGITALINTEL_MUSIC_CODER INDEO_AUDIO QDESIGN_MUSIC ON2_VP7_AUDIO ON2_VP6_AUDIO VME_VMPCMTPCLIGHTWAVE_LOSSLESSOLIGSMOLIADPCMOLICELPOLISBCOLIOPRLH_CODEC LH_CODEC_CELP LH_CODEC_SBC8LH_CODEC_SBC12LH_CODEC_SBC16NORRIS ISIAUDIO_2SOUNDSPACE_MUSICOMPRESS MPEG_ADTS_AAC MPEG_RAW_AAC MPEG_LOASNOKIA_MPEG_ADTS_AACNOKIA_MPEG_RAW_AACVODAFONE_MPEG_ADTS_AACVODAFONE_MPEG_RAW_AAC MPEG_HEAACVOXWARE_RT24_SPEECHSONICFOUNDRY_LOSSLESSINNINGS_TELECOM_ADPCMLUCENT_SX8300PLUCENT_SX5363SCUSEEMENTCSOFT_ALF2CM_ACMDVMDTS2MAKEAVISDIVIO_MPEG4_AACNOKIA_ADAPTIVE_MULTIRATE DIVIO_G726 LEAD_SPEECH LEAD_VORBIS WAVPACK_AUDIOOGG_VORBIS_MODE_1OGG_VORBIS_MODE_2OGG_VORBIS_MODE_3OGG_VORBIS_MODE_1_PLUSOGG_VORBIS_MODE_2_PLUSOGG_VORBIS_MODE_3_PLUSALAC _3COM_NBXOPUSFAAD_AACAMR_NBAMR_WBAMR_WP GSM_AMR_CBRGSM_AMR_VBR_SIDCOMVERSE_INFOSYS_G723_1COMVERSE_INFOSYS_AVQSBCCOMVERSE_INFOSYS_SBC SYMBOL_G729_AVOICEAGE_AMR_WBINGENIENT_G726 MPEG4_AAC ENCORE_G726 ZOLL_ASAO SPEEX_VOICE VIANIX_MASCWM9_SPECTRUM_ANALYZERWMF_SPECTRUM_ANAYZERGSM_610GSM_620GSM_660GSM_690GSM_ADAPTIVE_MULTIRATE_WB POLYCOM_G722 POLYCOM_G728POLYCOM_G729_A POLYCOM_SIRENGLOBAL_IP_ILBCRADIOTIME_TIME_SHIFT_RADIONICE_ACA NICE_ADPCM VOCORD_G721 VOCORD_G726 VOCORD_G722_1 VOCORD_G728 VOCORD_G729 VOCORD_G729_A VOCORD_G723_1 VOCORD_LBC NICE_G728FRACE_TELECOM_G729CODIANFLAC EXTENSIBLE DEVELOPMENTr r r r r sr cCsNzt|j}Wn ty|d}Ynwtd|ddddtD)Nz#06xzUnknown wave file format: z. Supported formats: z, css|]}|jVqdS)N)name).0xr r r =sz$_raise_bad_format..)r r ValueErrorjoinKNOWN_WAVE_FORMATS) format_tag format_namer r r _raise_bad_format6s   rcCsl|rd}nd}t|d|dd}|dkrtdt|d|d}d}|\}}}} } } |tjkr}|d kr}t|d |d d} |d 7}| d kry|d } |d 7}| d d }|rcd}nd}||rxt|d|ddd}ntd|tvrt|||kr|||t |||tj kr| || krtd|d| d| ||||| | | 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 Notes ----- Assumes file pointer is immediately after the 'fmt ' id ><Irrrz.Binary structure of wave file is not compliantHHIIHHrHr rrs 8qs 8qNz{WAV header is invalid: nAvgBytesPerSec must equal product of nSamplesPerSec and nBlockAlign, but file has nSamplesPerSec = z, nBlockAlign = z, and nAvgBytesPerSec = ) structunpackrrr rendswithrr_handle_pad_byterw)fid is_big_endianfmtsizeres bytes_readrchannelsfsbytes_per_second block_align bit_depthext_chunk_sizeextensible_chunk_dataraw_guidtailr r r _read_fmt_chunk@sP       rFcCs|rd}nd}|st|d|dd} n|} |dtd|dd} || |d||} | | } |tjkrld |krLdkrQnnd } n8| d vrXd } n1|d krd|d| } n%td|d|tjkr|dvr}|d| } n td|dt ||}|sz| d kr| n| }t j || |d}Wnt j y||dt j|| | d}Ynw| d kr| dkr|dn|d}t jt|| t |jfd d}|r|d| f|ddd| f<n|d| f|dd| df<|||jdd}n$| dvr-|}t j|| d|| fd}||| ntd| dt|| |d krE|d|}|S) a Notes ----- Assumes file pointer is immediately after the 'data' id It's possible to not use all available bits in a container, or to store samples in a container bigger than necessary, so bytes_per_sample uses the actual reported container size (nBlockAlign / nChannels). Real-world examples: Adobe Audition's "24-bit packed int (type 1, 20-bit)" nChannels = 2, nBlockAlign = 6, wBitsPerSample = 20 http://www-mmsp.ece.mcgill.ca/Documents/AudioFormats/WAVE/Samples/AFsp/M1F1-int12-AFsp.wav is: nChannels = 2, nBlockAlign = 4, wBitsPerSample = 12 http://www-mmsp.ece.mcgill.ca/Documents/AudioFormats/WAVE/Docs/multichaudP.pdf gives an example of: nChannels = 2, nBlockAlign = 8, wBitsPerSample = 20 rrrrrrrrrV1r9iz(Unsupported bit depth: the WAV file has z-bit integer data.>r"r9fz-bit floating-point data.)dtypecount)rri4i8N>r r rrc)rmodeoffsetshapezmmap=True not compatible with z-byte container size.)rrrtellseekr rwrryrnpfromfileioUnsupportedOperation frombufferzeroslenritemsizereshapeviewrmemmapr)rrrrris_rf64rmmaprrposbytes_per_sample n_samplesrstartrdatadtar r r _read_data_chunksz               rcCsJ|rd}nd}|d}|r#t||d}||dt||dSdS)N>Itjd|dd|ddtd d ntd t |dkrbd t |} |r^|r^tj| d td d nt| |dkrd}t ||} | dd\} } } | d}| d}nC|dkrt ||n9|dkrd}|stdt || | |||||}n!|dkrt ||n|dvrt ||n tjdtd d t ||||ks!Wt|ds|| |fS|d| |fSt|ds|w|dw)a Open a WAV file. Return the sample rate (in samples/sec) and data from an LPCM WAV file. Parameters ---------- filename : string or open file handle Input WAV file. mmap : bool, optional Whether to read data as memory-mapped (default: False). Not compatible with some bit depths; see Notes. Only to be used on real files. .. 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. Data is 1-D for 1-channel WAV, or 2-D of shape (Nsamples, Nchannels) otherwise. If a file-like input without a C-like file descriptor (e.g., :class:`python:io.BytesIO`) is passed, this will not be writeable. Notes ----- Common data types: [1]_ ===================== =========== =========== ============= WAV format Min Max NumPy dtype ===================== =========== =========== ============= 32-bit floating-point -1.0 +1.0 float32 32-bit integer PCM -2147483648 +2147483647 int32 24-bit integer PCM -2147483648 +2147483392 int32 16-bit integer PCM -32768 +32767 int16 8-bit integer PCM 0 255 uint8 ===================== =========== =========== ============= WAV files can specify arbitrary bit depth, and this function supports reading any integer PCM depth from 1 to 64 bits. Data is returned in the smallest compatible numpy int type, in left-justified format. 8-bit and lower is unsigned, while 9-bit and higher is signed. For example, 24-bit data will be stored as int32, with the MSB of the 24-bit data stored at the MSB of the int32, and typically the least significant byte is 0x00. (However, if a file actually contains data past its specified bit depth, those bits will be read and output, too. [2]_) This bit justification and sign matches WAV's native internal format, which allows memory mapping of WAV files that use 1, 2, 4, or 8 bytes per sample (so 24-bit files cannot be memory-mapped, but 32-bit can). IEEE float PCM in 32- or 64-bit format is supported, with or without mmap. Values exceeding [-1, +1] are not clipped. Non-linear PCM (mu-law, A-law) is not supported. 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 .. [2] Adobe Systems Incorporated, "Adobe Audition 3 User Guide", section "Audio file formats: 24-bit Packed Int (type 1, 20-bit)", 2007 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() rFrbrz%Reached EOF prematurely; finished at dz bytes, expected z bytes from header.r ) stacklevelzUnexpected end of file.zIncomplete chunk ID: z, ignoring it.fmt Tr rrfactdatazNo fmt chunk before datasLIST>FakeJUNKz-Chunk (non-data) not understood, skipping it.r)hasattropenrrrwarningswarnrrrrrrrcloser)filenamerrrrrfmt_chunk_receiveddata_chunk_receivedchunk_idmsg fmt_chunkrrrrrrr r r r8sx f              0   rc Cst|dr|}nt|d}|}zg|jj}gd}|jj|vr(td|jdd}|d7}|d7}|d 7}|d 7}|d krBtj}ntj}|j d krMd } n|j d } |jj d } || d | } | | d } t d|| || | | } |dks}|dks}| d7} |t dt| 7}|| 7}t|dd|j}|d dk}|rd}|d7}|d7}|d 7}|d7}|t dd7}|t dd7}|t d|j7}|t d|j d7}|t dd7}|d 7}|t dt| 7}|| 7}|dks|dks|d7}|t dd|j d7}|||d|t dt|jd|jjdks3|jjdkr7tjd kr7|}t|||}|sT|d|t d|d n|d!|t d|d Wt|dsq|d"S|dd"St|ds|w|dw)#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 >>> import numpy as np >>> 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.astype(np.int16)) rwb)float32float64uint8int16int32int64zUnsupported data type ''rsrrrr rzs2    P h) )