64414911d44b8964d4dc6b3d71aa4bdc7b23db45
[u/mrichter/AliRoot.git] / ITS / AliITStrackerSA.h
1 #ifndef ALIITSTRACKERSA_H
2 #define ALIITSTRACKERSA_H 
3
4
5
6 #include "AliITStrackerMI.h"
7
8 /* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
9  * See cxx source for full Copyright notice                               */
10
11 ////////////////////////////////////////////////////
12 //  Stand alone tracker class                     //
13 //  Origin:  Elisabetta Crescio                   //
14 //  e-mail:  crescio@to.infn.it                   //
15 ////////////////////////////////////////////////////
16
17 class AliITSclusterTable;
18 class AliITStrackSA;
19 class AliESDVertex;
20 class AliITSVertexer;
21 class TTree;
22 class TArrayD;
23
24 class AliITStrackerSA : public AliITStrackerMI {
25
26
27  public:
28
29   AliITStrackerSA();
30   AliITStrackerSA(AliITSgeom *geom);
31   AliITStrackerSA(AliITSgeom *geom,AliESDVertex *vert);
32   AliITStrackerSA(AliITSgeom *geom,AliITSVertexer *vertexer);
33   AliITStrackerSA(const AliITStrackerSA& tracker);
34   AliITStrackerSA& operator=(const AliITStrackerSA& /* trkr */);
35   virtual ~AliITStrackerSA();  
36   virtual Int_t Clusters2Tracks(AliESD *event){Int_t rc = AliITStrackerMI::Clusters2Tracks(event); if(!rc) rc=FindTracks(event); return rc;}
37   Int_t FindTracks(AliESD* event);
38
39   AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex);
40
41   Int_t FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const;
42   Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerMI::LoadClusters(cf); SetClusterTree(cf);SetSixPoints(kTRUE); return rc;}
43   void SetVertex(AliESDVertex *vtx){fVert = vtx;}
44   void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
45   void SetSixPoints(Bool_t sp = kFALSE){fSixPoints = sp;}
46   Bool_t GetSixPoints() const {return fSixPoints;}
47   void SetWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
48
49   enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
50
51  protected:
52
53   //Initialization
54   void Init();
55   void ResetForFinding();
56   void UpdatePoints();
57
58   static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
59                      Double_t x3,Double_t y3);
60
61   Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp); 
62
63   static Int_t   FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2, 
64                            Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
65   static Int_t   FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2, 
66                        Float_t x3, Float_t y3,Float_t& a, Float_t& b, 
67                        Float_t& c);
68  
69   static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
70   static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5, 
71               Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
72               Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16, 
73               Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
74  
75   Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, 
76                        AliITStrackSA* trs,Double_t zvertex,Int_t flagp); 
77
78  
79
80   void GetCoorAngles(AliITSclusterV2* cl,Int_t module,Double_t &phi,Double_t &lambda,Float_t &x,Float_t &y,Float_t &z,Double_t* vertex);
81   void GetCoorErrors(AliITSclusterV2* cl, Int_t module,Float_t &sx,Float_t &sy, Float_t &sz);
82
83   AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
84   void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
85
86
87   Double_t fPhiEstimate; //Estimation of phi angle on next layer
88   Float_t fPoint1[2];   //! coord. of 1-st point to evaluate the curvature
89   Float_t fPoint2[2];   //! coord. of 2-nd point to evaluate the curvature
90   Float_t fPoint3[2];   //! coord. of 3-rd point to evaluate the curvature
91   Float_t fPointc[2];   //! current point coord (for curvature eval.)
92   Double_t fLambdac;    //! current value of the Lambda angle in the window
93   Double_t fPhic;       //! current value of the Phi angle in the window
94   Float_t fCoef1;       //! param. of the equation of the circ. approx a layer
95   Float_t fCoef2;       //! param. of the equation of the circ. approx a layer
96   Float_t fCoef3;       //! param. of the equation of the circ. approx a layer
97   Int_t fNloop;         //  Number of iterqations on phi and lambda windows
98   Double_t *fPhiWin;    // phi window sizes
99   Double_t *fLambdaWin; // lambda window sizes
100   AliESDVertex *fVert;        //! primary vertex
101   AliITSVertexer *fVertexer;  //! vertexer 
102   AliITSgeom *fGeom;          //! ITS geometry
103   TObjArray *fListOfTracks;   //! container for found tracks 
104   TTree *fITSclusters;        //! pointer to ITS tree of clusters
105   Bool_t fSixPoints;          // If true 6/6 points are required (default). 5/6 otherwise
106
107   TClonesArray** fCluLayer; //! array with clusters 
108   TClonesArray** fCluCoord; //! array with cluster info
109
110   ClassDef(AliITStrackerSA,2)
111 };
112
113 #endif
114