97 lines · 2.8 KB · Python
RawDownload
#! /usr/bin/env python import osimport reimport globimport structimport numpy as npfrom tempfile import mkstemp DEFAULT_SOUND = '22050Hz_5s_brownnoise.wav' def is32bit():    return struct.calcsize("P") * 8 == 32 def array_from_text_file(filename, dtype = 'float'):    realpathname = os.path.join(os.path.dirname(__file__), filename)    return np.loadtxt(realpathname, dtype = dtype) def list_all_sounds(rel_dir):    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

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t