[PWGCF] Define + fill correlation tables for both MC gen and reco levels (#15818)

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: abmodak

PWGCF/TwoParticleCorrelations/DataModel/LongRangeDerived.h

@@ -18,16 +18,71 @@
 #ifndef PWGCF_TWOPARTICLECORRELATIONS_DATAMODEL_LONGRANGEDERIVED_H_
 #define PWGCF_TWOPARTICLECORRELATIONS_DATAMODEL_LONGRANGEDERIVED_H_
 
-#include <Framework/ASoA.h>
-#include <Framework/AnalysisDataModel.h>
+#include "Common/DataModel/Multiplicity.h"
 
-#include <cstdint>
+#include "Framework/ASoA.h"
+#include "Framework/AnalysisDataModel.h"
 
 namespace o2::aod
 {
+namespace lrcorrmccolltable
+{
+DECLARE_SOA_COLUMN(Multiplicity, multiplicity, float);
+} // namespace lrcorrmccolltable
+DECLARE_SOA_TABLE(LRMcCollisions, "AOD", "LRMCCOLLISION",
+                  o2::soa::Index<>,
+                  mccollision::PosZ,
+                  lrcorrmccolltable::Multiplicity,
+                  mult::MultMCFT0A,
+                  mult::MultMCFT0C);
+using LRMcCollision = LRMcCollisions::iterator;
+
+namespace lrcorrmctrktable
+{
+DECLARE_SOA_INDEX_COLUMN(LRMcCollision, lrMcCollision);
+DECLARE_SOA_COLUMN(Pt, pt, float);
+DECLARE_SOA_COLUMN(Eta, eta, float);
+DECLARE_SOA_COLUMN(Phi, phi, float);
+} // namespace lrcorrmctrktable
+
+DECLARE_SOA_TABLE(LRMidMcTracks, "AOD", "LRMIDMCTRACK",
+                  o2::soa::Index<>,
+                  lrcorrmctrktable::LRMcCollisionId,
+                  lrcorrmctrktable::Pt,
+                  lrcorrmctrktable::Eta,
+                  lrcorrmctrktable::Phi,
+                  mcparticle::PdgCode,
+                  mcparticle::Flags,
+                  mcparticle::IsPhysicalPrimary<mcparticle::Flags>);
+using LRMidMcTrack = LRMidMcTracks::iterator;
+
+DECLARE_SOA_TABLE(LRFt0aMcTracks, "AOD", "LRFT0AMCTRACK",
+                  o2::soa::Index<>,
+                  lrcorrmctrktable::LRMcCollisionId,
+                  lrcorrmctrktable::Pt,
+                  lrcorrmctrktable::Eta,
+                  lrcorrmctrktable::Phi);
+using LRFt0aMcTrack = LRFt0aMcTracks::iterator;
+
+DECLARE_SOA_TABLE(LRFt0cMcTracks, "AOD", "LRFT0CMCTRACK",
+                  o2::soa::Index<>,
+                  lrcorrmctrktable::LRMcCollisionId,
+                  lrcorrmctrktable::Pt,
+                  lrcorrmctrktable::Eta,
+                  lrcorrmctrktable::Phi);
+using LRFt0cMcTrack = LRFt0cMcTracks::iterator;
+
+DECLARE_SOA_TABLE(LRMftMcTracks, "AOD", "LRMFTMCTRACK",
+                  o2::soa::Index<>,
+                  lrcorrmctrktable::LRMcCollisionId,
+                  lrcorrmctrktable::Pt,
+                  lrcorrmctrktable::Eta,
+                  lrcorrmctrktable::Phi);
+using LRMftMcTrack = LRMftMcTracks::iterator;
+
 namespace lrcorrcolltable
 {
-DECLARE_SOA_COLUMN(Zvtx, zvtx, float);
+DECLARE_SOA_INDEX_COLUMN(LRMcCollision, lrMcCollision);
 DECLARE_SOA_COLUMN(Multiplicity, multiplicity, float);
 DECLARE_SOA_COLUMN(Centrality, centrality, float);
 DECLARE_SOA_COLUMN(TotalFT0AmplitudeA, totalFT0AmplitudeA, float); //! sum of amplitudes on A side of FT0
@@ -39,17 +94,22 @@ DECLARE_SOA_COLUMN(GapSide, gapSide, uint8_t);                     // 0 for side
 DECLARE_SOA_TABLE(LRCollisions, "AOD", "LRCOLLISION",
                   o2::soa::Index<>,
                   bc::RunNumber,
-                  lrcorrcolltable::Zvtx,
+                  collision::PosZ,
                   lrcorrcolltable::Multiplicity,
                   lrcorrcolltable::Centrality,
                   timestamp::Timestamp);
+DECLARE_SOA_TABLE(LRCollLabels, "AOD", "LRCOLLLABEL",
+                  lrcorrcolltable::LRMcCollisionId);
 using LRCollision = LRCollisions::iterator;
+using LRCollLabel = LRCollLabels::iterator;
+using LRCollisionsWithLabel = soa::Join<LRCollisions, LRCollLabels>;
+using LRCollisionWithLabel = LRCollisionsWithLabel::iterator;
 
 DECLARE_SOA_TABLE(UpcLRCollisions, "AOD", "UPCLRCOLLISION",
                   o2::soa::Index<>,
                   bc::GlobalBC,
                   bc::RunNumber,
-                  lrcorrcolltable::Zvtx,
+                  collision::PosZ,
                   lrcorrcolltable::Multiplicity,
                   lrcorrcolltable::TotalFT0AmplitudeA,
                   lrcorrcolltable::TotalFT0AmplitudeC,
@@ -92,7 +152,8 @@ enum TrackPid {
   kSpCharge,
   kSpPion,
   kSpKaon,
-  kSpProton
+  kSpProton,
+  kNoPid
 };
 enum V0TrackPid {
   kSpK0short,
@@ -210,7 +271,6 @@ DECLARE_SOA_TABLE(UpcLRMftBestTracks, "AOD", "UPCLRMFTBESTTRACK",
                   lrcorrtrktable::Eta,
                   lrcorrtrktable::Phi);
 using UpcLRMftBestTrack = UpcLRMftBestTracks::iterator;
-
 } // namespace o2::aod
 
 #endif // PWGCF_TWOPARTICLECORRELATIONS_DATAMODEL_LONGRANGEDERIVED_H_