O2Physics/PWGLF/DataModel/lambdaJetPolarizationIons.h at 4522db9e0e82987f23a45570d414b95232dbf2cc · alibuild/O2Physics · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
// 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_LAMBDAJETPOLARIZATIONIONS_H_
#define PWGLF_DATAMODEL_LAMBDAJETPOLARIZATIONIONS_H_

#include <Framework/ASoA.h>

#include <cmath>

namespace o2::aod
{

namespace lambdajetpol
{
// Collision information:
DECLARE_SOA_COLUMN(CentFT0M, centFT0M, float);
DECLARE_SOA_COLUMN(CentFT0C, centFT0C, 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, int);

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(RingCollisions, "AOD", "RINGCOLLISIONS",
                  o2::soa::Index<>, // self-index: auto-assigned row number
                  lambdajetpol::CentFT0M,
                  lambdajetpol::CentFT0C,
                  lambdajetpol::CentFV0A);

namespace lambdajetpol
{
DECLARE_SOA_INDEX_COLUMN(RingCollision, ringCollision); // Declare index after table is available
} // namespace lambdajetpol

DECLARE_SOA_TABLE(RingJets, "AOD", "RINGJETS",
                  lambdajetpol::RingCollisionId, // relational index -> RingCollisions
                  lambdajetpol::JetPt,
                  lambdajetpol::JetEta,
                  lambdajetpol::JetPhi,
                  lambdajetpol::JetNConstituents,
                  // Dynamic columns (explicitly bound to their static inputs):
                  lambdajetpol::JetPx<lambdajetpol::JetPt, lambdajetpol::JetPhi>,
                  lambdajetpol::JetPy<lambdajetpol::JetPt, lambdajetpol::JetPhi>,
                  lambdajetpol::JetPz<lambdajetpol::JetPt, lambdajetpol::JetEta>);

DECLARE_SOA_TABLE(RingLeadP, "AOD", "RINGLEADP",
                  lambdajetpol::RingCollisionId,
                  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::RingCollisionId,
                  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);

using RingCollision = RingCollisions::iterator; // Useful shorthand
} // namespace o2::aod

#endif // PWGLF_DATAMODEL_LAMBDAJETPOLARIZATIONIONS_H_