Skip to content

Please consider the following formatting changes to #17976 #7

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
christianreckziegel merged 1 commit into from
Sep 20, 2026
Merged

Please consider the following formatting changes to #17976 #7

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
51 changes: 25 additions & 26 deletions PWGJE/Tasks/hfFragmentationFunction.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,14 +17,13 @@
/// The task store data relevant to the calculation of hadronization observables radial
/// profile and/or jet momentum fraction for charmed hadrons

#include "PWGHF/Core/DecayChannels.h"
#include "PWGJE/Core/JetDerivedDataUtilities.h"
#include "PWGJE/Core/JetHFUtilities.h"
#include "PWGJE/Core/JetUtilities.h"
#include "PWGJE/DataModel/Jet.h"
#include "PWGJE/DataModel/JetReducedData.h"

#include "PWGHF/Core/DecayChannels.h"

#include "Common/Core/RecoDecay.h"

#include <CommonConstants/MathConstants.h>
Expand Down Expand Up @@ -65,24 +64,24 @@ double deltaPhi(double phi1, double phi2)
return std::abs(dphi);
}

//
//
/// Collision counter selection indexes
///
/// The collision selection is done and stored in multiple steps, for later QA analysis.
/// In order not to hard code which bins should be filled throughout different process
/// function, this namespace with enums is create
namespace collisionSelections
{
enum CollisionSelectionStep {
kMCCollisions = 0, ///< raw mccollisions with no selection, starts with 0
kMCCollisionsZCut, ///< mccollisions with z vtx selection
kMCCollisionsZCutSel8, ///< mccollisions with z vtx and sel8 mc emulated selections
kMCCollisionsZCutSel8HasCollisions, ///< mccollisions with z vtx and sel8 mc emulated selections, with at least one reconstructed collisions
kMCCollisionsZCutSel8SplitCollisions, ///< mccollisions with z vtx and sel8 mc emulated selections, with no split reconstructed collisions
kRecoCollisions, ///< raw reconstructed collisions after previous mccollisions selection
kRecoCollisionsZcut, ///< reconstructed collisions with z vtx selection after previous mccollisions selection
kRecoCollisionsZcutSel8 ///< reconstructed collisions with z vtx and sel8 selections after previous mccollisions selection
};
enum CollisionSelectionStep {
kMCCollisions = 0, ///< raw mccollisions with no selection, starts with 0
kMCCollisionsZCut, ///< mccollisions with z vtx selection
kMCCollisionsZCutSel8, ///< mccollisions with z vtx and sel8 mc emulated selections
kMCCollisionsZCutSel8HasCollisions, ///< mccollisions with z vtx and sel8 mc emulated selections, with at least one reconstructed collisions
kMCCollisionsZCutSel8SplitCollisions, ///< mccollisions with z vtx and sel8 mc emulated selections, with no split reconstructed collisions
kRecoCollisions, ///< raw reconstructed collisions after previous mccollisions selection
kRecoCollisionsZcut, ///< reconstructed collisions with z vtx selection after previous mccollisions selection
kRecoCollisionsZcutSel8 ///< reconstructed collisions with z vtx and sel8 selections after previous mccollisions selection
};
}
// creating table for storing distance data
namespace o2::aod
Expand Down Expand Up @@ -469,7 +468,7 @@ struct HfFragmentationFunction {
if (!hasSelectedCollision) {
continue;
}

// --- begin particle level jets storage
// hf particle level jets associated to same mccollision
const auto mcpJetsPerMCCollision = mcpjets.sliceBy(MCPJetsPerMCCollisionPreslice, mccollision.globalIndex());
Expand Down Expand Up @@ -498,7 +497,7 @@ struct HfFragmentationFunction {
} else if (mcdcand.candidateSelFlag() & BIT(1)) { // CandidateSelFlag == BIT(1) -> selected as HFbar
selectedAs = -1;
}

// store matched particle and detector level data in one single table (calculate angular distance in eta-phi plane on the fly)
matchJetTable(jetutilities::deltaR(mcpjet, mcpcand), mcpjet.pt(), mcpjet.eta(), mcpjet.phi(), mcpjet.template tracks_as<aod::JetParticles>().size() + mcpjet.template candidates_as<TCandidatesMCP>().size(), // particle level jet
mcpcand.pt(), mcpcand.eta(), mcpcand.phi(), mcpcand.y(), (mcpcand.originMcGen() == RecoDecay::OriginType::Prompt), // particle level HF
Expand All @@ -510,11 +509,11 @@ struct HfFragmentationFunction {
} else {
// store matched particle and detector level data in one single table (calculate angular distance in eta-phi plane on the fly)
matchJetTable(jetutilities::deltaR(mcpjet, mcpcand), mcpjet.pt(), mcpjet.eta(), mcpjet.phi(), mcpjet.template tracks_as<aod::JetParticles>().size() + mcpjet.template candidates_as<TCandidatesMCP>().size(), // particle level jet
mcpcand.pt(), mcpcand.eta(), mcpcand.phi(), mcpcand.y(), (mcpcand.originMcGen() == RecoDecay::OriginType::Prompt), // particle level HF
-2, -2, -2, -2, -2, // no detector-level jet found
-2, -2, -2, -2, -2, false, // no detector-level jet found
-2, -2, -2, // no detector-level jet found
-2, -2); // no detector-level jet found
mcpcand.pt(), mcpcand.eta(), mcpcand.phi(), mcpcand.y(), (mcpcand.originMcGen() == RecoDecay::OriginType::Prompt), // particle level HF
-2, -2, -2, -2, -2, // no detector-level jet found
-2, -2, -2, -2, -2, false, // no detector-level jet found
-2, -2, -2, // no detector-level jet found
-2, -2); // no detector-level jet found
}
} // end of mcpjets loop

Expand Down Expand Up @@ -551,12 +550,12 @@ struct HfFragmentationFunction {
}

// store matched particle and detector level data in one single table (calculate angular distance in eta-phi plane on the fly)
matchJetTable(-2, -2, -2, -2, -2, // particle level jet
-2, -2, -2, -2, false, // particle level HF
jetutilities::deltaR(mcdjet, mcdcand), mcdjet.pt(), mcdjet.eta(), mcdjet.phi(), mcdjet.template tracks_as<aod::JetTracks>().size() + mcdjet.template candidates_as<TCandidatesMCD>().size(), // detector level jet
mcdcand.pt(), mcdcand.eta(), mcdcand.phi(), mcdcand.m(), mcdcand.y(), (mcdcand.originMcRec() == RecoDecay::OriginType::Prompt), // detector level HF
mcdcand.mlScores()[0], mcdcand.mlScores()[1], mcdcand.mlScores()[2], // Machine Learning PID scores: background, prompt, non-prompt
static_cast<int>(mcdcand.flagMcMatchRec()), selectedAs); // HF = +1, HFbar = -1, neither = 0
matchJetTable(-2, -2, -2, -2, -2, // particle level jet
-2, -2, -2, -2, false, // particle level HF
jetutilities::deltaR(mcdjet, mcdcand), mcdjet.pt(), mcdjet.eta(), mcdjet.phi(), mcdjet.template tracks_as<aod::JetTracks>().size() + mcdjet.template candidates_as<TCandidatesMCD>().size(), // detector level jet
mcdcand.pt(), mcdcand.eta(), mcdcand.phi(), mcdcand.m(), mcdcand.y(), (mcdcand.originMcRec() == RecoDecay::OriginType::Prompt), // detector level HF
mcdcand.mlScores()[0], mcdcand.mlScores()[1], mcdcand.mlScores()[2], // Machine Learning PID scores: background, prompt, non-prompt
static_cast<int>(mcdcand.flagMcMatchRec()), selectedAs); // HF = +1, HFbar = -1, neither = 0
}
} // end of non-matched detector level jets loop
} // end of collisions loop
Expand Down
Loading