a DOg] @s^ddlZddlZddlZddgZGdddeZdZddZdd Zd d Z d d Z ddZ dZ ddZ ddZddZddZddZddZddZddlmZdd lmZed!d"Zd#ej_d$ej_d%ej_d&ej_d'ej_d(ej_Gd)d*d*Z Gd+d,d,Z!d>d-dZ"e#d.krZddl$Z$e$j%d/ds>e$j%&d0e$j%d/Z'e"e'd1Z(e)d2e'e)d3e(*e)d4e(+e)d5e(,e)d6e(-e)d7e(.e)d8e(/e$j%d9dr:e$j%d9Z0e)d:e0e"e0d;@Z1e12e(3e(4d<Z5e5sqe16e5qWdn1s(0Ye)d=Wdn1sP0YdS)?NErroropenc@s eZdZdS)rN)__name__ __module__ __qualname__rr/usr/lib64/python3.9/aifc.pyrsl@QEcCs8ztd|ddWStjy2tdYn0dS)N>lrstructunpackreaderrorEOFErrorfilerrr _read_longsrcCs8ztd|ddWStjy2tdYn0dS)N>Lr rr rrrr _read_ulongsrcCs8ztd|ddWStjy2tdYn0dS)N>hrr rrrr _read_shortsrcCs8ztd|ddWStjy2tdYn0dS)N>Hrrr rrrr _read_ushortsrcCs@t|d}|dkrd}n ||}|d@dkr<|d}|S)Nr)ordr)rlengthdatadummyrrr _read_strings   r!gcCst|}d}|dkr d}|d}t|}t|}||krN|krNdkrXnnd}n0|dkrft}n"|d}|d|td |d }||S) Nrrgi?lg@?)rr _HUGE_VALpow)fexponsignhimantlomantrrr _read_floats"r-cCs|td|dS)Nrwriter packr(xrrr _write_shortsr3cCs|td|dS)Nrr.r1rrr _write_ushortsr4cCs|td|dS)Nr r.r1rrr _write_longsr5cCs|td|dS)Nrr.r1rrr _write_ulongsr6cCsRt|dkrtd|tdt|||t|d@dkrN|ddS)Nz%string exceeds maximum pstring lengthBrr)len ValueErrorr/r r0)r(srrr _write_strings   r=c Csddl}|dkrd}|d}nd}|dkr8d}d}d}n||\}}|dks^|dks^||krp|dB}d}d}nh|d}|dkr|||}d}||B}||d}||}t|}|||d}||}t|}t||t||t||dS) Nrr#r"i@rr$i? )mathfrexpldexpfloorintr4r6) r(r2r?r*r)r+r,ZfmantZfsmantrrr _write_floats8       rD)Chunk) namedtuple _aifc_paramsz7nchannels sampwidth framerate nframes comptype compnamez3Number of audio channels (1 for mono, 2 for stereo)zSample width in byteszSampling frequencyzNumber of audio framesz(Compression type ("NONE" for AIFF files)zRA human-readable version of the compression type ('not compressed' for AIFF files)c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZdS)2 Aifc_readNcCs6d|_d|_g|_d|_||_t|}|dkr:td|d}|dkrTd|_ n|dkrdd|_ ntdd|_ d|_ d|_ zt|j}Wnt yYqYn0|}|d kr||d|_ nH|d kr||_ |d }d|_ n(|d krt||_n|d kr|||qx|j r*|j s2tddS)NrFORMz file does not start with FORM idr AIFFAIFCrznot an AIFF or AIFF-C fileCOMMSSNDFVERMARKz$COMM chunk and/or SSND chunk missing)_version_convert_markers _soundpos_filerEZgetnamerr_aifcZ_comm_chunk_read _ssnd_chunk_ssnd_seek_neededr_read_comm_chunkr _readmarkskip)selfrchunkZformdataZ chunknamer rrrinitfp4sH          zAifc_read.initfpcCsLt|tr>t|d}z||WqH|YqH0n ||dS)Nrb) isinstancestrbuiltinsrr^closer\r(Z file_objectrrr__init__\s   zAifc_read.__init__cCs|SNrr\rrr __enter__hszAifc_read.__enter__cGs |dSrfrcr\argsrrr__exit__kszAifc_read.__exit__cCs|jSrf)rUrgrrrgetfpqszAifc_read.getfpcCsd|_d|_dS)Nrr)rXrTrgrrrrewindtszAifc_read.rewindcCs |j}|durd|_|dSrf)rUrcr\rrrrrcxszAifc_read.closecCs|jSrf)rTrgrrrtell~szAifc_read.tellcCs|jSrf) _nchannelsrgrrr getnchannelsszAifc_read.getnchannelscCs|jSrf)_nframesrgrrr getnframesszAifc_read.getnframescCs|jSrf) _sampwidthrgrrr getsampwidthszAifc_read.getsampwidthcCs|jSrf) _frameratergrrr getframerateszAifc_read.getframeratecCs|jSrf _comptypergrrr getcomptypeszAifc_read.getcomptypecCs|jSrf _compnamergrrr getcompnameszAifc_read.getcompnamecCs*t||||||Srf)rGrrrvrxrtr{r~rgrrr getparamss  zAifc_read.getparamscCst|jdkrdS|jSNrr:rSrgrrr getmarkersszAifc_read.getmarkerscCs2|jD]}||dkr|Sqtd|dSNrzmarker {0!r} does not existrSrformatr\idmarkerrrrgetmarks   zAifc_read.getmarkcCs*|dks||jkrtd||_d|_dS)Nrzposition not in ranger)rsrrTrX)r\posrrrsetposszAifc_read.setposcCs|jrD|jd|jd}|j|j}|r>|j|dd|_|dkrPdS|j||j}|jrv|rv||}|jt||j|j |_|S)NrrNr) rXrWseekrrT _framesizerRr:rqru)r\nframesr rrrrr readframess       zAifc_read.readframescCsddl}||dSNrr)audioopZalaw2linr\rrrrr _alaw2linszAifc_read._alaw2lincCsddl}||dSr)rZulaw2linrrrr _ulaw2linszAifc_read._ulaw2lincCs2ddl}t|dsd|_||d|j\}|_|SNr _adpcmstater)rhasattrrZ adpcm2linrrrr _adpcm2lins  zAifc_read._adpcm2lincCsVt||_t||_t|dd|_tt||_|jdkrFtd|jdkrXtd|j|j|_ |j rFd}|j dkrd}t dd |_ |d |_|rt|jd}|d@dkr|d}|j ||_ |jd dt||_|jd krR|jd kr |j|_n4|jdvr |j|_n|jdvr6|j|_ntdd|_n d |_d|_dS)NrNrbad sample widthbad # of channelsrzWarning: bad COMM chunk sizer r"NONEG722ulawULAWalawALAWunsupported compression typernot compressed)rrqrrsrurCr-rwrrrV chunksizewarningswarnrrzrrrr!r}rrRrr)r\r]ZkludgerrrrrYsD                 zAifc_read._read_comm_chunkcCst|}zDt|D]6}t|}t|}t|}|s6|r|j|||fqWnBtydt|jt|jdkrvdnd|f}t |Yn0dS)Nz;Warning: MARK chunk contains only %s marker%s instead of %srr<) rrangerr!rSappendrr:rr)r\r]ZnmarkersirrnamewrrrrZs  zAifc_read._readmark)rrrrUr^rerhrlrmrnrcrprrrtrvrxr{r~rrrrrrrrrYrZrrrrrHs2$( *rHc@s0eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z dd?Z"d@dAZ#dBdCZ$dDdEZ%dFdGZ&dHdIZ'dS)J Aifc_writeNcCs\t|trNt|d}z||Wn|Yn0|drXd|_n ||dS)Nwbz.aiffr)r`rarbrr^rcendswithrVrdrrrre/s   zAifc_write.__init__cCs^||_t|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ g|_ d|_d|_dS)Nrrrr)rU _AIFC_versionrQrzr}rRrqrurwrs_nframeswritten _datawritten _datalengthrS _marklengthrVrorrrr^?szAifc_write.initfpcCs |dSrfrirgrrr__del__PszAifc_write.__del__cCs|SrfrrgrrrrhSszAifc_write.__enter__cGs |dSrfrirjrrrrlVszAifc_write.__exit__cCs|jrtdd|_dS)N0cannot change parameters after starting to writerrrrVrgrrraiff\szAifc_write.aiffcCs|jrtdd|_dS)NrrrrgrrraifcaszAifc_write.aifccCs(|jrtd|dkrtd||_dS)Nrrr)rrrq)r\ nchannelsrrr setnchannelsfs zAifc_write.setnchannelscCs|jstd|jS)Nznumber of channels not set)rqrrgrrrrrmszAifc_write.getnchannelscCs0|jrtd|dks|dkr&td||_dS)Nrrr r)rrru)r\ sampwidthrrr setsampwidthrs zAifc_write.setsampwidthcCs|jstd|jS)Nzsample width not set)rurrgrrrrvyszAifc_write.getsampwidthcCs(|jrtd|dkrtd||_dS)Nrrzbad frame rate)rrrw)r\ frameraterrr setframerate~s zAifc_write.setframeratecCs|jstd|jS)Nzframe rate not set)rwrrgrrrrxszAifc_write.getframeratecCs|jrtd||_dS)Nr)rrrs)r\rrrr setnframesszAifc_write.setnframescCs|jSrfrrgrrrrtszAifc_write.getnframescCs.|jrtd|dvrtd||_||_dSNr)rrrrrrr)rrrzr})r\comptypecompnamerrr setcomptypes zAifc_write.setcomptypecCs|jSrfryrgrrrr{szAifc_write.getcomptypecCs|jSrfr|rgrrrr~szAifc_write.getcompnamecCsf|\}}}}}}|jrtd|dvr.td|||||||||||dSr)rrrrrrr)r\paramsrrrrrrrrr setparamss    zAifc_write.setparamscCs8|jr|jr|jstdt|j|j|j|j|j|jS)Nznot all parameters set)rqrurwrrGrsrzr}rgrrrrs  zAifc_write.getparamscCs|dkrtd|dkr tdt|ts2tdtt|jD],}||j|dkr@|||f|j|<dSq@|j|||fdS)Nrzmarker ID must be > 0zmarker position must be >= 0zmarker name must be bytes)rr`bytesrr:rSr)r\rrrrrrrsetmarks zAifc_write.setmarkcCs2|jD]}||dkr|Sqtd|dSrrrrrrrs   zAifc_write.getmarkcCst|jdkrdS|jSrrrgrrrrszAifc_write.getmarkerscCs|jSrfrrgrrrrpszAifc_write.tellcCszt|ttfst|d}|t|t||j|j}|j rN| |}|j ||j ||_ |j t||_ dS)Nr8)r`r bytearray memoryviewcast_ensure_header_writtenr:rurqrRrUr/rr)r\rrrrrwriteframesraws   zAifc_write.writeframesrawcCs.|||j|jks"|j|jkr*|dSrf)rrrsrr _patchheader)r\rrrr writeframess    zAifc_write.writeframescCs|jdurdSzx|d|jd@r<|jd|jd|_||j|jksb|j|jksb|jrj| Wd|_ |j}d|_| nd|_ |j}d|_| 0dS)Nrrr9) rUrrr/ _writemarkersrrsrrrrRrc)r\r(rrrrcs,         zAifc_write.closecCsddl}||dSr)rZlin2alawrrrr _lin2alawszAifc_write._lin2alawcCsddl}||dSr)rZlin2ulawrrrr _lin2ulawszAifc_write._lin2ulawcCs2ddl}t|dsd|_||d|j\}|_|Sr)rrrZ lin2adpcmrrrr _lin2adpcms  zAifc_write._lin2adpcmcCsf|jsb|jdvr.|jsd|_|jdkr.td|js|jdkr|j|_n&|jdvr(|j|_n|jdvr:|j|_dS)Nrrr)rzrrRrrrgrrr_init_compressions      zAifc_write._init_compressionc CsH|jr|jdkr||jd|js<||j|j|_|j|j|j|_|jd@rf|jd|_|jr|jdvr|jd|_|jd@r|jd|_n0|jdkr|jdd|_|jd@r|jd|_z|j |_ Wnt t fyd|_ Yn0| |j}|jr@|jd |jd t|jdt|j|jn |jd |jd t|j|t|j|j|j dur|j |_t|j|j|jd vrt|jdnt|j|jdt|j|j|jr|j|jt|j|j|jd|j dur|j |_t|j|jdt|jdt|jddS)NrrIr)rrrrrrr rKrOrJrLrrNrMr)rVrzrrUr/rsrqrurrp_form_length_posAttributeErrorOSError_write_form_lengthr6rQr3 _nframes_posrDrwr=r}_ssnd_length_pos)r\Z initlength commlengthrrrr%s^                        zAifc_write._write_headercCs\|jr*dt|j}|d@r$|d}d}nd}d}t|jd||jd|d||S) Nrr rrr rN)rVr:r}r6rUr)r\ datalengthrZ verslengthrrrrXs"zAifc_write._write_form_lengthcCs|j}|jd@r,|jd}|jdn|j}||jkrd|j|jkrd|jdkrd|j|ddS|j|j d| |}|j|j dt |j|j|j|j dt |j|d|j|d|j|_||_dS)Nrr9rrN)rUrprr/rrsrrrrrrr6r)r\Zcurposrr rrrres*      zAifc_write._patchheadercCst|jdkrdS|jdd}|jD]:}|\}}}|t|dd}t|d@dkr(|d}q(t|j||d|_t|jt|j|jD]2}|\}}}t|j|t|j|t|j|qdS)NrrPrrrN)r:rSrUr/r6rr3r=)r\rrrrrrrrr{s"          zAifc_write._writemarkers)(rrrrUrer^rrhrlrrrrrrrvrrxrrtrr{r~rrrrrrprrrcrrrrrrrrrrrrrrsJ    3 rcCsJ|durt|dr|j}nd}|dvr.t|S|dvr>t|StddS)Nmoder_)rr_)rrz$mode must be 'r', 'rb', 'w', or 'wb')rrrHrr)r(rrrrrs __main__rz/usr/demos/data/audio/bach.aiffrZReadingz nchannels =z nframes =z sampwidth =z framerate =z comptype =z compname =rZWritingrizDone.)N)7r rbr__all__ Exceptionrrrrrrr!r&r-r3r4r5r6r=rDr]rE collectionsrFrGr__doc__rrrrrrHrrrsysargvrfnr(printrrrtrvrxr{r~Zgngrrrrrrrrrsx !            .