3 |
|
from printcolor import printc |
4 |
|
from samplesclass import sample |
5 |
|
import pickle |
6 |
+ |
import sys |
7 |
|
|
8 |
+ |
prefix='ZllH.May23.' |
9 |
|
|
10 |
< |
prefix='ZllH.May5Reg.' |
10 |
> |
lumi=4980 |
11 |
|
|
12 |
< |
lumi=4700 |
12 |
> |
#pathIN='/scratch/May08/' |
13 |
> |
#pathOUT='/scratch/May08/env/' |
14 |
|
|
15 |
< |
pathIN='/scratch/May05/' |
16 |
< |
pathOUT='/scratch/May05/env/' |
15 |
> |
pathIN=sys.argv[1] |
16 |
> |
pathOUT=sys.argv[2] |
17 |
|
|
18 |
|
weightexpression='((0.46502*PUweight+0.53498*PUweight2011B)*weightTrig)' |
19 |
|
|
20 |
< |
#this is only to speed it up, rwmove for final trees! |
21 |
< |
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 & deltaPullAngleAK7 < 10.' |
20 |
> |
#this is only to speed it up, remove for final trees! |
21 |
> |
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' |
22 |
|
|
23 |
|
|
24 |
|
|
28 |
|
sampletypes0 =['BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','SIG','SIG','SIG','SIG','SIG'] |
29 |
|
samplesgroup0=['VV','VV','VV','TT','ST','ST','ST','ST','ST','ST','ZH','ZH','ZH','ZH','ZH'] |
30 |
|
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] |
31 |
< |
SF0=[1.0,1.0,1.0,1.03387,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0] |
31 |
> |
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] |
32 |
|
Aprefix0 = [''] |
33 |
|
cut0 = ['(Vtype == 1 || Vtype == 0) && EVENT.json && hbhe'] |
34 |
|
|
39 |
|
sampletypes1 =['BKG']*3 |
40 |
|
samplesgroup1=['DYlight','DYb','DYc'] |
41 |
|
xsecs1 = [3048]*3 |
42 |
< |
SF1=[1.26223,0.927624,1.22381] |
42 |
> |
SF1=[1.00,0.95,1.24] |
43 |
|
Aprefix1 = ['udsg_120_', 'b_120_','c_120_'] |
44 |
|
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.'] |
45 |
|
|
49 |
|
sampletypes2 =['BKG']*3 |
50 |
|
samplesgroup2=['DYlight','DYb','DYc'] |
51 |
|
xsecs2 = [30]*3 |
52 |
< |
SF2=[1.26223,0.927624,1.22381] |
52 |
> |
SF2=[1.00,0.95,1.24] |
53 |
|
Aprefix2 = ['udsg_120_', 'b_120_','c_120_'] |
54 |
|
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.'] |
55 |
|
|
71 |
|
|
72 |
|
# i>1 because no WW sample!!! only temporary fix |
73 |
|
|
74 |
< |
for i in range(1,len(InFiles0)): |
75 |
< |
copytree(pathIN,pathOUT,prefix,InFiles0[i],Aprefix0[0],cut0[0]+Precut) |
74 |
> |
for i in range(0,len(InFiles0)): |
75 |
> |
#copytree(pathIN,pathOUT,prefix,InFiles0[i],Aprefix0[0],cut0[0]+Precut) |
76 |
|
info.append(sample(samplenames0[i],sampletypes0[i])) |
77 |
|
info[-1].path=pathOUT |
78 |
|
info[-1].identifier=Aprefix0[0]+InFiles0[i] |
85 |
|
info[-1].addtreecut(cut0[0]) |
86 |
|
|
87 |
|
for i in range(3): |
88 |
< |
copytree(pathIN,pathOUT,prefix,InFiles1[i],Aprefix1[i],cut1[i]+Precut) |
88 |
> |
#copytree(pathIN,pathOUT,prefix,InFiles1[i],Aprefix1[i],cut1[i]+Precut) |
89 |
|
info.append(sample(samplenames1[i],sampletypes1[i])) |
90 |
|
info[-1].path=pathOUT |
91 |
|
info[-1].identifier=Aprefix1[i]+InFiles1[i] |
98 |
|
info[-1].addtreecut(cut1[i]) |
99 |
|
|
100 |
|
for i in range(3): |
101 |
< |
copytree(pathIN,pathOUT,prefix,InFiles2[i],Aprefix2[i],cut2[i]+Precut) |
101 |
> |
#copytree(pathIN,pathOUT,prefix,InFiles2[i],Aprefix2[i],cut2[i]+Precut) |
102 |
|
info.append(sample(samplenames2[i],sampletypes2[i])) |
103 |
|
info[-1].path=pathOUT |
104 |
|
info[-1].identifier=Aprefix2[i]+InFiles2[i] |