[PWGLF] Add HI Lambda/AntiLambda polarization around jets workflow

PWGLF/DataModel/lambdaJetPolarizationIons.h

@@ -0,0 +1,150 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+///
+/// \file lambdaJetPolarizationIons.h
+/// \brief Derived Data table for Jet-induced polarization analysis (HI)
+/// \author Cicero Domenico Muncinelli (cicero.domenico.muncinelli@cern.ch)
+// Comments, questions, complaints, suggestions?
+// Please write to:
+// cicero.domenico.muncinelli@cern.ch
+//
+
+#ifndef PWGLF_DATAMODEL_LAMBDAJETPOL_H_
+#define PWGLF_DATAMODEL_LAMBDAJETPOL_H_
+
+#include <Framework/ASoA.h>
+
+namespace o2::aod
+{
+
+namespace lambdajetpol
+{
+
+// DECLARE_SOA_COLUMN(CollIdx, collIdx, uint64_t); // Using a regular SOA column instead of an index column for convenience
+// Collision information:
+DECLARE_SOA_INDEX_COLUMN(Collision, collision);
+DECLARE_SOA_COLUMN(CentFT0M, centFT0M, float);
+DECLARE_SOA_COLUMN(CentFT0C, centFT0C, float);
+DECLARE_SOA_COLUMN(CentFT0CVariant1, centFT0CVariant1, float);
+DECLARE_SOA_COLUMN(CentMFT, centMFT, float);
+DECLARE_SOA_COLUMN(CentNGlobal, centNGlobal, float);
+DECLARE_SOA_COLUMN(CentFV0A, centFV0A, float);
+
+// Jet (and jet proxies) information:
+DECLARE_SOA_COLUMN(JetPt, jetPt, float);
+DECLARE_SOA_COLUMN(JetEta, jetEta, float);
+DECLARE_SOA_COLUMN(JetPhi, jetPhi, float);
+DECLARE_SOA_COLUMN(JetNConstituents, jetNConstituents, uint64_t);
+
+DECLARE_SOA_COLUMN(LeadParticlePt, leadParticlePt, float);
+DECLARE_SOA_COLUMN(LeadParticleEta, leadParticleEta, float);
+DECLARE_SOA_COLUMN(LeadParticlePhi, leadParticlePhi, float);
+
+// V0 information:
+DECLARE_SOA_COLUMN(V0Pt, v0Pt, float);
+DECLARE_SOA_COLUMN(V0Eta, v0Eta, float);
+DECLARE_SOA_COLUMN(V0Phi, v0Phi, float);
+
+DECLARE_SOA_COLUMN(IsLambda, isLambda, bool);
+DECLARE_SOA_COLUMN(IsAntiLambda, isAntiLambda, bool);
+DECLARE_SOA_COLUMN(MassLambda, massLambda, float);
+DECLARE_SOA_COLUMN(MassAntiLambda, massAntiLambda, float);
+
+DECLARE_SOA_COLUMN(PosPt, posPt, float);
+DECLARE_SOA_COLUMN(PosEta, posEta, float);
+DECLARE_SOA_COLUMN(PosPhi, posPhi, float);
+DECLARE_SOA_COLUMN(NegPt, negPt, float);
+DECLARE_SOA_COLUMN(NegEta, negEta, float);
+DECLARE_SOA_COLUMN(NegPhi, negPhi, float);
+
+DECLARE_SOA_COLUMN(PosTPCNSigmaPr, posTPCNSigmaPr, float);
+DECLARE_SOA_COLUMN(PosTPCNSigmaPi, posTPCNSigmaPi, float);
+DECLARE_SOA_COLUMN(NegTPCNSigmaPr, negTPCNSigmaPr, float);
+DECLARE_SOA_COLUMN(NegTPCNSigmaPi, negTPCNSigmaPi, float);
+
+DECLARE_SOA_COLUMN(V0CosPA, v0cosPA, float);
+DECLARE_SOA_COLUMN(V0Radius, v0radius, float);
+DECLARE_SOA_COLUMN(DcaV0Daughters, dcaV0daughters, float);
+DECLARE_SOA_COLUMN(DcaPosToPV, dcaPosToPV, float);
+DECLARE_SOA_COLUMN(DcaNegToPV, dcaNegToPV, float);
+
+// Dynamic columns for jets (Px,Py,Pz):
+DECLARE_SOA_DYNAMIC_COLUMN(JetPx, jetPx, //! Jet px
+ [](float jetPt, float jetPhi) -> float { return jetPt * std::cos(jetPhi); });
+DECLARE_SOA_DYNAMIC_COLUMN(JetPy, jetPy, //! Jet py
+ [](float jetPt, float jetPhi) -> float { return jetPt * std::sin(jetPhi); });
+DECLARE_SOA_DYNAMIC_COLUMN(JetPz, jetPz, //! Jet pz
+ [](float jetPt, float jetEta) -> float { return jetPt * std::sinh(jetEta); });
+// Same for leading particles:
+DECLARE_SOA_DYNAMIC_COLUMN(LeadParticlePx, leadParticlePx, //! Leading particle px
+ [](float leadParticlePt, float leadParticlePhi) -> float { return leadParticlePt * std::cos(leadParticlePhi); });
+DECLARE_SOA_DYNAMIC_COLUMN(LeadParticlePy, leadParticlePy, //! Leading particle py
+ [](float leadParticlePt, float leadParticlePhi) -> float { return leadParticlePt * std::sin(leadParticlePhi); });
+DECLARE_SOA_DYNAMIC_COLUMN(LeadParticlePz, leadParticlePz, //! Leading particle pz
+ [](float leadParticlePt, float leadParticleEta) -> float { return leadParticlePt * std::sinh(leadParticleEta); });
+
+} // namespace lambdajetpol
+
+DECLARE_SOA_TABLE(RingJets, "AOD", "RINGJETS", // Renamed to follow convention on "s" at the end of table name.
+ lambdajetpol::CollisionId, // Changed to an internal O2 index, slightly different from usual o2::soa::Index<> though
+ lambdajetpol::JetPt,
+ lambdajetpol::JetEta,
+ lambdajetpol::JetPhi,
+ lambdajetpol::JetNConstituents,
+ // Dynamic columns
+ lambdajetpol::JetPx<lambdajetpol::JetPt, lambdajetpol::JetPhi>, // Explicitly binding to static columns
+ lambdajetpol::JetPy<lambdajetpol::JetPt, lambdajetpol::JetPhi>,
+ lambdajetpol::JetPz<lambdajetpol::JetPt, lambdajetpol::JetEta>);
+
+DECLARE_SOA_TABLE(RingLeadP, "AOD", "RINGLEADP", // Leading particle table
+ lambdajetpol::CollisionId,
+ lambdajetpol::LeadParticlePt,
+ lambdajetpol::LeadParticleEta,
+ lambdajetpol::LeadParticlePhi,
+ // Dynamic columns
+ lambdajetpol::LeadParticlePx<lambdajetpol::LeadParticlePt, lambdajetpol::LeadParticlePhi>,
+ lambdajetpol::LeadParticlePy<lambdajetpol::LeadParticlePt, lambdajetpol::LeadParticlePhi>,
+ lambdajetpol::LeadParticlePz<lambdajetpol::LeadParticlePt, lambdajetpol::LeadParticleEta>);
+
+DECLARE_SOA_TABLE(RingLaV0s, "AOD", "RINGLAV0S",
+ lambdajetpol::CollisionId,
+ lambdajetpol::V0Pt,
+ lambdajetpol::V0Eta,
+ lambdajetpol::V0Phi,
+ lambdajetpol::IsLambda,
+ lambdajetpol::IsAntiLambda,
+ lambdajetpol::MassLambda,
+ lambdajetpol::MassAntiLambda,
+ lambdajetpol::PosPt,
+ lambdajetpol::PosEta,
+ lambdajetpol::PosPhi,
+ lambdajetpol::NegPt,
+ lambdajetpol::NegEta,
+ lambdajetpol::NegPhi,
+ lambdajetpol::PosTPCNSigmaPr,
+ lambdajetpol::PosTPCNSigmaPi,
+ lambdajetpol::NegTPCNSigmaPr,
+ lambdajetpol::NegTPCNSigmaPi,
+ lambdajetpol::V0CosPA,
+ lambdajetpol::V0Radius,
+ lambdajetpol::DcaV0Daughters,
+ lambdajetpol::DcaPosToPV,
+ lambdajetpol::DcaNegToPV);
+
+DECLARE_SOA_TABLE(RingCollisions, "AOD", "RINGCOLLISIONS",
+ lambdajetpol::CollisionId,
+ lambdajetpol::CentFT0M,
+ lambdajetpol::CentFT0C,
+ lambdajetpol::CentFV0A);
+
+} // namespace o2::aod
+
+#endif // PWGLF_DATAMODEL_LAMBDAJETPOL_H_

PWGLF/TableProducer/Strangeness/CMakeLists.txt

@@ -147,6 +147,11 @@ o2physics_add_dpl_workflow(lambdajetpolarizationbuilder
  PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
  COMPONENT_NAME Analysis)
 
+o2physics_add_dpl_workflow(lambdajetpolarizationions
+ SOURCES lambdaJetPolarizationIons.cxx
+ PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore O2Physics::PWGJECore FastJet::FastJet O2Physics::AnalysisCCDB
+ COMPONENT_NAME Analysis)
+
 o2physics_add_dpl_workflow(stracents
  SOURCES stracents.cxx
  PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore