demos/aubiocut: start moving aubiocut to new python
authorPaul Brossier <piem@piem.org>
Wed, 13 Feb 2013 03:58:38 +0000 (22:58 -0500)
committerPaul Brossier <piem@piem.org>
Wed, 13 Feb 2013 03:58:38 +0000 (22:58 -0500)
python.old/aubiocut [deleted file]
python/demos/aubiocut [new file with mode: 0755]

diff --git a/python.old/aubiocut b/python.old/aubiocut
deleted file mode 100755 (executable)
index be1b926..0000000
+++ /dev/null
@@ -1,156 +0,0 @@
-#! /usr/bin/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="filename", 
-                          help="input sound file")
-        parser.add_option("-m","--mode", 
-                         action="store", 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=512, 
-                          help="buffer size [default=512]")
-        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.3, 
-                          help="onset peak picking threshold [default=0.3]")
-        parser.add_option("-C","--dcthreshold",
-                          action="store", dest="dcthreshold", default=1., 
-                          help="onset peak picking DC component [default=1.]")
-        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 3*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")
-        parser.add_option("-z","--zerocross",
-                          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",
-                          action="store", dest="xsize", default=1., 
-                          type='float', help="define xsize for plot")
-        parser.add_option("-y","--ysize",
-                          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",
-                          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")
-        # to be implemented
-        parser.add_option("-b","--beat",
-                          action="store_true", dest="beat", default=False,
-                          help="output beat locations")
-        (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
-params = taskparams()
-params.hopsize    = int(options.hopsize)
-params.bufsize    = int(options.bufsize)
-params.threshold  = float(options.threshold)
-params.dcthreshold = float(options.dcthreshold)
-params.zerothres  = float(options.zerothres)
-params.silence    = float(options.silence)
-params.mintol     = float(options.mintol)
-params.verbose    = options.verbose
-# default take back system delay
-if options.delay: params.delay = int(float(options.delay)/params.step)
-
-dotask = taskonset
-if options.beat:
-       dotask = taskbeat
-elif options.silencecut:
-       dotask = tasksilence
-elif options.plot or options.func: 
-       params.storefunc=True
-else:              
-       params.storefunc=False
-
-lonsets, lofunc = [], []
-wplot,oplots = [],[]
-modes = options.mode.split(',')
-for i in range(len(modes)):
-       params.onsetmode = modes[i] 
-       filetask = dotask(filename,params=params)
-       onsets = filetask.compute_all()
-
-        #lonsets.append(onsets)
-       if not options.silencecut:
-               ofunc = filetask.ofunc
-               lofunc.append(ofunc)
-
-       if options.plot:
-               if options.beat: 
-                       filetask.plot(oplots, onsets)
-               else:
-                       filetask.plot(onsets, ofunc, wplot, oplots, nplot=options.nplot)
-
-       if options.func: 
-               for i in ofunc: 
-                       print i 
-
-if options.outplot:
-  extension = options.outplot.split('.')[-1] 
-  outplot = '.'.join(options.outplot.split('.')[:-1])
-else:
-  extension,outplot = None,None
-if options.plot: filetask.plotplot(wplot, oplots, outplot=outplot, extension=extension,
-  xsize=options.xsize,ysize=options.ysize,spectro=options.spectro)
-
-if options.cut:
-        a = taskcut(filename,onsets,params=params)
-       a.compute_all()
diff --git a/python/demos/aubiocut b/python/demos/aubiocut
new file mode 100755 (executable)
index 0000000..e801733
--- /dev/null
@@ -0,0 +1,155 @@
+#! /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)