Add information for the Xi/Omega polarisation analysis (#7075)

Author: mpuccio

PWGLF/DataModel/cascqaanalysis.h

@@ -25,6 +25,8 @@
 #include "Common/DataModel/Multiplicity.h"
 #include "Common/DataModel/Centrality.h"
 #include "TRandom.h"
+#include "Math/Vector4D.h"
+#include "Math/Boost.h"
 
 namespace o2::aod
 {
@@ -113,7 +115,7 @@ namespace cascadesflow
 DECLARE_SOA_COLUMN(CentFT0A, centFT0A, float);
 DECLARE_SOA_COLUMN(CentFT0C, centFT0C, float);
 DECLARE_SOA_COLUMN(CentFT0M, centFT0M, float);
-DECLARE_SOA_COLUMN(Sign, sign, int);
+DECLARE_SOA_COLUMN(Sign, sign, int16_t);
 DECLARE_SOA_COLUMN(Pt, pt, float);
 DECLARE_SOA_COLUMN(Eta, eta, float);
 DECLARE_SOA_COLUMN(Phi, phi, float);
@@ -124,6 +126,9 @@ DECLARE_SOA_COLUMN(V2CSP, v2CSP, float);
 DECLARE_SOA_COLUMN(PsiT0C, psiT0C, float);
 DECLARE_SOA_COLUMN(BDTResponseXi, bdtResponseXi, float);
 DECLARE_SOA_COLUMN(BDTResponseOmega, bdtResponseOmega, float);
+DECLARE_SOA_COLUMN(CosThetaStarLambdaFromOmega, cosThetaStarLambdaFromOmega, float);
+DECLARE_SOA_COLUMN(CosThetaStarLambdaFromXi, cosThetaStarLambdaFromXi, float);
+DECLARE_SOA_COLUMN(CosThetaStarProton, cosThetaStarProton, float);
 
 } // namespace cascadesflow
 
@@ -155,7 +160,7 @@ DECLARE_SOA_TABLE(CascTraining, "AOD", "CascTraining", o2::soa::Index<>,
                   mycascades::DCABachToPV, mycascades::DCACascDaughters, mycascades::DCAV0Daughters, mycascades::DCAV0ToPV, mycascades::BachBaryonCosPA, mycascades::BachBaryonDCAxyToPV, mycascades::McPdgCode);
 
 DECLARE_SOA_TABLE(CascAnalysis, "AOD", "CascAnalysis", o2::soa::Index<>,
-                  cascadesflow::CentFT0C, cascadesflow::Sign, cascadesflow::Pt, cascadesflow::Eta, cascadesflow::Phi, cascadesflow::MassXi, cascadesflow::MassOmega, cascadesflow::V2CSP, cascadesflow::V2CEP, cascadesflow::PsiT0C, cascadesflow::BDTResponseXi, cascadesflow::BDTResponseOmega);
+                  cascadesflow::CentFT0C, cascadesflow::Sign, cascadesflow::Pt, cascadesflow::Eta, cascadesflow::Phi, cascadesflow::MassXi, cascadesflow::MassOmega, cascadesflow::V2CSP, cascadesflow::V2CEP, cascadesflow::PsiT0C, cascadesflow::BDTResponseXi, cascadesflow::BDTResponseOmega, cascadesflow::CosThetaStarLambdaFromOmega, cascadesflow::CosThetaStarLambdaFromXi, cascadesflow::CosThetaStarProton);
 
 namespace myMCcascades
 {

PWGLF/TableProducer/Strangeness/cascadeflow.cxx

@@ -298,6 +298,25 @@ struct cascadeFlow {
   template <class collision_t, class cascade_t>
   void fillAnalysedTable(collision_t coll, cascade_t casc, float v2CSP, float v2CEP, float PsiT0C, float BDTresponseXi, float BDTresponseOmega)
   {
+    double masses[2]{o2::constants::physics::MassXiMinus, o2::constants::physics::MassOmegaMinus};
+    ROOT::Math::PxPyPzMVector cascadeVector[2], lambdaVector, protonVector;
+    float cosThetaStarLambda[2], cosThetaStarProton;
+    lambdaVector.SetCoordinates(casc.pxlambda(), casc.pylambda(), casc.pzlambda(), o2::constants::physics::MassLambda);
+    ROOT::Math::Boost lambdaBoost{lambdaVector.BoostToCM()};
+    if (casc.sign() < 0) {
+      protonVector.SetCoordinates(casc.pxneg(), casc.pyneg(), casc.pzneg(), o2::constants::physics::MassProton);
+    } else {
+      protonVector.SetCoordinates(casc.pxpos(), casc.pypos(), casc.pzpos(), o2::constants::physics::MassProton);
+    }
+    auto boostedProton{lambdaBoost(protonVector)};
+    cosThetaStarProton = boostedProton.Pz() / boostedProton.P();
+    for (int i{0}; i < 2; ++i) {
+      cascadeVector[i].SetCoordinates(casc.px(), casc.py(), casc.pz(), masses[i]);
+      ROOT::Math::Boost cascadeBoost{cascadeVector[i].BoostToCM()};
+      auto boostedLambda{cascadeBoost(lambdaVector)};
+      cosThetaStarLambda[i] = boostedLambda.Pz() / boostedLambda.P();
+    }
+
     analysisSample(coll.centFT0C(),
                    casc.sign(),
                    casc.pt(),
@@ -309,7 +328,10 @@ struct cascadeFlow {
                    v2CEP,
                    PsiT0C,
                    BDTresponseXi,
-                   BDTresponseOmega);
+                   BDTresponseOmega,
+                   cosThetaStarLambda[0],
+                   cosThetaStarLambda[1],
+                   cosThetaStarProton);
   }
 
   void init(InitContext const&)