nowaf/RootFilesInUse/GetEventYield.py

import ROOT
import sys, re
sys.path.append("/afs/naf.desy.de/user/n/nowaf/UserCode/nowaf/PythonScripts/")
import definitions as Def
import styles as Style
import MakeRealTauEst_cff as RT
import math

if __name__ == "__main__":

    Def.SetGlobalStyles()

    #lumi = 180.
    #lumi = 4683.9
    lumi = 4992.1
    ### v4
    #lumi = 927.9


    makeTexFile = True

    cut = None
    #cut = 650.

    removeSamples = []
    #removeSamples.append( "TTbar" )
    removeSamples.append( "GVJets" )
    #removeSamples.append( "WWJets" )
    #removeSamples.append( "Zinv" ) 
    #removeSamples.append( "ZJets" )
    #removeSamples.append( "WJets" )
    #removeSamples.append( "QCDFlat" )
    #removeSamples.append( "Data" )

    addSample = []
    #addSample.append( "WWJets" )
    addSample.append( "LM13" )
    addSample.append( "LM2" )
    #addSample.append( "LM8" )

    allSamples = []
    allSamples = []
    allSamples.append( "TTbar" )
    #allSamples.append( "GVJets" )
    allSamples.append( "WWJets" )
    #allSamples.append( "Zinv" ) 
    allSamples.append( "ZJets" )
    allSamples.append( "WJets" )
    #allSamples.append( "QCDFlat" )
    #allSamples.append( "Data" )
    #allSamples.append( "LM13" )
    #allSamples.append( "LM2" )

    nameMap = {}
    nameMap[ "LM13" ] = "LMThirteen"
    nameMap[ "LM2" ] = "LMTwo"
    

    fileDir = "/scratch/hh/lustre/cms/user/nowaf/2011Data/VersionII/Plots/RealTauEst/"
    bDir = "/scratch/hh/lustre/cms/user/nowaf/2011Data/VersionI/mSugraScanTan50_TaNC_cuts/"

    #fileName = "_fakes_xxviii"
    fileName = "_clxix"

    ##add = "_xxxv"
    ###add = "_tightID"


    #dir = "finalPlotTauPreSelection/"
    #dir = "finalPlotTau/"

    dir = "finalPlotTauPreSelectionControl/"
    #dir = "finalPlotTauControl/"

    #dir = "finalPlotTauPreSelectionControlLepFakes/"

    #dir = "finalPlotTauPreSelectionControl20MHT40/"
    #dir = "finalPlotTauControl20MHT40/"

    #dir = "finalPlotTauPreSelectionControl40MHT60/"
    #dir = "finalPlotTauControl40MHT60/"

    #dir = "finalPlotTauPreSelectionControl60MHT100/"
    #dir = "finalPlotTauControl60MHT100/"

    #dir = "finalPlotTauPreSelectionControl60MHT80/"
    #dir = "finalPlotTauPreSelectionControl60MHT80HT500/"
    #dir = "finalPlotTauControl60MHT80/"

    #dir = "finalPlotTauPreSelectionControl80MHT100/"
    #dir = "finalPlotTauPreSelectionControl80MHT100HT500/"
    #dir = "finalPlotTauControl80MHT100/"

    #dir = "finalPlotTauPreSelectionControl100MHT120/"
    #dir = "finalPlotTauPreSelectionControl100MHT120HT500/"
    #dir = "finalPlotTauControl100MHT120/"

    #dir = "finalPlotTauPreSelectionControl120MHT140/"
    #dir = "finalPlotTauPreSelectionControl120MHT140HT500/"
    #dir = "finalPlotTauControl120MHT140/"

    #dir = "finalPlotTauPreSelectionMatchedTauControl/"
    #dir = "finalPlotTauMatchedTauControl/"

    #dir = "finalPlotTauInvertMHTControl/"
    #dir = "finalPlotTauGen/"

    #dir = "finalPlotTauPreSelectionGenVetoTau/"
    #dir = "finalPlotGenVetoTau/"

    Type = "Full"
    roundTo = 1
    if re.search( "Pre", dir ):
        Type = "BaseLine"
        roundTo = 0
        pass

    histList = []
    #histList.append( "MHTMHT" )
    histList.append( "HTHT" )
    

    statusDict = {}
    statusDict[ "QCDFlat" ] = "BG"
    statusDict[ "TTbar" ] = "BG"
    statusDict[ "GVJets" ] = "BG"
    statusDict[ "ZJets" ] = "BG"
    statusDict[ "Zinv" ] = "BG"
    statusDict[ "VVJets" ] = "BG"
    statusDict[ "WWJets" ] = "BG"
    statusDict[ "WJets" ] = "BG"
    statusDict[ "Data" ] = "Data"
    statusDict[ "LM13" ] = "Signal"
    statusDict[ "LM7" ] = "Signal"
    statusDict[ "LM8" ] = "Signal"
    statusDict[ "GMSBc" ] = "Signal"
    statusDict[ "LM3" ] = "Signal"
    statusDict[ "LM9" ] = "Signal"
    statusDict[ "LM12" ] = "Signal"
    statusDict[ "330_340" ] = "Signal"
    statusDict[ "330_390" ] = "Signal"
    statusDict[ "LM2" ] = "Signal"


    #helper = RT.ReadIn( fileDir, "TTbar" )
    #scalingFactors = helper.scalingFactors


    #histDict = {}
    #for hist in histList:
    #    print "----- ", hist
    #    histDict[ hist ] = {}
    #    for file in sampleList:
    #        rootfile = ROOT.TFile.Open( fileDict[ file ] )
    #        #print file
    #        histDict[ hist ][ file ] = rootfile.Get( dir + hist )
    #        histDict[ hist ][ file ].SetDirectory( 0 )
    #        print file, " scalingFactor=", scalingFactors[ file ]
    #        if file in scalingFactors.keys():
    #            histDict[ hist ][ file ].Scale( scalingFactors[ file ] )
    #            pass
    #        rootfile.Close()
    #        pass
    #    pass

    
    readIn = RT.ReadIn( fileDir, fileName, lumi=lumi )
    readIn.removeSamples( removeSamples )
    readIn.addSamples( addSample )
    readIn.removeHists( [ "MHT" ] )
    readIn.addHists( histList )
    #readIn.dontScale()
    histDict = readIn.getHists( dir )

    readInUnscaled = RT.ReadIn( fileDir, fileName )
    readInUnscaled.removeSamples( removeSamples )
    readInUnscaled.addSamples( addSample )
    readInUnscaled.removeHists( [ "MHT" ] )
    readInUnscaled.addHists( histList )
    readInUnscaled.dontScale()
    histDictUnscaled = readInUnscaled.getHists( dir )

    ################## tex file ############################################################

    if makeTexFile:
        TF = Def.MakeTexFile( fileName="tex/newCommandsSelection" + Type )
        TF.addCommand( name="Sel" + Type + "Lumi",
                       value=lumi/1000.,
                       roundTo=1 )
        pass

    bg = 0.
    bgerr = 0.
    lowestBin = 0
    if not cut == None:
        lowestBin = histDict[ histList[ 0 ] ][ "QCDFlat" ].FindBin( cut )
        pass
    print "#################EventYields###############"
    print "           Lumi=", lumi
    print "========", dir, "========="
    for sample in readIn.sampleList:
        if statusDict[ sample ] == "BG":
            nbins = histDict[ histList[ 0 ] ][ sample ].GetNbinsX() + 1
            err = ROOT.Double( 0 )
            val = histDict[ histList[ 0 ] ][ sample ].IntegralAndError( lowestBin, nbins, err )
            #bgerr += ( math.sqrt( histDictUnscaled[ histList[ 0 ] ][ sample ].Integral( lowestBin, nbins ) ) *\
            #      readIn.scalingFactors[ sample ] )**2
            bgerr += err**2
            bg += val
            print sample, ": ", round( val, 2 ), " +- ",round( err, 2 )

            if makeTexFile:
                TF.addCommand( name="Sel" + Type + sample,
                               value=val,
                               roundTo=roundTo )
                TF.addCommand( name="Sel" + Type + sample + "StatUncert",
                               value=err,
                               roundTo=roundTo )
                pass

            pass
        pass

    for s in allSamples:
        if makeTexFile and not s in readIn.sampleList:
            TF.addCommand( name="Sel" + Type + s,
                           value="-",
                           roundTo=None )
            TF.addCommand( name="Sel" + Type + s + "StatUncert",
                           value="-",
                           roundTo=None )
            pass
        pass
            

    print "-------- BG ---------"
    print "BG: ", round( bg, 2 ), " +- ", round( math.sqrt( bgerr ),2 )

    if makeTexFile:
        TF.addCommand( name="Sel" + Type + "All",
                       value=bg,
                       roundTo=roundTo )
        TF.addCommand( name="Sel" + Type + "All" + "StatUncert",
                       value=math.sqrt( bgerr ),
                       roundTo=roundTo )
        pass

    
    print "--------Data---------"
    if "Data" in readIn.sampleList:
        nbinsdata = histDict[ histList[ 0 ] ][ "Data" ].GetNbinsX() + 1
        print "Data: ", round( histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ), 2 ), \
              " +- ", round( math.sqrt( histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ) ), 2 )

        if makeTexFile:
            TF.addCommand( name="Sel" + Type + "Data",
                           value=histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ),
                           roundTo=0 )
            TF.addCommand( name="Sel" + Type + "Data" + "StatUncert",
                           value=math.sqrt( histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ) ),
                           roundTo=roundTo )
            pass
        
    print "-------Signal--------"
    for sample in readIn.sampleList:
        if statusDict[ sample ] == "Signal":
            nbinssignal = histDict[ histList[ 0 ] ][ sample ].GetNbinsX() + 1
            nserr = ROOT.Double( 0 )
            ns = histDict[ histList[ 0 ] ][ sample ].IntegralAndError( lowestBin, nbinssignal, nserr )
            print sample, ": ", round( ns, 2 ), " +- ", round( nserr, 2 )

            if makeTexFile:
                TF.addCommand( name="Sel" + Type + nameMap[ sample ],
                               value=ns,
                               roundTo=roundTo )
                TF.addCommand( name="Sel" + Type + nameMap[ sample ] + "StatUncert",
                               value=nserr,
                               roundTo=roundTo )
                pass
            
            pass
        pass
    print "###########################################"
    #print
    print "------ Number of used MC evens ----"
    #print
    mcall = 0
    for sample in readInUnscaled.sampleList:
        nbins = histDict[ histList[ 0 ] ][ sample ].GetNbinsX() + 1
        mc = histDictUnscaled[ histList[ 0 ] ][ sample ].GetEntries()
        mcall += mc
        print sample, ": ", mc
        s = sample
        if sample in nameMap.keys():
          s =  nameMap[ sample ]
          pass
        TF.addCommand( name="Sel" + Type + s + "MC",
                       value=mc,
                       roundTo=0 )
        pass
    TF.addCommand( name="Sel" + Type + "All" + "MC",
                   value=mcall,
                   roundTo=0 )

    if makeTexFile:
        TF.writeTexFile()
        pass

    
    pass
    
Revision:	1.1.1.1 (vendor branch)
Committed:	Tue Mar 20 13:12:08 2012 UTC (13 years, 1 month ago) by nowak
Content type:	text/x-python
Branch:	rootFilesInUse, MAIN
CVS Tags:	start, HEAD
Changes since 1.1:	+0 -0 lines
Log Message:	t files in use
#	Content
1	import ROOT
2	import sys, re
3	sys.path.append("/afs/naf.desy.de/user/n/nowaf/UserCode/nowaf/PythonScripts/")
4	import definitions as Def
5	import styles as Style
6	import MakeRealTauEst_cff as RT
7	import math
8
9	if __name__ == "__main__":
10
11	Def.SetGlobalStyles()
12
13	#lumi = 180.
14	#lumi = 4683.9
15	lumi = 4992.1
16	### v4
17	#lumi = 927.9
18
19
20	makeTexFile = True
21
22	cut = None
23	#cut = 650.
24
25	removeSamples = []
26	#removeSamples.append( "TTbar" )
27	removeSamples.append( "GVJets" )
28	#removeSamples.append( "WWJets" )
29	#removeSamples.append( "Zinv" )
30	#removeSamples.append( "ZJets" )
31	#removeSamples.append( "WJets" )
32	#removeSamples.append( "QCDFlat" )
33	#removeSamples.append( "Data" )
34
35	addSample = []
36	#addSample.append( "WWJets" )
37	addSample.append( "LM13" )
38	addSample.append( "LM2" )
39	#addSample.append( "LM8" )
40
41	allSamples = []
42	allSamples = []
43	allSamples.append( "TTbar" )
44	#allSamples.append( "GVJets" )
45	allSamples.append( "WWJets" )
46	#allSamples.append( "Zinv" )
47	allSamples.append( "ZJets" )
48	allSamples.append( "WJets" )
49	#allSamples.append( "QCDFlat" )
50	#allSamples.append( "Data" )
51	#allSamples.append( "LM13" )
52	#allSamples.append( "LM2" )
53
54	nameMap = {}
55	nameMap[ "LM13" ] = "LMThirteen"
56	nameMap[ "LM2" ] = "LMTwo"
57
58
59	fileDir = "/scratch/hh/lustre/cms/user/nowaf/2011Data/VersionII/Plots/RealTauEst/"
60	bDir = "/scratch/hh/lustre/cms/user/nowaf/2011Data/VersionI/mSugraScanTan50_TaNC_cuts/"
61
62	#fileName = "_fakes_xxviii"
63	fileName = "_clxix"
64
65	##add = "_xxxv"
66	###add = "_tightID"
67
68
69	#dir = "finalPlotTauPreSelection/"
70	#dir = "finalPlotTau/"
71
72	dir = "finalPlotTauPreSelectionControl/"
73	#dir = "finalPlotTauControl/"
74
75	#dir = "finalPlotTauPreSelectionControlLepFakes/"
76
77	#dir = "finalPlotTauPreSelectionControl20MHT40/"
78	#dir = "finalPlotTauControl20MHT40/"
79
80	#dir = "finalPlotTauPreSelectionControl40MHT60/"
81	#dir = "finalPlotTauControl40MHT60/"
82
83	#dir = "finalPlotTauPreSelectionControl60MHT100/"
84	#dir = "finalPlotTauControl60MHT100/"
85
86	#dir = "finalPlotTauPreSelectionControl60MHT80/"
87	#dir = "finalPlotTauPreSelectionControl60MHT80HT500/"
88	#dir = "finalPlotTauControl60MHT80/"
89
90	#dir = "finalPlotTauPreSelectionControl80MHT100/"
91	#dir = "finalPlotTauPreSelectionControl80MHT100HT500/"
92	#dir = "finalPlotTauControl80MHT100/"
93
94	#dir = "finalPlotTauPreSelectionControl100MHT120/"
95	#dir = "finalPlotTauPreSelectionControl100MHT120HT500/"
96	#dir = "finalPlotTauControl100MHT120/"
97
98	#dir = "finalPlotTauPreSelectionControl120MHT140/"
99	#dir = "finalPlotTauPreSelectionControl120MHT140HT500/"
100	#dir = "finalPlotTauControl120MHT140/"
101
102	#dir = "finalPlotTauPreSelectionMatchedTauControl/"
103	#dir = "finalPlotTauMatchedTauControl/"
104
105	#dir = "finalPlotTauInvertMHTControl/"
106	#dir = "finalPlotTauGen/"
107
108	#dir = "finalPlotTauPreSelectionGenVetoTau/"
109	#dir = "finalPlotGenVetoTau/"
110
111	Type = "Full"
112	roundTo = 1
113	if re.search( "Pre", dir ):
114	Type = "BaseLine"
115	roundTo = 0
116	pass
117
118	histList = []
119	#histList.append( "MHTMHT" )
120	histList.append( "HTHT" )
121
122
123	statusDict = {}
124	statusDict[ "QCDFlat" ] = "BG"
125	statusDict[ "TTbar" ] = "BG"
126	statusDict[ "GVJets" ] = "BG"
127	statusDict[ "ZJets" ] = "BG"
128	statusDict[ "Zinv" ] = "BG"
129	statusDict[ "VVJets" ] = "BG"
130	statusDict[ "WWJets" ] = "BG"
131	statusDict[ "WJets" ] = "BG"
132	statusDict[ "Data" ] = "Data"
133	statusDict[ "LM13" ] = "Signal"
134	statusDict[ "LM7" ] = "Signal"
135	statusDict[ "LM8" ] = "Signal"
136	statusDict[ "GMSBc" ] = "Signal"
137	statusDict[ "LM3" ] = "Signal"
138	statusDict[ "LM9" ] = "Signal"
139	statusDict[ "LM12" ] = "Signal"
140	statusDict[ "330_340" ] = "Signal"
141	statusDict[ "330_390" ] = "Signal"
142	statusDict[ "LM2" ] = "Signal"
143
144
145	#helper = RT.ReadIn( fileDir, "TTbar" )
146	#scalingFactors = helper.scalingFactors
147
148
149	#histDict = {}
150	#for hist in histList:
151	# print "----- ", hist
152	# histDict[ hist ] = {}
153	# for file in sampleList:
154	# rootfile = ROOT.TFile.Open( fileDict[ file ] )
155	# #print file
156	# histDict[ hist ][ file ] = rootfile.Get( dir + hist )
157	# histDict[ hist ][ file ].SetDirectory( 0 )
158	# print file, " scalingFactor=", scalingFactors[ file ]
159	# if file in scalingFactors.keys():
160	# histDict[ hist ][ file ].Scale( scalingFactors[ file ] )
161	# pass
162	# rootfile.Close()
163	# pass
164	# pass
165
166
167	readIn = RT.ReadIn( fileDir, fileName, lumi=lumi )
168	readIn.removeSamples( removeSamples )
169	readIn.addSamples( addSample )
170	readIn.removeHists( [ "MHT" ] )
171	readIn.addHists( histList )
172	#readIn.dontScale()
173	histDict = readIn.getHists( dir )
174
175	readInUnscaled = RT.ReadIn( fileDir, fileName )
176	readInUnscaled.removeSamples( removeSamples )
177	readInUnscaled.addSamples( addSample )
178	readInUnscaled.removeHists( [ "MHT" ] )
179	readInUnscaled.addHists( histList )
180	readInUnscaled.dontScale()
181	histDictUnscaled = readInUnscaled.getHists( dir )
182
183	################## tex file ############################################################
184
185	if makeTexFile:
186	TF = Def.MakeTexFile( fileName="tex/newCommandsSelection" + Type )
187	TF.addCommand( name="Sel" + Type + "Lumi",
188	value=lumi/1000.,
189	roundTo=1 )
190	pass
191
192	bg = 0.
193	bgerr = 0.
194	lowestBin = 0
195	if not cut == None:
196	lowestBin = histDict[ histList[ 0 ] ][ "QCDFlat" ].FindBin( cut )
197	pass
198	print "#################EventYields###############"
199	print " Lumi=", lumi
200	print "========", dir, "========="
201	for sample in readIn.sampleList:
202	if statusDict[ sample ] == "BG":
203	nbins = histDict[ histList[ 0 ] ][ sample ].GetNbinsX() + 1
204	err = ROOT.Double( 0 )
205	val = histDict[ histList[ 0 ] ][ sample ].IntegralAndError( lowestBin, nbins, err )
206	#bgerr += ( math.sqrt( histDictUnscaled[ histList[ 0 ] ][ sample ].Integral( lowestBin, nbins ) ) *\
207	# readIn.scalingFactors[ sample ] )**2
208	bgerr += err**2
209	bg += val
210	print sample, ": ", round( val, 2 ), " +- ",round( err, 2 )
211
212	if makeTexFile:
213	TF.addCommand( name="Sel" + Type + sample,
214	value=val,
215	roundTo=roundTo )
216	TF.addCommand( name="Sel" + Type + sample + "StatUncert",
217	value=err,
218	roundTo=roundTo )
219	pass
220
221	pass
222	pass
223
224	for s in allSamples:
225	if makeTexFile and not s in readIn.sampleList:
226	TF.addCommand( name="Sel" + Type + s,
227	value="-",
228	roundTo=None )
229	TF.addCommand( name="Sel" + Type + s + "StatUncert",
230	value="-",
231	roundTo=None )
232	pass
233	pass
234
235
236	print "-------- BG ---------"
237	print "BG: ", round( bg, 2 ), " +- ", round( math.sqrt( bgerr ),2 )
238
239	if makeTexFile:
240	TF.addCommand( name="Sel" + Type + "All",
241	value=bg,
242	roundTo=roundTo )
243	TF.addCommand( name="Sel" + Type + "All" + "StatUncert",
244	value=math.sqrt( bgerr ),
245	roundTo=roundTo )
246	pass
247
248
249	print "--------Data---------"
250	if "Data" in readIn.sampleList:
251	nbinsdata = histDict[ histList[ 0 ] ][ "Data" ].GetNbinsX() + 1
252	print "Data: ", round( histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ), 2 ), \
253	" +- ", round( math.sqrt( histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ) ), 2 )
254
255	if makeTexFile:
256	TF.addCommand( name="Sel" + Type + "Data",
257	value=histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ),
258	roundTo=0 )
259	TF.addCommand( name="Sel" + Type + "Data" + "StatUncert",
260	value=math.sqrt( histDict[ histList[ 0 ] ][ "Data" ].Integral( lowestBin, nbinsdata ) ),
261	roundTo=roundTo )
262	pass
263
264	print "-------Signal--------"
265	for sample in readIn.sampleList:
266	if statusDict[ sample ] == "Signal":
267	nbinssignal = histDict[ histList[ 0 ] ][ sample ].GetNbinsX() + 1
268	nserr = ROOT.Double( 0 )
269	ns = histDict[ histList[ 0 ] ][ sample ].IntegralAndError( lowestBin, nbinssignal, nserr )
270	print sample, ": ", round( ns, 2 ), " +- ", round( nserr, 2 )
271
272	if makeTexFile:
273	TF.addCommand( name="Sel" + Type + nameMap[ sample ],
274	value=ns,
275	roundTo=roundTo )
276	TF.addCommand( name="Sel" + Type + nameMap[ sample ] + "StatUncert",
277	value=nserr,
278	roundTo=roundTo )
279	pass
280
281	pass
282	pass
283	print "###########################################"
284	#print
285	print "------ Number of used MC evens ----"
286	#print
287	mcall = 0
288	for sample in readInUnscaled.sampleList:
289	nbins = histDict[ histList[ 0 ] ][ sample ].GetNbinsX() + 1
290	mc = histDictUnscaled[ histList[ 0 ] ][ sample ].GetEntries()
291	mcall += mc
292	print sample, ": ", mc
293	s = sample
294	if sample in nameMap.keys():
295	s = nameMap[ sample ]
296	pass
297	TF.addCommand( name="Sel" + Type + s + "MC",
298	value=mc,
299	roundTo=0 )
300	pass
301	TF.addCommand( name="Sel" + Type + "All" + "MC",
302	value=mcall,
303	roundTo=0 )
304
305	if makeTexFile:
306	TF.writeTexFile()
307	pass
308
309
310	pass
311