shithub: aubio

ref: c60e048f3aba852710b9763a6fb24adad2017f40
dir: /python/tests/utils.py/

#! /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'):
    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