Add AOD setter

diff --git a/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.cxx b/PWG2/FEMTOSCOPY/AliFemtoUser/AliFemtoPairCutAntiGamma.cxx
index 2a9abb2b122bffc327f4c712e608e4408570a0f7..cd8f86ab693087165414ba02e9d806a955ecd333 100644 (file)
--- 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 064cb794c017e735e4cd287374a1c1df7785c565..f0d85741ad645e37aa03d8ad3a9e6f9587594c00 100644 (file)
--- 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__