Skip to content
Merged
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
58 changes: 29 additions & 29 deletions PWGHF/D2H/Tasks/taskLcToK0sP.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,9 @@
///
/// \note based on taskD0.cxx, taskLc.cxx

#include "PWGHF/Core/CentralityEstimation.h"
#include "PWGHF/Core/HfHelper.h"
#include "PWGHF/Core/SelectorCuts.h"
#include "PWGHF/Core/CentralityEstimation.h"
#include "PWGHF/DataModel/AliasTables.h"
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
#include "PWGHF/DataModel/CandidateSelectionTables.h"
Expand All @@ -37,12 +37,12 @@
#include <Framework/InitContext.h>
#include <Framework/runDataProcessing.h>

#include <THnSparse.h>

#include <cstdlib>
#include <vector>
#include <string>
#include <numeric>

#include <THnSparse.h>
#include <string>
#include <vector>

using namespace o2;
using namespace o2::analysis;
Expand Down Expand Up @@ -258,19 +258,19 @@ struct HfTaskLcToK0sP {
// Data with ML: {mass, pt, centrality, bkg, prompt, non-prompt, numPvContr}
if (doprocessDataWithMl || doprocessDataWithMlWithFT0C || doprocessDataWithMlWithFT0M) {
registry.add("hnLcK0sPDataWithBdt", "THn for Lc->K0sP data with BDT", HistType::kTHnSparseF,
{thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr});
{thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr});
}

// MC Rec with ML: {mass, pt, centrality, bkg, prompt, non-prompt, numPvContr, origin}
if (doprocessMcWithMl || doprocessMcWithMlWithFT0C || doprocessMcWithMlWithFT0M) {
registry.add("hnLcK0sPRecMcWithBdt", "THn for Lc->K0sP MC rec with BDT", HistType::kTHnSparseF,
{thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr, thnAxisOrigin});
{thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr, thnAxisOrigin});
}

// MC Gen: {pt, centrality, rapidity, numPvContr, ptB, origin}
if (doprocessMcWithMl || doprocessMcWithMlWithFT0C || doprocessMcWithMlWithFT0M) {
registry.add("hnLcK0sPGenMc", "THn for Lc->K0sP MC gen", HistType::kTHnSparseF,
{thnAxisPt, thnAxisCentrality, thnAxisY, thnAxisNumPvContr, thnAxisPtB, thnAxisOrigin});
{thnAxisPt, thnAxisCentrality, thnAxisY, thnAxisNumPvContr, thnAxisPtB, thnAxisOrigin});
}
}
}
Expand Down Expand Up @@ -518,38 +518,38 @@ struct HfTaskLcToK0sP {
}

void processDataStd(Collisions const& collisions,
FilteredCandLcToPK0SWSelFlag const& candidates,
TracksWPid const&)
FilteredCandLcToPK0SWSelFlag const& candidates,
TracksWPid const&)
{
runAnalysisPerCollisionData<false>(collisions, candidates);
}
PROCESS_SWITCH(HfTaskLcToK0sP, processDataStd, "Process Data", false);

void processMcStd(CollisionsMc const& collisions,
FilteredCandLcToPK0SWSelFlagAndMc const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
FilteredCandLcToPK0SWSelFlagAndMc const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
{
runAnalysisPerCollisionMc<false>(collisions, candidates, mcParticles);
}
PROCESS_SWITCH(HfTaskLcToK0sP, processMcStd, "Process MC data", false);

// Data with ML
void processDataWithMl(Collisions const& collisions,
FilteredCandLcToPK0SWSelFlagAndMl const& candidates,
TracksWPid const&)
FilteredCandLcToPK0SWSelFlagAndMl const& candidates,
TracksWPid const&)
{
runAnalysisPerCollisionData<true>(collisions, candidates);
}
PROCESS_SWITCH(HfTaskLcToK0sP, processDataWithMl, "Process Data with ML", false);

// MC Rec and Gen with ML
void processMcWithMl(CollisionsMc const& collisions,
FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
{
runAnalysisPerCollisionMc<true>(collisions, candidates, mcParticles);
}
Expand Down Expand Up @@ -604,21 +604,21 @@ struct HfTaskLcToK0sP {

// MC Std with FT0M centrality
void processMcStdWithFT0M(CollisionsMcWithFT0M const& collisions,
FilteredCandLcToPK0SWSelFlagAndMc const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
FilteredCandLcToPK0SWSelFlagAndMc const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
{
runAnalysisPerCollisionMc<false>(collisions, candidates, mcParticles);
}
PROCESS_SWITCH(HfTaskLcToK0sP, processMcStdWithFT0M, "Process MC with FT0M centrality", false);

// MC with ML + FT0C centrality
void processMcWithMlWithFT0C(CollisionsMcWithFT0C const& collisions,
FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates,
soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles,
aod::TracksWMc const&,
TracksWPid const&)
{
runAnalysisPerCollisionMc<true>(collisions, candidates, mcParticles);
}
Expand All @@ -641,4 +641,4 @@ WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
return WorkflowSpec{
adaptAnalysisTask<HfTaskLcToK0sP>(cfgc),
};
}
}