5 |
|
import pickle |
6 |
|
import sys |
7 |
|
|
8 |
< |
prefix='ZllH.May23.' |
8 |
> |
prefix='ZllH.May5.' |
9 |
|
|
10 |
< |
lumi=4980 |
11 |
< |
|
12 |
< |
#pathIN='/scratch/May08/' |
13 |
< |
#pathOUT='/scratch/May08/env/' |
10 |
> |
lumi=4983.4 |
11 |
|
|
12 |
|
pathIN=sys.argv[1] |
13 |
|
pathOUT=sys.argv[2] |
15 |
|
weightexpression='((0.46502*PUweight+0.53498*PUweight2011B)*weightTrig)' |
16 |
|
|
17 |
|
#this is only to speed it up, remove for final trees! |
18 |
< |
Precut='& V.pt > 50 & V.mass > 75. & V.mass < 105 & hJet_pt[0] > 20. & hJet_pt[1] > 20. & abs(hJet_eta[0]) < 2.4 & abs(hJet_eta[1]) < 2.4' |
18 |
> |
Precut='' |
19 |
|
|
20 |
|
|
21 |
|
|
22 |
|
#Montecarlos |
23 |
< |
InFiles0 = ['WW_TuneZ2_7TeV_pythia6_tauola', 'WZ_TuneZ2_7TeV_pythia6_tauola','ZZ_TuneZ2_7TeV_pythia6_tauola','TTJets_TuneZ2_7TeV-madgraph-tauola','T_TuneZ2_s-channel_7TeV-powheg-tauola','T_TuneZ2_t-channel_7TeV-powheg-tauola', 'T_TuneZ2_tW-channel-DR_7TeV-powheg-tauola','Tbar_TuneZ2_s-channel_7TeV-powheg-tauola','Tbar_TuneZ2_t-channel_7TeV-powheg-tauola', 'Tbar_TuneZ2_tW-channel-DR_7TeV-powheg-tauola','ZH_ZToLL_HToBB_M-110_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-115_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-120_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-125_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-130_7TeV-powheg-herwigpp'] |
24 |
< |
samplenames0 =['WW','WZ','ZZ','TT','ST_s','ST_t','ST_tW','STbar_s','STbar_t','STbar_tW','ZH110','ZH115','ZH120','ZH125','ZH130'] |
25 |
< |
sampletypes0 =['BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','SIG','SIG','SIG','SIG','SIG'] |
26 |
< |
samplesgroup0=['VV','VV','VV','TT','ST','ST','ST','ST','ST','ST','ZH','ZH','ZH','ZH','ZH'] |
27 |
< |
xsecs0 = [42.9, 18.3, 5.9, 165, 3.19, 41.92, 7.87, 1.44, 22.65, 7.87, 0.4721*0.100974*0.745, 0.4107*0.100974*0.704, 0.3598*0.100974*0.648, 0.3158*0.100974*0.577, 0.2778*0.100974*0.493] |
28 |
< |
SF0=[1.0,1.0,1.0,1.04,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0] |
23 |
> |
InFiles0 = ['WW_TuneZ2_7TeV_pythia6_tauola', 'WZ_TuneZ2_7TeV_pythia6_tauola','ZZ_TuneZ2_7TeV_pythia6_tauola','TTJets_TuneZ2_7TeV-madgraph-tauola','T_TuneZ2_s-channel_7TeV-powheg-tauola','T_TuneZ2_t-channel_7TeV-powheg-tauola', 'T_TuneZ2_tW-channel-DR_7TeV-powheg-tauola','Tbar_TuneZ2_s-channel_7TeV-powheg-tauola','Tbar_TuneZ2_t-channel_7TeV-powheg-tauola', 'Tbar_TuneZ2_tW-channel-DR_7TeV-powheg-tauola','ZH_ZToLL_HToBB_M-110_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-115_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-120_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-125_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-130_7TeV-powheg-herwigpp','ZH_ZToLL_HToBB_M-135_7TeV-powheg-herwigpp'] |
24 |
> |
samplenames0 =['WW','WZ','ZZ','TT','ST_s','ST_t','ST_tW','STbar_s','STbar_t','STbar_tW','ZH110','ZH115','ZH120','ZH125','ZH130','ZH135'] |
25 |
> |
sampletypes0 =['BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','SIG','SIG','SIG','SIG','SIG','SIG'] |
26 |
> |
samplesgroup0=['VV','VV','VV','TT','ST','ST','ST','ST','ST','ST','ZH','ZH','ZH','ZH','ZH','ZH'] |
27 |
> |
xsecs0 = [42.9, 18.3, 5.9, 165, 3.19, 41.92, 7.87, 1.44, 22.65, 7.87, 0.4721*0.100974*0.745, 0.4107*0.100974*0.704, 0.3598*0.100974*0.648, 0.3158*0.100974*0.577, 0.2778*0.100974*0.493,0.2453*0.100974*0.403] |
28 |
> |
SF0=[1.0,1.0,1.0,1.032,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0] |
29 |
|
Aprefix0 = [''] |
30 |
< |
cut0 = ['(Vtype == 1 || Vtype == 0) && EVENT.json && hbhe'] |
30 |
> |
cut0 = ['(Vtype == 1 || Vtype == 0)'] |
31 |
|
|
32 |
|
#split DY in flavours |
33 |
|
#genZpt < 120 |
36 |
|
sampletypes1 =['BKG']*3 |
37 |
|
samplesgroup1=['DYlight','DYb','DYc'] |
38 |
|
xsecs1 = [3048]*3 |
39 |
< |
SF1=[1.00,0.95,1.24] |
39 |
> |
SF1=[1.108,0.984,1.108] |
40 |
|
Aprefix1 = ['udsg_120_', 'b_120_','c_120_'] |
41 |
< |
cut1 = ['(Vtype == 1 || Vtype == 0) & EVENT.json & hbhe & eventFlav != 5 & eventFlav != 4 & genZpt <= 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & hbhe & eventFlav == 5 & genZpt <= 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & hbhe & eventFlav == 4 & genZpt <= 120.'] |
41 |
> |
cut1 = ['(Vtype == 1 || Vtype == 0) & EVENT.json & eventFlav != 5 & eventFlav != 4 & genZpt <= 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & eventFlav == 5 & genZpt <= 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & eventFlav == 4 & genZpt <= 120.'] |
42 |
|
|
43 |
|
#genZpt > 120 |
44 |
|
InFiles2 = ['DYJetsToLL_PtZ-100_TuneZ2_7TeV-madgraph-tauola']*3 |
46 |
|
sampletypes2 =['BKG']*3 |
47 |
|
samplesgroup2=['DYlight','DYb','DYc'] |
48 |
|
xsecs2 = [30]*3 |
49 |
< |
SF2=[1.00,0.95,1.24] |
49 |
> |
SF2=[1.108,0.984,1.108] |
50 |
|
Aprefix2 = ['udsg_120_', 'b_120_','c_120_'] |
51 |
< |
cut2 = ['(Vtype == 1 || Vtype == 0) & EVENT.json & hbhe & eventFlav != 5 & eventFlav != 4 & genZpt > 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & hbhe & eventFlav == 5 & genZpt > 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & hbhe & eventFlav == 4 & genZpt > 120.'] |
51 |
> |
cut2 = ['(Vtype == 1 || Vtype == 0) & EVENT.json & eventFlav != 5 & eventFlav != 4 & genZpt > 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & eventFlav == 5 & genZpt > 120.','(Vtype == 1 || Vtype == 0) & EVENT.json & eventFlav == 4 & genZpt > 120.'] |
52 |
|
|
53 |
|
#The Data |
54 |
|
dataFiles = ['DataZee', 'DataZmm'] |
56 |
|
datatypes =['DATA','DATA'] |
57 |
|
datagroup=['DATA','DATA'] |
58 |
|
AdataPrefix = [''] |
59 |
< |
datacuts=['(Vtype==1 && EVENT.json && hbhe) && (triggerFlags[5] | triggerFlags[6])','(Vtype==0 && EVENT.json && hbhe)&&((EVENT.run<173198 && (triggerFlags[0]>0 || triggerFlags[13]>0 || triggerFlags[14]>0 || triggerFlags[20]>0 || triggerFlags[21]>0)) || (EVENT.run>=173198 && EVENT.run<175832 && (triggerFlags[13]>0 ||triggerFlags[14]>0 || triggerFlags[22]>0 || triggerFlags[23]>0))|| (EVENT.run>=175832 && EVENT.run<178390 && (triggerFlags[13]>0 ||triggerFlags[14]>0 ||triggerFlags[15]>0 || triggerFlags[21]>0 || triggerFlags[22]>0 || triggerFlags[23]>0)) || (EVENT.run>=178390 && (triggerFlags[14]>0 ||triggerFlags[15]>0 || triggerFlags[21]>0 || triggerFlags[22]>0 || triggerFlags[23]>0 || triggerFlags[24]>0 || triggerFlags[25]>0 || triggerFlags[26]>0 || triggerFlags[27]>0)))'] |
59 |
> |
datacuts=['(Vtype==1)','(Vtype==0)'] |
60 |
|
|
61 |
|
|
62 |
|
|
69 |
|
# i>1 because no WW sample!!! only temporary fix |
70 |
|
|
71 |
|
for i in range(0,len(InFiles0)): |
72 |
< |
#copytree(pathIN,pathOUT,prefix,InFiles0[i],Aprefix0[0],cut0[0]+Precut) |
72 |
> |
copytree(pathIN,pathOUT,prefix,InFiles0[i],Aprefix0[0],cut0[0]+Precut) |
73 |
|
info.append(sample(samplenames0[i],sampletypes0[i])) |
74 |
|
info[-1].path=pathOUT |
75 |
|
info[-1].identifier=Aprefix0[0]+InFiles0[i] |
82 |
|
info[-1].addtreecut(cut0[0]) |
83 |
|
|
84 |
|
for i in range(3): |
85 |
< |
#copytree(pathIN,pathOUT,prefix,InFiles1[i],Aprefix1[i],cut1[i]+Precut) |
85 |
> |
copytree(pathIN,pathOUT,prefix,InFiles1[i],Aprefix1[i],cut1[i]+Precut) |
86 |
|
info.append(sample(samplenames1[i],sampletypes1[i])) |
87 |
|
info[-1].path=pathOUT |
88 |
|
info[-1].identifier=Aprefix1[i]+InFiles1[i] |
95 |
|
info[-1].addtreecut(cut1[i]) |
96 |
|
|
97 |
|
for i in range(3): |
98 |
< |
#copytree(pathIN,pathOUT,prefix,InFiles2[i],Aprefix2[i],cut2[i]+Precut) |
98 |
> |
copytree(pathIN,pathOUT,prefix,InFiles2[i],Aprefix2[i],cut2[i]+Precut) |
99 |
|
info.append(sample(samplenames2[i],sampletypes2[i])) |
100 |
|
info[-1].path=pathOUT |
101 |
|
info[-1].identifier=Aprefix2[i]+InFiles2[i] |