Added method to import measurement data from a wave file

This commit is contained in:
Anne de Jong 2020-07-09 09:45:08 +02:00
parent 82ec7faa45
commit c30960b1a3
2 changed files with 52 additions and 1 deletions

View File

@ -73,7 +73,7 @@ class AvStream:
rtaudio_inputparams = None rtaudio_inputparams = None
rtaudio_outputparams = None rtaudio_outputparams = None
self.nframes_per_block = 512 self.nframes_per_block = 1024
if self.duplex_mode or avtype == AvType.audio_output: if self.duplex_mode or avtype == AvType.audio_output:
rtaudio_outputparams = {'deviceid': device.index, rtaudio_outputparams = {'deviceid': device.index,

View File

@ -489,6 +489,8 @@ class Measurement:
firstcoltime: If true, the first column is the treated as the firstcoltime: If true, the first column is the treated as the
sample time. sample time.
""" """
if os.path.splitext(mfn)[1] != '.h5':
mfn += '.h5'
if os.path.exists(mfn): if os.path.exists(mfn):
raise ValueError(f'File {mfn} already exist.') raise ValueError(f'File {mfn} already exist.')
if timestamp is None: if timestamp is None:
@ -497,6 +499,13 @@ class Measurement:
if data.ndim != 2: if data.ndim != 2:
data = data[:, np.newaxis] data = data[:, np.newaxis]
try:
len(sensitivity)
except:
raise ValueError('Sensitivity should be given as array-like data type')
sensitivity = np.asarray(sensitivity)
nchannels = data.shape[1] nchannels = data.shape[1]
if nchannels != sensitivity.shape[0]: if nchannels != sensitivity.shape[0]:
raise ValueError( raise ValueError(
@ -514,3 +523,45 @@ class Measurement:
compression='gzip') compression='gzip')
ad[0] = data ad[0] = data
return Measurement(mfn) return Measurement(mfn)
@staticmethod
def fromWaveFile(fn, newfn=None, force=False, timestamp=None):
"""Convert a measurement file to a wave file, and return the
measurement handle."""
if timestamp is None:
timestamp = int(time.time())
base_fn = os.path.splitext(fn)[0]
if newfn is None:
newfn = base_fn + '.h5'
if os.path.exists(newfn) and not force:
raise RuntimeError(f'Measurement file name {newfn} already exists in path, set "force" to true to overwrite')
with wave.open(fn, 'r') as wf:
nchannels = wf.getnchannels()
samplerate = wf.getframerate()
sensitivity = np.ones(nchannels)
sampwidth = wf.getsampwidth()
nframes = wf.getnframes()
if sampwidth == 2:
dtype = np.int16
elif sampwidth == 1:
dtype = np.int8
elif sampwidth == 4:
dtype = np.int32
with h5.File(newfn, 'w') as hf:
hf.attrs['samplerate'] = samplerate
hf.attrs['nchannels'] = nchannels
hf.attrs['time'] = timestamp
hf.attrs['blocksize'] = 1
hf.attrs['sensitivity'] = sensitivity
data = np.frombuffer(
wf.readframes(nframes), dtype=dtype).reshape(nframes, nchannels)
ad = hf.create_dataset('audio', (1, data.shape[0], data.shape[1]),
dtype=dtype,
maxshape=(1, data.shape[0], data.shape[1]),
compression='gzip')
ad[0] = data
return Measurement(newfn)