O2Physics/Tutorials/Skimming/DataModel/LFDerived.h at c1d3c08287eb3f2be2d6a6c3038f015f0985f6aa · AliceO2Group/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
// 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.
/// \author Nima Zardoshti <nima.zardoshti@cern.ch>, CERN

#ifndef TUTORIALS_SKIMMING_DATAMODEL_LFDERIVED_H_
#define TUTORIALS_SKIMMING_DATAMODEL_LFDERIVED_H_

#include <Framework/ASoA.h>
#include <Framework/AnalysisDataModel.h>

#include <TMath.h>

namespace o2::aod
{
DECLARE_SOA_TABLE(LFCollisions, "AOD", "LFCOLLISION", o2::soa::Index<>,
                  o2::aod::collision::PosZ);
using LFCollision = LFCollisions::iterator;

namespace lftrack
{
DECLARE_SOA_INDEX_COLUMN(LFCollision, lfCollision);
DECLARE_SOA_COLUMN(Pt, pt, float);
DECLARE_SOA_COLUMN(Eta, eta, float);
DECLARE_SOA_COLUMN(Phi, phi, float);
DECLARE_SOA_DYNAMIC_COLUMN(P, p, [](float pt, float eta) { return pt * TMath::CosH(eta); });
DECLARE_SOA_COLUMN(TpcNSigmaEl, tpcNSigmaEl, float);
DECLARE_SOA_COLUMN(TpcNSigmaMu, tpcNSigmaMu, float);
DECLARE_SOA_COLUMN(TpcNSigmaPi, tpcNSigmaPi, float);
DECLARE_SOA_COLUMN(TpcNSigmaKa, tpcNSigmaKa, float);
DECLARE_SOA_COLUMN(TpcNSigmaPr, tpcNSigmaPr, float);
DECLARE_SOA_COLUMN(TpcNSigmaDe, tpcNSigmaDe, float);
DECLARE_SOA_COLUMN(TpcNSigmaTr, tpcNSigmaTr, float);
DECLARE_SOA_COLUMN(TpcNSigmaHe, tpcNSigmaHe, float);
DECLARE_SOA_COLUMN(TpcNSigmaAl, tpcNSigmaAl, float);
DECLARE_SOA_COLUMN(TofNSigmaEl, tofNSigmaEl, float);
DECLARE_SOA_COLUMN(TofNSigmaMu, tofNSigmaMu, float);
DECLARE_SOA_COLUMN(TofNSigmaPi, tofNSigmaPi, float);
DECLARE_SOA_COLUMN(TofNSigmaKa, tofNSigmaKa, float);
DECLARE_SOA_COLUMN(TofNSigmaPr, tofNSigmaPr, float);
DECLARE_SOA_COLUMN(TofNSigmaDe, tofNSigmaDe, float);
DECLARE_SOA_COLUMN(TofNSigmaTr, tofNSigmaTr, float);
DECLARE_SOA_COLUMN(TofNSigmaHe, tofNSigmaHe, float);
DECLARE_SOA_COLUMN(TofNSigmaAl, tofNSigmaAl, float);
DECLARE_SOA_COLUMN(TpcNSigma1, tpcNSigma1, float);
DECLARE_SOA_COLUMN(TpcNSigma2, tpcNSigma2, float);
DECLARE_SOA_COLUMN(TpcNSigma3, tpcNSigma3, float);
DECLARE_SOA_COLUMN(TpcNSigma4, tpcNSigma4, float);
DECLARE_SOA_COLUMN(TofNSigma1, tofNSigma1, float);
DECLARE_SOA_COLUMN(TofNSigma2, tofNSigma2, float);
DECLARE_SOA_COLUMN(TofNSigma3, tofNSigma3, float);
DECLARE_SOA_COLUMN(TofNSigma4, tofNSigma4, float);
} // namespace lftrack
DECLARE_SOA_TABLE(LFTracks, "AOD", "LFTRACK", o2::soa::Index<>,
                  lftrack::Pt, lftrack::Eta, lftrack::Phi,
                  lftrack::TpcNSigmaEl, lftrack::TpcNSigmaMu,
                  lftrack::TpcNSigmaPi, lftrack::TpcNSigmaKa,
                  lftrack::TpcNSigmaPr, lftrack::TpcNSigmaDe,
                  lftrack::TpcNSigmaTr, lftrack::TpcNSigmaHe,
                  lftrack::TpcNSigmaAl,
                  lftrack::P<lftrack::Pt, lftrack::Eta>);
using LFTrack = LFTracks::iterator;

DECLARE_SOA_TABLE(LFSmallTracks, "AOD", "LFSMALLTRACK", o2::soa::Index<>,
                  lftrack::Pt, lftrack::Eta, lftrack::Phi,
                  lftrack::TpcNSigma1, lftrack::TpcNSigma2,
                  lftrack::TpcNSigma3, lftrack::TpcNSigma4,
                  lftrack::P<lftrack::Pt, lftrack::Eta>);
using LFSmallTrack = LFSmallTracks::iterator;

DECLARE_SOA_TABLE(LFSingleTracks, "AOD", "LFSINGLETRACK", o2::soa::Index<>,
                  lftrack::Pt, lftrack::Eta, lftrack::Phi,
                  lftrack::TpcNSigma1,
                  lftrack::P<lftrack::Pt, lftrack::Eta>);
using LFSingleTrack = LFSingleTracks::iterator;

DECLARE_SOA_TABLE(LFNucleiTracks, "AOD", "LFNUCLEITRACK", o2::soa::Index<>,
                  lftrack::LFCollisionId,
                  lftrack::Pt, lftrack::Eta, lftrack::Phi,
                  lftrack::TpcNSigmaPi, lftrack::TpcNSigmaKa,
                  lftrack::TpcNSigmaPr, lftrack::TpcNSigmaHe,
                  lftrack::TofNSigmaPi, lftrack::TofNSigmaKa,
                  lftrack::TofNSigmaPr, lftrack::TofNSigmaHe,
                  lftrack::P<lftrack::Pt, lftrack::Eta>);

using LFNucleiTrack = LFNucleiTracks::iterator;

} // namespace o2::aod

#endif // TUTORIALS_SKIMMING_DATAMODEL_LFDERIVED_H_