Adding pt and eta control plots, increasing ClassDef version

[u/mrichter/AliRoot.git] / PWG0 / esdTrackCuts / AliESDtrackCuts.h

diff --git a/PWG0/esdTrackCuts/AliESDtrackCuts.h b/PWG0/esdTrackCuts/AliESDtrackCuts.h
index 991e8d94b838e36b2762d078d841a16404ceb29d..74ce6964d97af87882c1f2fcc552d657d345ae8f 100644 (file)
--- a/PWG0/esdTrackCuts/AliESDtrackCuts.h
+++ b/PWG0/esdTrackCuts/AliESDtrackCuts.h
@@ -23,26 +23,28 @@
 #ifndef ALIESDTRACKCUTS_H
 #define ALIESDTRACKCUTS_H
 
-#include <TNamed.h>
 #include <TF1.h>
 #include <TH2.h>
+#include "AliAnalysisCuts.h"
 
 class AliESD;
+class AliESDEvent;
 class AliESDtrack;
 class AliLog;
 class TTree;
 
-class AliESDtrackCuts : public TNamed 
+class AliESDtrackCuts : public AliAnalysisCuts
 {
-
 public:
-  AliESDtrackCuts();
-  AliESDtrackCuts(Char_t* name, Char_t* title="");
+  AliESDtrackCuts(const Char_t* name = "AliESDtrackCuts", const Char_t* title = "");
   virtual ~AliESDtrackCuts();
-
+  Bool_t IsSelected(TObject* obj)
+       {return AcceptTrack((AliESDtrack*)obj);}
   Bool_t AcceptTrack(AliESDtrack* esdTrack);
   TObjArray* GetAcceptedTracks(AliESD* esd);
   Int_t CountAcceptedTracks(AliESD* esd);
+  TObjArray* GetAcceptedTracks(AliESDEvent* esd);
+  Int_t CountAcceptedTracks(AliESDEvent* esd);
 
   virtual Long64_t Merge(TCollection* list);
   virtual void Copy(TObject &c) const;
@@ -65,7 +67,11 @@ public:
   void SetMinNsigmaToVertex(Float_t sigma=1e99)       {fCutNsigmaToVertex = sigma;}
   void SetRequireSigmaToVertex(Bool_t b=kTRUE )       {fCutSigmaToVertexRequired = b;}
 
-  // track kinmatic cut setters  
+  // getters
+  Float_t GetMinNsigmaToVertex()       { return fCutNsigmaToVertex;}
+  Bool_t GetRequireSigmaToVertex( )    { return fCutSigmaToVertexRequired;}
+
+  // track kinmatic cut setters
   void SetPRange(Float_t r1=0, Float_t r2=1e99)       {fPMin=r1;   fPMax=r2;}
   void SetPtRange(Float_t r1=0, Float_t r2=1e99)      {fPtMin=r1;  fPtMax=r2;}
   void SetPxRange(Float_t r1=-1e99, Float_t r2=1e99)  {fPxMin=r1;  fPxMax=r2;}
@@ -74,8 +80,6 @@ public:
   void SetEtaRange(Float_t r1=-1e99, Float_t r2=1e99) {fEtaMin=r1; fEtaMax=r2;}
   void SetRapRange(Float_t r1=-1e99, Float_t r2=1e99) {fRapMin=r1; fRapMax=r2;}
 
-  Float_t GetMinNsigmaToVertex() { return fCutNsigmaToVertex; } 
-
   //######################################################
   void SetHistogramsOn(Bool_t b=kFALSE) {fHistogramsOn = b;}
   void DefineHistograms(Int_t color=1);
@@ -154,13 +158,16 @@ protected:
   TH1F* fhDZNormalized[2];            //->
   TH2F* fhDXYvsDZNormalized[2];       //->
   TH1F* fhNSigmaToVertex[2];          //->
+  
+  TH1F* fhPt[2];                      //-> pt of esd tracks
+  TH1F* fhEta[2];                     //-> eta of esd tracks
 
   TF1*  ffDTheoretical;               //-> theoretical distance to vertex normalized (2d gauss)
 
   TH1F*  fhCutStatistics;             //-> statistics of what cuts the tracks did not survive
   TH2F*  fhCutCorrelation;            //-> 2d statistics plot
 
-  ClassDef(AliESDtrackCuts, 1)
+  ClassDef(AliESDtrackCuts, 2)
 };