Feat: Add photons to femto framework

Author: ariedel-cern

PWGCF/Femto/Core/dataTypes.h

@@ -35,6 +35,9 @@ using V0MaskType = uint32_t;
 using V0MaskType001 = uint16_t; // old data type, was too narrow
 using V0Type = uint16_t;
 
+// datatypes for photons
+using PhotonMaskType = uint16_t;
+
 // datatypes for kinks
 using KinkMaskType = uint32_t;
 using KinkType = uint8_t;

PWGCF/Femto/Core/partitions.h

@@ -123,6 +123,18 @@
     (o2::aod::femtobase::stored::mass < (selection).massMax) && \
     ncheckbit(o2::aod::femtov0s::mask, (selection).mask)
 
+// partition for photons (PCM)
+// no sign/mass handling: photons are their own antiparticle and have no mass window
+// NOLINTNEXTLINE(cppcoreguidelines-macro-usage)
+#define MAKE_PHOTON_PARTITION(selection)                      \
+  (o2::aod::femtobase::stored::pt > (selection).ptMin) &&     \
+    (o2::aod::femtobase::stored::pt < (selection).ptMax) &&   \
+    (o2::aod::femtobase::stored::eta > (selection).etaMin) && \
+    (o2::aod::femtobase::stored::eta < (selection).etaMax) && \
+    (o2::aod::femtobase::stored::phi > (selection).phiMin) && \
+    (o2::aod::femtobase::stored::phi < (selection).phiMax) && \
+    ncheckbit(o2::aod::femtophotons::mask, (selection).mask)
+
 // NOLINTNEXTLINE(cppcoreguidelines-macro-usage)
 #define MAKE_CASCADE_PARTITION(selection)                                                                                                      \
   ifnode((selection).sign.node() != 0,                                                                                                         \

PWGCF/Femto/Core/photonBuilder.h

@@ -715,7 +715,9 @@ class V0Builder
         v0.v0radius(),
         v0.x(),
         v0.y(),
-        v0.z());
+        v0.z(),
+        v0.alpha(),
+        v0.qtarm());
     }
   }
 
@@ -754,7 +756,9 @@ class V0Builder
         v0.v0radius(),
         v0.x(),
         v0.y(),
-        v0.z());
+        v0.z(),
+        v0.alpha(),
+        v0.qtarm());
     }
   }
 

PWGCF/Femto/Core/photonHistManager.h

@@ -75,6 +75,7 @@ enum V0Hist {
   kK0shortMassVsAntiLambdaMass,
   kStrangeTofVsTofPosDau,
   kStrangeTofVsTofNegDau,
+  kArmenterosQtVsAlpha,
   // mc
   kOrigin,
   kPdg,
@@ -145,6 +146,8 @@ struct ConfV0QaBinning : o2::framework::ConfigurableGroup {
   o2::framework::ConfigurableAxis massAntiLambda{"massAntiLambda", {{100, 1, 1.2}}, "mass for antiparticle hypothesis"};
   o2::framework::ConfigurableAxis massK0short{"massK0short", {{200, 0.45, 0.55}}, "Mass for k0short hypothesis"};
   o2::framework::ConfigurableAxis strangeTof{"strangeTof", {{500, -5, 5}}, "Strangeness TOF vs TOF Nsigma for daughters"};
+  o2::framework::ConfigurableAxis armenterosAlpha{"armenterosAlpha", {{200, -1, 1}}, "Armenteros-Podolanski alpha"};
+  o2::framework::ConfigurableAxis armenterosQt{"armenterosQt", {{150, 0, 0.3}}, "Armenteros-Podolanski qt (GeV/#it{c})"};
 };
 
 constexpr const char PrefixLambdaQaBinning1[] = "LambdaQaBinning1";
@@ -188,6 +191,7 @@ constexpr std::array<histmanager::HistInfo<V0Hist>, kV0HistLast> HistTable = {
    {kLambdaMassVsAntiLambdaMass, o2::framework::HistType::kTH2F, "hLambdaMassVsAntiLambdaMass", "#Lambda mass vs #bar{#Lambda}; m_{p#pi^{-}} (GeV/#it{c}^{2}); m_{#bar{p}#pi^{+}} (GeV/#it{c}^{2})"},
    {kStrangeTofVsTofPosDau, o2::framework::HistType::kTH2F, "hStrangeTofVsTofPosDau", "TOF_{Strange} vs TOF_{Tracking} of positive Daughter; n#sigma_{TOF, strange}; n#sigma_{TOF, tracking}"},
    {kStrangeTofVsTofNegDau, o2::framework::HistType::kTH2F, "hStrangeTofVsTofNegDau", "TOF_{Strange} vs TOF_{Tracking} of negative Daughter; n#sigma_{TOF, strange}; n#sigma_{TOF, tracking}"},
+   {kArmenterosQtVsAlpha, o2::framework::HistType::kTH2F, "hArmenterosPodolanski", "Armenteros-Podolanski; #alpha; q_{T} (GeV/#it{c})"},
    {kOrigin, o2::framework::HistType::kTH1F, "hOrigin", "Status Codes (=Origin); Status Code; Entries"},
    {kPdg, o2::framework::HistType::kTH1F, "hPdg", "PDG Codes of reconstructed v0; PDG Code; Entries"},
    {kPdgMother, o2::framework::HistType::kTH1F, "hPdgMother", "PDG Codes of mother of reconstructed v0; PDG Code; Entries"},
@@ -251,7 +255,8 @@ constexpr std::array<histmanager::HistInfo<V0Hist>, kV0HistLast> HistTable = {
     {kK0shortMassVsLambdaMass, {(confQa).massK0short, (confQa).massLambda}},         \
     {kK0shortMassVsAntiLambdaMass, {(confQa).massK0short, (confQa).massAntiLambda}}, \
     {kStrangeTofVsTofPosDau, {(confQa).strangeTof, (confQa).strangeTof}},            \
-    {kStrangeTofVsTofNegDau, {(confQa).strangeTof, (confQa).strangeTof}},
+    {kStrangeTofVsTofNegDau, {(confQa).strangeTof, (confQa).strangeTof}},            \
+    {kArmenterosQtVsAlpha, {(confQa).armenterosAlpha, (confQa).armenterosQt}},
 
 // NOLINTNEXTLINE(cppcoreguidelines-macro-usage)
 #define V0_HIST_MC_QA_MAP(confAnalysis, confQa)                 \
@@ -518,6 +523,7 @@ class V0HistManager
       mHistogramRegistry->add(qaDir + getHistNameV2(kK0shortMassVsAntiLambdaMass, HistTable), getHistDesc(kK0shortMassVsAntiLambdaMass, HistTable), getHistType(kK0shortMassVsAntiLambdaMass, HistTable), {V0Specs.at(kK0shortMassVsAntiLambdaMass)});
       mHistogramRegistry->add(qaDir + getHistNameV2(kStrangeTofVsTofPosDau, HistTable), getHistDesc(kStrangeTofVsTofPosDau, HistTable), getHistType(kStrangeTofVsTofPosDau, HistTable), {V0Specs.at(kStrangeTofVsTofPosDau)});
       mHistogramRegistry->add(qaDir + getHistNameV2(kStrangeTofVsTofNegDau, HistTable), getHistDesc(kStrangeTofVsTofNegDau, HistTable), getHistType(kStrangeTofVsTofNegDau, HistTable), {V0Specs.at(kStrangeTofVsTofNegDau)});
+      mHistogramRegistry->add(qaDir + getHistNameV2(kArmenterosQtVsAlpha, HistTable), getHistDesc(kArmenterosQtVsAlpha, HistTable), getHistType(kArmenterosQtVsAlpha, HistTable), {V0Specs.at(kArmenterosQtVsAlpha)});
     }
   }
 
@@ -640,6 +646,7 @@ class V0HistManager
       mHistogramRegistry->fill(HIST(v0Prefix) + HIST(QaDir) + HIST(getHistName(kK0shortMassVsAntiLambdaMass, HistTable)), massK0short, massAntiLambda);
       mHistogramRegistry->fill(HIST(v0Prefix) + HIST(QaDir) + HIST(getHistName(kStrangeTofVsTofPosDau, HistTable)), v0candidate.strangeTofPosDau(), tofPosDau);
       mHistogramRegistry->fill(HIST(v0Prefix) + HIST(QaDir) + HIST(getHistName(kStrangeTofVsTofNegDau, HistTable)), v0candidate.strangeTofNegDau(), tofNegDau);
+      mHistogramRegistry->fill(HIST(v0Prefix) + HIST(QaDir) + HIST(getHistName(kArmenterosQtVsAlpha, HistTable)), v0candidate.alpha(), v0candidate.qtArm());
     }
   }
 

PWGCF/Femto/Core/v0Builder.h

@@ -637,6 +637,8 @@ DECLARE_SOA_DYNAMIC_COLUMN(DecayVtx, decayVtx,                 //! distance of d
                            [](float vtxX, float vtxY, float vtxZ) -> float {
                              return std::hypot(vtxX, vtxY, vtxZ);
                            });
+DECLARE_SOA_COLUMN(Alpha, alpha, float); //! Armenteros-Podolanski alpha
+DECLARE_SOA_COLUMN(QtArm, qtArm, float); //! Armenteros-Podolanski qt
 
 // id columns for Lambda daughter tracks
 DECLARE_SOA_INDEX_COLUMN_FULL(PosDau, posDau, int32_t, FTracks, "_PosDau"); //! index column for positive daughter track
@@ -740,6 +742,8 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLambdaExtras_001, "FLAMBDAEXTRA", 1, //! lam
                                    femtov0s::DecayVtxX,
                                    femtov0s::DecayVtxY,
                                    femtov0s::DecayVtxZ,
+                                   femtov0s::Alpha,
+                                   femtov0s::QtArm,
                                    femtov0s::DecayVtx<femtov0s::DecayVtxX, femtov0s::DecayVtxY, femtov0s::DecayVtxZ>,
                                    femtov0s::Pa<femtov0s::CosPa>);
 
@@ -800,11 +804,101 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FK0shortExtras_001, "FK0SHORTEXTRA", 1, //! k
                                    femtov0s::DecayVtxX,
                                    femtov0s::DecayVtxY,
                                    femtov0s::DecayVtxZ,
+                                   femtov0s::Alpha,
+                                   femtov0s::QtArm,
                                    femtov0s::DecayVtx<femtov0s::DecayVtxX, femtov0s::DecayVtxY, femtov0s::DecayVtxZ>,
                                    femtov0s::Pa<femtov0s::CosPa>);
 
 using FK0shortExtras = FK0shortExtras_001;
 
+namespace femtophotons
+{
+// columns for bit masks
+DECLARE_SOA_COLUMN(Mask, mask, o2::analysis::femto::datatypes::PhotonMaskType); //! Bitmask for photon (PCM) selections
+
+// columns for debug information
+DECLARE_SOA_COLUMN(CosPa, cosPa, float);                   //! Cosine of pointing angle of the V0 (conversion) to the primary vertex
+DECLARE_SOA_DYNAMIC_COLUMN(Pa, pa,                         //! pointing angle
+                           [](float cosPa) -> float {
+                             return std::acos(cosPa);
+                           });
+DECLARE_SOA_COLUMN(DcaToPvXY, dcaToPvXY, float);           //! DCAxy of the photon (V0) to the primary vertex
+DECLARE_SOA_COLUMN(DcaToPvZ, dcaToPvZ, float);             //! DCAz of the photon (V0) to the primary vertex
+DECLARE_SOA_COLUMN(Chi2Ndf, chi2Ndf, float);                //! Chi2 / NDF of the reconstructed conversion vertex (KF)
+DECLARE_SOA_COLUMN(V0Radius, v0Radius, float);             //! Transverse radius of the conversion point
+DECLARE_SOA_COLUMN(PosDauTpcNSigmaEl, posDauTpcNSigmaEl, float); //! TPC electron nSigma of the positive daughter
+DECLARE_SOA_COLUMN(NegDauTpcNSigmaEl, negDauTpcNSigmaEl, float); //! TPC electron nSigma of the negative daughter
+DECLARE_SOA_COLUMN(PosDauPt, posDauPt, float);                   //! pT of the positive daughter (at the conversion point)
+DECLARE_SOA_COLUMN(NegDauPt, negDauPt, float);                   //! pT of the negative daughter (at the conversion point)
+DECLARE_SOA_COLUMN(PosDauTpcInnerParam, posDauTpcInnerParam, float); //! momentum of the positive daughter at the inner wall of the TPC
+DECLARE_SOA_COLUMN(NegDauTpcInnerParam, negDauTpcInnerParam, float); //! momentum of the negative daughter at the inner wall of the TPC
+DECLARE_SOA_COLUMN(PosDauTpcSignal, posDauTpcSignal, float);     //! TPC dE/dx of the positive daughter
+DECLARE_SOA_COLUMN(NegDauTpcSignal, negDauTpcSignal, float);     //! TPC dE/dx of the negative daughter
+DECLARE_SOA_COLUMN(DecayVtxX, decayVtxX, float);           //! x coordinate of conversion point
+DECLARE_SOA_COLUMN(DecayVtxY, decayVtxY, float);           //! y coordinate of conversion point
+DECLARE_SOA_COLUMN(DecayVtxZ, decayVtxZ, float);           //! z coordinate of conversion point
+DECLARE_SOA_DYNAMIC_COLUMN(DecayVtx, decayVtx,             //! distance of conversion point from nominal interaction point
+                           [](float vtxX, float vtxY, float vtxZ) -> float {
+                             return std::hypot(vtxX, vtxY, vtxZ);
+                           });
+} // namespace femtophotons
+
+// table for basic PCM photon information (mass is fixed to 0 by construction, not stored)
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FPhotons_001, "FPHOTON", 1, //! femto PCM photons
+                                   o2::soa::Index<>,
+                                   femtobase::stored::FColId,
+                                   femtobase::stored::Pt,
+                                   femtobase::stored::Eta,
+                                   femtobase::stored::Phi,
+                                   femtobase::dynamic::P<femtobase::stored::Pt, femtobase::stored::Eta>,
+                                   femtobase::dynamic::Px<femtobase::stored::Pt, femtobase::stored::Phi>,
+                                   femtobase::dynamic::Py<femtobase::stored::Pt, femtobase::stored::Phi>,
+                                   femtobase::dynamic::Pz<femtobase::stored::Pt, femtobase::stored::Eta>,
+                                   femtobase::dynamic::Theta<femtobase::stored::Eta>);
+using FPhotons = FPhotons_001;
+using StoredFPhotons = StoredFPhotons_001;
+
+// table for basic PCM photon information with compressed/binned kinematics
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FLitePhotons_001, "FLITEPHOTON", 1,
+                                   o2::soa::Index<>,
+                                   femtobase::stored::FLiteColId,
+                                   femtobase::lite::UnsignedBinnedPt,
+                                   femtobase::lite::BinnedEta,
+                                   femtobase::lite::BinnedPhi,
+                                   femtobase::lite::unsignedpt::Pt<femtobase::lite::UnsignedBinnedPt>,
+                                   femtobase::lite::Eta<femtobase::lite::BinnedEta>,
+                                   femtobase::lite::Phi<femtobase::lite::BinnedPhi>);
+using FLitePhotons = FLitePhotons_001;
+using FLitePhoton = FLitePhotons::iterator;
+using StoredFLitePhotons = StoredFLitePhotons_001;
+
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FPhotonMasks_001, "FPHOTONMASK", 1, //! photon (PCM) masks
+                                   femtophotons::Mask);
+using FPhotonMasks = FPhotonMasks_001;
+using StoredFPhotonMasks = StoredFPhotonMasks_001;
+
+DECLARE_SOA_TABLE_STAGED_VERSIONED(FPhotonExtras_001, "FPHOTONEXTRA", 1, //! photon (PCM) extra information
+                                   femtophotons::CosPa,
+                                   femtophotons::DcaToPvXY,
+                                   femtophotons::DcaToPvZ,
+                                   femtophotons::Chi2Ndf,
+                                   femtophotons::V0Radius,
+                                   femtophotons::PosDauPt,
+                                   femtophotons::NegDauPt,
+                                   femtophotons::PosDauTpcInnerParam,
+                                   femtophotons::NegDauTpcInnerParam,
+                                   femtophotons::PosDauTpcSignal,
+                                   femtophotons::NegDauTpcSignal,
+                                   femtophotons::PosDauTpcNSigmaEl,
+                                   femtophotons::NegDauTpcNSigmaEl,
+                                   femtophotons::DecayVtxX,
+                                   femtophotons::DecayVtxY,
+                                   femtophotons::DecayVtxZ,
+                                   femtophotons::DecayVtx<femtophotons::DecayVtxX, femtophotons::DecayVtxY, femtophotons::DecayVtxZ>,
+                                   femtophotons::Pa<femtophotons::CosPa>);
+using FPhotonExtras = FPhotonExtras_001;
+using StoredFPhotonExtras = StoredFPhotonExtras_001;
+
 namespace femtokinks
 {
 // columns for bit masks

PWGCF/Femto/Core/v0HistManager.h

@@ -11,7 +11,7 @@
 
 o2physics_add_dpl_workflow(femto-producer
           SOURCES ./femtoProducer.cxx
-          PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::EventFilteringUtils
+          PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::EventFilteringUtils O2Physics::PWGEMPhotonMesonCore
           COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(femto-producer-derived-to-derived