python.DiTauCommonConfig Namespace Reference

Functions
def	AddDiTauLowPtCfg (flags, **kwargs)
def	AddDiTauIDDecorationCfg (flags, **kwargs)
def	AddDiTauChargeDecoratorCfg (flags, **kwargs)

Function Documentation

AddDiTauChargeDecoratorCfg()

def python.DiTauCommonConfig.AddDiTauChargeDecoratorCfg	(		flags,
		**	kwargs
	)

Decorate DiTau charge

Definition at line 53 of file DiTauCommonConfig.py.

def AddDiTauChargeDecoratorCfg(flags, **kwargs):
    """Decorate DiTau charge"""
 
    kwargs.setdefault("DiTauContainerName", "DiTauJets")
    kwargs.setdefault("prefix",           kwargs['DiTauContainerName'])
 
    acc = ComponentAccumulator()
   
    DiTauChargeDecorator = CompFactory.DerivationFramework.DiTauChargeDecorator
    DiTauChargeDecoratorKernel = CompFactory.DerivationFramework.CommonAugmentation
 
    prefix = kwargs['prefix']
    diTauChargeDecorator = DiTauChargeDecorator(name               = f"{prefix}_DiTauChargeDecorator",
                                                DiTauContainerName = kwargs['DiTauContainerName'])
    acc.addPublicTool(diTauChargeDecorator)
    acc.addEventAlgo(DiTauChargeDecoratorKernel(name              = f"{prefix}_DiTauIDDecorKernel",
                                                AugmentationTools = [diTauChargeDecorator]))
 
    return acc
 

AddDiTauIDDecorationCfg()

def python.DiTauCommonConfig.AddDiTauIDDecorationCfg	(		flags,
		**	kwargs
	)

Decorate ditau ID scores 

Definition at line 21 of file DiTauCommonConfig.py.

def AddDiTauIDDecorationCfg(flags, **kwargs):
    """Decorate ditau ID scores """
 
    acc = ComponentAccumulator()
 
    import DiTauRec.DiTauToolsConfig as DiTauTools
 
    diTauOnnxScoreCalculator = acc.popToolsAndMerge(DiTauTools.DiTauOnnxScoreCalculatorCfg(
            flags, 
            onnxModelPath                   = "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx",
        ))
 
    diTauWPDecorator = acc.popToolsAndMerge(DiTauTools.DiTauWPDecoratorCfg(
            flags,
            ))
 
    acc.addPublicTool(diTauOnnxScoreCalculator)
    acc.addPublicTool(diTauWPDecorator)
 
    DiTauIDDecoratorWrapper = CompFactory.DerivationFramework.DiTauIDDecoratorWrapper
    DiTauIDDecoratorKernel = CompFactory.DerivationFramework.CommonAugmentation
 
    DiTauIDDecoratorWrapper = DiTauIDDecoratorWrapper(name               = "DiTauIDDecoratorWrapper",
                                                      DiTauContainerName = kwargs['DiTauContainerName'],
                                                      DiTauOnnxDiscriminantTool = diTauOnnxScoreCalculator,
                                                      DiTauWPDecorator = diTauWPDecorator)
 
    acc.addPublicTool(DiTauIDDecoratorWrapper)
    acc.addEventAlgo(DiTauIDDecoratorKernel(name              = "DiTauIDDecorKernel",
                                            AugmentationTools = [DiTauIDDecoratorWrapper]))
    return acc
 

AddDiTauLowPtCfg()

def python.DiTauCommonConfig.AddDiTauLowPtCfg	(		flags,
		**	kwargs
	)

Configure the low-pt di-tau building

Definition at line 7 of file DiTauCommonConfig.py.

def AddDiTauLowPtCfg(flags, **kwargs):
    """Configure the low-pt di-tau building"""
 
    acc = ComponentAccumulator()
 
    from JetRecConfig.JetRecConfig import JetRecCfg
    from JetRecConfig.StandardLargeRJets import AntiKt10LCTopo
    acc.merge(JetRecCfg(flags,AntiKt10LCTopo))
 
    from DiTauRec.DiTauBuilderConfig import DiTauBuilderCfg
    acc.merge(DiTauBuilderCfg(flags, name="DiTauLowPtBuilder", doLowPt=True))
 
    return acc