Changes to the AliFemto code for DEtaDPhi angular analysis. Adding Nsigma values...

[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemtoUser / AliFemtoESDTrackCut.cxx

diff --git a/PWGCF/FEMTOSCOPY/AliFemtoUser/AliFemtoESDTrackCut.cxx b/PWGCF/FEMTOSCOPY/AliFemtoUser/AliFemtoESDTrackCut.cxx
index f0a9c78c86453714f0a080be94c81b2e866af35c..d01f333aa6f6f0d3cf8b4a8d3d1d6957932ef39a 100644 (file)
--- a/PWGCF/FEMTOSCOPY/AliFemtoUser/AliFemtoESDTrackCut.cxx
+++ b/PWGCF/FEMTOSCOPY/AliFemtoUser/AliFemtoESDTrackCut.cxx
@@ -113,7 +113,8 @@ ClassImp(AliFemtoESDTrackCut)
     fMinPforTPCpid(0.0),
     fMaxPforTPCpid(10000.0),
     fMinPforITSpid(0.0),
-    fMaxPforITSpid(10000.0)
+   fMaxPforITSpid(10000.0),
+   fElectronRejection(0)
 {
   // Default constructor
   fNTracksPassed = fNTracksFailed = 0;
@@ -359,6 +360,10 @@ bool AliFemtoESDTrackCut::Pass(const AliFemtoTrack* track)
     tMost[4] = track->PidProbProton()*PidFractionProton(track->P().Mag());
     float ipidmax = 0.0;
 
+    //****N Sigma Method -- electron rejection****
+    if(fElectronRejection) 
+      if(!IsElectron(track->NSigmaTPCE(),track->NSigmaTPCPi(),track->NSigmaTPCK(), track->NSigmaTPCP())) 
+       return false;
 
     //****N Sigma Method****
        if(fPIDMethod==0){
@@ -1042,6 +1047,7 @@ void AliFemtoESDTrackCut::SetNsigma(Double_t nsigma)
   fNsigma = nsigma;
 }
 
+
 void AliFemtoESDTrackCut::SetClusterRequirementITS(AliESDtrackCuts::Detector det, AliESDtrackCuts::ITSClusterRequirement req)
 {
   fCutClusterRequirementITS[det] = req;
@@ -1065,3 +1071,11 @@ Bool_t AliFemtoESDTrackCut::CheckITSClusterRequirement(AliESDtrackCuts::ITSClust
 
   return kFALSE;
 }
+
+bool AliFemtoESDTrackCut::IsElectron(float nsigmaTPCE, float nsigmaTPCPi,float nsigmaTPCK, float nsigmaTPCP)
+{
+   if(nsigmaTPCE<3 && nsigmaTPCPi>3 && nsigmaTPCK>3 && nsigmaTPCP>3)
+      return false;
+   else
+     return true;
+}