adding possibility to select only primary protons for momentum resolution correction...

diff --git a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.cxx b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.cxx
index c3c7aa4586aa525d6b45deb6ba90d53a2caa2cf6..45100d401175d41359bd688282f9a668da4f5d0c 100644 (file)
--- a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.cxx
+++ b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.cxx
@@ -63,6 +63,7 @@ AliFemtoEventReaderESDChainKine::AliFemtoEventReaderESDChainKine():
   fMagFieldSign(0),
   fReadV0(0),
   isKaonAnalysis(kFALSE),
+  isProtonAnalysis(kFALSE),
   fOnlyPrimaries(kFALSE)
 
 {
@@ -90,6 +91,7 @@ AliFemtoEventReaderESDChainKine::AliFemtoEventReaderESDChainKine(const AliFemtoE
   fMagFieldSign(0),
   fReadV0(0),
   isKaonAnalysis(kFALSE),
+  isProtonAnalysis(kFALSE),
   fOnlyPrimaries(kFALSE)
 
 
@@ -111,6 +113,7 @@ AliFemtoEventReaderESDChainKine::AliFemtoEventReaderESDChainKine(const AliFemtoE
   fMagFieldSign = aReader.fMagFieldSign;
   fReadV0 = aReader.fReadV0;
   isKaonAnalysis = aReader.isKaonAnalysis;
+  isProtonAnalysis = aReader.isProtonAnalysis;
   fOnlyPrimaries = aReader.fOnlyPrimaries;
 
 }
@@ -143,6 +146,7 @@ AliFemtoEventReaderESDChainKine& AliFemtoEventReaderESDChainKine::operator=(cons
   fMagFieldSign = aReader.fMagFieldSign;
   fReadV0 = aReader.fReadV0;
   isKaonAnalysis = aReader.isKaonAnalysis;
+  isProtonAnalysis = aReader.isProtonAnalysis;
   fOnlyPrimaries = aReader.fOnlyPrimaries;
 
   return *this;
@@ -818,6 +822,9 @@ AliFemtoEvent* AliFemtoEventReaderESDChainKine::ReturnHbtEvent()
     if (isKaonAnalysis == true && trackCopy->GetPDGPid() != 321) {
       trackAccept = false;
     }
+    if (isProtonAnalysis == true && TMath::Abs(trackCopy->GetPDGPid()) != 2212) {
+      trackAccept = false;
+    }
 
     if (tGoodMomentum==true  && trackAccept == true)
     {
@@ -1137,6 +1144,11 @@ void AliFemtoEventReaderESDChainKine::SetKaonAnalysis(Bool_t a)
   isKaonAnalysis = a;
 }
 
+void AliFemtoEventReaderESDChainKine::SetProtonAnalysis(Bool_t a)
+{
+  isProtonAnalysis = a;
+}
+
 void AliFemtoEventReaderESDChainKine::SetOnlyPrimaries(Bool_t a)
 {
   fOnlyPrimaries = a;

diff --git a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.h b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.h
index e5a645b18a8b7269f4e807e210420750c9551744..df34538161bdb12f755b83a5a07048f4d8eebf39 100644 (file)
--- a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.h
+++ b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChainKine.h
@@ -69,6 +69,7 @@ class AliFemtoEventReaderESDChainKine : public AliFemtoEventReader
   void GetGlobalPositionAtGlobalRadiiThroughTPC(AliESDtrack *track, Float_t bfield, Float_t globalPositionsAtRadii[9][3]);
   void SetMagneticFieldSign(int s);
   void SetKaonAnalysis(Bool_t);
+  void SetProtonAnalysis(Bool_t);
   void SetOnlyPrimaries(Bool_t);
 
  protected:
@@ -97,6 +98,7 @@ class AliFemtoEventReaderESDChainKine : public AliFemtoEventReader
   int            fMagFieldSign;     // Magnetic field sign
   bool           fReadV0;
   Bool_t isKaonAnalysis; // switch for Kaon analysis
+  Bool_t isProtonAnalysis; // switch for Proton analysis
   Bool_t fOnlyPrimaries; // switch to analyze only primaries
 
 #ifdef __ROOT__