ATLAS Offline Software
Loading...
Searching...
No Matches
AtlRunQueryRoot.py
Go to the documentation of this file.
1#!/bin/env python
2
3# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
4#
5# ----------------------------------------------------------------
6# Script : AtlRunQueryRoot.py
7# Project: AtlRunQuery
8# Purpose: ROOT interactions
9# Authors: Andreas Hoecker (CERN), Joerg Stelzer (DESY)
10# Created: Dec 13, 2008
11# ----------------------------------------------------------------
12#
13# ---------------------------------------------------------------------------------------------------
14# ROOT TTree making and plotting
15# ---------------------------------------------------------------------------------------------------
16
17import datetime, sys, os
18from array import array
19from CoolRunQuery.utils.AtlRunQueryUtils import importroot
20importroot() # needed before any root import to switch to batch mode
21from ROOT import TFile, TTree, TCanvas, gPad, TGraph, TColor, TStyle, TGaxis, gROOT, gStyle, TText, TH1F, TLegend, TF1
22
23# ---------------------------------------------------------------------------------------------------
24# Creation of ROOT output file
25# ---------------------------------------------------------------------------------------------------
26
27# general style
28def SetStyle( whiteCanvas = False ):
29
30 # colors
31 c_Canvas = TColor.GetColor( "#f0f0f0" ) # TColor.GetColor( "#e9e6da" )
32 if whiteCanvas:
33 c_Canvas = TColor.GetColor( 10 )
34 c_FrameFill = 10
35 c_TitleBox = TColor.GetColor( "#dae1e6" )
36
37 myStyle = TStyle( gROOT.GetStyle("Plain") )
38 myStyle.SetName("myStyle")
39
40 myStyle.SetLineStyleString( 5, "[52 12]" )
41 myStyle.SetLineStyleString( 6, "[22 12]" )
42 myStyle.SetLineStyleString( 7, "[22 10 7 10]" )
43
44 # pretty colors
45 myStyle.SetPalette(1)
46
47 # use plain black on white colors
48 myStyle.SetFrameBorderMode(0)
49 myStyle.SetCanvasBorderMode(0)
50 myStyle.SetPadBorderMode(0)
51 myStyle.SetPadColor(0)
52 myStyle.SetFillStyle(0)
53
54 myStyle.SetLegendBorderSize(0)
55
56 myStyle.SetTitleFillColor( c_TitleBox )
57 myStyle.SetFrameFillColor( c_FrameFill )
58 myStyle.SetCanvasColor( c_Canvas )
59
60 # set the paper & margin sizes
61 myStyle.SetPaperSize(20,26)
62 myStyle.SetPadTopMargin(0.11)
63 myStyle.SetPadRightMargin(0.05)
64 myStyle.SetPadBottomMargin(0.11)
65 myStyle.SetPadLeftMargin(0.12)
66
67 # use bold lines and markers
68 myStyle.SetMarkerStyle(21)
69 myStyle.SetMarkerSize(0.3)
70 myStyle.SetHistLineWidth(2)
71 myStyle.SetLineStyleString(2,"[12 12]") # postscript dashes
72
73 # do not display any of the standard histogram decorations
74 myStyle.SetOptTitle(1)
75 myStyle.SetTitleH(0.052)
76
77 myStyle.SetOptStat(0)
78 myStyle.SetOptFit(0)
79
80 # put tick marks on top and RHS of plots
81 myStyle.SetPadTickX(1)
82 myStyle.SetPadTickY(1)
83
84 # use this style
85 gROOT.GetListOfStyles().Add(myStyle)
86 gROOT.SetStyle("myStyle")
87
88def SaveGraphsToFile( filename, varnames, xvecs, yvecs, addparamName = None, addparamVal = None ):
89 f = TFile.Open( filename, 'RECREATE' )
90 if len(xvecs) != len(yvecs) or len(varnames) != len(xvecs):
91 print ('ERROR: wrong dimensions in "AtlRunQueryRoot.SaveGraphsToFile"')
92 return
93 for ik in range(len(xvecs)):
94 xv = xvecs[ik]
95 yv = yvecs[ik]
96 g = TGraph( len(xv) )
97 g.SetName ( "graph%i" % (ik+1) )
98 g.SetTitle( varnames[ik] )
99 for i in range(len(xv)):
100 g.SetPoint( i, xv[i], yv[i] )
101 g.Write()
102
103 # save additional parameter
104 if addparamName is not None:
105 addparam = TF1( addparamName, "[0]", 0, 1 )
106 addparam.SetParameter( 0, addparamVal )
107 addparam.Write()
108
109 f.Close()
110
111def MakePlots( tree, datapath ):
112
113 # batch mode -> no windows
114
115 gROOT.SetBatch( 1 )
116 if tree is None:
117 print ('ERROR: input tree is None')
118 return
119
120 # style --------------------------------------------
121 #steelBlue = TColor.GetColor( "#4863A0" )
122 #slateBlue = TColor.GetColor( "#574EC7" )
123 dogerBlue = TColor.GetColor( "#1569C7" )
124 dogerBlue3 = TColor.GetColor( "#1569C7" )
125 graphCol = dogerBlue
126 markerCol = dogerBlue3
127
128 SetStyle()
129 # --------------------------------------------------
130
131 # reference
132 reflist = [ 'Run' ]
133
134 # types that can be plotted
135 validtypes = [ 'int', 'float', 'double', 'short', 'long' ]
136
137 # get list of leaves for plotting
138 leafList = tree.GetListOfLeaves()
139 varlist = []
140
141 for leaf in leafList:
142 if leaf is not None:
143 # check type, only values are plotted
144 typename = leaf.GetTypeName().lower()
145 valid = False
146 for vt in validtypes:
147 if vt in typename:
148 valid = True
149 if valid:
150 varlist.append( leaf.GetName() )
151
152 # plot
153 TGaxis.SetMaxDigits(6)
154 c = []
155 g = []
156 fnames = []
157 hnames = []
158 for ref in reflist:
159 # sanity check
160 if ref not in varlist:
161 print ('Big troubles in "MakePlots" --> reference variable "%s" not in TTree' % ref)
162 sys.exit(1)
163
164 for var in varlist:
165 if var == ref:
166 continue
167
168 vreg = var.replace('-','_')
169 rreg = ref.replace('-','_')
170
171 hname = "h_%s_%s" % (ref, var)
172 c.append( TCanvas( "c_%s" % hname, "ATLAS Run Query: %s vs. %s" % (var, ref), 0, 0, 600, 500 ) )
173 tree.Draw( "%s:%s" % (vreg, rreg), "", "goff" )
174 g.append( TGraph( tree.GetSelectedRows(), tree.GetV2(), tree.GetV1() ) )
175 g[-1].SetTitle( "ATLAS Run Query: %s vs. %s" % (var, ref) )
176 g[-1].SetMarkerStyle( 20 )
177 g[-1].SetMarkerColor( markerCol )
178 g[-1].SetMarkerSize( 1.2 )
179 g[-1].SetLineWidth( 2 )
180 g[-1].SetLineColor( graphCol )
181 g[-1].GetXaxis().SetTitleOffset( 1.25 )
182 g[-1].GetYaxis().SetTitleOffset( 1.75 )
183 g[-1].GetXaxis().SetTitle( ref )
184 g[-1].GetYaxis().SetTitle( var )
185 g[-1].GetXaxis().SetLabelOffset( 0.012 )
186 g[-1].GetYaxis().SetLabelOffset( 0.012 )
187 g[-1].GetXaxis().SetNdivisions( 506 )
188
189 # some global style settings
190 scale = 1
191 gPad.SetTicks()
192 gPad.SetLeftMargin ( 0.138*scale )
193 gPad.SetRightMargin ( 0.050*scale )
194 gPad.SetBottomMargin( 0.120*scale )
195
196 g[-1].Draw("alp")
197
198 c[-1].Update()
199 fnames.append( '%s/atlrunquery_%s.png' % (datapath, hname) )
200 hnames.append( '%s versus %s' % (var, ref ) )
201 c[-1].Print( fnames[-1] )
202
203 return hnames, fnames
204
205# ---------------------------------------------------------------------------------------------------
206# Creation of ROOT output file
207# ---------------------------------------------------------------------------------------------------
208
209def makeLBPlotSummaryForLHC( lbrange, xvecStb, yvec, runNr, datapath, printText = '' ):
210
211 plotDevelop = False
212 if plotDevelop:
213 # pickle input
214 import json
215 store = (list(lbrange), xvecStb, yvec, runNr, datapath)
216 pf = open( '%s/plotinput.json' % datapath, 'w' )
217 try:
218 json.dump(store, pf)
219 except Exception as ex:
220 print ('ERROR: could not store plotinput data')
221 print("Reason: %s" % ex)
222 sys.exit(1)
223 pf.close()
224
225 # sanity check
226 if not lbrange or len(yvec)==0 or not yvec[0]:
227 return "None"
228
229 ylegend = ['Intensity Beam-1', 'Intensity Beam-2', 'Beam energy', 'Online inst. luminosity' ]
230
231 SetStyle()
232 name = 'LHCsummary_vs_lb_run_%i' % (runNr)
233 #print ("Attempt printing", name)
234 title = 'LHC summary vs. LB for run_%i' % (runNr)
235 szescale = 1.2
236 c = TCanvas( name, title, 0, 0, int(530*szescale), int(400*szescale) )
237 c.GetPad(0).SetTopMargin(0.13)
238 c.GetPad(0).SetLeftMargin(0.09)
239 c.GetPad(0).SetRightMargin(1.3)
240 # c.GetPad(0).SetGrid()
241 x1 = lbrange[0]
242 x2 = lbrange[-1]
243
244 h = TH1F( name, title, x2 - x1 + 1, x1, x2 + 1 ) # for stable beams
245 h.GetXaxis().SetTitle( 'Luminosity block number' )
246 h.GetXaxis().SetTitleOffset( 1.3 )
247 h.GetYaxis().SetTitle( 'Beam intensity (10^{11} protons)' )
248 h.GetYaxis().SetTitleOffset( 1.0 )
249 h.SetTitle( title )
250
251 hg = []
252 for iy,y in enumerate(yvec):
253 hg.append( TH1F( name + 'g%i' % iy, title, x2 - x1 + 1, x1, x2 + 1 ) )
254 for lb in lbrange:
255 hg[iy].SetBinContent( lb, y[lb] )
256
257 # first draw histogram
258 ymax = max(hg[0].GetMaximum(),hg[1].GetMaximum())
259 y2 = ymax*1.3
260 if y2 <= 0:
261 y2 = 1
262 h.SetMinimum(0)
263 h.SetMaximum(y2)
264 h.Draw("0")
265 gPad.SetTicks(1, 0)
266
267 # beam energy (rescale)
268 ebcol = TColor.GetColor( "#306754" )
269 heb = hg[2]
270 heb.Scale( 1./1000.0, "nosw2" ) # in TeV
271 if heb.GetMaximum()==0:
272 heb.SetMaximum(1)
273 ebmax = heb.GetMaximum()*1.3
274
275 # one more sanity check
276 if ebmax == 0:
277 return "None"
278
279 heb.SetLineColor( ebcol )
280 heb.SetLineWidth( 1 )
281 scale = h.GetMaximum()/ebmax
282 heb.SetFillColor( TColor.GetColor("#C3FDB8") )
283 heb.Scale( scale, "nosw2" )
284 heb.Draw("same")
285
286 if xvecStb:
287 for lb in lbrange:
288 if lbrange[lb] in xvecStb:
289 h.SetBinContent( lbrange[lb], yvec[2][lb]*scale/1000.0 ) # draw for beam energy (in TeV)
290 h.SetFillColor( TColor.GetColor( "#63bD58" ) )
291 h.SetFillStyle( 3007 )
292 h.SetLineColor( TColor.GetColor( "#C3FDB8" ) )
293 h.SetLineWidth( 1 )
294 h.SetLineStyle( 1 )
295 h.Draw("same")
296 hebp = TH1F( heb )
297 hebp.SetFillColor( 0 )
298 hebp.Draw("same")
299
300
301 colorList = [TColor.GetColor( "#255EC7" ), TColor.GetColor( "#E42217" ), TColor.GetColor( "#2212EE" ), TColor.GetColor( "#22EE33" )]
302
303 for ig in range(2):
304 if ig < len(colorList):
305 hg[ig].SetLineColor( colorList[ig] )
306 hg[ig].SetLineWidth( 2 )
307 hg[ig].Draw("same")
308
309 # logo
310 now = str(datetime.datetime.today())
311 t = TText( h.GetXaxis().GetXmax(), y2*1.01, 'ATLAS Run Query %s' % now[:now.find('.')] )
312 t.SetTextAlign( 31 )
313 t.SetTextSize( 0.035 )
314 t.SetTextSize( 0.030 )
315 t.SetTextColor( TColor.GetColor( "#888888" ) )
316 t.Draw()
317
318 drawLegend = False
319 dx = 1 - c.GetRightMargin() - c.GetLeftMargin() - 0.05
320 dy = 0.1
321 legend = TLegend( 1 - c.GetRightMargin() - dx, 1 - c.GetTopMargin() - dy - 0.035,
322 1 - c.GetRightMargin(), 1 - c.GetTopMargin() - 0.035)
323 # legend.SetFillStyle( 1001 )
324 # legend.SetFillColor( TColor.GetColor( "#FFFFFF" ) )
325 legend.SetNColumns(2)
326 legend.SetBorderSize( 1 )
327 legend.SetMargin( 0.15 )
328 legend.SetLineColor( 0 )
329 legend.SetTextColor( 1 )
330 for ig,hgg in enumerate(hg):
331 if ylegend[ig]:
332 opt = "l"
333 if ig == 2:
334 opt = "FL"
335 legend.AddEntry( hgg, ylegend[ig], opt )
336 drawLegend = True
337
338 if xvecStb:
339 legend.AddEntry( h, "LBs with stable beams","F" )
340 drawLegend = True
341 if drawLegend:
342 legend.Draw("same")
343
344 # redraw axis
345 h.Draw( "sameaxis" )
346
347 # draw beam energy axis on the right side
348 if h.GetMaximum() > 0:
349 axis = TGaxis( x2 + 1, 0, x2 + 1, h.GetMaximum(), 0, ebmax, 510, "+L" )
350 axis.SetLabelOffset( 0.007 )
351 axis.SetTitleOffset( 1.2 )
352 axis.SetTitleSize( h.GetXaxis().GetTitleSize() )
353 axis.SetLabelSize( 0.04 )
354 axis.SetLineColor( ebcol )
355 axis.SetTitleColor( ebcol )
356 axis.SetLabelColor( ebcol )
357 axis.SetTitle( "Beam energy (TeV)" )
358 axis.Draw()
359
360 c.Update()
361 fnames = '%s/atlrunquery_%s.png' % (datapath, name)
362 c.Print( fnames )
363 #print ("Printing",fnames)
364
365 return fnames
366
367def makeTimePlotList( xvec, xveclb, yvec, toffset, dt, ymin, ymax,
368 xtit, ytit, ylegend, name, title, datapath, printText = '' ):
369
370 # sanity check
371 if not xvec or len(yvec)==0 or not yvec[0]:
372 return "None"
373
374 SetStyle()
375
376 gStyle.SetPadTickX(0)
377 gStyle.SetPadTickY(1)
378
379 c = TCanvas( name, title, 0, 0, 530, 430 )
380 c.GetPad(0).SetTopMargin(0.22)
381 c.GetPad(0).SetGrid()
382
383 # create the graphs
384 graphs = []
385 y1 = 1e30
386 y2 = -1e30
387 for ig in range(len(yvec)):
388 n = len(yvec[ig])
389 g = TGraph( n )
390 for i in range(n):
391 g.SetPoint( i, xvec[ig][i], yvec[ig][i] )
392 if ymin == ymax:
393 y1 = min(y1,yvec[ig][i])
394 y2 = max(y2,yvec[ig][i])
395 graphs.append( g )
396
397 if ymin != ymax:
398 y1 = ymin
399 y2 = ymax
400
401 # create the frame
402 frame = TH1F( name, title, int(dt/2), 0, dt )
403
404 # x-axis is time format
405 frame.GetXaxis().SetTimeDisplay(1)
406 # use rather default: frame.GetXaxis().SetTimeFormat("%d-%Hh")
407
408 # =================================
409 # toffset += 7*3600 # BUG in ROOT ???
410 # =================================
411
412 frame.GetXaxis().SetTimeOffset(toffset, "gmt") # assumes time given was local
413
414 frame.GetXaxis().SetTitle( xtit )
415 frame.GetXaxis().SetTitleOffset( 1.3 )
416 frame.GetYaxis().SetTitle( ytit )
417 frame.GetYaxis().SetTitleOffset( 1.4 )
418 frame.SetTitle( title )
419 frame.SetMinimum(y1 - (y2-y1)*0.1)
420 frame.SetMaximum(y2 + (y2-y1)*0.3)
421 frame.Draw()
422
423 colorList = [TColor.GetColor( "#154EB7" ), TColor.GetColor( "#82CAFF" ), TColor.GetColor( "#D11B17" ), TColor.GetColor( "#FF9A4D" )]
424 styleList = [1,1,1,1]
425
426 for ig,g in enumerate(graphs):
427 if ig < len(colorList):
428 g.SetLineColor( colorList[ig] )
429 g.SetLineStyle( styleList[ig] )
430 g.SetLineWidth( 2 )
431 g.Draw("same")
432
433 if printText:
434 t = TText( frame.GetXaxis().GetXmax(), y2*1.01, printText )
435 t.SetTextAlign( 31 )
436 t.SetTextSize( 0.035 )
437 t.Draw()
438
439 # logo
440 now = str(datetime.datetime.today())
441 tlogo = TText( frame.GetXaxis().GetXmax()*1.025 - frame.GetXaxis().GetXmin()*0.025, y1,
442 'ATLAS Run Query %s' % now[:now.find('.')] )
443 tlogo.SetTextSize( 0.030 )
444 tlogo.SetTextColor( TColor.GetColor( "#aaaaaa" ) )
445 tlogo.SetTextAngle(90)
446 tlogo.Draw()
447
448 # axis for LBs
449 lbcol = TColor.GetColor( "#777777" )
450 axis = TGaxis( frame.GetXaxis().GetXmin(), frame.GetMaximum(),
451 frame.GetXaxis().GetXmax(), frame.GetMaximum(),
452 1, len(xveclb), 505, "-L" )
453 axis.SetLabelOffset( frame.GetXaxis().GetLabelOffset() )
454 axis.SetTitleOffset( 1.4 )
455 axis.SetTitleSize( frame.GetXaxis().GetTitleSize() )
456 axis.SetLabelSize( frame.GetXaxis().GetLabelSize() )
457 axis.SetLineColor( lbcol )
458 axis.SetTitleColor( lbcol )
459 axis.SetLabelColor( lbcol )
460 axis.SetTitle( "Luminosity block number (indicative)" )
461 axis.Draw()
462
463 # legend
464 drawLegend = False
465 dx = 1 - c.GetRightMargin() - c.GetLeftMargin() - 0.05
466 dy = 0.1
467 legend = TLegend( 1 - c.GetRightMargin() - dx, 1 - c.GetTopMargin() - dy - 0.035,
468 1 - c.GetRightMargin(), 1 - c.GetTopMargin() - 0.035)
469 # legend.SetFillStyle( 1001 )
470 # legend.SetFillColor( TColor.GetColor( "#FFFFFF" ) )
471 legend.SetNColumns(2)
472 legend.SetBorderSize( 1 )
473 legend.SetMargin( 0.15 )
474 legend.SetLineColor( 0 )
475 legend.SetTextColor( 1 )
476 legend.SetTextSize( 0.037 )
477 for ig,g in enumerate(graphs):
478 if ylegend[ig]:
479 h = g.GetHistogram()
480 h.SetLineColor( colorList[ig] )
481 h.SetLineStyle( styleList[ig] )
482 h.SetLineWidth( 2 )
483 legend.AddEntry( h, ylegend[ig], "l" )
484 drawLegend = True
485 if drawLegend:
486 legend.Draw("same")
487
488 # redraw axis
489 frame.Draw( "sameaxis" )
490
491 c.Update()
492 fnames = '%s/atlrunquery_%s.png' % (datapath, name)
493 c.Print( fnames )
494
495 return fnames
496
497
498def makeLBPlotList( xvec, xvecStb, yvec, xtitle, ytitle, ylegend, histname, histtitle, datapath, printText = '', ymin = None , ymax = None ):
499
500 # sanity check
501 if not xvec or len(yvec)==0 or not yvec[0]:
502 return "None"
503
504 SetStyle()
505 c = TCanvas( histname, histtitle, 0, 0, 530, 400 )
506 c.GetPad(0).SetTopMargin(0.13)
507 c.GetPad(0).SetGrid()
508 x1 = xvec[0]
509 x2 = xvec[-1]
510
511 h = TH1F( histname, histtitle, x2 - x1 + 1, x1, x2 + 1 ) # for stable beams
512 hg = []
513 for iy in range(len(yvec)):
514 hg.append( TH1F( histname + 'g%i' % iy, histtitle, x2 - x1 + 1, x1, x2 + 1 ) )
515 h.GetXaxis().SetTitle( xtitle )
516 h.GetXaxis().SetTitleOffset( 1.3 )
517 h.GetYaxis().SetTitle( ytitle )
518 h.GetYaxis().SetTitleOffset( 1.4 )
519 h.SetTitle( histtitle )
520
521
522 for i in range(len(xvec)):
523 for iy,y in enumerate(yvec):
524 if type(y[i]) is tuple:
525 val,valerr = y[i]
526 else:
527 val,valerr = y[i],0
528 hg[iy].SetBinContent( xvec[i], val )
529 hg[iy].SetBinError( xvec[i], valerr )
530 if xvec[i] in xvecStb:
531 h.SetBinContent( xvec[i], yvec[0][i] )
532
533 if not ymax:
534 ymax = max([h.GetMaximum() for h in hg])
535 if ymax <= 0:
536 ymax = 1
537 ymax *= 1.2
538
539 if not ymin:
540 ymin=0
541
542 # first draw histogram
543 h.SetMinimum(ymin)
544 h.SetMaximum(ymax)
545 if xvecStb:
546 h.SetFillColor( TColor.GetColor( "#98AFC7" ) )
547 h.SetLineColor( TColor.GetColor( "#788FA7" ) )
548 h.SetLineWidth( 1 )
549 h.SetLineStyle( 1 )
550 h.Draw("")
551 else:
552 h.Draw("0")
553
554 colorList = [TColor.GetColor( "#255EC7" ), TColor.GetColor( "#E42217" ), TColor.GetColor( "#2212EE" ), TColor.GetColor( "#22EE33" )]
555
556 for ig,hgg in enumerate(hg):
557 if ig < len(colorList):
558 hgg.SetLineColor( colorList[ig] )
559 hgg.SetLineWidth( 2 )
560 hgg.Draw("e2same")
561
562 if printText:
563 t = TText( h.GetXaxis().GetXmax(), (ymax-ymin)*1.01 + ymin, printText )
564 t.SetTextAlign( 31 )
565 t.SetTextSize( 0.035 )
566 t.Draw()
567
568 # logo
569 now = str(datetime.datetime.today())
570 tlogo = TText( h.GetXaxis().GetXmax()*1.025, (ymax-ymin)*0.0+ymin, 'ATLAS Run Query %s' % now[:now.find('.')] )
571 tlogo.SetTextSize( 0.030 )
572 tlogo.SetTextColor( TColor.GetColor( "#888888" ) )
573 tlogo.SetTextAngle(90)
574 tlogo.Draw()
575
576 drawLegend = False
577 dx = 0.4
578 dy = 0.10 + 0.03*(len(hg)-1)
579 legend = TLegend( 1 - c.GetRightMargin() - dx, 1 - c.GetTopMargin() - dy - 0.015,
580 1 - c.GetRightMargin(), 1 - c.GetTopMargin() - 0.015)
581 # legend.SetFillStyle( 1001 )
582 # legend.SetFillColor( TColor.GetColor( "#FFFFFF" ) )
583 legend.SetBorderSize( 1 )
584 legend.SetMargin( 0.15 )
585 legend.SetLineColor( 0 )
586 legend.SetTextColor( 1 )
587 for ig,hgg in enumerate(hg):
588 if ylegend[ig]:
589 legend.AddEntry( hgg, ylegend[ig], "l" )
590 drawLegend = True
591
592 if xvecStb:
593 legend.AddEntry( h, "LBs with stable beams","F" )
594 drawLegend = True
595 if drawLegend:
596 legend.Draw("same")
597
598 # redraw axis
599 h.Draw( "sameaxis" )
600
601 c.Update()
602 fnames = '%s/atlrunquery_%s.png' % (datapath, histname)
603 c.Print( fnames )
604
605 return fnames
606
607
608def makeBSPlots( xvec, yvec, xtitle, ytitle, histname, histtitle, datapath, ymin, ymax, printText = '' ):
609
610 # sanity check
611 if not xvec or len(yvec)==0 or not yvec[0]:
612 return None
613
614 ROOT = importroot()
615
616 SetStyle()
617 c = TCanvas( histname, histtitle, 0, 0, 530, 400 )
618 c.GetPad(0).SetTopMargin(0.13)
619 c.GetPad(0).SetGrid()
620
621 bounds = [1]
622 for nlb in xvec:
623 bounds += [bounds[-1]+nlb]
624
625 from array import array
626 h = TH1F( histname + 'g', histtitle, len(bounds)-1, array('f',bounds) )
627 h.GetXaxis().SetTitle( xtitle )
628 h.GetXaxis().SetTitleOffset( 1.3 )
629 h.GetYaxis().SetTitle( ytitle )
630 h.GetYaxis().SetTitleOffset( 1.4 )
631 h.SetTitle( histtitle )
632 h.SetMinimum(ymin)
633 h.SetMaximum(ymax)
634 h.SetFillColor( TColor.GetColor( "#56A5EC" ) )
635 for bin,y in enumerate(yvec):
636 val,valerr = y
637 h.SetBinContent( bin+1, val )
638 h.SetBinError ( bin+1, valerr )
639 h.Draw("e2")
640
641 hc = h.Clone()
642 hc.SetLineColor( ROOT.kBlack )
643 hc.SetLineWidth(2)
644 hc.Draw("e0same")
645
646 if printText:
647 t = TText( h.GetXaxis().GetXmax(), (ymax-ymin)*1.01 + ymin, printText )
648 t.SetTextAlign( 31 )
649 t.SetTextSize( 0.035 )
650 t.Draw()
651
652 # logo
653 now = str(datetime.datetime.today())
654 tlogo = TText( h.GetXaxis().GetXmax()*1.025, (ymax-ymin)*0.0+ymin, 'ATLAS Run Query %s' % now[:now.find('.')] )
655 tlogo.SetTextSize( 0.030 )
656 tlogo.SetTextColor( TColor.GetColor( "#888888" ) )
657 tlogo.SetTextAngle(90)
658 tlogo.Draw()
659
660 c.Update()
661 fnames = '%s/atlrunquery_%s.png' % (datapath, histname)
662 c.Print( fnames )
663
664 return fnames
665
666
667def makeLBPlot( xvec, xvecStb, yvecScalar,
668 xtitle, ytitle, ylegendScalar,
669 histname, histtitle,
670 datapath, printText = '', ymin = None, ymax = None ):
671 yvec = [yvecScalar]
672 ylegend = [ylegendScalar]
673 return makeLBPlotList( xvec, xvecStb, yvec, xtitle, ytitle, ylegend, histname, histtitle, datapath, printText, ymin, ymax )
674
675def InttypeTrf( tree, var, vlist, value, kcoord ):
676 # initialisation
677 if kcoord < 0:
678 vlist.append( array( 'i', [ 0 ] ) )
679 vreg = var.replace('-','_')
680 tree.Branch( vreg, vlist[-1], "%s/I" % vreg )
681 return
682 # fill tree
683 try:
684 vlist[kcoord][0] = int(value)
685 except (ValueError, TypeError):
686 vlist[kcoord][0] = -1
687
688def FloattypeTrf( tree, var, vlist, value, kcoord ):
689 # initialisation
690 if kcoord < 0:
691 vlist.append( array( 'f', [ 0 ] ) )
692 vreg = var.replace('-','_')
693 tree.Branch( vreg, vlist[-1], "%s/F" % vreg )
694 return
695 # fill tree
696 try:
697 vlist[kcoord][0] = float(value)
698 except (ValueError, TypeError):
699 vlist[kcoord][0] = -1
700
701def StringtypeTrf( tree, var, vlist, value, kcoord ):
702 # initialisation
703 if kcoord < 0:
704 vlist.append( array( 'b', 'ab\0'.encode() ) )
705 tree.Branch( var, vlist[-1], "%s/C" % var )
706 return
707 # fill tree
708 try:
709 vlist[kcoord] = array( 'b', (value + '\0').encode() )
710 tree.SetBranchAddress( var, vlist[kcoord] )
711 except (ValueError, TypeError):
712 vlist[kcoord] = array( 'b', 'unknown\0'.encode() )
713 tree.SetBranchAddress( var, vlist[kcoord] )
714
715def TimetypeTrf( tree, var, vlist, value, kcoord ):
716 return StringtypeTrf( tree, var, vlist, value, kcoord )
717
718def DurationtypeTrf( tree, var, vlist, value, kcoord ):
719 # original format: '[2d] 13h 15m 45s' --> transform in seconds
720 if kcoord >= 0:
721 try:
722 dur = value.split()
723 if len(dur) == 3:
724 h, m, s = dur
725 d = '0d'
726 elif len(dur) == 4:
727 d, h, m, s = dur
728 else:
729 print ('Unknown format in "DurationtypeTrf:"')
730 print (dur)
731 sys.exit(1)
732
733 value = str( ( (int(d.replace('d','').strip())*24 + int(h.replace('h','').strip()))*60 +
734 int(m.replace('m','').strip()) )*60 + int(s.replace('s','').strip()) )
735 except (ValueError, TypeError):
736 print (value)
737 sys.exit(1)
738 value = 0
739 InttypeTrf( tree, var, vlist, value, kcoord )
740
741
742def MakeHtml( hnames, fnames, uselarge = False, iconwidth = 70, nbreakline = 10 ):
743 windowstyle = 'openWindow'
744 if uselarge:
745 windowstyle = 'openLargeWindow'
746 iconwidth = 120
747 s = ''
748 s += '<table><tr>'
749 for i in range(0,len(fnames)):
750 f = fnames[i]
751 h = hnames[i]
752 s += """<td valign="top"><a STYLE="text-decoration: none" href=" """
753 s += """javascript:%s('Plot','<!DOCTYPE html PUBLIC &quot;-//W3C//DTD HTML 4.01 Transitional//EN&quot;><html xmlns:my><head><title>%s</title><LINK href=&quot;atlas-runquery.css&quot; rel=&quot;stylesheet&quot; type=&quot;text/css&quot;><body><table style=&quot;font-family: sans-serif; font-size: 85%%&quot;><tr><td valign=&quot;top&quot;>""" % (windowstyle,h)
754 s += """<img src=&quot;%s&quot;>""" % f
755 s += """<hr color=&quot;red&quot; size=1><font color=&quot;#777777&quot;><font size=&quot;-1&quot;><i><font size=&quot;-2&quot;>Created by AtlRunQuery on: %s</font></i></font></td></tr></table><script type=&quot;text/javascript&quot;></script></body></html>""" % str(datetime.datetime.now())
756 s += """')"""
757 s += """ "><img vspace=0 src="%s" title="Plot showing query result for %s" width="%i"></a><br><font size="-2">%s</font></td>""" % (f,h,iconwidth,h)
758 s += """<td>&nbsp;</td>"""
759 if (i+1)%nbreakline == 0:
760 s += '</tr><tr>'
761
762 s += '</tr></table>'
763 return s
764
765def GetActions():
766 # define all actions (choose small letters)
767 # format: 'column header' 'include' 'type' '...'
768 actions = { "run": ( True, InttypeTrf ),
769 "links": ( False, StringtypeTrf ),
770 "#lb": ( True, InttypeTrf ),
771 "#events": ( True, InttypeTrf ),
772 "smk": ( True, InttypeTrf ),
773 "start and endtime": ( True, TimetypeTrf ),
774 "duration": ( True, DurationtypeTrf ),
775 "torcurrent": ( True, FloattypeTrf ),
776 "solcurrent": ( True, FloattypeTrf ),
777 "detector systems": ( False, StringtypeTrf ),
778 "stream": ( True, InttypeTrf ),
779 "lumi": ( True, FloattypeTrf ),
780 "other": ( True, StringtypeTrf )
781 }
782 return actions
783
784def CreateRootFile( dic ):
785
786 from CoolRunQuery.selector.AtlRunQuerySelectorBase import DataKey
787 from CoolRunQuery.AtlRunQueryQueryConfig import QC
788 datapath = QC.datapath
789
790 actions = GetActions()
791
792 if 'data' not in os.listdir('.'):
793 os.mkdir('data')
794
795 f = open( '%s/dummyroot.txt' % datapath, 'w' )
796
797 froot = TFile( '%s/atlrunquery.root' % datapath, 'RECREATE' )
798
799 tree = TTree( 'RunQuery', 'ATLAS Run Query' )
800
801 varlist = []
802
803 # create branches
804 keylist = []
805 for data_key in dic:
806
807 #print ("DATA_KEY",type(data_key),data_key)
808 f.write( 'key: %s \n' % (data_key.ResultKey))
809
810 # the actual variable name used in the tree
811 var = data_key.ResultKey.replace('#',' ').strip().replace(' ','_').replace('STR:','').replace('-','_').replace(':','_')
812
813 k = data_key.ResultKey.lower()
814 if data_key.Type==DataKey.STREAM:
815 k = 'stream'
816 elif 'olc_' == data_key.ResultKey[0:4]:
817 k = 'lumi'
818
819 # is key included in 'actions' dictionary ?
820 if k not in actions:
821 k = 'other'
822
823 useit, function = actions[k]
824 if useit:
825 function( tree, var, varlist, -1, -1 ) # create branch
826 keylist.append( (data_key,var, function) )
827
828 # loop over runs
829 for ev in range(len(dic[DataKey('Run')])):
830 for k,(data_key,var,function) in enumerate(keylist):
831
832 val = dic[data_key][ev]
833 if data_key.Type==DataKey.STREAM:
834 val = val[0]
835
836 function( tree, var, varlist, val, k )
837
838 tree.Fill()
839
840 # make plots ?
841 hnames, fnames = MakePlots( tree, datapath )
842 htmlstr = MakeHtml( hnames, fnames )
843
844 f.close()
845 tree.Write()
846 froot.Close()
847
848 return froot.GetName(), htmlstr
849
850
851
852def makeRatePlot( v, lbduration, plottriggers, averrate, xtit, ytit, name, title, datapath, printText = '' ):
853
854
864
865 ROOT = importroot()
866
867 SetStyle()
868 c = TCanvas( name, title, 0, 0, 530, 400 )
869 c.GetPad(0).SetTopMargin(0.13)
870 c.GetPad(0).SetGrid()
871
872 firstlb = min([lbinfo[0][0] for lbinfo in v.values()])
873 lastlb = max([lbinfo[-1][0] for lbinfo in v.values()])
874
875 frame = TH1F( name, title, lastlb-firstlb+1, firstlb, lastlb+1 )
876 frame.GetXaxis().SetTitle( xtit )
877 frame.GetXaxis().SetTitleOffset( 1.3 )
878 frame.GetYaxis().SetTitle( ytit )
879 frame.GetYaxis().SetTitleOffset( 1.4 )
880
881
882 hTAP = {} # trigger after prescale (not realy, it contains (tap+tav)/2 with and error of (tap-tav)/2)
883 hTAV = {} # trigger after veto
884 ymax = -1
885
886 # see http://root.cern.ch/root/html526/TColorWheel.html
887 colorList = [ (ROOT.kBlue,ROOT.kAzure+1),
888 (ROOT.kGreen+2,ROOT.kGreen+2),
889 (ROOT.kRed+1,ROOT.kRed),
890 (ROOT.kTeal-1,ROOT.kTeal),
891 (ROOT.kOrange+8,ROOT.kOrange),
892 (ROOT.kYellow+1,ROOT.kYellow),
893 (ROOT.kOrange+3,ROOT.kOrange+5),
894 (ROOT.kPink+8,ROOT.kPink+9) ]
895
896 clrIdx = 0
897 for trname,avrate in plottriggers:
898 hTAP[trname] = TH1F( '%s_ap_%s' % (name,trname), title, lastlb-firstlb+1, firstlb, lastlb+1 )
899 hTAV[trname] = TH1F( '%s_av_%s' % (name,trname), title, lastlb-firstlb+1, firstlb, lastlb+1 )
900 hap = hTAP[trname]
901 hav = hTAV[trname]
902 hav.SetLineWidth( 2 )
903 if clrIdx < len(colorList):
904 dark,bright = colorList[clrIdx]
905 hav.SetLineColor( dark )
906 hap.SetFillColor( bright )
907 hap.SetMarkerColor( bright )
908 clrIdx += 1
909
910 counts = v[trname]
911 for (lb,bp,ap,av),(lb2,lbstart,dt) in zip(counts,lbduration):
912 _apr = float(ap)/dt
913 _avr = float(av)/dt
914 hap.SetBinContent( lb-firstlb+1, (_apr+_avr)/2 )
915 hap.SetBinError( lb-firstlb+1, (_apr-_avr)/2 )
916 hav.SetBinContent( lb-firstlb+1, _avr )
917 ymax = max(ymax,_apr)
918
919
920 if ymax <= 0:
921 ymax = 1
922
923 # first draw histogram
924 y2 = ymax*1.2
925 y1 = 0.01
926 frame.SetMinimum(y1)
927 frame.SetMaximum(y2)
928 frame.Draw("0")
929
930 for h in hTAP.values():
931 h.Draw("same e2")
932 for h in hTAV.values():
933 h.Draw("same")
934
935 if printText:
936 t = TText( frame.GetXaxis().GetXmax(), y2*1.01, printText )
937 t.SetTextAlign( 31 )
938 t.SetTextSize( 0.035 )
939 t.Draw()
940
941 # logo
942 now = str(datetime.datetime.today())
943 tlogo = TText( frame.GetXaxis().GetXmax()*1.025, y1, 'ATLAS Run QUERY %s' % now[:now.find('.')] )
944 tlogo.SetTextSize( 0.030 )
945 tlogo.SetTextColor( TColor.GetColor( "#888888" ) )
946 tlogo.SetTextAngle(90)
947 tlogo.Draw()
948
949 dx = 0.4
950 dy = 0.10 + 0.01*(len(hTAV)-1)
951 legend = TLegend( 1 - c.GetRightMargin() - dx, 1 - c.GetTopMargin() - dy - 0.015,
952 1 - c.GetRightMargin(), 1 - c.GetTopMargin() - 0.015)
953 legend.SetNColumns(2)
954 legend.SetBorderSize( 0 )
955 legend.SetMargin( 0.15 )
956 legend.SetLineColor( 0 )
957 legend.SetTextColor( 1 )
958 # fill legend - highest average rate first
959 for trname,avtrrate in averrate:
960 if trname not in hTAV:
961 continue
962 legend.AddEntry( hTAV[trname], trname, "l" )
963 legend.Draw("same")
964
965 # redraw axis
966 frame.Draw( "sameaxis" )
967
968 c.Update()
969 fnames = '%s/atlrunquery_%s.png' % (datapath, name)
970 c.Print( fnames )
971 frame.SetMaximum(ymax*5)
972 c.SetLogy(1)
973 c.Update()
974 fnames2 = '%s/atlrunquery_%s_log.png' % (datapath, name)
975 c.Print( fnames2 )
976
977 return fnames
978
979
980
981
982
983# command line driver for convenience
984if __name__=='__main__':
985
986 import json
987 datapath = "."
988 pf = open( '%s/plotinput.json' % datapath, 'r' )
989 (lbrange, xvecStb, yvec, runNr, datapath) = json.load(pf)
990 lbrange = range(len(lbrange))
991 datapath = "test"
992 makeLBPlotSummaryForLHC( lbrange, xvecStb, yvec, runNr, datapath, printText = '' )
print
void print(char *figname, TCanvas *c1)
Definition TRTCalib_StrawStatusPlots.cxx:26
min
#define min(a, b)
Definition cfImp.cxx:40
max
#define max(a, b)
Definition cfImp.cxx:41
array
STL class.
replace
std::string replace(std::string s, const std::string &s2, const std::string &s3)
Definition hcg.cxx:310
python.output.AtlRunQueryRoot.TimetypeTrf
TimetypeTrf(tree, var, vlist, value, kcoord)
Definition AtlRunQueryRoot.py:715
python.output.AtlRunQueryRoot.SaveGraphsToFile
SaveGraphsToFile(filename, varnames, xvecs, yvecs, addparamName=None, addparamVal=None)
Definition AtlRunQueryRoot.py:88
python.output.AtlRunQueryRoot.StringtypeTrf
StringtypeTrf(tree, var, vlist, value, kcoord)
Definition AtlRunQueryRoot.py:701
python.output.AtlRunQueryRoot.makeRatePlot
makeRatePlot(v, lbduration, plottriggers, averrate, xtit, ytit, name, title, datapath, printText='')
Definition AtlRunQueryRoot.py:852
python.output.AtlRunQueryRoot.FloattypeTrf
FloattypeTrf(tree, var, vlist, value, kcoord)
Definition AtlRunQueryRoot.py:688
python.output.AtlRunQueryRoot.DurationtypeTrf
DurationtypeTrf(tree, var, vlist, value, kcoord)
Definition AtlRunQueryRoot.py:718
python.output.AtlRunQueryRoot.makeLBPlotList
makeLBPlotList(xvec, xvecStb, yvec, xtitle, ytitle, ylegend, histname, histtitle, datapath, printText='', ymin=None, ymax=None)
Definition AtlRunQueryRoot.py:498
python.output.AtlRunQueryRoot.makeLBPlot
makeLBPlot(xvec, xvecStb, yvecScalar, xtitle, ytitle, ylegendScalar, histname, histtitle, datapath, printText='', ymin=None, ymax=None)
Definition AtlRunQueryRoot.py:670
python.output.AtlRunQueryRoot.makeBSPlots
makeBSPlots(xvec, yvec, xtitle, ytitle, histname, histtitle, datapath, ymin, ymax, printText='')
Definition AtlRunQueryRoot.py:608
python.output.AtlRunQueryRoot.MakeHtml
MakeHtml(hnames, fnames, uselarge=False, iconwidth=70, nbreakline=10)
Definition AtlRunQueryRoot.py:742
python.output.AtlRunQueryRoot.InttypeTrf
InttypeTrf(tree, var, vlist, value, kcoord)
Definition AtlRunQueryRoot.py:675
python.output.AtlRunQueryRoot.makeLBPlotSummaryForLHC
makeLBPlotSummaryForLHC(lbrange, xvecStb, yvec, runNr, datapath, printText='')
Definition AtlRunQueryRoot.py:209
python.output.AtlRunQueryRoot.makeTimePlotList
makeTimePlotList(xvec, xveclb, yvec, toffset, dt, ymin, ymax, xtit, ytit, ylegend, name, title, datapath, printText='')
Definition AtlRunQueryRoot.py:368
type
Generated on for ATLAS Offline Software by doxygen 1.14.0