X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=CORRFW%2FAliCFTrackIsPrimaryCuts.h;h=4aa46739d9043e69f32e899bbd423faa402e8319;hb=5435cbd26c7f9f20d8be24ff7ba7292a89a68f8e;hp=fe485cba9f2f8fda6ceb6352564a6d18f950492d;hpb=c2c45ca49af059cfbf5b1b2e3a82a757b527f90b;p=u%2Fmrichter%2FAliRoot.git

diff --git a/CORRFW/AliCFTrackIsPrimaryCuts.h b/CORRFW/AliCFTrackIsPrimaryCuts.h
index fe485cba9f2..4aa46739d90 100644
--- a/CORRFW/AliCFTrackIsPrimaryCuts.h
+++ b/CORRFW/AliCFTrackIsPrimaryCuts.h
@@ -57,7 +57,8 @@
 #include "AliESDtrackCuts.h"
 class TBits;
 class AliESDtrack;
-class AliESD;
+class AliAODTrack;
+class AliVEvent;
 
 class AliCFTrackIsPrimaryCuts : public AliCFCutBase
 {
@@ -97,7 +98,7 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
   // please use indices from the enumeration below
   void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
   void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
-  virtual void SetEvtInfo(TObject* esd) ;
+  virtual void SetRecEventInfo(const TObject* esd);
 
   // indeces/counters for single selections
   enum { 
@@ -119,11 +120,12 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
  private:
   void SelectionBitMap(TObject* obj);
   void GetDCA(AliESDtrack* esdTrack);
+  void GetDCA(AliAODTrack* aodTrack);
   void DefineHistograms(); 		// books histograms and TList
   void Initialise();			// sets everything to 0
   void FillHistograms(TObject* obj, Bool_t b);
 					// Fills histograms before and after cuts
-  AliESD*  fESD;			// pointer to event, needed for SPD vertex
+  AliVEvent*  fEvt;			// pointer to event, needed for SPD vertex and DCA in AODs
   Bool_t   fUseSPDvertex;		// flag: calculate dca to SPD-vertex, off by default
   Bool_t   fUseTPCvertex;		// flag: calculate dca to TPC-vertex, off by default
   Double_t fMinDCAToVertexXY;		// cut value: min distance to main vertex in transverse plane