Functions
def	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])))

string	date_string = "Run 2, #sqrt{s} = 13 TeV"

	grl = pt.get_grl("15")

string	out_tag = "run2"

string	outfile = "ZeeZmm_ratio_vs_mu_"+out_tag+".pdf"

	ymin

	ymax

Function Documentation

◆ main()

def plotting.yearwise_luminosity_vs_mu.main ( )

Definition at line 73 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)).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, "L_{Z #rightarrow ee}/L_{Z #rightarrow #mu#mu}", "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("L_{Z #rightarrow ee} / L_{Z #rightarrow #mu#mu}")
         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-005", size=22)
         
     leg.AddEntry(line1, "68% band", "f")
     leg.Draw()
  
     if comp:
         c1.SaveAs(outdir + outfile)
     else: 
         c1.SaveAs(outdir + channel + "ATLAS_ratio_vs_mu_"+out_tag+".pdf")
  