a
XC?hTF @ s d dl mZ d dlmZ d dlZd dlZd dlZd dlZg dZG dd de Z
dZ dZ ed d
Z
G d d d ZG d
d dZdddZdS ) )Chunk)
namedtupleN)openError Wave_read
Wave_writec @ s e Zd ZdS )r N)__name__
__module__ __qualname__ r r )/opt/alt/python39/lib64/python3.9/wave.pyr T s r )NbhNi _wave_paramsz7nchannels sampwidth framerate nframes comptype compnamec @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+S ),r c C s d | _ d| _t|dd| _| j dkr0td| jddkrHtdd| _d | _d| _ zt| jdd}W n t
y Y qY n0 | }|d kr| | d| _n2|d
kr| jstd || _|j | j
| _d| _ q| qT| jr| jstd d S )
Nr )Z bigendian RIFFz file does not start with RIFF id WAVEznot a WAVE filer
fmt datazdata chunk before fmt chunkz#fmt chunk and/or data chunk missing)_convert _soundposr _fileZgetnamer readZ_fmt_chunk_read _data_chunk_data_seek_neededEOFError_read_fmt_chunkZ chunksize
_framesize_nframesskip)selffilechunkZ chunknamer r r initfp} s8
zWave_read.initfpc C sR d | _ t|tr"t|d}|| _ z| | W n | j rF| Y n0 d S )Nrb_i_opened_the_file
isinstancestrbuiltinsr r% closer" fr r r __init__ s
zWave_read.__init__c C s | d S Nr, r" r r r __del__ s zWave_read.__del__c C s | S r0 r r2 r r r __enter__ s zWave_read.__enter__c G s | d S r0 r1 r" argsr r r __exit__ s zWave_read.__exit__c C s | j S r0 )r r2 r r r getfp s zWave_read.getfpc C s d| _ d| _d S )Nr
r )r r r2 r r r rewind s zWave_read.rewindc C s" d | _ | j}|r d | _| d S r0 )r r( r, r" r# r r r r, s
zWave_read.closec C s | j S r0 )r r2 r r r tell s zWave_read.tellc C s | j S r0 )
_nchannelsr2 r r r getnchannels s zWave_read.getnchannelsc C s | j S r0 )r r2 r r r
getnframes s zWave_read.getnframesc C s | j S r0 )
_sampwidthr2 r r r getsampwidth s zWave_read.getsampwidthc C s | j S r0 )
_framerater2 r r r getframerate s zWave_read.getframeratec C s | j S r0 _comptyper2 r r r getcomptype s zWave_read.getcomptypec C s | j S r0 _compnamer2 r r r getcompname s zWave_read.getcompnamec C s* t | | | | | | S r0 )r r= r@ rB r> rE rH r2 r r r getparams s zWave_read.getparamsc C s d S r0 r r2 r r r
getmarkers s zWave_read.getmarkersc C s t dd S Nzno marksr r" idr r r getmark s zWave_read.getmarkc C s* |dk s|| j krtd|| _d| _d S )Nr zposition not in ranger
)r r r r )r" posr r r setpos s zWave_read.setposc C s | j r8| jdd | j| j }|r2| j|d d| _ |dkrDdS | j|| j }| jdkrxtjdkrxt
|| j}| j r|r| |}| jt || j
| j | _|S )Nr r
big)r r seekr r r r? sys byteorderaudioopbyteswapr lenr< )r" nframesrP datar r r
readframes s
zWave_read.readframesc C s z$t d|d\}| _| _}}W n t jy> td Y n0 |tkrzt d|dd }W n t jy| td Y n0 |d d | _| jst dnt d |f | jst d
| j| j | _
d | _ d | _ d S )
Nz r@ rB rE rH rI rJ rO rQ r\ r r r r r r ^ s*
r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9S ):r c C sR d | _ t|tr"t|d}|| _ z| | W n | j rF| Y n0 d S )Nwbr' r- r r r r/ - s
zWave_write.__init__c C s@ || _ d | _d| _d| _d| _d| _d| _d| _d| _d| _ d S )Nr F)
r r r< r? rA r _nframeswritten _datawritten _datalength_headerwrittenr: r r r r% 9 s zWave_write.initfpc C s | d S r0 r1 r2 r r r r3 E s zWave_write.__del__c C s | S r0 r r2 r r r r4 H s zWave_write.__enter__c G s | d S r0 r1 r5 r r r r7 K s zWave_write.__exit__c C s( | j rtd|dk r td|| _d S )N0cannot change parameters after starting to writer
rb )rj r r< )r" nchannelsr r r setnchannelsQ s
zWave_write.setnchannelsc C s | j std| j S )Nznumber of channels not set)r< r r2 r r r r= X s zWave_write.getnchannelsc C s0 | j rtd|dk s |dkr&td|| _d S )Nrm r
r ra )rj r r? )r" rg r r r setsampwidth] s
zWave_write.setsampwidthc C s | j std| j S )Nzsample width not set)r? r r2 r r r r@ d s zWave_write.getsampwidthc C s0 | j rtd|dkr tdtt|| _d S )Nrm r zbad frame rate)rj r introundrA )r" framerater r r setframeratei s
zWave_write.setframeratec C s | j std| j S )Nzframe rate not set)rA r r2 r r r rB p s zWave_write.getframeratec C s | j rtd|| _d S Nrm )rj r r )r" rZ r r r
setnframesu s zWave_write.setnframesc C s | j S r0 ri r2 r r r r> z s zWave_write.getnframesc C s. | j rtd|dvr td|| _|| _d S )Nrm )rc z unsupported compression type)rj r rD rG )r" comptypecompnamer r r setcomptype} s zWave_write.setcomptypec C s | j S r0 rC r2 r r r rE s zWave_write.getcomptypec C s | j S r0 rF r2 r r r rH s zWave_write.getcompnamec C sV |\}}}}}}| j r td| | | | | | | | | || d S ru )rj r ro rp rt rv rz )r" paramsrn rg rs rZ rx ry r r r setparams s
zWave_write.setparamsc C s8 | j r| jr| jstdt| j | j| j| j| j| jS )Nznot all parameters set)r< r? rA r r r rD rG r2 r r r rI s
zWave_write.getparamsc C s t dd S )Nzsetmark() not supportedrL )r" rN rP namer r r setmark s zWave_write.setmarkc C s t dd S rK rL rM r r r rO s zWave_write.getmarkc C s d S r0 r r2 r r r rJ s zWave_write.getmarkersc C s | j S r0 rw r2 r r r r; s zWave_write.tellc C s t |ttfs t|d}| t| t|| j| j }| j rN| |}| jdkrpt
j dkrpt
|| j}| j| | jt|7 _| j| | _d S )NBr
rS )r) bytes bytearray
memoryviewcast_ensure_header_writtenrY r? r< r rU rV rW rX r writerj ri )r" r[ rZ r r r writeframesraw s
zWave_write.writeframesrawc C s" | | | j| jkr | d S r0 )r rk rj _patchheader)r" r[ r r r writeframes s
zWave_write.writeframesc C sv zP| j r0| d | j| jkr&| | j W d | _ | j}|rrd | _| n d | _ | j}|rpd | _| 0 d S )Nr )r r rk rj r flushr( r, r: r r r r, s
zWave_write.closec C s> | j s:| jstd| js"td| js0td| | d S )Nz# channels not specifiedzsample width not specifiedzsampling rate not specified)rl r< r r? rA
_write_header)r" Zdatasizer r r r s z!Wave_write._ensure_header_writtenc C s | j d | js$|| j| j | _| j| j | j | _z| j | _W n tt fyd d | _Y n0 | j t
dd| j dddt | j| j
| j| j
| j | j| j | jd d | jd ur| j | _| j t
d | j d
| _d S ) Nr z rz rE rH r| rI r~ rO rJ r; r r r, r r r r r r r r s8
r c C sJ |d u r t | dr| j}nd}|dv r.t| S |dv r>t| S tdd S )Nmoder& )rr& )wrh z$mode must be 'r', 'rb', 'w', or 'wb')hasattrr r r r )r. r r r r r s
r )N)r$ r collectionsr rW r+ rd rU __all__ Exceptionr rf Z _array_fmtsr r r r r r r r J s" 6 d