O2Physics/ALICE3/DataModel/OTFRICH.h at cc392db1e17a2be50ba1035c74bf24b5f1bbb761 · 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
// 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   OTFRICH.h
/// \author Nicola Nicassio, University and INFN Bari
/// \since  15/05/2023
/// \brief  Set of tables for the ALICE 3 OTFRICH information
///

#ifndef ALICE3_DATAMODEL_OTFRICH_H_
#define ALICE3_DATAMODEL_OTFRICH_H_

// O2 includes
#include "Framework/AnalysisDataModel.h"

namespace o2::aod
{
namespace upgrade_rich
{
DECLARE_SOA_COLUMN(NSigmaElectronRich, nSigmaElectronRich, float); //! NSigma electron BarrelRich
DECLARE_SOA_COLUMN(NSigmaMuonRich, nSigmaMuonRich, float);         //! NSigma muon BarrelRich
DECLARE_SOA_COLUMN(NSigmaPionRich, nSigmaPionRich, float);         //! NSigma pion BarrelRich
DECLARE_SOA_COLUMN(NSigmaKaonRich, nSigmaKaonRich, float);         //! NSigma kaon BarrelRich
DECLARE_SOA_COLUMN(NSigmaProtonRich, nSigmaProtonRich, float);     //! NSigma proton BarrelRich
DECLARE_SOA_DYNAMIC_COLUMN(NSigmaRich, nSigmaRich,                 //! General function to get the nSigma for the RICH
                           [](const float el,
                              const float mu,
                              const float pi,
                              const float ka,
                              const float pr,
                              const int id) -> float {
                             switch (std::abs(id)) {
                               case 0:
                                 return el;
                               case 1:
                                 return mu;
                               case 2:
                                 return pi;
                               case 3:
                                 return ka;
                               case 4:
                                 return pr;
                               default:
                                 LOG(fatal) << "Unrecognized PDG code for RICH";
                                 return 999.f;
                             }
                           });

DECLARE_SOA_COLUMN(HasSig, hasSig, bool);     //! Has signal in the barrel rich (is particle over threshold)
DECLARE_SOA_COLUMN(HasSigEl, hasSigEl, bool); //! Has nSigma electron BarrelRich (is electron over threshold)
DECLARE_SOA_COLUMN(HasSigMu, hasSigMu, bool); //! Has nSigma muon BarrelRich (is muon over threshold)
DECLARE_SOA_COLUMN(HasSigPi, hasSigPi, bool); //! Has nSigma pion BarrelRich (is pion over threshold)
DECLARE_SOA_COLUMN(HasSigKa, hasSigKa, bool); //! Has nSigma kaon BarrelRich (is kaon over threshold)
DECLARE_SOA_COLUMN(HasSigPr, hasSigPr, bool); //! Has nSigma proton BarrelRich (is proton over threshold)

} // namespace upgrade_rich
DECLARE_SOA_TABLE(UpgradeRichs, "AOD", "UPGRADERICH",
                  upgrade_rich::NSigmaElectronRich,
                  upgrade_rich::NSigmaMuonRich,
                  upgrade_rich::NSigmaPionRich,
                  upgrade_rich::NSigmaKaonRich,
                  upgrade_rich::NSigmaProtonRich,
                  upgrade_rich::NSigmaRich<upgrade_rich::NSigmaElectronRich,
                                           upgrade_rich::NSigmaMuonRich,
                                           upgrade_rich::NSigmaPionRich,
                                           upgrade_rich::NSigmaKaonRich,
                                           upgrade_rich::NSigmaProtonRich>);

using UpgradeRich = UpgradeRichs::iterator;

DECLARE_SOA_TABLE(UpgradeRichSignals, "AOD", "UPGRADERICHSIG",
                  upgrade_rich::HasSig,
                  upgrade_rich::HasSigEl,
                  upgrade_rich::HasSigMu,
                  upgrade_rich::HasSigPi,
                  upgrade_rich::HasSigKa,
                  upgrade_rich::HasSigPr);

using UpgradeRichSignal = UpgradeRichSignals::iterator;

} // namespace o2::aod

#endif // ALICE3_DATAMODEL_OTFRICH_H_