source: python/test/bench/pitch/bench-pitch @ 3043081

#! /usr/bin/python

from aubio.bench.node import *
from aubio.tasks import *

class benchpitch(bench):
        
        """ list of values to store per file """
        valuenames = ['mode']
        """ list of lists to store per file """
        valuelists = ['orig', 'mean', 'med']
        """ list of values to print per dir """
        printnames = [ 'mode']

        """ per dir """
        formats = {'mode': "%12s" , 'thres': "%5.4s", 
                'dist':  "%5.4s", 'prec': "%5.4s", 'recl':  "%5.4s",
                'Ttrue': "%5.4s", 'Tfp':   "%5.4s", 'Tfn':   "%5.4s", 
                'Tm':    "%5.4s", 'Td':    "%5.4s",
                'aTtrue':"%5.4s", 'aTfp':  "%5.4s", 'aTfn':  "%5.4s", 
                'aTm':   "%5.4s", 'aTd':   "%5.4s",
                'mean':  "%5.40s", 'smean': "%5.40s", 
                'amean':  "%5.40s", 'samean': "%5.40s"}

        def dir_eval(self):
                """ evaluate statistical data over the directory """
                v = self.v

                v['mode']      = self.params.pitchmode

        def file_exec(self,input,output):
                filetask = self.task(input,params=self.params)
                computed_data = filetask.compute_all()
                orig,mean,med = filetask.eval(computed_data)
                
                self.v['orig'].append(orig)
                self.v['mean'].append(mean)
                self.v['med'].append(med)
                #print results#, computed_data
                #print input, results, results - float(input.split('.')[-2])
                        
        def run_bench(self,modes=['schmitt']):
                from os.path import basename
                d = []
                self.modes = modes
                self.pretty_titles()
                for mode in self.modes:
                        self.params.pitchmode = mode
                        self.dir_eval_print()
                        self.plotpitchtessiture(d,
                                self.v['orig'], 
                                self.v['med'],
                                plottitle=self.v['mode'],
                                plotmode='points')
                #d.append('beta = .25,orig(x) title \"-2 octave\"')
                d.append('beta = .50,orig(x) title \"-1 octave\"')
                d.append('beta = 1.0,orig(x) title \"original\"')
                d.append('beta = 2.0,orig(x) title \"+1 octave\"')
                title = basename(self.datadir)
                outplot = "_-_".join(('pitchtessiture',title))
                for ext in ('ps','png','svg',''):
                        self.plotplotpitchtessiture(d,
                                plottitle=title,
                                outplot=outplot,
                                extension=ext)

        """
        Plot functions 
        """

        def plotpitchtessiture(self,d,lx,ly,plottitle="",plotmode='linespoints'):
                import Gnuplot, Gnuplot.funcutils
                d.append(Gnuplot.Data(lx, ly, with=plotmode, title="%s" % (plottitle) ))

        def plotplotpitchtessiture(self,d,plottitle='',outplot=0,extension=''):
                from aubio.gnuplot import gnuplot_create
                g = gnuplot_create(outplot=outplot,extension=extension) 
                g.title(plottitle)
                g('orig(x) = beta*x')
                g.xlabel('original pitch (Hz)')
                g.ylabel('detected pitch (Hz)')
                g('set log xy')
                g.plot(*d)


if __name__ == "__main__":
        import sys
        if len(sys.argv) > 1: datapath = sys.argv[1]
        else: print "error: a path is required"; sys.exit(1)
        if len(sys.argv) > 2:
                for each in sys.argv[3:-1]: print each
        modes = ['schmitt', 'yin', 'mcomb', 'fcomb']
        #modes = ['fcomb']

        params = taskparams()
        params.bufsize = 2048
        params.hopsize = params.bufsize/2
        benchpitch = benchpitch(datapath,params=params)
        benchpitch.task = taskpitch

        #benchpitch.titles  = [ 'mode', 'thres', 'avg', 'avgdist' ]
        #benchpitch.formats = ["%12s" , "| %6s", "| %6s", "| %6s", "| %6s", "| %6s" ]
        try:
                benchpitch.run_bench(modes=modes)
        except KeyboardInterrupt:
                print "Interrupted by user"
                sys.exit(1)

        sys.exit(0)