o wi4[@sdZ ddlmZddlZddlZddlZgdZGdddeZdZ dZ d Z d Z ed d Z d dZGdddZGdddZGdddZdddZdS)zGCopiped from https://github.com/python/cpython/blob/v3.13.2/Lib/wave.py) namedtupleN)openError Wave_read Wave_writec@s eZdZdS)rN)__name__ __module__ __qualname__r r P/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/kaldiio/python_wave.pyrTsris8q)NbhNi _wave_paramsz7nchannels sampwidth framerate nframes comptype compnamecCsTtt|}tdt||D]}t|D]}||||||d|<qqt|S)Nrr ) bytearraylenrangebytes)datawidth swapped_datarjr r r _byteswapbs  rc@sJeZdZdddZddZddZdd d Zd d ZdddZddZ dS)_ChunkTFc Csd|_||_|r d}nd}||_|d|_t|jdkrtzt|d|dd|_ Wn tj y:tdw|rC|j d|_ d|_ z|j |_ Wnttfy]d|_YdSwd|_dS) NF><LrT)closedalignfileread chunknamerEOFErrorstruct unpack_from chunksizeerror size_readtelloffsetAttributeErrorOSErrorseekable)selfr"r! bigendian inclheaderstrflagr r r __init__ms.     z_Chunk.__init__cC|jS)z*Return the name (ID) of the current chunk.)r$r0r r r getnamesz_Chunk.getnamecCs(|jsz |Wd|_dSd|_wdS)NT)r skipr6r r r closes  z _Chunk.closercCsv|jrtd|jstd|dkr||j}n |dkr!||j}|dks*||jkr,t|j|j |d||_dS)zSeek to specified position into the chunk. Default position is 0 (start of chunk). If the file is not seekable, this will result in an error. I/O operation on closed filez cannot seekr rN) r ValueErrorr/r.r*r( RuntimeErrorr"seekr,)r0poswhencer r r r>s   z _Chunk.seekcCs|jrtd|jS)Nr:)r r<r*r6r r r r+z _Chunk.tellcCs|jrtd|j|jkrdS|dkr|j|j}||j|jkr'|j|j}|j|}|jt||_|j|jkrQ|jrQ|jd@rQ|jd}|jt||_|S)zRead at most size bytes from the chunk. If size is omitted or negative, read until the end of the chunk. r:rr )r r<r*r(r"r#rr!)r0sizerdummyr r r r#s$      z _Chunk.readcCs|jrtd|jr6z"|j|j}|jr|jd@r|d}|j|d|j||_WdSty5Ynw|j|jkrVt d|j|j}| |}|sNt |j|jksr.minr#r%)r0nrEr r r r8s&     z _Chunk.skipN)TTF)r)rB) rrr r4r7r9r>r+r#r8r r r r rls    rc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,S)-raPVariables used in this class: These variables are available to the user though appropriate methods of this class: _file -- the open file with methods read(), close(), and seek() set through the __init__() method _nchannels -- the number of audio channels available through the getnchannels() method _nframes -- the number of audio frames available through the getnframes() method _sampwidth -- the number of bytes per audio sample available through the getsampwidth() method _framerate -- the sampling frequency available through the getframerate() method _comptype -- the AIFF-C compression type ('NONE' if AIFF) available through the getcomptype() method _compname -- the human-readable AIFF-C compression type available through the getcomptype() method _soundpos -- the position in the audio stream available through the tell() method, set through the setpos() method These variables are used internally only: _fmt_chunk_read -- 1 iff the FMT chunk has been read _data_seek_needed -- 1 iff positioned correctly in audio file for readframes() _data_chunk -- instantiation of a chunk class for the DATA chunk _framesize -- size of one frame in the file cCsd|_d|_t|dd|_|jdkrtd|jddkr$tdd|_d|_ d|_ z t|jdd}Wn t y@Yn0w|}|d krR| |d|_n|d krk|js]td ||_|j |j |_d|_ n|q+|jrv|jsztd dS) Nr)r1RIFFz file does not start with RIFF idrWAVEznot a WAVE filer fmt datazdata chunk before fmt chunkz#fmt chunk and/or data chunk missing)_convert _soundposr_filer7rr#_fmt_chunk_read _data_chunk_data_seek_neededr%_read_fmt_chunkr( _framesize_nframesr8)r0r"chunkr$r r r initfps@   zWave_read.initfpcCJd|_t|trt|d}||_z||WdS|jr$|)Nrb_i_opened_the_file isinstancestrbuiltinsrrVr9r0fr r r r4s  zWave_read.__init__cC |dSNr9r6r r r __del__& zWave_read.__del__cC|Srar r6r r r __enter__)zWave_read.__enter__cGr`rarbr0argsr r r __exit__,rdzWave_read.__exit__cCr5ra)rNr6r r r getfp2zWave_read.getfpcCsd|_d|_dS)Nr r)rQrMr6r r r rewind5s zWave_read.rewindcCs&d|_|j}|rd|_|dSdSra)rNrZr9r0r"r r r r99s  zWave_read.closecCr5ra)rMr6r r r r+@rlzWave_read.tellcCr5ra) _nchannelsr6r r r getnchannelsCrlzWave_read.getnchannelscCr5ra)rTr6r r r getnframesFrlzWave_read.getnframescCr5ra) _sampwidthr6r r r getsampwidthIrlzWave_read.getsampwidthcCr5ra) _framerater6r r r getframerateLrlzWave_read.getframeratecCr5ra _comptyper6r r r getcomptypeOrlzWave_read.getcomptypecCr5ra _compnamer6r r r getcompnameRrlzWave_read.getcompnamecCs*t||||||Sra)rrprsrurqrxr{r6r r r getparamsUs  zWave_read.getparamscCddl}|jddddS)NrWave_read.getmarkersremovewarnings _deprecatedr0rr r r getmarkersZr~cCddl}|jdddtd)NrWave_read.getmarkrrno marksrrrr0idrr r r getmark_rcCs*|dks ||jkr td||_d|_dS)Nrzposition not in ranger )rTrrMrQ)r0r?r r r setposds zWave_read.setposcCs|jr|jdd|j|j}|r|j|dd|_|dkr"dS|j||j}|jdkr;tjdkr;t ||j}|j rE|rE| |}|jt ||j |j|_|S)NrrCr big) rQrPr>rMrSr#rrsys byteorderrrLrro)r0nframesr?rr r r readframesjs    zWave_read.readframesc CsVztd|d\}|_|_}}Wn tjytdwz td|dd}Wn tjy6tdw|tkrztd|d\}}}|d} t| dkrUtWn tjyatdw| t krzddl } d | j | d } Wt | t yd } Yt | w|d d|_|jst d |jst d|j|j|_d|_d|_dS)Nzrrr&rr)r0curposr r r rs    zWave_write._patchheaderN) rrr rr4rVrcrfrjrrprrsrrurrqrrxr{rr|rrrr+rrr9rrrr r r r rs<     rcCsF|durt|dr |j}nd}|dvrt|S|dvrt|Std)NmoderX)rrX)wrz$mode must be 'r', 'rb', 'w', or 'wb')hasattrrrrr)r_rr r r rs rra)r collectionsrr]r&r__all__rrrrr _array_fmtsrrrrrrr r r r s, I oKk