DiTauToolsConfig Namespace Reference

Functions
	SeedJetBuilderCfg (flags, name="DiTauRec_SeedJetBuilder", jetCollection="")
	SubjetBuilderCfg (flags, name="DiTauRec_SubjetBuilder")
	TVAToolCfg (flags, name="TVATool_forDiTaus", **kwargs)
	JetAlgCfg (flags, name="DiTauRec_JetAlgorithm", **kwargs)
	VertexFinderCfg (flags, name="DiTauRec_VertexFinder", **kwargs)
	DiTauTrackFinderCfg (flags, name="DiTauRec_DiTauTrackFinder", **kwargs)
	CellFinderCfg (flags, name="DiTauRec_CellFinder")
	DiTauConstituentFinderCfg (flags, name="DiTauRec_DiTauConstituentFinder", **kwargs)
	DiTauExtraVarDecoratorCfg (flags, name="DiTauRec_ExtraVarDecorator", **kwargs)
	DiTauOnnxScoreCalculatorCfg (flags, name="DiTauRec_OnnxScoreCalculator", **kwargs)
	DiTauWPDecoratorCfg (flags, name="DiTauRec_WPDecorator", **kwargs)

Function Documentation

CellFinderCfg()

DiTauToolsConfig.CellFinderCfg	(		flags,
			name = "DiTauRec_CellFinder" )

Configure the cell finder

Definition at line 76 of file DiTauToolsConfig.py.

def CellFinderCfg(flags, name="DiTauRec_CellFinder"):
    """Configure the cell finder"""
    acc = ComponentAccumulator()
 
    CellFinder = CompFactory.CellFinder(name,
                                        Rsubjet = flags.DiTau.Rsubjet,)
    acc.setPrivateTools(CellFinder)
    return acc
 

DiTauConstituentFinderCfg()

DiTauToolsConfig.DiTauConstituentFinderCfg	(		flags,
			name = "DiTauRec_DiTauConstituentFinder",
		**	kwargs )

Configure the di-tau constituent finder

Definition at line 85 of file DiTauToolsConfig.py.

def DiTauConstituentFinderCfg(flags, name="DiTauRec_DiTauConstituentFinder", **kwargs):
    """Configure the di-tau constituent finder"""
    acc = ComponentAccumulator()
    kwargs.setdefault("Rsubjet", 0.2)
    kwargs.setdefault("UseRawConstit", True)
 
    acc.setPrivateTools(CompFactory.DiTauConstituentFinder(name, **kwargs))
    return acc
 

DiTauExtraVarDecoratorCfg()

DiTauToolsConfig.DiTauExtraVarDecoratorCfg	(		flags,
			name = "DiTauRec_ExtraVarDecorator",
		**	kwargs )

Configure the ExtraVarDecorator

Definition at line 94 of file DiTauToolsConfig.py.

def DiTauExtraVarDecoratorCfg(flags, name="DiTauRec_ExtraVarDecorator", **kwargs):
    """Configure the ExtraVarDecorator"""
    acc = ComponentAccumulator()
    acc.setPrivateTools(CompFactory.DiTauExtraVarDecorator(name, **kwargs))
    return acc
 

DiTauOnnxScoreCalculatorCfg()

DiTauToolsConfig.DiTauOnnxScoreCalculatorCfg	(		flags,
			name = "DiTauRec_OnnxScoreCalculator",
		**	kwargs )

Configure the OnnxScoreCalculator

Definition at line 100 of file DiTauToolsConfig.py.

def DiTauOnnxScoreCalculatorCfg(flags, name="DiTauRec_OnnxScoreCalculator", **kwargs):
    """Configure the OnnxScoreCalculator"""
    acc = ComponentAccumulator()
    kwargs.setdefault("onnxModelPath", "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx")
    kwargs.setdefault("maxTracks", 10)
    acc.setPrivateTools(CompFactory.DiTauOnnxDiscriminantTool(name, **kwargs))
    return acc
 

DiTauTrackFinderCfg()

DiTauToolsConfig.DiTauTrackFinderCfg	(		flags,
			name = "DiTauRec_DiTauTrackFinder",
		**	kwargs )

Configure the di-tau track finder

Definition at line 58 of file DiTauToolsConfig.py.

def DiTauTrackFinderCfg(flags, name="DiTauRec_DiTauTrackFinder", **kwargs):
    """Configure the di-tau track finder"""    
    acc = ComponentAccumulator()
    
    kwargs.setdefault("MaxDrJet", 1.0)
    kwargs.setdefault("MaxDrSubjet", 0.2)
    kwargs.setdefault("MaxNTracksSubjet", -1)
    kwargs.setdefault("TrackParticleContainer", "InDetTrackParticles")
 
    if "TrackSelectorTool" not in kwargs:
        from InDetConfig.InDetTrackSelectorToolConfig import TauRecInDetTrackSelectorToolCfg
        InDetTrackSelectorTool = acc.popToolsAndMerge(TauRecInDetTrackSelectorToolCfg(flags))
        acc.addPublicTool(InDetTrackSelectorTool)
        kwargs.setdefault("TrackSelectorTool", InDetTrackSelectorTool)
 
    acc.setPrivateTools(CompFactory.DiTauTrackFinder(name, **kwargs))
    return acc
 

DiTauWPDecoratorCfg()

DiTauToolsConfig.DiTauWPDecoratorCfg	(		flags,
			name = "DiTauRec_WPDecorator",
		**	kwargs )

Configure the WPDecorator

Definition at line 108 of file DiTauToolsConfig.py.

def DiTauWPDecoratorCfg(flags, name="DiTauRec_WPDecorator", **kwargs):
    """Configure the WPDecorator"""
    acc = ComponentAccumulator()
    kwargs.setdefault("ScoreName", "omni_score")
    kwargs.setdefault("DiTauContainerName", "DiTauJets")
    kwargs.setdefault("DecorWPNames", ["omni_score_VL", "omni_score_L", "omni_score_M", "omni_score_T"])
    kwargs.setdefault("DecorWPCuts", [0.40, 0.60, 0.80, 0.9]) #NOTE: these are dummy values for testing, to be updated when WPs values will be offficially decided 
    acc.setPrivateTools(CompFactory.DiTauWPDecorator(name, **kwargs))
    return acc
 

JetAlgCfg()

DiTauToolsConfig.JetAlgCfg	(		flags,
			name = "DiTauRec_JetAlgorithm",
		**	kwargs )

Configure the JetAlgorithm

Definition at line 35 of file DiTauToolsConfig.py.

def JetAlgCfg(flags, name="DiTauRec_JetAlgorithm", **kwargs): # Name changed wrt legacy config DiTauRec_TVATool
    """Configure the JetAlgorithm"""
    acc = ComponentAccumulator()
 
    tools = [acc.popToolsAndMerge(TVAToolCfg(flags))]
    kwargs.setdefault("Tools", tools)
 
    acc.addEventAlgo(CompFactory.JetAlgorithm(name, **kwargs))
    return acc
 
# require TrackVertexAssociation to be produced by TVA tool - see above

SeedJetBuilderCfg()

DiTauToolsConfig.SeedJetBuilderCfg	(		flags,
			name = "DiTauRec_SeedJetBuilder",
			jetCollection = "" )

Configure the seed jet builder

Definition at line 6 of file DiTauToolsConfig.py.

def SeedJetBuilderCfg(flags, name="DiTauRec_SeedJetBuilder", jetCollection=""):
    """Configure the seed jet builder"""
    acc = ComponentAccumulator()
    acc.setPrivateTools(CompFactory.SeedJetBuilder(name, 
                                                   JetCollection = jetCollection if jetCollection != "" else flags.DiTau.SeedJetCollection[0]))
    return acc
 
 

SubjetBuilderCfg()

DiTauToolsConfig.SubjetBuilderCfg	(		flags,
			name = "DiTauRec_SubjetBuilder" )

Configure the subjet builder

Definition at line 14 of file DiTauToolsConfig.py.

def SubjetBuilderCfg(flags, name="DiTauRec_SubjetBuilder"):
    """Configure the subjet builder"""
    acc = ComponentAccumulator()
    acc.setPrivateTools(CompFactory.SubjetBuilder(name, 
                                                  Rsubjet = flags.DiTau.Rsubjet,
                                                  ptminsubjet = flags.DiTau.PtMinSubjet))
    return acc
 

TVAToolCfg()

DiTauToolsConfig.TVAToolCfg	(		flags,
			name = "TVATool_forDiTaus",
		**	kwargs )

Configure the TVA tool

Definition at line 22 of file DiTauToolsConfig.py.

def TVAToolCfg(flags, name="TVATool_forDiTaus", **kwargs):
    """Configure the TVA tool"""
    acc = ComponentAccumulator()
 
    kwargs.setdefault("TrackParticleContainer", "InDetTrackParticles")
    kwargs.setdefault("TrackVertexAssociation", "JetTrackVtxAssoc_forDiTaus")
    kwargs.setdefault("VertexContainer", "PrimaryVertices")
    kwargs.setdefault("MaxTransverseDistance", 2.5) # in mm
    kwargs.setdefault("MaxLongitudinalDistance", 2) # in mm
 
    acc.setPrivateTools(CompFactory.TrackVertexAssociationTool(name, **kwargs))
    return acc
 

VertexFinderCfg()

DiTauToolsConfig.VertexFinderCfg	(		flags,
			name = "DiTauRec_VertexFinder",
		**	kwargs )

Configure the vertex finder

Definition at line 46 of file DiTauToolsConfig.py.

def VertexFinderCfg(flags, name="DiTauRec_VertexFinder", **kwargs):
    """Configure the vertex finder"""
    acc = ComponentAccumulator()
 
    kwargs.setdefault("PrimVtxContainerName", "PrimaryVertices")
    kwargs.setdefault("AssociatedTracks", "GhostTrack")
    kwargs.setdefault("TrackVertexAssociation", "JetTrackVtxAssoc_forDiTaus")
    kwargs.setdefault("UseTJVA", flags.Tau.doTJVA)
 
    acc.setPrivateTools(CompFactory.VertexFinder(name, **kwargs))
    return acc