forked from tiwariPC/hhbbgg_AwkwardAnalyzer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathvariables.py
123 lines (118 loc) · 4.31 KB
/
variables.py
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import copy
regions = ["preselection", "selection", "srbbgg", "srbbggMET", "crantibbgg", "crbbantigg"]
vardict = {
"dibjet_mass": "dibjet_mass",
"diphoton_mass": "diphoton_mass",
"bbgg_mass": "bbgg_mass",
"dibjet_pt": "dibjet_pt",
"diphoton_pt": "diphoton_pt",
"bbgg_pt": "bbgg_pt",
"bbgg_eta": "bbgg_eta",
"bbgg_phi": "bbgg_phi",
"lead_pho_pt": "lead_pho_pt",
"sublead_pho_pt": "sublead_pho_pt",
"lead_pho_eta": "lead_pho_eta",
"lead_pho_phi": "lead_pho_phi",
"sublead_pho_eta": "sublead_pho_eta",
"sublead_pho_phi": "sublead_pho_phi",
"diphoton_eta": "diphoton_eta",
"diphoton_phi": "diphoton_phi",
"dibjet_eta": "dibjet_eta",
"dibjet_phi": "dibjet_phi",
# ----bjet-----
"lead_bjet_pt": "lead_bjet_pt",
"sublead_bjet_pt": "sublead_bjet_pt",
"lead_bjet_eta": "lead_bjet_eta",
"lead_bjet_phi": "lead_bjet_phi",
"sublead_bjet_eta": "sublead_bjet_eta",
"sublead_bjet_phi": "sublead_bjet_phi",
"sublead_bjet_PNetB": "sublead_bjet_PNetB",
"lead_bjet_PNetB": "lead_bjet_PNetB",
"CosThetaStar_gg": "CosThetaStar_gg",
"CosThetaStar_jj": "CosThetaStar_jj",
"CosThetaStar_CS": "CosThetaStar_CS",
"DeltaR_jg_min": "DeltaR_jg_min",
"pholead_PtOverM": "pholead_PtOverM",
"phosublead_PtOverM": "phosublead_PtOverM",
"FirstJet_PtOverM": "FirstJet_PtOverM",
"SecondJet_PtOverM": "SecondJet_PtOverM",
"lead_pt_over_diphoton_mass": "lead_pt_over_diphoton_mass",
"sublead_pt_over_diphoton_mass": "sublead_pt_over_diphoton_mass",
"lead_pt_over_dibjet_mass": "lead_pt_over_dibjet_mass",
"sublead_pt_over_dibjet_mass": "sublead_pt_over_dibjet_mass",
"diphoton_bbgg_mass": "diphoton_bbgg_mass",
"dibjet_bbgg_mass": "dibjet_bbgg_mass",
"puppiMET_pt": "puppiMET_pt",
"puppiMET_phi": "puppiMET_phi",
"puppiMET_phiJERDown": "puppiMET_phiJERDown",
"puppiMET_phiJERUp": "puppiMET_phiJERUp",
"puppiMET_phiJESDown": "puppiMET_phiJESDown",
"puppiMET_phiJESUp": "puppiMET_phiJESUp",
"puppiMET_phiUnclusteredDown": "puppiMET_phiUnclusteredDown",
"puppiMET_phiUnclusteredUp": "puppiMET_phiUnclusteredUp",
"puppiMET_ptJERDown":"puppiMET_ptJERDown",
"puppiMET_ptJERUp":"puppiMET_ptJERUp",
"puppiMET_ptJESDown":"puppiMET_ptJESDown",
"puppiMET_ptJESUp":"puppiMET_ptJESUp",
"lead_pho_mvaID_WP90":"lead_pho_mvaID_WP90",
"lead_pho_mvaID_WP80":"lead_pho_mvaID_WP80",
"sublead_pho_mvaID_WP90":"sublead_pho_mvaID_WP90",
"sublead_pho_mvaID_WP80":"sublead_pho_mvaID_WP80",
"lead_pho_mvaID":"lead_pho_mvaID",
"sublead_pho_mvaID":"sublead_pho_mvaID",
}
variables_common = {
"preselection": [
"dibjet_mass",
"diphoton_mass",
"bbgg_mass",
"dibjet_pt",
"diphoton_pt",
"bbgg_pt",
"lead_pho_pt",
"sublead_pho_pt",
"bbgg_eta",
"bbgg_phi",
"lead_pho_eta",
"lead_pho_phi",
"sublead_pho_eta",
"sublead_pho_phi",
"diphoton_eta",
"diphoton_phi",
"dibjet_eta",
"dibjet_phi",
"lead_bjet_pt",
"sublead_bjet_pt",
"lead_bjet_eta",
"lead_bjet_phi",
"sublead_bjet_eta",
"sublead_bjet_phi",
"sublead_bjet_PNetB",
"lead_bjet_PNetB",
"CosThetaStar_gg",
"CosThetaStar_jj",
"CosThetaStar_CS",
"DeltaR_jg_min",
"pholead_PtOverM",
"phosublead_PtOverM",
"FirstJet_PtOverM",
"SecondJet_PtOverM",
"lead_pt_over_diphoton_mass",
"sublead_pt_over_diphoton_mass",
"lead_pt_over_dibjet_mass",
"sublead_pt_over_dibjet_mass",
"diphoton_bbgg_mass",
"dibjet_bbgg_mass",
"lead_pho_mvaID_WP90",
"lead_pho_mvaID_WP80",
"sublead_pho_mvaID_WP90",
"sublead_pho_mvaID_WP80",
"lead_pho_mvaID",
"sublead_pho_mvaID",
]
}
for ireg in regions:
print(ireg)
variables_common[ireg] = copy.deepcopy(variables_common["preselection"])
srbbggMET = ["puppiMET_pt", "puppiMET_phi", "puppiMET_phiJERDown", "puppiMET_phiJERUp","puppiMET_phiJESDown", "puppiMET_phiJESUp", "puppiMET_phiUnclusteredDown", "puppiMET_phiUnclusteredUp", "puppiMET_ptJERDown", "puppiMET_ptJERUp", "puppiMET_ptJESDown", "puppiMET_ptJESUp"]
variables_common["srbbggMET"] = variables_common["srbbggMET"] + srbbggMET