ActsHGTD_single_pad_clusterization_test.py

Go to the documentation of this file.

#!/usr/bin/env python
# Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
import argparse
 
if __name__ == "__main__":
 
    parser = argparse.ArgumentParser(description="Run reconstruction script with specified configuration.")
    parser.add_argument("--mode", choices=["legacy", "xAOD"], required=True, help="Specify the reconstruction mode.")
    args = parser.parse_args()
 
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
 
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags.Input.Files = defaultTestFiles.RDO_RUN4
    flags.Exec.MaxEvents = -1
 
    output_suffix = "_legacy" if args.mode == "legacy" else "_xAOD"
    flags.Output.HISTFileName = f"ActsMonitoringOutput{output_suffix}.root"
    flags.DQ.useTrigger = False
 
    flags.Acts.doMonitoring = True
    flags.HGTD.doMonitoring = True
 
    flags.lock()
    flags.dump()
 
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    acc = MainServicesCfg(flags)
 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    acc.merge(PoolReadCfg(flags))
 
    if args.mode == "legacy":
        from HGTD_Config.HGTD_PrepRawDataFormationConfig import PadClusterizationCfg
        acc.merge(PadClusterizationCfg(flags))
 
        # Add edm converter
        from InDetConfig.InDetPrepRawDataFormationConfig import HGTDInDetToXAODClusterConversionCfg
        acc.merge(HGTDInDetToXAODClusterConversionCfg(flags))
 
    elif args.mode == "xAOD":
        from ActsConfig.ActsClusterizationConfig import ActsHgtdClusterizationAlgCfg
        acc.merge(ActsHgtdClusterizationAlgCfg(flags))
 
    else :
        raise ValueError("Invalid mode selected. Please select either 'legacy' or 'xAOD'.")
 
    from ActsConfig.ActsAnalysisConfig import ActsHgtdClusterAnalysisAlgCfg
    acc.merge(ActsHgtdClusterAnalysisAlgCfg(flags))
 
    acc.printConfig(withDetails = True, summariseProps = True)
    acc.run()