@@ -187,7 +187,7 @@ struct HfFemtoDreamProducer {
187187
188188 using GeneratedMc = soa::Filtered<soa::Join<aod::McParticles, aod::HfCand3ProngMcGen>>;
189189
190- Filter filterSelectCandidateDplus = (aod::hf_sel_candidate_dplus::isSelDplusToPiKPi >= selectionFlagHadron || aod::hf_sel_candidate_dplus::isSelDplusToPiKPi >= selectionFlagHadron);
190+ Filter filterSelectCandidateDplus = (aod::hf_sel_candidate_dplus::isSelDplusToPiKPi >= selectionFlagHadron || aod::hf_sel_candidate_dplus::isSelDplusToPiKPi >= selectionFlagHadron);
191191 Filter filterSelectCandidateLc = (aod::hf_sel_candidate_lc::isSelLcToPKPi >= selectionFlagHadron || aod::hf_sel_candidate_lc::isSelLcToPiKP >= selectionFlagHadron);
192192
193193 HistogramRegistry qaRegistry{" QAHistos" , {}, OutputObjHandlingPolicy::AnalysisObject};
@@ -472,7 +472,7 @@ struct HfFemtoDreamProducer {
472472 auto trackPos1 = candidate.template prong0_as <TrackType>(); // positive daughter (negative for the antiparticles)
473473 auto trackNeg = candidate.template prong1_as <TrackType>(); // negative daughter (positive for the antiparticles)
474474 auto trackPos2 = candidate.template prong2_as <TrackType>(); // positive daughter (negative for the antiparticles)
475-
475+
476476 auto bc = col.template bc_as <aod::BCsWithTimestamps>();
477477 int64_t timeStamp = bc.timestamp ();
478478 auto fillTable = [&](int CandFlag,
@@ -510,34 +510,34 @@ struct HfFemtoDreamProducer {
510510 }
511511 } };
512512
513- if constexpr (channel == DecayChannel::DplusToPiKPi) {
513+ if constexpr (channel == DecayChannel::DplusToPiKPi) {
514514 if constexpr (useCharmMl) {
515515 // / fill with ML information
516- // / BDT index 0: bkg score; BDT index 1: prompt score; BDT index 2: non-prompt score
516+ // / BDT index 0: bkg score; BDT index 1: prompt score; BDT index 2: non-prompt score
517517 if (applyMlMode == FillMlFromSelector) {
518518 if (candidate.mlProbDplusToPiKPi ().size () > 0 ) {
519519 outputMlDplus.at (0 ) = candidate.mlProbDplusToPiKPi ()[0 ]; // / bkg score
520520 outputMlDplus.at (1 ) = candidate.mlProbDplusToPiKPi ()[1 ]; // / prompt score
521521 outputMlDplus.at (2 ) = candidate.mlProbDplusToPiKPi ()[2 ]; // / non-prompt score
522- }
522+ }
523523 } else if (applyMlMode == FillMlFromNewBDT) {
524- isSelectedMlDplusToPiKPi = false ;
525- if (candidate.mlProbDplusToPiKPi ().size () > 0 ) {
526- std::vector<float > inputFeaturesDplusToPiKPi = hfMlResponseDplus.getInputFeatures (candidate);
527- isSelectedMlDplusToPiKPi = hfMlResponseDplus.isSelectedMl (inputFeaturesDplusToPiKPi, candidate.pt (), outputMlDplus);
528- }
529- if (!isSelectedMlDplusToPiKPi)
530- continue ;
531- } else {
532- LOGF (fatal, " Please check your Ml configuration!!" );
524+ isSelectedMlDplusToPiKPi = false ;
525+ if (candidate.mlProbDplusToPiKPi ().size () > 0 ) {
526+ std::vector<float > inputFeaturesDplusToPiKPi = hfMlResponseDplus.getInputFeatures (candidate);
527+ isSelectedMlDplusToPiKPi = hfMlResponseDplus.isSelectedMl (inputFeaturesDplusToPiKPi, candidate.pt (), outputMlDplus);
533528 }
529+ if (!isSelectedMlDplusToPiKPi)
530+ continue ;
531+ } else {
532+ LOGF (fatal, " Please check your Ml configuration!!" );
534533 }
534+ }
535535 fillTable (0 , candidate.isSelDplusToPiKPi (), outputMlDplus.at (0 ), outputMlDplus.at (1 ), outputMlDplus.at (2 ));
536-
537- } else if constexpr (channel == DecayChannel::LcToPKPi) {
536+
537+ } else if constexpr (channel == DecayChannel::LcToPKPi) {
538538 if constexpr (useCharmMl) {
539539 // / fill with ML information
540- // / BDT index 0: bkg score; BDT index 1: prompt score; BDT index 2: non-prompt score
540+ // / BDT index 0: bkg score; BDT index 1: prompt score; BDT index 2: non-prompt score
541541 if (applyMlMode == FillMlFromSelector) {
542542 if (candidate.mlProbLcToPKPi ().size () > 0 ) {
543543 outputMlPKPi.at (0 ) = candidate.mlProbLcToPKPi ()[0 ]; // / bkg score
@@ -629,9 +629,9 @@ struct HfFemtoDreamProducer {
629629
630630 // / DplusToPiKPi
631631 void processDataDplusToPiKPi (FemtoFullCollision const & col,
632- aod::BCsWithTimestamps const &,
633- FemtoHFTracks const & tracks,
634- soa::Filtered<CandidateDplus> const & candidates)
632+ aod::BCsWithTimestamps const &,
633+ FemtoHFTracks const & tracks,
634+ soa::Filtered<CandidateDplus> const & candidates)
635635 {
636636 // get magnetic field for run
637637 getMagneticFieldTesla (col.bc_as <aod::BCsWithTimestamps>());
@@ -642,10 +642,10 @@ struct HfFemtoDreamProducer {
642642 " Provide experimental data for DplusToPiKPi femto" , false );
643643
644644 void processDataDplusToPiKPiWithML (FemtoFullCollision const & col,
645- aod::BCsWithTimestamps const &,
646- FemtoHFTracks const & tracks,
647- soa::Filtered<soa::Join<CandidateDplus,
648- aod::HfMlDplusToPiKPi>> const & candidates)
645+ aod::BCsWithTimestamps const &,
646+ FemtoHFTracks const & tracks,
647+ soa::Filtered<soa::Join<CandidateDplus,
648+ aod::HfMlDplusToPiKPi>> const & candidates)
649649 {
650650
651651 // get magnetic field for run
@@ -657,10 +657,10 @@ struct HfFemtoDreamProducer {
657657 " Provide experimental data for DplusToPiKPi with ml" , false );
658658
659659 void processMcDplusToPiKPi (FemtoFullCollisionMc const & col,
660- aod::BCsWithTimestamps const &,
661- FemtoHFMcTracks const & tracks,
662- aod::McParticles const &,
663- CandidateDplusMc const & candidates)
660+ aod::BCsWithTimestamps const &,
661+ FemtoHFMcTracks const & tracks,
662+ aod::McParticles const &,
663+ CandidateDplusMc const & candidates)
664664 {
665665 // get magnetic field for run
666666 getMagneticFieldTesla (col.bc_as <aod::BCsWithTimestamps>());
@@ -670,11 +670,11 @@ struct HfFemtoDreamProducer {
670670 PROCESS_SWITCH (HfFemtoDreamProducer, processMcDplusToPiKPi, " Provide Mc for DplusToPiKPi" , false );
671671
672672 void processMcDplusToPiKPiWithML (FemtoFullCollisionMc const & col,
673- aod::BCsWithTimestamps const &,
674- FemtoHFMcTracks const & tracks,
675- aod::McParticles const &,
676- soa::Join<CandidateDplusMc,
677- aod::HfMlDplusToPiKPi> const & candidates)
673+ aod::BCsWithTimestamps const &,
674+ FemtoHFMcTracks const & tracks,
675+ aod::McParticles const &,
676+ soa::Join<CandidateDplusMc,
677+ aod::HfMlDplusToPiKPi> const & candidates)
678678 {
679679 // get magnetic field for run
680680 getMagneticFieldTesla (col.bc_as <aod::BCsWithTimestamps>());
@@ -728,7 +728,7 @@ struct HfFemtoDreamProducer {
728728 // get magnetic field for run
729729 getMagneticFieldTesla (col.bc_as <aod::BCsWithTimestamps>());
730730
731- fillCharmHadronTable<DecayChannel::LcToPKPi,true , false >(col, tracks, candidates);
731+ fillCharmHadronTable<DecayChannel::LcToPKPi, true , false >(col, tracks, candidates);
732732 }
733733 PROCESS_SWITCH (HfFemtoDreamProducer, processMcLcToPKPi, " Provide Mc for lctopkpi" , false );
734734
0 commit comments