| | 1 | #!/usr/bin/env python |
| | 2 | # |
| | 3 | # Copyright (C) 2013 eNovance SAS <licensing@enovance.com> |
| | 4 | # Author: Erwan Velu <erwan@enovance.com> |
| | 5 | # |
| | 6 | # The license below covers all files distributed with fio unless otherwise |
| | 7 | # noted in the file itself. |
| | 8 | # |
| | 9 | # This program is free software; you can redistribute it and/or modify |
| | 10 | # it under the terms of the GNU General Public License version 2 as |
| | 11 | # published by the Free Software Foundation. |
| | 12 | # |
| | 13 | # This program is distributed in the hope that it will be useful, |
| | 14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| | 15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| | 16 | # GNU General Public License for more details. |
| | 17 | # |
| | 18 | # You should have received a copy of the GNU General Public License |
| | 19 | # along with this program; if not, write to the Free Software |
| | 20 | # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| | 21 | |
| | 22 | import os |
| | 23 | import fnmatch |
| | 24 | import sys |
| | 25 | import getopt |
| | 26 | import re |
| | 27 | import math |
| | 28 | import shutil |
| | 29 | |
| | 30 | def find_file(path, pattern): |
| | 31 | fio_data_file=[] |
| | 32 | # For all the local files |
| | 33 | for file in os.listdir(path): |
| | 34 | # If the file math the regexp |
| | 35 | if fnmatch.fnmatch(file, pattern): |
| | 36 | # Let's consider this file |
| | 37 | fio_data_file.append(file) |
| | 38 | |
| | 39 | return fio_data_file |
| | 40 | |
| | 41 | def generate_gnuplot_script(fio_data_file,title,gnuplot_output_filename,gnuplot_output_dir,mode,disk_perf,gpm_dir): |
| | 42 | if verbose: print "Generating rendering scripts" |
| | 43 | filename=gnuplot_output_dir+'mygraph' |
| | 44 | temporary_files.append(filename) |
| | 45 | f=open(filename,'w') |
| | 46 | |
| | 47 | # Plotting 3D or comparing graphs doesn't have a meaning unless if there is at least 2 traces |
| | 48 | if len(fio_data_file) > 1: |
| | 49 | f.write("call \'%s/graph3D.gpm\' \'%s' \'%s\' \'\' \'%s\' \'%s\'\n" % (gpm_dir,title,gnuplot_output_filename,gnuplot_output_filename,mode)) |
| | 50 | |
| | 51 | # Setting up the compare files that will be plot later |
| | 52 | compare=open(gnuplot_output_dir + 'compare.gnuplot','w') |
| | 53 | compare.write(''' |
| | 54 | set title '%s' |
| | 55 | set terminal png size 1280,1024 |
| | 56 | set ytics axis out auto |
| | 57 | set key top left reverse |
| | 58 | set xlabel "Time (Seconds)" |
| | 59 | set ylabel '%s' |
| | 60 | set yrange [0:] |
| | 61 | set style line 1 lt 1 lw 3 pt 3 linecolor rgb "green" |
| | 62 | '''% (title,mode)) |
| | 63 | compare.close() |
| | 64 | #Copying the common file for all kind of graph (raw/smooth/trend) |
| | 65 | compare_raw_filename="compare-%s-2Draw" % (gnuplot_output_filename) |
| | 66 | compare_smooth_filename="compare-%s-2Dsmooth" % (gnuplot_output_filename) |
| | 67 | compare_trend_filename="compare-%s-2Dtrend" % (gnuplot_output_filename) |
| | 68 | |
| | 69 | shutil.copy(gnuplot_output_dir+'compare.gnuplot',gnuplot_output_dir+compare_raw_filename+".gnuplot") |
| | 70 | shutil.copy(gnuplot_output_dir+'compare.gnuplot',gnuplot_output_dir+compare_smooth_filename+".gnuplot") |
| | 71 | shutil.copy(gnuplot_output_dir+'compare.gnuplot',gnuplot_output_dir+compare_trend_filename+".gnuplot") |
| | 72 | temporary_files.append(gnuplot_output_dir+compare_raw_filename+".gnuplot") |
| | 73 | temporary_files.append(gnuplot_output_dir+compare_smooth_filename+".gnuplot") |
| | 74 | temporary_files.append(gnuplot_output_dir+compare_trend_filename+".gnuplot") |
| | 75 | |
| | 76 | #Setting up a different output filename for each kind of graph |
| | 77 | compare_raw=open(gnuplot_output_dir+compare_raw_filename + ".gnuplot",'a') |
| | 78 | compare_raw.write("set output '%s.png'\n" % compare_raw_filename) |
| | 79 | compare_smooth=open(gnuplot_output_dir+compare_smooth_filename+".gnuplot",'a') |
| | 80 | compare_smooth.write("set output '%s.png'\n" % compare_smooth_filename) |
| | 81 | compare_trend=open(gnuplot_output_dir+compare_trend_filename+".gnuplot",'a') |
| | 82 | compare_trend.write("set output '%s.png'\n" % compare_trend_filename) |
| | 83 | |
| | 84 | # Let's plot the average value for all the traces |
| | 85 | global_disk_perf = sum(disk_perf, []) |
| | 86 | global_avg = average(global_disk_perf) |
| | 87 | compare_raw.write("plot %s w l ls 1 ti 'Global average value (%.2f)'" % (global_avg,global_avg)); |
| | 88 | compare_smooth.write("plot %s w l ls 1 ti 'Global average value (%.2f)'" % (global_avg,global_avg)); |
| | 89 | compare_trend.write("plot %s w l ls 1 ti 'Global average value (%.2f)'" % (global_avg,global_avg)); |
| | 90 | |
| | 91 | pos=0 |
| | 92 | # Let's create a temporary file for each selected fio file |
| | 93 | for file in fio_data_file: |
| | 94 | tmp_filename = "gnuplot_temp_file.%d" % pos |
| | 95 | |
| | 96 | # Plotting comparing graphs doesn't have a meaning unless if there is at least 2 traces |
| | 97 | if len(fio_data_file) > 1: |
| | 98 | # Adding the plot instruction for each kind of comparing graphs |
| | 99 | compare_raw.write(",\\\n'%s' using 2:3 with linespoints title '%s'" % (tmp_filename,fio_data_file[pos])) |
| | 100 | compare_smooth.write(",\\\n'%s' using 2:3 smooth csplines title '%s'" % (tmp_filename,fio_data_file[pos])) |
| | 101 | compare_trend.write(",\\\n'%s' using 2:3 smooth bezier title '%s'" % (tmp_filename,fio_data_file[pos])) |
| | 102 | |
| | 103 | png_file=file.replace('.log','') |
| | 104 | raw_filename = "%s-2Draw" % (png_file) |
| | 105 | smooth_filename = "%s-2Dsmooth" % (png_file) |
| | 106 | trend_filename = "%s-2Dtrend" % (png_file) |
| | 107 | avg = average(disk_perf[pos]) |
| | 108 | f.write("call \'%s/graph2D.gpm\' \'%s' \'%s\' \'%s\' \'%s\' \'%s\' \'%s\' \'%s\' \'%f\'\n" % (gpm_dir,title,tmp_filename,fio_data_file[pos],raw_filename,mode,smooth_filename,trend_filename,avg)) |
| | 109 | pos = pos +1 |
| | 110 | |
| | 111 | # Plotting comparing graphs doesn't have a meaning unless if there is at least 2 traces |
| | 112 | if len(fio_data_file) > 1: |
| | 113 | os.remove(gnuplot_output_dir+"compare.gnuplot") |
| | 114 | compare_raw.close() |
| | 115 | compare_smooth.close() |
| | 116 | compare_trend.close() |
| | 117 | f.close() |
| | 118 | |
| | 119 | def generate_gnuplot_math_script(title,gnuplot_output_filename,mode,average,gnuplot_output_dir,gpm_dir): |
| | 120 | filename=gnuplot_output_dir+'mymath'; |
| | 121 | temporary_files.append(filename) |
| | 122 | f=open(filename,'a') |
| | 123 | f.write("call \'%s/math.gpm\' \'%s' \'%s\' \'\' \'%s\' \'%s\' %s\n" % (gpm_dir,title,gnuplot_output_filename,gnuplot_output_filename,mode,average)) |
| | 124 | f.close() |
| | 125 | |
| | 126 | def compute_aggregated_file(fio_data_file, gnuplot_output_filename, gnuplot_output_dir): |
| | 127 | if verbose: print "Processing data file 2/2" |
| | 128 | temp_files=[] |
| | 129 | pos=0 |
| | 130 | |
| | 131 | # Let's create a temporary file for each selected fio file |
| | 132 | for file in fio_data_file: |
| | 133 | tmp_filename = "%sgnuplot_temp_file.%d" % (gnuplot_output_dir, pos) |
| | 134 | temp_files.append(open(tmp_filename,'r')) |
| | 135 | pos = pos +1 |
| | 136 | |
| | 137 | f = open(gnuplot_output_dir+gnuplot_output_filename, "w") |
| | 138 | temporary_files.append(gnuplot_output_dir+gnuplot_output_filename) |
| | 139 | index=0 |
| | 140 | # Let's add some information |
| | 141 | for tempfile in temp_files: |
| | 142 | f.write("# Disk%d was coming from %s\n" % (index,fio_data_file[index])) |
| | 143 | f.write(tempfile.read()) |
| | 144 | f.write("\n") |
| | 145 | tempfile.close() |
| | 146 | index = index + 1 |
| | 147 | f.close() |
| | 148 | |
| | 149 | def average(s): return sum(s) * 1.0 / len(s) |
| | 150 | |
| | 151 | def compute_temp_file(fio_data_file,disk_perf,gnuplot_output_dir, min_time, max_time): |
| | 152 | end_time=max_time |
| | 153 | if end_time == -1: |
| | 154 | end_time="infinite" |
| | 155 | if verbose: print "Processing data file 1/2 with %s<time<%s" % (min_time,end_time) |
| | 156 | files=[] |
| | 157 | temp_outfile=[] |
| | 158 | blk_size=0 |
| | 159 | for file in fio_data_file: |
| | 160 | files.append(open(file)) |
| | 161 | pos = len(files) - 1 |
| | 162 | tmp_filename = "%sgnuplot_temp_file.%d" % (gnuplot_output_dir,pos) |
| | 163 | temporary_files.append(tmp_filename) |
| | 164 | gnuplot_file=open(tmp_filename,'w') |
| | 165 | temp_outfile.append(gnuplot_file) |
| | 166 | gnuplot_file.write("#Temporary file based on file %s\n" % file) |
| | 167 | disk_perf.append([]) |
| | 168 | |
| | 169 | shall_break = False |
| | 170 | while True: |
| | 171 | current_line=[] |
| | 172 | nb_empty_files=0 |
| | 173 | nb_files=len(files) |
| | 174 | for myfile in files: |
| | 175 | s=myfile.readline().replace(',',' ').split() |
| | 176 | if not s: |
| | 177 | nb_empty_files+=1 |
| | 178 | s="-1, 0, 0, 0".replace(',',' ').split() |
| | 179 | |
| | 180 | if (nb_empty_files == nb_files): |
| | 181 | shall_break=True |
| | 182 | break; |
| | 183 | |
| | 184 | current_line.append(s); |
| | 185 | |
| | 186 | if shall_break == True: |
| | 187 | break |
| | 188 | |
| | 189 | last_time = -1 |
| | 190 | index=-1 |
| | 191 | perfs=[] |
| | 192 | for line in enumerate(current_line): |
| | 193 | # Index will be used to remember what file was featuring what value |
| | 194 | index=index+1 |
| | 195 | |
| | 196 | time, perf, x, block_size = line[1] |
| | 197 | if (blk_size == 0): |
| | 198 | try: |
| | 199 | blk_size=int(block_size) |
| | 200 | except: |
| | 201 | print "Error while reading the following line :" |
| | 202 | print line |
| | 203 | sys.exit(1); |
| | 204 | |
| | 205 | # We ignore the first 500msec as it doesn't seems to be part of the real benchmark |
| | 206 | # Time < 500 usually reports BW=0 breaking the min computing |
| | 207 | if (min_time == 0): |
| | 208 | min_time==0.5 |
| | 209 | |
| | 210 | # Then we estimate if the data we got is part of the time range we want to plot |
| | 211 | if ((float(time)>(float(min_time)*1000)) and ((int(time) < (int(max_time)*1000)) or max_time==-1)): |
| | 212 | disk_perf[index].append(int(perf)) |
| | 213 | perfs.append("%d %s %s"% (index, time, perf)) |
| | 214 | |
| | 215 | # If we reach this point, it means that all the traces are coherent |
| | 216 | for p in enumerate(perfs): |
| | 217 | index, perf_time,perf = p[1].split() |
| | 218 | temp_outfile[int(index)].write("%s %.2f %s\n" % (index, float(float(perf_time)/1000), perf)) |
| | 219 | |
| | 220 | |
| | 221 | for file in files: |
| | 222 | file.close() |
| | 223 | for file in temp_outfile: |
| | 224 | file.close() |
| | 225 | return blk_size |
| | 226 | |
| | 227 | def compute_math(fio_data_file, title,gnuplot_output_filename,gnuplot_output_dir,mode,disk_perf,gpm_dir): |
| | 228 | if verbose: print "Computing Maths" |
| | 229 | global_min=[] |
| | 230 | global_max=[] |
| | 231 | average_file=open(gnuplot_output_dir+gnuplot_output_filename+'.average', 'w') |
| | 232 | min_file=open(gnuplot_output_dir+gnuplot_output_filename+'.min', 'w') |
| | 233 | max_file=open(gnuplot_output_dir+gnuplot_output_filename+'.max', 'w') |
| | 234 | stddev_file=open(gnuplot_output_dir+gnuplot_output_filename+'.stddev', 'w') |
| | 235 | global_file=open(gnuplot_output_dir+gnuplot_output_filename+'.global','w') |
| | 236 | temporary_files.append(gnuplot_output_dir+gnuplot_output_filename+'.average') |
| | 237 | temporary_files.append(gnuplot_output_dir+gnuplot_output_filename+'.min') |
| | 238 | temporary_files.append(gnuplot_output_dir+gnuplot_output_filename+'.max') |
| | 239 | temporary_files.append(gnuplot_output_dir+gnuplot_output_filename+'.stddev') |
| | 240 | temporary_files.append(gnuplot_output_dir+gnuplot_output_filename+'.global') |
| | 241 | |
| | 242 | min_file.write('DiskName %s\n' % mode) |
| | 243 | max_file.write('DiskName %s\n'% mode) |
| | 244 | average_file.write('DiskName %s\n'% mode) |
| | 245 | stddev_file.write('DiskName %s\n'% mode ) |
| | 246 | for disk in xrange(len(fio_data_file)): |
| | 247 | # print disk_perf[disk] |
| | 248 | min_file.write("# Disk%d was coming from %s\n" % (disk,fio_data_file[disk])) |
| | 249 | max_file.write("# Disk%d was coming from %s\n" % (disk,fio_data_file[disk])) |
| | 250 | average_file.write("# Disk%d was coming from %s\n" % (disk,fio_data_file[disk])) |
| | 251 | stddev_file.write("# Disk%d was coming from %s\n" % (disk,fio_data_file[disk])) |
| | 252 | avg = average(disk_perf[disk]) |
| | 253 | variance = map(lambda x: (x - avg)**2, disk_perf[disk]) |
| | 254 | standard_deviation = math.sqrt(average(variance)) |
| | 255 | # print "Disk%d [ min=%.2f max=%.2f avg=%.2f stddev=%.2f \n" % (disk,min(disk_perf[disk]),max(disk_perf[disk]),avg, standard_deviation) |
| | 256 | average_file.write('%d %d\n' % (disk, avg)) |
| | 257 | stddev_file.write('%d %d\n' % (disk, standard_deviation)) |
| | 258 | local_min=min(disk_perf[disk]) |
| | 259 | local_max=max(disk_perf[disk]) |
| | 260 | min_file.write('%d %d\n' % (disk, local_min)) |
| | 261 | max_file.write('%d %d\n' % (disk, local_max)) |
| | 262 | global_min.append(int(local_min)) |
| | 263 | global_max.append(int(local_max)) |
| | 264 | |
| | 265 | global_disk_perf = sum(disk_perf, []) |
| | 266 | avg = average(global_disk_perf) |
| | 267 | variance = map(lambda x: (x - avg)**2, global_disk_perf) |
| | 268 | standard_deviation = math.sqrt(average(variance)) |
| | 269 | |
| | 270 | global_file.write('min=%.2f\n' % min(global_disk_perf)) |
| | 271 | global_file.write('max=%.2f\n' % max(global_disk_perf)) |
| | 272 | global_file.write('avg=%.2f\n' % avg) |
| | 273 | global_file.write('stddev=%.2f\n' % standard_deviation) |
| | 274 | global_file.write('values_count=%d\n' % len(global_disk_perf)) |
| | 275 | global_file.write('disks_count=%d\n' % len(fio_data_file)) |
| | 276 | #print "Global [ min=%.2f max=%.2f avg=%.2f stddev=%.2f \n" % (min(global_disk_perf),max(global_disk_perf),avg, standard_deviation) |
| | 277 | |
| | 278 | average_file.close() |
| | 279 | min_file.close() |
| | 280 | max_file.close() |
| | 281 | stddev_file.close() |
| | 282 | global_file.close() |
| | 283 | try: |
| | 284 | os.remove(gnuplot_output_dir+'mymath') |
| | 285 | except: |
| | 286 | True |
| | 287 | |
| | 288 | generate_gnuplot_math_script("Average values of "+title,gnuplot_output_filename+'.average',mode,int(avg),gnuplot_output_dir,gpm_dir) |
| | 289 | generate_gnuplot_math_script("Min values of "+title,gnuplot_output_filename+'.min',mode,average(global_min),gnuplot_output_dir,gpm_dir) |
| | 290 | generate_gnuplot_math_script("Max values of "+title,gnuplot_output_filename+'.max',mode,average(global_max),gnuplot_output_dir,gpm_dir) |
| | 291 | generate_gnuplot_math_script("Standard Deviation of "+title,gnuplot_output_filename+'.stddev',mode,int(standard_deviation),gnuplot_output_dir,gpm_dir) |
| | 292 | |
| | 293 | def parse_global_files(fio_data_file, global_search): |
| | 294 | max_result=0 |
| | 295 | max_file='' |
| | 296 | for file in fio_data_file: |
| | 297 | f=open(file) |
| | 298 | disk_count=0 |
| | 299 | search_value=-1 |
| | 300 | |
| | 301 | # Let's read the complete file |
| | 302 | while True: |
| | 303 | try: |
| | 304 | # We do split the name from the value |
| | 305 | name,value=f.readline().split("=") |
| | 306 | except: |
| | 307 | f.close() |
| | 308 | break |
| | 309 | # If we ended the file |
| | 310 | if not name: |
| | 311 | # Let's process what we have |
| | 312 | f.close() |
| | 313 | break |
| | 314 | else: |
| | 315 | # disks_count is not global_search item |
| | 316 | # As we need it for some computation, let's save it |
| | 317 | if name=="disks_count": |
| | 318 | disks_count=int(value) |
| | 319 | |
| | 320 | # Let's catch the searched item |
| | 321 | if global_search in name: |
| | 322 | search_value=float(value) |
| | 323 | |
| | 324 | # Let's process the avg value by estimated the global bandwidth per file |
| | 325 | # We keep the biggest in memory for reporting |
| | 326 | if global_search == "avg": |
| | 327 | if (disks_count > 0) and (search_value != -1): |
| | 328 | result=disks_count*search_value |
| | 329 | if (result > max_result): |
| | 330 | max_result=result |
| | 331 | max_file=file |
| | 332 | # Let's print the avg output |
| | 333 | if global_search == "avg": |
| | 334 | print "Biggest aggregated value of %s was %2.f in file %s\n" % (global_search, max_result, max_file) |
| | 335 | else: |
| | 336 | print "Global search %s is not yet implemented\n" % global_search |
| | 337 | |
| | 338 | def render_gnuplot(fio_data_file, gnuplot_output_dir): |
| | 339 | print "Running gnuplot Rendering" |
| | 340 | try: |
| | 341 | # Let's render all the compared files if some |
| | 342 | if len(fio_data_file) > 1: |
| | 343 | if verbose: print " |-> Rendering comparing traces" |
| | 344 | os.system("cd %s; for i in *.gnuplot; do gnuplot $i; done" % gnuplot_output_dir) |
| | 345 | if verbose: print " |-> Rendering math traces" |
| | 346 | os.system("cd %s; gnuplot mymath" % gnuplot_output_dir) |
| | 347 | if verbose: print " |-> Rendering 2D & 3D traces" |
| | 348 | os.system("cd %s; gnuplot mygraph" % gnuplot_output_dir) |
| | 349 | |
| | 350 | name_of_directory="the current" |
| | 351 | if gnuplot_output_dir != "./": |
| | 352 | name_of_directory=gnuplot_output_dir |
| | 353 | print "\nRendering traces are available in %s directory" % name_of_directory |
| | 354 | global keep_temp_files |
| | 355 | keep_temp_files=False |
| | 356 | except: |
| | 357 | print "Could not run gnuplot on mymath or mygraph !\n" |
| | 358 | sys.exit(1); |
| | 359 | |
| | 360 | def print_help(): |
| | 361 | print 'fio2gnuplot -ghbiodvk -t <title> -o <outputfile> -p <pattern> -G <type> -m <time> -M <time>' |
| | 362 | print |
| | 363 | print '-h --help : Print this help' |
| | 364 | print '-p <pattern> or --pattern <pattern> : A pattern in regexp to select fio input files' |
| | 365 | print '-b or --bandwidth : A predefined pattern for selecting *_bw.log files' |
| | 366 | print '-i or --iops : A predefined pattern for selecting *_iops.log files' |
| | 367 | print '-g or --gnuplot : Render gnuplot traces before exiting' |
| | 368 | print '-o or --outputfile <file> : The basename for gnuplot traces' |
| | 369 | print ' - Basename is set with the pattern if defined' |
| | 370 | print '-d or --outputdir <dir> : The directory where gnuplot shall render files' |
| | 371 | print '-t or --title <title> : The title of the gnuplot traces' |
| | 372 | print ' - Title is set with the block size detected in fio traces' |
| | 373 | print '-G or --Global <type> : Search for <type> in .global files match by a pattern' |
| | 374 | print ' - Available types are : min, max, avg, stddev' |
| | 375 | print ' - The .global extension is added automatically to the pattern' |
| | 376 | print '-m or --min_time <time> : Only consider data starting from <time> seconds (default is 0)' |
| | 377 | print '-M or --max_time <time> : Only consider data ending before <time> seconds (default is -1 aka nolimit)' |
| | 378 | print '-v or --verbose : Increasing verbosity' |
| | 379 | print '-k or --keep : Keep all temporary files from gnuplot\'s output dir' |
| | 380 | |
| | 381 | def main(argv): |
| | 382 | mode='unknown' |
| | 383 | pattern='' |
| | 384 | pattern_set_by_user=False |
| | 385 | title='No title' |
| | 386 | gnuplot_output_filename='result' |
| | 387 | gnuplot_output_dir='./' |
| | 388 | gpm_dir="/usr/share/fio/" |
| | 389 | disk_perf=[] |
| | 390 | run_gnuplot=False |
| | 391 | parse_global=False |
| | 392 | global_search='' |
| | 393 | min_time=0 |
| | 394 | max_time=-1 |
| | 395 | global verbose |
| | 396 | verbose=False |
| | 397 | global temporary_files |
| | 398 | temporary_files=[] |
| | 399 | global keep_temp_files |
| | 400 | keep_temp_files=True |
| | 401 | force_keep_temp_files=False |
| | 402 | |
| | 403 | if not os.path.isfile(gpm_dir+'math.gpm'): |
| | 404 | gpm_dir="/usr/local/share/fio/" |
| | 405 | if not os.path.isfile(gpm_dir+'math.gpm'): |
| | 406 | print "Looks like fio didn't got installed properly as no gpm files found in '/usr/share/fio' or '/usr/local/share/fio'\n" |
| | 407 | sys.exit(3) |
| | 408 | |
| | 409 | try: |
| | 410 | opts, args = getopt.getopt(argv[1:],"ghkbivo:d:t:p:G:m:M:",['bandwidth', 'iops', 'pattern', 'outputfile', 'outputdir', 'title', 'min_time', 'max_time', 'gnuplot', 'Global', 'help', 'verbose','keep']) |
| | 411 | except getopt.GetoptError: |
| | 412 | print "Error: One of the option passed to the cmdline was not supported" |
| | 413 | print "Please fix your command line or read the help (-h option)" |
| | 414 | sys.exit(2) |
| | 415 | |
| | 416 | for opt, arg in opts: |
| | 417 | if opt in ("-b", "--bandwidth"): |
| | 418 | pattern='*_bw.log' |
| | 419 | elif opt in ("-i", "--iops"): |
| | 420 | pattern='*_iops.log' |
| | 421 | elif opt in ("-v", "--verbose"): |
| | 422 | verbose=True |
| | 423 | elif opt in ("-k", "--keep"): |
| | 424 | #User really wants to keep the temporary files |
| | 425 | force_keep_temp_files=True |
| | 426 | elif opt in ("-p", "--pattern"): |
| | 427 | pattern_set_by_user=True |
| | 428 | pattern=arg |
| | 429 | pattern=pattern.replace('\\','') |
| | 430 | elif opt in ("-o", "--outputfile"): |
| | 431 | gnuplot_output_filename=arg |
| | 432 | elif opt in ("-d", "--outputdir"): |
| | 433 | gnuplot_output_dir=arg |
| | 434 | if not gnuplot_output_dir.endswith('/'): |
| | 435 | gnuplot_output_dir=gnuplot_output_dir+'/' |
| | 436 | if not os.path.exists(gnuplot_output_dir): |
| | 437 | os.makedirs(gnuplot_output_dir) |
| | 438 | elif opt in ("-t", "--title"): |
| | 439 | title=arg |
| | 440 | elif opt in ("-m", "--min_time"): |
| | 441 | min_time=arg |
| | 442 | elif opt in ("-M", "--max_time"): |
| | 443 | max_time=arg |
| | 444 | elif opt in ("-g", "--gnuplot"): |
| | 445 | run_gnuplot=True |
| | 446 | elif opt in ("-G", "--Global"): |
| | 447 | parse_global=True |
| | 448 | global_search=arg |
| | 449 | elif opt in ("-h", "--help"): |
| | 450 | print_help() |
| | 451 | sys.exit(1) |
| | 452 | |
| | 453 | # Adding .global extension to the file |
| | 454 | if parse_global==True: |
| | 455 | if not gnuplot_output_filename.endswith('.global'): |
| | 456 | pattern = pattern+'.global' |
| | 457 | |
| | 458 | fio_data_file=find_file('.',pattern) |
| | 459 | if len(fio_data_file) == 0: |
| | 460 | print "No log file found with pattern %s!" % pattern |
| | 461 | sys.exit(1) |
| | 462 | else: |
| | 463 | print "%d files Selected with pattern '%s'" % (len(fio_data_file), pattern) |
| | 464 | |
| | 465 | fio_data_file=sorted(fio_data_file, key=str.lower) |
| | 466 | for file in fio_data_file: |
| | 467 | print ' |-> %s' % file |
| | 468 | if "_bw.log" in file : |
| | 469 | mode="Bandwidth (KB/sec)" |
| | 470 | if "_iops.log" in file : |
| | 471 | mode="IO per Seconds (IO/sec)" |
| | 472 | if (title == 'No title') and (mode != 'unknown'): |
| | 473 | if "Bandwidth" in mode: |
| | 474 | title='Bandwidth benchmark with %d fio results' % len(fio_data_file) |
| | 475 | if "IO" in mode: |
| | 476 | title='IO benchmark with %d fio results' % len(fio_data_file) |
| | 477 | |
| | 478 | print |
| | 479 | #We need to adjust the output filename regarding the pattern required by the user |
| | 480 | if (pattern_set_by_user == True): |
| | 481 | gnuplot_output_filename=pattern |
| | 482 | # As we do have some regexp in the pattern, let's make this simpliest |
| | 483 | # We do remove the simpliest parts of the expression to get a clear file name |
| | 484 | gnuplot_output_filename=gnuplot_output_filename.replace('-*-','-') |
| | 485 | gnuplot_output_filename=gnuplot_output_filename.replace('*','-') |
| | 486 | gnuplot_output_filename=gnuplot_output_filename.replace('--','-') |
| | 487 | gnuplot_output_filename=gnuplot_output_filename.replace('.log','') |
| | 488 | # Insure that we don't have any starting or trailing dash to the filename |
| | 489 | gnuplot_output_filename = gnuplot_output_filename[:-1] if gnuplot_output_filename.endswith('-') else gnuplot_output_filename |
| | 490 | gnuplot_output_filename = gnuplot_output_filename[1:] if gnuplot_output_filename.startswith('-') else gnuplot_output_filename |
| | 491 | |
| | 492 | if parse_global==True: |
| | 493 | parse_global_files(fio_data_file, global_search) |
| | 494 | else: |
| | 495 | blk_size=compute_temp_file(fio_data_file,disk_perf,gnuplot_output_dir,min_time,max_time) |
| | 496 | title="%s @ Blocksize = %dK" % (title,blk_size/1024) |
| | 497 | compute_aggregated_file(fio_data_file, gnuplot_output_filename, gnuplot_output_dir) |
| | 498 | compute_math(fio_data_file,title,gnuplot_output_filename,gnuplot_output_dir,mode,disk_perf,gpm_dir) |
| | 499 | generate_gnuplot_script(fio_data_file,title,gnuplot_output_filename,gnuplot_output_dir,mode,disk_perf,gpm_dir) |
| | 500 | |
| | 501 | if (run_gnuplot==True): |
| | 502 | render_gnuplot(fio_data_file, gnuplot_output_dir) |
| | 503 | |
| | 504 | # Shall we clean the temporary files ? |
| | 505 | if keep_temp_files==False and force_keep_temp_files==False: |
| | 506 | # Cleaning temporary files |
| | 507 | if verbose: print "Cleaning temporary files" |
| | 508 | for f in enumerate(temporary_files): |
| | 509 | if verbose: print " -> %s"%f[1] |
| | 510 | try: |
| | 511 | os.remove(f[1]) |
| | 512 | except: |
| | 513 | True |
| | 514 | |
| | 515 | #Main |
| | 516 | if __name__ == "__main__": |
| | 517 | sys.exit(main(sys.argv)) |
projects / fio.git / blame_incremental