Adding include path to allow compilation of CleanGeom task
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALTracker.h
1 //========================================================================  
2 // Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved.  
3 // See cxx source for full Copyright notice                                
4 //========================================================================  
5 //                       
6 //                       Class AliEMCALTracker 
7 //                      -----------------------
8 // Implementation of the track matching method between barrel tracks and
9 // EMCAL clusters.
10 // Besides algorithm implementation, some cuts are required to be set
11 // in order to define, for each track, an acceptance window where clusters
12 // are searched to find best match (if any).
13 // The class accepts as input an ESD container, and works directly on it,
14 // simply setting, for each of its tracks, the fEMCALindex flag, for each
15 // track which is matched to a cluster.
16 // In order to use method, one must launch PropagateBack().
17 //
18 // ------------------------------------------------------------------------
19 // author: A. Pulvirenti (alberto.pulvirenti@ct.infn.it)
20 //=========================================================================
21
22 #ifndef ALIEMCALTRACKER_H
23 #define ALIEMCALTRACKER_H
24
25 #include "AliTracker.h"
26 #include <TMath.h>
27 #include <TVector3.h>
28 class TList;
29 class TTree;
30 class TObjArray;
31 class AliESDEvent;
32 class AliVCluster;
33 class AliESDCaloCluster;
34 class AliEMCALRecPoint;
35 class AliEMCALGeometry;
36
37 class AliEMCALTracker : public AliTracker 
38 {
39  public:
40   AliEMCALTracker();
41   AliEMCALTracker(const AliEMCALTracker &t);
42   AliEMCALTracker& operator=(const AliEMCALTracker &source);
43   virtual ~AliEMCALTracker() {Clear();}
44         
45   virtual void        Clear(Option_t *option="ALL");
46   virtual Int_t       Clusters2Tracks(AliESDEvent*) {return -1;}
47           void        InitParameters();
48   virtual Int_t       LoadClusters(TTree*);
49           Int_t       LoadClusters(AliESDEvent* esd);
50           Int_t       LoadTracks(AliESDEvent* esd);
51   virtual Int_t       PropagateBack(AliESDEvent* esd);
52   virtual Int_t       RefitInward(AliESDEvent*) {return -1;}
53   virtual void        UnloadClusters();
54   virtual AliCluster* GetCluster(Int_t) const {return NULL;}
55           void        SetCutEta(Double_t value) {fCutEta=value;}
56           void        SetCutPhi(Double_t value) {fCutPhi=value;}
57           void        SetGeometry(AliEMCALGeometry *geom) {fGeom=geom;}
58           void        SetCutPt(Double_t value)   {fCutPt=value;}
59           void        SetCutNITS(Double_t value) {fCutNITS=value;}
60           void        SetCutNTPC(Double_t value) {fCutNTPC=value;}
61           void        SetStepLength(Float_t length) {fStep=length;}
62           void        SetTrackCorrectionMode(Option_t *option);
63           void        SetEMCalSurfaceDistance(Double_t d) {fEMCalSurfaceDistance = d;}
64
65   enum {        kUnmatched = -99999 };
66         
67   class  AliEMCALMatchCluster : public TObject
68   {
69    public:
70     AliEMCALMatchCluster(Int_t ID, AliEMCALRecPoint *recPoint);
71     AliEMCALMatchCluster(Int_t ID, AliESDCaloCluster *caloCluster);
72     virtual ~AliEMCALMatchCluster() { }
73     //----------------------------------------------------------------------------
74     Int_t     Index() const {return fIndex;}
75     Double_t  X() const {return fX;}
76     Double_t  Y() const {return fY;} 
77     Double_t  Z() const {return fZ;}
78    private:
79     Int_t     fIndex;  // index of cluster in its native container (ESD or TClonesArray)
80     Double_t  fX;      // global X position
81     Double_t  fY;      // global Y position
82     Double_t  fZ;      // global Z position
83   };
84    
85  private:
86   Int_t  FindMatchedCluster(AliESDtrack *track);
87   enum ETrackCorr { 
88     kTrackCorrNone  = 0, // do not correct for energy loss
89     kTrackCorrMMB   = 1, // use MeanMaterialBudget() function to evaluate correction
90   };
91
92   Double_t    fCutPt;           // mimimum pT cut on tracks
93   Double_t    fCutNITS;         // mimimum number of track hits in the ITS
94   Double_t    fCutNTPC;         // mimimum number of track hits in the TPC
95         
96   Float_t     fStep;            // Length of each step in propagation
97   ETrackCorr  fTrackCorrMode;   // Material budget correction mode
98   Float_t     fEMCalSurfaceDistance; // EMCal surface distance
99   Double_t    fClusterWindow;   // Select clusters in the window to be matched to tracks
100   Double_t    fCutEta;        // cut on eta difference
101   Double_t    fCutPhi;        // cut on phi difference
102   Bool_t            fITSTrackSA;      // If ITS Tracks  
103         
104   TObjArray  *fTracks;          //! collection of tracks
105   TObjArray  *fClusters;        //! collection of EMCAL clusters (ESDCaloCluster or EMCALRecPoint)
106         
107   AliEMCALGeometry *fGeom;      //! EMCAL geometry
108   
109   ClassDef(AliEMCALTracker, 7)  // EMCAL "tracker"
110 };
111 #endif