|
18 | 18 | /// |
19 | 19 | /// \note based on taskD0.cxx, taskLc.cxx |
20 | 20 |
|
| 21 | +#include "PWGHF/Core/CentralityEstimation.h" |
21 | 22 | #include "PWGHF/Core/HfHelper.h" |
22 | 23 | #include "PWGHF/Core/SelectorCuts.h" |
23 | | -#include "PWGHF/Core/CentralityEstimation.h" |
24 | 24 | #include "PWGHF/DataModel/AliasTables.h" |
25 | 25 | #include "PWGHF/DataModel/CandidateReconstructionTables.h" |
26 | 26 | #include "PWGHF/DataModel/CandidateSelectionTables.h" |
|
37 | 37 | #include <Framework/InitContext.h> |
38 | 38 | #include <Framework/runDataProcessing.h> |
39 | 39 |
|
| 40 | +#include <THnSparse.h> |
| 41 | + |
40 | 42 | #include <cstdlib> |
41 | | -#include <vector> |
42 | | -#include <string> |
43 | 43 | #include <numeric> |
44 | | - |
45 | | -#include <THnSparse.h> |
| 44 | +#include <string> |
| 45 | +#include <vector> |
46 | 46 |
|
47 | 47 | using namespace o2; |
48 | 48 | using namespace o2::analysis; |
@@ -258,19 +258,19 @@ struct HfTaskLcToK0sP { |
258 | 258 | // Data with ML: {mass, pt, centrality, bkg, prompt, non-prompt, numPvContr} |
259 | 259 | if (doprocessDataWithMl || doprocessDataWithMlWithFT0C || doprocessDataWithMlWithFT0M) { |
260 | 260 | registry.add("hnLcK0sPDataWithBdt", "THn for Lc->K0sP data with BDT", HistType::kTHnSparseF, |
261 | | - {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr}); |
| 261 | + {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr}); |
262 | 262 | } |
263 | 263 |
|
264 | 264 | // MC Rec with ML: {mass, pt, centrality, bkg, prompt, non-prompt, numPvContr, origin} |
265 | 265 | if (doprocessMcWithMl || doprocessMcWithMlWithFT0C || doprocessMcWithMlWithFT0M) { |
266 | 266 | registry.add("hnLcK0sPRecMcWithBdt", "THn for Lc->K0sP MC rec with BDT", HistType::kTHnSparseF, |
267 | | - {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr, thnAxisOrigin}); |
| 267 | + {thnAxisMass, thnAxisPt, thnAxisCentrality, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt, thnAxisNumPvContr, thnAxisOrigin}); |
268 | 268 | } |
269 | 269 |
|
270 | 270 | // MC Gen: {pt, centrality, rapidity, numPvContr, ptB, origin} |
271 | 271 | if (doprocessMcWithMl || doprocessMcWithMlWithFT0C || doprocessMcWithMlWithFT0M) { |
272 | 272 | registry.add("hnLcK0sPGenMc", "THn for Lc->K0sP MC gen", HistType::kTHnSparseF, |
273 | | - {thnAxisPt, thnAxisCentrality, thnAxisY, thnAxisNumPvContr, thnAxisPtB, thnAxisOrigin}); |
| 273 | + {thnAxisPt, thnAxisCentrality, thnAxisY, thnAxisNumPvContr, thnAxisPtB, thnAxisOrigin}); |
274 | 274 | } |
275 | 275 | } |
276 | 276 | } |
@@ -518,38 +518,38 @@ struct HfTaskLcToK0sP { |
518 | 518 | } |
519 | 519 |
|
520 | 520 | void processDataStd(Collisions const& collisions, |
521 | | - FilteredCandLcToPK0SWSelFlag const& candidates, |
522 | | - TracksWPid const&) |
| 521 | + FilteredCandLcToPK0SWSelFlag const& candidates, |
| 522 | + TracksWPid const&) |
523 | 523 | { |
524 | 524 | runAnalysisPerCollisionData<false>(collisions, candidates); |
525 | 525 | } |
526 | 526 | PROCESS_SWITCH(HfTaskLcToK0sP, processDataStd, "Process Data", false); |
527 | 527 |
|
528 | 528 | void processMcStd(CollisionsMc const& collisions, |
529 | | - FilteredCandLcToPK0SWSelFlagAndMc const& candidates, |
530 | | - soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
531 | | - aod::TracksWMc const&, |
532 | | - TracksWPid const&) |
| 529 | + FilteredCandLcToPK0SWSelFlagAndMc const& candidates, |
| 530 | + soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
| 531 | + aod::TracksWMc const&, |
| 532 | + TracksWPid const&) |
533 | 533 | { |
534 | 534 | runAnalysisPerCollisionMc<false>(collisions, candidates, mcParticles); |
535 | 535 | } |
536 | 536 | PROCESS_SWITCH(HfTaskLcToK0sP, processMcStd, "Process MC data", false); |
537 | 537 |
|
538 | 538 | // Data with ML |
539 | 539 | void processDataWithMl(Collisions const& collisions, |
540 | | - FilteredCandLcToPK0SWSelFlagAndMl const& candidates, |
541 | | - TracksWPid const&) |
| 540 | + FilteredCandLcToPK0SWSelFlagAndMl const& candidates, |
| 541 | + TracksWPid const&) |
542 | 542 | { |
543 | 543 | runAnalysisPerCollisionData<true>(collisions, candidates); |
544 | 544 | } |
545 | 545 | PROCESS_SWITCH(HfTaskLcToK0sP, processDataWithMl, "Process Data with ML", false); |
546 | 546 |
|
547 | 547 | // MC Rec and Gen with ML |
548 | 548 | void processMcWithMl(CollisionsMc const& collisions, |
549 | | - FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates, |
550 | | - soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
551 | | - aod::TracksWMc const&, |
552 | | - TracksWPid const&) |
| 549 | + FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates, |
| 550 | + soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
| 551 | + aod::TracksWMc const&, |
| 552 | + TracksWPid const&) |
553 | 553 | { |
554 | 554 | runAnalysisPerCollisionMc<true>(collisions, candidates, mcParticles); |
555 | 555 | } |
@@ -604,21 +604,21 @@ struct HfTaskLcToK0sP { |
604 | 604 |
|
605 | 605 | // MC Std with FT0M centrality |
606 | 606 | void processMcStdWithFT0M(CollisionsMcWithFT0M const& collisions, |
607 | | - FilteredCandLcToPK0SWSelFlagAndMc const& candidates, |
608 | | - soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
609 | | - aod::TracksWMc const&, |
610 | | - TracksWPid const&) |
| 607 | + FilteredCandLcToPK0SWSelFlagAndMc const& candidates, |
| 608 | + soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
| 609 | + aod::TracksWMc const&, |
| 610 | + TracksWPid const&) |
611 | 611 | { |
612 | 612 | runAnalysisPerCollisionMc<false>(collisions, candidates, mcParticles); |
613 | 613 | } |
614 | 614 | PROCESS_SWITCH(HfTaskLcToK0sP, processMcStdWithFT0M, "Process MC with FT0M centrality", false); |
615 | 615 |
|
616 | 616 | // MC with ML + FT0C centrality |
617 | 617 | void processMcWithMlWithFT0C(CollisionsMcWithFT0C const& collisions, |
618 | | - FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates, |
619 | | - soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
620 | | - aod::TracksWMc const&, |
621 | | - TracksWPid const&) |
| 618 | + FilteredCandLcToPK0SWSelFlagAndMcAndMl const& candidates, |
| 619 | + soa::Join<aod::McParticles, aod::HfCandCascadeMcGen> const& mcParticles, |
| 620 | + aod::TracksWMc const&, |
| 621 | + TracksWPid const&) |
622 | 622 | { |
623 | 623 | runAnalysisPerCollisionMc<true>(collisions, candidates, mcParticles); |
624 | 624 | } |
|
0 commit comments