From ac8deadfc02466ac8501edece84f6f37e02e759c Mon Sep 17 00:00:00 2001
From: akisiel <akisiel@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Fri, 4 Nov 2011 10:33:12 +0000
Subject: [PATCH] Add AOD setter

---
 .../AliFemtoUser/AliFemtoPairCutAntiGamma.cxx | 27 +++++++++++++------
 .../AliFemtoUser/AliFemtoPairCutAntiGamma.h   |  2 ++
 2 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.cxx b/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.cxx
index 2a9abb2b122..cd8f86ab693 100644
--- a/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.cxx
+++ b/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.cxx
@@ -35,7 +35,8 @@ AliFemtoPairCutAntiGamma::AliFemtoPairCutAntiGamma():
   AliFemtoShareQualityPairCut(),
   fMaxEEMinv(0.0),
   fMaxDTheta(0.0),
-  fDTPCMin(0)
+  fDTPCMin(0),
+  fUseAOD(kFALSE)
 {
 }
 //__________________
@@ -43,11 +44,13 @@ AliFemtoPairCutAntiGamma::AliFemtoPairCutAntiGamma(const AliFemtoPairCutAntiGamm
   AliFemtoShareQualityPairCut(c),
   fMaxEEMinv(0.0),
   fMaxDTheta(0.0),
-  fDTPCMin(0)
+  fDTPCMin(0),
+  fUseAOD(kFALSE)
 { 
   fMaxEEMinv = c.fMaxEEMinv;
   fMaxDTheta = c.fMaxDTheta;
   fDTPCMin = c.fDTPCMin;
+  fUseAOD = c.fUseAOD;
 }
 
 //__________________
@@ -79,13 +82,16 @@ bool AliFemtoPairCutAntiGamma::Pass(const AliFemtoPair* pair){
   }
 
   bool tempTPCEntrance = true;
-  
-  double distx = pair->Track1()->Track()->NominalTpcEntrancePoint().x() - pair->Track2()->Track()->NominalTpcEntrancePoint().x();
-  double disty = pair->Track1()->Track()->NominalTpcEntrancePoint().y() - pair->Track2()->Track()->NominalTpcEntrancePoint().y();
-  double distz = pair->Track1()->Track()->NominalTpcEntrancePoint().z() - pair->Track2()->Track()->NominalTpcEntrancePoint().z();
-  double dist = sqrt(distx*distx + disty*disty + distz*distz);
+ 
+  if(!fUseAOD)
+    {
+      double distx = pair->Track1()->Track()->NominalTpcEntrancePoint().x() - pair->Track2()->Track()->NominalTpcEntrancePoint().x();
+      double disty = pair->Track1()->Track()->NominalTpcEntrancePoint().y() - pair->Track2()->Track()->NominalTpcEntrancePoint().y();
+      double distz = pair->Track1()->Track()->NominalTpcEntrancePoint().z() - pair->Track2()->Track()->NominalTpcEntrancePoint().z();
+      double dist = sqrt(distx*distx + disty*disty + distz*distz);
 
-  tempTPCEntrance = dist > fDTPCMin;
+      tempTPCEntrance = dist > fDTPCMin;
+    }
 
 
   if (temp && tempTPCEntrance) {
@@ -140,3 +146,8 @@ void AliFemtoPairCutAntiGamma::SetTPCEntranceSepMinimum(double dtpc)
 {
   fDTPCMin = dtpc;
 }
+
+void AliFemtoPairCutAntiGamma::SetUseAOD(Bool_t UseAOD)
+{
+  fUseAOD = UseAOD;
+}
diff --git a/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.h b/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.h
index 064cb794c01..f0d85741ad6 100644
--- a/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.h
+++ b/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.h
@@ -41,11 +41,13 @@ public:
   void SetMaxEEMinv(Double_t maxeeminv);
   void SetMaxThetaDiff(Double_t maxdtheta);
   void SetTPCEntranceSepMinimum(double dtpc);
+  void SetUseAOD(Bool_t UseAOD);
   
  protected:
   Double_t fMaxEEMinv; // Maximum allowed ee Minv
   Double_t fMaxDTheta; // Maximum polar angle difference
   Double_t fDTPCMin;          // Minimum allowed pair nominal separation at the entrance to the TPC
+  Bool_t fUseAOD; // Use AODs
 
 
 #ifdef __ROOT__
-- 
2.43.0