1#! /usr/bin/env python 2# -*- coding: utf-8 -*- 3 4import sys, os 5import numpy as np 6from aubio import fvec, sink, float_type 7 8if __name__ == '__main__': 9 if len(sys.argv) < 1: 10 print('usage: %s' % sys.argv[0]) 11 sys.exit(1) 12 13 samplerate = 44100 14 hop_size = 256 15 16 # create python/tests/sounds if needed 17 output_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) 18 output_dir = os.path.join(output_dir, 'tests', 'sounds') 19 if not os.path.isdir(output_dir): 20 os.makedirs(output_dir) 21 22 filenames = ['44100Hz_1f_silence.wav', 23 '22050Hz_5s_brownnoise.wav', 24 '32000Hz_127f_sine440.wav', 25 ] 26 samplerates = [44100, 22050, 32000] 27 durations = [1, 5*22050, 127] 28 29 for fname, samplerate, duration in zip(filenames, samplerates, durations): 30 output_name = os.path.join(output_dir, fname) 31 g = sink(output_name, samplerate) 32 total_frames = 0 33 while total_frames < duration: 34 write = min(hop_size, duration - total_frames) 35 if 'brownnoise' in fname: 36 vec = np.random.rand(write).astype(float_type) * 2. - 1. 37 elif 'sine' in fname: 38 freq = 440 39 t = np.arange(write).astype(float_type) + total_frames 40 vec = np.sin(2. * np.pi * freq * t / float(samplerate)) 41 else: 42 # silence 43 vec = fvec(write) 44 g(vec, write) 45 total_frames += write 46 outstr = "wrote {:2f}s".format(total_frames / float(samplerate)) 47 outstr += " ({:d} frames".format(total_frames) 48 outstr += " at {:d}Hz)".format(g.samplerate) 49 outstr += " to {:s}".format(g.uri) 50 print(outstr) 51