Changes to compute efficiencies for particle or antiparticle separately (ChiaraZ...

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliCFVertexingHF.cxx

diff --git a/PWG3/vertexingHF/AliCFVertexingHF.cxx b/PWG3/vertexingHF/AliCFVertexingHF.cxx
index 7712cd6e8b53c1bff9c8e2b943836e7247fe8e70..98df8cb48c2a8f10e8511b18322d6fb4e24dcf9b 100644 (file)
--- a/PWG3/vertexingHF/AliCFVertexingHF.cxx
+++ b/PWG3/vertexingHF/AliCFVertexingHF.cxx
@@ -279,7 +279,6 @@ Int_t AliCFVertexingHF::CheckOrigin() const
        if (!((abspdgGranma > 500 && abspdgGranma < 600) || (abspdgGranma > 5000 && abspdgGranma < 6000))){
                if (fKeepDfromBOnly) return -999;
        }
-
        return pdgGranma;
 }
 
@@ -625,7 +624,7 @@ Bool_t AliCFVertexingHF::FillUnfoldingMatrix(Double_t *fill) const
 }
 //___________________________________________________________
 
-Int_t AliCFVertexingHF::CheckReflexion()
+Int_t AliCFVertexingHF::CheckReflexion(Char_t isSign)
 {
        //
        // check for reflexion (particle/antiparticle)
@@ -645,9 +644,22 @@ Int_t AliCFVertexingHF::CheckReflexion()
                }    
        }
        
-       if(fmcPartCandidate->GetPdgCode()>0) return 1;  // particle
-       else if(fmcPartCandidate->GetPdgCode()<0) return 2;  // antiparticle
+       if(fmcPartCandidate->GetPdgCode()>0) {
+               if (isSign == 1){ // I ask for antiparticle only
+                       AliDebug(2,"candidate is particle, I ask for antiparticle only");
+                       return 0;
+               }
+               return 1;  // particle
+       }
+       else if(fmcPartCandidate->GetPdgCode()<0) {
+               if (isSign == 0){ // I ask for particle only
+                       AliDebug(2,"candidate is antiparticle, I ask for particle only");
+                       return 0;
+               }
+               return 2;  // antiparticle
+       }
        else return 0;  // ....shouldn't be...
+
 }
 //___________________________________________________________