| 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, 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 |
|
|
| 25 |
|
#Montecarlos |
| 26 |
|
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'] |
| 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.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 |
< |
Aprefix1 = ['udscg_120_', 'b_120_','c_120_'] |
| 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 |
|
|
| 46 |
|
#genZpt > 120 |
| 49 |
|
sampletypes2 =['BKG']*3 |
| 50 |
|
samplesgroup2=['DYlight','DYb','DYc'] |
| 51 |
|
xsecs2 = [30]*3 |
| 52 |
< |
Aprefix2 = ['udscg_120_', 'b_120_','c_120_'] |
| 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 |
|
|
| 56 |
|
#The Data |
| 61 |
|
AdataPrefix = [''] |
| 62 |
|
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)))'] |
| 63 |
|
|
| 64 |
+ |
|
| 65 |
+ |
|
| 66 |
+ |
|
| 67 |
+ |
|
| 68 |
|
info = [] |
| 69 |
|
|
| 70 |
|
#prepare the files: |
| 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]) |
| 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] |
| 81 |
|
info[-1].lumi=lumi |
| 82 |
|
info[-1].prefix=prefix |
| 83 |
|
info[-1].xsec=xsecs0[i] |
| 84 |
+ |
info[-1].sf=SF0[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]) |
| 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] |
| 94 |
|
info[-1].lumi=lumi |
| 95 |
|
info[-1].prefix=prefix |
| 96 |
|
info[-1].xsec=xsecs1[i] |
| 97 |
+ |
info[-1].sf=SF1[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]) |
| 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] |
| 107 |
|
info[-1].lumi=lumi |
| 108 |
|
info[-1].prefix=prefix |
| 109 |
|
info[-1].xsec=xsecs2[i] |
| 110 |
+ |
info[-1].sf=SF2[i] |
| 111 |
|
info[-1].addtreecut(cut2[i]) |
| 112 |
|
|
| 113 |
|
for i in range(0,len(dataFiles)): |
| 114 |
< |
#copytree(pathIN,pathOUT,prefix,dataFiles[i],AdataPrefix[0],datacuts[i]) |
| 114 |
> |
copytree(pathIN,pathOUT,prefix,dataFiles[i],AdataPrefix[0],datacuts[i]+Precut) |
| 115 |
|
info.append(sample(datanames[i],datatypes[i])) |
| 116 |
|
info[-1].path=pathOUT |
| 117 |
|
info[-1].identifier=AdataPrefix[0]+dataFiles[i] |
