python.InDetCommonConfig Namespace Reference

Functions
	InDetCommonCfg (flags, **kwargs)

Function Documentation

InDetCommonCfg()

python.InDetCommonConfig.InDetCommonCfg	(		flags,
		**	kwargs )

Main config for common ID track decorations

Definition at line 16 of file InDetCommonConfig.py.

def InDetCommonCfg(flags, **kwargs):
    """Main config for common ID track decorations"""
 
    acc = ComponentAccumulator()
    CommonAugmentation = CompFactory.DerivationFramework.CommonAugmentation
 
    have_PV_container = (
        flags.Beam.Type is not BeamType.Cosmics and
        (any('PrimaryVertices' in elements for elements
             in flags.Input.TypedCollections)))
    if not have_PV_container and kwargs['DoVertexFinding']:
        have_PV_container = (
            any('PixelRDOs' in elements for elements
                in flags.Input.TypedCollections) or
            any('SCT_RDOs' in elements for elements in
                flags.Input.TypedCollections))
 
    if have_PV_container:
        # ====================================================================
        # LABELLING TRACKS WITH OUTCOME OF SELECTOR TOOL
        # ====================================================================
        from DerivationFrameworkInDet.InDetToolsConfig import (
            InDetTrackSelectionToolWrapperCfg)
        DFCommonTrackSelection = acc.getPrimaryAndMerge(
            InDetTrackSelectionToolWrapperCfg(
                flags,
                name           = "DFCommonTrackSelection",
                CutLevel  = "TightPrimary",
                DecorationName = "DFCommonTightPrimary"))
 
        if kwargs['AddPseudoTracks']:
            from DerivationFrameworkInDet.InDetToolsConfig import (
                PseudoTrackSelectorCfg)
            PseudoTrackSelectorTool = acc.getPrimaryAndMerge(
                PseudoTrackSelectorCfg(
                    flags,
                    name                                 = "PseudoTrackSelectorTool",
                    RecoTrackParticleLocation            = "InDetTrackParticles",
                    PseudoTrackParticleLocation          = "InDetPseudoTrackParticles",
                    OutputRecoReplacedWithPseudo         = "InDetReplacedWithPseudoTrackParticles",
                    OutputRecoReplacedWithPseudoFromB    = "InDetReplacedWithPseudoFromBTrackParticles",
                    OutputRecoReplacedWithPseudoNotFromB = "InDetReplacedWithPseudoNotFromBTrackParticles",
                    OutputRecoPlusPseudo                 = "InDetPlusPseudoTrackParticles",
                    OutputRecoPlusPseudoFromB            = "InDetPlusPseudoFromBTrackParticles",
                    OutputRecoPlusPseudoNotFromB         = "InDetPlusPseudoNotFromBTrackParticles",
                    OutputRecoNoFakes                    = "InDetNoFakesTrackParticles",
                    OutputRecoNoFakesFromB               = "InDetNoFakesFromBTrackParticles",
                    OutputRecoNoFakesNotFromB            = "InDetNoFakesNotFromBTrackParticles"))
 
            acc.addEventAlgo(CommonAugmentation(
                "InDetSelectedPseudo",
                AugmentationTools=[PseudoTrackSelectorTool]))
 
        # ====================================================================
        # EXPRESSION OF Z0 AT THE PRIMARY VERTEX
        # ====================================================================
        from DerivationFrameworkInDet.InDetToolsConfig import (
            TrackParametersAtPVCfg)
        DFCommonZ0AtPV = acc.getPrimaryAndMerge(TrackParametersAtPVCfg(
            flags,
            name                       = "DFCommonZ0AtPV",
            TrackParticleContainerName = "InDetTrackParticles",
            VertexContainerName        = "PrimaryVertices",
            Z0SGEntryName              = "DFCommonInDetTrackZ0AtPV"))
 
        # ====================================================================
        # DECORATE THE HARDSCATTER VERTEX WITH A FLAG
        # ====================================================================
        from InDetConfig.InDetHardScatterSelectionToolConfig import (
            InDetHardScatterSelectionToolCfg)
        DFCommonHSSelectionTool = acc.popToolsAndMerge(
            InDetHardScatterSelectionToolCfg(flags,
                                             name="DFCommonHSSelectionTool"))
        acc.addPublicTool(DFCommonHSSelectionTool)
 
        from DerivationFrameworkInDet.InDetToolsConfig import HardScatterVertexDecoratorCfg
        acc.merge(HardScatterVertexDecoratorCfg(flags))
 
        # ====================================================================
        # DECORATE THE TRACKS WITH USED-IN-FIT TTVA VARIABLES
        # ====================================================================
        from InDetConfig.InDetUsedInFitTrackDecoratorToolConfig import (
            InDetUsedInFitTrackDecoratorToolCfg)
        DFCommonUsedInFitDecoratorTool = acc.popToolsAndMerge(
            InDetUsedInFitTrackDecoratorToolCfg(
                flags,
                name                 = "DFCommonUsedInFitDecoratorTool",
                AMVFVerticesDecoName = "TTVA_AMVFVertices",
                AMVFWeightsDecoName  = "TTVA_AMVFWeights",
                TrackContainer       = "InDetTrackParticles",
                VertexContainer      = "PrimaryVertices"))
        acc.addPublicTool(DFCommonUsedInFitDecoratorTool)
 
        from DerivationFrameworkInDet.InDetToolsConfig import (
            UsedInVertexFitTrackDecoratorCfg)
        DFCommonUsedInFitDecorator = acc.getPrimaryAndMerge(
            UsedInVertexFitTrackDecoratorCfg(
                flags,
                name="DFCommonUsedInFitDecorator",
                UsedInFitDecoratorTool=DFCommonUsedInFitDecoratorTool))
 
        if kwargs['AddPseudoTracks']:
            PseudoTrackContainers = [
                "InDetPseudoTrackParticles",
                "InDetReplacedWithPseudoTrackParticles",
                "InDetReplacedWithPseudoFromBTrackParticles",
                "InDetReplacedWithPseudoNotFromBTrackParticles",
                "InDetPlusPseudoTrackParticles",
                "InDetPlusPseudoFromBTrackParticles",
                "InDetPlusPseudoNotFromBTrackParticles",
                "InDetNoFakesTrackParticles",
                "InDetNoFakesFromBTrackParticles",
                "InDetNoFakesNotFromBTrackParticles"
            ]
            PseudoTrackDecorators = []
            for t in PseudoTrackContainers:
                InDetDecorator = acc.popToolsAndMerge(
                    InDetUsedInFitTrackDecoratorToolCfg(
                        flags,
                        name="DFCommonUsedInFitDecoratorTool" +
                        t.replace('InDetPseudo', 'Pseudo').replace(
                            'InDet', 'Reco').replace('TrackParticles', ''),
                        AMVFVerticesDecoName="TTVA_AMVFVertices",
                        AMVFWeightsDecoName="TTVA_AMVFWeights",
                        TrackContainer=t,
                        VertexContainer="PrimaryVertices"))
                DerivDecorator = acc.getPrimaryAndMerge(
                    UsedInVertexFitTrackDecoratorCfg(
                        flags,
                        name="DFCommonUsedInFitDecorator" +
                        t.replace('InDetPseudo', 'Pseudo').replace(
                            'InDet', 'Reco').replace('TrackParticles', ''),
                        UsedInFitDecoratorTool=InDetDecorator))
                PseudoTrackDecorators.append(DerivDecorator)
 
        # Turned off by defult for the LRT tracks as they are not used. Added this option to turn it on for future studies.
        if (kwargs['DecoLRTTTVA'] and kwargs['DoR3LargeD0'] and
                kwargs['StoreSeparateLargeD0Container']):
 
            # ====================================================================
            # DECORATE THE LRT TRACKS WITH USED-IN-FIT TTVA VARIABLES
            # ====================================================================
            DFCommonUsedInFitDecoratorToolLRT = acc.popToolsAndMerge(
                InDetUsedInFitTrackDecoratorToolCfg(
                    flags,
                    name                 = "DFCommonUsedInFitDecoratorToolLRT",
                    AMVFVerticesDecoName = "TTVA_AMVFVertices",
                    AMVFWeightsDecoName  = "TTVA_AMVFWeights",
                    TrackContainer       = "InDetLargeD0TrackParticles",
                    VertexContainer      = "PrimaryVertices"))
 
            DFCommonUsedInFitDecoratorLRT = acc.getPrimaryAndMerge(
                UsedInVertexFitTrackDecoratorCfg(
                    flags,
                    name="DFCommonUsedInFitDecoratorLRT",
                    UsedInFitDecoratorTool=DFCommonUsedInFitDecoratorToolLRT))
            # =======================================
            # CREATE THE DERIVATION KERNEL ALGORITHM
            # =======================================
            for tool in [DFCommonTrackSelection,
                                   DFCommonZ0AtPV,
                                   DFCommonUsedInFitDecorator,
                                   DFCommonUsedInFitDecoratorLRT]:
                acc.addEventAlgo(CommonAugmentation("InDetCommonKernel"+tool.name,
                                 AugmentationTools=[tool]))
        else:
            AugTools = [DFCommonTrackSelection,
                        DFCommonZ0AtPV,
                        DFCommonUsedInFitDecorator]
            if kwargs['AddPseudoTracks']:
                AugTools += PseudoTrackDecorators
            for tool in AugTools:
                acc.addEventAlgo(CommonAugmentation("InDetCommonKernel"+tool.name,
                                                AugmentationTools=[tool]))
 
    # Add LRT merger job to the sequence when the LRT track particle is supposed to be made already
    if (kwargs['MergeLRT'] and kwargs['DoR3LargeD0'] and
        kwargs['StoreSeparateLargeD0Container']):
        # ====================================================================
        # Merge the LRT and standard track using track particle merger
        # ====================================================================
        from DerivationFrameworkInDet.InDetToolsConfig import InDetLRTMergeCfg
        acc.merge(InDetLRTMergeCfg(flags))
 
    return acc