U
e5d6G � @ s� d dl Z dddddgZG dd� de�ZdZd Zd dlZd dlZd dlZd d
l m
Z
d d l m Z d dl
Z
e d d
�ZG dd� d�ZG dd� d�Zddd�Zddd�ZdS )� N�open�openfp�Error� Wave_read�
Wave_writec @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r
r
� /usr/lib64/python3.8/wave.pyr N s � )N�b�hN�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�| _| j�� dkr0td��| j�d�dkrHtd��d| _d | _d| _ zt| jdd�}W n t
k
r� Y q�Y nX |�� }|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 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 | � � d S �N�r- �r# r
r
r �__del__� s zWave_read.__del__c C s | S r1 r
r3 r
r
r � __enter__� s zWave_read.__enter__c G s | � � d S r1 r2 �r# �argsr
r
r �__exit__� s zWave_read.__exit__c C s | j S r1 )r r3 r
r
r �getfp� s zWave_read.getfpc C s d| _ d| _d S )Nr r )r r r3 r
r
r �rewind� s zWave_read.rewindc C s" d | _ | j}|r d | _|�� d S r1 )r r) r- �r# r$ r
r
r r- � s
zWave_read.closec C s | j S r1 )r r3 r
r
r �tell� s zWave_read.tellc C s | j S r1 )�
_nchannelsr3 r
r
r � getnchannels� s zWave_read.getnchannelsc C s | j S r1 )r! r3 r
r
r �
getnframes� s zWave_read.getnframesc C s | j S r1 )�
_sampwidthr3 r
r
r � getsampwidth� s zWave_read.getsampwidthc C s | j S r1 )�
_framerater3 r
r
r � getframerate� s zWave_read.getframeratec C s | j S r1 �� _comptyper3 r
r
r � getcomptype� s zWave_read.getcomptypec C s | j S r1 �� _compnamer3 r
r
r � getcompname� s zWave_read.getcompnamec C s* t | �� | �� | �� | �� | �� | �� �S r1 )r r> rA rC r? rF rI r3 r
r
r � getparams� s �zWave_read.getparamsc C s d S r1 r
r3 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# �nframesrQ �datar
r
r �
readframes� s
zWave_read.readframesc C s� z$t �d|�d��\}| _| _}}W n t jk
r@ td �Y nX |tkr�zt �d|�d��d }W n t jk
r� td �Y nX |d d | _| js�t d��nt d |f ��| js�t d
��| j| j | _
d | _ d | _ d S )
Nz r? rA rC rF rI rJ rK rP rR r] 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"t�|d�}|| _ z| �|� W n | j rF|�� � Y nX d S )N�wbr( r. r
r
r r0 . s
zWave_write.__init__c C s@ || _ d | _d| _d| _d| _d| _d| _d| _d| _d| _ d S )Nr F)
r r r= r@ rB r! �_nframeswritten� _datawritten� _datalength�_headerwrittenr; r
r
r r& : s zWave_write.initfpc C s | � � d S r1 r2 r3 r
r
r r4 F s zWave_write.__del__c C s | S r1 r
r3 r
r
r r5 I s zWave_write.__enter__c G s | � � d S r1 r2 r6 r
r
r r8 L s zWave_write.__exit__c C s( | j rtd��|dk r td��|| _d S )N�0cannot change parameters after starting to writer rc )rk r r= )r# � nchannelsr
r
r � setnchannelsR s
zWave_write.setnchannelsc C s | j std��| j S )Nznumber of channels not set)r= r r3 r
r
r r> Y s zWave_write.getnchannelsc C s0 | j rtd��|dk s |dkr&td��|| _d S )Nrn r r rb )rk r r@ )r# rh r
r
r � setsampwidth^ s
zWave_write.setsampwidthc C s | j std��| j S )Nzsample width not set)r@ r r3 r
r
r rA e s zWave_write.getsampwidthc C s0 | j rtd��|dkr td��tt|��| _d S )Nrn r zbad frame rate)rk r �int�roundrB )r# � framerater
r
r � setframeratej s
zWave_write.setframeratec C s | j std��| j S )Nzframe rate not set)rB r r3 r
r
r rC q s zWave_write.getframeratec C s | j rtd��|| _d S �Nrn )rk r r! )r# r[ r
r
r �
setnframesv s zWave_write.setnframesc C s | j S r1 �rj r3 r
r
r r? { s zWave_write.getnframesc C s. | j rtd��|dkr td��|| _|| _d S )Nrn )rd z unsupported compression type)rk r rE rH )r# �comptype�compnamer
r
r � setcomptype~ s zWave_write.setcomptypec C s | j S r1 rD r3 r
r
r rF � s zWave_write.getcomptypec C s | j S r1 rG r3 r
r
r rI � s zWave_write.getcompnamec C sV |\}}}}}}| j r td��| �|� | �|� | �|� | �|� | �||� d S rv )rk r rp rq ru rw r{ )r# Zparamsro rh rt r[ ry rz 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@ rB r r r! rE rH r3 r
r
r rJ � s �zWave_write.getparamsc C s t d��d S )Nzsetmark() not supportedrM )r# rO rQ �namer
r
r �setmark� s zWave_write.setmarkc C s t d��d S rL rM rN r
r
r rP � s zWave_write.getmarkc C s d S r1 r
r3 r
r
r rK � s zWave_write.getmarkersc C s | j S r1 rx r3 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 rT )r* �bytes� bytearray�
memoryview�cast�_ensure_header_writtenrZ r@ r= r rV rW rX rY r �writerk rj )r# r\ r[ r
r
r �writeframesraw� s
zWave_write.writeframesrawc C s" | � |� | j| jkr | �� d S r1 )r� rl rk � _patchheader)r# r\ r
r
r � writeframes� s
zWave_write.writeframesc C sX z2| j r0| �d� | j| jkr&| �� | j �� W 5 d | _ | j}|rRd | _|�� X d S )Nr )r r) r- r� rl rk r� �flushr; 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)rm r= r r@ rB �
_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 fk
rf d | _Y nX | j �t
� dd| j dddt | j| j
| j| j
| j | j| j | jd d� � | jd k r�| j �� | _| j �t
� d | j�� d
| _d S ) Nr z rq rA ru rC rw r? r{ rF rI r| rJ r~ rP rK r< r� r� r- r� r� r� 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' )�rr' )�wri z$mode must be 'r', 'rb', 'w', or 'wb')�hasattrr� r r r �r/ r� r
r
r r � s
c C s t jdtdd� t| |d�S )NzBwave.openfp is deprecated since Python 3.7. Use wave.open instead.r_ )�
stacklevel)r� )�warnings�warn�DeprecationWarningr r� r
r
r r s
�)N)N)r, �__all__� Exceptionr rg Z _array_fmtsrX re rV r% r � collectionsr r� r r r r r r
r
r
r �J s& � 6 d