Add configurable to disable 3body tracking

creetz16 · Mar 26, 2024 · 98a48a8 · 98a48a8
1 parent d685dc8
commit 98a48a8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/...ertexing/StrangenessTracking/include/StrangenessTracking/StrangenessTrackingConfigParam.h b/...ertexing/StrangenessTracking/include/StrangenessTracking/StrangenessTrackingConfigParam.h
@@ -33,6 +33,7 @@ struct StrangenessTrackingParamConfig : public o2::conf::ConfigurableParamHelper
   float mMaxChi2 = 50;         // Maximum matching chi2
   bool mVertexMatching = true; // Flag to enable/disable vertex matching
   bool mSkipTPC = true;        // Flag to enable/disable TPC only tracks
+  bool mSkip3Body = false;     // Flag to enable/disable Decay3Body tracking
 
   O2ParamDef(StrangenessTrackingParamConfig, "strtracker");
 };

diff --git a/Detectors/Vertexing/StrangenessTracking/src/StrangenessTracker.cxx b/Detectors/Vertexing/StrangenessTracking/src/StrangenessTracker.cxx
@@ -312,9 +312,11 @@ void StrangenessTracker::process()
   }
 
   // Loop over 3bodys
-  for (int i3Body{0}; i3Body < mInput3BodyTracks.size(); i3Body++) {
-    LOG(debug) << "Analysing 3-Body: " << i3Body + 1 << "/" << mInput3BodyTracks.size();
-    process3Body(i3Body, mInput3BodyTracks[i3Body], mInput3BodyIndices[i3Body]);
+  if (!mStrParams->mSkip3Body) {
+    for (int i3Body{0}; i3Body < mInput3BodyTracks.size(); i3Body++) {
+      LOG(debug) << "Analysing 3-Body: " << i3Body + 1 << "/" << mInput3BodyTracks.size();
+      process3Body(i3Body, mInput3BodyTracks[i3Body], mInput3BodyIndices[i3Body]);
+    }
   }
 }