-
Notifications
You must be signed in to change notification settings - Fork 30
/
Copy pathBoostedWSamples.py
204 lines (177 loc) · 15.8 KB
/
BoostedWSamples.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
#!/usr/bin/env python
########################################
##
## Author: Wei Zou
##
## Email: [email protected]
#######################################
from ROOT import *
####### define the general structure for samples
class Samples:
def __init__(self, CHANNEL):
self.filenames = {} ## same of the input file
self.filepath = "" ## file name path
self.luminosity = 0. ## luminosity to be considered
self.treename = "" ## name of the tree
self.channel = CHANNEL ## channel
def SetFilePath(self,path):
self.filepath = path
def SetLumi(self,lumi):
self.luminosity = lumi
def SetTreeName(self,tree):
self.treename = tree
def SetFileNames(self):
if self.channel == "mu":
self.filenames["data"] = self.filepath + "RD_WmunuJets_DataAll_GoldenJSON_19p3invfb.root"
self.filenames["TTbar"] = self.filepath + "RD_mu_TTbar_CMSSW532.root"
self.filenames["TTbar_matchDn"] = self.filepath + "RD_mu_TTbar_matchingdown_CMSSW532.root"
self.filenames["TTbar_matchUp"] = self.filepath + "RD_mu_TTbar_matchingup_CMSSW532.root"
self.filenames["TTbar_Powheg"] = self.filepath + "RD_mu_TTbar_powheg_CMSSW532.root"
self.filenames["TTbar_scaleUp"] = self.filepath + "RD_mu_TTbar_scaleup_CMSSW532.root"
self.filenames["TTbar_scaleDn"] = self.filepath + "RD_mu_TTbar_scaledown_CMSSW532.root"
self.filenames["WJets_Pythia"] = self.filepath + "RD_mu_WJets_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_matchUp"] = self.filepath + "RD_mu_WJets_matchingup_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_matchDw"] = self.filepath + "RD_mu_WJets_matchingdown_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_scaleUp"] = self.filepath + "RD_mu_WJets_scaleup_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_scaleDw"] = self.filepath + "RD_mu_WJets_scaledown_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia180"] = self.filepath + "RD_mu_WpJPt180_CMSSW532.root"
self.filenames["WJets_Pythia100"] = self.filepath + "RD_mu_WpJPt100_CMSSW532.root";
self.filenames["WJets_Herwig"] = self.filepath + "RD_mu_WpJPt100_herwig_CMSSW532.root";
self.filenames["W1Jets_Pythia"] = self.filepath + "RD_mu_W1Jets_CMSSW532.root";
self.filenames["W2Jets_Pythia"] = self.filepath + "RD_mu_W2Jets_CMSSW532.root";
self.filenames["W3Jets_Pythia"] = self.filepath + "RD_mu_W3Jets_CMSSW532.root";
self.filenames["W4Jets_Pythia"] = self.filepath + "RD_mu_W4Jets_CMSSW532.root";
self.filenames["ZJets"] = self.filepath + "RD_mu_ZpJ_CMSSW532.root"
self.filenames["WW"] = self.filepath + "RD_mu_WW_CMSSW532.root"
self.filenames["WZ"] = self.filepath + "RD_mu_WZ_CMSSW532.root"
self.filenames["ZZ"] = self.filepath + "RD_mu_ZZ_CMSSW532.root"
self.filenames["tch"] = self.filepath + "RD_mu_STopT_T_CMSSW532.root"
self.filenames["tWch"] = self.filepath + "RD_mu_STopTW_T_CMSSW532.root"
self.filenames["sch"] = self.filepath + "RD_mu_STopS_T_CMSSW532.root"
self.filenames["tch_bar"] = self.filepath + "RD_mu_STopT_Tbar_CMSSW532.root"
self.filenames["tWch_bar"] = self.filepath + "RD_mu_STopTW_Tbar_CMSSW532.root"
self.filenames["sch_bar"] = self.filepath + "RD_mu_STopS_Tbar_CMSSW532.root"
self.filenames["ggH600"] = self.filepath + "RD_mu_HWWMH600_CMSSW532_private.root"
self.filenames["ggH700"] = self.filepath + "RD_mu_HWWMH700_CMSSW532_private.root"
self.filenames["ggH800"] = self.filepath + "RD_mu_HWWMH800_CMSSW532_private.root"
self.filenames["ggH900"] = self.filepath + "RD_mu_HWWMH900_CMSSW532_private.root"
self.filenames["ggH1000"] = self.filepath + "RD_mu_HWWMH1000_CMSSW532_private.root"
self.filenames["vbfH600"] = self.filepath + "RD_mu_VBFHWWMH600_CMSSW532_private.root"
self.filenames["vbfH700"] = self.filepath + "RD_mu_VBFHWWMH700_CMSSW532_private.root"
self.filenames["vbfH800"] = self.filepath + "RD_mu_VBFHWWMH800_CMSSW532_private.root"
self.filenames["vbfH900"] = self.filepath + "RD_mu_VBFHWWMH900_CMSSW532_private.root"
self.filenames["vbfH1000"] = self.filepath + "RD_mu_VBFHWWMH1000_CMSSW532_private.root"
self.filenames["rsg1000_kMpl01_py"] = self.filepath + "RD_mu_RSGravitonToWW_kMpl01_M-1000_pythia_CMSSW532_private.root"
self.filenames["rsg1000_kMpl01_hw"] = self.filepath + "RD_mu_RSGravitonToWW_kMpl01_M-1000_herwig_CMSSW532_private.root"
self.filenames["rsg1500_kMpl01_py"] = self.filepath + "RD_mu_RSGravitonToWW_kMpl01_M-1500_pythia_CMSSW532_private.root"
self.filenames["rsg1500_kMpl01_hw"] = self.filepath + "RD_mu_RSGravitonToWW_kMpl01_M-1500_herwig_CMSSW532_private.root"
self.filenames["rsg2000_kMpl01_py"] = self.filepath + "RD_mu_RSGravitonToWW_kMpl01_M-2000_pythia_CMSSW532_private.root"
self.filenames["BulkG_c0p2_M600"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M600.root"
self.filenames["BulkG_c0p2_M700"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M700.root"
self.filenames["BulkG_c0p2_M800"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M800.root"
self.filenames["BulkG_c0p2_M900"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M900.root"
self.filenames["BulkG_c0p2_M1000"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1000.root"
self.filenames["BulkG_c0p2_M1100"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1100.root"
self.filenames["BulkG_c0p2_M1200"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1200.root"
self.filenames["BulkG_c0p2_M1300"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1300.root"
self.filenames["BulkG_c0p2_M1400"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1400.root"
self.filenames["BulkG_c0p2_M1500"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1500.root"
self.filenames["BulkG_c0p2_M1600"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1600.root"
self.filenames["BulkG_c0p2_M1700"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1700.root"
self.filenames["BulkG_c0p2_M1800"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1800.root"
self.filenames["BulkG_c0p2_M1900"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M1900.root"
self.filenames["BulkG_c0p2_M2000"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M2000.root"
self.filenames["BulkG_c0p2_M2100"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M2100.root"
self.filenames["BulkG_c0p2_M2200"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M2200.root"
self.filenames["BulkG_c0p2_M2300"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M2300.root"
self.filenames["BulkG_c0p2_M2400"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M2400.root"
self.filenames["BulkG_c0p2_M2500"] = self.filepath + "RD_mu_BulkG_WW_lvjj_c0p2_M2500.root"
elif self.channel == "el":
self.filenames["data"] = self.filepath + "RD_WenuJets_DataAllSingleElectronTrigger_GoldenJSON_19p2invfb.root"
self.filenames["TTbar"] = self.filepath + "RD_el_TTbar_CMSSW532.root";
self.filenames["TTbar_matchDn"] = self.filepath + "RD_el_TTbar_matchingdown_CMSSW532.root"
self.filenames["TTbar_matchUp"] = self.filepath + "RD_el_TTbar_matchingup_CMSSW532.root"
self.filenames["TTbar_Powheg"] = self.filepath + "RD_el_TTbar_powheg_CMSSW532.root"
self.filenames["TTbar_scaleDn"] = self.filepath + "RD_el_TTbar_scaleup_CMSSW532.root"
self.filenames["TTbar_scaleUp"] = self.filepath + "RD_el_TTbar_scaledown_CMSSW532.root"
self.filenames["WJets_Pythia"] = self.filepath + "RD_el_WJets_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_matchUp"] = self.filepath + "RD_el_WJets_matchingup_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_matchDw"] = self.filepath + "RD_el_WJets_matchingdown_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_scaleUp"] = self.filepath + "RD_el_WJets_scaleup_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia_scaleDw"] = self.filepath + "RD_el_WJets_scaledown_madgraph_CMSSW532.root";
self.filenames["WJets_Pythia180"] = self.filepath + "RD_el_WpJPt180_CMSSW532.root"
self.filenames["WJets_Pythia100"] = self.filepath + "RD_el_WpJPt100_CMSSW532.root";
self.filenames["WJets_Herwig"] = self.filepath + "RD_el_WpJPt100_herwig_CMSSW532.root";
self.filenames["W1Jets_Pythia"] = self.filepath + "RD_el_W1Jets_CMSSW532.root";
self.filenames["W2Jets_Pythia"] = self.filepath + "RD_el_W2Jets_CMSSW532.root";
self.filenames["W3Jets_Pythia"] = self.filepath + "RD_el_W3Jets_CMSSW532.root";
self.filenames["W4Jets_Pythia"] = self.filepath + "RD_el_W4Jets_CMSSW532.root";
self.filenames["ZJets"] = self.filepath + "RD_el_ZpJ_CMSSW532.root"
self.filenames["WW"] = self.filepath + "RD_el_WW_CMSSW532.root"
self.filenames["WZ"] = self.filepath + "RD_el_WZ_CMSSW532.root"
self.filenames["ZZ"] = self.filepath + "RD_el_ZZ_CMSSW532.root"
self.filenames["tch"] = self.filepath + "RD_el_STopT_T_CMSSW532.root"
self.filenames["tWch"] = self.filepath + "RD_el_STopTW_T_CMSSW532.root"
self.filenames["sch"] = self.filepath + "RD_el_STopS_T_CMSSW532.root"
self.filenames["tch_bar"] = self.filepath + "RD_el_STopT_Tbar_CMSSW532.root"
self.filenames["tWch_bar"] = self.filepath + "RD_el_STopTW_Tbar_CMSSW532.root"
self.filenames["sch_bar"] = self.filepath + "RD_el_STopS_Tbar_CMSSW532.root"
self.filenames["ggH600"] = self.filepath + "RD_el_HWWMH600_CMSSW532_private.root"
self.filenames["ggH700"] = self.filepath + "RD_el_HWWMH700_CMSSW532_private.root"
self.filenames["ggH800"] = self.filepath + "RD_el_HWWMH800_CMSSW532_private.root"
self.filenames["ggH900"] = self.filepath + "RD_el_HWWMH900_CMSSW532_private.root"
self.filenames["ggH1000"] = self.filepath + "RD_el_HWWMH1000_CMSSW532_private.root"
self.filenames["vbfH600"] = self.filepath + "RD_el_VBFHWWMH600_CMSSW532_private.root"
self.filenames["vbfH700"] = self.filepath + "RD_el_VBFHWWMH700_CMSSW532_private.root"
self.filenames["vbfH800"] = self.filepath + "RD_el_VBFHWWMH800_CMSSW532_private.root"
self.filenames["vbfH900"] = self.filepath + "RD_el_VBFHWWMH900_CMSSW532_private.root"
self.filenames["vbfH1000"] = self.filepath + "RD_el_VBFHWWMH1000_CMSSW532_private.root"
self.filenames["rsg1000_kMpl01_py"] = self.filepath + "RD_el_RSGravitonToWW_kMpl01_M-1000_pythia_CMSSW532_private.root"
self.filenames["rsg1000_kMpl01_hw"] = self.filepath + "RD_el_RSGravitonToWW_kMpl01_M-1000_herwig_CMSSW532_private.root"
self.filenames["rsg1500_kMpl01_py"] = self.filepath + "RD_el_RSGravitonToWW_kMpl01_M-1500_pythia_CMSSW532_private.root"
self.filenames["rsg1500_kMpl01_hw"] = self.filepath + "RD_el_RSGravitonToWW_kMpl01_M-1500_herwig_CMSSW532_private.root"
self.filenames["rsg2000_kMpl01_py"] = self.filepath + "RD_el_RSGravitonToWW_kMpl01_M-2000_pythia_CMSSW532_private.root"
self.filenames["BulkG_c0p2_M600"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M600.root"
self.filenames["BulkG_c0p2_M700"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M700.root"
self.filenames["BulkG_c0p2_M800"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M800.root"
self.filenames["BulkG_c0p2_M900"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M900.root"
self.filenames["BulkG_c0p2_M1000"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1000.root"
self.filenames["BulkG_c0p2_M1100"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1100.root"
self.filenames["BulkG_c0p2_M1200"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1200.root"
self.filenames["BulkG_c0p2_M1300"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1300.root"
self.filenames["BulkG_c0p2_M1400"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1400.root"
self.filenames["BulkG_c0p2_M1500"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1500.root"
self.filenames["BulkG_c0p2_M1600"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1600.root"
self.filenames["BulkG_c0p2_M1700"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1700.root"
self.filenames["BulkG_c0p2_M1800"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1800.root"
self.filenames["BulkG_c0p2_M1900"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M1900.root"
self.filenames["BulkG_c0p2_M2000"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M2000.root"
self.filenames["BulkG_c0p2_M2100"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M2100.root"
self.filenames["BulkG_c0p2_M2200"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M2200.root"
self.filenames["BulkG_c0p2_M2300"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M2300.root"
self.filenames["BulkG_c0p2_M2400"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M2400.root"
self.filenames["BulkG_c0p2_M2500"] = self.filepath + "RD_el_BulkG_WW_lvjj_c0p2_M2500.root"
def GetLumiScaleFactor(self,txtfile,keyname): ### get lumi scale factor from the file
multiplicitylabel = 1.0
scalefactor = 1.0
samplename ="";
SFfile = open(txtfile)
for sfline in SFfile:
if sfline.find("#")!=-1: continue
if(sfline.find(keyname) != -1):
samplename = string(sfline.split()[0])
if(samplename !=keyname): continue;
scalefactor = float(sfline.split()[1])
if len(sfline.split()) > 2:
multiplicitylabel = float(sfline.split()[2])
scalefactor = scalefactor * multiplicitylabel
break
SFfile.close()
return scalefactor
### other get functions
def GetFileNames(self):
return self.filenames
def GetLumi(self):
return self.luminosity
def GetTreeName(self):
return self.treename