Skip to content

Commit 45c5ba8

Browse files
authored
[PWGHF] Add missing conditions for different process functions (#10590)
1 parent 78fd12a commit 45c5ba8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

PWGHF/TableProducer/trackIndexSkimCreator.cxx

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)