Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
73 changes: 52 additions & 21 deletions PWGLF/DataModel/LFSigmaTables.h
Original file line number Diff line number Diff line change
Expand Up @@ -35,16 +35,11 @@ using std::array;
namespace o2::aod
{

// Indexing
namespace sigma0Core
{
DECLARE_SOA_INDEX_COLUMN_FULL(PhotonV0, photonV0, int, V0Cores, "_PhotonV0"); //!
DECLARE_SOA_INDEX_COLUMN_FULL(LambdaV0, lambdaV0, int, V0Cores, "_LambdaV0"); //!
} // namespace sigma0Core

// for real data
namespace sigma0Core
{
DECLARE_SOA_COLUMN(PhotonV0ID, photonV0ID, int);
DECLARE_SOA_COLUMN(LambdaV0ID, lambdaV0ID, int);
DECLARE_SOA_COLUMN(X, x, float);
DECLARE_SOA_COLUMN(Y, y, float);
DECLARE_SOA_COLUMN(Z, z, float);
Expand Down Expand Up @@ -163,6 +158,9 @@ DECLARE_SOA_DYNAMIC_COLUMN(LambdaPhi, lambdaPhi, //! Phi in the range [0, 2pi)
} // namespace sigma0Core

DECLARE_SOA_TABLE(Sigma0Cores, "AOD", "SIGMA0CORES",
// Indices for debug
sigma0Core::PhotonV0ID, sigma0Core::LambdaV0ID,

// Basic properties
sigma0Core::X, sigma0Core::Y, sigma0Core::Z, sigma0Core::DCADaughters,
sigma0Core::PhotonPx, sigma0Core::PhotonPy, sigma0Core::PhotonPz, sigma0Core::PhotonMass,
Expand Down Expand Up @@ -502,6 +500,42 @@ DECLARE_SOA_TABLE(Sigma0PhotonExtras, "AOD", "SIGMA0PHOTON",
sigma0PhotonExtra::PhotonNegTrackCode,
sigma0PhotonExtra::PhotonV0Type);

// For EMCal Photon extra info
namespace sigma0EMPhoton
{
//______________________________________________________
// REGULAR COLUMNS FOR SIGMA0EMPHOTON
DECLARE_SOA_COLUMN(PhotonID, photonID, int);
DECLARE_SOA_COLUMN(PhotonEnergy, photonEnergy, float);
DECLARE_SOA_COLUMN(PhotonEMCEta, photonEMCEta, float);
DECLARE_SOA_COLUMN(PhotonEMCPhi, photonEMCPhi, float);
DECLARE_SOA_COLUMN(PhotonM02, photonM02, float);
DECLARE_SOA_COLUMN(PhotonM20, photonM20, float);
DECLARE_SOA_COLUMN(PhotonNCells, photonNCells, int);
DECLARE_SOA_COLUMN(PhotonTime, photonTime, float);
DECLARE_SOA_COLUMN(PhotonIsExotic, photonIsExotic, bool);
DECLARE_SOA_COLUMN(PhotonDistToBad, photonDistToBad, float);
DECLARE_SOA_COLUMN(PhotonNLM, photonNLM, int);
DECLARE_SOA_COLUMN(PhotonDefinition, photonDefinition, int);
DECLARE_SOA_COLUMN(PhotonHasAssocTrk, photonHasAssocTrk, bool);

} // namespace sigma0EMPhoton

DECLARE_SOA_TABLE(Sigma0EMPhotons, "AOD", "SIGMA0EMPHOTON",
sigma0EMPhoton::PhotonID,
sigma0EMPhoton::PhotonEnergy,
sigma0EMPhoton::PhotonEMCEta,
sigma0EMPhoton::PhotonEMCPhi,
sigma0EMPhoton::PhotonM02,
sigma0EMPhoton::PhotonM20,
sigma0EMPhoton::PhotonNCells,
sigma0EMPhoton::PhotonTime,
sigma0EMPhoton::PhotonIsExotic,
sigma0EMPhoton::PhotonDistToBad,
sigma0EMPhoton::PhotonNLM,
sigma0EMPhoton::PhotonDefinition,
sigma0EMPhoton::PhotonHasAssocTrk);

// For Lambda extra info
namespace sigma0LambdaExtra
{
Expand Down Expand Up @@ -567,6 +601,7 @@ DECLARE_SOA_TABLE(Sigma0LambdaExtras, "AOD", "SIGMA0LAMBDA",
// for MC
namespace sigma0MCCore
{
DECLARE_SOA_COLUMN(ParticleIdMC, particleIdMC, int); //! V0 Particle ID
DECLARE_SOA_COLUMN(MCradius, mcradius, float);
DECLARE_SOA_COLUMN(PDGCode, pdgCode, int);
DECLARE_SOA_COLUMN(PDGCodeMother, pdgCodeMother, int);
Expand All @@ -576,6 +611,9 @@ DECLARE_SOA_COLUMN(IsProducedByGenerator, isProducedByGenerator, bool);
DECLARE_SOA_COLUMN(PhotonMCPx, photonmcpx, float);
DECLARE_SOA_COLUMN(PhotonMCPy, photonmcpy, float);
DECLARE_SOA_COLUMN(PhotonMCPz, photonmcpz, float);
DECLARE_SOA_COLUMN(PhotonAmplitudeA, photonAmplitudeA, float); // Energy fraction deposited by a particle inside this calo cell.
DECLARE_SOA_COLUMN(PhotonPDGCodePos, photonPDGCodePos, int);
DECLARE_SOA_COLUMN(PhotonPDGCodeNeg, photonPDGCodeNeg, int);
DECLARE_SOA_COLUMN(IsPhotonPrimary, isPhotonPrimary, bool);
DECLARE_SOA_COLUMN(PhotonPDGCode, photonPDGCode, int);
DECLARE_SOA_COLUMN(PhotonPDGCodeMother, photonPDGCodeMother, int);
Expand All @@ -585,6 +623,8 @@ DECLARE_SOA_COLUMN(LambdaMCPx, lambdamcpx, float);
DECLARE_SOA_COLUMN(LambdaMCPy, lambdamcpy, float);
DECLARE_SOA_COLUMN(LambdaMCPz, lambdamcpz, float);
DECLARE_SOA_COLUMN(IsLambdaPrimary, isLambdaPrimary, bool);
DECLARE_SOA_COLUMN(LambdaPDGCodePos, lambdaPDGCodePos, int);
DECLARE_SOA_COLUMN(LambdaPDGCodeNeg, lambdaPDGCodeNeg, int);
DECLARE_SOA_COLUMN(LambdaPDGCode, lambdaPDGCode, int);
DECLARE_SOA_COLUMN(LambdaPDGCodeMother, lambdaPDGCodeMother, int);
DECLARE_SOA_COLUMN(LambdaIsCorrectlyAssoc, lambdaIsCorrectlyAssoc, bool);
Expand Down Expand Up @@ -691,15 +731,17 @@ DECLARE_SOA_DYNAMIC_COLUMN(LambdaMCPhi, lambdaMCPhi, //! Phi in the range [0, 2p
} // namespace sigma0MCCore

DECLARE_SOA_TABLE(Sigma0MCCores, "AOD", "SIGMA0MCCORES",
// MC particle index for debug
sigma0MCCore::ParticleIdMC,

// Basic properties
sigma0MCCore::MCradius, sigma0MCCore::PDGCode, sigma0MCCore::PDGCodeMother, sigma0MCCore::MCprocess, sigma0MCCore::IsProducedByGenerator,

sigma0MCCore::PhotonMCPx, sigma0MCCore::PhotonMCPy, sigma0MCCore::PhotonMCPz,
sigma0MCCore::IsPhotonPrimary, sigma0MCCore::PhotonPDGCode, sigma0MCCore::PhotonPDGCodeMother, sigma0MCCore::PhotonIsCorrectlyAssoc,
sigma0MCCore::PhotonMCPx, sigma0MCCore::PhotonMCPy, sigma0MCCore::PhotonMCPz, sigma0MCCore::PhotonAmplitudeA,
sigma0MCCore::PhotonPDGCodePos, sigma0MCCore::PhotonPDGCodeNeg, sigma0MCCore::IsPhotonPrimary, sigma0MCCore::PhotonPDGCode, sigma0MCCore::PhotonPDGCodeMother, sigma0MCCore::PhotonIsCorrectlyAssoc,

sigma0MCCore::LambdaMCPx, sigma0MCCore::LambdaMCPy, sigma0MCCore::LambdaMCPz,
sigma0MCCore::IsLambdaPrimary, sigma0MCCore::LambdaPDGCode, sigma0MCCore::LambdaPDGCodeMother, sigma0MCCore::LambdaIsCorrectlyAssoc,
sigma0MCCore::LambdaPDGCodePos, sigma0MCCore::LambdaPDGCodeNeg, sigma0MCCore::IsLambdaPrimary, sigma0MCCore::LambdaPDGCode, sigma0MCCore::LambdaPDGCodeMother, sigma0MCCore::LambdaIsCorrectlyAssoc,

// Dynamic columns
sigma0MCCore::IsSigma0<sigma0MCCore::PDGCode>,
Expand Down Expand Up @@ -728,11 +770,6 @@ DECLARE_SOA_TABLE(Sigma0MCCores, "AOD", "SIGMA0MCCORES",
sigma0MCCore::LambdaMCY<sigma0MCCore::LambdaMCPx, sigma0MCCore::LambdaMCPy, sigma0MCCore::LambdaMCPz>,
sigma0MCCore::LambdaMCPhi<sigma0MCCore::LambdaMCPx, sigma0MCCore::LambdaMCPy>);

namespace sigma0MCCore
{
DECLARE_SOA_INDEX_COLUMN(McParticle, mcParticle); //! MC particle for Sigma0
}

// for MC
namespace kstarMCCore
{
Expand Down Expand Up @@ -923,12 +960,6 @@ DECLARE_SOA_TABLE(KStarGens, "AOD", "KSTARGENS",
DECLARE_SOA_TABLE(SigmaCollRef, "AOD", "SIGMACOLLREF", //! optional table to refer back to a collision
o2::soa::Index<>, v0data::StraCollisionId);

DECLARE_SOA_TABLE(SigmaIndices, "AOD", "SIGMAINDEX", //! index table when using AO2Ds
o2::soa::Index<>, sigma0Core::PhotonV0Id, sigma0Core::LambdaV0Id, o2::soa::Marker<1>);

DECLARE_SOA_TABLE(SigmaMCLabels, "AOD", "SIGMAMCLABEL", //! optional table to refer to mcparticles
o2::soa::Index<>, sigma0MCCore::McParticleId);

DECLARE_SOA_TABLE(SigmaGenCollRef, "AOD", "SIGMAGENCOLLREF", //! optional table to refer back to a collision
o2::soa::Index<>, v0data::StraMCCollisionId);

Expand Down
2 changes: 1 addition & 1 deletion PWGLF/TableProducer/Strangeness/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ o2physics_add_dpl_workflow(cascademlselection

o2physics_add_dpl_workflow(sigma0builder
SOURCES sigma0builder.cxx
PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::MLCore O2Physics::AnalysisCCDB
PUBLIC_LINK_LIBRARIES O2::Framework O2::EMCALBase O2Physics::AnalysisCore O2Physics::MLCore O2Physics::AnalysisCCDB
COMPONENT_NAME Analysis)

o2physics_add_dpl_workflow(lambdajetpolarizationbuilder
Expand Down
Loading
Loading