|
18 | 18 | /// \author Daniel Samitz <daniel.samitz@cern.ch> |
19 | 19 | /// \author Elisa Meninno <elisa.meninno@cern.ch> |
20 | 20 |
|
21 | | - |
22 | 21 | #include "PWGHF/Core/HfHelper.h" |
23 | 22 | #include "PWGHF/DataModel/AliasTables.h" |
24 | 23 | #include "PWGHF/DataModel/CandidateReconstructionTables.h" |
@@ -92,7 +91,7 @@ DECLARE_SOA_COLUMN(OriginMcGen, originMcGen, int8_t); |
92 | 91 | DECLARE_SOA_COLUMN(MlScoreFirstClass, mlScoreFirstClass, float); |
93 | 92 | DECLARE_SOA_COLUMN(MlScoreSecondClass, mlScoreSecondClass, float); |
94 | 93 | DECLARE_SOA_COLUMN(MlScoreThirdClass, mlScoreThirdClass, float); |
95 | | -DECLARE_SOA_COLUMN(SigBgStatus, sigBgStatus, int); //! 0 bg, 1 prompt, 2 non-prompt,-1 default value (impossible, should not be the case), -999 for data |
| 94 | +DECLARE_SOA_COLUMN(SigBgStatus, sigBgStatus, int); //! 0 bg, 1 prompt, 2 non-prompt,-1 default value (impossible, should not be the case), -999 for data |
96 | 95 | // Events |
97 | 96 | DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int); |
98 | 97 | DECLARE_SOA_COLUMN(RunNumber, runNumber, int); |
@@ -140,7 +139,6 @@ DECLARE_SOA_TABLE(HfCandCascLites, "AOD", "HFCANDCASCLITE", |
140 | 139 | full::MlScoreSecondClass, |
141 | 140 | full::MlScoreThirdClass, |
142 | 141 | full::SigBgStatus); |
143 | | - |
144 | 142 |
|
145 | 143 | DECLARE_SOA_TABLE(HfCandCascFulls, "AOD", "HFCANDCASCFULL", |
146 | 144 | collision::BCId, |
@@ -273,7 +271,7 @@ struct HfTreeCreatorLcToK0sP { |
273 | 271 | SigBgStatus status{Default}; |
274 | 272 |
|
275 | 273 | if (std::abs(flag) != 1) { |
276 | | - return Background; // combinatorial or wrong decay |
| 274 | + return Background; // combinatorial or wrong decay |
277 | 275 | } |
278 | 276 | // Otherwise it is signal: distinguish between Prompt/ NonPrompt |
279 | 277 | if (origin == RecoDecay::OriginType::Prompt) { |
@@ -489,11 +487,11 @@ struct HfTreeCreatorLcToK0sP { |
489 | 487 | const int flag = candidate.flagMcMatchRec(); |
490 | 488 |
|
491 | 489 | // get status (Prompt / NonPrompt / Background) |
492 | | - SigBgStatus sigbgstatus = determineSignalBgStatus(candidate); |
493 | | - |
| 490 | + SigBgStatus sigbgstatus = determineSignalBgStatus(candidate); |
| 491 | + |
494 | 492 | bool isSignal = (std::abs(flag) == 1); |
495 | 493 | bool isBackground = !isSignal; |
496 | | - |
| 494 | + |
497 | 495 | if ((fillOnlySignal && isSignal) || (fillOnlyBackground && isBackground) || (!fillOnlySignal && !fillOnlyBackground)) { |
498 | 496 | fillCandidate(candidate, bach, candidate.flagMcMatchRec(), candidate.originMcRec(), candidateMlScore, sigbgstatus); |
499 | 497 | } |
|
0 commit comments