Feat: add Sigma particles to femto framework

Author: Henrik Fribert

PWGCF/Femto/Core/dataTypes.h

@@ -35,6 +35,10 @@ using TrackType = uint16_t;
 using V0MaskType = uint16_t;
 using V0Type = uint8_t;
 
+// datatypes for kinks
+using KinkMaskType = uint32_t;
+using KinkType = uint8_t;
+
 // datatypes for two track resonances
 using TwoTrackResonanceMaskType = uint32_t;
 // two track resonance types

PWGCF/Femto/Core/kinkBuilder.h

@@ -69,7 +69,8 @@ enum class Track : o2::aod::femtodatatypes::TrackType {
   kPrimaryTrack,
   kV0Daughter,
   kCascadeBachelor,
-  kResonanceDaughter
+  kResonanceDaughter,
+  kKinkDaughter
 };
 
 enum class V0 : o2::aod::femtodatatypes::V0Type {
@@ -78,6 +79,10 @@ enum class V0 : o2::aod::femtodatatypes::V0Type {
   kK0short
 };
 
+enum class Kink : o2::aod::femtodatatypes::KinkType {
+  kSigma
+};
+
 enum class Cascade : o2::aod::femtodatatypes::CascadeType {
   kXi,
   kOmega
@@ -95,14 +100,16 @@ enum class Pairs : o2::aod::femtodatatypes::PairType {
   kTrackTrack,
   kTrackV0,
   kTrackResonance,
-  kTrackCascade
+  kTrackCascade,
+  kTrackKink
 };
 
 enum class TrackPairs : o2::aod::femtodatatypes::PairType {
   kTrackTrack,
   kTrackPosDaughter,
   kTrackNegDaughter,
-  kTrackBachelor
+  kTrackBachelor,
+  kTrackChaDaughter
 };
 
 }; // namespace modes

PWGCF/Femto/Core/kinkHistManager.h

@@ -110,4 +110,16 @@
     (femtobase::stored::mass < selection.massMax) &&                                                         \
     ncheckbit(femtocascades::mask, selection.mask)
 
+#define MAKE_SIGMA_PARTITION(selection)                                                                      \
+  ifnode(selection.sign.node() > 0, femtobase::stored::signedPt > 0.f, femtobase::stored::signedPt < 0.f) && \
+    (nabs(femtobase::stored::signedPt) > selection.ptMin) &&                                                 \
+    (nabs(femtobase::stored::signedPt) < selection.ptMax) &&                                                 \
+    (femtobase::stored::eta > selection.etaMin) &&                                                           \
+    (femtobase::stored::eta < selection.etaMax) &&                                                           \
+    (femtobase::stored::phi > selection.phiMin) &&                                                           \
+    (femtobase::stored::phi < selection.phiMax) &&                                                           \
+    (femtobase::stored::mass > selection.massMin) &&                                                         \
+    (femtobase::stored::mass < selection.massMax) &&                                                         \
+    ncheckbit(femtokinks::mask, selection.mask)
+
 #endif // PWGCF_FEMTO_CORE_PARTITIONS_H_

PWGCF/Femto/Core/modes.h

@@ -129,16 +129,19 @@ constexpr const char PrefixV0NegDauBinning[] = "V0NegDauBinning";
 constexpr const char PrefixCascadePosDauBinning[] = "CascadePosDauBinning";
 constexpr const char PrefixCascadeNegDauBinning[] = "CascadeNegDauBinning";
 constexpr const char PrefixCascadeBachelorBinning[] = "CascadeBachelorBinning";
+constexpr const char PrefixKinkChaDauBinning[] = "KinkChaDauBinning";
 
 using ConfTrackBinning1 = ConfTrackBinning<PrefixTrackBinning1>;
 using ConfTrackBinning2 = ConfTrackBinning<PrefixTrackBinning2>;
 using ConfResonancePosDauBinning = ConfTrackBinning<PrefixResonancePosDauBinning>;
 using ConfResonanceNegDauBinning = ConfTrackBinning<PrefixResonanceNegDauBinning>;
 using ConfV0PosDauBinning = ConfTrackBinning<PrefixV0PosDauBinning>;
 using ConfV0NegDauBinning = ConfTrackBinning<PrefixV0NegDauBinning>;
+using ConfKinkChaDauBinning = ConfTrackBinning<PrefixKinkChaDauBinning>;
 using ConfCascadePosDauBinning = ConfTrackBinning<PrefixCascadePosDauBinning>;
 using ConfCascadeNegDauBinning = ConfTrackBinning<PrefixCascadeNegDauBinning>;
 using ConfCascadeBachelorBinning = ConfTrackBinning<PrefixCascadeBachelorBinning>;
+using ConfKinkChaDauBinning = ConfTrackBinning<PrefixKinkChaDauBinning>;
 
 template <const char* Prefix>
 struct ConfTrackQaBinning : o2::framework::ConfigurableGroup {
@@ -203,6 +206,7 @@ constexpr const char PrefixV0NegDauQaBinning[] = "V0NegDauQaBinning";
 constexpr const char PrefixCascadePosDauQaBinning[] = "CascadePosDauQaBinning";
 constexpr const char PrefixCascadeNegDauQaBinning[] = "CascadeNegDauQaBinning";
 constexpr const char PrefixCascadeBachelorQaBinning[] = "CascadeBachelorQaBinning";
+constexpr const char PrefixKinkChaDauQaBinning[] = "KinkChaDauQaBinning";
 
 using ConfTrackQaBinning1 = ConfTrackQaBinning<PrefixTrackQaBinning1>;
 using ConfTrackQaBinning2 = ConfTrackQaBinning<PrefixTrackQaBinning2>;
@@ -213,6 +217,7 @@ using ConfV0NegDauQaBinning = ConfTrackQaBinning<PrefixV0NegDauQaBinning>;
 using ConfCascadePosDauQaBinning = ConfTrackQaBinning<PrefixCascadePosDauQaBinning>;
 using ConfCascadeNegDauQaBinning = ConfTrackQaBinning<PrefixCascadeNegDauQaBinning>;
 using ConfCascadeBachelorQaBinning = ConfTrackQaBinning<PrefixCascadeBachelorQaBinning>;
+using ConfKinkChaDauQaBinning = ConfTrackQaBinning<PrefixKinkChaDauQaBinning>;
 
 // must be in sync with enum TrackVariables
 // the enum gives the correct index in the array
@@ -374,6 +379,9 @@ constexpr char PrefixCascadePosDaughterQa[] = "CascadePosDauQa/";
 constexpr char PrefixCascadeNegDaughterQa[] = "CascadeNegDauQa/";
 constexpr char PrefixCascadeBachelorQa[] = "CascadeBachelorQa/";
 
+constexpr char PrefixKinkChaDaughter[] = "KinkChaDau/";
+constexpr char PrefixKinkChaDaughterQa[] = "KinkChaDauQa/";
+
 constexpr std::string_view AnalysisDir = "Kinematics/";
 constexpr std::string_view QaDir = "QA/";
 constexpr std::string_view PidDir = "PID/";

PWGCF/Femto/Core/partitions.h

@@ -488,6 +488,57 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FK0shortExtras_001, "FK0SHORTEXTRA", 1, //! k
 
 using FK0shortExtras = FK0shortExtras_001;
 
+namespace femtokinks
+{
+// columns for bit masks
+DECLARE_SOA_COLUMN(Mask, mask, femtodatatypes::KinkMaskType); //! Bitmask for kink selections
+
+// columns for debug information
+DECLARE_SOA_COLUMN(KinkAngle, kinkAngle, float);     //! Kink angle between mother and charged daughter at decay vertex
+DECLARE_SOA_COLUMN(DcaMothToPV, dcaMothToPV, float); //! DCA of the mother track to the primary vertex
+DECLARE_SOA_COLUMN(DcaDaugToPV, dcaDaugToPV, float); //! DCA of the charged daughter track to the primary vertex
+DECLARE_SOA_COLUMN(DecayVtxX, decayVtxX, float);     //! x coordinate of decay vertex (relative to PV)
+DECLARE_SOA_COLUMN(DecayVtxY, decayVtxY, float);     //! y coordinate of decay vertex (relative to PV)
+DECLARE_SOA_COLUMN(DecayVtxZ, decayVtxZ, float);     //! z coordinate of decay vertex (relative to PV)
+DECLARE_SOA_COLUMN(TransRadius, transRadius, float); //! Transverse decay radius from PV
+
+// id column for charged daughter track
+DECLARE_SOA_INDEX_COLUMN_FULL(ChaDau, chaDau, int32_t, FTracks, "_ChaDau"); //!
+} // namespace femtokinks
+
+// table for basic sigma minus information
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FSigmas_001, "FSIGMA", 1,
+                                  o2::soa::Index<>,
+                                  femtobase::stored::CollisionId, // use sign to differentiate between sigma minus (-1) and anti sigma minus (+1)
+                                  femtobase::stored::SignedPt,
+                                  femtobase::stored::Eta,
+                                  femtobase::stored::Phi,
+                                  femtobase::stored::Mass,
+                                  femtokinks::ChaDauId,
+                                  femtobase::dynamic::Sign<femtobase::stored::SignedPt>,
+                                  femtobase::dynamic::Pt<femtobase::stored::SignedPt>,
+                                  femtobase::dynamic::P<femtobase::stored::SignedPt, femtobase::stored::Eta>,
+                                  femtobase::dynamic::Px<femtobase::stored::SignedPt, femtobase::stored::Eta>,
+                                  femtobase::dynamic::Py<femtobase::stored::SignedPt, femtobase::stored::Eta>,
+                                  femtobase::dynamic::Pz<femtobase::stored::SignedPt, femtobase::stored::Eta>,
+                                  femtobase::dynamic::Theta<femtobase::stored::Eta>);
+using FSigmas = FSigmas_001;
+
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FSigmaMasks_001, "FSIGMAMASKS", 1,
+                                  femtokinks::Mask);
+using FSigmaMasks = FSigmaMasks_001;
+
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FSigmaExtras_001, "FSIGMAEXTRAS", 1,
+                                  femtokinks::KinkAngle,
+                                  femtokinks::DcaDaugToPV,
+                                  femtokinks::DcaMothToPV,
+                                  femtokinks::DecayVtxX,
+                                  femtokinks::DecayVtxY,
+                                  femtokinks::DecayVtxZ,
+                                  femtokinks::TransRadius);
+
+using FSigmaExtras = FSigmaExtras_001;
+
 namespace femtocascades
 {
 // columns for cascade bit masks

PWGCF/Femto/Core/trackHistManager.h

@@ -19,7 +19,9 @@
 #include "PWGCF/Femto/Core/trackBuilder.h"
 #include "PWGCF/Femto/Core/twoTrackResonanceBuilder.h"
 #include "PWGCF/Femto/Core/v0Builder.h"
+#include "PWGCF/Femto/Core/kinkBuilder.h"
 #include "PWGLF/DataModel/LFStrangenessTables.h"
+#include "PWGLF/DataModel/LFKinkDecayTables.h"
 
 #include "Common/DataModel/Centrality.h"
 #include "Common/DataModel/EventSelection.h"
@@ -70,6 +72,8 @@ using Run3PpVzeros = V0Datas;
 
 using Run3PpCascades = CascDatas;
 
+using Run3PpKinks = KinkCands;
+
 } // namespace consumeddata
 } // namespace o2::analysis::femto
 
@@ -116,6 +120,13 @@ struct FemtoProducer {
   cascadebuilder::ConfOmegaBits confOmegaBits;
   cascadebuilder::CascadeBuilder<modes::Cascade::kOmega> omegaBuilder;
 
+  // kink builder
+  kinkbuilder::KinkBuilderProducts kinkBuilderProducts;
+  kinkbuilder::ConfKinkTables confKinkTables;
+  kinkbuilder::ConfKinkFilters confKinkFilters;
+  kinkbuilder::ConfSigmaBits confSigmaBits;
+  kinkbuilder::KinkBuilder<modes::Kink::kSigma> sigmaBuilder;
+
   // resonance daughter filters and partitions
   twotrackresonancebuilder::ConfTwoTrackResonanceDaughterFilters confResonanceDaughterFilters;
   // caching and preslicing
@@ -191,6 +202,9 @@ struct FemtoProducer {
     lambdaBuilder.init(confLambdaBits, confV0Filters, confV0Tables, context);
     antilambdaBuilder.init(confLambdaBits, confV0Filters, confV0Tables, context);
 
+    // configure kink builder
+    sigmaBuilder.init(confSigmaBits, confKinkFilters, confKinkTables, context);
+
     // cascade selections
     xiBuilder.init(confXiBits, confCascadeFilters, confCascadeTables, context);
     omegaBuilder.init(confOmegaBits, confCascadeFilters, confCascadeTables, context);
@@ -201,11 +215,14 @@ struct FemtoProducer {
     kstar0Builder.init(confKstar0Bits, confKstarFilters, confResonanceDaughterFilters, confTwoTrackResonanceTables, context);
     kstar0barBuilder.init(confKstar0Bits, confKstarFilters, confResonanceDaughterFilters, confTwoTrackResonanceTables, context);
 
-    if ((xiBuilder.fillAnyTable() || omegaBuilder.fillAnyTable()) && !doprocessTracksV0sCascadesRun3pp) {
-      LOG(fatal) << "At least one cascade tabel is enabled, but wrong process function is enabled. Breaking...";
+    if ((xiBuilder.fillAnyTable() || omegaBuilder.fillAnyTable()) && (!doprocessTracksV0sCascadesRun3pp && !doprocessTracksV0sCascadesKinksRun3pp)) {
+      LOG(fatal) << "At least one cascade table is enabled, but wrong process function is enabled. Breaking...";
+    }
+    if ((lambdaBuilder.fillAnyTable() || antilambdaBuilder.fillAnyTable() || k0shortBuilder.fillAnyTable()) && (!doprocessTracksV0sCascadesRun3pp && !doprocessTracksV0sRun3pp && !doprocessTracksV0sCascadesKinksRun3pp)) {
+      LOG(info) << "At least one v0 table is enabled, but wrong process function is enabled. Breaking...";
     }
-    if ((lambdaBuilder.fillAnyTable() || antilambdaBuilder.fillAnyTable() || k0shortBuilder.fillAnyTable()) && (!doprocessTracksV0sCascadesRun3pp && !doprocessTracksV0sRun3pp)) {
-      LOG(info) << "At least one v0 tabel is enbaled, but wrong process function is enabled. Breaking...";
+    if (sigmaBuilder.fillAnyTable() && !doprocessTracksKinksRun3pp && !doprocessTracksV0sCascadesKinksRun3pp) {
+      LOG(fatal) << "At least one kink table is enabled, but wrong process function is enabled. Breaking...";
     }
   }
 
@@ -244,6 +261,14 @@ struct FemtoProducer {
     k0shortBuilder.fillV0s(collisionBuilderProducts, trackBuilderProducts, v0builderProducts, v0s, tracks, trackBuilder, indexMapTracks);
   }
 
+  // add kinks
+  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5>
+  void processTracksKinks(T1 const& col, T2 const& bcs, T3 const& tracks, T4 const& tracksWithItsPid, T5 const& kinks)
+  {
+    processTracks<system>(col, bcs, tracks, tracksWithItsPid);
+    sigmaBuilder.fillKinks(collisionBuilderProducts, trackBuilderProducts, kinkBuilderProducts, kinks, tracks, trackBuilder, indexMapTracks);
+  }
+
   // add cascades
   template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5, typename T6>
   void processTracksV0sCascades(T1 const& col, T2 const& bcs, T3 const& tracks, T4 const& tracksWithItsPid, T5 const& v0s, T6 const& cascades)
@@ -255,6 +280,18 @@ struct FemtoProducer {
                               cascades, tracks, col, trackBuilder, indexMapTracks);
   }
 
+  // add kinks
+  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5, typename T6, typename T7>
+  void processTracksV0sCascadesKinks(T1 const& col, T2 const& bcs, T3 const& tracks, T4 const& tracksWithItsPid, T5 const& v0s, T6 const& cascades, T7 const& kinks)
+  {
+    processTracksV0s<system>(col, bcs, tracks, tracksWithItsPid, v0s);
+    sigmaBuilder.fillKinks(collisionBuilderProducts, trackBuilderProducts, kinkBuilderProducts, kinks, tracks, trackBuilder, indexMapTracks);
+    xiBuilder.fillCascades(collisionBuilderProducts, trackBuilderProducts, cascadeBuilderProducts,
+                           cascades, tracks, col, trackBuilder, indexMapTracks);
+    omegaBuilder.fillCascades(collisionBuilderProducts, trackBuilderProducts, cascadeBuilderProducts,
+                              cascades, tracks, col, trackBuilder, indexMapTracks);
+  }
+
   // proccess functions
   void processTracksRun3pp(consumeddata::Run3PpCollisions::iterator const& col,
                            BCsWithTimestamps const& bcs,
@@ -280,7 +317,20 @@ struct FemtoProducer {
   };
   PROCESS_SWITCH(FemtoProducer, processTracksV0sRun3pp, "Process tracks and v0s", false);
 
-  // process tracks, v0s and casacades
+  // process tracks and kinks
+  void processTracksKinksRun3pp(consumeddata::Run3PpCollisions::iterator const& col,
+                                BCsWithTimestamps const& bcs,
+                                consumeddata::Run3FullPidTracks const& tracks,
+                                consumeddata::Run3PpKinks const& kinks)
+  {
+    // its pid information is generated dynamically, so we need to add it here
+    auto tracksWithItsPid = o2::soa::Attach<consumeddata::Run3FullPidTracks, pidits::ITSNSigmaEl, pidits::ITSNSigmaPi,
+                                            pidits::ITSNSigmaKa, pidits::ITSNSigmaPr, pidits::ITSNSigmaDe, pidits::ITSNSigmaTr, pidits::ITSNSigmaHe>(tracks);
+    processTracksKinks<modes::System::kPP_Run3>(col, bcs, tracks, tracksWithItsPid, kinks);
+  }
+  PROCESS_SWITCH(FemtoProducer, processTracksKinksRun3pp, "Process tracks and kinks", false);
+
+  // process tracks, v0s and cascades
   void processTracksV0sCascadesRun3pp(consumeddata::Run3PpCollisions::iterator const& col,
                                       BCsWithTimestamps const& bcs,
                                       consumeddata::Run3FullPidTracks const& tracks,
@@ -293,6 +343,21 @@ struct FemtoProducer {
     processTracksV0sCascades<modes::System::kPP_Run3>(col, bcs, tracks, tracksWithItsPid, v0s, cascades);
   }
   PROCESS_SWITCH(FemtoProducer, processTracksV0sCascadesRun3pp, "Provide Tracks, V0s and Cascades for Run3", false);
+
+  // process tracks, v0s, cascades and kinks
+  void processTracksV0sCascadesKinksRun3pp(consumeddata::Run3PpCollisions::iterator const& col,
+                                      BCsWithTimestamps const& bcs,
+                                      consumeddata::Run3FullPidTracks const& tracks,
+                                      consumeddata::Run3PpVzeros const& v0s,
+                                      consumeddata::Run3PpCascades const& cascades,
+                                      consumeddata::Run3PpKinks const& kinks)
+  {
+    // its pid information is generated dynamically, so we need to add it here
+    auto tracksWithItsPid = o2::soa::Attach<consumeddata::Run3FullPidTracks, pidits::ITSNSigmaEl, pidits::ITSNSigmaPi,
+                                            pidits::ITSNSigmaKa, pidits::ITSNSigmaPr, pidits::ITSNSigmaDe, pidits::ITSNSigmaTr, pidits::ITSNSigmaHe>(tracks);
+    processTracksV0sCascadesKinks<modes::System::kPP_Run3>(col, bcs, tracks, tracksWithItsPid, v0s, cascades, kinks);
+  }
+  PROCESS_SWITCH(FemtoProducer, processTracksV0sCascadesKinksRun3pp, "Provide Tracks, V0s and Cascades for Run3", false);
 };
 
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)

PWGCF/Femto/DataModel/FemtoTables.h

@@ -24,6 +24,11 @@ o2physics_add_dpl_workflow(femto-v0-qa
           PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
           COMPONENT_NAME Analysis)
 
+o2physics_add_dpl_workflow(femto-kink-qa
+	  SOURCES femtoKinkQa.cxx
+          PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
+          COMPONENT_NAME Analysis)          
+
 o2physics_add_dpl_workflow(femto-cascade-qa
 	  SOURCES femtoCascadeQa.cxx
           PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore