a DOgTF@sddlmZddlmZddlZddlZddlZddlZgdZGddde Z dZ dZ ed d Z Gd d d ZGd ddZdddZdS))Chunk) namedtupleN)openError Wave_read Wave_writec@s eZdZdS)rN)__name__ __module__ __qualname__r r )/opt/alt/python39/lib64/python3.9/wave.pyrTsr)NbhNi _wave_paramsz7nchannels sampwidth framerate nframes comptype compnamec@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+S),rcCsd|_d|_t|dd|_|jdkr0td|jddkrHtdd|_d|_d|_ zt|jdd}Wnt yYqYn0|}|d kr| |d|_n2|d kr|jstd ||_|j |j |_d|_ q|qT|jr|jstd dS) Nr)Z bigendianRIFFz file does not start with RIFF idWAVEznot a WAVE filer fmt datazdata chunk before fmt chunkz#fmt chunk and/or data chunk missing)_convert _soundposr_fileZgetnamerreadZ_fmt_chunk_read _data_chunk_data_seek_neededEOFError_read_fmt_chunkZ chunksize _framesize_nframesskip)selffilechunkZ chunknamer r r initfp}s8     zWave_read.initfpcCsRd|_t|tr"t|d}||_z||Wn|jrF|Yn0dS)Nrb_i_opened_the_file isinstancestrbuiltinsrr%closer"fr r r __init__s  zWave_read.__init__cCs |dSNr,r"r r r __del__szWave_read.__del__cCs|Sr0r r2r r r __enter__szWave_read.__enter__cGs |dSr0r1r"argsr r r __exit__szWave_read.__exit__cCs|jSr0)rr2r r r getfpszWave_read.getfpcCsd|_d|_dS)Nr r)rrr2r r r rewindszWave_read.rewindcCs"d|_|j}|rd|_|dSr0)rr(r,r"r#r r r r,s zWave_read.closecCs|jSr0)rr2r r r tellszWave_read.tellcCs|jSr0) _nchannelsr2r r r getnchannelsszWave_read.getnchannelscCs|jSr0)r r2r r r getnframesszWave_read.getnframescCs|jSr0) _sampwidthr2r r r getsampwidthszWave_read.getsampwidthcCs|jSr0) _framerater2r r r getframerateszWave_read.getframeratecCs|jSr0 _comptyper2r r r getcomptypeszWave_read.getcomptypecCs|jSr0 _compnamer2r r r getcompnameszWave_read.getcompnamecCs*t||||||Sr0)rr=r@rBr>rErHr2r r r getparamss  zWave_read.getparamscCsdSr0r r2r r r getmarkersszWave_read.getmarkerscCs tddSNzno marksrr"idr r r getmarkszWave_read.getmarkcCs*|dks||jkrtd||_d|_dS)Nrzposition not in ranger )r rrr)r"posr r r setposszWave_read.setposcCs|jr8|jdd|j|j}|r2|j|dd|_|dkrDdS|j||j}|jdkrxtjdkrxt ||j}|j r|r| |}|jt ||j |j|_|S)Nrr big)rrseekrrrr?sys byteorderaudioopbyteswaprlenr<)r"nframesrPdatar r r readframess   zWave_read.readframescCsz$td|d\}|_|_}}Wntjy>tdYn0|tkrztd|dd}Wntjy|tdYn0|dd|_|jst dnt d |f|jst d |j|j|_ d |_ d |_ dS) Nzr@rBrErHrIrJrOrQr\rr r r r r^s* rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9S):rcCsRd|_t|tr"t|d}||_z||Wn|jrF|Yn0dS)Nwbr'r-r r r r/-s  zWave_write.__init__cCs@||_d|_d|_d|_d|_d|_d|_d|_d|_d|_ dS)NrF) rrr<r?rAr _nframeswritten _datawritten _datalength_headerwrittenr:r r r r%9szWave_write.initfpcCs |dSr0r1r2r r r r3EszWave_write.__del__cCs|Sr0r r2r r r r4HszWave_write.__enter__cGs |dSr0r1r5r r r r7KszWave_write.__exit__cCs(|jrtd|dkrtd||_dS)N0cannot change parameters after starting to writer rb)rjrr<)r" nchannelsr r r setnchannelsQs zWave_write.setnchannelscCs|jstd|jS)Nznumber of channels not set)r<rr2r r r r=XszWave_write.getnchannelscCs0|jrtd|dks|dkr&td||_dS)Nrmr rra)rjrr?)r"rgr r r setsampwidth]s zWave_write.setsampwidthcCs|jstd|jS)Nzsample width not set)r?rr2r r r r@dszWave_write.getsampwidthcCs0|jrtd|dkrtdtt||_dS)Nrmrzbad frame rate)rjrintroundrA)r" framerater r r setframerateis zWave_write.setframeratecCs|jstd|jS)Nzframe rate not set)rArr2r r r rBpszWave_write.getframeratecCs|jrtd||_dSNrm)rjrr )r"rZr r r setnframesuszWave_write.setnframescCs|jSr0rir2r r r r>zszWave_write.getnframescCs.|jrtd|dvrtd||_||_dS)Nrm)rczunsupported compression type)rjrrDrG)r"comptypecompnamer r r setcomptype}s zWave_write.setcomptypecCs|jSr0rCr2r r r rEszWave_write.getcomptypecCs|jSr0rFr2r r r rHszWave_write.getcompnamecCsV|\}}}}}}|jrtd|||||||||||dSru)rjrrorprtrvrz)r"paramsrnrgrsrZrxryr r r setparamss    zWave_write.setparamscCs8|jr|jr|jstdt|j|j|j|j|j|jS)Nznot all parameters set)r<r?rArrr rDrGr2r r r rIs  zWave_write.getparamscCs tddS)Nzsetmark() not supportedrL)r"rNrPnamer r r setmarkszWave_write.setmarkcCs tddSrKrLrMr r r rOszWave_write.getmarkcCsdSr0r r2r r r rJszWave_write.getmarkerscCs|jSr0rwr2r r r r;szWave_write.tellcCst|ttfst|d}|t|t||j|j}|j rN| |}|jdkrpt j dkrpt ||j}|j||jt|7_|j||_dS)NBr rS)r)bytes bytearray memoryviewcast_ensure_header_writtenrYr?r<rrUrVrWrXrwriterjri)r"r[rZr r r writeframesraws  zWave_write.writeframesrawcCs"|||j|jkr|dSr0)rrkrj _patchheader)r"r[r r r writeframess  zWave_write.writeframescCsvzP|jr0|d|j|jkr&||jWd|_|j}|rrd|_|n d|_|j}|rpd|_|0dS)Nr)rrrkrjrflushr(r,r:r r r r,s     zWave_write.closecCs>|js:|jstd|js"td|js0td||dS)Nz# channels not specifiedzsample width not specifiedzsampling rate not specified)rlr<rr?rA _write_header)r"Zdatasizer r r rsz!Wave_write._ensure_header_writtencCs|jd|js$||j|j|_|j|j|j|_z|j|_Wntt fydd|_Yn0|jt dd|jdddt |j|j |j|j |j|j|j|jdd |jdur|j|_|jt d |jd |_dS) NrzrzrErHr|rIr~rOrJr;rrr,rrrr r r r rs8     rcCsJ|durt|dr|j}nd}|dvr.t|S|dvr>t|StddS)Nmoder&)rr&)wrhz$mode must be 'r', 'rb', 'w', or 'wb')hasattrrrrr)r.rr r r rs r)N)r$r collectionsrrWr+rdrU__all__ ExceptionrrfZ _array_fmtsrrrrr r r r Js"  6d