python.InDetGNNHardScatterSelectionConfig Namespace Reference

Functions
def	AsgPtEtaSelectionToolCfg (flags, name="AsgPtEtaSelectionTool", **kwargs)
def	AsgPtEtaSelectionToolGapCfg (flags, name="AsgPtEtaSelectionTool", **kwargs)
def	AsgViewFromSelectionAlgCfg (flags, name="AsgViewFromSelectionAlg", **kwargs)
def	GNNHSSelectionAlgCfg (flags, input, minPt)
def	GNNHSOverlapRemovalToolCfg (flags, name="GNNHS_OverlapRemovalToolCfg", **kwargs)
def	GNNHSOverlapRemovalAlgCfg (flags, name="GNNHS_OverlapRemovalAlg", overlapInputNames=None, overlapOutputNames=None, **kwargs)
def	GNNToolCfg (flags, name="HardScatterSelectionGNNTool", **kwargs)
def	GNNHSVertexDecoratorAlgCfg (flags, name="GNNHS_VertexDecoratorAlg", **kwargs)
def	GNNSequenceCfg (flags, **kwargs)

Variables
	flags = initConfigFlags()
	Files
	MaxEvents
	top_acc = MainServicesCfg(flags)
	OutputLevel
	withDetails
	True
	summariseProps
	sc = top_acc.run(1)

Function Documentation

AsgPtEtaSelectionToolCfg()

def python.InDetGNNHardScatterSelectionConfig.AsgPtEtaSelectionToolCfg	(		flags,
			name = "AsgPtEtaSelectionTool",
		**	kwargs
	)

Definition at line 11 of file InDetGNNHardScatterSelectionConfig.py.

def AsgPtEtaSelectionToolCfg(flags, name="AsgPtEtaSelectionTool", **kwargs):
    cfg = ComponentAccumulator()
    kwargs.setdefault("maxEta", 2.5)
    cfg.setPrivateTools(CompFactory.CP.AsgPtEtaSelectionTool(name, **kwargs))
    return cfg
 

AsgPtEtaSelectionToolGapCfg()

def python.InDetGNNHardScatterSelectionConfig.AsgPtEtaSelectionToolGapCfg	(		flags,
			name = "AsgPtEtaSelectionTool",
		**	kwargs
	)

Definition at line 17 of file InDetGNNHardScatterSelectionConfig.py.

def AsgPtEtaSelectionToolGapCfg(flags, name="AsgPtEtaSelectionTool", **kwargs):
    kwargs.setdefault("etaGapLow", 1.37)
    kwargs.setdefault("etaGapHigh", 1.52)
    return AsgPtEtaSelectionToolCfg(flags, name, **kwargs)
 

AsgViewFromSelectionAlgCfg()

def python.InDetGNNHardScatterSelectionConfig.AsgViewFromSelectionAlgCfg	(		flags,
			name = "AsgViewFromSelectionAlg",
		**	kwargs
	)

Definition at line 22 of file InDetGNNHardScatterSelectionConfig.py.

def AsgViewFromSelectionAlgCfg(flags, name="AsgViewFromSelectionAlg", **kwargs):
    cfg = ComponentAccumulator()
    kwargs.setdefault("selection", "selectPtEta")
    kwargs.setdefault("deepCopy", False)
    cfg.addEventAlgo(CompFactory.CP.AsgViewFromSelectionAlg(name, **kwargs))
    return cfg
 

GNNHSOverlapRemovalAlgCfg()

def python.InDetGNNHardScatterSelectionConfig.GNNHSOverlapRemovalAlgCfg	(		flags,
			name = "GNNHS_OverlapRemovalAlg",
			overlapInputNames = None,
			overlapOutputNames = None,
		**	kwargs
	)

Definition at line 81 of file InDetGNNHardScatterSelectionConfig.py.

def GNNHSOverlapRemovalAlgCfg(flags, name="GNNHS_OverlapRemovalAlg",
                              overlapInputNames = None, overlapOutputNames = None, **kwargs):
    cfg = ComponentAccumulator()
 
    kwargs.setdefault("OutputLabel", "passesOR")
    kwargs.setdefault("affectingSystematicsFilter", ".*")
 
    for obj in overlapInputNames:
        kwargs.setdefault(obj, overlapInputNames[obj])
        kwargs.setdefault(obj+"Decoration", kwargs["OutputLabel"] + ",as_char")
 
    kwargs.setdefault("overlapTool", cfg.popToolsAndMerge(GNNHSOverlapRemovalToolCfg(flags)))
    
    cfg.addEventAlgo(CompFactory.CP.OverlapRemovalAlg(name, **kwargs))
 
    for obj in overlapInputNames:
        cfg.addEventAlgo(CompFactory.CP.AsgViewFromSelectionAlg(
            name = "GNNHS_"+obj+"_ORSelectionAlg",
            input = overlapInputNames[obj],
            output = overlapOutputNames[obj],
            selection = [kwargs["OutputLabel"]+",as_char"],
            deepCopy = True))
    
    return cfg
 
 
# GNNTool + VertexDecoratorAlg configs
 

GNNHSOverlapRemovalToolCfg()

def python.InDetGNNHardScatterSelectionConfig.GNNHSOverlapRemovalToolCfg	(		flags,
			name = "GNNHS_OverlapRemovalToolCfg",
		**	kwargs
	)

Definition at line 49 of file InDetGNNHardScatterSelectionConfig.py.

def GNNHSOverlapRemovalToolCfg(flags, name="GNNHS_OverlapRemovalToolCfg", **kwargs):
    cfg = ComponentAccumulator()
 
    kwargs.setdefault("InputLabel", "selectPtEta")
    kwargs.setdefault("OutputLabel", "passesOR")
    kwargs.setdefault("OutputPassValue", True)
 
    subtool_kwargs={}
    for prop in ["InputLabel", "OutputLabel", "OutputPassValue"]:
        subtool_kwargs[prop] = kwargs[prop]
 
    kwargs.setdefault("EleEleORT", CompFactory.ORUtils.EleEleOverlapTool(**subtool_kwargs))
    kwargs.setdefault("EleMuORT", CompFactory.ORUtils.EleMuSharedTrkOverlapTool(**subtool_kwargs))
    kwargs.setdefault("EleJetORT", CompFactory.ORUtils.EleJetOverlapTool(**subtool_kwargs))
    kwargs.setdefault(
        "MuJetORT",
        CompFactory.ORUtils.MuJetOverlapTool(
            PVContainerName=(
                "PrimaryVertices_initial" 
                if flags.Tracking.PriVertex.sortingSetup is VertexSortingSetup.GNNSorting 
                else "PrimaryVertices"
            ),
            **subtool_kwargs
        )
    )
    kwargs.setdefault("PhoEleORT", CompFactory.ORUtils.DeltaROverlapTool(**subtool_kwargs))
    kwargs.setdefault("PhoMuORT", CompFactory.ORUtils.DeltaROverlapTool(**subtool_kwargs))
    kwargs.setdefault("PhoJetORT", CompFactory.ORUtils.DeltaROverlapTool(**subtool_kwargs))
 
    cfg.setPrivateTools(CompFactory.ORUtils.OverlapRemovalTool(name, **kwargs))
    return cfg
 

GNNHSSelectionAlgCfg()

def python.InDetGNNHardScatterSelectionConfig.GNNHSSelectionAlgCfg	(		flags,
			input,
			minPt
	)

Definition at line 29 of file InDetGNNHardScatterSelectionConfig.py.

def GNNHSSelectionAlgCfg(flags, input, minPt):
    cfg = ComponentAccumulator()
 
    selectionTool = None
    if input in ["Electrons", "Photons", "AntiKt4EMTopoJets"]:
        selectionTool = cfg.popToolsAndMerge(
            AsgPtEtaSelectionToolGapCfg(flags, minPt = minPt))
    elif input in ["Muons"]:
        selectionTool = cfg.popToolsAndMerge(
            AsgPtEtaSelectionToolCfg(flags, minPt = minPt))
    
    cfg.addEventAlgo(CompFactory.CP.AsgSelectionAlg(
        name = "GNNHS_"+input+"_SelectionAlg",
        selectionTool = selectionTool,
        selectionDecoration = "selectPtEta,as_char",
        particles = input))
 
    return cfg
 
 

GNNHSVertexDecoratorAlgCfg()

def python.InDetGNNHardScatterSelectionConfig.GNNHSVertexDecoratorAlgCfg	(		flags,
			name = "GNNHS_VertexDecoratorAlg",
		**	kwargs
	)

Definition at line 114 of file InDetGNNHardScatterSelectionConfig.py.

def GNNHSVertexDecoratorAlgCfg(flags, name="GNNHS_VertexDecoratorAlg", **kwargs):
    cfg = ComponentAccumulator()
 
    kwargs.setdefault("photonsIn", "Photons")
 
    if "gnnTool" not in kwargs:
        kwargs.setdefault("gnnTool", cfg.popToolsAndMerge(
            GNNToolCfg(flags,
                       nnFile="InDetGNNHardScatterSelection/v0/HSGN2_export_090824.onnx")))
 
    if "TrackVertexAssociationTool" not in kwargs:
        from TrackVertexAssociationTool.TrackVertexAssociationToolConfig import TTVAToolCfg
        kwargs.setdefault(
            "TrackVertexAssociationTool",
            cfg.popToolsAndMerge(
                TTVAToolCfg(
                    flags,
                    "TrackVertexAssociationTool_GNNHS",
                    VertexContName=(
                        "PrimaryVertices_initial" 
                        if flags.Tracking.PriVertex.sortingSetup is VertexSortingSetup.GNNSorting 
                        else "PrimaryVertices"
                    ),
                )
            ),
        )
 
 
    cfg.addEventAlgo(
        CompFactory.InDetGNNHardScatterSelection.VertexDecoratorAlg(name, **kwargs))
    return cfg
        
 
# Global Sequence

GNNSequenceCfg()

def python.InDetGNNHardScatterSelectionConfig.GNNSequenceCfg	(		flags,
		**	kwargs
	)

Definition at line 148 of file InDetGNNHardScatterSelectionConfig.py.

def GNNSequenceCfg(flags, **kwargs):
    cfg = ComponentAccumulator()
 
    sysSvc = CompFactory.CP.SystematicsSvc("SystematicsSvc")
    selectionSvc = CompFactory.CP.SelectionNameSvc("SelectionNameSvc")
    cfg.addService(sysSvc)
    cfg.addService(selectionSvc)
 
    inputCollections = {
        "jets": (
            "AntiKt4EMTopoCustomVtxGNNJets" 
            if flags.Tracking.PriVertex.sortingSetup is VertexSortingSetup.GNNSorting 
            else "AntiKt4EMTopoJets"
        ),
        "electrons": "Electrons",
        "muons": "Muons",
        "photons": "Photons",
    }
 
    ptThresholds = {
        "jets": 15000,
        "electrons": 4500,
        "muons": 3000,
        "photons": 10000,
    }
 
    for obj in inputCollections:
        cfg.merge(GNNHSSelectionAlgCfg(flags, input = inputCollections[obj],
                                       minPt = ptThresholds[obj]))
 
    overlapOutputNames = {
        "muons": f'{inputCollections["muons"]}_OR',
        "electrons": f'{inputCollections["electrons"]}_OR',
        "photons": f'{inputCollections["photons"]}_OR',
        "jets": f'{inputCollections["jets"]}_OR',
    }
 
    cfg.merge(GNNHSOverlapRemovalAlgCfg(flags, overlapInputNames = inputCollections,
                                        overlapOutputNames = overlapOutputNames))
 
    from PhotonVertexSelection.PhotonVertexSelectionConfig import DecoratePhotonPointingAlgCfg
    cfg.merge(DecoratePhotonPointingAlgCfg(flags, PhotonContainerKey=overlapOutputNames["photons"]))
 
    cfg.merge(
        GNNHSVertexDecoratorAlgCfg(
            flags,
            vertexIn=(
                "PrimaryVertices_initial" 
                if flags.Tracking.PriVertex.sortingSetup is VertexSortingSetup.GNNSorting 
                else "PrimaryVertices"
            ),
            electronsIn=overlapOutputNames["electrons"],
            muonsIn=overlapOutputNames["muons"],
            photonsIn=overlapOutputNames["photons"],
            jetsIn=overlapOutputNames["jets"],
        )
    )
 
    return cfg
 
 

GNNToolCfg()

def python.InDetGNNHardScatterSelectionConfig.GNNToolCfg	(		flags,
			name = "HardScatterSelectionGNNTool",
		**	kwargs
	)

Definition at line 109 of file InDetGNNHardScatterSelectionConfig.py.

def GNNToolCfg(flags, name="HardScatterSelectionGNNTool", **kwargs):
    acc = ComponentAccumulator()
    acc.setPrivateTools(CompFactory.InDetGNNHardScatterSelection.GNNTool(name, **kwargs))
    return acc
 

Variable Documentation

Files

python.InDetGNNHardScatterSelectionConfig.Files

Definition at line 215 of file InDetGNNHardScatterSelectionConfig.py.

flags

python.InDetGNNHardScatterSelectionConfig.flags = initConfigFlags()

Definition at line 211 of file InDetGNNHardScatterSelectionConfig.py.

MaxEvents

python.InDetGNNHardScatterSelectionConfig.MaxEvents

Definition at line 216 of file InDetGNNHardScatterSelectionConfig.py.

OutputLevel

python.InDetGNNHardScatterSelectionConfig.OutputLevel

Definition at line 232 of file InDetGNNHardScatterSelectionConfig.py.

sc

python.InDetGNNHardScatterSelectionConfig.sc = top_acc.run(1)

Definition at line 238 of file InDetGNNHardScatterSelectionConfig.py.

summariseProps

python.InDetGNNHardScatterSelectionConfig.summariseProps

Definition at line 233 of file InDetGNNHardScatterSelectionConfig.py.

top_acc

python.InDetGNNHardScatterSelectionConfig.top_acc = MainServicesCfg(flags)

Definition at line 220 of file InDetGNNHardScatterSelectionConfig.py.

True

python.InDetGNNHardScatterSelectionConfig.True

Definition at line 233 of file InDetGNNHardScatterSelectionConfig.py.

withDetails

python.InDetGNNHardScatterSelectionConfig.withDetails

Definition at line 233 of file InDetGNNHardScatterSelectionConfig.py.