From c43e865abf8ffa87767579eac91c93b4c87a1cfb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20K=C3=B6hler?= <nicolas.koehler@cern.ch>
Date: Fri, 14 Aug 2020 14:13:29 +0200
Subject: [PATCH] add CaloMuonScoreTool to new configuration
---
.../python/MuonCombinedRecToolsConfig.py | 16 +++++++++++++++-
1 file changed, 15 insertions(+), 1 deletion(-)
diff --git a/Reconstruction/MuonIdentification/MuonCombinedConfig/python/MuonCombinedRecToolsConfig.py b/Reconstruction/MuonIdentification/MuonCombinedConfig/python/MuonCombinedRecToolsConfig.py
index bee7b31c20d..1f1f7d60515 100644
--- a/Reconstruction/MuonIdentification/MuonCombinedConfig/python/MuonCombinedRecToolsConfig.py
+++ b/Reconstruction/MuonIdentification/MuonCombinedConfig/python/MuonCombinedRecToolsConfig.py
@@ -816,6 +816,16 @@ def CaloMuonLikelihoodToolCfg(flags, name='CaloMuonLikelihoodTool', **kwargs ):
result.setPrivateTools(tool)
return result
+def CaloMuonScoreToolCfg(flags, name='CaloMuonScoreTool', **kwargs ):
+ from TrackToCalo.TrackToCaloConfig import ParticleCaloCellAssociationToolCfg
+ result = ParticleCaloCellAssociationToolCfg(flags)
+ kwargs.setdefault("ParticleCaloCellAssociationTool", result.popPrivateTools())
+ caloMuonScoreSvc = CompFactory.CaloMuonScoreONNXRuntimeSvc(name="CaloMuonScoreONNXRuntimeSvc")
+ result.addService(caloMuonScoreSvc)
+ tool = CompFactory.CaloMuonScoreTool(name, **kwargs )
+ result.setPrivateTools(tool)
+ return result
+
def MuonCaloTagToolCfg(flags, name='MuonCaloTagTool', **kwargs ):
result = ComponentAccumulator()
kwargs.setdefault("CaloMuonTagLoose", CompFactory.CaloMuonTag(name = "CaloMuonTagLoose", TagMode="Loose") )
@@ -827,6 +837,10 @@ def MuonCaloTagToolCfg(flags, name='MuonCaloTagTool', **kwargs ):
kwargs.setdefault("CaloMuonLikelihoodTool", acc.popPrivateTools() )
result.addPublicTool(kwargs['CaloMuonLikelihoodTool'])
+ acc = CaloMuonScoreToolCfg(flags)
+ kwargs.setdefault("CaloMuonScoreTool", acc.popPrivateTools() )
+ result.addPublicTool(kwargs['CaloMuonScoreTool'])
+
result.merge(acc)
acc = TrackDepositInCaloToolCfg(flags)
trackDepositInCaloTool = acc.popPrivateTools()
@@ -838,7 +852,7 @@ def MuonCaloTagToolCfg(flags, name='MuonCaloTagTool', **kwargs ):
result.addPublicTool(calotrkmuidalgtrackselectortool)
kwargs.setdefault("TrackSelectorTool", calotrkmuidalgtrackselectortool )
result.merge(acc)
- kwargs.setdefault("doCaloLR", True )
+ kwargs.setdefault("doCaloLR", True)
tool = CompFactory.MuonCombined.MuonCaloTagTool(name, **kwargs )
result.setPrivateTools(tool)
return result
--
GitLab