plot_timing_data.py

Go to the documentation of this file.

#!/bin/env python

from ROOT import *

import code

gROOT.Macro('~/rootlogon.C')

gs = [] # globals

times = {} # Map from tuple to list of times


def time_format(hrs):
    if hrs < 1: return str(int(hrs*60+.5))+'m'
    return str(int(hrs))+'h'+str(int(hrs*60+.5)%60)+'m'

# Skips blank entries without superfluous spaces
def join_with_spaces(vec):
    if len(vec) == 0: return ''
    if len(vec) == 1: return vec[0]
    if vec[0] == '': return join_with_spaces(vec[1:])
    return vec[0]+' '+join_with_spaces(vec[1:])


for line in open('timing_data.txt'):
    try:
        fname, time = line[:-1].split(' ')
    except:
        print 'Invalid line:'
        print '  ', line
        continue

    det = ''
    if 'fd' in fname or 'fardet' in fname: det = 'Far detector'
    if 'nd' in fname or 'neardet' in fname: det = 'Near detector'
    if det == '': print 'Unknown detector: ', fname

    datamc = ''
    if 'sim' in fname: datamc = 'MC'
    if 'data' in fname: datamc = 'Data'
    if datamc == '': print 'Unknown data/MC: ', fname

    horn = ''
    if 'fhc' in fname: horn = 'FHC'
    if 'rhc' in fname: horn = 'RHC'
    if 'cosmics' in fname or 't02' in fname: horn = 'Cosmics'
    if 't00' in fname: horn = 'NuMI'
    if horn == '': print 'Unknown horn: ', fname

    swap = ''
    if 'swap' in fname: swap = 'fluxswap'
    if 'nonswap' in fname: swap = 'nonswap'
    if 'tau' in fname: swap = 'tau'
    if datamc != 'Data' and horn != 'Cosmics' and swap == '':
        print 'Unknown swap ', fname

    version = ''
    if 'v1' in fname: version = 'v1'
    if 'v2' in fname: version = 'v2'
    if 'v3' in fname: version = 'v3'
    if 'v4' in fname: version = 'v4'
    if 'v5' in fname: version = 'v5'
    if 'v6' in fname: version = 'v6'
    if 'v7' in fname: version = 'v7'
    if 'v8' in fname: version = 'v8'
    if 'v9' in fname: version = 'v9'
    if version == '': print 'Unknown version ', fname

    release = ''
    if 'S' in fname:
        release = fname[fname.find('S'):fname.find('S')+9]
    else:
        release = fname[fname.find('FA'):fname.find('FA')+10]
    if release == '': print 'Unknown release ', fname

    mrcc = ''
    if 'mrcclemsum' in fname: mrcc = 'MRCC'
    if 'mrelemsum' in fname: mrcc = 'MRE'

    special = ''
    if 'geojittered' in fname: special = 'geojittered'
    if 'nomask' in fname: special = 'nomask'
    if 'calib_shift_nominal' in fname: special = 'calib_shift_nominal'
    if 'calib_shift_cell' in fname: special = 'calib_shift_cell'
    if 'calib_shift20pc' in fname: special = 'calib_shift20pc'
    if 'calib_shift20dn' in fname: special = 'calib_shift20dn'
    if 'calib_shift_slope20up' in fname: special = 'calib_shift_slope20up'
    if 'calib_shift_slope20dn' in fname: special = 'calib_shift_slope20dn'
    if 'FTF_BIC' in fname: special = 'FTF_BIC'
    if 'QGSP_BIC_HP' in fname: special = 'QGSP_BIC_HP'

    title = join_with_spaces([det, horn, swap, special, mrcc, release, datamc])

    times.setdefault(title, []).append(float(time)/(60*60)) # convert to hours

total_time = 0

for title in times:
    fname = 'plots/'+title.lower().replace(' ', '_')

    gs.append(TCanvas())
    c = gs[-1]

    gs.append(TH1F('', title+';Hours to process through LEM;Number of files', 24*60, 0, 24))
    h = gs[-1]

    for t in times[title]:
        h.Fill(t)
        total_time += t
    h.Draw()

    mean = h.GetMean()

#    rebin = int((3*mean)/100)
#    if rebin > 1: h.Rebin(rebin)

#    gPad.SetLogy(True)
#    gPad.Print(fname+'_log.eps')
#    gPad.SetLogy(False)

    h.GetXaxis().SetRangeUser(0, mean*3)

    gs.append(TLatex(.6, .8, 'Total files '+str(len(times[title]))))
    lat = gs[-1]
    lat.SetNDC()
    lat.Draw()

    gs.append(TLatex(.6, .7, 'Average time '+time_format(mean)))
    lat = gs[-1]
    lat.SetNDC()
    lat.Draw()

    gPad.Print(fname+'.eps')
    gPad.Print(fname+'.png')

    c.Update()


    print title, len(times[title]), mean


print 'Total hours', total_time

# Just want the console to hang around so the user can look at the plots
# without them disappearing. Ctrl-D or "exit()" to quit.
code.interact(local = locals())