python/demos: add demo_pitch.py and demo_waveform_plot.py

Paul Brossier committed 13 years ago

Commit 5d5d6b91 parent 2cedc835 files · +149 −68Browse files

Showing the changes to python (5 of 5 files). Show all

Unified · Split

5 files

python/demos/demo_onset.py +1 −0
python/demos/demo_pitch.py +37 −0
python/demos/demo_simple_robot_voice.py +1 −0
python/demos/demo_specdesc.py +68 −68
python/demos/demo_waveform_plot.py +42 −0

@@ -17,6 +17,7 @@ if len( sys.argv ) > 2: samplerate = int(sys.argv[2]) · 16 unmodified lines
17		17
18	s = source(filename, samplerate, hop_s)	18	s = source(filename, samplerate, hop_s)
19	samplerate = s.samplerate	19	samplerate = s.samplerate
		20
20	o = onset("default", win_s, hop_s, samplerate)	21	o = onset("default", win_s, hop_s, samplerate)
21		22
22	# onset detection delay, in samples	23	# onset detection delay, in samples

python/demos/demo_pitch.py+37−0

@@ -0,0 +1,37 @@
	1	#! /usr/bin/env python
	2
	3	import sys
	4	from aubio import source, pitch
	5
	6	win_s = 1024 # fft size
	7	hop_s = win_s # hop size
	8
	9	if len(sys.argv) < 2:
	10	print "Usage: %s <filename> [samplerate]" % sys.argv[0]
	11	sys.exit(1)
	12
	13	filename = sys.argv[1]
	14
	15	samplerate = 0
	16	if len( sys.argv ) > 2: samplerate = int(sys.argv[2])
	17
	18	s = source(filename, samplerate, hop_s)
	19	samplerate = s.samplerate
	20
	21	pitch_o = pitch("default", win_s, hop_s, samplerate)
	22	pitch_o.set_unit("midi")
	23
	24	pitches = []
	25
	26
	27	# total number of frames read
	28	total_frames = 0
	29	while True:
	30	samples, read = s()
	31	pitch = pitch_o(samples)[0]
	32	print "%f %f" % (total_frames / float(samplerate), pitch)
	33	#pitches += [pitches]
	34	total_frames += read
	35	if read < hop_s: break
	36
	37	#print pitches

python/demos/demo_simple_robot_voice.py+1−0

@@ -19,6 +19,7 @@ if __name__ == '__main__': · 18 unmodified lines
19	while read:	19	while read:
20	samples, read = f()	20	samples, read = f()
21	spectrum = pv(samples) # compute spectrum	21	spectrum = pv(samples) # compute spectrum
		22	#spectrum.norm *= .8 # reduce amplitude a bit
22	spectrum.phas[:] = 0. # zero phase	23	spectrum.phas[:] = 0. # zero phase
23	new_samples = pv.rdo(spectrum) # compute modified samples	24	new_samples = pv.rdo(spectrum) # compute modified samples
24	g(new_samples, read) # write to output	25	g(new_samples, read) # write to output

python/demos/demo_specdesc.py+68−68

 #! /usr/bin/env python
-from numpy import random, sin, arange, ones, zeros
+import sys
-from math import pi
+from aubio import fvec, source, pvoc, specdesc
-from aubio import fvec, onset
+from numpy import hstack
-def build_sinusoid(length, freqs, samplerate):
+win_s = 512                 # fft size
-  return sin( 2. * pi * arange(length) * freqs / samplerate)
+hop_s = win_s / 4           # hop size
-def run_onset(p, input_vec):
+if len(sys.argv) < 2:
-  f = fvec (p.hop_size)
+    print "Usage: %s <filename> [samplerate]" % sys.argv[0]
-  cands = []
+    sys.exit(1)
-  count = 0
-  for vec_slice in input_vec.reshape((-1, p.hop_size)):
-    f[:] = vec_slice
-    cands.append(o(f))
-  return cands
-methods = ['default',
+filename = sys.argv[1]
-           'energy',
-           'complex',
-           'phase',
-           'specdiff',
-           'kl',
-           'mkl',
-           'specflux',
-           'centroid',
-           'spread',
-           'skewness',
-           'kurtosis',
-           'slope',
-           'decrease',
-           'rolloff',
-          ]
-cands = {}
+samplerate = 0
-buf_size = 2048
+if len( sys.argv ) > 2: samplerate = int(sys.argv[2])
-hop_size = 512
-samplerate = 44100
-sin_length = (samplerate * 10) % 512 * 512
-freqs = zeros(sin_length)
-partition = sin_length / 8
+s = source(filename, samplerate, hop_s)
-pointer = 0
+samplerate = s.samplerate
-pointer += partition
+pv = pvoc(win_s, hop_s)
-freqs[pointer: pointer + partition] = 440
-pointer += partition
+methods = ['default', 'energy', 'hfc', 'complex', 'phase', 'specdiff', 'kl', 'mkl',
-pointer += partition
+    'specflux', 'centroid', 'spread', 'skewness', 'kurtosis', 'slope', 'decrease',
-freqs[ pointer : pointer + partition ] = 740
+    'rolloff', ]
-pointer += partition
+all_descs = {}
-freqs[ pointer : pointer + partition ] = 1480
+o = {}
-pointer += partition
-pointer += partition
-freqs[ pointer : pointer + partition ] = 400 + 5 * random.random(sin_length/8)
-a = build_sinusoid(sin_length, freqs, samplerate)
 for method in methods:
-  o = onset(method, buf_size, hop_size, samplerate)
+    cands = []
-  cands[method] = run_onset(o, a)
+    all_descs[method] = fvec(0)
+    o[method] = specdesc(method, win_s)
+total_frames = 0
+downsample = 2
-print "done computing"
+while True:
+    samples, read = s()
+    fftgrain = pv(samples)
+    print "%f" % ( total_frames / float(samplerate) ),
+    for method in methods:
+        specdesc_val = o[method](fftgrain)[0]
+        all_descs[method] = hstack ( [all_descs[method], specdesc_val] )
+        print "%f" % specdesc_val,
+    print
+    total_frames += read
+    if read < hop_s: break
 if 1:
-  from pylab import plot, show, xlabel, ylabel, legend, ylim, subplot
+    print "done computing, now plotting"
-  subplot (211)
+    import matplotlib.pyplot as plt
-  legend(methods+['ground truth'], 'upper right')
+    from demo_waveform_plot import get_waveform_plot
-  xlabel('time (s)')
+    fig = plt.figure()
-  ylabel('amplitude')
+    plt.rc('lines',linewidth='.8')
-  ramp = arange(0, sin_length).astype('float') / samplerate
+    wave = plt.axes([0.1, 0.75, 0.8, 0.19])
-  plot(ramp, a, ':')
+    get_waveform_plot(filename, samplerate, ax = wave )
-  subplot (212)
+    wave.yaxis.set_visible(False)
-  ramp = arange(0, sin_length / hop_size).astype('float') * hop_size / samplerate
+    wave.xaxis.set_visible(False)
-  for method in methods:
-    plot(ramp, cands[method],'.-')
-    legend(methods, 'upper right')
-    xlabel('time (s)')
-  ylabel('spectral descriptor value')
-  show()
+    all_desc_times = [ x * hop_s  for x in range(len(all_descs["default"])) ]
+    n_methods = len(methods)
+    for i, method in enumerate(methods):
+        #ax = fig.add_subplot (n_methods, 1, i)
+        #plt2 = plt.axes([0.1, 0.1, 0.8, 0.65], sharex = plt1)
+        ax = plt.axes ( [0.1, 0.75 - ((i+1) * 0.65 / n_methods),  0.8, 0.65 / n_methods], sharex = wave )
+        ax.plot(all_desc_times, all_descs[method], '-', label = method)
+        #ax.set_ylabel(method, rotation = 0)
+        ax.xaxis.set_visible(False)
+        ax.yaxis.set_visible(False)
+        ax.axis(xmax = all_desc_times[-1], xmin = all_desc_times[0])
+        ax.annotate(method, xy=(-10, 10),  xycoords='axes points',
+                horizontalalignment='right', verticalalignment='bottom',
+                )
+    if all_desc_times[-1] / float(samplerate) > 60:
+        plt.xlabel('time (mm:ss)')
+        ax.set_xticklabels([ "%02d:%02d" % (t/float(samplerate)/60, (t/float(samplerate))%60) for t in ax.get_xticks()[:-1]], rotation = 50)
+    else:
+        plt.xlabel('time (ss.mm)')
+        ax.set_xticklabels([ "%02d.%02d" % (t/float(samplerate), 100*((t/float(samplerate))%1) ) for t in ax.get_xticks()[:-1]], rotation = 50)
+    #plt.ylabel('spectral descriptor value')
+    ax.xaxis.set_visible(True)
+    plt.show()

python/demos/demo_waveform_plot.py+42−0

@@ -0,0 +1,42 @@
	1	#! /usr/bin/env python
	2
	3	import sys
	4	from aubio import pvoc, source
	5	from numpy import zeros, hstack
	6
	7	def get_waveform_plot(filename, samplerate = 0, ax = None):
	8	import matplotlib.pyplot as plt
	9	if not ax:
	10	fig = plt.figure()
	11	ax = fig.add_subplot(111)
	12	hop_s = 512 # fft window size
	13
	14	allsamples_max = zeros(0,)
	15	downsample = 2 # to plot n samples / hop_s
	16
	17	a = source(filename, samplerate, hop_s) # source file
	18	if samplerate == 0: samplerate = a.samplerate
	19
	20	total_frames = 0
	21	while True:
	22	samples, read = a()
	23	# keep some data to plot it later
	24	new_maxes = (abs(samples.reshape(hop_s/downsample, downsample))).max(axis=0)
	25	allsamples_max = hstack([allsamples_max, new_maxes])
	26	total_frames += read
	27	if read < hop_s: break
	28
	29	allsamples_max = (allsamples_max > 0) * allsamples_max
	30	allsamples_max_times = [ ( float (t) / downsample ) * hop_s for t in range(len(allsamples_max)) ]
	31
	32	ax.plot(allsamples_max_times, allsamples_max, '-b')
	33	ax.plot(allsamples_max_times, -allsamples_max, '-b')
	34
	35	if __name__ == '__main__':
	36	if len(sys.argv) < 2:
	37	print "Usage: %s <filename>" % sys.argv[0]
	38	else:
	39	for soundfile in sys.argv[1:]:
	40	get_waveform_plot(soundfile)
	41	# display graph
	42	show()

python/demos: add demo_pitch.py and demo_waveform_plot.py

Keyboard shortcuts

General

Issues and pull requests

Issues, pull requests and Inbox

Repository code