|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#!/usr/bin/env python
#
# Launch the script with the command: ./compareDQM.py
# Set below the two DQM input files (DQMfileOld,DQMfileNew)
#
# This script compares the plots cointained in two DQM files and save the superimposed plots
#
DQMfileOld="/afs/cern.ch/user/s/sdonato/AFSwork/public/DQM_V0001_R000000001__CMSSW_X_Y_Z__RelVal__TrigVal.root"
DQMfileNew="/afs/cern.ch/user/s/sdonato/AFSwork/public/DQM_V0001_R000000002__CMSSW_X_Y_Z__RelVal__TrigVal.root"
labelNew = "New"
labelOld = "Old"
########################## load libraries #################################
import os, string, re, sys, math
try:
import ROOT
except:
print("\nCannot load PYROOT, make sure you have setup ROOT in the path")
print("and pyroot library is also defined in the variable PYTHONPATH, try:\n")
if (os.getenv("PYTHONPATH")):
print(" setenv PYTHONPATH ${PYTHONPATH}:$ROOTSYS/lib\n")
else:
print(" setenv PYTHONPATH $ROOTSYS/lib\n")
sys.exit()
folder="plots"
try:
os.mkdir(folder)
except:
print("folder " + folder + " already exist")
from ROOT import TFile
from ROOT import TCanvas
from ROOT import TLegend
from ROOT import TH1F
from ROOT import TGraphErrors
########################## define a function that return plots given a TFile #################################
def GetPlots(_file0):
# _file0=TFile(filename)
dir1 = _file0.Get("DQMData")
dir2 = dir1.Get("Run 1")
dir3 = dir2.Get("HLT")
dir4 = dir3.Get("Run summary")
plots=[]
for type in dir4.GetListOfKeys():
dirType= dir4.Get(type.GetName())
for triggerKey in dirType.GetListOfKeys():
triggerDir=dirType.Get(triggerKey.GetName())
for plotKey in triggerDir.GetListOfKeys():
plotPointer=triggerDir.Get(plotKey.GetName())
plot=plotPointer
if(plot.GetName()=="efficiency"):
for plotEfficiencyKey in plotPointer.GetListOfKeys():
plot=plotPointer.Get(plotEfficiencyKey.GetName())
plots=plots+[plot.Clone(triggerKey.GetName() + "_" + plot.GetName())]
else:
plots=plots+[plot.Clone(triggerKey.GetName() + "_" + plot.GetName())]
return plots
########################## read DQM plots #################################
fileNew=TFile(DQMfileOld)
plotsNew=0
plotsOld=0
try:
plotsNew = GetPlots(fileNew)
except:
print("Problem with ", fileNew)
fileOld=TFile(DQMfileNew)
try:
plotsOld = GetPlots(fileOld)
except:
print("Problem with ", fileOld)
##### for kind of plots save a .png superimposing the New with the Old #####
ROOT.gROOT.SetBatch()
ROOT.gStyle.SetOptStat(0)
c1 = TCanvas("c1","",1280,720)
c1.SetGridx()
c1.SetGridy()
legend = TLegend(0.07,0.85,0.2,0.93);
first=True
for plotNew in plotsNew:
for plotOld in plotsOld:
if(plotNew.GetName()==plotOld.GetName()):
plotOld.SetLineColor(4)
plotOld.SetMarkerColor(4)
# plotOld.SetFillColor(4)
plotNew.SetLineColor(2)
plotNew.SetMarkerColor(2)
# plotNew.SetFillColor(2)
# plotNew.SetFillStyle(3002)
plotNew.SetLineWidth(2)
plotOld.SetLineWidth(2)
if first:
legend.AddEntry(plotNew,labelNew,"l");
legend.AddEntry(plotOld,labelOld,"l");
if plotNew.GetName().rfind("mistagrate)")>0:
plotOld.SetMinimum(0.001)
plotNew.SetMinimum(0.001)
c1.SetLogy(1)
else:
c1.SetLogy(0)
plotOld.SetMaximum(1.05*max(plotOld.GetMaximum(),plotNew.GetMaximum(),1))
plotOld.Draw()
plotNew.Draw("same")
legend.Draw()
c1.SaveAs(folder+"/"+plotNew.GetName()+".png")
first=False
|