| 1 |
import ROOT, sys
|
| 2 |
sys.path.append("/afs/naf.desy.de/user/n/nowaf/UserCode/nowaf/PythonScripts/")
|
| 3 |
import definitions as Def
|
| 4 |
import styles as Style
|
| 5 |
|
| 6 |
if __name__ == "__main__":
|
| 7 |
|
| 8 |
Def.SetGlobalStyles()
|
| 9 |
|
| 10 |
saveName = "./plots/HPSVeryLoose_Misc_EffOnly"
|
| 11 |
|
| 12 |
dir_ = "MuonAnalyzerPre"
|
| 13 |
dirEff_ = "EffRatePre"
|
| 14 |
|
| 15 |
rootfile = ROOT.TFile.Open( "./WJets_muonAcc.root" )
|
| 16 |
hist1 = rootfile.Get( dir_ + "/MuonPtPrompt" )
|
| 17 |
hist1.SetDirectory( 0 )
|
| 18 |
hist1.SetName( "MuonPtPrompt" )
|
| 19 |
hist2 = rootfile.Get( dir_ + "/GenMuonPtMatched" )
|
| 20 |
hist2.SetDirectory( 0 )
|
| 21 |
hist2.SetName( "MuonPtReco" )
|
| 22 |
hist3 = rootfile.Get( dir_ + "/MuonPtPromptNotFromTau" )
|
| 23 |
hist3.SetDirectory( 0 )
|
| 24 |
hist3.SetName( "MuonPtNoTau" )
|
| 25 |
hist4 = rootfile.Get( dir_ + "/MuonDRPromptNotFromTau" )
|
| 26 |
hist4.SetDirectory( 0 )
|
| 27 |
hist4.SetName( "MuonDRNoTau" )
|
| 28 |
hist5 = rootfile.Get( dir_ + "/GenMuonDRMatched" )
|
| 29 |
hist5.SetDirectory( 0 )
|
| 30 |
hist5.SetName( "MuonDRReco" )
|
| 31 |
hist7 = rootfile.Get( dir_ + "/GenMuonPtMatchedDR01to03" )
|
| 32 |
hist7.SetDirectory( 0 )
|
| 33 |
hist7.SetName( "GenMuonPtMatchedDR01to03" )
|
| 34 |
hist8 = rootfile.Get( dir_ + "/GenMuonPtMatchedDR03to05" )
|
| 35 |
hist8.SetDirectory( 0 )
|
| 36 |
hist8.SetName( "GenMuonPtMatchedDR03to05" )
|
| 37 |
hist9 = rootfile.Get( dir_ + "/GenMuonPtMatchedDR05toInf" )
|
| 38 |
hist9.SetDirectory( 0 )
|
| 39 |
hist9.SetName( "GenMuonPtMatchedDR05toInf" )
|
| 40 |
rootfile.Close()
|
| 41 |
|
| 42 |
rootfile = ROOT.TFile.Open( "./WJets_HPS_eff.root" )
|
| 43 |
hist6 = rootfile.Get( dirEff_ + "/TauPt" )
|
| 44 |
hist6.SetDirectory( 0 )
|
| 45 |
hist6.SetName( "TauEff" )
|
| 46 |
hist6U = rootfile.Get( dirEff_ + "/TauPtErrUp" )
|
| 47 |
hist6U.SetDirectory( 0 )
|
| 48 |
hist6U.SetName( "TauEffErrUp" )
|
| 49 |
hist6D = rootfile.Get( dirEff_ + "/TauPtErrDwn" )
|
| 50 |
hist6D.SetDirectory( 0 )
|
| 51 |
hist6D.SetName( "TauEffErrDwn" )
|
| 52 |
rootfile.Close()
|
| 53 |
|
| 54 |
|
| 55 |
|
| 56 |
histList = []
|
| 57 |
#histList.append( hist1 )
|
| 58 |
#histList.append( hist2 )
|
| 59 |
#histList.append( hist3 )
|
| 60 |
#histList.append( hist4 )
|
| 61 |
#histList.append( hist5 )
|
| 62 |
##histList.append( hist6 )
|
| 63 |
##histList.append( hist6U )
|
| 64 |
##histList.append( hist6D )
|
| 65 |
#histList.append( hist7 )
|
| 66 |
#histList.append( hist8 )
|
| 67 |
#histList.append( hist9 )
|
| 68 |
|
| 69 |
errHistList = []
|
| 70 |
errHistList.append( hist6 )
|
| 71 |
errHistList.append( hist6U )
|
| 72 |
errHistList.append( hist6D )
|
| 73 |
|
| 74 |
fakeList = []
|
| 75 |
fakeDir = "FakeRateDiJetsLoose3Jet/"
|
| 76 |
#fakeList.append( "JetTauPtJPT" )
|
| 77 |
#fakeList.append( "JetTauEtaJPT" )
|
| 78 |
#fakeList.append( "JetTauEtaJPTPt20" )
|
| 79 |
#fakeList.append( "JetTauEtaJPTPt40" )
|
| 80 |
#fakeList.append( "JetTauEtaJPTPt60" )
|
| 81 |
#fakeList.append( "JetTauEtaJPTPt80" )
|
| 82 |
#fakeList.append( "JetTauEtaJPTPt100" )
|
| 83 |
#fakeList.append( "JetTauEtaJPTPt200" )
|
| 84 |
#fakeList.append( "JetTauEtaJPTPt500" )
|
| 85 |
#fakeList.append( "JetTauEtaJPTPtInf" )
|
| 86 |
rootfile = ROOT.TFile.Open( "./Data_fakes.root" )
|
| 87 |
for fh in fakeList:
|
| 88 |
h = rootfile.Get( fakeDir + fh )
|
| 89 |
h.SetDirectory( 0 )
|
| 90 |
hU = rootfile.Get( fakeDir + fh + "ErrUp" )
|
| 91 |
hU.SetDirectory( 0 )
|
| 92 |
hD = rootfile.Get( fakeDir + fh + "ErrDwn" )
|
| 93 |
hD.SetDirectory( 0 )
|
| 94 |
errHistList.append( h )
|
| 95 |
errHistList.append( hU )
|
| 96 |
errHistList.append( hD )
|
| 97 |
|
| 98 |
rootfile.Close()
|
| 99 |
|
| 100 |
|
| 101 |
cDict = {}
|
| 102 |
for hist in histList:
|
| 103 |
#if not hist.GetName() in [ "TauEff", "TauEffErrUp", "TauEffErrDwn" ]:
|
| 104 |
cDict[ hist.GetName() ] = ROOT.TCanvas( hist.GetName(), hist.GetName() )
|
| 105 |
cDict[ hist.GetName() ].cd()
|
| 106 |
hist.Draw()
|
| 107 |
pass
|
| 108 |
#for hist in errHistList :
|
| 109 |
for i in range( 0, len( errHistList ) ):
|
| 110 |
print i
|
| 111 |
if not i%3:
|
| 112 |
#print "selected: ", i
|
| 113 |
#cDict[ errHistList[ i + 1 ].GetName() ] = ROOT.TCanvas( errHistList[ i + 1 ].GetName(),
|
| 114 |
# errHistList[ i + 1 ].GetName() )
|
| 115 |
#cDict[ errHistList[ i + 1 ].GetName() ].cd()
|
| 116 |
#errHistList[ i + 1 ].Draw()
|
| 117 |
#errHistList[ i ].Draw( "same" )
|
| 118 |
#errHistList[ i + 2 ].Draw( "same" )
|
| 119 |
cDict[ errHistList[ i ].GetName() ] = ROOT.TCanvas( errHistList[ i ].GetName(),
|
| 120 |
errHistList[ i ].GetName() )
|
| 121 |
errHistList[ i ].GetXaxis().SetRangeUser( 0,155 )
|
| 122 |
|
| 123 |
errHistList[ i ].Draw()
|
| 124 |
pass
|
| 125 |
pass
|
| 126 |
|
| 127 |
|
| 128 |
|
| 129 |
Style.saveToEPS( saveName, cDict )
|
| 130 |
|
| 131 |
Def.DontQuit()
|
