python.DiTauCommonConfig Namespace Reference

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

Function Documentation

AddDiTauChargeDecoratorCfg()

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

Decorate DiTau charge

Definition at line 60 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()

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

Decorate ditau ID scores 

Definition at line 22 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,
            ))
    decorWPNames = diTauWPDecorator.DecorWPNames
    diTauContainerName = kwargs.setdefault("DiTauContainerName", "DiTauJets")
    wpDecorationKeys = [diTauContainerName + "." + WP for WP in decorWPNames] # FIXME Once WriteDecorHandleKeyArrays are updated, this can switch to wpDecorationKeys = diTauWPDecorator.DecorWPNames
    decorWPCuts = diTauWPDecorator.DecorWPCuts
    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,
                                                      WPDecorationKeys = wpDecorationKeys,
                                                      DecorWPCuts = decorWPCuts)
 
    acc.addPublicTool(DiTauIDDecoratorWrapper)
    acc.addEventAlgo(DiTauIDDecoratorKernel(name              = "DiTauIDDecorKernel",
                                            AugmentationTools = [DiTauIDDecoratorWrapper]))
    return acc
 
 

AddDiTauLowPtCfg()

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