From 72bf3a9773e4b417c74a1d8e6ae0f05171f43b57 Mon Sep 17 00:00:00 2001
From: Junlee Kim <jikim1290@gmail.com>
Date: Tue, 29 Oct 2024 04:31:24 +0100
Subject: [PATCH] [PWGLF] adding configurable to test flow and polarization
 (#8184)

Co-authored-by: junleekim <junleekim@maccbookjunlee2024.dyndns.cern.ch>
Co-authored-by: ALICE Action Bot <alibuild@cern.ch>
---
 PWGLF/Tasks/Strangeness/lambdapolarization.cxx | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/PWGLF/Tasks/Strangeness/lambdapolarization.cxx b/PWGLF/Tasks/Strangeness/lambdapolarization.cxx
index b42b45a5b4a..7cf6677b1bc 100644
--- a/PWGLF/Tasks/Strangeness/lambdapolarization.cxx
+++ b/PWGLF/Tasks/Strangeness/lambdapolarization.cxx
@@ -121,6 +121,9 @@ struct lambdapolarization {
   Configurable<std::string> cfgQvecRefAName{"cfgQvecRefAName", "TPCpos", "The name of detector for reference A"};
   Configurable<std::string> cfgQvecRefBName{"cfgQvecRefBName", "TPCneg", "The name of detector for reference B"};
 
+  Configurable<bool> cfgPhiDepStudy{"cfgPhiDepStudy", false, "cfg for phi dependent study"};
+  Configurable<float> cfgPhiDepSig{"cfgPhiDepSig", 0.2, "cfg for significance on phi dependent study"};
+
   Configurable<bool> cfgShiftCorr{"cfgShiftCorr", false, "additional shift correction"};
   Configurable<bool> cfgShiftCorrDef{"cfgShiftCorrDef", false, "additional shift correction definition"};
   Configurable<std::string> cfgShiftPath{"cfgShiftPath", "Users/j/junlee/Qvector/QvecCalib/Shift", "Path for Shift"};
@@ -549,6 +552,10 @@ struct lambdapolarization {
         relphi = TVector2::Phi_0_2pi(static_cast<float>(nmode) * (LambdaVec.Phi() - psidefFT0C - deltapsiFT0C));
       }
 
+      if (cfgPhiDepStudy && cfgPhiDepSig * std::abs(TMath::Sin(relphi)) > gRandom->Uniform(0, 1)) {
+        continue;
+      }
+
       if (nmode == 2) { ////////////
         if (LambdaTag) {
           histos.fill(HIST("psi2/h_lambda_cos"), v0.mLambda(), v0.pt(), angle, centrality, relphi);