o
6��f�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 �Zd S ) � )�
namedtuple�
_sunau_paramsz7nchannels sampwidth framerate nframes comptype compnameidns.� � � � � � � � � � � � l �� c @ s e Zd ZdS )�ErrorN)�__name__�
__module__� __qualname__� r r �,/opt/alt/python310/lib64/python3.10/sunau.pyr � s r c C s8 d}t d�D ]}| �d�}|st�|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 �@ t |�t d�krdd l}|�|d�}d| _nd| _| �|� d S )N� r �rbTF��type�builtins�open�_opened�initfp��self�fr. r r r �__init__� � zAu_read.__init__c C s | j r | �� d S d S �N��_file�close�r3 r r r �__del__� s �zAu_read.__del__c C � | S r7 r r; r r r � __enter__� � zAu_read.__enter__c G � | � � d S r7 �r: �r3 �argsr r r �__exit__� � zAu_read.__exit__c C s� || _ d| _tt|��}|tkrtd��tt|��| _| jdk r$td��| jdkr-td��t|�| _| jtkr=t| j�| _tt|��| _ | j t
vrMtd��| j t t fv r[d| _
d | _n4| j tkrgd | _| _
n(| j tkrsd | _| _
n | j tkrd
| _| _
n| j tkr�d | _| _
ntd ��tt|��| _tt|��| _| js�td
��| j| j | _| jdkr�|�| jd �| _| j�d�\| _}}nd| _z|�� | _W d S ttfy� d | _Y d S w )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� � ) r9 � _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)r3 r �magic�_r r r r1 � sX
�
�zAu_read.initfpc C � | j S r7 )r9 r; r r r �getfp� � z
Au_read.getfpc C ra r7 )rX r; r r r � getnchannels� rc zAu_read.getnchannelsc C ra r7 )rQ r; r r r � getsampwidth� rc zAu_read.getsampwidthc C ra r7 )rW r; r r r � getframerate� rc zAu_read.getframeratec C s( | j tkrtS | jtv r| j | j S dS �Nr )rK rL rM rN rR r; r r r �
getnframes� s
zAu_read.getnframesc C � | j tkrdS | j tkrdS dS )N�ULAW�ALAW�NONE�rM rO rP r; r r r � getcomptype� �
zAu_read.getcomptypec C ri )N�CCITT G.711 u-law�CCITT G.711 A-law�not compressedrm r; r r r � getcompname� ro zAu_read.getcompnamec C �* t | �� | �� | �� | �� | �� | �� �S r7 �r rd re rf rh rn rs r; r r r � getparams � �zAu_read.getparamsc C s d S r7 r r; r r r �
getmarkers r? zAu_read.getmarkersc C s t d��)Nzno marks)r )r3 �idr r r �getmark s zAu_read.getmarkc C sp | j tv r6|tkr| j�� }n | j�|| j �}| jt|�| j 7 _| j tkr4dd l }|�
|| j �}|S d S rg ) rM rN rL r9 r rR rI �lenrO �audioopZulaw2linrQ )r3 �nframesr$ r| r r r �
readframes s
zAu_read.readframesc C s* | j d u r td��| j�| j � d| _d S )N� cannot seekr )r\ r^ r9 �seekrI r; r r r �rewind s
zAu_read.rewindc C ra r7 )rI r; r r r r[ ! rc z Au_read.tellc C sP |dk s
|| � � krtd��| jd u rtd��| j�| j|| j � || _d S )Nr zposition not in ranger )rh r r\ r^ r9 r� rR rI )r3 �posr r r �setpos$ s
zAu_read.setposc C s* | j }|rd | _ | jr|�� d S d S d S r7 )r9 r0 r: �r3 r r r r r: , s �z
Au_read.closeN)r r r r5 r< r> rD r1 rb rd re rf rh rn rs rv rx rz r~ r� r[ 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� Zd3S )4�Au_writec C r) )Nr* r �wbTFr, r2 r r r r5 5 r6 zAu_write.__init__c C s | j r| �� d | _ d S r7 r8 r; r r r r< > �
zAu_write.__del__c C r= r7 r r; r r r r> C r? zAu_write.__enter__c G r@ r7 rA rB r r r rD F rE zAu_write.__exit__c C sF || _ d| _d| _d| _d| _t| _d| _d| _d| _ d| _
d| _ d S )Nr rH rj ) r9 rW rX rQ rR rL �_nframes�_nframeswritten� _datawritten� _datalengthrY � _comptyper� r r r r1 I s
zAu_write.initfpc C �( | j rtd��|dvrtd��|| _d S )N�0cannot change parameters after starting to write)r r r z"only 1, 2, or 4 channels supported)r� r rX )r3 � nchannelsr r r � setnchannelsV �
zAu_write.setnchannelsc C � | j std��| j S )Nznumber of channels not set)rX r r; r r r rd ] � zAu_write.getnchannelsc C r� )Nr� )r r r r zbad sample width)r� r rQ )r3 � sampwidthr r r � setsampwidthb r� zAu_write.setsampwidthc C s | j std��| jS )N�sample width not specified)rW r rQ r; r r r re i r� zAu_write.getsampwidthc C s | j rtd��|| _d S )Nr� )r� r rW )r3 � framerater r r � setframeraten r� zAu_write.setframeratec C r� )Nzframe rate not set)rW r r; r r r rf s r� zAu_write.getframeratec C s( | j rtd��|dk rtd��|| _d S )Nr� r z # of frames cannot be negative)r� r r� )r3 r} r r r �
setnframesx r� zAu_write.setnframesc C ra r7 �r� r; r r r rh rc zAu_write.getnframesc C s |dv r || _ d S td��)N)rl rj zunknown compression type)r� r )r3 r- �namer r r � setcomptype� s
zAu_write.setcomptypec C ra r7 �r� r; r r r rn � rc zAu_write.getcomptypec C s | j dkrdS | j dkrdS dS )Nrj rp rk rq rr r� r; r r r rs � ro zAu_write.getcompnamec C sH |\}}}}}}| � |� | �|� | �|� | �|� | �||� d S r7 )r� r� r� r� r� )r3 �paramsr� r� r� r} ZcomptypeZcompnamer r r � setparams� s
zAu_write.setparamsc C rt r7 ru r; r r r rv � rw zAu_write.getparamsc C ra r7 r� r; r r r r[ � rc z
Au_write.tellc C s~ t |ttf�st|��d�}| �� | jdkr"dd l}|�|| j �}t
|�| j }| j �
|� | j| | _| jt
|� | _d S )N�Brj r )�
isinstancer# � bytearray�
memoryview�cast�_ensure_header_writtenr� r| Zlin2ulawrQ r{ rR r9 r" r� r� )r3 r$ r| r} r r r �writeframesraw� s
zAu_write.writeframesrawc C s2 | � |� | j| jks| j| jkr| �� d S d S r7 )r� r� r� r� r� � _patchheader)r3 r$ r r r � writeframes� s
�zAu_write.writeframesc C s� | j r>z+| �� | j| jks| j| jkr| �� | j �� W | j }d | _ | jr-|� � d S d S | j }d | _ | jr=|� � w w d S r7 )
r9 r� r� r� r� r� r� �flushr0 r: r� r r r r: � s$ ��
��zAu_write.closec C s@ | j s | js
td��| jstd��| jstd��| �� d S d S )Nz# of channels not specifiedr� zframe rate not specified)r� rX r rQ rW �
_write_headerr; r r r r� � s �z Au_write._ensure_header_writtenc C s| | j dkr5| jdkrt}d| _n4| jdkrt}d| _n)| jdkr&t}d| _n | jdkr1t}d| _ntd��| j dkr@t}d| _ntd��| j| j | _t
| j t � dt
| j� }|d d
@ }t
| j |� | jtkrlt}n| j| j }z| j �� | _W n ttfy� d | _Y nw t
| j |� || _t
| j |� t
| j | j� t
| j | j � | j �| j� | j �d |t
| j� d � d S )
Nrl r r r r zinternal errorrj r
r
i����rG r )r� rQ rS rR rT rU rV r rO rX r' r9 rJ r{ rY r� rL r[ �_form_length_posr] r^ r� rW r" )r3 �encoding� header_sizeZlengthr r r r� � sL
� "zAu_write._write_headerc C sH | j d u r td��| j�| j � t| j| j� | j| _| j�dd� d S )Nr r r )r� r^ r9 r� r' r� r� r; r r r r� � s
zAu_write._patchheaderN) r r r r5 r< r> rD r1 r� rd r� re r� rf r� rh r� rn rs r� rv r[ r� r� r: r� r� r� r r r r r� 3 s4
*r� Nc C sF |d u rt | d�r
| j}nd}|dv rt| �S |dv r t| �S td��)N�moder+ )�rr+ )�wr� z$mode must be 'r', 'rb', 'w', or 'wb')�hasattrr� r( r� r )r4 r� r r r r/ s
r/ r7 )� collectionsr r rJ rO rS rT rU rV 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_5rP rL rN � Exceptionr r r' r( r� r/ r r r r �<module> sD i�� Q