Showing the changes to python (1 of 1 file). Show all
Unified · Split
| @@ -19,6 +19,9 @@ onsets = [] · 18 unmodified lines | |||
| 19 | s = source(filename, samplerate, hop_s) | 19 | s = source(filename, samplerate, hop_s) |
| 20 | o = onset("default", win_s, hop_s) | 20 | o = onset("default", win_s, hop_s) |
| 21 | 21 | ||
| 22 | desc = [] | ||
| 23 | tdesc = [] | ||
| 24 | |||
| 22 | block_read = 0 | 25 | block_read = 0 |
| 23 | allsamples_max = zeros(0,) | 26 | allsamples_max = zeros(0,) |
| 24 | while True: | 27 | while True: |
| @@ -26,6 +29,8 @@ while True: · 1 unmodified line | |||
| 26 | new_maxes = (abs(samples.reshape(hop_s/downsample, downsample))).max(axis=0) | 29 | new_maxes = (abs(samples.reshape(hop_s/downsample, downsample))).max(axis=0) |
| 27 | allsamples_max = hstack([allsamples_max, new_maxes]) | 30 | allsamples_max = hstack([allsamples_max, new_maxes]) |
| 28 | isbeat = o(samples) | 31 | isbeat = o(samples) |
| 32 | desc.append(o.get_descriptor()) | ||
| 33 | tdesc.append(o.get_thresholded_descriptor()) | ||
| 29 | if isbeat: | 34 | if isbeat: |
| 30 | thisbeat = (block_read - 4. + isbeat[0]) * hop_s / samplerate | 35 | thisbeat = (block_read - 4. + isbeat[0]) * hop_s / samplerate |
| 31 | print "%.4f" % thisbeat | 36 | print "%.4f" % thisbeat |
| @@ -35,14 +40,24 @@ while True: · 3 unmodified lines | |||
| 35 | 40 | ||
| 36 | # do plotting | 41 | # do plotting |
| 37 | from numpy import arange | 42 | from numpy import arange |
| 38 | from pylab import plot, show, xlabel, ylabel, legend, ylim, subplot, axis | 43 | import matplotlib.pyplot as plt |
| 39 | allsamples_max = (allsamples_max > 0) * allsamples_max | 44 | allsamples_max = (allsamples_max > 0) * allsamples_max |
| 40 | allsamples_max_times = [ float(t) * hop_s / downsample / samplerate for t in range(len(allsamples_max)) ] | 45 | allsamples_max_times = [ float(t) * hop_s / downsample / samplerate for t in range(len(allsamples_max)) ] |
| 41 | plot(allsamples_max_times, allsamples_max, '-b') | 46 | plt1 = plt.axes([0.1, 0.75, 0.8, 0.19]) |
| 42 | plot(allsamples_max_times, -allsamples_max, '-b') | 47 | plt2 = plt.axes([0.1, 0.1, 0.8, 0.65], sharex = plt1) |
| 43 | axis(xmin = 0., xmax = max(allsamples_max_times) ) | 48 | plt.rc('lines',linewidth='.8') |
| 44 | for stamp in onsets: plot([stamp, stamp], [-1., 1.], '.-r') | 49 | plt1.plot(allsamples_max_times, allsamples_max, '-b') |
| 45 | xlabel('time (s)') | 50 | plt1.plot(allsamples_max_times, -allsamples_max, '-b') |
| 46 | ylabel('amplitude') | 51 | for stamp in onsets: plt1.plot([stamp, stamp], [-1., 1.], '-r') |
| 47 | show() | 52 | plt1.axis(xmin = 0., xmax = max(allsamples_max_times) ) |
| 48 | 53 | plt1.xaxis.set_visible(False) | |
| 54 | desc_times = [ float(t) * hop_s / samplerate for t in range(len(desc)) ] | ||
| 55 | desc_plot = [d / max(desc) for d in desc] | ||
| 56 | plt2.plot(desc_times, desc_plot, '-g') | ||
| 57 | tdesc_plot = [d / max(desc) for d in tdesc] | ||
| 58 | for stamp in onsets: plt2.plot([stamp, stamp], [min(tdesc_plot), max(desc_plot)], '-r') | ||
| 59 | plt2.plot(desc_times, tdesc_plot, '-y') | ||
| 60 | plt2.axis(ymin = min(tdesc_plot), ymax = max(desc_plot)) | ||
| 61 | plt.xlabel('time (s)') | ||
| 62 | #plt.savefig('/tmp/t.png', dpi=200) | ||
| 63 | plt.show() | ||