def AppendOutputList(HIAODItemList=[]): """Adds HIJetOutputList to the list passed in as an argument""" if HIJetFlags.WriteClusters(): AddToOutputList(HIJetFlags.HIClusterKey(), "xAOD::CaloClusterContainer") from RecExConfig.RecFlags import rec if not rec.doESD(): for R in HIJetFlags.AntiKtRValues(): AddToOutputList("AntiKt%dHIJets" % int(10 * R)) if jetFlags.useTruth(): AddToOutputList("AntiKt%dTruthJets" % int(10 * R)) AddToOutputList(HIJetFlags.TrackJetContainerName()) HIAODItemList += HIJetFlags.HIJetOutputList()
def HIClusterGetter(tower_key="CombinedTower", cell_key="AllCalo", cluster_key=""): """Function to equip HI cluster builder from towers and cells, adds to output AOD stream""" if cluster_key == "": cluster_key = HIJetFlags.HIClusterKey() HIClusterMaker = CompFactory.HIClusterMaker theAlg = HIClusterMaker() theAlg.InputTowerKey = tower_key theAlg.CaloCellContainerKey = cell_key theAlg.OutputContainerKey = cluster_key from AthenaCommon.AlgSequence import AlgSequence topSequence = AlgSequence() topSequence += theAlg if HIJetFlags.WriteClusters(): AddToOutputList(HIJetFlags.HIClusterKey(), "xAOD::CaloClusterContainer") return theAlg