Changes related to the extraction of the V0 finder into a separate class (A. Dainese...
[u/mrichter/AliRoot.git] / ITS / AliITStrackerSA.h
1 #ifndef ALIITSTRACKERSA_H
2 #define ALIITSTRACKERSA_H 
3
4
5
6 #include "AliITSgeomTGeo.h"
7 #include "AliITStrackerMI.h"
8
9 /* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
10  * See cxx source for full Copyright notice                               */
11
12 /* $Id$ */
13
14
15 ////////////////////////////////////////////////////
16 //  Stand alone tracker class                     //
17 //  Origin:  Elisabetta Crescio                   //
18 //  e-mail:  crescio@to.infn.it                   //
19 //  adapted for cosmics by A.Dainese              //
20 ////////////////////////////////////////////////////
21
22 class AliITSclusterTable;
23 class AliITStrackSA;
24 class AliESDVertex;
25 class AliESDEvent;
26 class AliITSVertexer;
27 class TTree;
28 class TArrayD;
29
30 class AliITStrackerSA : public AliITStrackerMI {
31
32
33  public:
34
35   AliITStrackerSA();
36   AliITStrackerSA(const Char_t *geom);
37   AliITStrackerSA(const Char_t *geom,AliESDVertex *vert);
38   AliITStrackerSA(const Char_t *geom,AliITSVertexer *vertexer);
39   AliITStrackerSA(const AliITStrackerSA& tracker);
40   AliITStrackerSA& operator=(const AliITStrackerSA& source);
41   virtual ~AliITStrackerSA();  
42   virtual Int_t Clusters2Tracks(AliESDEvent *event);
43   Int_t FindTracks(AliESDEvent* event);
44
45   AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
46   void StoreTrack(AliITStrackV2 *t,AliESDEvent *event) const; 
47   Int_t FindTrackLowChiSquare() const;
48   Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerMI::LoadClusters(cf); SetClusterTree(cf); return rc;}
49   void SetVertex(AliESDVertex *vtx){fVert = vtx;}
50   void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
51
52   void SetOutwardFinding() {fInwardFlag=kFALSE;}
53   void SetInwardFinding() {fInwardFlag=kTRUE;}
54   void SetOuterStartLayer(Int_t osl = 0) {
55     if(osl>(AliITSgeomTGeo::GetNLayers()-2)) AliWarning("Minimum Number of layers is 2. OuterStartLayer set to Nlayers-2");
56     fOuterStartLayer = TMath::Min(AliITSgeomTGeo::GetNLayers()-2,osl);
57   }
58   Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
59   void SetInnerStartLayer(Int_t isl = 5) {
60     if(isl<1) AliWarning("Minimum Number of layers is 2. InnerStartLayer set to 1");
61     fInnerStartLayer = TMath::Max(1,isl);
62   }
63   Int_t GetInnerStartLayer() const {return fInnerStartLayer;}
64
65   void SetSAFlag(Bool_t fl){fITSStandAlone=fl;}  // StandAlone flag setter
66   Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag getter
67   void SetFixedWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
68   void SetCalculatedWindowSizes(Int_t n=10, Float_t phimin=0.002, Float_t phimax=0.0145, Float_t lambdamin=0.003, Float_t lambdamax=0.008);
69
70   void SetMinNPoints(Int_t np){fMinNPoints=np;}
71   Int_t GetMinNPoints() const {return fMinNPoints;}
72   void SetMinimumChargeSDDSSD(Float_t minq=0.){fMinQ=minq;}
73   enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
74  protected:
75
76   //Initialization
77   void Init();
78   void ResetForFinding();
79   void UpdatePoints();
80   Bool_t SetFirstPoint(Int_t lay, Int_t clu, Double_t* primaryVertex);
81   static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
82                      Double_t x3,Double_t y3);
83
84   Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp); 
85
86   static Int_t   FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2, 
87                            Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
88   static Int_t   FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2, 
89                        Float_t x3, Float_t y3,Float_t& a, Float_t& b, 
90                        Float_t& c);
91  
92   static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
93   static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5, 
94               Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
95               Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16, 
96               Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
97  
98   Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, 
99                        AliITStrackSA* trs,Double_t zvertex,Int_t flagp); 
100
101   void GetCoorAngles(AliITSRecPoint* cl,Double_t &phi,Double_t &lambda,Float_t &x,Float_t &y,Float_t &z,Double_t* vertex);
102   void GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz);
103
104   AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
105   void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
106
107
108   Double_t fPhiEstimate; //Estimation of phi angle on next layer
109   Bool_t fITSStandAlone; //Tracking is performed in the ITS alone if kTRUE
110   Float_t fPoint1[2];   //! coord. of 1-st point to evaluate the curvature
111   Float_t fPoint2[2];   //! coord. of 2-nd point to evaluate the curvature
112   Float_t fPoint3[2];   //! coord. of 3-rd point to evaluate the curvature
113   Float_t fPointc[2];   //! current point coord (for curvature eval.)
114   Double_t fLambdac;    //! current value of the Lambda angle in the window
115   Double_t fPhic;       //! current value of the Phi angle in the window
116   Float_t fCoef1;       //! param. of the equation of the circ. approx a layer
117   Float_t fCoef2;       //! param. of the equation of the circ. approx a layer
118   Float_t fCoef3;       //! param. of the equation of the circ. approx a layer
119   Int_t fNloop;         //  Number of iterqations on phi and lambda windows
120   Double_t *fPhiWin;    // phi window sizes
121   Double_t *fLambdaWin; // lambda window sizes
122   AliESDVertex *fVert;        //! primary vertex
123   AliITSVertexer *fVertexer;  //! vertexer 
124   TObjArray *fListOfTracks;   //! container for found tracks 
125   TObjArray *fListOfSATracks; //! container for found SA tracks 
126   TTree *fITSclusters;        //! pointer to ITS tree of clusters
127   Bool_t fInwardFlag;       // set to kTRUE for inward track finding
128   Int_t fOuterStartLayer;     // Outward search for tracks with <6 points: outer layer to start from
129   Int_t fInnerStartLayer;     // Inward search for tracks with <6 points: inner layer to start from
130   Int_t fMinNPoints;        // minimum number of clusters for a track
131   Float_t fMinQ;              // lower cut on cluster charge (SDD and SSD)
132
133   TClonesArray** fCluLayer; //! array with clusters 
134   TClonesArray** fCluCoord; //! array with cluster info
135
136   ClassDef(AliITStrackerSA,9)
137 };
138
139 #endif