ATLAS Offline Software
04Plot.py
Go to the documentation of this file.
1 #!/usr/bin/env python
2 
3 # Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
4 import TopExamples.plot
5 import ROOT
6 
7 integratedLuminosity = 1. #fb-1
8 dilepton = True
9 
10 stack = [
11  TopExamples.plot.Proc('test1/Data15.root', ROOT.kBlack, 'Data', False),
12  TopExamples.plot.Proc('test1/13TeV_FS_ttbar_PowhegPythiaEvtGen.root', ROOT.kRed, 'Powheg+Pythia t#bar{t}'),
13 # TopExamples.plot.Proc('test1/147770.Sherpa_CT10_Zee.e1434_s1933_s1911_r5591_r5625.22_08_1_output.root', ROOT.kBlue, 'Sherpa Zee'),
14 # TopExamples.plot.Proc('test1/147771.Sherpa_CT10_Zmumu.e1434_s1933_s1911_r5591_r5625.22_08_1_output.root', ROOT.kGreen, 'Sherpa Zmumu'),
15 ]
16 
17 channels = ['ejets', 'mujets']
18 if dilepton:
19  channels = ['ee', 'mumu', 'emu']
20 
21 p = TopExamples.plot.Plotter('index.html', stack, integratedLuminosity)
22 
23 if dilepton:
24  p.h2('Some control plots')
25 
26  p.h3('met in Z window, at least two jets')
27  p.plot('ee/control_met')
28  p.plot('mumu/control_met')
29  p.br()
30 
31  p.h3('invariant mass for met < 60 + >= 2 jets')
32  p.plot('ee/control_invmass')
33  p.plot('mumu/control_invmass')
34  p.br()
35 
36  p.h3('njet for met < 60 in Z-window')
37  p.plot('ee/control_njet')
38  p.plot('mumu/control_njet')
39  p.br()
40 
41  p.h3('dilepton invariant mass for met < 60 GeV, >= 2 jets')
42  p.plot('ee/control_invmass')
43  p.plot('mumu/control_invmass')
44  p.br()
45 
46  p.h3('in Z window, met > 30, >= 2 jet')
47  p.plot('ee/dy_dphi')
48  p.plot('mumu/dy_dphi')
49  p.br()
50 
51  p.plot('ee/dy_dphi_zpt')
52  p.plot('mumu/dy_dphi_zpt')
53  p.br()
54 
55  p.plot('ee/dy_counter')
56  p.plot('mumu/dy_counter')
57  p.br()
58 
59  p.h2('All cuts except <something> style plots')
60 
61  p.h3('all cuts except Z window (then with Z window)')
62  p.plot('ee/control_sig_invmass')
63  p.plot('mumu/control_sig_invmass')
64  p.br()
65 
66  p.plot('ee/control_sig_invmass_cut')
67  p.plot('mumu/control_sig_invmass_cut')
68  p.br()
69 
70  p.h3('all cuts except njet')
71  p.plot('ee/control_sig_njet')
72  p.plot('mumu/control_sig_njet')
73  p.br()
74 
75  p.h3('all cuts except met')
76  p.plot('ee/control_sig_met')
77  p.plot('mumu/control_sig_met')
78  p.br()
79 
80  p.h3('emu channel ht')
81  p.plot('emu/ht')
82  p.br()
83 
84 for channel in channels:
85  p.h2(channel)
86 
87  p.h3('event')
88  p.plot(channel + '/mu')
89  p.plot(channel + '/mc_weight')
90  p.br()
91 
92  p.h3('el')
93  p.plot(channel + '/el_n')
94  p.plot(channel + '/el_pt')
95  p.plot(channel + '/el_eta')
96  p.plot(channel + '/el_phi')
97  p.plot(channel + '/el_e')
98  p.plot(channel + '/el_m')
99  p.plot(channel + '/el_charge')
100  p.br()
101 
102  p.h3('mu')
103  p.plot(channel + '/mu_n')
104  p.plot(channel + '/mu_pt')
105  p.plot(channel + '/mu_eta')
106  p.plot(channel + '/mu_phi')
107  p.plot(channel + '/mu_e')
108  p.plot(channel + '/mu_m')
109  p.plot(channel + '/mu_charge')
110  p.br()
111 
112  p.h3('jet')
113  p.plot(channel + '/jet_n')
114  p.plot(channel + '/jet_pt')
115  p.plot(channel + '/jet_eta')
116  p.plot(channel + '/jet_phi')
117  p.plot(channel + '/jet_e')
118  p.plot(channel + '/jet_mv2c20')
119  p.br()
120 
121  p.h3('leading jet')
122  p.plot(channel + '/jet0_pt')
123  p.plot(channel + '/jet0_eta')
124  p.plot(channel + '/jet0_phi')
125  p.plot(channel + '/jet0_e')
126  p.br()
127 
128  p.h3('sub-leading jet')
129  p.plot(channel + '/jet1_pt')
130  p.plot(channel + '/jet1_eta')
131  p.plot(channel + '/jet1_phi')
132  p.plot(channel + '/jet1_e')
133  p.br()
134 
135  p.h3('sub-sub-leading jet')
136  p.plot(channel + '/jet2_pt')
137  p.plot(channel + '/jet2_eta')
138  p.plot(channel + '/jet2_phi')
139  p.plot(channel + '/jet2_e')
140  p.br()
141 
142  p.h3('sub-sub-sub-leading jet')
143  p.plot(channel + '/jet3_pt')
144  p.plot(channel + '/jet3_eta')
145  p.plot(channel + '/jet3_phi')
146  p.plot(channel + '/jet3_e')
147  p.br()
148 
149  p.h3('met')
150  p.plot(channel + '/met_et')
151  p.plot(channel + '/met_phi')
152 
153 p.doneAndDusted()