Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2022-03-14 23:57:23

0001 import ROOT
0002 import os, sys
0003 import argparse
0004 import array
0005 
0006 
0007 bitDigiOcc = 0
0008 
0009 listBitConfFilter = [
0010     { "name": "AMC13", "error": 1, "warning": -1 }, 
0011     { "name": "AMC",   "error": 2, "warning": 3  }, 
0012     { "name": "OH",    "error": 4, "warning": 5  }, 
0013     { "name": "VFAT",  "error": 6, "warning": 7  }, 
0014 ]
0015 
0016 
0017 parser = argparse.ArgumentParser()
0018 
0019 parser.add_argument("src", default="", help="")
0020 parser.add_argument("dst", default="", help="")
0021 
0022 opts = parser.parse_args()
0023 
0024 fDQM = ROOT.TFile.Open(opts.src)
0025 dirMain = None
0026 
0027 for k1 in fDQM.GetListOfKeys(): 
0028     d1 = fDQM.Get(k1.GetName())
0029     if not isinstance(d1, ROOT.TDirectoryFile): 
0030         continue
0031     
0032     for k2 in d1.GetListOfKeys(): 
0033         d2 = d1.Get(k2.GetName())
0034         if not isinstance(d1, ROOT.TDirectoryFile): 
0035             continue
0036         
0037         dirMain = d2
0038         break
0039     
0040     if dirMain: 
0041         break
0042 
0043 dirEventInfo = dirMain.Get("GEM/Run summary/EventInfo")
0044 listHistLumi = [ k.GetName() for k in dirEventInfo.GetListOfKeys() ]
0045 listHistLumi = [ s for s in listHistLumi if s.startswith("chamberStatus_inLumi") ]
0046 
0047 fOut = ROOT.TFile.Open(opts.dst, "RECREATE")
0048 
0049 for name in listHistLumi: 
0050     histCurr = dirEventInfo.Get(name)
0051     
0052     listFillLumi = [ histCurr.GetBinContent(i + 1, 0) for i in range(histCurr.GetNbinsX()) ]
0053     numLumi = max([ i for i, x in enumerate(listFillLumi) if abs(x) > 0 ]) + 1
0054     
0055     listBinLumi = [ histCurr.GetXaxis().GetBinLowEdge(i + 1) for i in range(numLumi              + 1) ]
0056     listBinY    = [ histCurr.GetYaxis().GetBinLowEdge(i + 1) for i in range(histCurr.GetNbinsY() + 1) ]
0057     
0058     for dicConf in listBitConfFilter: 
0059         fOut.cd()
0060         
0061         numBinY = len(listBinY) - 1
0062         
0063         histNew = ROOT.TH2S(
0064             dicConf[ "name" ] + "_" + name, 
0065             histCurr.GetTitle() + " ({})".format(dicConf[ "name" ]), 
0066             numLumi, 
0067             array.array("d", listBinLumi), 
0068             numBinY, 
0069             array.array("d", listBinY), 
0070         )
0071         
0072         histNew.GetXaxis().SetTitle(histCurr.GetXaxis().GetTitle())
0073         histNew.GetYaxis().SetTitle(histCurr.GetYaxis().GetTitle())
0074         
0075         for i in range(len(listBinY) - 1): 
0076             histNew.GetYaxis().SetBinLabel(i + 1, histCurr.GetYaxis().GetBinLabel(i + 1))
0077         
0078         for j in range(numBinY): 
0079             for i in range(numLumi): 
0080                 val  = int(histCurr.GetBinContent(i + 1, j + 1))
0081                 occ  = val & ( 1 << bitDigiOcc           ) != 0
0082                 err  = val & ( 1 << dicConf[ "error"   ] ) != 0 if dicConf[ "error"   ] >= 0 else False
0083                 warn = val & ( 1 << dicConf[ "warning" ] ) != 0 if dicConf[ "warning" ] >= 0 else False
0084                 print(dicConf[ "name" ], val, occ, err, warn)
0085                 
0086                 out = 0
0087                 if err: 
0088                     out = 2
0089                 elif warn: 
0090                     out = 3
0091                 elif occ: 
0092                     out = 1
0093                 
0094                 histNew.SetBinContent(i + 1, j + 1, out)
0095         
0096         histNew.Write()
0097 
0098 fOut.Write()
0099 fOut.Close()
0100 fDQM.Close()
0101 
0102