Skip to content

Commit

Permalink
[PWGLF] Add PID info of daughter tracks in the table (AliceO2Group#8437)
Browse files Browse the repository at this point in the history
  • Loading branch information
@BongHwi
BongHwi authored Nov 15, 2024
1 parent 9c3524c commit 0ac4c85
Show file tree
Hide file tree
Showing 2 changed files with 93 additions and 13 deletions.
74 changes: 61 additions & 13 deletions PWGLF/DataModel/LFResonanceTables.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -123,19 +123,37 @@ DECLARE_SOA_COLUMN(HasITS, hasITS, bool); //! Has ITS
DECLARE_SOA_COLUMN(HasTPC, hasTPC, bool); //! Has TPC
DECLARE_SOA_COLUMN(HasTOF, hasTOF, bool); //! Has TOF
DECLARE_SOA_COLUMN(TPCCrossedRowsOverFindableCls, tpcCrossedRowsOverFindableCls, float);
DECLARE_SOA_COLUMN(DaughDCA, daughDCA, float); //! DCA between daughters
DECLARE_SOA_COLUMN(CascDaughDCA, cascdaughDCA, float); //! DCA between daughters from cascade
DECLARE_SOA_COLUMN(V0CosPA, v0CosPA, float); //! V0 Cosine of Pointing Angle
DECLARE_SOA_COLUMN(CascCosPA, cascCosPA, float); //! Cascade Cosine of Pointing Angle
DECLARE_SOA_COLUMN(MLambda, mLambda, float); //! The invariant mass of V0 candidate, assuming lambda
DECLARE_SOA_COLUMN(MAntiLambda, mAntiLambda, float); //! The invariant mass of V0 candidate, assuming antilambda
DECLARE_SOA_COLUMN(MK0Short, mK0Short, float); //! The invariant mass of V0 candidate, assuming k0s
DECLARE_SOA_COLUMN(MXi, mXi, float); //! The invariant mass of Xi candidate
DECLARE_SOA_COLUMN(TransRadius, transRadius, float); //! Transverse radius of the decay vertex
DECLARE_SOA_COLUMN(CascTransRadius, casctransRadius, float); //! Transverse radius of the decay vertex from cascade
DECLARE_SOA_COLUMN(DecayVtxX, decayVtxX, float); //! X position of the decay vertex
DECLARE_SOA_COLUMN(DecayVtxY, decayVtxY, float); //! Y position of the decay vertex
DECLARE_SOA_COLUMN(DecayVtxZ, decayVtxZ, float); //! Z position of the decay vertex
DECLARE_SOA_COLUMN(DaughDCA, daughDCA, float); //! DCA between daughters
DECLARE_SOA_COLUMN(CascDaughDCA, cascdaughDCA, float); //! DCA between daughters from cascade
DECLARE_SOA_COLUMN(V0CosPA, v0CosPA, float); //! V0 Cosine of Pointing Angle
DECLARE_SOA_COLUMN(CascCosPA, cascCosPA, float); //! Cascade Cosine of Pointing Angle
DECLARE_SOA_COLUMN(MLambda, mLambda, float); //! The invariant mass of V0 candidate, assuming lambda
DECLARE_SOA_COLUMN(MAntiLambda, mAntiLambda, float); //! The invariant mass of V0 candidate, assuming antilambda
DECLARE_SOA_COLUMN(MK0Short, mK0Short, float); //! The invariant mass of V0 candidate, assuming k0s
DECLARE_SOA_COLUMN(MXi, mXi, float); //! The invariant mass of Xi candidate
DECLARE_SOA_COLUMN(TransRadius, transRadius, float); //! Transverse radius of the decay vertex
DECLARE_SOA_COLUMN(CascTransRadius, casctransRadius, float); //! Transverse radius of the decay vertex from cascade
DECLARE_SOA_COLUMN(DecayVtxX, decayVtxX, float); //! X position of the decay vertex
DECLARE_SOA_COLUMN(DecayVtxY, decayVtxY, float); //! Y position of the decay vertex
DECLARE_SOA_COLUMN(DecayVtxZ, decayVtxZ, float); //! Z position of the decay vertex
DECLARE_SOA_COLUMN(DaughterTPCNSigmaPi1, daughterTPCNSigmaPi1, float); //! TPC PID of the first daughter as Pion
DECLARE_SOA_COLUMN(DaughterTPCNSigmaKa1, daughterTPCNSigmaKa1, float); //! TPC PID of the first daughter as Kaon
DECLARE_SOA_COLUMN(DaughterTPCNSigmaPr1, daughterTPCNSigmaPr1, float); //! TPC PID of the first daughter as Proton
DECLARE_SOA_COLUMN(DaughterTPCNSigmaPi2, daughterTPCNSigmaPi2, float); //! TPC PID of the second daughter as Pion
DECLARE_SOA_COLUMN(DaughterTPCNSigmaKa2, daughterTPCNSigmaKa2, float); //! TPC PID of the second daughter as Kaon
DECLARE_SOA_COLUMN(DaughterTPCNSigmaPr2, daughterTPCNSigmaPr2, float); //! TPC PID of the second daughter as Proton
DECLARE_SOA_COLUMN(DaughterTPCNSigmaPiBach, daughterTPCNSigmaPiBach, float); //! TPC PID of the bachelor daughter as Pion
DECLARE_SOA_COLUMN(DaughterTPCNSigmaKaBach, daughterTPCNSigmaKaBach, float); //! TPC PID of the bachelor daughter as Kaon
DECLARE_SOA_COLUMN(DaughterTPCNSigmaPrBach, daughterTPCNSigmaPrBach, float); //! TPC PID of the bachelor daughter as Proton
DECLARE_SOA_COLUMN(DaughterTOFNSigmaPi1, daughterTOFNSigmaPi1, float); //! TOF PID of the first daughter as Pion
DECLARE_SOA_COLUMN(DaughterTOFNSigmaKa1, daughterTOFNSigmaKa1, float); //! TOF PID of the first daughter as Kaon
DECLARE_SOA_COLUMN(DaughterTOFNSigmaPr1, daughterTOFNSigmaPr1, float); //! TOF PID of the first daughter as Proton
DECLARE_SOA_COLUMN(DaughterTOFNSigmaPi2, daughterTOFNSigmaPi2, float); //! TOF PID of the second daughter as Pion
DECLARE_SOA_COLUMN(DaughterTOFNSigmaKa2, daughterTOFNSigmaKa2, float); //! TOF PID of the second daughter as Kaon
DECLARE_SOA_COLUMN(DaughterTOFNSigmaPr2, daughterTOFNSigmaPr2, float); //! TOF PID of the second daughter as Proton
DECLARE_SOA_COLUMN(DaughterTOFNSigmaPiBach, daughterTOFNSigmaPiBach, float); //! TOF PID of the bachelor daughter as Pion
DECLARE_SOA_COLUMN(DaughterTOFNSigmaKaBach, daughterTOFNSigmaKaBach, float); //! TOF PID of the bachelor daughter as Kaon
DECLARE_SOA_COLUMN(DaughterTOFNSigmaPrBach, daughterTOFNSigmaPrBach, float); //! TOF PID of the bachelor daughter as Proton
// For MC
DECLARE_SOA_INDEX_COLUMN(McParticle, mcParticle); //! Index of the corresponding MC particle
DECLARE_SOA_COLUMN(IsPhysicalPrimary, isPhysicalPrimary, bool);
Expand Down Expand Up @@ -200,6 +218,18 @@ DECLARE_SOA_TABLE(ResoV0s, "AOD", "RESOV0S",
resodaughter::Eta,
resodaughter::Phi,
resodaughter::Indices,
resodaughter::DaughterTPCNSigmaPi1,
resodaughter::DaughterTPCNSigmaKa1,
resodaughter::DaughterTPCNSigmaPr1,
resodaughter::DaughterTPCNSigmaPi2,
resodaughter::DaughterTPCNSigmaKa2,
resodaughter::DaughterTPCNSigmaPr2,
resodaughter::DaughterTOFNSigmaPi1,
resodaughter::DaughterTOFNSigmaKa1,
resodaughter::DaughterTOFNSigmaPr1,
resodaughter::DaughterTOFNSigmaPi2,
resodaughter::DaughterTOFNSigmaKa2,
resodaughter::DaughterTOFNSigmaPr2,
resodaughter::V0CosPA,
resodaughter::DaughDCA,
v0data::DCAPosToPV,
Expand All @@ -224,6 +254,24 @@ DECLARE_SOA_TABLE(ResoCascades, "AOD", "RESOCASCADES",
resodaughter::Eta,
resodaughter::Phi,
resodaughter::CascadeIndices,
resodaughter::DaughterTPCNSigmaPi1,
resodaughter::DaughterTPCNSigmaKa1,
resodaughter::DaughterTPCNSigmaPr1,
resodaughter::DaughterTPCNSigmaPi2,
resodaughter::DaughterTPCNSigmaKa2,
resodaughter::DaughterTPCNSigmaPr2,
resodaughter::DaughterTPCNSigmaPiBach,
resodaughter::DaughterTPCNSigmaKaBach,
resodaughter::DaughterTPCNSigmaPrBach,
resodaughter::DaughterTOFNSigmaPi1,
resodaughter::DaughterTOFNSigmaKa1,
resodaughter::DaughterTOFNSigmaPr1,
resodaughter::DaughterTOFNSigmaPi2,
resodaughter::DaughterTOFNSigmaKa2,
resodaughter::DaughterTOFNSigmaPr2,
resodaughter::DaughterTOFNSigmaPiBach,
resodaughter::DaughterTOFNSigmaKaBach,
resodaughter::DaughterTOFNSigmaPrBach,
resodaughter::V0CosPA,
resodaughter::CascCosPA,
resodaughter::DaughDCA,
Expand Down
32 changes: 32 additions & 0 deletions PWGLF/TableProducer/Resonances/LFResonanceInitializer.cxx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@
///
/// \author Bong-Hwi Lim <[email protected]>

#include <string>
#include <vector>
#include "Common/DataModel/PIDResponse.h"
#include "Common/Core/TrackSelection.h"
#include "Common/DataModel/Centrality.h"
Expand Down Expand Up @@ -511,6 +513,18 @@ struct reso2initializer {
v0.eta(),
v0.phi(),
childIDs,
v0.template posTrack_as<TrackType>().tpcNSigmaPi(),
v0.template posTrack_as<TrackType>().tpcNSigmaKa(),
v0.template posTrack_as<TrackType>().tpcNSigmaPr(),
v0.template posTrack_as<TrackType>().tofNSigmaPi(),
v0.template posTrack_as<TrackType>().tofNSigmaKa(),
v0.template posTrack_as<TrackType>().tofNSigmaPr(),
v0.template negTrack_as<TrackType>().tpcNSigmaPi(),
v0.template negTrack_as<TrackType>().tpcNSigmaKa(),
v0.template negTrack_as<TrackType>().tpcNSigmaPr(),
v0.template negTrack_as<TrackType>().tofNSigmaPi(),
v0.template negTrack_as<TrackType>().tofNSigmaKa(),
v0.template negTrack_as<TrackType>().tofNSigmaPr(),
v0.v0cosPA(),
v0.dcaV0daughters(),
v0.dcapostopv(),
Expand Down Expand Up @@ -545,6 +559,24 @@ struct reso2initializer {
casc.eta(),
casc.phi(),
childIDs,
casc.template posTrack_as<TrackType>().tpcNSigmaPi(),
casc.template posTrack_as<TrackType>().tpcNSigmaKa(),
casc.template posTrack_as<TrackType>().tpcNSigmaPr(),
casc.template posTrack_as<TrackType>().tofNSigmaPi(),
casc.template posTrack_as<TrackType>().tofNSigmaKa(),
casc.template posTrack_as<TrackType>().tofNSigmaPr(),
casc.template negTrack_as<TrackType>().tpcNSigmaPi(),
casc.template negTrack_as<TrackType>().tpcNSigmaKa(),
casc.template negTrack_as<TrackType>().tpcNSigmaPr(),
casc.template negTrack_as<TrackType>().tofNSigmaPi(),
casc.template negTrack_as<TrackType>().tofNSigmaKa(),
casc.template negTrack_as<TrackType>().tofNSigmaPr(),
casc.template bachelor_as<TrackType>().tpcNSigmaPi(),
casc.template bachelor_as<TrackType>().tpcNSigmaKa(),
casc.template bachelor_as<TrackType>().tpcNSigmaPr(),
casc.template bachelor_as<TrackType>().tofNSigmaPi(),
casc.template bachelor_as<TrackType>().tofNSigmaKa(),
casc.template bachelor_as<TrackType>().tofNSigmaPr(),
casc.v0cosPA(collision.posX(), collision.posY(), collision.posZ()),
casc.casccosPA(collision.posX(), collision.posY(), collision.posZ()),
casc.dcaV0daughters(),
Expand Down

0 comments on commit 0ac4c85

Please sign in to comment.