3
\-E � @ s� d dl Z dddddgZG dd� de�ZdZdZd dlZd dlZd dlZd d l m
Z
d d
l m Z e dd�Z
G dd� d�ZG dd� d�Zddd�ZeZdS )� N�open�openfp�Error� Wave_read�
Wave_writec @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r
r
� /usr/lib64/python3.6/wave.pyr N s � �b�h�i)�Chunk)�
namedtuple� _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�| _| jj� dkr0td��| jjd�dkrHtd��d| _d | _x�d| _ yt| jdd�}W n t
k
r� P Y nX |j� }|d kr�| j |� d| _n2|d
kr�| js�td ��|| _|j | j
| _d| _ P |j� qVW | j s�| j �r td ��d S )
Nr )Z bigendians RIFFz file does not start with RIFF id� s WAVEznot a WAVE filer s fmt s datazdata chunk before fmt chunkz#fmt chunk and/or data chunk missing)�_convert� _soundposr �_fileZgetnamer �readZ_fmt_chunk_read� _data_chunk�_data_seek_needed�EOFError�_read_fmt_chunkZ chunksize�
_framesize�_nframes�skip)�self�file�chunkZ chunknamer
r
r �initfp} s:
zWave_read.initfpc C sR d | _ t|t�r"tj|d�}|| _ y| j|� W n | j rF|j� � Y nX d S )N�rb)�_i_opened_the_file�
isinstance�str�builtinsr r" �close)r �fr
r
r �__init__� s
zWave_read.__init__c C s | j � d S )N)r( )r r
r
r �__del__� s zWave_read.__del__c C s | S )Nr
)r r
r
r � __enter__� s zWave_read.__enter__c G s | j � d S )N)r( )r �argsr
r
r �__exit__� s zWave_read.__exit__c C s | j S )N)r )r r
r
r �getfp� s zWave_read.getfpc C s d| _ d| _d S )Nr r )r r )r r
r
r �rewind� s zWave_read.rewindc C s" d | _ | j}|r d | _|j� d S )N)r r$ r( )r r r
r
r r( � s
zWave_read.closec C s | j S )N)r )r r
r
r �tell� s zWave_read.tellc C s | j S )N)�
_nchannels)r r
r
r � getnchannels� s zWave_read.getnchannelsc C s | j S )N)r )r r
r
r �
getnframes� s zWave_read.getnframesc C s | j S )N)�
_sampwidth)r r
r
r � getsampwidth� s zWave_read.getsampwidthc C s | j S )N)�
_framerate)r r
r
r � getframerate� s zWave_read.getframeratec C s | j S )N)� _comptype)r r
r
r � getcomptype� s zWave_read.getcomptypec C s | j S )N)� _compname)r r
r
r � getcompname� s zWave_read.getcompnamec C s* t | j� | j� | j� | j� | j� | j� �S )N)r r3 r6 r8 r4 r: r< )r r
r
r � getparams� s zWave_read.getparamsc C s d S )Nr
)r r
r
r �
getmarkers� s zWave_read.getmarkersc C s t d��d S )Nzno marks)r )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| jjdd� | j| j }|r2| jj|d� d| _ |dkrDdS | jj|| j �}| jdkrxtjdkrxt j
|| j�}| j r�|r�| j |�}| jt |�| j
| j | _|S )Nr � r �big)r r �seekr r r r5 �sys� byteorder�audioop�byteswapr �lenr2 )r �nframesrA �datar
r
r �
readframes� s
zWave_read.readframesc C sz t jd|jd��\}| _| _}}|tkrNt jd|jd��d }|d d | _ntd|f ��| j| j | _d | _ d
| _
d S ) Nz<HHLLH� z<H� r � � zunknown format: %r�NONEznot compressed) �structZ unpack_fromr r2 r7 �WAVE_FORMAT_PCMr5 r r r9 r; )r r! Z
wFormatTagZdwAvgBytesPerSecZ wBlockAlign� sampwidthr
r
r r � s zWave_read._read_fmt_chunkN)r r r r" r* r+ r, r. r/ r0 r( r1 r3 r4 r6 r8 r: r< r= r> r@ rB rM r r
r
r
r r ^ s*
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|t�r"tj|d�}|| _ y| j|� W n | j rF|j� � Y nX d S )N�wb)r$ r% r& r' r r" r( )r 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 r2 r5 r7 r �_nframeswritten� _datawritten� _datalength�_headerwritten)r r r
r
r r" / s zWave_write.initfpc C s | j � d S )N)r( )r r
r
r r+ ; s zWave_write.__del__c C s | S )Nr
)r r
r
r r, > s zWave_write.__enter__c G s | j � d S )N)r( )r r- r
r
r r. A s zWave_write.__exit__c C s( | j rtd��|dk r td��|| _d S )Nz0cannot change parameters after starting to writer zbad # of channels)rX r r2 )r � nchannelsr
r
r � setnchannelsG s
zWave_write.setnchannelsc C s | j std��| j S )Nznumber of channels not set)r2 r )r r
r
r r3 N s zWave_write.getnchannelsc C s0 | j rtd��|dk s |dkr&td��|| _d S )Nz0cannot change parameters after starting to writer r zbad sample width)rX r r5 )r rU r
r
r � setsampwidthS s
zWave_write.setsampwidthc C s | j std��| j S )Nzsample width not set)r5 r )r r
r
r r6 Z s zWave_write.getsampwidthc C s0 | j rtd��|dkr td��tt|��| _d S )Nz0cannot change parameters after starting to writer zbad frame rate)rX r �int�roundr7 )r � framerater
r
r � setframerate_ s
zWave_write.setframeratec C s | j std��| j S )Nzframe rate not set)r7 r )r r
r
r r8 f s zWave_write.getframeratec C s | j rtd��|| _d S )Nz0cannot change parameters after starting to write)rX r r )r rK r
r
r �
setnframesk s zWave_write.setnframesc C s | j S )N)rW )r r
r
r r4 p s zWave_write.getnframesc C s. | j rtd��|dkr td��|| _|| _d S )Nz0cannot change parameters after starting to writerR z unsupported compression type)rR )rX r r9 r; )r �comptype�compnamer
r
r � setcomptypes s zWave_write.setcomptypec C s | j S )N)r9 )r r
r
r r: { s zWave_write.getcomptypec C s | j S )N)r; )r r
r
r r< ~ s zWave_write.getcompnamec C sV |\}}}}}}| j r td��| j|� | j|� | j|� | j|� | j||� d S )Nz0cannot change parameters after starting to write)rX r r\ r] ra rb re )r Zparamsr[ rU r` rK rc rd r
r
r � setparams� s
zWave_write.setparamsc C s> | j s| j s| j r td��t| j | j| j| j| j| j�S )Nznot all parameters set)r2 r5 r7 r r r r9 r; )r r
r
r r= � s zWave_write.getparamsc C s t d��d S )Nzsetmark() not supported)r )r r? rA �namer
r
r �setmark� s zWave_write.setmarkc C s t d��d S )Nzno marks)r )r r? r
r
r r@ � s zWave_write.getmarkc C s d S )Nr
)r r
r
r r> � s zWave_write.getmarkersc C s | j S )N)rW )r r
r
r r1 � s zWave_write.tellc C s� t |ttf�s t|�jd�}| jt|�� t|�| j| j }| j rN| j |�}| jdkrpt
j dkrpt j
|| j�}| jj|� | jt|�7 _| j| | _d S )N�Br rD )r% �bytes� bytearray�
memoryview�cast�_ensure_header_writtenrJ r5 r2 r rF rG rH rI r �writerX rW )r rL rK r
r
r �writeframesraw� s
zWave_write.writeframesrawc C s" | j |� | j| jkr | j� d S )N)rp rY rX � _patchheader)r rL r
r
r � writeframes� s
zWave_write.writeframesc C sX z2| j r0| jd� | j| jkr&| j� | j j� W d d | _ | j}|rRd | _|j� X d S )Nr )r rn rY rX rq �flushr$ r( )r r r
r
r r( � s
zWave_write.closec C s> | j s:| jstd��| js"td��| js0td��| j|� d S )Nz# channels not specifiedzsample width not specifiedzsampling rate not specified)rZ r2 r r5 r7 �
_write_header)r Zdatasizer
r
r rn � s z!Wave_write._ensure_header_writtenc C s� | j jd� | js$|| j| j | _| j| j | j | _y| j j� | _W n tt fk
rf d | _Y nX | j jt
j dd| j dddt | j| j
| j| j
| j | j| j | jd d� � | jd k r�| j j� | _| j jt
j d | j�� d
| _d S ) Ns RIFFz<L4s4sLHHLLHH4s�$ s WAVEs fmt � rQ s dataz<LT)r ro r r2 r5 rY r1 �_form_length_pos�AttributeError�OSErrorrS �packrT r7 �_data_length_posrZ )r Z
initlengthr
r
r rt � s$
zWave_write._write_headerc C s� | j | jkrd S | jj� }| jj| jd� | jjtjdd| j �� | jj| j d� | jjtjd| j �� | jj|d� | j | _d S )Nr z<Lru )
rX rY r r1 rE rw ro rS rz r{ )r Zcurposr
r
r rq � s
zWave_write._patchheaderN) r r r r* r" r+ r, r. r\ r3 r] r6 ra r8 rb r4 re r: r< rf r= rh r@ r> r1 rp rr r( rn rt rq r
r
r
r r s8
c C sJ |d kr t | d�r| j}nd}|dkr.t| �S |dkr>t| �S td��d S ) N�moder# �r�wrV z$mode must be 'r', 'rb', 'w', or 'wb')r} r# )r~ rV )�hasattrr| r r r )r) r| r
r
r r � s
)Nr
r Nr )N)r' �__all__� Exceptionr rT Z _array_fmtsrH rS rF r! r � collectionsr r r r r r r
r
r
r �<module>J s" , d