CPAlgorithm: add IOStatsAlg from TopCPToolkit

CPAlgorithm: add IOStatsAlg from TopCPToolkit

PhysicsAnalysis/Algorithms/AnalysisAlgorithmsConfig/data/example_config.yaml

@@ -62,17 +62,17 @@ TauJets:
       tauID: 'Tight'
       triggerChainsPerYear:
         '2015':
-	  - 'HLT_tau25_medium1_tracktwo'
-	  - 'HLT_tau35_medium1_tracktwo'
-	'2016':
-	  - 'HLT_tau25_medium1_tracktwo'
-	  - 'HLT_tau35_medium1_tracktwo'
-	'2017':
-	  - 'HLT_tau25_medium1_tracktwo'
-	  - 'HLT_tau35_medium1_tracktwo'
-	'2018':
-	  - 'HLT_tau25_medium1_tracktwoEF_OR_mediumRNN_tracktwoMVA'
-	  - 'HLT_tau35_medium1_tracktwoEF_OR_mediumRNN_tracktwoMVA'
+          - 'HLT_tau25_medium1_tracktwo'
+          - 'HLT_tau35_medium1_tracktwo'
+        '2016':
+          - 'HLT_tau25_medium1_tracktwo'
+          - 'HLT_tau35_medium1_tracktwo'
+        '2017':
+          - 'HLT_tau25_medium1_tracktwo'
+          - 'HLT_tau35_medium1_tracktwo'
+        '2018':
+          - 'HLT_tau25_medium1_tracktwoEF_OR_mediumRNN_tracktwoMVA'
+          - 'HLT_tau35_medium1_tracktwoEF_OR_mediumRNN_tracktwoMVA'
     PtEtaSelection: {}
 
 GeneratorLevelAnalysis: {}

PhysicsAnalysis/Algorithms/AnalysisAlgorithmsConfig/python/ConfigFactory.py

@@ -340,4 +340,8 @@ class ConfigFactory():
         self.addAlgConfigBlock(algName="Output", alg=OutputAnalysisConfig,
             defaults={'configName': 'Output'})
 
+        # IOStats printouts
+        from AsgAnalysisAlgorithms.AsgAnalysisConfig import IOStatsBlock
+        self.addAlgConfigBlock(algName="IOStats", alg=IOStatsBlock)
+
         return

PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/AsgAnalysisAlgorithms/AsgAnalysisAlgorithmsDict.h

@@ -41,5 +41,6 @@
 #include <AsgAnalysisAlgorithms/SystObjectUnioniserAlg.h>
 #include <AsgAnalysisAlgorithms/TreeFillerAlg.h>
 #include <AsgAnalysisAlgorithms/TreeMakerAlg.h>
+#include <AsgAnalysisAlgorithms/IOStatsAlg.h>
 
 #endif

PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/AsgAnalysisAlgorithms/IOStatsAlg.h

+/*
+  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
+*/
+
+/// @author Zhengcheng Tao
+
+
+#ifndef ASG_IOSTATSALG_H
+#define ASG_IOSTATSALG_H
+
+#include <AnaAlgorithm/AnaAlgorithm.h>
+#include <AsgTools/PropertyWrapper.h>
+
+namespace CP {
+
+  class IOStatsAlg final : public EL::AnaAlgorithm {
+
+  public:
+    using EL::AnaAlgorithm::AnaAlgorithm;
+
+    // finalize function: run once at the end of a job after event loop
+    virtual StatusCode finalize() override;
+
+  private:
+    Gaudi::Property<std::string> m_printOption {this, "printOption", "Summary", "option to pass the standard ROOT printing function. Can be 'Summary', 'ByEntries' or 'ByBytes'."};
+
+  };
+
+} // namespace
+
+#endif

PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/AsgAnalysisAlgorithms/selection.xml

@@ -34,5 +34,6 @@
    <class name="CP::SystObjectUnioniserAlg" />
    <class name="CP::TreeFillerAlg" />
    <class name="CP::TreeMakerAlg" />
+   <class name="CP::IOStatsAlg" />
 
 </lcgdict>

PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/Root/IOStatsAlg.cxx

+/*
+  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
+*/
+
+/// @author Zhengcheng Tao
+
+#include <AsgAnalysisAlgorithms/IOStatsAlg.h>
+
+#include "xAODCore/tools/IOStats.h"
+#include "xAODCore/tools/ReadStats.h"
+
+namespace CP {
+
+  StatusCode IOStatsAlg::finalize() {
+
+    xAOD::IOStats::instance().merged().Print(m_printOption.value().c_str());
+
+    return StatusCode::SUCCESS;
+  }
+
+} // namespace

PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/python/AsgAnalysisConfig.py

@@ -82,6 +82,19 @@ class CommonServicesConfig (ConfigBlock) :
             sysDumper.histogramName = self.systematicsHistogram
 
 
+class IOStatsBlock(ConfigBlock):
+    """Print what branches are used in analysis"""
+
+    def __init__(self):
+        super(IOStatsBlock, self).__init__()
+        self.addOption("printOption", "Summary", type=str,
+                       info='option to pass the standard ROOT printing function. Can be "Summary", "ByEntries" or "ByBytes".')
+
+    def makeAlgs(self, config):
+        alg = config.createAlgorithm('CP::IOStatsAlg', 'IOStatsAlg')
+        alg.printOption = self.printOption
+
+
 class PileupReweightingBlock (ConfigBlock):
     """the ConfigBlock for pileup reweighting"""
 

PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/src/components/AsgAnalysisAlgorithms_entries.cxx

@@ -37,6 +37,7 @@
 #include <AsgAnalysisAlgorithms/TreeMakerAlg.h>
 #include "AsgAnalysisAlgorithms/SystObjectLinkerAlg.h"
 #include "AsgAnalysisAlgorithms/SystObjectUnioniserAlg.h"
+#include "AsgAnalysisAlgorithms/IOStatsAlg.h"
 
 DECLARE_COMPONENT (CP::AsgFlagSelectionTool)
 DECLARE_COMPONENT (CP::AsgMaskSelectionTool)
@@ -69,6 +70,7 @@ DECLARE_COMPONENT (CP::PMGTruthWeightAlg)
 DECLARE_COMPONENT (CP::SysListDumperAlg)
 DECLARE_COMPONENT (CP::TreeFillerAlg)
 DECLARE_COMPONENT (CP::TreeMakerAlg)
+DECLARE_COMPONENT (CP::IOStatsAlg)
 DECLARE_COMPONENT (CP::SystObjectLinkerAlg)
 // Concrete classes of SystObjectUnioniserAlg
 DECLARE_COMPONENT (CP::SystJetUnioniserAlg)

PhysicsAnalysis/Algorithms/ComponentFactoryPreloader/Root/ComponentFactoryPreloader.cxx

@@ -38,6 +38,7 @@
 #include <AsgAnalysisAlgorithms/SystObjectLinkerAlg.h>
 #include <AsgAnalysisAlgorithms/TreeFillerAlg.h>
 #include <AsgAnalysisAlgorithms/TreeMakerAlg.h>
+#include <AsgAnalysisAlgorithms/IOStatsAlg.h>
 #include <EgammaAnalysisAlgorithms/EgammaCalibrationAndSmearingAlg.h>
 #include <EgammaAnalysisAlgorithms/EgammaIsGoodOQSelectionTool.h>
 #include <EgammaAnalysisAlgorithms/EgammaIsolationCorrectionAlg.h>
@@ -151,6 +152,7 @@ namespace CP
     ANA_CHECK (asg::registerAlgorithmFactory<CP::TrigEventSelectionAlg>("CP::TrigEventSelectionAlg"));
     ANA_CHECK (asg::registerAlgorithmFactory<CP::TrigGlobalEfficiencyAlg>("CP::TrigGlobalEfficiencyAlg"));
     ANA_CHECK (asg::registerAlgorithmFactory<CP::VertexSelectionAlg>("CP::VertexSelectionAlg"));
+    ANA_CHECK (asg::registerAlgorithmFactory<CP::IOStatsAlg>("CP::IOStatsAlg"));
 
     ANA_CHECK (asg::registerToolFactory<AsgDeadHVCellRemovalTool> ("AsgDeadHVCellRemovalTool"));
     ANA_CHECK (asg::registerToolFactory<BTaggingSelectionTool> ("BTaggingSelectionTool"));