3
\-E @ s d dl Z dddddgZG dd deZdZdZd dlZd dlZd dlZd d l m
Z
d d
l m Z e ddZ
G dd dZG dd dZdddZeZdS ) NopenopenfpError Wave_read
Wave_writec @ s e Zd ZdS )r N)__name__
__module__ __qualname__ r
r
/usr/lib64/python3.6/wave.pyr N s bhi)Chunk)
namedtuple _wave_paramsz7nchannels sampwidth framerate nframes comptype compnamec @ 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+S ),r c C s d | _ d| _t|dd| _| jj dkr0td| jjddkrHtdd| _d | _xd| _ yt| jdd}W n t
k
r P Y nX |j }|d kr| j | d| _n2|d
kr| jstd || _|j | j
| _d| _ P |j qVW | j s| j r td d S )
Nr )Z bigendians RIFFz file does not start with RIFF id s WAVEznot a WAVE filer s fmt s datazdata chunk before fmt chunkz#fmt chunk and/or data chunk missing)_convert _soundposr _fileZgetnamer readZ_fmt_chunk_read _data_chunk_data_seek_neededEOFError_read_fmt_chunkZ chunksize
_framesize_nframesskip)selffilechunkZ chunknamer
r
r initfp} s:
zWave_read.initfpc C sR d | _ t|tr"tj|d}|| _ y| j| W n | j rF|j Y nX d S )Nrb)_i_opened_the_file
isinstancestrbuiltinsr r" close)r fr
r
r __init__ s
zWave_read.__init__c C s | j d S )N)r( )r r
r
r __del__ s zWave_read.__del__c C s | S )Nr
)r r
r
r __enter__ s zWave_read.__enter__c G s | j d S )N)r( )r argsr
r
r __exit__ s zWave_read.__exit__c C s | j S )N)r )r r
r
r getfp s zWave_read.getfpc C s d| _ d| _d S )Nr r )r r )r r
r
r rewind s zWave_read.rewindc C s" d | _ | j}|r d | _|j d S )N)r r$ r( )r r r
r
r r( s
zWave_read.closec C s | j S )N)r )r r
r
r tell s zWave_read.tellc C s | j S )N)
_nchannels)r r
r
r getnchannels s zWave_read.getnchannelsc C s | j S )N)r )r r
r
r
getnframes s zWave_read.getnframesc C s | j S )N)
_sampwidth)r r
r
r getsampwidth s zWave_read.getsampwidthc C s | j S )N)
_framerate)r r
r
r getframerate s zWave_read.getframeratec C s | j S )N) _comptype)r r
r
r getcomptype s zWave_read.getcomptypec C s | j S )N) _compname)r r
r
r getcompname s zWave_read.getcompnamec C s* t | j | j | j | j | j | j S )N)r r3 r6 r8 r4 r: r< )r r
r
r getparams s zWave_read.getparamsc C s d S )Nr
)r r
r
r
getmarkers s zWave_read.getmarkersc C s t dd S )Nzno marks)r )r idr
r
r getmark s zWave_read.getmarkc C s* |dk s|| j krtd|| _d| _d S )Nr zposition not in ranger )r r r r )r posr
r
r setpos s zWave_read.setposc C s | j r8| jjdd | j| j }|r2| jj|d d| _ |dkrDdS | jj|| j }| jdkrxtjdkrxt j
|| j}| j r|r| j |}| jt || j
| j | _|S )Nr r big)r r seekr r r r5 sys byteorderaudioopbyteswapr lenr2 )r nframesrA datar
r
r
readframes s
zWave_read.readframesc C sz t jd|jd\}| _| _}}|tkrNt jd|jdd }|d d | _ntd|f | j| j | _d | _ d
| _
d S ) Nz r@ rB rM r r
r
r
r r ^ s*
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 Zd3d4 Z d5d6 Z d7d8 Z d9S ):r c C sR d | _ t|tr"tj|d}|| _ y| j| W n | j rF|j Y nX d S )Nwb)r$ r% r& r' r r" r( )r r) r
r
r r* # s
zWave_write.__init__c C s@ || _ d | _d| _d| _d| _d| _d| _d| _d| _d| _ d S )Nr F)
r r r2 r5 r7 r _nframeswritten _datawritten _datalength_headerwritten)r r r
r
r r" / s zWave_write.initfpc C s | j d S )N)r( )r r
r
r r+ ; s zWave_write.__del__c C s | S )Nr
)r r
r
r r, > s zWave_write.__enter__c G s | j d S )N)r( )r r- r
r
r r. A s zWave_write.__exit__c C s( | j rtd|dk r td|| _d S )Nz0cannot change parameters after starting to writer zbad # of channels)rX r r2 )r nchannelsr
r
r setnchannelsG s
zWave_write.setnchannelsc C s | j std| j S )Nznumber of channels not set)r2 r )r r
r
r r3 N s zWave_write.getnchannelsc C s0 | j rtd|dk s |dkr&td|| _d S )Nz0cannot change parameters after starting to writer r zbad sample width)rX r r5 )r rU r
r
r setsampwidthS s
zWave_write.setsampwidthc C s | j std| j S )Nzsample width not set)r5 r )r r
r
r r6 Z s zWave_write.getsampwidthc C s0 | j rtd|dkr tdtt|| _d S )Nz0cannot change parameters after starting to writer zbad frame rate)rX r introundr7 )r framerater
r
r setframerate_ s
zWave_write.setframeratec C s | j std| j S )Nzframe rate not set)r7 r )r r
r
r r8 f s zWave_write.getframeratec C s | j rtd|| _d S )Nz0cannot change parameters after starting to write)rX r r )r rK r
r
r
setnframesk s zWave_write.setnframesc C s | j S )N)rW )r r
r
r r4 p s zWave_write.getnframesc C s. | j rtd|dkr td|| _|| _d S )Nz0cannot change parameters after starting to writerR z unsupported compression type)rR )rX r r9 r; )r comptypecompnamer
r
r setcomptypes s zWave_write.setcomptypec C s | j S )N)r9 )r r
r
r r: { s zWave_write.getcomptypec C s | j S )N)r; )r r
r
r r< ~ s zWave_write.getcompnamec C sV |\}}}}}}| j r td| j| | j| | j| | j| | j|| d S )Nz0cannot change parameters after starting to write)rX r r\ r] ra rb re )r Zparamsr[ rU r` rK rc rd r
r
r setparams s
zWave_write.setparamsc C s> | j s| j s| j r tdt| j | j| j| j| j| jS )Nznot all parameters set)r2 r5 r7 r r r r9 r; )r r
r
r r= s zWave_write.getparamsc C s t dd S )Nzsetmark() not supported)r )r r? rA namer
r
r setmark s zWave_write.setmarkc C s t dd S )Nzno marks)r )r r? r
r
r r@ s zWave_write.getmarkc C s d S )Nr
)r r
r
r r> s zWave_write.getmarkersc C s | j S )N)rW )r r
r
r r1 s zWave_write.tellc C s t |ttfs t|jd}| jt| t|| j| j }| j rN| j |}| jdkrpt
j dkrpt j
|| j}| jj| | jt|7 _| j| | _d S )NBr rD )r% bytes bytearray
memoryviewcast_ensure_header_writtenrJ r5 r2 r rF rG rH rI r writerX rW )r rL rK r
r
r writeframesraw s
zWave_write.writeframesrawc C s" | j | | j| jkr | j d S )N)rp rY rX _patchheader)r rL r
r
r writeframes s
zWave_write.writeframesc C sX z2| j r0| jd | j| jkr&| j | j j W d d | _ | j}|rRd | _|j X d S )Nr )r rn rY rX rq flushr$ r( )r r r
r
r r( s
zWave_write.closec C s> | j s:| jstd| js"td| js0td| j| d S )Nz# channels not specifiedzsample width not specifiedzsampling rate not specified)rZ r2 r r5 r7
_write_header)r Zdatasizer
r
r rn s z!Wave_write._ensure_header_writtenc C s | j jd | js$|| j| j | _| j| j | j | _y| j j | _W n tt fk
rf d | _Y nX | j jt
j dd| j dddt | j| j
| j| j
| j | j| j | jd d | jd k r| j j | _| j jt
j d | j d
| _d S ) Ns RIFFz r1 rp rr r( rn rt rq r
r
r
r r s8
c C sJ |d kr t | dr| j}nd}|dkr.t| S |dkr>t| S tdd S ) Nmoder# rwrV z$mode must be 'r', 'rb', 'w', or 'wb')r} r# )r~ rV )hasattrr| r r r )r) r| r
r
r r s
)Nr
r Nr )N)r' __all__ Exceptionr rT Z _array_fmtsrH rS rF r! r collectionsr r r r r r r
r
r
r J s" , d