Update taskCharmHadronsTrackFemtoDream.cxx

Author: zhangbiao-phy

PWGHF/HFC/Tasks/taskCharmHadronsTrackFemtoDream.cxx

@@ -767,7 +767,30 @@ struct HfTaskCharmHadronsTrackFemtoDream {
                             : NegativeCharge;
 
       timeStamp = part.timeStamp();
-
+      float tpcNSigma = 999.f;
+      float tofNSigma = 999.f;
+      switch (trackSel.pdgCodeTrack1.value) {
+        case kProton:
+          tpcNSigma = part.tpcNSigmaPr();
+          tofNSigma = part.tofNSigmaPr();
+          break;
+        case kPiPlus:
+          tpcNSigma = part.tpcNSigmaPi();
+          tofNSigma = part.tofNSigmaPi();
+          break;
+        case kKPlus:
+          tpcNSigma = part.tpcNSigmaKa();
+          tofNSigma = part.tofNSigmaKa();
+          break;
+        case kDeuteron:
+          tpcNSigma = part.tpcNSigmaDe();
+          tofNSigma = part.tofNSigmaDe();
+          break;
+        default:
+          LOG(fatal) << "Unhandled PDG code in PID switch: "
+                     << trackSel.pdgCodeTrack1.value;
+          break;
+      }
       rowFemtoResultTrk(
         col.globalIndex(),
         timeStamp,
@@ -779,8 +802,8 @@ struct HfTaskCharmHadronsTrackFemtoDream {
         part.tpcNClsFound(),
         part.tpcNClsFindable(),
         part.tpcNClsCrossedRows(),
-        part.tpcNSigmaPr(),
-        part.tofNSigmaPr());
+        tpcNSigma,
+        tofNSigma);
     }
 
     // ---- Fill Collision Table ----
@@ -874,9 +897,12 @@ struct HfTaskCharmHadronsTrackFemtoDream {
       auto sliceCharmHad = partitionCharmHadron2Prong->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
       if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
         continue;
+      } else {
+        fillTables<false, DecayChannel::D0ToPiK>(col, sliceTrk1, sliceCharmHad);
+      }
+      if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
+        doSameEvent<false, DecayChannel::D0ToPiK, FilteredCharmCand2Prongs>(sliceCharmHad, sliceTrk1, parts, col);
       }
-      fillTables<false, DecayChannel::D0ToPiK>(col, sliceTrk1, sliceCharmHad);
-      doSameEvent<false, DecayChannel::D0ToPiK, FilteredCharmCand2Prongs>(sliceCharmHad, sliceTrk1, parts, col);
     }
     if (mixSetting.doMixEvent) {
       switch (mixSetting.mixingBinPolicy) {
@@ -906,9 +932,12 @@ struct HfTaskCharmHadronsTrackFemtoDream {
       auto sliceCharmHad = partitionCharmHadronDstar->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
       if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
         continue;
+      } else {
+        fillTables<false, DecayChannel::DstarToD0Pi>(col, sliceTrk1, sliceCharmHad);
+      }
+      if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
+        doSameEvent<false, DecayChannel::DstarToD0Pi, FilteredCharmCandDstars>(sliceCharmHad, sliceTrk1, parts, col);
       }
-      fillTables<false, DecayChannel::DstarToD0Pi>(col, sliceTrk1, sliceCharmHad);
-      doSameEvent<false, DecayChannel::DstarToD0Pi, FilteredCharmCandDstars>(sliceCharmHad, sliceTrk1, parts, col);
     }
     if (mixSetting.doMixEvent) {
       switch (mixSetting.mixingBinPolicy) {