TCCAccumulator.py

Go to the documentation of this file.

#Matt A's custom python accumulator for R22 
# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from pprint import pprint
 
    
 
def MyAlgConfigNeutral(inputFlags,**kwargs,):
    # this is based on the PFRun3Config, so parts of it are gonna be used as is
    result=ComponentAccumulator()
 
    StoreGateSvc=CompFactory.StoreGateSvc
 
    result.addService(StoreGateSvc("DetectorStore"))
    
    #need to setup the track to vertex association tool (TTVA). 
    # seems you can do this from the AOD
    #    from TrackVertexAssociationTool.TTVAToolConfig import usedInFitDecoratorCfg
    #    result.addEventAlgo(usedInFitDecoratorCfg(cfgFlags, "TTVA"))
 
 
    from TrackCaloClusterRecTools.TrackCaloClusterConfig import runTCCReconstruction
    TCCReco=runTCCReconstruction(inputFlags, outputTCCName="TrackCaloClusters",doNeutral=True,doCharged=False,doCombined=False)
    result.merge(TCCReco)
 
    StoreGateSvc.Dump = True
    return result
 
 
 
 
def MyAlgConfigCharged(inputFlags,**kwargs,):
    # this is based on the PFRun3Config, so parts of it are gonna be used as is
    result=ComponentAccumulator()
 
    StoreGateSvc=CompFactory.StoreGateSvc
 
    result.addService(StoreGateSvc("DetectorStore"))
    
    #need to setup the track to vertex association tool (TTVA). 
    # seems you can do this from the AOD
    #    from TrackVertexAssociationTool.TTVAToolConfig import usedInFitDecoratorCfg
    #    result.addEventAlgo(usedInFitDecoratorCfg(cfgFlags, "TTVA"))
 
 
    from TrackCaloClusterRecTools.TrackCaloClusterConfig import runTCCReconstruction
    TCCReco=runTCCReconstruction(inputFlags, outputTCCName="TrackCaloClusters",doNeutral=False,doCharged=True,doCombined=False)
    result.merge(TCCReco)
    
    StoreGateSvc.Dump = True
    return result
 
 
def LinkerAlgConfig(inputFlags,**kwargs):
    CA=ComponentAccumulator()
    StoreGateSvc=CompFactory.StoreGateSvc
    CA.addService(StoreGateSvc("DetectorStore"))
 
    from eflowRec.PFCfg import getEGamFlowElementAssocAlgorithm
    from eflowRec.PFCfg import getMuonFlowElementAssocAlgorithm
    
    CA.addEventAlgo(getEGamFlowElementAssocAlgorithm(inputFlags,AODTest=False,doTCC=True))
    CA.addEventAlgo(getMuonFlowElementAssocAlgorithm(inputFlags,LinkNeutralFEClusters=False,useMuonTopoClusters=False,AODTest=False,doTCC=True))    
    CA.printConfig()
    #print("IO")
    #    iolist=result.getIO()
    #    for param in iolist:
    #        key=param["key"]
    #        prop=param["prop"]
    #        print(key,prop)
 
    return CA
 
 
def TauLinkerConfig(inputFlags, **kwargs):
    #separate linker, can merge to main one for no loss of generality
    TauLinkerCA=ComponentAccumulator()
    # add Storegate
    StoreGateSvc=CompFactory.StoreGateSvc
    TauLinkerCA.addService(StoreGateSvc("DetectorStore"))
    
    from eflowRec.PFCfg import getTauFlowElementAssocAlgorithm
    TauLinkerCA.addEventAlgo(getTauFlowElementAssocAlgorithm(inputFlags,doTCC=True))
    return TauLinkerCA
 
 
 
if __name__=="__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    cfgFlags = initConfigFlags()
    cfgFlags.Input.isMC=True
    cfgFlags.Input.Format="AOD"
    #    cfgFlags.Input.Files=["/scratch/anthony/TEST_AOD/TCC/mc16_13TeV.410470.PhPy8EG_A14_ttbar_hdamp258p75_nonallhad.recon.AOD.e6337_s3126_r10724/mc16_13TeV/AOD.14795494._007454.pool.root.1"]
    #    cfgFlags.Input.Files=["/scratch/anthony/TEST_AOD/TCC/valid1_test/valid1/AOD.24855256._000001.pool.root.1"]
    cfgFlags.Input.Files=["/scratch/anthony/GPF_CODE//TCC_STUDIES/CustomAlg/myAOD.root"]
    cfgFlags.Output.AODFileName="/scratch/anthony/GPF_CODE/TCC_STUDIES/CustomAlg/output_AOD.root"
    cfgFlags.Exec.MaxEvents=20
    cfgFlags.Output.doWriteAOD=True
    cfgFlags.Common.ProductionStep=""
    cfgFlags.dump()
    cfgFlags.lock()
    
 
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    cfg=MainServicesCfg(cfgFlags)
 
    from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
    inputList=["xAOD::TrackParticleContainer#InDetTrackParticles", "xAOD::TrackParticleAuxContainer#InDetTrackParticlesAux."]
    inputList.append("xAOD::FlowElementContainer#JetETMissChargedParticleFlowObjects")
    inputList.append("xAOD::FlowElementContainer#JetETMissNeutralParticleFlowObjects")
    inputList.append("xAOD::FlowElementAuxContainer#JetETMissNeutralParticleFlowObjectsAux.")
    inputList.append("xAOD::FlowElementAuxContainer#JetETMissChargedParticleFlowObjectsAux.")
    inputList.append("xAOD::FlowElementContainer#TrackCaloClustersCharged")
    inputList.append("xAOD::FlowElementAuxContainer#TrackCaloClustersChargedAux.")
    inputList.append("xAOD::FlowElementContainer#TrackCaloClustersNeutral")
    inputList.append("xAOD::FlowElementAuxContainer#TrackCaloClustersNeutralAux.")
 
    inputList.append("xAOD::MuonContainer#Muons")
    inputList.append("xAOD::MuonAuxContainer#*")
    inputList.append("xAOD::PhotonContainer#Photons")
    inputList.append("xAOD::PhotonAuxContainer#*")
    inputList.append("xAOD::ElectronContainer#Electrons")
    inputList.append("xAOD::ElectronAuxContainer#*")
    inputList.append("xAOD::TauJetContainer#*")
    inputList.append("xAOD::TauJetAuxContainer#*")
    
    
    cfg.merge(OutputStreamCfg(cfgFlags,"AOD",ItemList=inputList))
    
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
 
    
    cfg.merge(PoolReadCfg(cfgFlags))
    cfg.merge(MyAlgConfigNeutral(cfgFlags))
    cfg.merge(MyAlgConfigCharged(cfgFlags))
    cfg.merge(LinkerAlgConfig(cfgFlags)) # El/Phot/Muon linkers
    #cfg.merge(TauLinkerConfig(cfgFlags)) # Tau linkers
    
    cfg.printConfig() # print full CA
 
    cfg.run()