#! /usr/bin/env python
+import os
+import re
+import glob
+import numpy as np
+from tempfile import mkstemp
+
+DEFAULT_SOUND = '22050Hz_5s_brownnoise.wav'
+
def array_from_text_file(filename, dtype = 'float'):
- import os.path
- from numpy import array
- filename = os.path.join(os.path.dirname(__file__), filename)
- return array([line.split() for line in open(filename).readlines()],
- dtype = dtype)
+ realpathname = os.path.join(os.path.dirname(__file__), filename)
+ return np.loadtxt(realpathname, dtype = dtype)
def list_all_sounds(rel_dir):
- import os.path, glob
datadir = os.path.join(os.path.dirname(__file__), rel_dir)
return glob.glob(os.path.join(datadir,'*.*'))
+
+def get_default_test_sound(TestCase, rel_dir = 'sounds'):
+ all_sounds = list_all_sounds(rel_dir)
+ if len(all_sounds) == 0:
+ TestCase.skipTest("please add some sounds in \'python/tests/sounds\'")
+ else:
+ default_sound = all_sounds[0]
+ if DEFAULT_SOUND in map(os.path.basename, all_sounds):
+ while os.path.basename(default_sound) != DEFAULT_SOUND:
+ default_sound = all_sounds.pop(0)
+ return default_sound
+
+def get_tmp_sink_path():
+ fd, path = mkstemp()
+ os.close(fd)
+ return path
+
+def del_tmp_sink_path(path):
+ try:
+ os.unlink(path)
+ except WindowsError as e:
+ # removing the temporary directory sometimes fails on windows
+ import warnings
+ errmsg = "failed deleting temporary file {:s} ({:s})"
+ warnings.warn(UserWarning(errmsg.format(path, repr(e))))
+
+def array_from_yaml_file(filename):
+ import yaml
+ f = open(filename)
+ yaml_data = yaml.safe_load(f)
+ f.close()
+ return yaml_data
+
+def count_samples_in_file(file_path):
+ from aubio import source
+ hopsize = 256
+ s = source(file_path, 0, hopsize)
+ total_frames = 0
+ while True:
+ _, read = s()
+ total_frames += read
+ if read < hopsize: break
+ return total_frames
+
+def count_samples_in_directory(samples_dir):
+ total_frames = 0
+ for f in os.walk(samples_dir):
+ if len(f[2]):
+ for each in f[2]:
+ file_path = os.path.join(f[0], each)
+ if file_path:
+ total_frames += count_samples_in_file(file_path)
+ return total_frames
+
+def count_files_in_directory(samples_dir):
+ total_files = 0
+ for f in os.walk(samples_dir):
+ if len(f[2]):
+ for each in f[2]:
+ file_path = os.path.join(f[0], each)
+ if file_path:
+ total_files += 1
+ return total_files
+
+def parse_file_samplerate(soundfile):
+ samplerate = None
+ # parse samplerate
+ re_sr = re.compile(r'/([0-9]{4,})Hz_.*')
+ match_samplerate = re_sr.findall(soundfile)
+ if match_samplerate:
+ samplerate = int(match_samplerate[0])
+ else:
+ import warnings
+ warnings.warn(UserWarning("could not parse samplerate for {:s}"
+ .format(soundfile)))
+ return samplerate