VHbb/python/prepare_environment.py

#!/usr/bin/env python
from copytree import copytree
from printcolor import printc
from samplesclass import sample
import pickle
import sys

prefix='ZllH.May5.'

lumi=4983.4

pathIN=sys.argv[1]
pathOUT=sys.argv[2]

weightexpression='((0.46502*PUweight+0.53498*PUweight2011B)*weightTrig)'

#this is only to speed it up, remove for final trees!
Precut=''


#Montecarlos
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']
samplenames0 =['WW','WZ','ZZ','TT','ST_s','ST_t','ST_tW','STbar_s','STbar_t','STbar_tW','ZH110','ZH115','ZH120','ZH125','ZH130','ZH135']
sampletypes0 =['BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','BKG','SIG','SIG','SIG','SIG','SIG','SIG']
samplesgroup0=['VV','VV','VV','TT','ST','ST','ST','ST','ST','ST','ZH','ZH','ZH','ZH','ZH','ZH']
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]
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]
Aprefix0 = ['']
cut0 = ['(Vtype == 1 || Vtype == 0)']

#split DY in flavours
#genZpt < 120
InFiles1 = ['DYJetsToLL_TuneZ2_M-50_7TeV-madgraph-tauola']*3
samplenames1 =['Zudsg','Zbb','Zcc']
sampletypes1 =['BKG']*3
samplesgroup1=['DYlight','DYb','DYc']
xsecs1 = [3048]*3
SF1=[1.108,0.984,1.108]
Aprefix1 = ['udsg_120_', 'b_120_','c_120_']
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.']

#genZpt > 120
InFiles2 = ['DYJetsToLL_PtZ-100_TuneZ2_7TeV-madgraph-tauola']*3
samplenames2 =['Zudsg_pt120','Zbb_pt120','Zcc_pt120']
sampletypes2 =['BKG']*3
samplesgroup2=['DYlight','DYb','DYc']
xsecs2 = [30]*3
SF2=[1.108,0.984,1.108]
Aprefix2 = ['udsg_120_', 'b_120_','c_120_']
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.']

#The Data
dataFiles = ['DataZee', 'DataZmm']
datanames =['Zee','Zmm']
datatypes =['DATA','DATA']
datagroup=['DATA','DATA']
AdataPrefix = ['']
datacuts=['(Vtype==1)','(Vtype==0)']


info = []

#prepare the files:

# i>1 because no WW sample!!! only temporary fix

for i in range(0,len(InFiles0)):
    copytree(pathIN,pathOUT,prefix,InFiles0[i],Aprefix0[0],cut0[0]+Precut)
    info.append(sample(samplenames0[i],sampletypes0[i]))
    info[-1].path=pathOUT
    info[-1].identifier=Aprefix0[0]+InFiles0[i]
    info[-1].weightexpression=weightexpression
    info[-1].group=samplesgroup0[i]
    info[-1].lumi=lumi
    info[-1].prefix=prefix
    info[-1].xsec=xsecs0[i]
    info[-1].sf=SF0[i]
    info[-1].addtreecut(cut0[0])

for i in range(3):
    copytree(pathIN,pathOUT,prefix,InFiles1[i],Aprefix1[i],cut1[i]+Precut)
    info.append(sample(samplenames1[i],sampletypes1[i]))
    info[-1].path=pathOUT
    info[-1].identifier=Aprefix1[i]+InFiles1[i]
    info[-1].weightexpression=weightexpression
    info[-1].group=samplesgroup1[i]
    info[-1].lumi=lumi
    info[-1].prefix=prefix
    info[-1].xsec=xsecs1[i]
    info[-1].sf=SF1[i]
    info[-1].addtreecut(cut1[i])
    
for i in range(3):
    copytree(pathIN,pathOUT,prefix,InFiles2[i],Aprefix2[i],cut2[i]+Precut)
    info.append(sample(samplenames2[i],sampletypes2[i]))
    info[-1].path=pathOUT
    info[-1].identifier=Aprefix2[i]+InFiles2[i]
    info[-1].weightexpression=weightexpression
    info[-1].group=samplesgroup2[i]
    info[-1].lumi=lumi
    info[-1].prefix=prefix
    info[-1].xsec=xsecs2[i]
    info[-1].sf=SF2[i]
    info[-1].addtreecut(cut2[i])
 
for i in range(0,len(dataFiles)):
    copytree(pathIN,pathOUT,prefix,dataFiles[i],AdataPrefix[0],datacuts[i]+Precut)
    info.append(sample(datanames[i],datatypes[i]))
    info[-1].path=pathOUT
    info[-1].identifier=AdataPrefix[0]+dataFiles[i]
    info[-1].group=datagroup[i]
    info[-1].lumi=lumi
    info[-1].prefix=prefix
    info[-1].addtreecut(datacuts[i])


#dump info   
infofile = open(pathOUT+'/samples.info','w')
pickle.dump(info,infofile)
infofile.close()
Revision:	1.11
Committed:	Wed Jan 16 16:22:46 2013 UTC (12 years, 4 months ago) by peller
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.10:	+0 -0 lines
State:	*FILE REMOVED*
Log Message:	reorganized the whole repository. Macros im myutils, config files in subdirectories. Config file split in parts. Path config file restructured. Moved all path options to the path config. Changed the code accordingly.
#	User	Rev	Content
1	peller	1.1	#!/usr/bin/env python
2			from copytree import copytree
3			from printcolor import printc
4			from samplesclass import sample
5			import pickle
6	peller	1.6	import sys
7	peller	1.1
8	peller	1.9	prefix='ZllH.May5.'
9	peller	1.1
10	nmohr	1.10	lumi=4983.4
11	peller	1.1
12	peller	1.6	pathIN=sys.argv[1]
13			pathOUT=sys.argv[2]
14	peller	1.1
15			weightexpression='((0.46502PUweight+0.53498PUweight2011B)*weightTrig)'
16
17	peller	1.3	#this is only to speed it up, remove for final trees!
18	nmohr	1.10	Precut=''
19	peller	1.2
20
21	peller	1.1
22			#Montecarlos
23	peller	1.7	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.47210.1009740.745, 0.41070.1009740.704, 0.35980.1009740.648, 0.31580.1009740.577, 0.27780.1009740.493,0.24530.1009740.403]
28	nmohr	1.10	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	peller	1.1	Aprefix0 = ['']
30	nmohr	1.10	cut0 = ['(Vtype == 1 \|\| Vtype == 0)']
31	peller	1.1
32			#split DY in flavours
33			#genZpt < 120
34			InFiles1 = ['DYJetsToLL_TuneZ2_M-50_7TeV-madgraph-tauola']*3
35			samplenames1 =['Zudsg','Zbb','Zcc']
36			sampletypes1 =['BKG']*3
37			samplesgroup1=['DYlight','DYb','DYc']
38			xsecs1 = [3048]*3
39	nmohr	1.10	SF1=[1.108,0.984,1.108]
40	peller	1.2	Aprefix1 = ['udsg_120_', 'b_120_','c_120_']
41	nmohr	1.10	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	peller	1.1
43			#genZpt > 120
44			InFiles2 = ['DYJetsToLL_PtZ-100_TuneZ2_7TeV-madgraph-tauola']*3
45			samplenames2 =['Zudsg_pt120','Zbb_pt120','Zcc_pt120']
46			sampletypes2 =['BKG']*3
47			samplesgroup2=['DYlight','DYb','DYc']
48			xsecs2 = [30]*3
49	nmohr	1.10	SF2=[1.108,0.984,1.108]
50	peller	1.2	Aprefix2 = ['udsg_120_', 'b_120_','c_120_']
51	nmohr	1.10	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	peller	1.1
53			#The Data
54			dataFiles = ['DataZee', 'DataZmm']
55			datanames =['Zee','Zmm']
56			datatypes =['DATA','DATA']
57			datagroup=['DATA','DATA']
58			AdataPrefix = ['']
59	nmohr	1.10	datacuts=['(Vtype==1)','(Vtype==0)']
60	peller	1.1
61	peller	1.2
62
63
64
65	peller	1.1	info = []
66
67			#prepare the files:
68
69			# i>1 because no WW sample!!! only temporary fix
70
71	peller	1.3	for i in range(0,len(InFiles0)):
72	peller	1.8	copytree(pathIN,pathOUT,prefix,InFiles0[i],Aprefix0[0],cut0[0]+Precut)
73	peller	1.1	info.append(sample(samplenames0[i],sampletypes0[i]))
74			info[-1].path=pathOUT
75			info[-1].identifier=Aprefix0[0]+InFiles0[i]
76			info[-1].weightexpression=weightexpression
77			info[-1].group=samplesgroup0[i]
78			info[-1].lumi=lumi
79			info[-1].prefix=prefix
80			info[-1].xsec=xsecs0[i]
81	peller	1.2	info[-1].sf=SF0[i]
82	peller	1.1	info[-1].addtreecut(cut0[0])
83
84			for i in range(3):
85	peller	1.8	copytree(pathIN,pathOUT,prefix,InFiles1[i],Aprefix1[i],cut1[i]+Precut)
86	peller	1.1	info.append(sample(samplenames1[i],sampletypes1[i]))
87			info[-1].path=pathOUT
88			info[-1].identifier=Aprefix1[i]+InFiles1[i]
89			info[-1].weightexpression=weightexpression
90			info[-1].group=samplesgroup1[i]
91			info[-1].lumi=lumi
92			info[-1].prefix=prefix
93			info[-1].xsec=xsecs1[i]
94	peller	1.2	info[-1].sf=SF1[i]
95	peller	1.1	info[-1].addtreecut(cut1[i])
96
97			for i in range(3):
98	peller	1.8	copytree(pathIN,pathOUT,prefix,InFiles2[i],Aprefix2[i],cut2[i]+Precut)
99	peller	1.1	info.append(sample(samplenames2[i],sampletypes2[i]))
100			info[-1].path=pathOUT
101			info[-1].identifier=Aprefix2[i]+InFiles2[i]
102			info[-1].weightexpression=weightexpression
103			info[-1].group=samplesgroup2[i]
104			info[-1].lumi=lumi
105			info[-1].prefix=prefix
106			info[-1].xsec=xsecs2[i]
107	peller	1.2	info[-1].sf=SF2[i]
108	peller	1.1	info[-1].addtreecut(cut2[i])
109
110			for i in range(0,len(dataFiles)):
111	peller	1.2	copytree(pathIN,pathOUT,prefix,dataFiles[i],AdataPrefix[0],datacuts[i]+Precut)
112	peller	1.1	info.append(sample(datanames[i],datatypes[i]))
113			info[-1].path=pathOUT
114			info[-1].identifier=AdataPrefix[0]+dataFiles[i]
115			info[-1].group=datagroup[i]
116			info[-1].lumi=lumi
117			info[-1].prefix=prefix
118			info[-1].addtreecut(datacuts[i])
119
120
121			#dump info
122			infofile = open(pathOUT+'/samples.info','w')
123			pickle.dump(info,infofile)
124			infofile.close()