| 1 | from config import * |
|---|
| 2 | import commands,sys |
|---|
| 3 | import re |
|---|
| 4 | |
|---|
| 5 | def runcommand(cmd,debug=0): |
|---|
| 6 | if VERBOSE >= VERBOSE_CMD or debug: print cmd |
|---|
| 7 | if debug: return |
|---|
| 8 | status, output = commands.getstatusoutput(cmd) |
|---|
| 9 | if status == 0 or VERBOSE >= VERBOSE_OUT: |
|---|
| 10 | output = output.split('\n') |
|---|
| 11 | if VERBOSE >= VERBOSE_OUT: |
|---|
| 12 | for i in output: |
|---|
| 13 | if i: print i |
|---|
| 14 | if not status == 0: |
|---|
| 15 | print 'error:',status,output |
|---|
| 16 | print 'command returning error was',cmd |
|---|
| 17 | #sys.exit(1) |
|---|
| 18 | if output == '' or output == ['']: return |
|---|
| 19 | return output |
|---|
| 20 | |
|---|
| 21 | def list_files(datapath,filter='f', maxdepth = -1): |
|---|
| 22 | if not os.path.exists(datapath): |
|---|
| 23 | print |
|---|
| 24 | print "ERR: no directory %s were found" % datapath |
|---|
| 25 | sys.exit(1) |
|---|
| 26 | if maxdepth >= 0: maxstring = " -maxdepth %d " % maxdepth |
|---|
| 27 | else: maxstring = "" |
|---|
| 28 | cmd = '%s' * 5 % ('find ',datapath,maxstring,' -type ',filter) |
|---|
| 29 | return runcommand(cmd) |
|---|
| 30 | |
|---|
| 31 | def list_wav_files(datapath,maxdepth = -1): |
|---|
| 32 | return list_files(datapath, filter="f -name '*.wav'",maxdepth = maxdepth) |
|---|
| 33 | |
|---|
| 34 | sndfile_filter = "f -name '*.wav' -o -name '*.aif' -o -name '*.aiff'" |
|---|
| 35 | |
|---|
| 36 | def list_snd_files(datapath,maxdepth = -1): |
|---|
| 37 | return list_files(datapath, filter=sndfile_filter, |
|---|
| 38 | maxdepth = maxdepth) |
|---|
| 39 | |
|---|
| 40 | def list_res_files(datapath,maxdepth = -1): |
|---|
| 41 | return list_files(datapath, filter="f -name '*.txt'", maxdepth = maxdepth) |
|---|
| 42 | |
|---|
| 43 | def list_dirs(datapath): |
|---|
| 44 | return list_files(datapath, filter="d") |
|---|
| 45 | |
|---|
| 46 | def mkdir(path): |
|---|
| 47 | cmd = '%s%s' % ('mkdir -p ',path) |
|---|
| 48 | return runcommand(cmd) |
|---|
| 49 | |
|---|
| 50 | def act_on_data (action,datapath,respath=None,suffix='.txt',filter='f',sub='\.wav$',**keywords): |
|---|
| 51 | """ execute action(datafile,resfile) on all files in datapath """ |
|---|
| 52 | dirlist = list_files(datapath,filter=filter) |
|---|
| 53 | if dirlist == ['']: dirlist = [] |
|---|
| 54 | if respath: |
|---|
| 55 | respath_in_datapath = re.split(datapath, respath,maxsplit=1)[1:] |
|---|
| 56 | if(respath_in_datapath and suffix == ''): |
|---|
| 57 | print 'error: respath in datapath and no suffix used' |
|---|
| 58 | for i in dirlist: |
|---|
| 59 | j = re.split(datapath, i,maxsplit=1)[1] |
|---|
| 60 | j = re.sub(sub,'',j) |
|---|
| 61 | #j = "%s%s%s"%(respath,j,suffix) |
|---|
| 62 | if respath: |
|---|
| 63 | j = "%s%s"%(respath,j) |
|---|
| 64 | if sub != '': |
|---|
| 65 | j = re.sub(sub,suffix,j) |
|---|
| 66 | else: |
|---|
| 67 | j = "%s%s" % (j,suffix) |
|---|
| 68 | action(i,j,**keywords) |
|---|
| 69 | |
|---|
| 70 | def act_on_results (action,datapath,respath,filter='d'): |
|---|
| 71 | """ execute action(respath) an all subdirectories in respath """ |
|---|
| 72 | dirlist = list_files(datapath,filter='d') |
|---|
| 73 | respath_in_datapath = re.split(datapath, respath,maxsplit=1)[1:] |
|---|
| 74 | if(respath_in_datapath and not filter == 'd' and suffix == ''): |
|---|
| 75 | print 'warning: respath is in datapath' |
|---|
| 76 | for i in dirlist: |
|---|
| 77 | s = re.split(datapath, i ,maxsplit=1)[1] |
|---|
| 78 | action("%s%s%s"%(respath,'/',s)) |
|---|
| 79 | |
|---|
| 80 | def act_on_files (action,listfiles,listres=None,suffix='.txt',filter='f',sub='\.wav$',**keywords): |
|---|
| 81 | """ execute action(respath) an all subdirectories in respath """ |
|---|
| 82 | if listres and len(listfiles) <= len(listres): |
|---|
| 83 | for i in listfiles: |
|---|
| 84 | action(i,listres[i],**keywords) |
|---|
| 85 | else: |
|---|
| 86 | for i in listfiles: |
|---|
| 87 | action(i,None,**keywords) |
|---|
| 88 | |
|---|
| 89 | class bench: |
|---|
| 90 | """ class to run benchmarks on directories """ |
|---|
| 91 | def __init__(self,datadir,resdir=None,checkres=False,checkanno=False): |
|---|
| 92 | self.datadir = datadir |
|---|
| 93 | # path to write results path to |
|---|
| 94 | self.resdir = resdir |
|---|
| 95 | # list of annotation files |
|---|
| 96 | self.reslist = [] |
|---|
| 97 | # list used to gather results |
|---|
| 98 | self.results = [] |
|---|
| 99 | print "Checking data directory", self.datadir |
|---|
| 100 | self.checkdata() |
|---|
| 101 | if checkanno: self.checkanno() |
|---|
| 102 | if checkres: self.checkres() |
|---|
| 103 | |
|---|
| 104 | def checkdata(self): |
|---|
| 105 | if os.path.isfile(self.datadir): |
|---|
| 106 | self.dirlist = os.path.dirname(self.datadir) |
|---|
| 107 | print "DBG: found a file" |
|---|
| 108 | elif os.path.isdir(self.datadir): |
|---|
| 109 | self.dirlist = list_dirs(self.datadir) |
|---|
| 110 | print "DBG: found a dir" |
|---|
| 111 | # allow dir* matching through find commands? |
|---|
| 112 | else: |
|---|
| 113 | print "ERR: path not understood" |
|---|
| 114 | sys.exit(1) |
|---|
| 115 | print "Listing directories in data directory", |
|---|
| 116 | if self.dirlist: |
|---|
| 117 | print " (%d elements)" % len(self.dirlist) |
|---|
| 118 | else: |
|---|
| 119 | print " (0 elements)" |
|---|
| 120 | print "ERR: no directory %s were found" % self.datadir |
|---|
| 121 | sys.exit(1) |
|---|
| 122 | print "Listing sound files in data directory", |
|---|
| 123 | self.sndlist = list_snd_files(self.datadir) |
|---|
| 124 | if self.sndlist: |
|---|
| 125 | print " (%d elements)" % len(self.sndlist) |
|---|
| 126 | else: |
|---|
| 127 | print " (0 elements)" |
|---|
| 128 | print "ERR: no sound files were found in", self.datadir |
|---|
| 129 | sys.exit(1) |
|---|
| 130 | |
|---|
| 131 | def checkanno(self): |
|---|
| 132 | print "Listing annotations in data directory", |
|---|
| 133 | self.reslist = list_res_files(self.datadir) |
|---|
| 134 | print " (%d elements)" % len(self.reslist) |
|---|
| 135 | #for each in self.reslist: print each |
|---|
| 136 | if not self.reslist or len(self.reslist) < len (self.sndlist): |
|---|
| 137 | print "ERR: not enough annotations" |
|---|
| 138 | return -1 |
|---|
| 139 | else: |
|---|
| 140 | print "Found enough annotations" |
|---|
| 141 | |
|---|
| 142 | def checkres(self): |
|---|
| 143 | print "Creating results directory" |
|---|
| 144 | act_on_results(mkdir,self.datadir,self.resdir,filter='d') |
|---|
| 145 | |
|---|
| 146 | def pretty_print(self,values): |
|---|
| 147 | for i in range(len(values)): |
|---|
| 148 | print self.formats[i] % values[i], |
|---|
| 149 | print |
|---|
| 150 | |
|---|
| 151 | def dir_exec(self): |
|---|
| 152 | """ run file_exec on every input file """ |
|---|
| 153 | self.orig, self.missed, self.merged, self.expc, \ |
|---|
| 154 | self.bad, self.doubled = 0, 0, 0, 0, 0, 0 |
|---|
| 155 | act_on_files(self.file_exec,self.sndlist,self.reslist, \ |
|---|
| 156 | suffix='',filter=sndfile_filter) |
|---|
| 157 | |
|---|
| 158 | def dir_eval(self): |
|---|
| 159 | pass |
|---|
| 160 | |
|---|
| 161 | def file_exec(self): |
|---|
| 162 | pass |
|---|
| 163 | |
|---|
| 164 | def file_eval(self): |
|---|
| 165 | pass |
|---|
| 166 | |
|---|
| 167 | def file_plot(self): |
|---|
| 168 | pass |
|---|
| 169 | |
|---|
| 170 | def dir_plot(self): |
|---|
| 171 | pass |
|---|
| 172 | |
|---|
| 173 | def run_bench(self): |
|---|
| 174 | for mode in self.modes: |
|---|
| 175 | self.params.mode = mode |
|---|
| 176 | self.dir_exec() |
|---|
| 177 | self.dir_eval() |
|---|
| 178 | self.dir_plot() |
|---|
