a
R�fTF � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlZg d�ZG dd� de �Z
dZ dZ ed d
�Z
G d d � d �ZG d
d� d�Zddd�ZdS )� )�Chunk)�
namedtupleN)�open�Error� 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 � )N�b�hN�i� _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��| j�d�dkrHtd��d| _d | _d| _ zt| jdd�}W n t
y� Y q�Y n0 |�� }|d kr�| � |� d| _n2|d
kr�| js�td ��|| _|j | j
| _d| _ q�|�� qT| jr�| js�td ��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_needed�EOFError�_read_fmt_chunkZ chunksize�
_framesize�_nframes�skip)�self�file�chunkZ chunknamer r r �initfp} s8
zWave_read.initfpc C sR d | _ t|t�r"t�|d�}|| _ z| �|� W n | j rF|�� � Y n0 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 | � � d S �N�r, �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 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| j�dd� | 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� byteorder�audioop�byteswapr �lenr< )r" �nframesrP �datar r r �
readframes� s
zWave_read.readframesc C s� z$t �d|�d��\}| _| _}}W n t jy> td �Y n0 |tkr�zt �d|�d��d }W n t jy| td �Y n0 |d d | _| js�t d��nt d |f ��| js�t d
��| j| j | _
d | _ d | _ d S )
Nz<HHLLH� z<H� r � � �bad sample widthzunknown format: %r�bad # of channels�NONEznot compressed)
�structZ unpack_fromr r< rA �errorr �WAVE_FORMAT_PCMr? r r rD rG )r" r$ Z
wFormatTagZdwAvgBytesPerSecZ wBlockAlign� sampwidthr r r r � s$ $
zWave_read._read_fmt_chunkN)r r r
r% r/ r3 r4 r7 r8 r9 r, r; r= r> 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|t�r"t�|d�}|| _ z| �|� W n | j rF|�� � Y n0 d S )N�wbr' 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 )N�0cannot 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 td��tt|��| _d S )Nrm r zbad frame rate)rj r �int�roundrA )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" �comptype�compnamer 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| jstd��t| j | j| j| j| j| j�S )Nznot all parameters set)r< r? rA r r r rD rG r2 r r r rI � s
�zWave_write.getparamsc C s t d��d S )Nzsetmark() not supportedrL )r" rN rP �namer r r �setmark� s zWave_write.setmarkc C s t d��d 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 |ttf�s t|��d�}| �t|�� t|�| j| j }| j rN| � |�}| jdkrpt
j dkrpt �
|| j�}| j�|� | jt|�7 _| j| | _d S )N�Br
rS )r) �bytes� bytearray�
memoryview�cast�_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<L4s4sLHHLLHH4s�$ r r � r` r �<LT)r r� r r< r? rk r; �_form_length_pos�AttributeError�OSErrorrd �packrf rA �_data_length_posrl )r" Z
initlengthr r r r� � s&
�
zWave_write._write_headerc C s� | j | jkrd S | j�� }| j�| jd� | j�t�dd| j �� | j�| j d� | j�t�d| j �� | j�|d� | j | _d S )Nr r� r� )
rj rk r r; rT r� r� rd r� r� )r" Zcurposr r r r� � s
zWave_write._patchheaderN) r r r
r/ r% r3 r4 r7 ro r= rp r@ rt rB rv r> 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 | d�r| j}nd}|dv r.t| �S |dv r>t| �S td��d S )N�moder& )�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 �<module>J s" � 6 d