adding map include file to solve compilation issue (bug https://savannah.cern.ch...
[u/mrichter/AliRoot.git] / HLT / ITS / tracking / AliITStrackerHLT.h
1 #ifndef ALIITSTRACKERHLT_H
2 #define ALIITSTRACKERHLT_H
3 /* Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6
7 class TTree;
8 class TTreeSRedirector;
9 class AliESDEvent;
10 class AliESDtrack;
11
12 class AliITSChannelStatus;
13 class AliITSDetTypeRec;
14 class AliITSRecoParam;
15 #include "AliHLTITSTrack.h" 
16 #include "AliHLTITSDetector.h"
17 #include "AliHLTITSLayer.h"
18
19 #include <TObjArray.h>
20
21 #include "AliITSRecPoint.h"
22 #include "AliTracker.h"
23 #include "AliHLTITSTrack.h"
24
25 //-------------------------------------------------------------------------
26 class AliITStrackerHLT : public AliTracker {
27 public:
28
29   
30   void StartLoadClusters( Int_t NClusters );
31   void LoadCluster( const AliITSRecPoint &cluster);
32   void Reconstruct( AliExternalTrackParam *tracksTPC, int *tracksTPCLab, int nTPCTracks );
33
34   AliHLTITSTrack *Tracks() const { return fTracks;}
35   Int_t NTracks() const { return fNTracks;}
36   AliHLTITSTrack *ITSOutTracks() const { return fITSOutTracks;}
37   Int_t NITSOutTracks() const { return fNITSOutTracks;}
38
39   Bool_t TransportToX( AliExternalTrackParam *t, double x ) const;
40   Bool_t TransportToPhiX( AliExternalTrackParam *t, double phi, double x ) const;
41
42   AliITStrackerHLT();
43   AliITStrackerHLT(const Char_t *geom);
44   virtual ~AliITStrackerHLT();
45   AliCluster *GetCluster(Int_t index) const;
46   virtual Bool_t GetTrackPoint(Int_t index, AliTrackPoint& p) const;
47   virtual Bool_t GetTrackPointTrackingError(Int_t index, 
48                         AliTrackPoint& p, const AliESDtrack *t);
49   AliITSRecPoint *GetClusterLayer(Int_t layn, Int_t ncl) const
50                   {return fLayers[layn].GetCluster(ncl);}
51   Int_t GetNumberOfClustersLayer(Int_t layn) const 
52                         {return fLayers[layn].GetNumberOfClusters();}
53   Int_t LoadClusters(TTree *cf);
54   void UnloadClusters();
55   
56   Int_t Clusters2Tracks(AliESDEvent *event);
57   Int_t PropagateBack(AliESDEvent *event);
58   Int_t RefitInward(AliESDEvent *event);
59   
60
61   static Int_t CorrectForTPCtoITSDeadZoneMaterial(AliHLTITSTrack *t);
62
63
64   AliHLTITSLayer    & GetLayer(Int_t layer) const;
65   AliHLTITSDetector & GetDetector(Int_t layer, Int_t n) const {return GetLayer(layer).GetDetector(n); }
66  
67   void FollowProlongationTree(AliHLTITSTrack * otrack);
68   Int_t FitOutward(AliHLTITSTrack * track );
69
70   void Init();
71
72
73 protected:
74
75   const AliITSRecoParam *GetRecoParam() const { return fRecoParam; }
76   Bool_t ComputeRoad(AliHLTITSTrack* track,Int_t ilayer,Int_t idet,Double_t &zmin,Double_t &zmax,Double_t &ymin,Double_t &ymax) const;
77   
78   
79   void CookLabel(AliKalmanTrack *t,Float_t wrong) const;
80   void CookLabel(AliHLTITSTrack *t,Float_t wrong) const;
81
82   void BuildMaterialLUT(TString material);
83   
84   Int_t CorrectForPipeMaterial(AliHLTITSTrack *t, bool InwardDirection=1);
85   Int_t CorrectForShieldMaterial(AliHLTITSTrack *t, Int_t  shieldindex, bool InwardDirection=1);
86   Int_t CorrectForLayerMaterial(AliHLTITSTrack *t, Int_t layerindex, bool InwardDirection=1);
87   void UpdateESDtrack(AliESDtrack *tESD,AliHLTITSTrack* track, ULong_t flags) const;
88   
89   Bool_t LocalModuleCoord(Int_t ilayer,Int_t idet,const AliHLTITSTrack *track,
90                           Float_t &xloc,Float_t &zloc) const;
91
92
93
94 // method to be used for Plane Efficiency evaluation
95
96   // 
97
98   const AliITSRecoParam *fRecoParam; //! 
99
100   AliHLTITSLayer* fLayers; //!
101   
102   Double_t fSPDdetzcentre[4];            // centres of SPD modules in z
103   
104   Int_t fUseTGeo;                        // use TGeo to get material budget
105
106   Float_t fxOverX0Pipe;                  // material budget
107   Float_t fxTimesRhoPipe;                // material budget
108   Float_t fxOverX0Shield[2];             // material budget
109   Float_t fxTimesRhoShield[2];           // material budget
110   Float_t fxOverX0Layer[6];              // material budget
111   Float_t fxTimesRhoLayer[6];            // material budget
112
113   AliHLTITSTrack *fTracks; // array of its-updated tracks
114   AliHLTITSTrack *fITSOutTracks; // array of tracks, fitted outward with ITS only
115   int fNTracks;// n tracks
116   int fNITSOutTracks;// n out tracks
117   double fLoadTime; //
118   double fRecoTime; //
119   int fNEvents; //
120   AliITSRecPoint *fClusters; //!
121   int fNClusters; //
122
123 private:
124   AliITStrackerHLT(const AliITStrackerHLT &tracker);
125   AliITStrackerHLT & operator=(const AliITStrackerHLT &tracker);  
126   ClassDef(AliITStrackerHLT,0)   //HLT ITS tracker
127 };
128
129
130
131
132 /////////////////////////////////////////////////////////
133 /////////////////////////////////////////////////////////
134 /////////////////////////////////////////////////////////
135
136
137
138
139
140 inline void AliITStrackerHLT::CookLabel(AliKalmanTrack *t,Float_t wrong) const {
141   //--------------------------------------------------------------------
142   //This function "cooks" a track label. If label<0, this track is fake.
143   //--------------------------------------------------------------------
144    Int_t tpcLabel=t->GetLabel();
145    if (tpcLabel<0) return;
146    AliTracker::CookLabel(t,wrong);
147    if (tpcLabel!=TMath::Abs(t->GetLabel())){
148      t->SetFakeRatio(1.);
149    }
150    if (tpcLabel !=t->GetLabel()) {
151      t->SetLabel(-tpcLabel);      
152    }
153 }
154
155
156
157 #endif