Added check on PDG code of daughters for MC

diff --git a/PWG3/vertexingHF/AliCFHeavyFlavourTask.cxx b/PWG3/vertexingHF/AliCFHeavyFlavourTask.cxx
index 629dd40c87cb8e0497dc90f36817364c11aff387..6e012067aafc44aa15f33ab1748951c6028946ff 100644 (file)
--- a/PWG3/vertexingHF/AliCFHeavyFlavourTask.cxx
+++ b/PWG3/vertexingHF/AliCFHeavyFlavourTask.cxx
@@ -165,6 +165,13 @@ void AliCFHeavyFlavourTask::UserExec(Option_t *)
                        AliWarning("At least one Daughter Particle not found in tree, skipping"); 
                        continue;  
                }
+               if (!(TMath::Abs(mcPartDaughter0->GetPdgCode())==321 &&
+                     TMath::Abs(mcPartDaughter1->GetPdgCode())==211) && 
+                   !(TMath::Abs(mcPartDaughter0->GetPdgCode())==211 &&
+                     TMath::Abs(mcPartDaughter1->GetPdgCode())==321)) {
+                 AliDebug(2, "The D0 MC doesn't come from a Kpi decay, skipping!!");
+                 continue;  
+               }
 
                // fill the container for Gen-level selection
                containerInput[0] = mcPart->Pt();

diff --git a/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVar.cxx b/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVar.cxx
index 0cc35d94ac7ecae57a5adea1f79a9c73867714d8..8ede588bccad0de3a211fcfc8f63aeff8d78c9ab 100644 (file)
--- a/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVar.cxx
+++ b/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVar.cxx
@@ -600,7 +600,14 @@ Bool_t AliCFHeavyFlavourTaskMultiVar::GetGeneratedValuesFromMCParticle(AliAODMCP
                AliWarning("At least one Daughter Particle not found in tree, skipping"); 
                return isOk;  
        }
-       
+       if (!(TMath::Abs(mcPartDaughter0->GetPdgCode())==321 &&
+             TMath::Abs(mcPartDaughter1->GetPdgCode())==211) && 
+           !(TMath::Abs(mcPartDaughter0->GetPdgCode())==211 &&
+             TMath::Abs(mcPartDaughter1->GetPdgCode())==321)) {
+         AliDebug(2, "The D0 MC doesn't come from a Kpi decay, skipping!!");
+         return isOk;  
+       }
+
        Double_t vtx1[3] = {0,0,0};   // primary vertex         
        Double_t vtx2daughter0[3] = {0,0,0};   // secondary vertex from daughter 0
        Double_t vtx2daughter1[3] = {0,0,0};   // secondary vertex from daughter 1

diff --git a/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.cxx b/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.cxx
index 1e81a7c2e285a69dd44621f373c81c4043543178..afcc7da36f9b7f96b6f92dfd8c3776b0a72efdce 100644 (file)
--- a/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.cxx
+++ b/PWG3/vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.cxx
@@ -1053,7 +1053,14 @@ Bool_t AliCFHeavyFlavourTaskMultiVarMultiStep::GetGeneratedValuesFromMCParticle(
                AliWarning("At least one Daughter Particle not found in tree, skipping"); 
                return isOk;  
        }
-       
+       if (!(TMath::Abs(mcPartDaughter0->GetPdgCode())==321 &&
+             TMath::Abs(mcPartDaughter1->GetPdgCode())==211) && 
+           !(TMath::Abs(mcPartDaughter0->GetPdgCode())==211 &&
+             TMath::Abs(mcPartDaughter1->GetPdgCode())==321)) {
+         AliDebug(2, "The D0 MC doesn't come from a Kpi decay, skipping!!");
+         return isOk;  
+       }
+
        Double_t vtx1[3] = {0,0,0};   // primary vertex         
        Double_t vtx2daughter0[3] = {0,0,0};   // secondary vertex from daughter 0
        Double_t vtx2daughter1[3] = {0,0,0};   // secondary vertex from daughter 1