3
\F @ s d dl mZ eddZdZdZdZdZdZd Zd
Z d Z
d Z d
Z dZ
dZdZdZeeeeeegZG dd deZdd Zdd ZG dd dZG dd dZd 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 dD ]&}| jd}|s$t|d t| }qW |S )Nr r r )rangereadEOFErrorord)filexiZbyter r r _read_u32 s
r c C sJ g }x2t dD ]&}t|d\}}|jdt| |}qW | jt| d S )Nr r r )r divmodinsertintwritebytes)r r datar dmr 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 dkr,dd l}|j|d}d| _nd| _| j| d S )N r rbTF)typebuiltinsopen_openedinitfp)selffr, r r r __init__ s zAu_read.__init__c C s | j r| j d S )N)_fileclose)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(tdtt|| _| jdk rHtd| jdkrZtdt|| _| jtkrzt| j| _tt|| _ | j t
krtd| j t t fkrd| _
d | _nj| j tkrd | _| _
nR| j tkrd | _| _
n:| j tkrd
| _| _
n"| j tkrd | _| _
ntd tt|| _tt|| _| j| j | _| jdkr~|j| jd | _| jjd
\| _}}nd| _y|j | _W n ttfk
r d | _Y nX d S )Nr zbad magic numberr zheader size too smalld z header size ridiculously largez encoding not (yet) supportedr r r r zunknown encoding ) r3 _soundposr! r AUDIO_FILE_MAGICr Z _hdr_size
_data_sizeAUDIO_UNKNOWN_SIZE _encoding_simple_encodingsAUDIO_FILE_ENCODING_MULAW_8AUDIO_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 partitiontell _data_posAttributeErrorOSError)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 )NULAWALAWNONE)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 dd 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 )4Au_writec C s@ t |t dkr,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|| _ ntdd 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 |ttfs t|jd}| j | jdkrDdd l}|j|| j }t
|| j }| j j
| | j| | _| jt
| | _d S )NBrY r )
isinstancer# bytearray
memoryviewcast_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| _qtdn | j dkrt}d| _ntd| j| j | _t
| j t dt
| j }|d d @ }t
| j | | jtkrt}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 | dr| j}nd}|dkr.t| S |dkr>t| S tdd S ) Nmoder* rwrk 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