Adding Centrality information in Mc Rec

PWGDQ/DataModel/ReducedTablesAlice3.h → ALICE3/DataModel/ReducedTablesAlice3.h

@@ -15,8 +15,8 @@
 ///
 /// \author Alexander Tiekoetter (atiekoet@cern.ch) University of Muenster
 
-#ifndef PWGDQ_DATAMODEL_REDUCEDTABLESALICE3_H_
-#define PWGDQ_DATAMODEL_REDUCEDTABLESALICE3_H_
+#ifndef ALICE3_DATAMODEL_REDUCEDTABLESALICE3_H_
+#define ALICE3_DATAMODEL_REDUCEDTABLESALICE3_H_
 
 #include "PWGDQ/DataModel/ReducedInfoTables.h"
 
@@ -36,9 +36,9 @@
 namespace reducedeventalice3
 {
 DECLARE_SOA_COLUMN(MultDensity, multDensity, float);
 DECLARE_SOA_COLUMN(MCPosX, mcPosX, float); //!  MC event position X
 DECLARE_SOA_COLUMN(MCPosY, mcPosY, float); //!  MC event position Y
 DECLARE_SOA_COLUMN(MCPosZ, mcPosZ, float); //!  MC event position Z
 } // namespace reducedeventalice3
 
 namespace reducedeventmcalice3
@@ -50,7 +50,7 @@
 DECLARE_SOA_COLUMN(MultMCNParticlesEta40, multMCNParticlesEta40, float);
 } // namespace reducedeventmcalice3
 
-DECLARE_SOA_TABLE_STAGED(ReducedA3Events, "REA3EVENTS", //!   Main event information table
+DECLARE_SOA_TABLE_STAGED(ReA3Events, "REA3EVENT", //!   Main event information table
                          o2::soa::Index<>,
                          collision::PosX, collision::PosY, collision::PosZ, collision::NumContrib,
                          collision::CollisionTime, collision::CollisionTimeRes, reducedeventalice3::MultDensity);
@@ -62,19 +62,19 @@
 DECLARE_SOA_TABLE(ReducedA3EventsInfo, "AOD", "REA3EVENTINFO", //!   Main event index table
                   reducedevent::CollisionId);
 
-DECLARE_SOA_TABLE(ReducedA3MCEvents, "AOD", "REA3MCEVTS", //!   Event level MC truth information
+DECLARE_SOA_TABLE(ReA3MCEvents, "AOD", "REA3MCEVENT", //!   Event level MC truth information
                   o2::soa::Index<>,
                   mccollision::GeneratorsID, reducedeventalice3::MCPosX, reducedeventalice3::MCPosY, reducedeventalice3::MCPosZ,
                   mccollision::T, mccollision::Weight, mccollision::ImpactParameter);
 
-using ReducedA3MCEvent = ReducedA3MCEvents::iterator;
-using ReducedA3Event = ReducedA3Events::iterator;
+using ReducedA3MCEvent = ReA3MCEvents::iterator;
+using ReA3Event = ReA3Events::iterator;
 
 namespace reducedtrackalice3
 {
 // basic track information
-DECLARE_SOA_INDEX_COLUMN(ReducedA3Event, reduceda3event); //!
-DECLARE_SOA_INDEX_COLUMN(Track, track);                   //!
+DECLARE_SOA_INDEX_COLUMN(ReA3Event, rea3event); //!
+DECLARE_SOA_INDEX_COLUMN(Track, track);         //!
 // ----  flags reserved for storing various information during filtering
 DECLARE_SOA_BITMAP_COLUMN(FilteringFlags, filteringFlags, 64); //!
 // -----------------------------------------------------
@@ -111,8 +111,8 @@
 } // namespace reducedtrackalice3
 
 // basic track information
-DECLARE_SOA_TABLE(ReducedA3Tracks, "AOD", "REA3TRACK", //!
-                  o2::soa::Index<>, reducedtrackalice3::ReducedA3EventId, reducedtrackalice3::FilteringFlags,
+DECLARE_SOA_TABLE(ReA3Tracks, "AOD", "REA3TRACK", //!
+                  o2::soa::Index<>, reducedtrackalice3::ReA3EventId, reducedtrackalice3::FilteringFlags,
                   reducedtrackalice3::Pt, reducedtrackalice3::Eta, reducedtrackalice3::Phi, reducedtrackalice3::Sign, reducedtrackalice3::IsAmbiguous,
                   reducedtrackalice3::Px<reducedtrackalice3::Pt, reducedtrackalice3::Phi>,
                   reducedtrackalice3::Py<reducedtrackalice3::Pt, reducedtrackalice3::Phi>,
@@ -126,19 +126,19 @@
 
 namespace reducedA3trackMC
 {
-DECLARE_SOA_INDEX_COLUMN(ReducedA3MCEvent, reducedA3MCEvent);                                 //!
-DECLARE_SOA_COLUMN(McReducedFlags, mcReducedFlags, uint16_t);                                 //! Flags to hold compressed MC selection information
-DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Mother0, mother0, int, "ReducedA3MCTracks_Mother0");       //! Track index of the first mother
-DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Mother1, mother1, int, "ReducedA3MCTracks_Mother1");       //! Track index of the last mother
-DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Daughter0, daughter0, int, "ReducedA3MCTracks_Daughter0"); //! Track index of the first daughter
-DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Daughter1, daughter1, int, "ReducedA3MCTracks_Daughter1"); //! Track index of the last daughter
-DECLARE_SOA_SELF_ARRAY_INDEX_COLUMN(Mothers, mothers);                                        //! Mother tracks (possible empty) array. Iterate over mcParticle.mothers_as<aod::McParticles>())
-DECLARE_SOA_SELF_SLICE_INDEX_COLUMN(Daughters, daughters);                                    //! Daughter tracks (possibly empty) slice. Check for non-zero with mcParticle.has_daughters(). Iterate over mcParticle.daughters_as<aod::McParticles>())
-DECLARE_SOA_COLUMN(Pt, pt, float);                                                            //!
-DECLARE_SOA_COLUMN(Eta, eta, float);                                                          //!
-DECLARE_SOA_COLUMN(Phi, phi, float);                                                          //!
-DECLARE_SOA_COLUMN(E, e, float);                                                              //!
-DECLARE_SOA_DYNAMIC_COLUMN(Px, px,                                                            //!
+DECLARE_SOA_INDEX_COLUMN(ReA3MCEvent, reA3MCEvent);                                      //!
+DECLARE_SOA_COLUMN(McReducedFlags, mcReducedFlags, uint16_t);                            //! Flags to hold compressed MC selection information
+DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Mother0, mother0, int, "ReA3MCTracks_Mother0");       //! Track index of the first mother
+DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Mother1, mother1, int, "ReA3MCTracks_Mother1");       //! Track index of the last mother
+DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Daughter0, daughter0, int, "ReA3MCTracks_Daughter0"); //! Track index of the first daughter
+DECLARE_SOA_SELF_INDEX_COLUMN_FULL(Daughter1, daughter1, int, "ReA3MCTracks_Daughter1"); //! Track index of the last daughter
+DECLARE_SOA_SELF_ARRAY_INDEX_COLUMN(Mothers, mothers);                                   //! Mother tracks (possible empty) array. Iterate over mcParticle.mothers_as<aod::McParticles>())
+DECLARE_SOA_SELF_SLICE_INDEX_COLUMN(Daughters, daughters);                               //! Daughter tracks (possibly empty) slice. Check for non-zero with mcParticle.has_daughters(). Iterate over mcParticle.daughters_as<aod::McParticles>())
+DECLARE_SOA_COLUMN(Pt, pt, float);                                                       //!
+DECLARE_SOA_COLUMN(Eta, eta, float);                                                     //!
+DECLARE_SOA_COLUMN(Phi, phi, float);                                                     //!
+DECLARE_SOA_COLUMN(E, e, float);                                                         //!
+DECLARE_SOA_DYNAMIC_COLUMN(Px, px,                                                       //!
                            [](float pt, float phi) -> float { return pt * std::cos(phi); });
 DECLARE_SOA_DYNAMIC_COLUMN(Py, py, //!
                            [](float pt, float phi) -> float { return pt * std::sin(phi); });
@@ -159,8 +159,8 @@
 
 // NOTE: This table is nearly identical to the one from Framework (except that it points to the event ID, not the BC id)
 //       This table contains all MC truth tracks (both barrel and muon)
-DECLARE_SOA_TABLE(ReducedA3MCTracks, "AOD", "REA3MCTRACK", //!  MC track information (on disk)
-                  o2::soa::Index<>, reducedA3trackMC::ReducedA3MCEventId,
+DECLARE_SOA_TABLE(ReA3MCTracks, "AOD", "REA3MCTRACK", //!  MC track information (on disk)
+                  o2::soa::Index<>, reducedA3trackMC::ReA3MCEventId,
                   mcparticle::PdgCode, mcparticle::StatusCode, mcparticle::Flags,
                   reducedA3trackMC::MothersIds, reducedA3trackMC::DaughtersIdSlice,
                   mcparticle::Weight,
@@ -179,18 +179,18 @@
                   mcparticle::GetHepMCStatusCode<mcparticle::Flags, mcparticle::StatusCode>,
                   mcparticle::IsPhysicalPrimary<mcparticle::Flags>);
 
-using ReducedA3MCTrack = ReducedA3MCTracks::iterator;
+using ReA3MCTrack = ReA3MCTracks::iterator;
 
 namespace reduceda3barreltracklabel
 {
-DECLARE_SOA_INDEX_COLUMN(ReducedA3MCTrack, reducedA3MCTrack); //!
+DECLARE_SOA_INDEX_COLUMN(ReA3MCTrack, reA3MCTrack); //!
 DECLARE_SOA_COLUMN(McMask, mcMask, uint16_t);
 } // namespace reduceda3barreltracklabel
 
 // NOTE: MC labels. This table has one entry for each reconstructed track (joinable with the track tables)
 //          The McParticleId points to the position of the MC truth track from the ReducedTracksMC table
 DECLARE_SOA_TABLE(ReducedA3TracksBarrelLabels, "AOD", "REA3BARLA", //!
-                  reduceda3barreltracklabel::ReducedA3MCTrackId, reduceda3barreltracklabel::McMask, reducedA3trackMC::McReducedFlags);
+                  reduceda3barreltracklabel::ReA3MCTrackId, reduceda3barreltracklabel::McMask, reducedA3trackMC::McReducedFlags);
 
 using ReducedA3TrackBarrelLabel = ReducedA3TracksBarrelLabels::iterator;
 
@@ -206,31 +206,31 @@
 DECLARE_SOA_TABLE(ReducedA3TracksBarrelInfo, "AOD", "REA3BARRELINFO",
                   reducedtrackalice3::CollisionId, collision::PosX, collision::PosY, collision::PosZ, reducedtrackalice3::TrackId);
 
-using ReducedA3Track = ReducedA3Tracks::iterator;
+using ReducedA3Track = ReA3Tracks::iterator;
 using ReducedA3TrackBarrel = ReducedA3TracksBarrel::iterator;
 using ReducedA3TrackBarrelCov = ReducedA3TracksBarrelCov::iterator;
 using ReducedA3TrackBarrelInfo = ReducedA3TracksBarrelInfo::iterator;
 
 namespace reducedeventlabela3
 {
-DECLARE_SOA_INDEX_COLUMN(ReducedA3MCEvent, reducedA3MCEvent); //! MC collision
-DECLARE_SOA_COLUMN(McMask, mcMask, uint16_t);                 //! Bit mask to indicate collision mismatches (bit ON means mismatch). Bit 15: indicates negative label
+DECLARE_SOA_INDEX_COLUMN(ReA3MCEvent, reA3MCEvent); //! MC collision
+DECLARE_SOA_COLUMN(McMask, mcMask, uint16_t);       //! Bit mask to indicate collision mismatches (bit ON means mismatch). Bit 15: indicates negative label
 } // namespace reducedeventlabela3
 
 DECLARE_SOA_TABLE(ReducedA3MCEventLabels, "AOD", "REA3MCCOLLBL", //! Table joined to the ReducedEvents table containing the MC index
-                  reducedeventlabela3::ReducedA3MCEventId, reducedeventlabela3::McMask);
+                  reducedeventlabela3::ReA3MCEventId, reducedeventlabela3::McMask);
 
 using ReducedA3MCEventLabel = ReducedA3MCEventLabels::iterator;
 
 namespace reducedA3track_association
 {
-DECLARE_SOA_INDEX_COLUMN(ReducedA3Event, reducedA3event); //! ReducedEvent index
-DECLARE_SOA_INDEX_COLUMN(ReducedA3Track, reducedA3track); //! ReducedTrack index
+DECLARE_SOA_INDEX_COLUMN(ReA3Event, reA3event); //! ReducedEvent index
+DECLARE_SOA_INDEX_COLUMN(ReA3Track, reA3track); //! ReducedTrack index
 } // namespace reducedA3track_association
 
 DECLARE_SOA_TABLE(ReducedA3TracksAssoc, "AOD", "REA3ASSOC", //! Table for reducedtrack-to-reducedcollision association
-                  reducedA3track_association::ReducedA3EventId,
-                  reducedA3track_association::ReducedA3TrackId);
+                  reducedA3track_association::ReA3EventId,
+                  reducedA3track_association::ReA3TrackId);
 
 DECLARE_SOA_TABLE(ReducedA3PIDTOF, "AOD", "REA3PIDTOF",
                   upgrade_tof::TOFEventTime,
@@ -332,4 +332,4 @@
 
 } // namespace o2::aod
 
-#endif // PWGDQ_DATAMODEL_REDUCEDTABLESALICE3_H_
+#endif // ALICE3_DATAMODEL_REDUCEDTABLESALICE3_H_

ALICE3/TableProducer/CMakeLists.txt

@@ -11,22 +11,22 @@
 
 add_subdirectory(OTF)
 
 o2physics_add_dpl_workflow(alice3-trackselection
                     SOURCES alice3-trackselection.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(alice3-trackextension
                     SOURCES alice3-trackextension.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore O2::ReconstructionDataFormats
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(alice3-centrality
                     SOURCES alice3-centrality.cxx
                     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(alice3-decaypreselector
                     SOURCES alice3-decaypreselector.cxx
                     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2::DCAFitter
                     COMPONENT_NAME Analysis)
@@ -62,7 +62,7 @@
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(alice3-dq-table-maker
-                    SOURCES alice3-dq-table-maker.cxx
+                    SOURCES alice3DqTableMaker.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore O2::DetectorsBase O2Physics::AnalysisCCDB O2Physics::PWGDQCore
                     COMPONENT_NAME Analysis)
 

ALICE3/TableProducer/OTF/onTheFlyDecayer.cxx

@@ -20,6 +20,7 @@
 #include "ALICE3/Core/TrackUtilities.h"
 #include "ALICE3/DataModel/tracksAlice3.h"
 
+#include <CommonConstants/PhysicsConstants.h>
 #include <Framework/AnalysisDataModel.h>
 #include <Framework/AnalysisHelpers.h>
 #include <Framework/AnalysisTask.h>
@@ -47,13 +48,19 @@
 using namespace o2;
 using namespace o2::framework;
 
-static constexpr int NumDecays = 7;
+static constexpr int NumDecays = 13;
 static constexpr int NumParameters = 1;
-static constexpr int DefaultParameters[NumDecays][NumParameters]{{1}, {1}, {1}, {1}, {1}, {1}, {1}};
+static constexpr std::array<std::array<int, NumParameters>, NumDecays> DefaultParameters{{{1}, {1}, {1}, {1}, {1}, {1}, {1}, {1}, {1}, {1}, {1}, {1}, {1}}};
 static const std::vector<std::string> parameterNames{"enable"};
 static const std::vector<std::string> particleNames{"K0s",
                                                     "Lambda",
                                                     "Anti-Lambda",
+                                                    "SigmaPlus",
+                                                    "Anti-SigmaPlus",
+                                                    "SigmaMinus",
+                                                    "Anti-SigmaMinus",
+                                                    "Xi0",
+                                                    "Anti-Xi0",
                                                     "Xi",
                                                     "Anti-Xi",
                                                     "Omega",
@@ -62,6 +69,12 @@ static const std::vector<std::string> particleNames{"K0s",
 static const std::vector<int> pdgCodes{PDG_t::kK0Short,
                                        PDG_t::kLambda0,
                                        PDG_t::kLambda0Bar,
+                                       PDG_t::kSigmaPlus,
+                                       PDG_t::kSigmaBarMinus,
+                                       PDG_t::kSigmaMinus,
+                                       PDG_t::kSigmaBarPlus,
+                                       o2::constants::physics::kXi0,
+                                       -o2::constants::physics::kXi0,
                                        PDG_t::kXiMinus,
                                        PDG_t::kXiPlusBar,
                                        PDG_t::kOmegaMinus,
@@ -78,13 +91,13 @@ struct OnTheFlyDecayer {
   Produces<aod::OTFDecayerBits> tableOTFDecayerBits;
 
   o2::upgrade::Decayer decayer;
-  Service<o2::framework::O2DatabasePDG> pdgDB;
+  Service<o2::framework::O2DatabasePDG> pdgDB{};
   HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};
 
   Configurable<int> seed{"seed", 0, "Set seed for particle decayer"};
   Configurable<float> magneticField{"magneticField", 20., "Magnetic field (kG)"};
   Configurable<LabeledArray<int>> enabledDecays{"enabledDecays",
-                                                {DefaultParameters[0], NumDecays, NumParameters, particleNames, parameterNames},
+                                                {DefaultParameters[0].data(), NumDecays, NumParameters, particleNames, parameterNames},
                                                 "Enable option for particle to be decayed: 0 - no, 1 - yes"};
 
   std::size_t indexOffset = 0;
@@ -99,7 +112,7 @@ struct OnTheFlyDecayer {
     decayer.setSeed(seed);
     decayer.setBField(magneticField);
     for (int i = 0; i < NumDecays; ++i) {
-      if (enabledDecays->get(particleNames[i].c_str(), "enable")) {
+      if (enabledDecays->get(particleNames[i].c_str(), "enable") != 0) {
         LOG(info) << " --- Decay enabled: " << pdgCodes[i];
         mEnabledDecays.push_back(pdgCodes[i]);
       }
@@ -146,9 +159,9 @@ struct OnTheFlyDecayer {
 
       const float decayRadius = decayer.getDecayRadius();
       const float trackVelocity = o2::upgrade::computeParticleVelocity(particle.p(), pdgDB->GetParticle(particle.pdgCode())->Mass());
-      const int charge = pdgDB->GetParticle(particle.pdgCode())->Charge() / 3;
+      const int charge = static_cast<int>(pdgDB->GetParticle(particle.pdgCode())->Charge() / 3);
       float trackLength{-1.f};
-      if (!charge) {
+      if (charge == 0) {
         const float dx = particle.vx() - decayer.getSecondaryVertexX();
         const float dy = particle.vy() - decayer.getSecondaryVertexY();
         const float dz = particle.vz() - decayer.getSecondaryVertexZ();