ref: 2d04b8612f7e5da67002f1956112ab9dc361ca07
dir: /python/aubiocut/
#! /usr/bin/python
""" this file was written by Paul Brossier
it is released under the GNU/GPL license.
"""
import sys
from aubio.aubioclass import *
usage = "usage: %s [options] -i soundfile" % sys.argv[0]
def check_mode(option, opt, value, parser):
nvalue = parser.rargs[0]
if nvalue == 'complexdomain' :
setattr(parser.values, option.dest, complexdomain)
elif nvalue == 'hfc' :
setattr(parser.values, option.dest, hfc)
elif nvalue == 'phase' :
setattr(parser.values, option.dest, phase)
elif nvalue == 'specdiff' :
setattr(parser.values, option.dest, specdiff)
elif nvalue == 'energy' :
setattr(parser.values, option.dest, energy)
elif nvalue == 'kl' :
setattr(parser.values, option.dest, kl)
elif nvalue == 'mkl' :
setattr(parser.values, option.dest, mkl)
elif nvalue == 'dual' :
setattr(parser.values, option.dest, 'dual')
else:
print "unknown detection function selected\n", usage
sys.exit(1)
def parse_args():
from optparse import OptionParser
parser = OptionParser(usage=usage)
parser.add_option("-i","--input",
action="store", dest="filename",
help="input sound file")
parser.add_option("-m","--mode", action="callback",
callback=check_mode, dest="mode", default='dual',
help="onset detection mode [default=dual] \
complexdomain|hfc|phase|specdiff|energy|kl|mkl|dual")
parser.add_option("-B","--bufsize",
action="store", dest="bufsize", default=1024,
help="buffer size [default=1024]")
parser.add_option("-H","--hopsize",
action="store", dest="hopsize", default=256,
help="overlap size [default=256]")
parser.add_option("-t","--threshold",
action="store", dest="threshold", default=0.650,
help="onset peak picking threshold [default=0.650]")
parser.add_option("-s","--silence",
action="store", dest="silence", default=-70,
help="silence threshold [default=-70]")
parser.add_option("-M","--mintol",
action="store", dest="mintol", default=0.048,
help="minimum inter onset interval [default=0.048]")
parser.add_option("-D","--delay",
action="store", dest="delay",
help="number of seconds to take back [default=system]\
default system delay is 2*hopsize/samplerate")
parser.add_option("-L","--localmin",
action="store_true", dest="localmin", default=False,
help="use local minima after peak detection")
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","--derivate",
action="store_true", dest="derivate", default=False,
help="derivate onset detection function")
parser.add_option("-S","--silencecut",
action="store_true", dest="silencecut", default=False,
help="outputs silence locations")
# to be implemented
parser.add_option("-z","--zerocross",
action="store_true", dest="zerocross", default=False,
help="NOT IMPLEMENTED zero crossing matching")
parser.add_option("-b","--beat",
action="store_true", dest="beat", default=False,
help="NOT IMPLEMENTED output beat locations")
# plotting functions
parser.add_option("-p","--plot",
action="store_true", dest="plot", default=False,
help="draw plot")
parser.add_option("-O","--outplot",
action="store", dest="outplot", default=None,
help="save plot to output.{ps,png}")
parser.add_option("-v","--verbose",
action="store_true", dest="verbose", default=False,
help="make lots of noise [default]")
parser.add_option("-q","--quiet",
action="store_false", dest="verbose", default=False,
help="be quiet")
(options, args) = parser.parse_args()
if not options.filename:
print "no file name given\n", usage
sys.exit(1)
return options, args
options, args = parse_args()
filename = options.filename
samplerate = float(sndfile(filename).samplerate())
hopsize = int(options.hopsize)
bufsize = int(options.bufsize)
step = float(samplerate)/float(hopsize)
threshold = float(options.threshold)
silence = float(options.silence)
mintol = float(options.mintol)*step
# default take back system delay
if options.delay: delay = float(options.delay)
else: delay = 2./step
if options.beat:
#onsets = getbeats(filename,threshold,silence,mode=options.mode)
exit("not implemented yet")
elif options.silencecut:
onsets = getsilences(filename,hopsize=hopsize,silence=silence)
elif options.plot:
onsets, ofunc = getonsets(filename,threshold,silence,
mode=options.mode,localmin=options.localmin,
derivate=options.derivate,
bufsize=bufsize,hopsize=hopsize,storefunc=True)
else:
onsets = getonsets(filename,threshold,silence,
mode=options.mode,localmin=options.localmin,
derivate=options.derivate,
bufsize=bufsize,hopsize=hopsize)
# take back system delay
if delay != 0:
for i in range(len(onsets)):
onsets[i] -= delay*step
# prune doubled
if mintol > 0:
last = -2*mintol
newonsets = []
for new in onsets:
if (new - last > mintol):
newonsets.append(new)
last = new
onsets = newonsets
# print times in second
if options.verbose:
for i in onsets: print "%f" % (i/step)
if options.plot:
from aubio.gnuplot import plot_onsets
plot_onsets(filename, onsets, ofunc,
samplerate=samplerate, hopsize=hopsize, outplot=options.outplot)
if options.cut:
cutfile(filename,onsets,bufsize=bufsize,hopsize=hopsize)