B
: @ s d dl mZ d dlZeddZdZdZdZdZd Zd
Z d Z
d Z d
Z dZ
dZdZdZdZeeeee egZG dd deZdd Zdd ZG dd dZG 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 */opt/alt/python37/lib64/python3.7/sunau.pyr s r c C s< d}x2t dD ]&}| d}|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\}}|dt| |}qW | t| 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}||d}d| _nd| _| | d S )N r rbTF)typebuiltinsopen_openedinitfp)selffr, r r r __init__ s zAu_read.__init__c C s | j r| 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 | 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|| _| jsLtd
| j| j | _| jdkr|| jd | _| jd\| _}}nd| _y| | _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 encodingzbad # of channels ) 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/ sT
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 | | | | | | 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 | j }n| j|| j }| jt|| j 7 _| j tkrhdd l }|
|| 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| j| 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|| kr td| jd kr.td| j| 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 | 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}||d}d| _nd| _| | d S )Nr) r wbTF)r+ r, r- r. r/ )r0 r1 r, r r r r2 5 s zAu_write.__init__c C s | j r| 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 C s zAu_write.__enter__c G s | d S )N)r4 )r0 r7 r r r r8 F 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/ I s zAu_write.initfpc C s( | j rtd|dkr td|| _d S )Nz0cannot change parameters after starting to write)r r r z"only 1, 2, or 4 channels supported)rm r rK )r0 nchannelsr r r setnchannelsV s
zAu_write.setnchannelsc C s | j std| j S )Nznumber of channels not set)rK r )r0 r r r rU ] s zAu_write.getnchannelsc C s( | j rtd|dkr td|| _d S )Nz0cannot change parameters after starting to write)r r r r zbad sample width)rm r rD )r0 sampwidthr r r setsampwidthb s
zAu_write.setsampwidthc C s | j std| jS )Nzsample width not specified)rJ r rD )r0 r r r rV i s zAu_write.getsampwidthc C s | j rtd|| _d S )Nz0cannot change parameters after starting to write)rm r rJ )r0 framerater r r setframeraten s zAu_write.setframeratec C s | j std| j S )Nzframe rate not set)rJ r )r0 r r r rW s 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
setnframesx s
zAu_write.setnframesc C s | j S )N)rm )r0 r r r rX s zAu_write.getnframesc C s |dkr|| _ ntdd S )N)r[ rY zunknown compression type)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 |\}}}}}}| | | | | | | | | || d S )N)rr rt rv rw ry )r0 paramsrq rs ru rd ZcomptypeZcompnamer r r setparams s
zAu_write.setparamsc C s* t | | | | | | 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|d}| | jdkrDdd l}||| j }t
|| 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| jks"| j| jkr*| d S )N)r rm rl ro rn _patchheader)r0 r$ r r r writeframes s
zAu_write.writeframesc C s^ | j rZz6| | j| jks(| j| jkr0| | j W d | j }d | _ | jrX| 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| 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 | _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 )
Nr[ r r r r zinternal errorrY r
r
ir: r )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_sizelengthr r r r sJ
zAu_write._write_headerc C sH | j d krtd| j| j t| j| j | j| _| jdd 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] r{ r^ rN r r r4 r r r r r r r rj 3 s2
*rj c C sJ |d kr t | dr| j}nd}|dkr.t| S |dkr>t| S tdd S )Nmoder* )rr* )wrk z$mode must be 'r', 'rb', 'w', or 'wb')hasattrr r( rj r )r1 r r r r r- s
r- c C s t jdtdd t| |dS )NzDsunau.openfp is deprecated since Python 3.7. Use sunau.open instead.r )
stacklevel)r )warningswarnDeprecationWarningr- )r1 r r r r openfp s
r )N)N) collectionsr r 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- r r r r r j s@ Q