diff --git a/PWGHF/TableProducer/treeCreatorToXiPi.cxx b/PWGHF/TableProducer/treeCreatorToXiPi.cxx index b875db47a41..8ad8f6935d6 100644 --- a/PWGHF/TableProducer/treeCreatorToXiPi.cxx +++ b/PWGHF/TableProducer/treeCreatorToXiPi.cxx @@ -15,9 +15,13 @@ /// /// \author Federica Zanone , Heidelberg University +#include "CommonConstants/PhysicsConstants.h" #include "Framework/AnalysisTask.h" #include "Framework/runDataProcessing.h" +#include "Common/DataModel/Centrality.h" +#include "Common/DataModel/Multiplicity.h" + #include "Common/Core/RecoDecay.h" #include "PWGHF/DataModel/CandidateReconstructionTables.h" @@ -31,9 +35,20 @@ namespace o2::aod namespace full { // collision info +DECLARE_SOA_INDEX_COLUMN(Collision, collision); +DECLARE_SOA_INDEX_COLUMN(McCollision, mcCollision); DECLARE_SOA_COLUMN(IsEventSel8, isEventSel8, bool); DECLARE_SOA_COLUMN(IsEventSelZ, isEventSelZ, bool); +DECLARE_SOA_COLUMN(RunNumber, runNumber, int); +DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int); +DECLARE_SOA_COLUMN(CentFT0A, centFT0A, float); +DECLARE_SOA_COLUMN(CentFT0C, centFT0C, float); +DECLARE_SOA_COLUMN(CentFT0M, centFT0M, float); +DECLARE_SOA_COLUMN(CentFV0A, centFV0A, float); +DECLARE_SOA_COLUMN(CentFDDM, centFDDM, float); +DECLARE_SOA_COLUMN(MultZeqNTracksPV, multZeqNTracksPV, float); // from creator +DECLARE_SOA_COLUMN(Cent, cent, float); DECLARE_SOA_COLUMN(XPv, xPv, float); DECLARE_SOA_COLUMN(YPv, yPv, float); DECLARE_SOA_COLUMN(ZPv, zPv, float); @@ -147,9 +162,24 @@ DECLARE_SOA_COLUMN(TofNSigmaPrFromLambda, tofNSigmaPrFromLambda, float); } // namespace full DECLARE_SOA_TABLE(HfToXiPiEvs, "AOD", "HFTOXIPIEV", - full::IsEventSel8, full::IsEventSelZ); + full::IsEventSel8, full::IsEventSelZ, + full::CollisionId, + full::McCollisionId, + collision::NumContrib, + collision::PosX, + collision::PosY, + collision::PosZ, + full::IsEventReject, + full::RunNumber, + full::CentFT0A, + full::CentFT0C, + full::CentFT0M, + full::CentFV0A, + full::CentFDDM, + full::MultZeqNTracksPV); DECLARE_SOA_TABLE(HfToXiPiFulls, "AOD", "HFTOXIPIFULL", + full::CollisionId, full::XPv, full::YPv, full::ZPv, collision::NumContrib, collision::Chi2, full::XDecayVtxCharmBaryon, full::YDecayVtxCharmBaryon, full::ZDecayVtxCharmBaryon, full::XDecayVtxCascade, full::YDecayVtxCascade, full::ZDecayVtxCascade, @@ -219,8 +249,9 @@ struct HfTreeCreatorToXiPi { Configurable zPvCut{"zPvCut", 10., "Cut on absolute value of primary vertex z coordinate"}; + using Cents = soa::Join; + using MyEventTable = soa::Join; using MyTrackTable = soa::Join; - using MyEventTable = soa::Join; void init(InitContext const&) { @@ -232,13 +263,40 @@ struct HfTreeCreatorToXiPi { template void fillEvent(const T& collision, float cutZPv) { - rowEv(collision.sel8(), std::abs(collision.posZ()) < cutZPv); + float centFT0A = -1.f; + float centFT0C = -1.f; + float centFT0M = -1.f; + float centFV0A = -1.f; + float centFDDM = -1.f; + centFT0A = collision.centFT0A(); + centFT0C = collision.centFT0C(); + centFT0M = collision.centFT0M(); + centFV0A = collision.centFV0A(); + centFDDM = collision.centFDDM(); + + rowEv( + collision.sel8(), std::abs(collision.posZ()) < cutZPv, + collision.globalIndex(), + -1, + collision.numContrib(), + collision.posX(), + collision.posY(), + collision.posZ(), + 0, + 1, + centFT0A, + centFT0C, + centFT0M, + centFV0A, + centFDDM, + collision.multZeqNTracksPV()); } template void fillCandidate(const T& candidate, int8_t flagMc, int8_t debugMc, int8_t originMc, bool collisionMatched) { rowCandidateFull( + candidate.collisionId(), candidate.xPv(), candidate.yPv(), candidate.zPv(),