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 /usr/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_chunk 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|Sr1r r3r r r __enter__szWave_read.__enter__cGs |dSr1r2r#argsr r r __exit__szWave_read.__exit__cCs|jSr1)rr3r r r getfpszWave_read.getfpcCsd|_d|_dS)Nr r)rrr3r r r rewindszWave_read.rewindcCs"d|_|j}|rd|_|dSr1)rr)r-r#r$r r r r-s zWave_read.closecCs|jSr1)rr3r r r tellszWave_read.tellcCs|jSr1) _nchannelsr3r r r getnchannelsszWave_read.getnchannelscCs|jSr1)r!r3r r r getnframesszWave_read.getnframescCs|jSr1) _sampwidthr3r r r getsampwidthszWave_read.getsampwidthcCs|jSr1) _framerater3r r r getframerateszWave_read.getframeratecCs|jSr1 _comptyper3r r r getcomptypeszWave_read.getcomptypecCs|jSr1 _compnamer3r r r getcompnameszWave_read.getcompnamecCs*t||||||Sr1)rr>rArCr?rFrIr3r r r getparamss  zWave_read.getparamscCsdSr1r r3r 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)rrseekrr rr@sys byteorderaudioopbyteswaprlenr=)r#nframesrQdatar 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?rArCrFrIrJrKrPrRr]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 r0-s  zWave_write.__init__cCs@||_d|_d|_d|_d|_d|_d|_d|_d|_d|_ dS)NrF) rrr=r@rBr!_nframeswritten _datawritten _datalength_headerwrittenr;r r r r&9szWave_write.initfpcCs |dSr1r2r3r r r r4EszWave_write.__del__cCs|Sr1r r3r r r r5HszWave_write.__enter__cGs |dSr1r2r6r r r r8KszWave_write.__exit__cCs(|jrtd|dkrtd||_dS)N0cannot change parameters after starting to writer rc)rlrr=)r# nchannelsr r r setnchannelsQs zWave_write.setnchannelscCs|jstd|jS)Nznumber of channels not set)r=rr3r r r r>XszWave_write.getnchannelscCs0|jrtd|dks|dkr&td||_dS)Nror rrb)rlrr@)r#rir r r setsampwidth]s zWave_write.setsampwidthcCs|jstd|jS)Nzsample width not set)r@rr3r r r rAdszWave_write.getsampwidthcCs0|jrtd|dkrtdtt||_dS)Nrorzbad frame rate)rlrintroundrB)r# framerater r r setframerateis zWave_write.setframeratecCs|jstd|jS)Nzframe rate not set)rBrr3r r r rCpszWave_write.getframeratecCs|jrtd||_dSNro)rlrr!)r#r[r r r setnframesuszWave_write.setnframescCs|jSr1rkr3r r r r?zszWave_write.getnframescCs.|jrtd|dvrtd||_||_dS)Nro)rdzunsupported compression type)rlrrErH)r#comptypecompnamer r r setcomptype}s zWave_write.setcomptypecCs|jSr1rDr3r r r rFszWave_write.getcomptypecCs|jSr1rGr3r r r rIszWave_write.getcompnamecCsV|\}}}}}}|jrtd|||||||||||dSrw)rlrrqrrrvrxr|)r#paramsrprirur[rzr{r r r setparamss    zWave_write.setparamscCs8|jr|jr|jstdt|j|j|j|j|j|jS)Nznot all parameters set)r=r@rBrrr!rErHr3r r r rJs  zWave_write.getparamscCs tddS)Nzsetmark() not supportedrM)r#rOrQnamer r r setmarkszWave_write.setmarkcCs tddSrLrMrNr r r rPszWave_write.getmarkcCsdSr1r r3r r r rKszWave_write.getmarkerscCs|jSr1ryr3r r r r<szWave_write.tellcCst|ttfst|d}|t|t||j|j}|j rN| |}|jdkrpt j dkrpt ||j}|j||jt|7_|j||_dS)NBr rT)r*bytes bytearray memoryviewcast_ensure_header_writtenrZr@r=rrVrWrXrYrwriterlrk)r#r\r[r r r writeframesraws  zWave_write.writeframesrawcCs"|||j|jkr|dSr1)rrmrl _patchheader)r#r\r r r writeframess  zWave_write.writeframescCsvzP|jr0|d|j|jkr&||jWd|_|j}|rrd|_|n d|_|j}|rpd|_|0dS)Nr)rrrmrlrflushr)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)rnr=rr@rB _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) NrzrrrArvrCrxr?r|rFrIr~rJrrPrKr<rrr-rrrr r r r rs8     rcCsJ|durt|dr|j}nd}|dvr.t|S|dvr>t|StddS)Nmoder')rr')wrjz$mode must be 'r', 'rb', 'w', or 'wb')hasattrrrrr)r/rr r r rs r)N)r%r collectionsrrXr,rerV__all__ ExceptionrrhZ _array_fmtsrrrrr r r r Js"  6d