SiStripMonitorClient/scripts/getGTfromDQMFile_V2.py

0001 #!/usr/bin/env python3
0002 from math import *
0003 from ROOT import TFile, TObject, TTree
0004 from array import array
0005 from ROOT import gDirectory
0006 import sys
0007 import os
0008
0009 def getGTfromDQMFile(DQMfile, RunNumber, globalTagVar):
0010     if not os.path.isfile(DQMfile):#    print "Error: file", DQMfile, "not found, exit"
0011         sys.exit(0)
0012     thefile = TFile( DQMfile )
0013     globalTagDir = 'DQMData/Run ' + RunNumber + '/Info/Run summary/CMSSWInfo'
0014     if not gDirectory.GetDirectory( globalTagDir ):
0015         #    print "Warning: globalTag not found in DQM file"
0016         sys.exit(0)
0017     keys = gDirectory.GetDirectory( globalTagDir ).GetListOfKeys()
0018     key = keys[0]
0019     globalTag = ''
0020     while key:
0021         obj = key.ReadObj()
0022         if globalTagVar in obj.GetName():
0023             globalTag = obj.GetName()[len("<"+globalTagVar+">s="):-len("</"+globalTagVar+">")]
0024             break
0025         key = keys.After(key)
0026     if len(globalTag) > 1:
0027         if globalTag.find('::') >= 0:
0028             print(globalTag[0:globalTag.find('::')])
0029         else:
0030             print(globalTag)
0031     return globalTag
0032