From 8d267b2615255d44803c0053b34f00ff5a9df2a5 Mon Sep 17 00:00:00 2001
From: victorvalenciatorres
 <118812999+victorvalenciatorres@users.noreply.github.com>
Date: Mon, 25 Nov 2024 12:26:34 +0100
Subject: [PATCH] [PWGDQ] Adding checks for M11 and M1111 weights for cumulants
 calculation (#8606)

---
 PWGDQ/Core/VarManager.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/PWGDQ/Core/VarManager.h b/PWGDQ/Core/VarManager.h
index 8d0a1805063..8c18ab0394a 100644
--- a/PWGDQ/Core/VarManager.h
+++ b/PWGDQ/Core/VarManager.h
@@ -4018,6 +4018,10 @@ void VarManager::FillQVectorFromGFW(C const& /*collision*/, A const& compA11, A
   values[kCORR4REF] = (pow(norm(compA21), 2) + norm(compA42) - 2. * (compA42 * conj(compA21) * conj(compA21)).real() + 8. * (compA23 * conj(compA21)).real() - 4. * S12A * norm(compA21) - 6. * S14A - 2. * S22A) / values[kM1111REF];
   values[kCORR2REF] = std::isnan(values[kCORR2REF]) || std::isinf(values[kCORR2REF]) ? 0 : values[kCORR2REF];
   values[kCORR4REF] = std::isnan(values[kCORR4REF]) || std::isinf(values[kCORR4REF]) ? 0 : values[kCORR4REF];
+  values[kM11REF] = std::isnan(values[kM11REF]) || std::isinf(values[kM11REF]) ? 0 : values[kM11REF];
+  values[kM1111REF] = std::isnan(values[kM1111REF]) || std::isinf(values[kM1111REF]) ? 0 : values[kM1111REF];
+  values[kM11REF] = std::isnan(values[kCORR2REF]) || std::isinf(values[kCORR2REF]) ? 0 : values[kM11REF];
+  values[kM1111REF] = std::isnan(values[kCORR4REF]) || std::isinf(values[kCORR4REF]) ? 0 : values[kM1111REF];
   values[kCORR2REFw] = values[kCORR2REF] * values[kM11REF];
   values[kCORR2REFsquared] = values[kCORR2REF] * values[kCORR2REF];
   values[kCORR2REFsquaredw] = values[kCORR2REF] * values[kCORR2REF] * values[kM11REF];