CorrelationsDerived.h

// 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.
#ifndef PWGCF_DATAMODEL_CORRELATIONSDERIVED_H_
#define PWGCF_DATAMODEL_CORRELATIONSDERIVED_H_

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

#include <cstdint>
#include <vector>

namespace o2::aod
{
namespace cfmccollision
{
DECLARE_SOA_COLUMN(Multiplicity, multiplicity, float); //! Centrality/multiplicity value
} // namespace cfmccollision
DECLARE_SOA_TABLE(CFMcCollisions, "AOD", "CFMCCOLLISION", //! Reduced MC collision table
                  o2::soa::Index<>,
                  mccollision::PosZ, cfmccollision::Multiplicity);
using CFMcCollision = CFMcCollisions::iterator;

namespace cfmcparticle
{
DECLARE_SOA_INDEX_COLUMN(CFMcCollision, cfMcCollision); //! Index to reduced MC collision
DECLARE_SOA_COLUMN(Pt, pt, float);                      //! pT (GeV/c)
DECLARE_SOA_COLUMN(Eta, eta, float);                    //! Pseudorapidity
DECLARE_SOA_COLUMN(Phi, phi, float);                    //! Phi angle
DECLARE_SOA_COLUMN(Sign, sign, int8_t);                 //! Sign (positive, negative)
const uint8_t kReconstructed = 0x80;
} // namespace cfmcparticle
DECLARE_SOA_TABLE(CFMcParticles, "AOD", "CFMCPARTICLE", //! Reduced MC particle table
                  o2::soa::Index<>,
                  cfmcparticle::CFMcCollisionId,
                  cfmcparticle::Pt, cfmcparticle::Eta, cfmcparticle::Phi, cfmcparticle::Sign,
                  mcparticle::PdgCode, mcparticle::Flags,
                  mcparticle::IsPhysicalPrimary<mcparticle::Flags>);
using CFMcParticle = CFMcParticles::iterator;

namespace cfcollision
{
DECLARE_SOA_INDEX_COLUMN(CFMcCollision, cfMcCollision); //! Index to reduced MC collision
DECLARE_SOA_COLUMN(Multiplicity, multiplicity, float);  //! Centrality/multiplicity value
} // namespace cfcollision
DECLARE_SOA_TABLE(CFCollisions, "AOD", "CFCOLLISION", //! Reduced collision table
                  o2::soa::Index<>,
                  bc::RunNumber, collision::PosZ,
                  cfcollision::Multiplicity, timestamp::Timestamp);
DECLARE_SOA_TABLE(CFCollLabels, "AOD", "CFCOLLLABEL", //! Labels for reduced collision table
                  cfcollision::CFMcCollisionId);
using CFCollision = CFCollisions::iterator;
using CFCollLabel = CFCollLabels::iterator;
using CFCollisionsWithLabel = soa::Join<CFCollisions, CFCollLabels>;
using CFCollisionWithLabel = CFCollisionsWithLabel::iterator;

namespace cftrack
{
DECLARE_SOA_INDEX_COLUMN(CFCollision, cfCollision);   //! Index to collision
DECLARE_SOA_INDEX_COLUMN(CFMcParticle, cfMCParticle); //! Index to MC particle
DECLARE_SOA_COLUMN(Pt, pt, float);                    //! pT (GeV/c)
DECLARE_SOA_COLUMN(Eta, eta, float);                  //! Pseudorapidity
DECLARE_SOA_COLUMN(Phi, phi, float);                  //! Phi angle
DECLARE_SOA_COLUMN(Sign, sign, int8_t);               //! Sign (positive, negative)
} // namespace cftrack
DECLARE_SOA_TABLE(CFTracks, "AOD", "CFTRACK", //! Reduced track table
                  o2::soa::Index<>,
                  cftrack::CFCollisionId,
                  cftrack::Pt, cftrack::Eta, cftrack::Phi,
                  cftrack::Sign, track::TrackType);
DECLARE_SOA_TABLE(CFTrackLabels, "AOD", "CFTRACKLABEL", //! Labels for reduced track table
                  cftrack::CFMcParticleId);
using CFTrack = CFTracks::iterator;
using CFTrackLabel = CFTrackLabels::iterator;
using CFTracksWithLabel = soa::Join<CFTracks, CFTrackLabels>;
using CFTrackWithLabel = CFTracksWithLabel::iterator;

//------transient CF-filter to CF-2prong-filter
DECLARE_SOA_TABLE(CFCollRefs, "AOD", "CFCOLLREF", o2::soa::Index<>, track::CollisionId); //! Transient cf collision index table

//------multiplicity set
namespace cfmultset
{
DECLARE_SOA_COLUMN(Multiplicities, multiplicities, std::vector<float>); //! List of auxiliary multiplicities
enum MultiplicityEstimators : uint8_t {
  CentFT0C = 0x1,
  MultFV0A = 0x2,
  MultNTracksPV = 0x4,
  MultNTracksGlobal = 0x8,
  CentFT0M = 0x10,
};

inline constexpr uint32_t NMultiplicityEstimators = __builtin_ctz(CentFT0M) + 1;

} // namespace cfmultset
DECLARE_SOA_TABLE(CFMultSets, "AOD", "CFMULTSET", cfmultset::Multiplicities); //! Auxilary multiplicity set table

using CFMultSet = CFMultSets::iterator;

// Reco

using CFCollRef = CFCollRefs::iterator;

namespace cftrackref
{
DECLARE_SOA_INDEX_COLUMN(Track, track);
} // namespace cftrackref
DECLARE_SOA_TABLE(CFTrackRefs, "AOD", "CFTRACKREF", o2::soa::Index<>, track::CollisionId, cftrackref::TrackId); //! Transient cf track index table

using CFTrackRef = CFTrackRefs::iterator;

// MC

namespace cfmcparticleref
{
DECLARE_SOA_INDEX_COLUMN(McParticle, mcParticle);
} // namespace cfmcparticleref
DECLARE_SOA_TABLE(CFMcParticleRefs, "AOD", "CFMCPARTICLEREF", o2::soa::Index<>, mcparticle::McCollisionId, cfmcparticleref::McParticleId); //! Transient cf track index table

using CFMcParticleRef = CFMcParticleRefs::iterator;
//------

namespace cf2prongtrack
{
DECLARE_SOA_INDEX_COLUMN_FULL(CFTrackProng0, cfTrackProng0, int, CFTracks, "_0"); //! Index to prong 1 CFTrack
DECLARE_SOA_INDEX_COLUMN_FULL(CFTrackProng1, cfTrackProng1, int, CFTracks, "_1"); //! Index to prong 2 CFTrack
DECLARE_SOA_COLUMN(Pt, pt, float);                                                //! pT (GeV/c)
DECLARE_SOA_COLUMN(Eta, eta, float);                                              //! Pseudorapidity
DECLARE_SOA_COLUMN(Phi, phi, float);                                              //! Phi angle
DECLARE_SOA_COLUMN(InvMass, invMass, float);                                      //! Invariant mass
DECLARE_SOA_COLUMN(Decay, decay, uint8_t);                                        //! Particle decay
enum ParticleDecay {
  D0ToPiK,
  D0barToKPi, // note: often duplicate to D0ToPiK. Choose D0barToKPiExclusive to select uniquely reconstructed D0bars
  JPsiToEE,
  JPsiToMuMu,
  Generic2Prong,
  PhiToKKPID1,
  PhiToKKPID2,
  PhiToKKPID3,
  PhiToKKPID3Loose,
  PhiToKKPID3Tight,
  K0stoPiPi,
  LambdatoPPi,
  AntiLambdatoPiP,
  K0stoPiPiLoose,
  K0stoPiPiTight,
  LambdaToPPiLoose,
  LambdaToPPiTight,
  AntiLambdaToPiPLoose,
  AntiLambdaToPiPTight,
  D0barToKPiExclusive
};
} // namespace cf2prongtrack
DECLARE_SOA_TABLE(CF2ProngTracks, "AOD", "CF2PRONGTRACK", //! Reduced track table
                  o2::soa::Index<>,
                  cftrack::CFCollisionId,
                  cf2prongtrack::CFTrackProng0Id,
                  cf2prongtrack::CFTrackProng1Id,
                  cf2prongtrack::Pt, cf2prongtrack::Eta, cf2prongtrack::Phi, cf2prongtrack::InvMass, cf2prongtrack::Decay);
using CF2ProngTrack = CF2ProngTracks::iterator;
//------

namespace cf2prongtrackml
{
DECLARE_SOA_COLUMN(MlProbD0, mlProbD0, std::vector<float>);       //!
DECLARE_SOA_COLUMN(MlProbD0bar, mlProbD0bar, std::vector<float>); //!
} // namespace cf2prongtrackml
DECLARE_SOA_TABLE(CF2ProngTrackmls, "AOD", "CF2PRONGTRACKML", //! Reduced track table
                  o2::soa::Index<>,
                  cftrack::CFCollisionId,
                  cf2prongtrackml::MlProbD0, cf2prongtrackml::MlProbD0bar);
using CF2ProngTrackml = CF2ProngTrackmls::iterator;
//------

namespace cf2prongmcpart
{
DECLARE_SOA_INDEX_COLUMN_FULL(CFParticleDaugh0, cfParticleDaugh0, int, CFMcParticles, "_0");         //! Index to prong 1 CFMcParticle
DECLARE_SOA_INDEX_COLUMN_FULL(CFParticleDaugh1, cfParticleDaugh1, int, CFMcParticles, "_1");         //! Index to prong 2 CFMcParticle
DECLARE_SOA_COLUMN(Decay, decay, uint8_t);                                                           //! Particle decay and flags
DECLARE_SOA_DYNAMIC_COLUMN(McDecay, mcDecay, [](uint8_t decay) -> uint8_t { return decay & 0x7f; }); //! MC particle decay
enum ParticleDecayFlags {
  Prompt = 0x80
};
} // namespace cf2prongmcpart
DECLARE_SOA_TABLE(CF2ProngMcParts, "AOD", "CF2PRONGMCPART", //! Table for the daughter particles of a 2-prong particle, to be joined with CFMcParticles
                  o2::soa::Index<>,
                  cf2prongmcpart::CFParticleDaugh0Id,
                  cf2prongmcpart::CFParticleDaugh1Id,
                  cf2prongmcpart::Decay,
                  cf2prongmcpart::McDecay<cf2prongmcpart::Decay>)
using CF2ProngMcPart = CF2ProngMcParts::iterator;

} // namespace o2::aod

#endif // PWGCF_DATAMODEL_CORRELATIONSDERIVED_H_