Changeset d4a0cc4 for python

Timestamp:

Jun 3, 2005, 2:53:41 AM (19 years ago)

Author:

Paul Brossier <piem@altern.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

21cc311

Parents:

d53e4df

Message:

update aubiopitch, adds first draft to plotpitches to aubio/gnuplot.py
update aubiopitch, adds first draft to plotpitches to aubio/gnuplot.py

Location:

python

Files:

• aubio/gnuplot.py (modified) (1 diff)
• aubiopitch (modified) (1 diff)

python/aubio/gnuplot.py

@@ -224 +224 @@
 
         g('unset multiplot')
+
+
+def plot_pitch(filename, pitch, samplerate=44100., hopsize=512, outplot=None):
+        import aubio.txtfile
+        import os.path
+        import numarray
+
+        # onset detection function 
+        downtime = (hopsize/samplerate)*numarray.arange(len(pitch))
+        d = Gnuplot.Data(downtime,pitch,with='lines') 
+
+        # check if datafile exists truth
+        datafile = filename.replace('.wav','.txt')
+        if not os.path.isfile(datafile):
+                title = "truth file not found"
+                t = Gnuplot.Data(0,0,with='impulses') 
+        else:
+                title = "truth file plotting not implemented yet"
+                t = Gnuplot.Data(0,0,with='impulses') 
+                #times,pitch = aubio.txtfile.read_datafile(datafile)
+                #t = Gnuplot.Data(times,pitch,with='lines') 
+                
+                #orig, missed, merged, expc, bad, doubled = \
+                #        onset_roc(x2,x1,tol)
+                #title = "GD %2.3f%% FP %2.3f%%" % \
+                #        ((100*float(orig-missed-merged)/(orig)),
+                #         (100*float(bad+doubled)/(orig)))
+                #print  orig, missed, merged, expc, bad, doubled
+                #print "GD %2.8f\t"        % (100*float(orig-missed-merged)/(orig)),
+                #print "FP %2.8f\t"        % (100*float(bad+doubled)/(orig))       , 
+                #print "GD-merged %2.8f\t" % (100*float(orig-missed)/(orig))       , 
+                #print "FP-pruned %2.8f\t" % (100*float(bad)/(orig))                
+
+        # audio data
+        time,data = audio_to_array(filename)
+        f = make_audio_plot(time,data)
+
+        # prepare the plot
+        g = Gnuplot.Gnuplot(debug=1, persist=1)
+        if outplot:
+                extension = outplot.split('.')[-1]
+                if extension == 'ps': extension = 'postscript'
+                g('set terminal %s' % extension)
+                g('set output \'%s\'' % outplot)
+
+        g('set title \'%s %s\'' % (filename,title))
+
+        g('set multiplot')
+
+        # hack to align left axis
+        g('set lmargin 15')
+
+        # plot waveform and onsets
+        g('set size 1,0.3')
+        g('set origin 0,0.7')
+        g('set xrange [0:%f]' % max(time)) 
+        g('set yrange [-1:1]') 
+        g.ylabel('amplitude')
+        g.plot(f)
+        
+        g('unset title')
+
+        # plot onset detection function
+        g('set size 1,0.7')
+        g('set origin 0,0')
+        g('set xrange [0:%f]' % (hopsize/samplerate*len(pitch)))
+        g('set yrange [0:%f]' % (max(pitch)*1.01))
+        g.xlabel('time')
+        g.ylabel('frequency (Hz)')
+        g.plot(d,t)
+
+        g('unset multiplot')

python/aubiopitch

@@ -1 +1 @@
 #!/usr/bin/python
 
-def do(filein):
-    from aubio import aubioclass
-    hopsize   = 512
-    bufsize   = 4096
-    frameread = 0
-    filei     = aubioclass.sndfile(filein)
-    srate     = filei.samplerate()
-    channels  = filei.channels()
-    myvec     = aubioclass.fvec(hopsize,channels)
-    readsize  = filei.read(hopsize,myvec)
-    ppick     = aubioclass.pitchpick(bufsize,hopsize,channels,myvec,srate)
-    while(readsize==hopsize):
-        readsize = filei.read(hopsize,myvec)
-        val = ppick.do(myvec)
-        freq = aubioclass.bintofreq(val,srate,bufsize)
-        now = (frameread)*hopsize/(srate+0.)
-        print "%.3f     %.2f" % (now,freq)
-        frameread += 1
+""" this file was written by Paul Brossier 
+  it is released under the GNU/GPL license.
+"""
 
-if __name__ == "__main__":
-    import sys
-    do(sys.argv[1])
+
+import sys
+import numarray
+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 == 'mcomb' :
+                 setattr(parser.values, option.dest, aubio_mcomb)
+        elif nvalue == 'yin'           :
+                 setattr(parser.values, option.dest, aubio_yin)
+        elif nvalue == 'fcomb'         :
+                 setattr(parser.values, option.dest, aubio_fcomb)
+        elif nvalue == 'schmitt'      :
+                 setattr(parser.values, option.dest, aubio_schmitt)
+
+
+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=aubio_schmitt, 
+                          help="pitch detection mode [default=dual] \
+                          mcomb|yin|fcomb|schmitt")
+        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=512, 
+                          help="overlap size [default=512]")
+        parser.add_option("-t","--threshold",
+                          action="store", dest="threshold", default=0.1, 
+                          help="pitch threshold (for yin) [default=0.1]")
+        parser.add_option("-s","--silence",
+                          action="store", dest="silence", default=-70, 
+                          help="silence threshold [default=-70]")
+        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")
+        # to be implemented
+        parser.add_option("-n","--note",
+                          action="store_true", dest="note", default=False,
+                          help="NOT IMPLEMENTED output notes")
+        # plotting functions
+        parser.add_option("-p","--plot",
+                          action="store_true", dest="plot", default=False, 
+                          help="NOT IMPLEMENTED draw plot")
+        parser.add_option("-O","--outplot",
+                          action="store", dest="outplot", default=None, 
+                          help="NOT IMPLEMENTED 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.note:
+        exit("not implemented yet")
+else:
+        pitch = getpitch(filename, #threshold,silence,
+                mode=options.mode,
+                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 range(len(pitch)): 
+                print "%f\t%f" % (i/step,pitch[i])
+
+if options.plot:
+        from aubio.gnuplot import plot_pitch
+        plot_pitch(filename, pitch, 
+                samplerate=samplerate, hopsize=hopsize, outplot=options.outplot)
+