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
22 changes: 9 additions & 13 deletions PWGJE/Core/JetFindingUtilities.h
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
#include <array>
#include <vector>
#include <string>
#include <optional>
#include <cmath>
#include <memory>
#include <TRandom3.h>
Expand Down Expand Up @@ -99,15 +98,14 @@ constexpr bool isEMCALClusterTable()
*/

template <typename T, typename U>
void analyseTracks(std::vector<fastjet::PseudoJet>& inputParticles, T const& tracks, int trackSelection, double trackingEfficinecy, std::optional<U> const& candidate = std::nullopt)
void analyseTracks(std::vector<fastjet::PseudoJet>& inputParticles, T const& tracks, int trackSelection, double trackingEfficinecy, const U* candidate = nullptr)
{
for (auto& track : tracks) {
if (!jetderiveddatautilities::selectTrack(track, trackSelection)) {
continue;
}
if (candidate != std::nullopt) {
auto cand = candidate.value();
if (jetcandidateutilities::isDaughterTrack(track, cand, tracks)) {
if (candidate != nullptr) {
if (jetcandidateutilities::isDaughterTrack(track, *candidate, tracks)) {
continue;
}
}
Expand Down Expand Up @@ -337,7 +335,7 @@ void findJets(JetFinder& jetFinder, std::vector<fastjet::PseudoJet>& inputPartic
* @param candidate optional hf candidiate
*/
template <bool checkIsDaughter, typename T, typename U>
void analyseParticles(std::vector<fastjet::PseudoJet>& inputParticles, std::string particleSelection, int jetTypeParticleLevel, T const& particles, o2::framework::Service<o2::framework::O2DatabasePDG> pdgDatabase, std::optional<U> const& candidate = std::nullopt)
void analyseParticles(std::vector<fastjet::PseudoJet>& inputParticles, std::string particleSelection, int jetTypeParticleLevel, T const& particles, o2::framework::Service<o2::framework::O2DatabasePDG> pdgDatabase, const U* candidate = nullptr)
{
for (auto& particle : particles) {
if (particleSelection == "PhysicalPrimary" && !particle.isPhysicalPrimary()) { // CHECK : Does this exclude the HF hadron?
Expand All @@ -361,23 +359,21 @@ void analyseParticles(std::vector<fastjet::PseudoJet>& inputParticles, std::stri
continue;
}
if constexpr (jetcandidateutilities::isMcCandidate<U>() && !jetv0utilities::isV0McCandidate<U>()) {
if (candidate != std::nullopt) {
auto cand = candidate.value();
if (cand.mcParticleId() == particle.globalIndex()) {
if (candidate != nullptr) {
if ((*candidate).mcParticleId() == particle.globalIndex()) {
continue;
}
if constexpr (checkIsDaughter) {
auto hfParticle = cand.template mcParticle_as<T>();
auto hfParticle = (*candidate).template mcParticle_as<T>();
if (jetcandidateutilities::isDaughterParticle(hfParticle, particle.globalIndex())) {
continue;
}
}
}
}
if constexpr (jetv0utilities::isV0McTable<U>()) { // note that for V0s the candidate table is given to this function, not a single candidate
if (candidate != std::nullopt) {
auto cands = candidate.value();
for (auto const& cand : cands) {
if (candidate != nullptr) {
for (auto const& cand : (*candidate)) {
if (cand.mcParticleId() == particle.globalIndex()) {
continue;
}
Expand Down
6 changes: 3 additions & 3 deletions PWGJE/JetFinders/jetFinderHF.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ struct JetFinderHFTask {
if constexpr (isEvtWiseSub) {
jetfindingutilities::analyseTracks<U, typename U::iterator>(inputParticles, tracks, trackSelection, trackingEfficiency);
} else {
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, trackingEfficiency, &candidate);
}
jetfindingutilities::findJets(jetFinder, inputParticles, minJetPt, maxJetPt, jetRadius, jetAreaFractionMin, collision, jetsTableInput, constituentsTableInput, registry.get<THn>(HIST("hJet")), fillTHnSparse, true);
}
Expand All @@ -212,9 +212,9 @@ struct JetFinderHFTask {
return;
}
if constexpr (checkIsDaughter) {
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, jetTypeParticleLevel, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, jetTypeParticleLevel, particles, pdgDatabase, &candidate);
} else {
jetfindingutilities::analyseParticles<false>(inputParticles, particleSelection, jetTypeParticleLevel, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<false>(inputParticles, particleSelection, jetTypeParticleLevel, particles, pdgDatabase, &candidate);
}
jetfindingutilities::findJets(jetFinder, inputParticles, minJetPt, maxJetPt, jetRadius, jetAreaFractionMin, collision, jetsTable, constituentsTable, registry.get<THn>(HIST("hJetMCP")), fillTHnSparse, true);
}
Expand Down
2 changes: 1 addition & 1 deletion PWGJE/JetFinders/jetFinderV0.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,7 @@ struct JetFinderV0Task {
if (!jetfindingutilities::analyseV0s(inputParticles, candidates, candPtMin, candPtMax, candYMin, candYMax, candIndex)) {
return;
}
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, jetTypeParticleLevel, particles, pdgDatabase, std::optional{candidates});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, jetTypeParticleLevel, particles, pdgDatabase, &candidates);
jetfindingutilities::findJets(jetFinder, inputParticles, minJetPt, maxJetPt, jetRadius, jetAreaFractionMin, collision, jetsTable, constituentsTable, registry.get<THn>(HIST("hJetMCP")), fillTHnSparse, true);
}

Expand Down
4 changes: 2 additions & 2 deletions PWGJE/TableProducer/eventwiseConstituentSubtractor.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ struct eventWiseConstituentSubtractorTask {
for (auto& candidate : candidates) {
inputParticles.clear();
tracksSubtracted.clear();
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, trackingEfficiency, &candidate);

tracksSubtracted = eventWiseConstituentSubtractor.JetBkgSubUtils::doEventConstSub(inputParticles, candidate.rho(), candidate.rhoM());
for (auto const& trackSubtracted : tracksSubtracted) {
Expand All @@ -105,7 +105,7 @@ struct eventWiseConstituentSubtractorTask {
for (auto& candidate : candidates) {
inputParticles.clear();
tracksSubtracted.clear();
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, std::optional{candidate}); // currently only works for charged analyses
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, &candidate); // currently only works for charged analyses

tracksSubtracted = eventWiseConstituentSubtractor.JetBkgSubUtils::doEventConstSub(inputParticles, candidate.rho(), candidate.rhoM());
for (auto const& trackSubtracted : tracksSubtracted) {
Expand Down
20 changes: 10 additions & 10 deletions PWGJE/TableProducer/rhoEstimator.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -218,7 +218,7 @@ struct RhoEstimatorTask {
continue;
}
inputParticles.clear();
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoD0Table(rho, rhoM);
Expand All @@ -230,7 +230,7 @@ struct RhoEstimatorTask {
{
for (auto& candidate : candidates) {
inputParticles.clear();
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoD0McTable(rho, rhoM);
Expand All @@ -246,7 +246,7 @@ struct RhoEstimatorTask {
continue;
}
inputParticles.clear();
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoDplusTable(rho, rhoM);
Expand All @@ -258,7 +258,7 @@ struct RhoEstimatorTask {
{
for (auto& candidate : candidates) {
inputParticles.clear();
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoDplusMcTable(rho, rhoM);
Expand All @@ -274,7 +274,7 @@ struct RhoEstimatorTask {
continue;
}
inputParticles.clear();
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoLcTable(rho, rhoM);
Expand All @@ -286,7 +286,7 @@ struct RhoEstimatorTask {
{
for (auto& candidate : candidates) {
inputParticles.clear();
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoLcMcTable(rho, rhoM);
Expand All @@ -302,7 +302,7 @@ struct RhoEstimatorTask {
continue;
}
inputParticles.clear();
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoBplusTable(rho, rhoM);
Expand All @@ -314,7 +314,7 @@ struct RhoEstimatorTask {
{
for (auto& candidate : candidates) {
inputParticles.clear();
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoBplusMcTable(rho, rhoM);
Expand All @@ -330,7 +330,7 @@ struct RhoEstimatorTask {
continue;
}
inputParticles.clear();
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, std::optional{candidate});
jetfindingutilities::analyseTracks(inputParticles, tracks, trackSelection, config.trackingEfficiency, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoDielectronTable(rho, rhoM);
Expand All @@ -342,7 +342,7 @@ struct RhoEstimatorTask {
{
for (auto& candidate : candidates) {
inputParticles.clear();
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, std::optional{candidate});
jetfindingutilities::analyseParticles<true>(inputParticles, particleSelection, 1, particles, pdgDatabase, &candidate);

auto [rho, rhoM] = bkgSub.estimateRhoAreaMedian(inputParticles, config.doSparse);
rhoDielectronMcTable(rho, rhoM);
Expand Down