LongExerciseTauReco/trackAlgoPlot.py at master · brandiskip/LongExerciseTauReco · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import ROOT
import numpy as np
import sys
import argparse
import cms_style
import time
from PhysicsTools.Heppy.physicsutils.TauDecayModes import tauDecayModes

#Command "True" prevents plots from popping up while running. Make "false" to turn off.
ROOT.gROOT.SetBatch(True)
cms_style.setTDRStyle()

#open and read file
tch = ROOT.TChain("tree")
tch.Add("tau_gentau_tuple_HLTstau200_gmsb_0.root")

# define selections for generated and l1 taus
pass_gen  = 'tau_gen_vis_pt>20 && abs(tau_gen_vis_eta)<2.1'

# selections for l1
pass_l1 = 'tau_l1_pt>20 && abs(tau_l1_eta)<2.1'
# selections for reco tau
pass_reco = 'tau_reco_pt>20'
# selections for hlt
pass_hlt  = 'tau_hltPFdispltau_pt>20'
# prompt tau
pass_prompt = 'tau_gen_lxy<0.1'
# decay modes
dm0 = 'tau_gen_decaymode==0'
dm1 = 'tau_gen_decaymode==1'
dm2 = 'tau_gen_decaymode==2'
dm10 = 'tau_gen_decaymode==10'
dmall = 'tau_gen_decaymode>=0 && tau_gen_decaymode<=10'

# create selections that gives events fully reconstructed offline but fail hlt
pass_all_cut = pass_gen + "&&" + pass_l1 + "&&" + pass_reco + "&&" + pass_hlt
pass_all_but_hlt = pass_gen + "&&" + pass_l1 + "&&" + pass_reco + "&& !(" + pass_hlt + ")"
# print(pass_all_but_hlt)

c = ROOT.TCanvas("c","c",600,500)
c.SetGrid()

# den is the number of taus passing only generated tau selection
den = ROOT.TH1F("den", "den", 40, 0, 40)
den.SetFillColor(ROOT.kBlue)

# 2D histogram
dhist = ROOT.TH2F("2dhist", "2dhist", 40, 0, 40, 30, -3, 3)

tch.Draw("log10(tau_gen_lxy):tau_reco_origAlgo >> 2dhist", pass_all_cut, "colz")
c.SetRightMargin(0.2)

# add axis labels
den.GetXaxis().SetTitle("algorithm index")
den.GetYaxis().SetTitle("taus")

# fill histogram
#tch.Draw("tau_reco_origAlgo >> den", pass_all_cut + "&&" + pass_prompt, "E")
#tch.Draw("tau_reco_origAlgo >> den", pass_all_but_hlt + "&&" + pass_prompt, "E")

# create box for legend
legend = ROOT.TLegend(.75, .75, .9, .95)
legend.SetBorderSize(0)
legend.SetFillStyle(0)
legend.SetTextSize(0.03)
legend.AddEntry(tch,"HLT tau","l")
#legend.AddEntry(tch,"no HLT tau","l")
legend.Draw()

# save the current canvas
c.SaveAs('track_origAlgo.pdf')
c.SaveAs('track_origAlgo.png')
c.Clear()