U
e5d�G � @ s� d dl mZ d dlZedd�ZdZdZdZdZd Zd
Z d Z
d Z d
Z dZ
dZdZdZdZeeeee egZG dd� de�Zdd� Zdd� ZG dd� d�ZG dd � d �Zd!d d �Zd"d d �ZdS )#� )�
namedtupleN�
_sunau_paramsz7nchannels sampwidth framerate nframes comptype compnameidns.� � � � � � � � � � � � l �� c @ s e Zd ZdS )�ErrorN)�__name__�
__module__� __qualname__� r r � /usr/lib64/python3.8/sunau.pyr � s r c C s8 d}t d�D ]&}| �d�}|s"t�|d t|� }q |S )Nr r r � )�range�read�EOFError�ord)�file�x�iZbyter r r � _read_u32� s
r c C sF g }t d�D ]&}t|d�\}}|�dt|�� |}q | �t|�� d S )Nr r r )r �divmod�insert�int�write�bytes)r r �datar �d�mr r r �
_write_u32� 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)S )*�Au_readc C s@ t |�t d�kr,dd l}|�|d�}d| _nd| _| �|� d S )N� r �rbTF��type�builtins�open�_opened�initfp��self�fr- r r r �__init__� s zAu_read.__init__c C s | j r| �� d S �N��_file�close�r2 r r r �__del__� s zAu_read.__del__c C s | S r5 r r9 r r r � __enter__� s zAu_read.__enter__c G s | � � d S r5 �r8 �r2 �argsr r r �__exit__� s zAu_read.__exit__c C s� || _ d| _tt|��}|tkr(td��tt|��| _| jdk rHtd��| jdkrZtd��t|�| _| jtkrzt| j�| _tt|��| _ | j t
kr�td��| j t t fkr�d| _
d | _nj| j tkr�d | _| _
nR| j tkr�d | _| _
n:| j tkr�d
| _| _
n"| j tk�rd | _| _
ntd ��tt|��| _tt|��| _| j�sLtd
��| j| j | _| jdk�r�|�| jd �| _| j�d�\| _}}nd| _z|�� | _W n ttfk
�r� d | _Y nX d S )Nr zbad magic numberr zheader size too small�d z header size ridiculously largez encoding not (yet) supportedr r r r zunknown encodingzbad # of channels� � ) r7 � _soundposr! r �AUDIO_FILE_MAGICr Z _hdr_size�
_data_size�AUDIO_UNKNOWN_SIZE� _encoding�_simple_encodings�AUDIO_FILE_ENCODING_MULAW_8�AUDIO_FILE_ENCODING_ALAW_8�
_sampwidth�
_framesize� AUDIO_FILE_ENCODING_LINEAR_8� AUDIO_FILE_ENCODING_LINEAR_16� AUDIO_FILE_ENCODING_LINEAR_24� AUDIO_FILE_ENCODING_LINEAR_32�
_framerate�
_nchannelsr �_info� partition�tell� _data_pos�AttributeError�OSError)r2 r �magic�_r r r r0 � sV
�
zAu_read.initfpc C s | j S r5 )r7 r9 r r r �getfp� s z
Au_read.getfpc C s | j S r5 )rR r9 r r r � getnchannels� s zAu_read.getnchannelsc C s | j S r5 )rK r9 r r r � getsampwidth� s zAu_read.getsampwidthc C s | j S r5 )rQ r9 r r r � getframerate� s zAu_read.getframeratec C s( | j tkrtS | jtkr$| j | j S dS �Nr )rE rF rG rH rL r9 r r r �
getnframes� s
zAu_read.getnframesc C s$ | j tkrdS | j tkr dS dS d S )N�ULAW�ALAW�NONE�rG rI rJ r9 r r r � getcomptype� s
zAu_read.getcomptypec C s$ | j tkrdS | j tkr dS dS d S )N�CCITT G.711 u-law�CCITT G.711 A-law�not compressedrd r9 r r r � getcompname� s
zAu_read.getcompnamec C s* t | �� | �� | �� | �� | �� | �� �S r5 �r r\ r] r^ r` re ri r9 r r r � getparams s �zAu_read.getparamsc C s d S r5 r r9 r r r �
getmarkers s zAu_read.getmarkersc C s t d��d S )Nzno marks)r )r2 �idr r r �getmark s zAu_read.getmarkc C sp | j tkrl|tkr | j�� }n| j�|| j �}| jt|�| j 7 _| j tkrhdd l }|�
|| j �}|S d S r_ ) rG rH rF r7 r rL rC �lenrI �audioopZulaw2linrK )r2 �nframesr$ rp r r r �
readframes s
zAu_read.readframesc C s* | j d krtd��| j�| j � d| _d S )N� cannot seekr )rV rX r7 �seekrC r9 r r r �rewind s
zAu_read.rewindc C s | j S r5 )rC r9 r r r rU ! s z Au_read.tellc C sP |dk s|| � � kr td��| jd kr.td��| j�| j|| j � || _d S )Nr zposition not in rangers )r` r rV rX r7 rt rL rC )r2 �posr r r �setpos$ s
zAu_read.setposc C s" | j }|r d | _ | jr |�� d S r5 )r7 r/ r8 �r2 r r r r r8 , s
z
Au_read.closeN)r r r r4 r: r; r? r0 r[ r\ r] r^ r` re ri rk rl rn rr ru rU rw r8 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� Zd3S )4�Au_writec C s@ t |�t d�kr,dd l}|�|d�}d| _nd| _| �|� d S )Nr) r �wbTFr+ r1 r r r r4 5 s zAu_write.__init__c C s | j r| �� d | _ d S r5 r6 r9 r r r r: > s zAu_write.__del__c C s | S r5 r r9 r r r r; C s zAu_write.__enter__c G s | � � d S r5 r< r= r r r r? F s zAu_write.__exit__c C sF || _ d| _d| _d| _d| _t| _d| _d| _d| _ d| _
d| _ d S )Nr rB ra ) r7 rQ rR rK rL rF �_nframes�_nframeswritten� _datawritten� _datalengthrS � _comptyperx r r r r0 I s zAu_write.initfpc C s( | j rtd��|dkr td��|| _d S )N�0cannot change parameters after starting to write)r r r z"only 1, 2, or 4 channels supported)r| r rR )r2 � nchannelsr r r � setnchannelsV s
zAu_write.setnchannelsc C s | j std��| j S )Nznumber of channels not set)rR r r9 r r r r\ ] s zAu_write.getnchannelsc C s( | j rtd��|dkr td��|| _d S )Nr� )r r r r zbad sample width)r| r rK )r2 � sampwidthr r r � setsampwidthb s
zAu_write.setsampwidthc C s | j std��| jS )N�sample width not specified)rQ r rK r9 r r r r] i s zAu_write.getsampwidthc C s | j rtd��|| _d S )Nr� )r| r rQ )r2 � framerater r r � setframeraten s zAu_write.setframeratec C s | j std��| j S )Nzframe rate not set)rQ r r9 r r r r^ s s zAu_write.getframeratec C s( | j rtd��|dk r td��|| _d S )Nr� r z # of frames cannot be negative)r| r r{ )r2 rq r r r �
setnframesx s
zAu_write.setnframesc C s | j S r5 �r| r9 r r r r` s zAu_write.getnframesc C s |dkr|| _ ntd��d S )N)rc ra zunknown compression type)r r )r2 r, �namer r r � setcomptype� s zAu_write.setcomptypec C s | j S r5 �r r9 r r r re � s zAu_write.getcomptypec C s$ | j dkrdS | j dkr dS dS d S )Nra rf rb rg rh r� r9 r r r ri � s
zAu_write.getcompnamec C sH |\}}}}}}| � |� | �|� | �|� | �|� | �||� d S r5 )r� r� r� r� r� )r2 Zparamsr� r� r� rq ZcomptypeZcompnamer r r � setparams� s
zAu_write.setparamsc C s* t | �� | �� | �� | �� | �� | �� �S r5 rj r9 r r r rk � s �zAu_write.getparamsc C s | j S r5 r� r9 r r r rU � s z
Au_write.tellc C s~ t |ttf�s t|��d�}| �� | jdkrDdd l}|�|| j �}t
|�| j }| j �
|� | j| | _| jt
|� | _d S )N�Bra r )�
isinstancer# � bytearray�
memoryview�cast�_ensure_header_writtenr rp Zlin2ulawrK ro rL r7 r" r| r} )r2 r$ rp rq r r r �writeframesraw� s
zAu_write.writeframesrawc C s. | � |� | j| jks"| j| jkr*| �� d S r5 )r� r| r{ r~ r} � _patchheader)r2 r$ r r r � writeframes� s
�zAu_write.writeframesc C s^ | j rZz6| �� | j| jks(| j| jkr0| �� | j � � W 5 | j }d | _ | jrX|�� X d S r5 )
r7 r/ r8 r� r| r{ r~ r} r� �flushrx r r r r8 � s
�zAu_write.closec C s< | j s8| jstd��| js"td��| js0td��| �� d S )Nz# of channels not specifiedr� zframe rate not specified)r| rR r rK rQ �
_write_headerr9 r r r r� � s z Au_write._ensure_header_writtenc C s� | j dkrl| jdkr t}d| _q�| jdkr6t}d| _q�| jdkrLt}d| _q�| jdkrbt}d| _q�td��n | j dkr�t}d| _ntd��| j| j | _t
| j t � dt
| j� }|d d
@ }t
| j |� | jtkr�t}n | j| j }z| j �� | _W n ttfk
�r d | _Y nX t
| j |� || _t
| j |� t
| j | j� t
| j | j � | j �| j� | j �d |t
| j� d � d S )
Nrc r r r r zinternal errorra r
r
i����rA r )r rK rM rL rN rO rP r rI rR r' r7 rD ro rS r{ rF rU �_form_length_posrW rX r~ rQ r" )r2 �encoding� header_sizeZlengthr r r r� � sJ
zAu_write._write_headerc C sH | j d krtd��| j�| j � t| j| j� | j| _| j�dd� d S )Nrs r r )r� rX r7 rt r' r} r~ r9 r r r r� � s
zAu_write._patchheaderN) r r r r4 r: r; r? r0 r� r\ r� r] r� r^ r� r` r� re ri r� rk rU r� r� r8 r� r� r� r r r r ry 3 s2
*ry c C sJ |d kr t | d�r| j}nd}|dkr.t| �S |dkr>t| �S td��d S )N�moder* )�rr* )�wrz z$mode must be 'r', 'rb', 'w', or 'wb')�hasattrr� r( ry r �r3 r� r r r r. s
r. c C s t jdtdd� t| |d�S )NzDsunau.openfp is deprecated since Python 3.7. Use sunau.open instead.r )�
stacklevel)r� )�warnings�warn�DeprecationWarningr. r� r r r �openfp s
�r� )N)N)� collectionsr r� r rD rI rM rN rO rP ZAUDIO_FILE_ENCODING_FLOATZAUDIO_FILE_ENCODING_DOUBLEZ AUDIO_FILE_ENCODING_ADPCM_G721Z AUDIO_FILE_ENCODING_ADPCM_G722Z AUDIO_FILE_ENCODING_ADPCM_G723_3Z AUDIO_FILE_ENCODING_ADPCM_G723_5rJ rF rH � Exceptionr r r' r( ry r. r� r r r r �j sD �� Q