o wiF@s2ddlZddlZddlmZdddZddZdS)NFc Cst|}|}|}|}|dkrd}n |dkrd}ntd||}dt|}t j |t |d} |dkrD| d|} |rL|| f|fS|| fS) Nuint8int16z%bytes_per_sample must be 1, 2, 4 or 8,)dtype) waveopen getframerate getnchannels getsampwidth ValueError readframes getnframeslennp frombufferrreshape) fd return_sizewdrate nchannelsnbytesrdatasizearrayrJ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/kaldiio/wavio.pyread_wavs"    r cCs|jtjkr d}n|jtjkrd}ntd|j|jdkr%|jd}n|jdkr-d}ntd|jt |}| || || || }||dt|S)NrrzNot Supported dtype {}z+Not Supported dimension: 0 or 1, but got {}r)rrrrrformatndimshaper Wave_write setnchannels setsampwidth setframeratetobytes writeframesr)rrr sampwidthrwrrrr write_wavs&            r,)F)numpyrr kaldiio.python_wave python_waver r,rrrrs