Adding option to save multiple MFT-MUON matching candidates

AliceO2Group · Jan 11, 2025 · 0220a49 · 0220a49
1 parent e40ee48
commit 0220a49
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/MC/bin/o2dpg_sim_config.py b/MC/bin/o2dpg_sim_config.py
@@ -80,6 +80,9 @@ def add(cfg, flatconfig):
         add(config, {"FwdMatching.matchFcn" : "matchsXYPhiTanl"})
     if args.fwdmatching_cut_4_param == True:
         add(config, {"FwdMatching.cutFcn" : "cut3SigmaXYAngles"})
+    if int(args.fwdmatchingNcandidates) > 0:
+        add(config, {"FwdMatching.saveMode" : 3})
+        add(config, {"FwdMatching.nCandidates" : int(args.fwdmatchingNcandidates)})
 
     # deal with larger combinatorics
     if args.col == "PbPb" or (args.embedding and args.colBkg == "PbPb"):

diff --git a/MC/bin/o2dpg_sim_workflow.py b/MC/bin/o2dpg_sim_workflow.py
@@ -151,6 +151,7 @@
 
 # Matching training for machine learning
 parser.add_argument('--fwdmatching-save-trainingdata', action='store_true', help='enables saving parameters at plane for matching training with machine learning')
+parser.add_argument('--fwdmatchingNcandidates', type=int, help='enables saving provided number of matching candidates', default=0)
 
 args = parser.parse_args()
 print (args)