Skip to content

[PWGHF] Add centrality information to Xic0 #9128

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
fgrosa merged 16 commits into from
Jan 23, 2025
Merged

[PWGHF] Add centrality information to Xic0 #9128

Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
fcf5cbf
Add centrality information to Xic0
Tao-Fang Dec 25, 2024
bcb88d4
Update treeCreatorToXiPi.cxx
Tao-Fang Dec 25, 2024
24b83de
Update treeCreatorToXiPi.cxx
Tao-Fang Dec 25, 2024
c4b4a37
Please consider the following formatting changes
alibuild Dec 25, 2024
0421db5
Merge pull request #1 from alibuild/alibot-cleanup-9128
Tao-Fang Dec 25, 2024
127c897
Update whether to use centrality
Tao-Fang Dec 30, 2024
dfd1f60
Update treeCreatorToXiPi.cxx
Tao-Fang Dec 30, 2024
1fa7c23
Update treeCreatorToXiPi.cxx
Tao-Fang Dec 30, 2024
6e6b37d
Update treeCreatorToXiPi.cxx
Tao-Fang Jan 17, 2025
79ba9eb
Update treeCreatorToXiPi.cxx
Tao-Fang Jan 17, 2025
c7606b6
Update treeCreatorToXiPi.cxx
Tao-Fang Jan 17, 2025
63830e4
Update PWGHF/TableProducer/treeCreatorToXiPi.cxx
Tao-Fang Jan 21, 2025
3c3d46a
Update PWGHF/TableProducer/treeCreatorToXiPi.cxx
Tao-Fang Jan 21, 2025
6c19fef
Update PWGHF/TableProducer/treeCreatorToXiPi.cxx
Tao-Fang Jan 21, 2025
0fe7dbc
Update treeCreatorToXiPi.cxx
Tao-Fang Jan 21, 2025
01951ed
Update treeCreatorToXiPi.cxx of bulid
Tao-Fang Jan 23, 2025
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
126 changes: 115 additions & 11 deletions PWGHF/TableProducer/treeCreatorToXiPi.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,13 @@
///
/// \author Federica Zanone <federica.zanone@cern.ch>, 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"
Expand All @@ -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);
Expand Down Expand Up @@ -147,9 +162,23 @@ DECLARE_SOA_COLUMN(TofNSigmaPrFromLambda, tofNSigmaPrFromLambda, float);
} // namespace full

DECLARE_SOA_TABLE(HfToXiPiEvs, "AOD", "HFTOXIPIEV",
full::IsEventSel8, full::IsEventSelZ);
full::IsEventSel8, full::IsEventSelZ,
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,
Expand Down Expand Up @@ -219,8 +248,13 @@ struct HfTreeCreatorToXiPi {

Configurable<float> zPvCut{"zPvCut", 10., "Cut on absolute value of primary vertex z coordinate"};

SliceCache cache;
Preslice<aod::HfCandToXiPi> candXicPerCollision = aod::hf_cand_xic0_omegac0::collisionId;

using Cents = soa::Join<aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs, aod::CentFDDMs>;
using MyTrackTable = soa::Join<aod::Tracks, aod::TrackSelection, aod::TracksExtra>;
using MyEventTable = soa::Join<aod::Collisions, aod::EvSels>;
using MyEventTableWithCent = soa::Join<aod::Collisions, aod::EvSels, aod::PVMultZeqs, Cents>;

void init(InitContext const&)
{
Expand All @@ -229,16 +263,46 @@ struct HfTreeCreatorToXiPi {
}
}

template <typename T>
template <bool useCentrality, typename T>
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;
float multZeqNTracksPV = -1.f;
if constexpr (useCentrality) {
centFT0A = collision.centFT0A();
centFT0C = collision.centFT0C();
centFT0M = collision.centFT0M();
centFV0A = collision.centFV0A();
centFDDM = collision.centFDDM();
multZeqNTracksPV = collision.multZeqNTracksPV();
}

rowEv(
collision.sel8(), std::abs(collision.posZ()) < cutZPv,
-1,
collision.numContrib(),
collision.posX(),
collision.posY(),
collision.posZ(),
0,
1,
centFT0A,
centFT0C,
centFT0M,
centFV0A,
centFDDM,
multZeqNTracksPV);
}

template <typename T>
void fillCandidate(const T& candidate, int8_t flagMc, int8_t debugMc, int8_t originMc, bool collisionMatched)
{
rowCandidateFull(
rowEv.lastIndex(),
candidate.xPv(),
candidate.yPv(),
candidate.zPv(),
Expand Down Expand Up @@ -434,7 +498,7 @@ struct HfTreeCreatorToXiPi {
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
fillEvent(collision, zPvCut);
fillEvent<false>(collision, zPvCut);
}

// Filling candidate properties
Expand All @@ -443,15 +507,35 @@ struct HfTreeCreatorToXiPi {
fillCandidate(candidate, -7, -7, RecoDecay::OriginType::None, false);
}
}
PROCESS_SWITCH(HfTreeCreatorToXiPi, processDataFull, "Process data with full information", true);
PROCESS_SWITCH(HfTreeCreatorToXiPi, processDataFull, "Process data with full information w/o centrality", true);

void processDataFullWithCentrality(MyEventTableWithCent const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi> const& candidates)
{
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
auto thisCollId = collision.globalIndex();
auto groupedXicCandidates = candidates.sliceBy(candXicPerCollision, thisCollId);
auto sizeTableCand = groupedXicCandidates.size();
fillEvent<true>(collision, zPvCut);

// Filling candidate properties
rowCandidateFull.reserve(sizeTableCand);
for (const auto& candidate : groupedXicCandidates) {
fillCandidate(candidate, -7, -7, RecoDecay::OriginType::None, false);
}
}
}
PROCESS_SWITCH(HfTreeCreatorToXiPi, processDataFullWithCentrality, "Process data with full information with centrality", false);

void processMcFullXic0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
{
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
fillEvent(collision, zPvCut);
fillEvent<false>(collision, zPvCut);
}

// Filling candidate properties
Expand All @@ -460,15 +544,35 @@ struct HfTreeCreatorToXiPi {
fillCandidate(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originRec(), candidate.collisionMatched());
}
}
PROCESS_SWITCH(HfTreeCreatorToXiPi, processMcFullXic0, "Process MC with full information for xic0", false);
PROCESS_SWITCH(HfTreeCreatorToXiPi, processMcFullXic0, "Process MC with full information for xic0 w/o centrality", false);

void processMcFullXic0WithCentrality(MyEventTableWithCent const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
{
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
auto thisCollId = collision.globalIndex();
auto groupedXicCandidates = candidates.sliceBy(candXicPerCollision, thisCollId);
auto sizeTableCand = groupedXicCandidates.size();
fillEvent<true>(collision, zPvCut);

// Filling candidate properties
rowCandidateFull.reserve(sizeTableCand);
for (const auto& candidate : groupedXicCandidates) {
fillCandidate(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originRec(), candidate.collisionMatched());
}
}
}
PROCESS_SWITCH(HfTreeCreatorToXiPi, processMcFullXic0WithCentrality, "Process MC with full information for xic0 with centrality", false);

void processMcFullOmegac0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfOmegacToXiPiMCRec> const& candidates)
{
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
fillEvent(collision, zPvCut);
fillEvent<false>(collision, zPvCut);
}

// Filling candidate properties
Expand All @@ -485,7 +589,7 @@ struct HfTreeCreatorToXiPi {
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
fillEvent(collision, zPvCut);
fillEvent<false>(collision, zPvCut);
}

// Filling candidate properties
Expand All @@ -502,7 +606,7 @@ struct HfTreeCreatorToXiPi {
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
fillEvent(collision, zPvCut);
fillEvent<false>(collision, zPvCut);
}

// Filling candidate properties
Expand All @@ -519,7 +623,7 @@ struct HfTreeCreatorToXiPi {
// Filling event properties
rowEv.reserve(collisions.size());
for (const auto& collision : collisions) {
fillEvent(collision, zPvCut);
fillEvent<false>(collision, zPvCut);
}

// Filling candidate properties
Expand Down