def configure(self): mlog = logging.getLogger('egammaTrackSlimmer.py::configure:') mlog.info('entering') try: from egammaRec.egammaRecConf import egammaTrackSlimmer theEgammaTrackSlimmer = egammaTrackSlimmer( "egammaTrackSlimmer", InputElectronContainerName=egammaKeys.outputElectronKey(), InputPhotonContainerName=egammaKeys.outputPhotonKey(), TrackParticleContainerName=egammaKeys.outputTrackParticleKey(), VertexContainerName=egammaKeys.outputConversionKey(), doThinning=True) print theEgammaTrackSlimmer except Exception: mlog.error("could not get handle to egammaTrackSlimmer") print traceback.format_exc() return False # add to topsequence mlog.info("now adding to topSequence") from AthenaCommon.AlgSequence import AlgSequence topSequence = AlgSequence() topSequence += theEgammaTrackSlimmer return True
def getGainDecorations(obj, collections=[ egammaKeys.outputElectronKey(), egammaKeys.outputPhotonKey() ], info=["E", "nCells"]): """getGainDecorations( obj, collections=["Electrons", "Photons"] ) -> Return a list with the 'ExtraContent' to be added to the decorations to save the gain information per layer""" return ["{part}.{info}".format(part=part, info=info) for part in collections \ for info in getGainLayerNames(obj) ]
def configure(self): mlog = logging.getLogger ('egammaGetter.py::configure:') mlog.info('entering') if doSuperclusters() : egammaBuilder = AlgFactory(egammaRecConf.topoEgammaBuilder, name = 'egamma', # Keys ElectronOutputName = egammaKeys.outputElectronKey(), PhotonOutputName = egammaKeys.outputPhotonKey(), #Super Cluster Tools TopoClusterCopier = egammaTopoClusterCopier, electronSuperClusterBuilder = electronSuperClusterBuilder, photonSuperClusterBuilder = photonSuperClusterBuilder, InputTopoClusterContainerName = egammaRecFlags.egammaTopoClusterCollection(), # Builder tools BremCollectionBuilderTool = EMBremCollectionBuilder, TrackMatchBuilderTool = EMTrackMatchBuilder, VertexBuilder = EMVertexBuilder if doConversions() else None, ConversionBuilderTool = EMConversionBuilder if doConversions() else None, AmbiguityTool = EGammaAmbiguityTool, # Decoration tools egammaTools = FcnWrapper(egammaDecorationTools), ElectronTools = FcnWrapper(electronDecorationTools), PhotonTools = FcnWrapper(photonDecorationTools), # Flags and other properties # Track matching depending if ID is on/off doTrackMatching = DetFlags.detdescr.ID_on(), # conversions building/matching depending if ID is on/off doVertexCollection=doConversions(), doConversions = doConversions() ) else : egammaBuilder = AlgFactory(egammaRecConf.egammaBuilder, name = 'egamma', # Keys ElectronOutputName = egammaKeys.outputElectronKey(), PhotonOutputName = egammaKeys.outputPhotonKey(), TopoSeededClusterContainerName = getTopoSeededCollectionName(), # Builder tools BremCollectionBuilderTool = EMBremCollectionBuilder, TrackMatchBuilderTool = EMTrackMatchBuilder, VertexBuilder = EMVertexBuilder if doConversions() else None, ConversionBuilderTool = EMConversionBuilder if doConversions() else None, AmbiguityTool = EGammaAmbiguityTool, # Decoration tools egammaTools = FcnWrapper(egammaDecorationTools), ElectronTools = FcnWrapper(electronDecorationTools), PhotonTools = FcnWrapper(photonDecorationTools), # Flags and other properties # Track matching depending if ID is on/off doTrackMatching = DetFlags.detdescr.ID_on(), # conversions building/matching depending if ID is on/off doVertexCollection=doConversions(), doConversions = doConversions(), # clusterEnergyCut = 10*MeV, doTopoSeededPhotons = egammaRecFlags.doTopoCaloSeeded() ) # configure egamma here: try: self._egammaBuilderHandle = egammaBuilder() except Exception: mlog.error("could not get handle to egamma") import traceback print traceback.format_exc() return False print self._egammaBuilderHandle return True
egammaSwTool = ToolFactory(egammaToolsConf.egammaSwTool, postInit=[configureClusterCorrections]) from egammaMVACalib import egammaMVACalibConf egammaMVATool = ToolFactory( egammaMVACalibConf.egammaMVATool, folder=jobproperties.egammaRecFlags.calibMVAVersion()) EMClusterTool = ToolFactory( egammaToolsConf.EMClusterTool, OutputClusterContainerName=egammaKeys.outputClusterKey(), OutputTopoSeededClusterContainerName=egammaKeys.outputTopoSeededClusterKey( ), ElectronContainerName=egammaKeys.outputElectronKey(), PhotonContainerName=egammaKeys.outputPhotonKey(), ClusterCorrectionToolName=FullNameWrapper(egammaSwTool), doSuperCluster=jobproperties.egammaRecFlags.doSuperclusters(), MVACalibTool=egammaMVATool) egammaCheckEnergyDepositTool = ToolFactory( egammaToolsConf.egammaCheckEnergyDepositTool, UseThrFmax=True, ThrF0max=0.9, ThrF1max=0.8, ThrF2max=0.98, ThrF3max=0.8) from EMBremCollectionBuilder import egammaBremCollectionBuilder from egammaTrackTools.egammaTrackToolsFactories import EMExtrapolationTools EMBremCollectionBuilder = ToolFactory(
# from DerivationFrameworkCalo.DerivationFrameworkCaloFactories import GainDecorator, getExtraContent # GainDecoratorTool = GainDecorator() # MySlimmingHelper.ExtraVariables.extend( getGainDecorations(GainDecoratorTool) ) # Currently it adds: # ['Electrons.E_Lr0_HiG', 'Electrons.E_Lr0_MedG', 'Electrons.E_Lr0_LowG', 'Electrons.E_Lr1_HiG', 'Electrons.E_Lr1_MedG', 'Electrons.E_Lr1_LowG', 'Electrons.E_Lr2_HiG', 'Electrons.E_Lr2_MedG', 'Electrons.E_Lr2_LowG', 'Electrons.E_Lr3_HiG', 'Electrons.E_Lr3_MedG', 'Electrons.E_Lr3_LowG', 'Electrons.nCells_Lr0_HiG', 'Electrons.nCells_Lr0_MedG', 'Electrons.nCells_Lr0_LowG', 'Electrons.nCells_Lr1_HiG', 'Electrons.nCells_Lr1_MedG', 'Electrons.nCells_Lr1_LowG', 'Electrons.nCells_Lr2_HiG', 'Electrons.nCells_Lr2_MedG', 'Electrons.nCells_Lr2_LowG', 'Electrons.nCells_Lr3_HiG', 'Electrons.nCells_Lr3_MedG', 'Electrons.nCells_Lr3_LowG', 'Photons.E_Lr0_HiG', 'Photons.E_Lr0_MedG', 'Photons.E_Lr0_LowG', 'Photons.E_Lr1_HiG', 'Photons.E_Lr1_MedG', 'Photons.E_Lr1_LowG', 'Photons.E_Lr2_HiG', 'Photons.E_Lr2_MedG', 'Photons.E_Lr2_LowG', 'Photons.E_Lr3_HiG', 'Photons.E_Lr3_MedG', 'Photons.E_Lr3_LowG', 'Photons.nCells_Lr0_HiG', 'Photons.nCells_Lr0_MedG', 'Photons.nCells_Lr0_LowG', 'Photons.nCells_Lr1_HiG', 'Photons.nCells_Lr1_MedG', 'Photons.nCells_Lr1_LowG', 'Photons.nCells_Lr2_HiG', 'Photons.nCells_Lr2_MedG', 'Photons.nCells_Lr2_LowG', 'Photons.nCells_Lr3_HiG', 'Photons.nCells_Lr3_MedG', 'Photons.nCells_Lr3_LowG'] from egammaRec.Factories import ToolFactory, getPropertyValue from egammaRec import egammaKeys from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__GainDecorator, DerivationFramework__ClusterEnergyPerLayerDecorator from CaloClusterCorrection.CaloClusterCorrectionConf import CaloFillRectangularCluster as CFRC from GaudiKernel.Constants import DEBUG GainDecorator = ToolFactory(DerivationFramework__GainDecorator, SGKey_electrons=egammaKeys.outputElectronKey(), SGKey_photons=egammaKeys.outputPhotonKey(), OutputLevel=DEBUG) # Define defaults for CaloFillRectangularCluster and ClusterEnergyPerLayerDecorator # will be overridden later depending on the number of cells neta X nphi CaloFillRectangularCluster = ToolFactory(CFRC, cells_name=egammaKeys.caloCellKey(), fill_cluster=True) ClusterEnergyPerLayerDecorator = ToolFactory( DerivationFramework__ClusterEnergyPerLayerDecorator, CaloFillRectangularTool=CaloFillRectangularCluster, SGKey_electrons=egammaKeys.outputElectronKey(), SGKey_photons=egammaKeys.outputPhotonKey(), SGKey_caloCells=egammaKeys.caloCellKey())
# Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration __doc__ = """ToolFactory to instantiate the two supercluster builders with default configuration""" __author__ = "Jovan Mitrevski" from egammaAlgs import egammaAlgsConf from egammaRec.Factories import AlgFactory from egammaRec import egammaKeys from egammaTools.egammaToolsFactories import EGammaAmbiguityTool topoEgammaBuilder = AlgFactory( egammaAlgsConf.topoEgammaBuilder, name='topoEgammaBuilder', SuperElectronRecCollectionName=egammaKeys.ElectronSuperRecKey(), SuperPhotonRecCollectionName=egammaKeys.PhotonSuperRecKey(), ElectronOutputName=egammaKeys.outputElectronKey(), PhotonOutputName=egammaKeys.outputPhotonKey(), AmbiguityTool=EGammaAmbiguityTool)