+++ /dev/null
-#! /usr/bin/env python
-
-""" this file was written by Paul Brossier
- it is released under the GNU/GPL license.
-"""
-
-import sys
-#from aubio.task import *
-
-usage = "usage: %s [options] -i soundfile" % sys.argv[0]
-
-def parse_args():
- from optparse import OptionParser
- parser = OptionParser(usage=usage)
- parser.add_option("-i", "--input", action = "store", dest = "source_file",
- help="input sound file to analyse", metavar = "<source_file>")
- parser.add_option("-m","--method",
- action="store", dest="onset_method", default='default',
- metavar = "<onset_method>",
- help="onset detection method [default=default] \
- complexdomain|hfc|phase|specdiff|energy|kl|mkl")
- # cutting methods
- """
- parser.add_option("-b","--beat",
- action="store_true", dest="beat", default=False,
- help="use beat locations")
- parser.add_option("-S","--silencecut",
- action="store_true", dest="silencecut", default=False,
- help="use silence locations")
- parser.add_option("-s","--silence",
- metavar = "<value>",
- action="store", dest="silence", default=-70,
- help="silence threshold [default=-70]")
- """
- # algorithm parameters
- parser.add_option("--samplerate",
- metavar = "<freq>", type='int',
- action="store", dest="samplerate", default=0,
- help="samplerate at which the file should be represented")
- parser.add_option("-B","--bufsize",
- action="store", dest="bufsize", default=512,
- metavar = "<size>",
- help="buffer size [default=512]")
- parser.add_option("-H","--hopsize",
- metavar = "<size>",
- action="store", dest="hopsize", default=256,
- help="overlap size [default=256]")
- parser.add_option("-t","--threshold",
- metavar = "<value>", type="float",
- action="store", dest="threshold", default=0.3,
- help="onset peak picking threshold [default=0.3]")
- """
- parser.add_option("-c","--cut",
- action="store_true", dest="cut", default=False,
- help="cut input sound file at detected labels \
- best used with option -L")
- parser.add_option("-D","--delay",
- action = "store", dest = "delay", type = "float",
- metavar = "<seconds>", default=0,
- help="number of seconds to take back [default=system]\
- default system delay is 3*hopsize/samplerate")
- parser.add_option("-C","--dcthreshold",
- metavar = "<value>",
- action="store", dest="dcthreshold", default=1.,
- help="onset peak picking DC component [default=1.]")
- parser.add_option("-M","--mintol",
- metavar = "<value>",
- action="store", dest="mintol", default=0.048,
- help="minimum inter onset interval [default=0.048]")
- parser.add_option("-L","--localmin",
- action="store_true", dest="localmin", default=False,
- help="use local minima after peak detection")
- parser.add_option("-d","--derivate",
- action="store_true", dest="derivate", default=False,
- help="derivate onset detection function")
- parser.add_option("-z","--zerocross",
- metavar = "<value>",
- action="store", dest="zerothres", default=0.008,
- help="zero-crossing threshold for slicing [default=0.00008]")
- """
- # plotting functions
- """
- parser.add_option("-p","--plot",
- action="store_true", dest="plot", default=False,
- help="draw plot")
- parser.add_option("-x","--xsize",
- metavar = "<size>",
- action="store", dest="xsize", default=1.,
- type='float', help="define xsize for plot")
- parser.add_option("-y","--ysize",
- metavar = "<size>",
- action="store", dest="ysize", default=1.,
- type='float', help="define ysize for plot")
- parser.add_option("-f","--function",
- action="store_true", dest="func", default=False,
- help="print detection function")
- parser.add_option("-n","--no-onsets",
- action="store_true", dest="nplot", default=False,
- help="do not plot detected onsets")
- parser.add_option("-O","--outplot",
- metavar = "<output_image>",
- action="store", dest="outplot", default=None,
- help="save plot to output.{ps,png}")
- parser.add_option("-F","--spectrogram",
- action="store_true", dest="spectro", default=False,
- help="add spectrogram to the plot")
- """
- parser.add_option("-v","--verbose",
- action="store_true", dest="verbose", default=True,
- help="make lots of noise [default]")
- parser.add_option("-q","--quiet",
- action="store_false", dest="verbose", default=True,
- help="be quiet")
- (options, args) = parser.parse_args()
- if not options.source_file:
- print "no file name given\n", usage
- sys.exit(1)
- return options, args
-
-if __name__ == '__main__':
- options, args = parse_args()
-
- hopsize = options.hopsize
- bufsize = options.bufsize
- samplerate = options.samplerate
- source_file = options.source_file
-
- from aubio import onset, source
-
- s = source(source_file, samplerate, hopsize)
- if samplerate == 0: samplerate = s.get_samplerate()
-
- o = onset(options.onset_method, bufsize, hopsize)
- o.set_threshold(options.threshold)
-
-
- timestamps = []
- block_read = 0
- while True:
- samples, read = s()
- is_onset = o(samples)
- if is_onset:
- this_onset = (block_read * hopsize - is_onset[0])
- if options.verbose:
- print "%.4f" % ( this_onset / samplerate )
- timestamps.append (this_onset)
- block_read += 1
- if read < hopsize: break
-
- # print some info
- duration = float ( block_read * hopsize + read ) / samplerate
- nstamps = len(timestamps)
- info = 'found %(nstamps)d timestamps in %(source_file)s' % locals()
- info += ' (read %(duration).2fs at %(samplerate)dHz)\n' % locals()
- sys.stderr.write(info)
s = source(filename, samplerate, hop_s)
samplerate = s.samplerate
-o = onset("default", win_s, hop_s)
+o = onset("default", win_s, hop_s, samplerate)
-# onset detection delay, in blocks
+# onset detection delay, in samples
+# default to 4 blocks delay to catch up with
delay = 4. * hop_s
+# list of onsets, in samples
onsets = []
# storage for plotted data
allsamples_max = zeros(0,)
downsample = 2 # to plot n samples / hop_s
+# total number of frames read
total_frames = 0
while True:
samples, read = s()
if is_onset:
this_onset = int(total_frames - delay + is_onset[0] * hop_s)
print "%f" % (this_onset / float(samplerate))
- onsets.append(this_onset / float(samplerate))
+ onsets.append(this_onset)
# keep some data to plot it later
new_maxes = (abs(samples.reshape(hop_s/downsample, downsample))).max(axis=0)
allsamples_max = hstack([allsamples_max, new_maxes])
plt.rc('lines',linewidth='.8')
plt1.plot(allsamples_max_times, allsamples_max, '-b')
plt1.plot(allsamples_max_times, -allsamples_max, '-b')
- for stamp in onsets: plt1.plot([stamp, stamp], [-1., 1.], '-r')
+ for stamp in onsets:
+ stamp /= float(samplerate)
+ plt1.plot([stamp, stamp], [-1., 1.], '-r')
plt1.axis(xmin = 0., xmax = max(allsamples_max_times) )
plt1.xaxis.set_visible(False)
plt1.yaxis.set_visible(False)
desc_plot = [d / max(desc) for d in desc]
plt2.plot(desc_times, desc_plot, '-g')
tdesc_plot = [d / max(desc) for d in tdesc]
- for stamp in onsets: plt2.plot([stamp, stamp], [min(tdesc_plot), max(desc_plot)], '-r')
+ for stamp in onsets:
+ stamp /= float(samplerate)
+ plt2.plot([stamp, stamp], [min(tdesc_plot), max(desc_plot)], '-r')
plt2.plot(desc_times, tdesc_plot, '-y')
plt2.axis(ymin = min(tdesc_plot), ymax = max(desc_plot))
plt.xlabel('time (s)')
--- /dev/null
+#! /usr/bin/env python
+
+import sys
+from aubio import tempo, source
+
+win_s = 512 # fft size
+hop_s = win_s / 2 # hop size
+
+if len(sys.argv) < 2:
+ print "Usage: %s <filename> [samplerate]" % sys.argv[0]
+ sys.exit(1)
+
+filename = sys.argv[1]
+
+samplerate = 0
+if len( sys.argv ) > 2: samplerate = int(sys.argv[2])
+
+s = source(filename, samplerate, hop_s)
+samplerate = s.samplerate
+o = tempo("default", win_s, hop_s, samplerate)
+
+# tempo detection delay, in samples
+# default to 4 blocks delay to catch up with
+delay = 4. * hop_s
+
+# list of beats, in samples
+beats = []
+
+# total number of frames read
+total_frames = 0
+while True:
+ samples, read = s()
+ is_beat = o(samples)
+ if is_beat:
+ this_beat = int(total_frames - delay + is_beat[0] * hop_s)
+ #print "%f" % (this_beat / float(samplerate))
+ beats.append(this_beat)
+ total_frames += read
+ if read < hop_s: break
+
+#convert samples to seconds
+beats = map( lambda x: x / float(samplerate), beats)
+
+bpms = [60./(b - a) for a,b in zip(beats[:-1],beats[1:])]
+
+if len(bpms):
+ print 'mean period:', "%.2f" % mean(bpms), 'bpm', 'median', "%.2f" % median(bpms), 'bpm'
+ print 'plotting', filename
+ # do plotting
+ from numpy import array, arange, mean, median
+ import matplotlib.pyplot as plt
+ plt1 = plt.axes([0.1, 0.75, 0.8, 0.19])
+ plt2 = plt.axes([0.1, 0.1, 0.8, 0.65], sharex = plt1)
+ plt.rc('lines',linewidth='.8')
+ for stamp in beats: plt1.plot([stamp, stamp], [-1., 1.], '-r')
+ plt1.axis(xmin = 0., xmax = total_frames / float(samplerate) )
+ plt1.xaxis.set_visible(False)
+ plt1.yaxis.set_visible(False)
+
+ # plot actual periods
+ plt2.plot(beats[1:], bpms, '-', label = 'raw')
+
+ # plot moving median of 5 last periods
+ median_win_s = 5
+ bpms_median = [ median(bpms[i:i + median_win_s:1]) for i in range(len(bpms) - median_win_s ) ]
+ plt2.plot(beats[median_win_s+1:], bpms_median, '-', label = 'median of %d' % median_win_s)
+ # plot moving median of 10 last periods
+ median_win_s = 20
+ bpms_median = [ median(bpms[i:i + median_win_s:1]) for i in range(len(bpms) - median_win_s ) ]
+ plt2.plot(beats[median_win_s+1:], bpms_median, '-', label = 'median of %d' % median_win_s)
+
+ plt2.axis(ymin = min(bpms), ymax = max(bpms))
+ #plt2.axis(ymin = 40, ymax = 240)
+ plt.xlabel('time (mm:ss)')
+ plt.ylabel('beats per minute (bpm)')
+ plt2.set_xticklabels([ "%02d:%02d" % (t/60, t%60) for t in plt2.get_xticks()[:-1]], rotation = 50)
+
+ #plt.savefig('/tmp/t.png', dpi=200)
+ plt2.legend()
+ plt.show()
+
+else:
+ print 'mean period:', "%.2f" % 0, 'bpm', 'median', "%.2f" % 0, 'bpm',
+ print 'nothing to plot, file too short?'
projects / aubio.git / commitdiff