Functions
	main ()

Variables
	parser = argparse.ArgumentParser()
	type
	str
	help
	action
	args = parser.parse_args()
	year = args.year
	channel = args.channel
	comp = args.comp
	indir = args.indir
	outdir = args.outdir
float	xmin = 0.2
	bins = np.concatenate((np.array([0, 10]), np.linspace(11, 17, 7), np.array([18, 22])))
str	date_string = "Run 2,#kern[-0.5]{ }#sqrt{s} = 13 TeV"
	grl = pt.get_grl("15")
str	out_tag = "run2"
str	outfile = "ZeeZmm_ratio_vs_mu_"+out_tag+".pdf"
	ymin
	ymax

Function Documentation

◆ main()

plotting.yearwise_luminosity_vs_mu.main ( )

Definition at line 80 of file yearwise_luminosity_vs_mu.py.

def main():
    dflist = []
    for run in grl: 
        livetime, zlumi, zerr, olumi, timestamp, dfz_small = pt.get_dfz(args.indir, year, run, channel)
 
        # Cut out short runs
        if livetime < pt.runlivetimecut:
            if livetime >= 0.: print(f"Skip Run {run} because of live time {livetime/60:.1f} min")
            continue
            
        # Cut out early 2016 runs with "strange" bunch structure
        if (year == "16" and dfz_small['LBStart'].iloc[0] < 1463184000) or run == "310247": 
            continue
        
        dflist.append(dfz_small)
   
    df = pd.concat(dflist)
    df['OffMu'] = df['OffMu'].round(0)
    df = df.groupby(pd.cut(df.OffMu, bins, right=False), observed=False).sum()
    df.reset_index(drop=True, inplace=True)
    if comp:
        df['ZeeLumiErr'] = np.sqrt(df['ZeeLumiErr'])
        df['ZmumuLumiErr'] = np.sqrt(df['ZmumuLumiErr'])
        df['Ratio']    = df['ZeeLumi'] / df['ZmumuLumi']
        df['RatioErr'] = df['Ratio'] * np.sqrt(pow(df['ZeeLumiErr'] / df['ZeeLumi'], 2) + pow(df['ZmumuLumiErr'] / df['ZmumuLumi'], 2))
    else:
        df['ZLumiErr'] = np.sqrt(df['ZLumiErr'])
        df['Bin'] = pd.Series(bins)
        
        norm = df['ZLumi'].sum() / df['OffLumi'].sum()
        df['Ratio']    = df['ZLumi'] / df['OffLumi'] / norm
        df['RatioErr'] = df['ZLumiErr'] / df['OffLumi'] / norm
 
    h_total = R.TH1F("h_total", "", len(bins)-1, bins)
 
    nan_list = df[df['Ratio'].isnull()].index.tolist()
    
    arr_ratio = []
    
    for xbin in range(0, h_total.GetNbinsX()):
        
        if xbin in nan_list:            
            continue
            
        try:
            h_total.SetBinContent(xbin+1, df['Ratio'][xbin])
            h_total.SetBinError(xbin+1, df['RatioErr'][xbin])
            arr_ratio.append(df['Ratio'][xbin])
            
        except KeyError:
            print("Cannot do ratio for", xbin)
 
    arr_ratio = np.array(arr_ratio)
 
    median = np.median(arr_ratio)
    stdev = np.percentile(abs(arr_ratio - median), 68)
    
    c1 = R.TCanvas()
    h_total.GetXaxis().SetTitle("<#mu>")
    h_total.Draw("E0")
    R.gStyle.SetErrorX()
    line = R.TLine(h_total.GetXaxis().GetXmin(), median, h_total.GetXaxis().GetXmax(), median)
    line.SetLineColor(R.kRed)
    line.Draw()
    
    if comp:
        h_total.GetYaxis().SetRangeUser(ymin, ymax)
        leg = R.TLegend(0.54, 0.72, 0.805, 0.92)
        leg.SetTextSize(18)
        leg.AddEntry(h_total, pt.Leemumuratiolabel, "ep")
        leg.AddEntry(line, f"Median = {median:.3f}#pm{stdev:.3f}", "l")
    else: 
        h_total.GetYaxis().SetRangeUser(0.95, 1.05)
        leg = R.TLegend(0.20, 0.18, 0.45, 0.35)
 
    print(f"Year = {year} channel = {channel}: median +- 68% percentile = {median:.3f} +- {stdev:.3f}")
 
    line1 = pt.make_bands(bins, stdev, median)
    line1.Draw("same 3")
    if comp: line.Draw()
    h_total.Draw('same E0')
    
    leg.SetBorderSize(0)
    leg.SetTextSize(0.05)
    if comp: 
        h_total.GetYaxis().SetTitle(pt.Leemumuratiolabel)
        zstring = ""
    else:
        h_total.GetYaxis().SetTitle("L_{"+pt.plotlabel[channel]+"} / L_{ATLAS}")
        leg.AddEntry(h_total, "L_{"+pt.plotlabel[channel]+"}^{year-normalised}/L_{ATLAS}", "ep")
        zstring = pt.plotlabel[channel]+" counting"
 
    if comp:
        pt.drawAtlasLabel(0.2, 0.88, "Internal")
        pt.drawText(0.2, 0.83, date_string, size=22)
        pt.drawText(0.2, 0.78, zstring, size=22)
    else:
        pt.drawAtlasLabel(xmin, 0.88, "Internal")
        pt.drawText(xmin, 0.83, date_string, size=22)
        pt.drawText(xmin, 0.78, zstring, size=22)
        pt.drawText(xmin, 0.71, "OflLumi-Run3-006", size=22)
        
    leg.AddEntry(line1, "68% band (#pm" + str(round(stdev, 3)) + ")", "f")
    leg.Draw()
 
    if comp:
        c1.SaveAs(outdir + outfile)
    else: 
        c1.SaveAs(outdir + channel + "ATLAS_ratio_vs_mu_"+out_tag+".pdf")
 