a
R�f] � @ s^ d dl Z d dlZd dlZddgZG dd� de�ZdZdd� Zdd � Zd
d � Z d d
� Z
dd� Z dZ dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d lmZ d d lmZ ed!d"�Zd#ej_d$ej_d%ej _d&ej _d'ej _d(ej _G d)d*� d*�Z G d+d,� d,�Z!d>d-d�Z"e#d.k�rZd dl$Z$e$j%d/d� �s>e$j%�&d0� e$j%d/ Z'e"e'd1���Z(e)d2e'� e)d3e(�*� � e)d4e(�+� � e)d5e(�,� � e)d6e(�-� � e)d7e(�.� � e)d8e(�/� � e$j%d9d� �r:e$j%d9 Z0e)d:e0� e"e0d;��@Z1e1�2e(�3� � e(�4d<�Z5e5�s�qe1�6e5� �q�W d � n1 �s(0 Y e)d=� W d � n1 �sP0 Y dS )?� N�Error�openc @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r �)/opt/alt/python39/lib64/python3.9/aifc.pyr � s l @Q E c C s8 zt �d| �d��d W S t jy2 td �Y n0 d S )N�>l� r ��structZunpack�read�error�EOFError��filer r r �
_read_long� s r c C s8 zt �d| �d��d W S t jy2 td �Y n0 d S )N�>Lr
r r r r r r � _read_ulong� s r c C s8 zt �d| �d��d W S t jy2 td �Y n0 d S )N�>h� r r r r r r � _read_short� s r c C s8 zt �d| �d��d W S t jy2 td �Y n0 d S )N�>Hr r r r r r r � _read_ushort� s r c C s@ t | �d��}|dkr d}n
| �|�}|d@ dkr<| �d�}|S )N� r � )�ordr
)r �length�data�dummyr r r � _read_string� s
r g�����c C s� t | �}d}|dk r d}|d }t| �}t| �}|| krN| krNdkrXn nd} n0|dkrft} n"|d }|d | td |d
� } || S ) Nr r ���� � g � i�? l g @�? )r r � _HUGE_VAL�pow)�f�expon�sign�himant�lomantr r r � _read_float� s "r, c C s | � t�d|�� d S )Nr ��writer �pack�r' �xr r r � _write_short� s r2 c C s | � t�d|�� d S )Nr r- r0 r r r �
_write_ushort� s r3 c C s | � t�d|�� d S )Nr r- r0 r r r � _write_long� s r4 c C s | � t�d|�� d S )Nr r- r0 r r r � _write_ulong� s r5 c C sR t |�dkrtd��| �t�dt |��� | �|� t |�d@ dkrN| �d� d S )N� z%string exceeds maximum pstring length�Br r � )�len�
ValueErrorr. r r/ )r' �sr r r �
_write_string� s
r< c C s� dd l }|dk r d}|d }nd}|dkr8d}d}d}n�|�|�\}}|dks^|dks^||krp|dB }d}d}nh|d }|dk r�|�||�}d}||B }|�|d�}|�|�}t|�}|�|| d�}|�|�}t|�}t| |� t| |� t| |� d S ) Nr r" r! i @ r r# i�? � )�mathZfrexpZldexpZfloor�intr3 r5 ) r' r1 r> r) r( r* r+ ZfmantZfsmantr r r � _write_float� s8
r@ )�Chunk)�
namedtuple� _aifc_paramsz7nchannels sampwidth framerate nframes comptype compnamez3Number of audio channels (1 for mono, 2 for stereo)zSample width in byteszSampling frequencyzNumber of audio framesz(Compression type ("NONE" for AIFF files)zRA human-readable version of the compression type
('not compressed' for AIFF files)c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � 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.d/� Zd0d1� ZdS )2� Aifc_readNc C s6 d| _ d | _g | _d| _|| _t|�}|�� dkr:td��|�d�}|dkrTd| _ n|dkrdd| _ ntd��d| _
d | _ d| _ zt| j�}W n t
y� Y �qY n0 |�� }|d kr�| �|� d| _
nH|d
kr�|| _ |�d �}d| _ n(|d kr�t|�| _ n|d
k�r| �|� |�� qx| j
�r*| j �s2td��d S )Nr � FORMz file does not start with FORM idr
� AIFF� AIFCr znot an AIFF or AIFF-C file� COMM� SSND� � FVER� MARKz$COMM chunk and/or SSND chunk missing)�_version�_convert�_markers� _soundpos�_filerA Zgetnamer r
�_aifcZ_comm_chunk_read� _ssnd_chunk�_ssnd_seek_neededr �_read_comm_chunkr � _readmark�skip)�selfr �chunkZformdataZ chunknamer r r r �initfp4 sH
zAifc_read.initfpc C sL t |t�r>t�|d�}z| �|� W qH |�� � Y qH0 n
| �|� d S )N�rb)�
isinstance�str�builtinsr rZ �close�rX r' Z file_objectr r r �__init__\ s
zAifc_read.__init__c C s | S �Nr �rX r r r � __enter__h s zAifc_read.__enter__c G s | � � d S rb �r_ �rX �argsr r r �__exit__k s zAifc_read.__exit__c C s | j S rb )rQ rc r r r �getfpq s zAifc_read.getfpc C s d| _ d| _d S )Nr r )rT rP rc r r r �rewindt s zAifc_read.rewindc C s | j }|d ur d | _ |�� d S rb )rQ r_ �rX r r r r r_ x s zAifc_read.closec C s | j S rb )rP rc r r r �tell~ s zAifc_read.tellc C s | j S rb )�
_nchannelsrc r r r � getnchannels� s zAifc_read.getnchannelsc C s | j S rb )�_nframesrc r r r �
getnframes� s zAifc_read.getnframesc C s | j S rb )�
_sampwidthrc r r r � getsampwidth� s zAifc_read.getsampwidthc C s | j S rb )�
_frameraterc r r r � getframerate� s zAifc_read.getframeratec C s | j S rb �� _comptyperc r r r � getcomptype� s zAifc_read.getcomptypec C s | j S rb �� _compnamerc r r r � getcompname� s zAifc_read.getcompnamec C s* t | �� | �� | �� | �� | �� | �� �S rb )rC rn rr rt rp rw rz rc r r r � getparams� s �zAifc_read.getparamsc C s t | j�dkrd S | jS �Nr �r9 rO rc r r r �
getmarkers� s zAifc_read.getmarkersc C s2 | j D ]}||d kr| S qtd�|���d S �Nr zmarker {0!r} does not exist�rO r �format�rX �id�markerr r r �getmark� s
zAifc_read.getmarkc C s* |dk s|| j krtd��|| _d| _d S )Nr zposition not in ranger )ro r rP rT )rX �posr r r �setpos� s zAifc_read.setposc C s� | j rD| j�d� | j�d�}| j| j }|r>| j�|d � d| _ |dkrPdS | j�|| j �}| jrv|rv| �|�}| jt|�| j| j | _|S )Nr rJ r )
rT rS �seekr
rP �
_framesizerN r9 rm rq )rX �nframesr r� r r r r �
readframes� s
�
zAifc_read.readframesc C s dd l }|�|d�S �Nr r )�audioopZalaw2lin�rX r r� r r r � _alaw2lin� s zAifc_read._alaw2linc C s dd l }|�|d�S r� )r� Zulaw2linr� r r r � _ulaw2lin� s zAifc_read._ulaw2linc C s2 dd l }t| d�sd | _|�|d| j�\}| _|S �Nr � _adpcmstater )r� �hasattrr� Z adpcm2linr� r r r �
_adpcm2lin� s
zAifc_read._adpcm2linc C sV t |�| _t|�| _t |�d d | _tt|��| _| jdkrFtd��| jdkrXtd��| j| j | _ | j
�rFd}|j dkr�d}t �
d� d |_ |�d
�| _|r�t|j�d��}|d@ dkr�|d }|j | |_ |j�d d� t|�| _| jd k�rR| jd
k�r
| j| _n4| jdv �r | j| _n | jdv �r6| j| _ntd��d| _n d | _d| _d S )N� rJ r �bad sample width�bad # of channels� r z Warning: bad COMM chunk size� r
r! � NONE� G722�� ulaw� ULAW�� alaw� ALAW� unsupported compression typer � not compressed)r rm r ro rq r? r, rs r r� rR Z chunksize�warnings�warnr
rv r r r� r ry r� rN r� r� )rX rY Zkludger r r r rU � sD
zAifc_read._read_comm_chunkc C s� t |�}zDt|�D ]6}t |�}t|�}t|�}|s6|r| j�|||f� qW nB ty� dt| j�t| j�dkrvdnd|f }t� |� Y n0 d S )Nz;Warning: MARK chunk contains only %s marker%s instead of %sr � r; )
r �ranger r rO �appendr r9 r� r� )rX rY Znmarkers�ir� r� �name�wr r r rV � s ��zAifc_read._readmark) r r r rQ rZ ra rd rh ri rj r_ rl rn rp rr rt rw rz r{ r~ r� r� r� r� r� r� rU rV r r r r rD s2 $( *rD c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � 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.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dS )J�
Aifc_writeNc C s\ t |t�rNt�|d�}z| �|� W n |�� � Y n0 |�d�rXd| _n
| �|� d S )N�wbz.aiffr )r\ r] r^ r rZ r_ �endswithrR r` r r r ra / s
zAifc_write.__init__c C s^ || _ t| _d| _d| _d | _d| _d| _d| _d| _ d| _
d| _ d| _ g | _
d| _d| _d S )Nr� r� r r )rQ �
_AIFC_versionrM rv ry rN rm rq rs ro �_nframeswritten� _datawritten� _datalengthrO � _marklengthrR rk r r r rZ ? s zAifc_write.initfpc C s | � � d S rb re rc r r r �__del__P s zAifc_write.__del__c C s | S rb r rc r r r rd S s zAifc_write.__enter__c G s | � � d S rb re rf r r r rh V s zAifc_write.__exit__c C s | j rtd��d| _d S )N�0cannot change parameters after starting to writer �r� r rR rc r r r �aiff\ s zAifc_write.aiffc C s | j rtd��d| _d S )Nr� r r� rc r r r �aifca s zAifc_write.aifcc C s( | j rtd��|dk r td��|| _d S )Nr� r r� )r� r rm )rX � nchannelsr r r � setnchannelsf s
zAifc_write.setnchannelsc C s | j std��| j S )Nznumber of channels not set)rm r rc r r r rn m s zAifc_write.getnchannelsc C s0 | j rtd��|dk s |dkr&td��|| _d S )Nr� r r
r� )r� r rq )rX � sampwidthr r r � setsampwidthr s
zAifc_write.setsampwidthc C s | j std��| j S )Nzsample width not set)rq r rc r r r rr y s zAifc_write.getsampwidthc C s( | j rtd��|dkr td��|| _d S )Nr� r zbad frame rate)r� r rs )rX � framerater r r � setframerate~ s
zAifc_write.setframeratec C s | j std��| j S )Nzframe rate not set)rs r rc r r r rt � s zAifc_write.getframeratec C s | j rtd��|| _d S )Nr� )r� r ro )rX r� r r r �
setnframes� s zAifc_write.setnframesc C s | j S rb �r� rc r r r rp � s zAifc_write.getnframesc C s. | j rtd��|dvr td��|| _|| _d S �Nr� )r� r� r� r� r� r� r� )r� r rv ry )rX �comptype�compnamer r r � setcomptype� s zAifc_write.setcomptypec C s | j S rb ru rc r r r rw � s zAifc_write.getcomptypec C s | j S rb rx rc r r r rz � s zAifc_write.getcompnamec C sf |\}}}}}}| j r td��|dvr.td��| �|� | �|� | �|� | �|� | �||� d S r� )r� r r� r� r� r� r� )rX �paramsr� r� r� r� r� r� r r r � setparams� s
zAifc_write.setparamsc C s8 | j r| jr| jstd��t| j | j| j| j| j| j�S )Nznot all parameters set)rm rq rs r rC ro rv ry rc r r r r{ � s
�zAifc_write.getparamsc C s� |dkrt d��|dk r t d��t|t�s2t d��tt| j��D ],}|| j| d kr@|||f| j|< d S q@| j�|||f� d S )Nr zmarker ID must be > 0z marker position must be >= 0zmarker name must be bytes)r r\ �bytesr� r9 rO r� )rX r� r� r� r� r r r �setmark� s
zAifc_write.setmarkc C s2 | j D ]}||d kr| S qtd�|���d S r r� r� r r r r� � s
zAifc_write.getmarkc C s t | j�dkrd S | jS r| r} rc r r r r~ � s zAifc_write.getmarkersc C s | j S rb r� rc r r r rl � s zAifc_write.tellc C sz t |ttf�s t|��d�}| �t|�� t|�| j| j }| j rN| � |�}| j
� |� | j | | _ | j
t|� | _
d S )Nr7 )r\ r� � bytearray�
memoryview�cast�_ensure_header_writtenr9 rq rm rN rQ r. r� r� )rX r r� r r r �writeframesraw� s
zAifc_write.writeframesrawc C s. | � |� | j| jks"| j| jkr*| �� d S rb )r� r� ro r� r� � _patchheader)rX r r r r � writeframes� s
�zAifc_write.writeframesc C s� | j d u rd S zx| �d� | jd@ r<| j �d� | jd | _| �� | j| jksb| j| jksb| jrj| � � W d | _
| j }d | _ |� � n d | _
| j }d | _ |� � 0 d S )Nr r r8 ) rQ r� r� r. �
_writemarkersr� ro r� r� r� rN r_ )rX r' r r r r_ � s,
��
�zAifc_write.closec C s dd l }|�|d�S r� )r� Zlin2alawr� r r r � _lin2alaw� s zAifc_write._lin2alawc C s dd l }|�|d�S r� )r� Zlin2ulawr� r r r � _lin2ulaw s zAifc_write._lin2ulawc C s2 dd l }t| d�sd | _|�|d| j�\}| _|S r� )r� r� r� Z lin2adpcmr� r r r �
_lin2adpcm s
zAifc_write._lin2adpcmc C sf | j sb| jdv r.| js d| _| jdkr.td��| js<td��| jsJtd��| jsXtd��| �|� d S )N�r� r� r� r� r� r zRsample width must be 2 when compressing with ulaw/ULAW, alaw/ALAW or G7.22 (ADPCM)z# channels not specifiedzsample width not specifiedzsampling rate not specified)r� rv rq r rm rs �
_write_header)rX Zdatasizer r r r�
s
z!Aifc_write._ensure_header_writtenc C s> | j dkr| j| _n&| j dv r(| j| _n| j dv r:| j| _d S )Nr� r� r� )rv r� rN r� r� rc r r r �_init_compression s
z Aifc_write._init_compressionc C sH | j r| jdkr| �� | j�d� | js<|| j| j | _| j| j | j | _| jd@ rf| jd | _| j r�| jdv r�| jd | _| jd@ r�| jd | _n0| jdkr�| jd d | _| jd@ r�| jd | _z| j� � | _
W n t t fy� d | _
Y n0 | �
| j�}| j �r@| j�d � | j�d
� t| jd� t| j| j� n | j�d � | j�d � t| j|� t| j| j� | j
d u�r�| j� � | _t| j| j� | jd
v �r�t| jd� nt| j| jd � t| j| j� | j �r�| j�| j� t| j| j� | j�d� | j
d u�r| j� � | _t| j| jd � t| jd� t| jd� d S )Nr� rE r )r� r� r� r� r r� � r
rG rK rF rH r� rJ rI r )rR rv r� rQ r. ro rm rq r� rl �_form_length_pos�AttributeError�OSError�_write_form_lengthr5 rM r2 � _nframes_posr@ rs r< ry �_ssnd_length_pos)rX Z
initlength�
commlengthr r r r� % s^
zAifc_write._write_headerc C s\ | j r*dt| j� }|d@ r$|d }d}nd}d}t| jd| | j d | d | � |S ) Nr� r � r� r r
rJ � )rR r9 ry r5 rQ r� )rX �
datalengthr� Z
verslengthr r r r� X s" ����z Aifc_write._write_form_lengthc C s� | j �� }| jd@ r,| jd }| j �d� n| j}|| jkrd| j| jkrd| jdkrd| j �|d� d S | j �| j d� | �
|�}| j �| j d� t | j | j� | j �| j
d� t | j |d � | j �|d� | j| _|| _d S )Nr r8 r rJ )rQ rl r� r. r� ro r� r� r� r� r� r� r5 r� )rX Zcurposr� r r r r r� e s*
��
zAifc_write._patchheaderc C s� t | j�dkrd S | j�d� d}| jD ]:}|\}}}|t |� d d }t |�d@ dkr(|d }q(t| j|� |d | _t| jt | j�� | jD ]2}|\}}}t| j|� t| j|� t| j|� q�d S )Nr rL r r � rJ )r9 rO rQ r. r5 r� r2 r< )rX r r� r� r� r� r r r r� { s"
zAifc_write._writemarkers)(r r r rQ ra rZ r� rd rh r� r� r� rn r� rr r� rt r� rp r� rw rz r� r{ r� r� r~ rl r� r� r_ r� r� r� r� r� r� r� r� r� r r r r r� sJ
3
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[ )r� r� z$mode must be 'r', 'rb', 'w', or 'wb')r� r� rD r� r )r' r� r r r r � s
�__main__r z /usr/demos/data/audio/bach.aiffr� ZReadingz nchannels =z nframes =z sampwidth =z framerate =z comptype =z compname =r ZWritingr� i zDone.)N)7r r^ r� �__all__� Exceptionr r� r r r r r r% r, r2 r3 r4 r5 r< r@ rY rA � collectionsrB rC r� �__doc__r� r� r� r� r� rD r� r r �sys�argvr� �fnr' �printrn rp rr rt rw rz Zgn�gr� r{ r� r r� r r r r �<module>� sx
! �
.