1818// /
1919// / \note based on taskD0.cxx, taskLc.cxx
2020
21+ #include " PWGHF/Core/CentralityEstimation.h"
2122#include " PWGHF/Core/HfHelper.h"
2223#include " PWGHF/Core/SelectorCuts.h"
23- #include " PWGHF/Core/CentralityEstimation.h"
2424#include " PWGHF/DataModel/AliasTables.h"
2525#include " PWGHF/DataModel/CandidateReconstructionTables.h"
2626#include " PWGHF/DataModel/CandidateSelectionTables.h"
3737#include < Framework/InitContext.h>
3838#include < Framework/runDataProcessing.h>
3939
40+ #include < THnSparse.h>
41+
4042#include < cstdlib>
41- #include < vector>
4243#include < numeric>
43-
44- #include < THnSparse.h>
44+ #include < vector>
4545
4646using namespace o2 ;
4747using namespace o2 ::analysis;
@@ -256,19 +256,19 @@ struct HfTaskLcToK0sP {
256256 // Data with ML: {mass, pt, centrality, bkg, prompt, non-prompt, numPvContr}
257257 if (doprocessDataWithMl || doprocessDataWithMlWithFT0C || doprocessDataWithMlWithFT0M) {
258258 registry.add (" hnLcK0sPDataWithBdt" , " THn for Lc->K0sP data with BDT" , HistType::kTHnSparseF ,
259- {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr});
259+ {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr});
260260 }
261261
262262 // MC Rec with ML: {mass, pt, centrality, bkg, prompt, non-prompt, numPvContr, origin}
263263 if (doprocessMcWithMl || doprocessMcWithMlWithFT0C || doprocessMcWithMlWithFT0M) {
264264 registry.add (" hnLcK0sPRecMcWithBdt" , " THn for Lc->K0sP MC rec with BDT" , HistType::kTHnSparseF ,
265- {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr, thnAxisOrigin});
265+ {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr, thnAxisOrigin});
266266 }
267267
268268 // MC Gen: {pt, centrality, rapidity, numPvContr, ptB, origin}
269269 if (doprocessMcWithMl || doprocessMcWithMlWithFT0C || doprocessMcWithMlWithFT0M) {
270270 registry.add (" hnLcK0sPGenMc" , " THn for Lc->K0sP MC gen" , HistType::kTHnSparseF ,
271- {thnAxisPt, thnAxisCentrality, thnAxisY, thnAxisNumPvContr, thnAxisPtB, thnAxisOrigin});
271+ {thnAxisPt, thnAxisCentrality, thnAxisY, thnAxisNumPvContr, thnAxisPtB, thnAxisOrigin});
272272 }
273273 }
274274 }
@@ -516,38 +516,38 @@ struct HfTaskLcToK0sP {
516516 }
517517
518518 void processDataStd (Collisions const & collisions,
519- FilteredCandLcToPK0SWSelFlag const & candidates,
520- TracksWPid const &)
519+ FilteredCandLcToPK0SWSelFlag const & candidates,
520+ TracksWPid const &)
521521 {
522522 runAnalysisPerCollisionData<false >(collisions, candidates);
523523 }
524524 PROCESS_SWITCH (HfTaskLcToK0sP, processDataStd, " Process Data" , false );
525525
526526 void processMcStd (CollisionsMc const & collisions,
527- FilteredCandLcToPK0SWSelFlagAndMc const & candidates,
528- soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
529- aod::TracksWMc const &,
530- TracksWPid const &)
527+ FilteredCandLcToPK0SWSelFlagAndMc const & candidates,
528+ soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
529+ aod::TracksWMc const &,
530+ TracksWPid const &)
531531 {
532532 runAnalysisPerCollisionMc<false >(collisions, candidates, mcParticles);
533533 }
534534 PROCESS_SWITCH (HfTaskLcToK0sP, processMcStd, " Process MC data" , false );
535535
536536 // Data with ML
537537 void processDataWithMl (Collisions const & collisions,
538- FilteredCandLcToPK0SWSelFlagAndMl const & candidates,
539- TracksWPid const &)
538+ FilteredCandLcToPK0SWSelFlagAndMl const & candidates,
539+ TracksWPid const &)
540540 {
541541 runAnalysisPerCollisionData<true >(collisions, candidates);
542542 }
543543 PROCESS_SWITCH (HfTaskLcToK0sP, processDataWithMl, " Process Data with ML" , false );
544544
545545 // MC Rec and Gen with ML
546546 void processMcWithMl (CollisionsMc const & collisions,
547- FilteredCandLcToPK0SWSelFlagAndMcAndMl const & candidates,
548- soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
549- aod::TracksWMc const &,
550- TracksWPid const &)
547+ FilteredCandLcToPK0SWSelFlagAndMcAndMl const & candidates,
548+ soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
549+ aod::TracksWMc const &,
550+ TracksWPid const &)
551551 {
552552 runAnalysisPerCollisionMc<true >(collisions, candidates, mcParticles);
553553 }
@@ -602,21 +602,21 @@ struct HfTaskLcToK0sP {
602602
603603 // MC Std with FT0M centrality
604604 void processMcStdWithFT0M (CollisionsMcWithFT0M const & collisions,
605- FilteredCandLcToPK0SWSelFlagAndMc const & candidates,
606- soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
607- aod::TracksWMc const &,
608- TracksWPid const &)
605+ FilteredCandLcToPK0SWSelFlagAndMc const & candidates,
606+ soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
607+ aod::TracksWMc const &,
608+ TracksWPid const &)
609609 {
610610 runAnalysisPerCollisionMc<false >(collisions, candidates, mcParticles);
611611 }
612612 PROCESS_SWITCH (HfTaskLcToK0sP, processMcStdWithFT0M, " Process MC with FT0M centrality" , false );
613613
614614 // MC with ML + FT0C centrality
615615 void processMcWithMlWithFT0C (CollisionsMcWithFT0C const & collisions,
616- FilteredCandLcToPK0SWSelFlagAndMcAndMl const & candidates,
617- soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
618- aod::TracksWMc const &,
619- TracksWPid const &)
616+ FilteredCandLcToPK0SWSelFlagAndMcAndMl const & candidates,
617+ soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const & mcParticles,
618+ aod::TracksWMc const &,
619+ TracksWPid const &)
620620 {
621621 runAnalysisPerCollisionMc<true >(collisions, candidates, mcParticles);
622622 }
@@ -639,4 +639,4 @@ WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
639639 return WorkflowSpec{
640640 adaptAnalysisTask<HfTaskLcToK0sP>(cfgc),
641641 };
642- }
642+ }
0 commit comments