Skip to content
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

Jump to bottom

PWGHF: adapt MC validation task to remove dependence on trackIndexSkimCreator #6971

Merged
merged 3 commits into from
Jul 24, 2024
Merged

PWGHF: adapt MC validation task to remove dependence on trackIndexSkimCreator #6971

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
e99af81
PWGHF: adapt MC validation task to remove dependence on trackIndexSki…
fgrosa Jul 24, 2024
98c8eb2
Please consider the following formatting changes
alibuild Jul 24, 2024
d29b3ce
Merge pull request #89 from alibuild/alibot-cleanup-6971
fgrosa Jul 24, 2024
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
2 changes: 1 addition & 1 deletion PWGHF/Tasks/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ o2physics_add_dpl_workflow(task-mc-gen-pt-rap-shapes

o2physics_add_dpl_workflow(task-mc-validation
SOURCES taskMcValidation.cxx
PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::EventFilteringUtils
COMPONENT_NAME Analysis)

# o2physics_add_dpl_workflow(task-sel-optimisation
Expand Down
119 changes: 103 additions & 16 deletions PWGHF/Tasks/taskMcValidation.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,16 +24,19 @@
#include "Framework/runDataProcessing.h"
#include "Framework/StaticFor.h"

#include "CCDB/BasicCCDBManager.h"
#include "Common/DataModel/CollisionAssociationTables.h"

#include "PWGHF/DataModel/CandidateReconstructionTables.h"
#include "PWGHF/DataModel/CandidateSelectionTables.h"
#include "PWGHF/Utils/utilsEvSelHf.h"

using namespace o2;
using namespace o2::analysis;
using namespace o2::aod;
using namespace o2::framework;
using namespace o2::framework::expressions;
using namespace o2::hf_evsel;

namespace
{
Expand Down Expand Up @@ -393,12 +396,15 @@ struct HfTaskMcValidationRec {

using HfCand2ProngWithMCRec = soa::Join<aod::HfCand2Prong, aod::HfCand2ProngMcRec>;
using HfCand3ProngWithMCRec = soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec>;
using CollisionsWithMCLabels = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::HfSelCollision>;
using CollisionsWithMCLabels = soa::Join<aod::Collisions, aod::EvSels, aod::McCollisionLabels>;
using TracksWithSel = soa::Join<aod::TracksWMc, aod::TracksExtra, aod::TrackSelection, aod::TrackCompColls>;

Partition<TracksWithSel> tracksFilteredGlobalTrackWoDCA = requireGlobalTrackWoDCAInFilter();
Partition<TracksWithSel> tracksInAcc = requireTrackCutInFilter(TrackSelectionFlags::kInAcceptanceTracks);

Service<o2::ccdb::BasicCCDBManager> ccdb;
HfEventSelection hfEvSel; // event selection and monitoring

AxisSpec axisDeltaMom{2000, -1., 1.};
AxisSpec axisOrigin{4, -1.5, 2.5};
AxisSpec axisEta{40, -1., 1.};
Expand Down Expand Up @@ -478,6 +484,16 @@ struct HfTaskMcValidationRec {

void init(InitContext&)
{
std::array<bool, 3> procCollisions = {doprocessColl, doprocessCollWithCentFTOC, doprocessCollWithCentFTOM};
if (std::accumulate(procCollisions.begin(), procCollisions.end(), 0) > 1) {
LOGP(fatal, "At most one process function for collision study can be enabled at a time.");
}

std::array<bool, 3> procCollAccoc = {doprocessCollAssoc, doprocessCollAssocWithCentFTOC, doprocessCollAssocWithCentFTOM};
if (std::accumulate(procCollAccoc.begin(), procCollAccoc.end(), 0) > 1) {
LOGP(fatal, "At most one process for collision association study function can be enabled at a time.");
}

histOriginTracks[0] = registry.add<THnSparse>("TrackToCollChecks/histOriginNonAssociatedTracks", ";origin;#it{p}_{T}^{reco} (GeV/#it{c});#it{#eta}^{reco};#it{Z}_{vtx}^{reco}#minus#it{Z}_{vtx}^{gen} (cm); is PV contributor; has TOF; number of ITS hits", HistType::kTHnSparseF, {axisOrigin, axisPt, axisEta, axisDeltaVtx, axisDecision, axisDecision, axisITShits}); // tracks not associated to any collision
histOriginTracks[1] = registry.add<THnSparse>("TrackToCollChecks/histOriginAssociatedTracks", ";origin;#it{p}_{T}^{reco} (GeV/#it{c});#it{#eta}^{reco};#it{Z}_{vtx}^{reco}#minus#it{Z}_{vtx}^{gen} (cm); is PV contributor; has TOF; number of ITS hits", HistType::kTHnSparseF, {axisOrigin, axisPt, axisEta, axisDeltaVtx, axisDecision, axisDecision, axisITShits}); // tracks associasted to a collision
histOriginTracks[2] = registry.add<THnSparse>("TrackToCollChecks/histOriginGoodAssociatedTracks", ";origin;#it{p}_{T}^{reco} (GeV/#it{c});#it{#eta}^{reco};#it{Z}_{vtx}^{reco}#minus#it{Z}_{vtx}^{gen} (cm); is PV contributor; has TOF; number of ITS hits", HistType::kTHnSparseF, {axisOrigin, axisPt, axisEta, axisDeltaVtx, axisDecision, axisDecision, axisITShits}); // tracks associated to the correct collision considering only first reco collision (based on the MC collision index)
Expand Down Expand Up @@ -519,19 +535,31 @@ struct HfTaskMcValidationRec {
histContributors = registry.add<TH1>("TrackToCollChecks/histContributors", "PV contributors from correct/wrong MC collision;;entries", HistType::kTH1F, {axisDecision});
histContributors->GetXaxis()->SetBinLabel(1, "correct MC collision");
histContributors->GetXaxis()->SetBinLabel(2, "wrong MC collision");
hfEvSel.addHistograms(registry); // collision monitoring

ccdb->setURL("http://alice-ccdb.cern.ch");
ccdb->setCaching(true);
ccdb->setLocalObjectValidityChecking();
}

void process(CollisionsWithMCLabels::iterator const& collision,
aod::McCollisions const&)
template <o2::hf_centrality::CentralityEstimator centEstimator, typename Coll>
void checkCollisions(Coll const& collision,
aod::McCollisions const&,
aod::BCsWithTimestamps const&)
{
// check that collision is selected by hf-track-index-skim-creator-tag-sel-collisions
if (collision.whyRejectColl() != 0) {
// apply event selection
if (!collision.has_mcCollision()) {
return;
}
if (!collision.has_mcCollision()) {

float centrality{-1.f};
const auto rejectionMask = hfEvSel.getHfCollisionRejectionMask<true, centEstimator, aod::BCsWithTimestamps>(collision, centrality, ccdb);
if (rejectionMask != 0) {
/// at least one event selection not satisfied --> reject the candidate
return;
}
auto mcCollision = collision.mcCollision_as<aod::McCollisions>();

auto mcCollision = collision.template mcCollision_as<aod::McCollisions>();
if (eventGeneratorType >= 0 && mcCollision.getSubGeneratorId() != eventGeneratorType) {
return;
}
Expand All @@ -540,24 +568,30 @@ struct HfTaskMcValidationRec {
registry.fill(HIST("histYvtxReco"), collision.posY());
registry.fill(HIST("histZvtxReco"), collision.posZ());
registry.fill(HIST("histDeltaZvtx"), collision.numContrib(), collision.posZ() - mcCollision.posZ());
} // end process
}

void processCollAssoc(CollisionsWithMCLabels const& collisions,
TracksWithSel const&,
aod::McParticles const& mcParticles,
aod::McCollisions const&,
aod::BCs const&)
template <o2::hf_centrality::CentralityEstimator centEstimator, typename Colls>
void checkCollisionAssociation(Colls const& collisions,
TracksWithSel const&,
aod::McParticles const& mcParticles,
aod::McCollisions const&,
aod::BCsWithTimestamps const&)
{
// loop over collisions
for (auto collision = collisions.begin(); collision != collisions.end(); ++collision) {
for (const auto& collision : collisions) {
// check that collision is selected by hf-track-index-skim-creator-tag-sel-collisions
if (collision.whyRejectColl() != 0) {

float centrality{-1.f};
const auto rejectionMask = hfEvSel.getHfCollisionRejectionMask<true, centEstimator, aod::BCsWithTimestamps>(collision, centrality, ccdb);
if (rejectionMask != 0) {
/// at least one event selection not satisfied --> reject the candidate
continue;
}

if (!collision.has_mcCollision()) {
continue;
}
auto mcCollision = collision.mcCollision_as<aod::McCollisions>();
auto mcCollision = collision.template mcCollision_as<aod::McCollisions>();
if (eventGeneratorType >= 0 && mcCollision.getSubGeneratorId() != eventGeneratorType) {
continue;
}
Expand Down Expand Up @@ -678,8 +712,61 @@ struct HfTaskMcValidationRec {
}
}
}

void processColl(CollisionsWithMCLabels::iterator const& collision,
aod::McCollisions const& mcCollisions,
aod::BCsWithTimestamps const& bcs)
{
checkCollisions<o2::hf_centrality::CentralityEstimator::None>(collision, mcCollisions, bcs);
} // end process
PROCESS_SWITCH(HfTaskMcValidationRec, processColl, "Process collision information without centrality selection", true);

void processCollWithCentFTOC(soa::Join<CollisionsWithMCLabels, aod::CentFT0Cs>::iterator const& collision,
aod::McCollisions const& mcCollisions,
aod::BCsWithTimestamps const& bcs)
{
checkCollisions<o2::hf_centrality::CentralityEstimator::FT0C>(collision, mcCollisions, bcs);
} // end process
PROCESS_SWITCH(HfTaskMcValidationRec, processCollWithCentFTOC, "Process collision information with centrality selection with FT0C", false);

void processCollWithCentFTOM(soa::Join<CollisionsWithMCLabels, aod::CentFT0Ms>::iterator const& collision,
aod::McCollisions const& mcCollisions,
aod::BCsWithTimestamps const& bcs)
{
checkCollisions<o2::hf_centrality::CentralityEstimator::FT0M>(collision, mcCollisions, bcs);
} // end process
PROCESS_SWITCH(HfTaskMcValidationRec, processCollWithCentFTOM, "Process collision information with centrality selection with FT0M", false);

void processCollAssoc(CollisionsWithMCLabels const& collisions,
TracksWithSel const& tracks,
aod::McParticles const& mcParticles,
aod::McCollisions const& mcCollisions,
aod::BCsWithTimestamps const& bcs)
{
checkCollisionAssociation<o2::hf_centrality::CentralityEstimator::None>(collisions, tracks, mcParticles, mcCollisions, bcs);
}
PROCESS_SWITCH(HfTaskMcValidationRec, processCollAssoc, "Process collision-association information, requires extra table from TrackToCollisionAssociation task (fillTableOfCollIdsPerTrack=true)", false);

void processCollAssocWithCentFTOC(soa::Join<CollisionsWithMCLabels, aod::CentFT0Cs> const& collisions,
TracksWithSel const& tracks,
aod::McParticles const& mcParticles,
aod::McCollisions const& mcCollisions,
aod::BCsWithTimestamps const& bcs)
{
checkCollisionAssociation<o2::hf_centrality::CentralityEstimator::FT0C>(collisions, tracks, mcParticles, mcCollisions, bcs);
}
PROCESS_SWITCH(HfTaskMcValidationRec, processCollAssocWithCentFTOC, "Process collision-association information with centrality selection with FT0C, requires extra table from TrackToCollisionAssociation task (fillTableOfCollIdsPerTrack=true)", false);

void processCollAssocWithCentFTOM(soa::Join<CollisionsWithMCLabels, aod::CentFT0Ms> const& collisions,
TracksWithSel const& tracks,
aod::McParticles const& mcParticles,
aod::McCollisions const& mcCollisions,
aod::BCsWithTimestamps const& bcs)
{
checkCollisionAssociation<o2::hf_centrality::CentralityEstimator::FT0M>(collisions, tracks, mcParticles, mcCollisions, bcs);
}
PROCESS_SWITCH(HfTaskMcValidationRec, processCollAssocWithCentFTOM, "Process collision-association information with centrality selection with FT0M, requires extra table from TrackToCollisionAssociation task (fillTableOfCollIdsPerTrack=true)", false);

void processEff(HfCand2ProngWithMCRec const& cand2Prongs,
HfCand3ProngWithMCRec const& cand3Prongs,
aod::TracksWMc const&,
Expand Down
Loading