| 1 |
import ROOT
|
| 2 |
import sys, re, math
|
| 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 |
from copy import deepcopy
|
| 8 |
|
| 9 |
if __name__ == "__main__":
|
| 10 |
|
| 11 |
#taufile = ROOT.TFile.Open( "/afs/naf.desy.de/group/cms/scratch/nowaf/CMSSW_4_2_4/src/anothertest.root" );
|
| 12 |
#muonfile = ROOT.TFile.Open( "/afs/naf.desy.de/group/cms/scratch/nowaf/CMSSW_4_2_4/src/anothertest2.root" );
|
| 13 |
#TauDR = taufile->Get( "TauSelectionTaudRJet" )
|
| 14 |
#MuonDR = ( TH1F* ) muonfile->Get( "MuonDRJet" )
|
| 15 |
|
| 16 |
Def.SetGlobalStyles()
|
| 17 |
|
| 18 |
|
| 19 |
filename = "/afs/naf.desy.de/group/cms/scratch/nowaf/CMSSW_4_2_4/src/RA2/Configuration/FinalPlots/TauFinalPlot.root"
|
| 20 |
filename = "/scratch/hh/lustre/cms/user/nowaf/2011Data/VersionII/Plots/RealTauEst/TTbar_cxxxix.root"
|
| 21 |
rootfile = ROOT.TFile.Open( filename )
|
| 22 |
TauDR = rootfile.Get( "finalPlotTauPreSelectionMatchedTauControl/TauSelectionTaudRJet" )
|
| 23 |
MuonDR = rootfile.Get( "MuonTemplateAnalyzerPre/MuonDRJet" )
|
| 24 |
MuonDR.Scale( 0.65 )
|
| 25 |
|
| 26 |
|
| 27 |
TauDR.SetDirectory( 0 )
|
| 28 |
MuonDR.SetDirectory( 0 )
|
| 29 |
|
| 30 |
|
| 31 |
weightHist = TauDR.Clone()
|
| 32 |
weightHist.SetDirectory( 0 )
|
| 33 |
weightHist.Divide( MuonDR.Clone() )
|
| 34 |
|
| 35 |
TwoDPlot = rootfile.Get( "MuonTemplateAnalyzerPre/MuonDRJetVsDPhiMHTTau" )
|
| 36 |
TwoDPlot.SetDirectory( 0 )
|
| 37 |
|
| 38 |
TauDPhi = rootfile.Get( "finalPlotTauPreSelectionMatchedTauControl/TauSelectiondPhiMHTTau" )
|
| 39 |
TauDPhi.SetDirectory( 0 )
|
| 40 |
|
| 41 |
TauDPhiWeighted = ROOT.TH1F( "MDRWeighted", "MDRWeighted", TauDPhi.GetNbinsX(), 0, TauDPhi.GetXaxis().GetXmax() )
|
| 42 |
TauDPhiWeighted.SetDirectory( 0 )
|
| 43 |
|
| 44 |
values = [ 0 ] * ( TwoDPlot.GetNbinsX() + 1 )
|
| 45 |
for xbin in range( 0, TwoDPlot.GetNbinsX() + 1 ):
|
| 46 |
xbincenter = TwoDPlot.GetXaxis().GetBinCenter( xbin )
|
| 47 |
#print "xcenter=", xbincenter
|
| 48 |
for ybin in range( 0, TwoDPlot.GetNbinsY() + 1 ):
|
| 49 |
ybincenter = TwoDPlot.GetYaxis().GetBinCenter( ybin )
|
| 50 |
#print "\tycenter=", ybincenter
|
| 51 |
xbinvalue = TwoDPlot.GetBinContent( xbin, ybin )
|
| 52 |
#print "\tval=", xbinvalue
|
| 53 |
weightbin = weightHist.FindBin( ybincenter )
|
| 54 |
weight = weightHist.GetBinContent( weightbin )
|
| 55 |
values[ xbin ] = values[ xbin ] + xbinvalue * weight
|
| 56 |
#values[ xbin ] = values[ xbin ] + xbinvalue
|
| 57 |
#print "xbinvalue=", xbinvalue, " weight=", weight
|
| 58 |
pass
|
| 59 |
pass
|
| 60 |
for i in range( 0, len( values ) ):
|
| 61 |
TauDPhiWeighted.SetBinContent( i, values[ i ] )
|
| 62 |
#print "bin=", i, " val=", values[ i ]
|
| 63 |
pass
|
| 64 |
TauDPhiWeighted.Scale( 0.65 )
|
| 65 |
|
| 66 |
TauDPhi.Rebin( 2 )
|
| 67 |
TauDPhiWeighted.Rebin( 2 )
|
| 68 |
|
| 69 |
#TauDR.Draw()
|
| 70 |
#TauDR.SetMarkerStyle( 22 )
|
| 71 |
#MuonDR.Draw()
|
| 72 |
TauDPhi.Draw()
|
| 73 |
TauDPhi.SetMarkerStyle( 22 )
|
| 74 |
TauDPhiWeighted.SetLineColor( 8 )
|
| 75 |
TauDPhiWeighted.Draw( "same" )
|
| 76 |
ROOT.gPad.SetLogy( 0 )
|
| 77 |
ROOT.gPad.Update()
|
| 78 |
#weightHist.Draw()
|
| 79 |
|
| 80 |
Def.DontQuit()
|
| 81 |
|
