U
ifG @ 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/python38/lib64/python3.8/sunau.pyr s r c C s8 d}t dD ]&}| d}|s"t|d t| }q |S )Nr r r )rangereadEOFErrorord)filexiZbyter r r _read_u32 s
r c C sF g }t dD ]&}t|d\}}|dt| |}q | 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 rbTFtypebuiltinsopen_openedinitfpselffr- r r r __init__ s zAu_read.__init__c C s | j r| d S N_filecloser2 r r r __del__ s zAu_read.__del__c C s | S r5 r r9 r r r __enter__ s zAu_read.__enter__c G s | d S r5 r8 r2 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| _z| | _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 ) r7 _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)r2 r magic_r r r r0 sV
zAu_read.initfpc C s | j S r5 )r7 r9 r r r getfp s z
Au_read.getfpc C s | j S r5 )rR r9 r r r getnchannels s zAu_read.getnchannelsc C s | j S r5 )rK r9 r r r getsampwidth s zAu_read.getsampwidthc C s | j S r5 )rQ r9 r r r getframerate s zAu_read.getframeratec C s( | j tkrtS | jtkr$| j | j S dS Nr )rE rF rG rH rL r9 r r r
getnframes s
zAu_read.getnframesc C s$ | j tkrdS | j tkr dS dS d S )NULAWALAWNONErG rI rJ r9 r r r getcomptype s
zAu_read.getcomptypec C s$ | j tkrdS | j tkr dS dS d S )NCCITT G.711 u-lawCCITT G.711 A-lawnot compressedrd r9 r r r getcompname s
zAu_read.getcompnamec C s* t | | | | | | S r5 r r\ r] r^ r` re ri r9 r r r getparams s zAu_read.getparamsc C s d S r5 r r9 r r r
getmarkers s zAu_read.getmarkersc C s t dd S )Nzno marks)r )r2 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 r_ ) rG rH rF r7 r rL rC lenrI audioopZulaw2linrK )r2 nframesr$ rp r r r
readframes s
zAu_read.readframesc C s* | j d krtd| j| j d| _d S )N cannot seekr )rV rX r7 seekrC r9 r r r rewind s
zAu_read.rewindc C s | j S r5 )rC r9 r r r rU ! s z Au_read.tellc C sP |dk s|| kr td| jd kr.td| j| j|| j || _d S )Nr zposition not in rangers )r` r rV rX r7 rt rL rC )r2 posr r r setpos$ s
zAu_read.setposc C s" | j }|r d | _ | jr | d S r5 )r7 r/ r8 r2 r r r r r8 , s
z
Au_read.closeN)r r r r4 r: r; r? r0 r[ r\ r] r^ r` re ri rk rl rn rr ru rU rw r8 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 wbTFr+ r1 r r r r4 5 s zAu_write.__init__c C s | j r| d | _ d S r5 r6 r9 r r r r: > s zAu_write.__del__c C s | S r5 r r9 r r r r; C s zAu_write.__enter__c G s | d S r5 r< r= r r r r? F s zAu_write.__exit__c C sF || _ d| _d| _d| _d| _t| _d| _d| _d| _ d| _
d| _ d S )Nr rB ra ) r7 rQ rR rK rL rF _nframes_nframeswritten _datawritten _datalengthrS _comptyperx r r r r0 I s zAu_write.initfpc C s( | j rtd|dkr td|| _d S )N0cannot change parameters after starting to write)r r r z"only 1, 2, or 4 channels supported)r| r rR )r2 nchannelsr r r setnchannelsV s
zAu_write.setnchannelsc C s | j std| j S )Nznumber of channels not set)rR r r9 r r r r\ ] s zAu_write.getnchannelsc C s( | j rtd|dkr td|| _d S )Nr )r r r r zbad sample width)r| r rK )r2 sampwidthr r r setsampwidthb s
zAu_write.setsampwidthc C s | j std| jS )Nsample width not specified)rQ r rK r9 r r r r] i s zAu_write.getsampwidthc C s | j rtd|| _d S )Nr )r| r rQ )r2 framerater r r setframeraten s zAu_write.setframeratec C s | j std| j S )Nzframe rate not set)rQ r r9 r r r r^ s s zAu_write.getframeratec C s( | j rtd|dk r td|| _d S )Nr r z # of frames cannot be negative)r| r r{ )r2 rq r r r
setnframesx s
zAu_write.setnframesc C s | j S r5 r| r9 r r r r` s zAu_write.getnframesc C s |dkr|| _ ntdd S )N)rc ra zunknown compression type)r r )r2 r, namer r r setcomptype s zAu_write.setcomptypec C s | j S r5 r r9 r r r re s zAu_write.getcomptypec C s$ | j dkrdS | j dkr dS dS d S )Nra rf rb rg rh r r9 r r r ri s
zAu_write.getcompnamec C sH |\}}}}}}| | | | | | | | | || d S r5 )r r r r r )r2 Zparamsr r r rq ZcomptypeZcompnamer r r setparams s
zAu_write.setparamsc C s* t | | | | | | S r5 rj r9 r r r rk s zAu_write.getparamsc C s | j S r5 r r9 r r r rU s z
Au_write.tellc C s~ t |ttfs t|d}| | jdkrDdd l}||| j }t
|| j }| j
| | j| | _| jt
| | _d S )NBra r )
isinstancer# bytearray
memoryviewcast_ensure_header_writtenr rp Zlin2ulawrK ro rL r7 r" r| r} )r2 r$ rp rq r r r writeframesraw s
zAu_write.writeframesrawc C s. | | | j| jks"| j| jkr*| d S r5 )r r| r{ r~ r} _patchheader)r2 r$ r r r writeframes s
zAu_write.writeframesc C s^ | j rZz6| | j| jks(| j| jkr0| | j W 5 | j }d | _ | jrX| X d S r5 )
r7 r/ r8 r r| r{ r~ r} r flushrx r r r r8 s
zAu_write.closec C s< | j s8| jstd| js"td| js0td| d S )Nz# of channels not specifiedr zframe rate not specified)r| rR r rK rQ
_write_headerr9 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 }z| 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 )
Nrc r r r r zinternal errorra r
r
irA r )r rK rM rL rN rO rP r rI rR r' r7 rD ro rS r{ rF rU _form_length_posrW rX r~ rQ r" )r2 encoding header_sizeZlengthr r r r sJ
zAu_write._write_headerc C sH | j d krtd| j| j t| j| j | j| _| jdd d S )Nrs r r )r rX r7 rt r' r} r~ r9 r r r r s
zAu_write._patchheaderN) r r r r4 r: r; r? r0 r r\ r r] r r^ r r` r re ri r rk rU r r r8 r r r r r r r ry 3 s2
*ry c C sJ |d kr t | dr| j}nd}|dkr.t| S |dkr>t| S tdd S )Nmoder* )rr* )wrz z$mode must be 'r', 'rb', 'w', or 'wb')hasattrr r( ry r r3 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. r r r r openfp s
r )N)N) collectionsr r r rD rI rM rN rO rP 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_5rJ rF rH Exceptionr r r' r( ry r. r r r r r