Skip to content
Merged
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
31 changes: 26 additions & 5 deletions PWGHF/TableProducer/treeCreatorLcToPKPi.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
#include "PWGHF/Core/HfHelper.h"
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
#include "PWGHF/DataModel/CandidateSelectionTables.h"
#include "Common/DataModel/Multiplicity.h"

using namespace o2;
using namespace o2::framework;
Expand Down Expand Up @@ -83,6 +84,11 @@ DECLARE_SOA_INDEX_COLUMN_FULL(Candidate, candidate, int, HfCand3Prong, "_0");
DECLARE_SOA_INDEX_COLUMN(McCollision, mcCollision);
DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
DECLARE_SOA_COLUMN(MultZeqFT0A, multZeqFT0A, float);
DECLARE_SOA_COLUMN(MultZeqFT0C, multZeqFT0C, float);
DECLARE_SOA_COLUMN(MultFT0M, multFT0M, float);
DECLARE_SOA_COLUMN(MultZeqFV0A, multZeqFV0A, float);
DECLARE_SOA_COLUMN(MultZeqNTracksPV, multZeqNTracksPV, float);
} // namespace full

DECLARE_SOA_TABLE(HfCandLcFulls, "AOD", "HFCANDLCFULL",
Expand Down Expand Up @@ -167,7 +173,12 @@ DECLARE_SOA_TABLE(HfCandLcFullEvs, "AOD", "HFCANDLCFULLEV",
collision::PosY,
collision::PosZ,
full::IsEventReject,
full::RunNumber);
full::RunNumber,
full::MultZeqFT0A,
full::MultZeqFT0C,
full::MultFT0M,
full::MultZeqFV0A,
full::MultZeqNTracksPV);

DECLARE_SOA_TABLE(HfCandLcFullPs, "AOD", "HFCANDLCFULLP",
full::McCollisionId,
Expand Down Expand Up @@ -197,7 +208,7 @@ struct HfTreeCreatorLcToPKPi {
{
}

void processMc(soa::Join<aod::Collisions, aod::McCollisionLabels> const& collisions,
void processMc(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::Mults, aod::MultZeqs> const& collisions,
aod::McCollisions const& mcCollisions,
soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec, aod::HfSelLc> const& candidates,
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& particles,
Expand All @@ -215,7 +226,12 @@ struct HfTreeCreatorLcToPKPi {
collision.posY(),
collision.posZ(),
0,
collision.bc().runNumber());
collision.bc().runNumber(),
collision.multZeqFT0A(),
collision.multZeqFT0C(),
collision.multFT0M(),
collision.multZeqFV0A(),
collision.multZeqNTracksPV());
}

// Filling candidate properties
Expand Down Expand Up @@ -330,7 +346,7 @@ struct HfTreeCreatorLcToPKPi {
}
PROCESS_SWITCH(HfTreeCreatorLcToPKPi, processMc, "Process MC tree writer", true);

void processData(aod::Collisions const& collisions,
void processData(soa::Join<aod::Collisions, aod::Mults, aod::MultZeqs> const& collisions,
soa::Join<aod::HfCand3Prong, aod::HfSelLc> const& candidates,
TracksWPid const& tracks, aod::BCs const&)
{
Expand All @@ -346,7 +362,12 @@ struct HfTreeCreatorLcToPKPi {
collision.posY(),
collision.posZ(),
0,
collision.bc().runNumber());
collision.bc().runNumber(),
collision.multZeqFT0A(),
collision.multZeqFT0C(),
collision.multFT0M(),
collision.multZeqFV0A(),
collision.multZeqNTracksPV());
}

// Filling candidate properties
Expand Down