U mdh@sdZddlZddlZddlZddlZddlZddlmZdddgZGddde Z Gdd d eZ e j e j hZd d Zd d ZdddZddZddZddZdddZddZddZdS)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 I/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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 cCsTzt|j}Wntk r*|d}YnXtd|ddddtDdS)Nz#06xzUnknown wave file format: z. Supported formats: z, css|] }|jVqdS)N)name).0xr r r =sz$_raise_bad_format..)r rz ValueErrorjoinKNOWN_WAVE_FORMATS) format_tag format_namer r r _raise_bad_format6s rcCst|r d}nd}t|d|dd}|dkr8tdt|d|d}d}|\}}}} } } |tjkr|d krt|d |d d} |d 7}| d kr|d } |d 7}| d d }|rd}nd}||rt|d|ddd}ntd|tkr t|||kr$|||t |||tj krb| || krbtd|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 compliantZHHIIHHrHr rrs 8qs 8qNz{WAV header is invalid: nAvgBytesPerSec must equal product of nSamplesPerSec and nBlockAlign, but file has nSamplesPerSec = z, nBlockAlign = z, and nAvgBytesPerSec = ) structunpackrr~r ryendswithrr_handle_pad_byterw)fid is_big_endianfmtsizeresZ bytes_readrchannelsfsbytes_per_second block_align bit_depthZext_chunk_sizeZextensible_chunk_dataZraw_guidtailr r r _read_fmt_chunk@sB        rFcCsX|r d}nd}t|d|dd}||} || } |tjkrd|krVdkr`nnd} q| d krnd } q|d kr|d | } qtd |dn<|tjkr|dkr|d| } qtd |dnt||} |sz$| d kr|n| } t j || | d}Wn8t j k r>| | dt j||| d}YnX| d kr4| dkr^|dn|d}t jt|| t |jfd d}|r|d| f|ddd| f<n |d| f|dd| df<|||jdd}nH| dkr$|} t j|| d| | fd}| | |ntd| dt|||dkrT|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 rrrrrr ru1>rrrrZV1r9iz(Unsupported bit depth: the WAV file has z-bit integer data.>r"r9fz-bit floating-point data.)dtypecount)rri4i8N>rr r rc)rmodeoffsetshapezmmap=True not compatible with z-byte container size.)rrrr rwr~rxrtellnumpyfromfileioUnsupportedOperationseekZ frombufferZzeroslenritemsizeZreshapeviewrZmemmapr)rrrrrrmmaprrZbytes_per_sampleZ n_samplesrstartrdatadtar r r _read_data_chunks^           rcCsF|r d}nd}|d}|rBt||d}||dt||dS)N>I>> 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() rFrbrrzQReached EOF prematurely; finished at {:d} bytes, expected {:d} 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.)hasattropencloserrrrwarningswarnformatrr~rrrrr)filenamerrrrZfmt_chunk_receivedZdata_chunk_receivedZchunk_idmsgZ fmt_chunkrrrrrrr r r rsvf                    c Cs6t|dr|}n t|d}|}z|jj}|dks\|dks\|dkrN|jjdks\td|jd }|d 7}|d 7}|d 7}|d 7}|dkrtj }ntj }|j dkrd}n |j d}|jjd} || d|} || d} t d|||| | | } |dks |dks | d7} |t dt| 7}|| 7}|dksZ|dksZ|d7}|t dd|j d7}t|ddd|jdkrtd|||d|t d|j|jjdks|jjdkrtjdkr|}t|||} |d|t d| dW5t|ds&|n |dXdS)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)) rwbrrrur zUnsupported data type '%s'rsrrrzs6    P _ *