3
\�F � @ s� d dl mZ edd�ZdZdZdZdZdZd Zd
Z d Z
d Z d
Z dZ
dZdZdZeeeeeegZG dd� de�Zdd� Zdd� ZG dd� d�ZG dd� d�Zd d d �ZeZd S ) � )�
namedtuple�
_sunau_paramsz7nchannels sampwidth framerate nframes comptype compnameidns.� � � � � � � � � � � � l �� c @ s e Zd ZdS )�ErrorN)�__name__�
__module__� __qualname__� r r � /usr/lib64/python3.6/sunau.pyr � s r c C s< d}x2t d�D ]&}| jd�}|s$t�|d t|� }qW |S )Nr r r � )�range�read�EOFError�ord)�file�x�iZbyter r r � _read_u32� s
r c C sJ g }x2t d�D ]&}t|d�\}}|jdt|�� |}qW | jt|�� 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}|j|d�}d| _nd| _| j|� 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| j� d S )N)�_file�close)r0 r r r �__del__� s zAu_read.__del__c C s | S )Nr )r0 r r r � __enter__� s zAu_read.__enter__c G s | j � d S )N)r4 )r0 �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| j | _| jdk�r~|j| jd �| _| jjd
�\| _}}nd| _y|j� | _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 encoding� � ) r3 � _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)r0 r �magic�_r r r r/ � sP
zAu_read.initfpc C s | j S )N)r3 )r0 r r r �getfp� s z
Au_read.getfpc C s | j S )N)rK )r0 r r r � getnchannels� s zAu_read.getnchannelsc C s | j S )N)rD )r0 r r r � getsampwidth� s zAu_read.getsampwidthc C s | j S )N)rJ )r0 r r r � getframerate� s zAu_read.getframeratec C s( | j tkrtS | jtkr$| j | j S dS )Nr )r> r? r@ rA rE )r0 r r r �
getnframes� s
zAu_read.getnframesc C s$ | j tkrdS | j tkr dS dS d S )N�ULAW�ALAW�NONE)r@ rB rC )r0 r r r � getcomptype� s
zAu_read.getcomptypec C s$ | j tkrdS | j tkr dS dS d S )NzCCITT G.711 u-lawzCCITT G.711 A-lawznot compressed)r@ rB rC )r0 r r r � getcompname� s
zAu_read.getcompnamec C s* t | j� | j� | j� | j� | j� | j� �S )N)r rU rV rW rX r\ r] )r0 r r r � getparams s zAu_read.getparamsc C s d S )Nr )r0 r r r �
getmarkers s zAu_read.getmarkersc C s t d��d S )Nzno marks)r )r0 �idr r r �getmark s zAu_read.getmarkc C sp | j tkrl|tkr | jj� }n| jj|| j �}| jt|�| j 7 _| j tkrhdd l }|j
|| j �}|S d S )Nr ) r@ rA r? r3 r rE r< �lenrB �audioopZulaw2linrD )r0 �nframesr$ rc r r r �
readframes s
zAu_read.readframesc C s* | j d krtd��| jj| j � d| _d S )Nz cannot seekr )rO rQ r3 �seekr< )r0 r r r �rewind s
zAu_read.rewindc C s | j S )N)r< )r0 r r r rN s z Au_read.tellc C sP |dk s|| j � kr td��| jd kr.td��| jj| j|| j � || _d S )Nr zposition not in rangez cannot seek)rX r rO rQ r3 rf rE r< )r0 �posr r r �setpos! s
zAu_read.setposc C s" | j }|r d | _ | jr |j� d S )N)r3 r. r4 )r0 r r r r r4 ) s
z
Au_read.closeN)r r r r2 r5 r6 r8 r/ rT rU rV rW rX r\ r] r^ r_ ra re rg rN ri r4 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}|j|d�}d| _nd| _| j|� d S )Nr) r �wbTF)r+ r, r- r. r/ )r0 r1 r, r r r r2 2 s zAu_write.__init__c C s | j r| j� d | _ d S )N)r3 r4 )r0 r r r r5 ; s zAu_write.__del__c C s | S )Nr )r0 r r r r6 @ s zAu_write.__enter__c G s | j � d S )N)r4 )r0 r7 r r r r8 C s zAu_write.__exit__c C sF || _ d| _d| _d| _d| _t| _d| _d| _d| _ d| _
d| _ d S )Nr r; rY ) r3 rJ rK rD rE r? �_nframes�_nframeswritten� _datawritten� _datalengthrL � _comptype)r0 r r r r r/ F s zAu_write.initfpc C s( | j rtd��|dkr td��|| _d S )Nz0cannot change parameters after starting to writer r r z"only 1, 2, or 4 channels supported)r r r )rm r rK )r0 � nchannelsr r r � setnchannelsS s
zAu_write.setnchannelsc C s | j std��| j S )Nznumber of channels not set)rK r )r0 r r r rU Z s zAu_write.getnchannelsc C s( | j rtd��|dkr td��|| _d S )Nz0cannot change parameters after starting to writer r r r zbad sample width)r r r r )rm r rD )r0 � sampwidthr r r � setsampwidth_ s
zAu_write.setsampwidthc C s | j std��| jS )Nzsample width not specified)rJ r rD )r0 r r r rV f s zAu_write.getsampwidthc C s | j rtd��|| _d S )Nz0cannot change parameters after starting to write)rm r rJ )r0 � framerater r r � setframeratek s zAu_write.setframeratec C s | j std��| j S )Nzframe rate not set)rJ r )r0 r r r rW p s zAu_write.getframeratec C s( | j rtd��|dk r td��|| _d S )Nz0cannot change parameters after starting to writer z # of frames cannot be negative)rm r rl )r0 rd r r r �
setnframesu s
zAu_write.setnframesc C s | j S )N)rm )r0 r r r rX | s zAu_write.getnframesc C s |dkr|| _ ntd��d S )Nr[ rY zunknown compression type)r[ rY )rp r )r0 r+ �namer r r � setcomptype s zAu_write.setcomptypec C s | j S )N)rp )r0 r r r r\ � s zAu_write.getcomptypec C s$ | j dkrdS | j dkr dS dS d S )NrY zCCITT G.711 u-lawrZ zCCITT G.711 A-lawznot compressed)rp )r0 r r r r] � s
zAu_write.getcompnamec C sH |\}}}}}}| j |� | j|� | j|� | j|� | j||� d S )N)rr rt rv rw ry )r0 Zparamsrq rs ru rd ZcomptypeZcompnamer r r � setparams� s
zAu_write.setparamsc C s* t | j� | j� | j� | j� | j� | j� �S )N)r rU rV rW rX r\ r] )r0 r r r r^ � s zAu_write.getparamsc C s | j S )N)rm )r0 r r r rN � s z
Au_write.tellc C s~ t |ttf�s t|�jd�}| j� | jdkrDdd l}|j|| j �}t
|�| j }| j j
|� | j| | _| jt
|� | _d S )N�BrY r )�
isinstancer# � bytearray�
memoryview�cast�_ensure_header_writtenrp rc Zlin2ulawrD rb rE r3 r" rm rn )r0 r$ rc rd r r r �writeframesraw� s
zAu_write.writeframesrawc C s. | j |� | j| jks"| j| jkr*| j� d S )N)r� rm rl ro rn � _patchheader)r0 r$ r r r � writeframes� s
zAu_write.writeframesc C s^ | j rZz6| j� | j| jks(| j| jkr0| j� | j j� W d | j }d | _ | jrX|j � X d S )N)
r3 r� rm rl ro rn r� �flushr. r4 )r0 r r r r r4 � s zAu_write.closec C s< | j s8| jstd��| js"td��| js0td��| j� d S )Nz# of channels not specifiedzsample width not specifiedzframe rate not specified)rm rK r rD rJ �
_write_header)r0 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 }y| j j� | _W n ttfk
�r d | _Y nX t
| j |� || _t
| j |� t
| j | j� t
| j | j � | j j| j� | j jd
|t
| j� d � d S )
Nr[ r r r r zinternal errorrY r
r
r: r i����)rp rD rF rE rG rH rI r rB rK r' r3 r= rb rL rl r? rN �_form_length_posrP rQ ro rJ r" )r0 �encodingZ header_sizeZlengthr r r r� � sJ
zAu_write._write_headerc C sH | j d krtd��| jj| j � t| j| j� | j| _| jjdd� d S )Nz cannot seekr r )r� rQ r3 rf r' rn ro )r0 r r r r� � s
zAu_write._patchheaderN) r r r r2 r5 r6 r8 r/ rr rU rt rV rv rW rw rX ry r\ r] rz r^ rN r� r� r4 r� r� r� r r r r rj 0 s2
*rj Nc C sJ |d kr t | d�r| j}nd}|dkr.t| �S |dkr>t| �S td��d S ) N�moder* �r�wrk z$mode must be 'r', 'rb', 'w', or 'wb')r� r* )r� rk )�hasattrr� r( rj r )r1 r� r r r r- s
r- )N)� collectionsr r r= rB rF rG rH rI 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_5rC r? rA � Exceptionr r r' r( rj r- Zopenfpr r r r �<module>j s> Q