Adding functionality so that AliHLTMUONRawDataHistoComponent histograms raw data...

[u/mrichter/AliRoot.git] / ANALYSIS / AliESDtrackCuts.h

diff --git a/ANALYSIS/AliESDtrackCuts.h b/ANALYSIS/AliESDtrackCuts.h
index 2563f6a134ae2532925c2fe30035203a826b801a..e8a3d1e7a2272841be18845235fedecdcd38e524 100644 (file)
--- a/ANALYSIS/AliESDtrackCuts.h
+++ b/ANALYSIS/AliESDtrackCuts.h
@@ -5,18 +5,16 @@
 //  track-to-vertex cut and a number of kinematic cuts. Two methods
 //  can be used to figure out if an ESD track survives the cuts:
 //  AcceptTrack which takes a single AliESDtrack as argument and
-//  returns kTRUE/kFALSE or GetAcceptedTracks which takes an AliESD
+//  returns kTRUE/kFALSE or GetAcceptedTracks which takes an AliESDEvent
 //  object and returns an TObjArray (of AliESDtracks) with the tracks
 //  in the ESD that survived the cuts.
 //
 //
-//  TODO: 
+//  TODO:
 //  - add functionality to save and load cuts
-//  - add different ways to make track to vertex cut
 //  - add histograms for kinematic cut variables?
 //  - upper and lower cuts for all (non-boolean) cuts
 //  - update print method
-//  - is there a smarter way to manage the cuts?
 //  - put comments to each variable
 //
 
@@ -27,7 +25,6 @@
 #include <TH2.h>
 #include "AliAnalysisCuts.h"
 
-class AliESD;
 class AliESDEvent;
 class AliESDtrack;
 class AliLog;
@@ -38,15 +35,17 @@ class AliESDtrackCuts : public AliAnalysisCuts
 public:
   AliESDtrackCuts(const Char_t* name = "AliESDtrackCuts", const Char_t* title = "");
   virtual ~AliESDtrackCuts();
+
   Bool_t IsSelected(TObject* obj)
        {return AcceptTrack((AliESDtrack*)obj);}
   Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
+
   Bool_t AcceptTrack(AliESDtrack* esdTrack);
-  TObjArray* GetAcceptedTracks(AliESD* esd);
-  Int_t CountAcceptedTracks(AliESD* esd);
-  TObjArray* GetAcceptedTracks(AliESDEvent* esd);
+  TObjArray* GetAcceptedTracks(AliESDEvent* esd,Bool_t bTPC = kFALSE);
   Int_t CountAcceptedTracks(AliESDEvent* esd);
 
+  static AliESDtrack* GetTPCOnlyTrack(AliESDEvent* esd, Int_t iTrack);
+
   virtual Long64_t Merge(TCollection* list);
   virtual void Copy(TObject &c) const;
   AliESDtrackCuts(const AliESDtrackCuts& pd);  // Copy Constructor
@@ -67,11 +66,33 @@ public:
   // track to vertex cut setters
   void SetMinNsigmaToVertex(Float_t sigma=1e10)       {fCutNsigmaToVertex = sigma;}
   void SetRequireSigmaToVertex(Bool_t b=kTRUE )       {fCutSigmaToVertexRequired = b;}
+  void SetDCAToVertex(Float_t dist=1e10)              {fCutDCAToVertex = dist;}
+  void SetDCAToVertexXY(Float_t dist=1e10)            {fCutDCAToVertexXY = dist;}
 
   // getters
-  Float_t GetMinNsigmaToVertex()       { return fCutNsigmaToVertex;}
-  Bool_t GetRequireSigmaToVertex( )    { return fCutSigmaToVertexRequired;}
 
+  Int_t   GetMinNClusterTPC()        const   { return fCutMinNClusterTPC;}
+  Int_t   GetMinNClustersITS()       const   { return fCutMinNClusterITS;}
+  Float_t GetMaxChi2PerClusterTPC()  const   { return fCutMaxChi2PerClusterTPC;}
+  Float_t GetMaxChi2PerClusterITS()  const   { return fCutMaxChi2PerClusterITS;}
+  Bool_t  GetRequireTPCRefit()       const   { return fCutRequireTPCRefit;}
+  Bool_t  GetRequireITSRefit()       const   { return fCutRequireITSRefit;}
+  Bool_t  GetAcceptKingDaughters()   const   { return fCutAcceptKinkDaughters;}
+  void    GetMaxCovDiagonalElements(Float_t& c1, Float_t& c2, Float_t& c3, Float_t& c4, Float_t& c5) 
+      {c1 = fCutMaxC11; c2 = fCutMaxC22; c3 = fCutMaxC33; c4 = fCutMaxC44; c5 = fCutMaxC55;}
+  Float_t GetMinNsigmaToVertex()     const   { return fCutNsigmaToVertex;}
+  Float_t GetDCAToVertex()           const   { return fCutDCAToVertex;}
+  Float_t GetDCAToVertexXY()         const   { return fCutDCAToVertexXY;}
+  Bool_t  GetRequireSigmaToVertex( ) const   { return fCutSigmaToVertexRequired;}
+
+  void GetPRange(Float_t& r1, Float_t& r2)   {r1=fPMin;   r2=fPMax;}
+  void GetPtRange(Float_t& r1, Float_t& r2)  {r1=fPtMin;  r2=fPtMax;}
+  void GetPxRange(Float_t& r1, Float_t& r2)  {r1=fPxMin;  r2=fPxMax;}
+  void GetPyRange(Float_t& r1, Float_t& r2)  {r1=fPyMin;  r2=fPyMax;}
+  void GetPzRange(Float_t& r1, Float_t& r2)  {r1=fPzMin;  r2=fPzMax;}
+  void GetEtaRange(Float_t& r1, Float_t& r2) {r1=fEtaMin; r2=fEtaMax;}
+  void GetRapRange(Float_t& r1, Float_t& r2) {r1=fRapMin; r2=fRapMax;}
+  
   // track kinmatic cut setters
   void SetPRange(Float_t r1=0, Float_t r2=1e10)       {fPMin=r1;   fPMax=r2;}
   void SetPtRange(Float_t r1=0, Float_t r2=1e10)      {fPtMin=r1;  fPtMax=r2;}
@@ -88,7 +109,7 @@ public:
   void SaveHistograms(const Char_t* dir = 0);
   void DrawHistograms();
 
-  Float_t GetSigmaToVertex(AliESDtrack* esdTrack);
+  static Float_t GetSigmaToVertex(AliESDtrack* esdTrack);
   
   static void EnableNeededBranches(TTree* tree);
 
@@ -100,7 +121,7 @@ public:
 protected:
   void Init(); // sets everything to 0
 
-  enum { kNCuts = 21 };
+  enum { kNCuts = 23 };
 
   //######################################################
   // esd track quality cuts
@@ -125,6 +146,8 @@ protected:
   // track to vertex cut
   Float_t fCutNsigmaToVertex;         // max number of estimated sigma from track-to-vertex
   Bool_t  fCutSigmaToVertexRequired;  // cut track if sigma from track-to-vertex could not be calculated
+  Float_t fCutDCAToVertex;            // track-to-vertex cut in absolute distance
+  Float_t fCutDCAToVertexXY;          // track-to-vertex cut in absolute distance in xy-plane
 
   // esd kinematics cuts
   Float_t fPMin,   fPMax;             // definition of the range of the P
@@ -153,13 +176,14 @@ protected:
 
   TH1F* fhDXY[2];                     //->
   TH1F* fhDZ[2];                      //->
+  TH1F* fhDXYDZ[2];                   //-> absolute distance sqrt(dxy**2 + dz**2) to vertex
   TH2F* fhDXYvsDZ[2];                 //->
 
   TH1F* fhDXYNormalized[2];           //->
   TH1F* fhDZNormalized[2];            //->
   TH2F* fhDXYvsDZNormalized[2];       //->
   TH1F* fhNSigmaToVertex[2];          //->
-  
+
   TH1F* fhPt[2];                      //-> pt of esd tracks
   TH1F* fhEta[2];                     //-> eta of esd tracks