From 3ae2431bb65a65e2aebc300d9d60dc0ad38325a1 Mon Sep 17 00:00:00 2001
From: Yash Patley <52608802+yashpatley@users.noreply.github.com>
Date: Tue, 20 Oct 2026 19:32:23 +0530
Subject: [PATCH] [PWGCF]: Lambda R2 Correlation MC on the fly update

Added process to run on on-the-fly MC gen data
---
 .../Tasks/lambdaR2Correlation.cxx                        | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/PWGCF/TwoParticleCorrelations/Tasks/lambdaR2Correlation.cxx b/PWGCF/TwoParticleCorrelations/Tasks/lambdaR2Correlation.cxx
index bcf8852121d..0b25d067843 100644
--- a/PWGCF/TwoParticleCorrelations/Tasks/lambdaR2Correlation.cxx
+++ b/PWGCF/TwoParticleCorrelations/Tasks/lambdaR2Correlation.cxx
@@ -363,7 +363,7 @@ struct LambdaTableProducer {
     histos.add("QA/Lambda/DCA/h3f_Dca_vs_pT_cent", "DCA", kTH3F, {axisCent, axisV0Pt, axisDcaLambda});
 
     // MC Generated Histograms
-    if (doprocessMCRecoGen || doprocessMCReco) {
+    if (doprocessMCRecoGen || doprocessMCReco || doprocessMCGen) {
       // DCA analysis
       histos.add("QA/Lambda/DCA/h3f_Prm_Dca_vs_pT_cent", "Primary DCA", kTH3F, {axisCent, axisV0Pt, axisDcaLambda});
       histos.add("QA/Lambda/DCA/h3f_Scd_Dca_vs_pT_cent", "Weak Decay DCA", kTH3F, {axisCent, axisV0Pt, axisDcaLambda});
@@ -923,6 +923,13 @@ struct LambdaTableProducer {
   }
 
   PROCESS_SWITCH(LambdaTableProducer, processMCRecoGen, "Process for MC RecoGen", false);
+
+  void processMCGen(aod::McCollisions::iterator const& mcCollision, aod::McParticles const& mcParticles)
+  {
+    fillLambdaMcGenTables(mcCollision, mcParticles);
+  }
+
+  PROCESS_SWITCH(LambdaTableProducer, processMCGen, "Process for MC Gen", false);
 };
 
 struct LambdaTracksExtProducer {