[e5afd47] | 1 | #! /usr/bin/env python |
---|
| 2 | |
---|
| 3 | import sys |
---|
[8fb567c] | 4 | from aubio import source, pvoc, filterbank |
---|
[e5afd47] | 5 | from numpy import vstack, zeros |
---|
| 6 | |
---|
| 7 | win_s = 512 # fft size |
---|
| 8 | hop_s = win_s / 4 # hop size |
---|
| 9 | |
---|
| 10 | if len(sys.argv) < 2: |
---|
| 11 | print "Usage: %s <filename> [samplerate]" % sys.argv[0] |
---|
| 12 | sys.exit(1) |
---|
| 13 | |
---|
| 14 | filename = sys.argv[1] |
---|
| 15 | |
---|
| 16 | samplerate = 0 |
---|
| 17 | if len( sys.argv ) > 2: samplerate = int(sys.argv[2]) |
---|
| 18 | |
---|
| 19 | s = source(filename, samplerate, hop_s) |
---|
| 20 | samplerate = s.samplerate |
---|
| 21 | |
---|
| 22 | pv = pvoc(win_s, hop_s) |
---|
| 23 | |
---|
| 24 | f = filterbank(40, win_s) |
---|
| 25 | f.set_mel_coeffs_slaney(samplerate) |
---|
| 26 | |
---|
| 27 | energies = zeros((40,)) |
---|
| 28 | o = {} |
---|
| 29 | |
---|
| 30 | total_frames = 0 |
---|
| 31 | downsample = 2 |
---|
| 32 | |
---|
| 33 | while True: |
---|
| 34 | samples, read = s() |
---|
| 35 | fftgrain = pv(samples) |
---|
| 36 | new_energies = f(fftgrain) |
---|
| 37 | print '%f' % (total_frames / float(samplerate) ), |
---|
| 38 | print ' '.join(['%f' % b for b in new_energies]) |
---|
| 39 | energies = vstack( [energies, new_energies] ) |
---|
| 40 | total_frames += read |
---|
| 41 | if read < hop_s: break |
---|
| 42 | |
---|
| 43 | if 1: |
---|
| 44 | print "done computing, now plotting" |
---|
| 45 | import matplotlib.pyplot as plt |
---|
| 46 | from demo_waveform_plot import get_waveform_plot |
---|
[6ff6d18] | 47 | from demo_waveform_plot import set_xlabels_sample2time |
---|
[e5afd47] | 48 | fig = plt.figure() |
---|
| 49 | plt.rc('lines',linewidth='.8') |
---|
| 50 | wave = plt.axes([0.1, 0.75, 0.8, 0.19]) |
---|
| 51 | get_waveform_plot(filename, samplerate, block_size = hop_s, ax = wave ) |
---|
| 52 | wave.yaxis.set_visible(False) |
---|
| 53 | wave.xaxis.set_visible(False) |
---|
| 54 | |
---|
| 55 | n_plots = len(energies.T) |
---|
| 56 | all_desc_times = [ x * hop_s for x in range(len(energies)) ] |
---|
| 57 | for i, band in enumerate(energies.T): |
---|
| 58 | ax = plt.axes ( [0.1, 0.75 - ((i+1) * 0.65 / n_plots), 0.8, 0.65 / n_plots], sharex = wave ) |
---|
| 59 | ax.plot(all_desc_times, band, '-', label = 'band %d' % i) |
---|
| 60 | #ax.set_ylabel(method, rotation = 0) |
---|
| 61 | ax.xaxis.set_visible(False) |
---|
| 62 | ax.yaxis.set_visible(False) |
---|
| 63 | ax.axis(xmax = all_desc_times[-1], xmin = all_desc_times[0]) |
---|
[4f2c28c] | 64 | ax.annotate('band %d' % i, xy=(-10, 0), xycoords='axes points', |
---|
[e5afd47] | 65 | horizontalalignment='right', verticalalignment='bottom', |
---|
[4f2c28c] | 66 | size = 'xx-small', |
---|
[e5afd47] | 67 | ) |
---|
| 68 | set_xlabels_sample2time( ax, all_desc_times[-1], samplerate) |
---|
| 69 | #plt.ylabel('spectral descriptor value') |
---|
| 70 | ax.xaxis.set_visible(True) |
---|
| 71 | plt.show() |
---|