remove obsolete function and factorise gnuplot init remove obsolete function and factorise gnuplot init

Paul Brossier committed 21 years ago

Commit 733c2f81 parent d45520a1 file · +41 −120Browse files

Showing the changes to python (1 of 1 file). Show all

Unified · Split

 	 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 """
-__notesheight = 0.25
-from numarray import *
-import Gnuplot, Gnuplot.funcutils
-def plotnote(la,title=None) :
-	if la[0,:].size() == 3:
-	        d = plotnote_withends(la, plot_title=title)
-	else:
-	    # scale data if in freq (for REF.txt files)
-	    if max(la[:,1] > 128 ):
-	        print "scaling frequency data to midi range"
-	        la[:,1] /= 6.875
-	        la[:,1] = log(la[:,1])/0.6931
-	        la[:,1] *= 12
-	        la[:,1] -= 3
-	    d = plotnote_withoutends(la, plot_title=title)
-	return d
-def plotnote_multi(lalist,title=None,fileout=None) :
-	d=list()
-	for i in range(len(lalist)):
-	    d.append(plotnote(lalist[i], title=title))
-	return d
-def plotnote_withends(la,plot_title=None) :
-	d=[]
-	x_widths = array(la[:,1]-la[:,0])/2.
-	d.append(Gnuplot.Data(
-	        la[:,0]+x_widths,               # x centers
-	        la[:,2],                        # y centers
-	        x_widths,                       # x errors
-	        __notesheight*ones(len(la)),    # y errors
-	        title=plot_title,with=('boxxyerrorbars fs 3')))
-	return d
+__notesheight = 0.25
-def plotnote_withoutends(la,plot_title=None) :
-        """ bug: fails drawing last note """
-        d=[]
-        x_widths = array(la[1:,0]-la[:-1,0])/2;
-        d.append(Gnuplot.Data(
-                la[:-1,0]+x_widths,             # x centers
-                la[:-1,1],                      # y centers
-                x_widths,                       # x errors
-                __notesheight*ones(len(la)-1),  # y errors
-                title=plot_title,with=('boxxyerrorbars fs 3')))
-        return d
-def plotnote_do(d,fileout=None):
-    g = Gnuplot.Gnuplot(debug=1, persist=1)
-    g.gnuplot('set style fill solid border 1; \
-    set size ratio 1/6; \
-    set boxwidth 0.9 relative; \
-    set mxtics 2.5; \
-    set mytics 2.5; \
-    set xtics 5; \
-    set ytics 1; \
-    set grid xtics ytics mxtics mytics')
-    g.xlabel('Time (s)')
-    g.ylabel('Midi pitch')
-    # do the plot
-    #g.gnuplot('set multiplot')
-    #for i in d:
-    g.plot(d[0])
-    #g.gnuplot('set nomultiplot')
-    if fileout != None:
-        g.hardcopy(fileout, enhanced=1, color=0)
 def audio_to_array(filename):
 	import aubio.aubioclass
+        import numarray
 	hopsize  = 2048
 	filei    = aubio.aubioclass.sndfile(filename)
 	framestep = 1/(filei.samplerate()+0.)
 		while (curpos < readsize):
 			data.append(myvec.get(curpos,i))
 			curpos+=1
-	time = arange(len(data))*framestep
+	time = numarray.arange(len(data))*framestep
 	return time,data
 def plot_audio(filenames, fileout=None, start=0, end=None, noaxis=None):
-	g = Gnuplot.Gnuplot(debug=1, persist=1)
+	g = gnuplot_init(fileout)
 	d = []
 	todraw = len(filenames)
 	xorig = 0.
 	xsize = 1./todraw
 	g.gnuplot('set multiplot;')
 	while (len(filenames)):
-                d.append(plot_audio_make(filenames.pop(0)))
+	        time,data = audio_to_array(filenames.pop(0))
+                d.append(make_audio_plot(time,data))
 		if not noaxis and todraw==1:
 			g.xlabel('Time (s)')
 			g.ylabel('Amplitude')
 		g.gnuplot('set origin %f,0.;' % (xorig) )
 		g.gnuplot('set style data lines; \
 			set yrange [-1.:1.]; \
-			set xrange [0:%f]' % b[-1])
+			set xrange [0:%f]' % time[-1])
 		g.plot(d.pop(0))
 		xorig += 1./todraw
 	g.gnuplot('unset multiplot;')
-	if fileout != None:
-		g.hardcopy(fileout, enhanced=1, color=0)
 def make_audio_plot(time,data,maxpoints=10000):
 	""" create gnuplot plot from an audio file """
+	import numarray
+	import Gnuplot, Gnuplot.funcutils
         length = len(time)
 	downsample = length/maxpoints
         if downsample == 0: downsample = 1
-        x = array(time).resize(length)[0:-1:downsample]
+        x = numarray.array(time).resize(length)[0:-1:downsample]
-        y = array(data).resize(length)[0:-1:downsample]
+        y = numarray.array(data).resize(length)[0:-1:downsample]
 	return Gnuplot.Data(x,y,with='lines')
 def plot_onsets(filename, onsets, ofunc, samplerate=44100., hopsize=512, outplot=None):
+	import Gnuplot, Gnuplot.funcutils
         import aubio.txtfile
         import os.path
         import numarray
+	import re
         from aubio.onsetcompare import onset_roc
         d,d2 = [],[]
                 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)
         d2.append(make_audio_plot(time,data))
         # prepare the plot
-        g = Gnuplot.Gnuplot(debug=1, persist=1)
+        g = gnuplot_init(outplot)
-        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 title \'%s %s\'' % (re.sub('.*/','',filename),title))
         g('set multiplot')
         import aubio.txtfile
         import os.path
         import numarray
+	import Gnuplot
+	import re
         d = []
         maxpitch = 100
                         title=('%d' % i)))
                 maxpitch = max(maxpitch,max(pitch[i][:])*1.1)
-        # check if datafile exists truth
+        # check if ground truth exists
         datafile = filename.replace('.wav','.txt')
         if not os.path.isfile(datafile):
                 title = "truth file not found"
         else:
                 title = "truth file plotting not implemented yet"
                 values = aubio.txtfile.read_datafile(datafile)
-                time, pitch = [], []
+		if (len(datafile[0])) > 1:
-                for i in range(len(values)):
+                        time, pitch = [], []
-                        time.append(values[i][0])
+                        for i in range(len(values)):
-                        pitch.append(values[i][1])
+                                time.append(values[i][0])
-                d.append(Gnuplot.Data(time,pitch,with='lines',title='ground truth'))
+                                pitch.append(values[i][1])
+                        d.append(Gnuplot.Data(time,pitch,with='lines',
+				title='ground truth'))
-                #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_init(outplot)
-        g = Gnuplot.Gnuplot(debug=1, persist=1)
+        g('set title \'%s %s\'' % (re.sub('.*/','',filename),title))
-        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 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.xlabel('time')
         g.ylabel('frequency (Hz)')
         g.plot(*d)
         g('unset multiplot')
+def gnuplot_init(outplot,debug=0,persist=1):
+	import Gnuplot
+        # prepare the plot
+        g = Gnuplot.Gnuplot(debug=debug, persist=persist)
+	if outplot == 'stdout':
+                g("set terminal png fontfile 'p052023l.pfb'")
+                #g('set output \'%s\'' % outplot)
+        elif outplot:
+                extension = outplot.split('.')[-1]
+                if extension == 'ps': extension = 'postscript'
+                g('set terminal %s' % extension)
+                g('set output \'%s\'' % outplot)
+	return g

remove obsolete function and factorise gnuplot init remove obsolete function and factorise gnuplot init

Keyboard shortcuts

General

Issues and pull requests

Issues, pull requests and Inbox

Repository code