4 from aubio import onset, source
5 from numpy import array, hstack, zeros
8 hop_s = win_s / 2 # hop size
11 print "Usage: %s <filename> [samplerate]" % sys.argv[0]
14 filename = sys.argv[1]
17 if len( sys.argv ) > 2: samplerate = int(sys.argv[2])
19 s = source(filename, samplerate, hop_s)
20 samplerate = s.samplerate
21 o = onset("default", win_s, hop_s, samplerate)
23 # list of onsets, in samples
26 # storage for plotted data
29 allsamples_max = zeros(0,)
30 downsample = 2 # to plot n samples / hop_s
32 # total number of frames read
37 print "%f" % (o.get_last_s())
38 onsets.append(o.get_last())
39 # keep some data to plot it later
40 new_maxes = (abs(samples.reshape(hop_s/downsample, downsample))).max(axis=0)
41 allsamples_max = hstack([allsamples_max, new_maxes])
42 desc.append(o.get_descriptor())
43 tdesc.append(o.get_thresholded_descriptor())
45 if read < hop_s: break
49 from numpy import arange
50 import matplotlib.pyplot as plt
51 allsamples_max = (allsamples_max > 0) * allsamples_max
52 allsamples_max_times = [ float(t) * hop_s / downsample / samplerate for t in range(len(allsamples_max)) ]
53 plt1 = plt.axes([0.1, 0.75, 0.8, 0.19])
54 plt2 = plt.axes([0.1, 0.1, 0.8, 0.65], sharex = plt1)
55 plt.rc('lines',linewidth='.8')
56 plt1.plot(allsamples_max_times, allsamples_max, '-b')
57 plt1.plot(allsamples_max_times, -allsamples_max, '-b')
59 stamp /= float(samplerate)
60 plt1.plot([stamp, stamp], [-1., 1.], '-r')
61 plt1.axis(xmin = 0., xmax = max(allsamples_max_times) )
62 plt1.xaxis.set_visible(False)
63 plt1.yaxis.set_visible(False)
64 desc_times = [ float(t) * hop_s / samplerate for t in range(len(desc)) ]
65 desc_plot = [d / max(desc) for d in desc]
66 plt2.plot(desc_times, desc_plot, '-g')
67 tdesc_plot = [d / max(desc) for d in tdesc]
69 stamp /= float(samplerate)
70 plt2.plot([stamp, stamp], [min(tdesc_plot), max(desc_plot)], '-r')
71 plt2.plot(desc_times, tdesc_plot, '-y')
72 plt2.axis(ymin = min(tdesc_plot), ymax = max(desc_plot))
73 plt.xlabel('time (s)')
74 #plt.savefig('/tmp/t.png', dpi=200)