@@ -1268,7 +1268,7 @@ struct HfTrackIndexSkimCreator {
12681268
12691269 void init (InitContext const &)
12701270 {
1271- if (!doprocess2And3ProngsWithPvRefit && !doprocess2And3ProngsNoPvRefit) {
1271+ if (!doprocess2And3ProngsWithPvRefit && !doprocess2And3ProngsNoPvRefit && !doprocess2And3ProngsWithPvRefitWithPidForHfFiltersBdt && !doprocess2And3ProngsNoPvRefitWithPidForHfFiltersBdt ) {
12721272 return ;
12731273 }
12741274
@@ -1356,7 +1356,7 @@ struct HfTrackIndexSkimCreator {
13561356 registry.add (" hMassDstarToD0Pi" , " D^{*#plus} candidates;inv. mass (K #pi #pi) - mass (K #pi) (GeV/#it{c}^{2});entries" , {HistType::kTH1D , {{500 , 0.135 , 0.185 }}});
13571357
13581358 // needed for PV refitting
1359- if (doprocess2And3ProngsWithPvRefit) {
1359+ if (doprocess2And3ProngsWithPvRefit || doprocess2And3ProngsWithPvRefitWithPidForHfFiltersBdt ) {
13601360 AxisSpec axisCollisionX{100 , -20 .f , 20 .f , " X (cm)" };
13611361 AxisSpec axisCollisionY{100 , -20 .f , 20 .f , " Y (cm)" };
13621362 AxisSpec axisCollisionZ{100 , -20 .f , 20 .f , " Z (cm)" };
@@ -1916,7 +1916,7 @@ struct HfTrackIndexSkimCreator {
19161916 bool pvRefitDoable = vertexer.prepareVertexRefit (vecPvContributorTrackParCov, primVtx);
19171917 if (!pvRefitDoable) {
19181918 LOG (info) << " Not enough tracks accepted for the refit" ;
1919- if (doprocess2And3ProngsWithPvRefit && config.fillHistograms ) {
1919+ if (( doprocess2And3ProngsWithPvRefit || doprocess2And3ProngsWithPvRefitWithPidForHfFiltersBdt) && config.fillHistograms ) {
19201920 registry.fill (HIST (" PvRefit/hNContribPvRefitNotDoable" ), collision.numContrib ());
19211921 }
19221922 }
@@ -1927,7 +1927,7 @@ struct HfTrackIndexSkimCreator {
19271927 // / PV refitting, if the tracks contributed to this at the beginning
19281928 o2::dataformats::VertexBase primVtxBaseRecalc;
19291929 bool recalcPvRefit = false ;
1930- if (doprocess2And3ProngsWithPvRefit && pvRefitDoable) {
1930+ if (( doprocess2And3ProngsWithPvRefit || doprocess2And3ProngsWithPvRefitWithPidForHfFiltersBdt) && pvRefitDoable) {
19311931 if (config.fillHistograms ) {
19321932 registry.fill (HIST (" PvRefit/verticesPerCandidate" ), 2 );
19331933 }
0 commit comments