From 61803e62d9d47d99214d65202b6f1978d74e24ed Mon Sep 17 00:00:00 2001
From: Baptiste Ravina <baptiste.ravina@cern.ch>
Date: Wed, 20 Dec 2023 15:52:25 +0100
Subject: [PATCH] improve the IFFLeptonDecorationBlock
---
.../AsgAnalysisAlgorithms/python/AsgAnalysisConfig.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/python/AsgAnalysisConfig.py b/PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/python/AsgAnalysisConfig.py
index 2624c24b78ab..1b7ee5a516f9 100644
--- a/PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/python/AsgAnalysisConfig.py
+++ b/PhysicsAnalysis/Algorithms/AsgAnalysisAlgorithms/python/AsgAnalysisConfig.py
@@ -313,20 +313,25 @@ class IFFLeptonDecorationBlock (ConfigBlock):
def __init__ (self, containerName='') :
super (IFFLeptonDecorationBlock, self).__init__( 'IFFLeptonDecoration' )
self.containerName = containerName
+ self.addOption ('separateChargeFlipElectrons', True, type=bool)
+ self.addOption ('decoration', 'IFFClass_%SYS%', type=str)
def makeAlgs (self, config) :
+ # the classification is only for MC
+ if config.dataType() is DataType.Data: return
+
particles = config.readName(self.containerName)
alg = config.createAlgorithm( 'CP::AsgClassificationDecorationAlg', 'IFFClassifierAlg' + self.containerName )
# the IFF classification tool
config.addPrivateTool( 'tool', 'TruthClassificationTool')
# label charge-flipped electrons as such
- alg.tool.separateChargeFlipElectrons = True
- alg.decoration = 'IFFClass_%SYS%'
+ alg.tool.separateChargeFlipElectrons = self.separateChargeFlipElectrons
+ alg.decoration = self.decoration
alg.particles = particles
# write the decoration only once to the output
- config.addOutputVar(self.containerName, alg.decoration, 'IFFClass', noSys=True)
+ config.addOutputVar(self.containerName, alg.decoration, alg.decoration.split("_%SYS%")[0], noSys=True)
def makeCommonServicesConfig( seq ):
--
GitLab