#!/usr/bin/python
import os,sys,glob
from fpformat import fix
from ROOT import *

exagg = 27310-2731 #Hz (p.146 of Rob's thesis, 2nd paragraph from bottom

## following are AFTER suppression of bad runs, using mean_columns.py
## on the directory /home/jbueno/Reports/ts/systematics/mrate/rates_per_set_br_removed
data_rates = {'68':2065.95199,'70':2324.83817,'71':2582.85438,
              '72':2674.23253,'74':2592.88293,'75':2686.61719,
              '76':2740.91466,
              '83':4221.84799,'84':4452.33555,'86':4972.34878,
              '87':4024.04726,
              '91':4202.46627,'92':4305.30669,'93':3809.54449}

d = '/home/e614/analysis_2006_2007/treesum/pass1/'
fs = glob.glob(d+'*_clk_raw.root')
di = {}
for fn in fs:
    f = TFile(fn)
    h = f.Get("Status/EvtType_Beginning")
    simple_clean_1 = h.GetBinContent(1+1)
    mult_muon_15   = h.GetBinContent(15+1)
    r_mu = mult_muon_15/(mult_muon_15+simple_clean_1)
    clean_fn = fn.split('/')[-1]
    just_set_gen = int(clean_fn.split('-')[0].lstrip('sum'))
    di[just_set_gen] = r_mu
    print just_set_gen,'     ',r_mu


print '#set  data_rate  gen    gen_rate  sim_minus_data   relative_diff    data_rate   error   scale'
for set,gen in [(68,468),(70,470),(71,471),(72,472),(74,474),(75,475),(76,476),
                (83,583),(84,584),(86,586),(87,587),(91,591),(92,592),(93,593)]:
    print set,'  ',fix(di[set],5),'  ',gen,'  ',fix(di[gen],5),'  ',
    diff = di[gen]-di[set]
    rel_diff = diff/di[set]
    print fix(diff,5),'     ',fix(rel_diff,5),'       ',
    if str(set) in data_rates.keys():
        error = rel_diff*data_rates[str(set)]
        print fix(data_rates[str(set)],1),'  ',fix(error,1),'  ',fix(exagg/error,1)
    else:
        print