Merge branch 'DCube_PRD' into 'master'

MuonPRDTest: Update the digit and reco DCube histograms

See merge request !46657

MuonSpectrometer/MuonValidation/MuonPRDTest/scripts/createDCubeDigitHistograms_withSel.py

@@ -7,7 +7,7 @@ import math
 from DCubeHistograms import MyHistoFiller
 
 if __name__ == "__main__":
-    parser = argparse.ArgumentParser(prog='createDCubeSimHistograms', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser = argparse.ArgumentParser(prog='createDCubeDigitHistograms', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
     parser.add_argument('-i', '--inputFile', help='choose input ROOT file', default='NSWPRDValAlg.digi.ntuple.root', type=str)
     parser.add_argument('-o', '--outputFile', help='choose output ROOT file', default='NSWPRDValAlg.digi.dcube.root', type=str)
     parser.add_argument('--doCSC', help='turn off CSC if using Run4 input ROOT file', default=False, action='store_true')
@@ -63,8 +63,13 @@ if __name__ == "__main__":
     #Filling
     for i in range(inputTree.GetEntries()):
         inputTree.GetEntry(i)
-        cschists = []
-        tgchists = []
+        cscDigitHists = []
+        cscSDOHists = []
+        cscRDOHists = []
+        tgcDigitHists = []
+        tgcSDOHists = []
+        tgcRDOHists = []
+
 
         # CSCs
         if Options.doCSC == True:
@@ -80,9 +85,19 @@ if __name__ == "__main__":
             else:
                 csc_sector_sel = lambda s: s.Digits_CSC_stationPhi[ncscDigit] == CSC_sector
 
-            for ncscDigit in range(0,len(inputTree.Digits_CSC_truth_localPosX)):
-                cschists += [MyHistoFiller( chamber_name = "CSC_Digit", eta_sel = csc_eta_sel, sector_sel = csc_sector_sel )]
-                cschists[ncscDigit].fill(inputTree, ncscDigit)
+            for ncscDigit in range(0,len(inputTree.Digits_CSC_localPosX)):
+                cscDigitHists += [MyHistoFiller( chamber_name = "CSC_Digit", eta_sel = csc_eta_sel, sector_sel = csc_sector_sel )]
+                cscDigitHists[ncscDigit].fill(inputTree, ncscDigit)
+
+            for ncscSDO in range(0,len(inputTree.SDO_CSC_localPosX)):
+                cscSDOHists += [MyHistoFiller( chamber_name = "CSC_SDO", eta_sel = csc_eta_sel, sector_sel = csc_sector_sel )]
+                cscSDOHists[ncscSDO].fill(inputTree, ncscSDO)
+                
+            for ncscRDO in range(0,len(inputTree.RDO_CSC_localPosX)):
+                cscRDOHists += [MyHistoFiller( chamber_name = "CSC_RDO", eta_sel = csc_eta_sel, sector_sel = csc_sector_sel )]
+                cscRDOHists[ncscRDO].fill(inputTree, ncscRDO)
+                
+
                 
         # TGCs
         if TGC_eta == "positive":
@@ -98,14 +113,34 @@ if __name__ == "__main__":
             tgc_sector_sel = lambda s: s.Digits_TGC_stationPhi[ntgcDigit] == TGC_sector
 
         for ntgcDigit in range(0,len(inputTree.Digits_TGC_localPosX)):
-            tgchists += [MyHistoFiller( chamber_name = "TGC_Digit", eta_sel = tgc_eta_sel, sector_sel = tgc_sector_sel )]
-            tgchists[ntgcDigit].fill(inputTree, ntgcDigit)
+            tgcDigitHists += [MyHistoFiller( chamber_name = "TGC_Digit", eta_sel = tgc_eta_sel, sector_sel = tgc_sector_sel )]
+            tgcDigitHists[ntgcDigit].fill(inputTree, ntgcDigit)
+            
+        for ntgcSDO in range(0,len(inputTree.SDO_TGC_localPosX)):
+            tgcSDOHists += [MyHistoFiller( chamber_name = "TGC_SDO", eta_sel = tgc_eta_sel, sector_sel = tgc_sector_sel )]
+            tgcSDOHists[ntgcSDO].fill(inputTree, ntgcSDO)
+            
+        for ntgcRDO in range(0,len(inputTree.RDO_TGC_localPosX)):
+            tgcRDOHists += [MyHistoFiller( chamber_name = "TGC_RDO", eta_sel = tgc_eta_sel, sector_sel = tgc_sector_sel )]
+            tgcRDOHists[ntgcRDO].fill(inputTree, ntgcRDO)
 
     #Writing
     if Options.doCSC == True:
-        cschist = MyHistoFiller( chamber_name = "CSC_Digit", eta_sel = None, sector_sel = None )
-        cschist.write(ODir)
+        cscDigitHist = MyHistoFiller( chamber_name = "CSC_Digit", eta_sel = None, sector_sel = None )
+        cscDigitHist.write(ODir)
+        
+        cscSDOHist = MyHistoFiller( chamber_name = "CSC_SDO", eta_sel = None, sector_sel = None )
+        cscSDOHist.write(ODir)
+        
+        cscRDOHist = MyHistoFiller( chamber_name = "CSC_RDO", eta_sel = None, sector_sel = None )
+        cscRDOHist.write(ODir)
+    
+    tgcDigitHist = MyHistoFiller( chamber_name = "TGC_Digit", eta_sel = None, sector_sel = None )
+    tgcDigitHist.write(ODir)
+    
+    tgcSDOHist = MyHistoFiller( chamber_name = "TGC_SDO", eta_sel = None, sector_sel = None )
+    tgcSDOHist.write(ODir)
     
-    tgchist = MyHistoFiller( chamber_name = "TGC_Digit", eta_sel = None, sector_sel = None )
-    tgchist.write(ODir)
+    tgcRDOHist = MyHistoFiller( chamber_name = "TGC_RDO", eta_sel = None, sector_sel = None )
+    tgcRDOHist.write(ODir)
     
\ No newline at end of file

MuonSpectrometer/MuonValidation/MuonPRDTest/scripts/createDCubeRecoHistograms_withSel.py

+# Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
+
+# this script can be used to create DCube histograms from the output ntuples of NSWPRDValAlg
+
+import os, sys, ROOT, argparse
+import math
+from DCubeHistograms import MyHistoFiller
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(prog='createDCubeRecoHistograms', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser.add_argument('-i', '--inputFile', help='choose input ROOT file', default='NSWPRDValAlg.reco.ntuple.root', type=str)
+    parser.add_argument('-o', '--outputFile', help='choose output ROOT file', default='NSWPRDValAlg.reco.dcube.root', type=str)
+    parser.add_argument('--doCSC', help='turn off CSC if using Run4 input ROOT file', default=True, action='store_true')
+    parser.add_argument('--CSCsel', help='Choose eta_sector selections for CSC, e.g. positive_1 for positive eta and sector 1, None_None for no selection', default='None_None', type=str)
+    parser.add_argument('--TGCsel', help='Choose eta_sector selections for TGC, e.g. positive_1 for positive eta and sector 1, None_None for no selection', default='None_None', type=str)
+
+    Options = parser.parse_args()
+
+    ROOT.gROOT.SetBatch(True)
+
+    if not os.path.exists(Options.inputFile):
+        print ('ERROR: File %s does not exist'%Options.inputFile)
+        sys.exit(1)
+
+    inputFile = ROOT.TFile(Options.inputFile, "READ")
+    if not inputFile:
+        print ('ERROR: Failed to open file %s'%Options.inputFile)
+        sys.exit(1)
+    inputTree = inputFile.Get("NSWValTree")
+    if not inputTree:
+        print ('ERROR: NSWValTree does not exist in file %s'%Options.inputFile)
+        sys.exit(1)
+
+    nEntries = inputTree.GetEntries()
+    if nEntries==0:
+        print ('ERROR: NSWValTree of file %s has 0 entries'%Options.inputFile)
+        sys.exit(1)
+
+    outputFile = ROOT.TFile(Options.outputFile, "RECREATE")
+    if not outputFile:
+        print ('ERROR: Failed to open file %s'%Options.outputFile)
+        sys.exit(1)
+
+    outputFile.cd()
+    outputFile.mkdir("reconstruction/")
+    ODir = outputFile.GetDirectory("reconstruction/")
+    ODir.cd()
+    
+    CSCselections = Options.CSCsel.split("_")
+    CSC_eta = CSCselections[0]
+    if CSCselections[1] != "None":
+        CSC_sector = int (CSCselections[1])
+    else:
+        CSC_sector = CSCselections[1]
+        
+    TGCselections = Options.TGCsel.split("_")
+    TGC_eta = TGCselections[0]
+    if TGCselections[1] != "None":
+        TGC_sector = int (TGCselections[1])
+    else:
+        TGC_sector = TGCselections[1]
+        
+    #Filling
+    for i in range(inputTree.GetEntries()):
+        inputTree.GetEntry(i)
+        cscPRDHists = []
+        tgcPRDHists = []
+
+        # CSCs
+        if Options.doCSC == True:
+            if CSC_eta == "positive":
+                csc_eta_sel = lambda t: t.PRD_CSC_stationEta[ncscPRD] >= 0
+            elif CSC_eta == "negative":
+                csc_eta_sel = lambda t: t.PRD_CSC_stationEta[ncscPRD] < 0
+            else:
+                csc_eta_sel = lambda t: t.PRD_CSC_stationEta[ncscPRD] < 9
+
+            if CSC_sector == "None":
+                csc_sector_sel = lambda s: s.PRD_CSC_stationPhi[ncscPRD] < 10
+            else:
+                csc_sector_sel = lambda s: s.PRD_CSC_stationPhi[ncscPRD] == CSC_sector
+
+            for ncscPRD in range(0,len(inputTree.PRD_CSC_localPosX)):
+                cscPRDHists += [MyHistoFiller( chamber_name = "CSC_PRD", eta_sel = csc_eta_sel, sector_sel = csc_sector_sel )]
+                cscPRDHists[ncscPRD].fill(inputTree, ncscPRD)
+                
+
+                
+        # TGCs
+        if TGC_eta == "positive":
+            tgc_eta_sel = lambda t: t.PRD_TGC_stationEta[ntgcPRD] >= 0
+        elif TGC_eta == "negative":
+            tgc_eta_sel = lambda t: t.PRD_TGC_stationEta[ntgcPRD] < 0
+        else:
+            tgc_eta_sel = lambda t: t.PRD_TGC_stationEta[ntgcPRD] < 9
+
+        if TGC_sector == "None":
+            tgc_sector_sel = lambda s: s.PRD_TGC_stationPhi[ntgcPRD] < 51
+        else:
+            tgc_sector_sel = lambda s: s.PRD_TGC_stationPhi[ntgcPRD] == TGC_sector
+            
+        for ntgcPRD in range(0,len(inputTree.PRD_TGC_localPosX)):
+            tgcPRDHists += [MyHistoFiller( chamber_name = "TGC_PRD", eta_sel = tgc_eta_sel, sector_sel = tgc_sector_sel )]
+            tgcPRDHists[ntgcPRD].fill(inputTree, ntgcPRD)
+
+
+    #Writing
+    if Options.doCSC == True:
+        cscPRDHist = MyHistoFiller( chamber_name = "CSC_PRD", eta_sel = None, sector_sel = None )
+        cscPRDHist.write(ODir)
+    
+    tgcPRDHist = MyHistoFiller( chamber_name = "TGC_PRD", eta_sel = None, sector_sel = None )
+    tgcPRDHist.write(ODir)
+    
\ No newline at end of file

MuonSpectrometer/MuonValidation/MuonPRDTest/src/NSWPRDValAlg.h

@@ -146,7 +146,7 @@ class NSWPRDValAlg: public AthAlgorithm
   Gaudi::Property<std::string> m_TGC_SDOContainerName{this, "TGC_SDOContainerName", "TGC_SDO" };
   Gaudi::Property<std::string> m_TGC_DigitContainerName{this, "TGC_DigitContainerName","TGC_DIGITS"  };
   Gaudi::Property<std::string> m_TGC_RDOContainerName{this,"TGC_RDOContainerName", "TGCRDO" };
-  Gaudi::Property<std::string> m_TGC_PRDContainerName{this, "TGC_PRDContainerName","TGCPRD" };
+  Gaudi::Property<std::string> m_TGC_PRDContainerName{this, "TGC_PRDContainerName","TGC_Measurements" };
 
   // Matching algorithm