ITS-upgrade simulation/reconstruction code (C. Terrevoli, A. Mastroserio)
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITStrackerUpgrade.h
CommitLineData
1d9af2d5 1#ifndef ALIITSTRACKERUPGRADE_H
2#define ALIITSTRACKERUPGRADE_H
3
4
5
6#include "AliITStrackerMI.h"
7#include "AliITSlayerUpgrade.h"
8#include "AliITStrackMI.h"
9
10/* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
11 * See cxx source for full Copyright notice */
12
13/* $Id$ */
14////////////////////////////////////////////////////
15// ITS Upgrade Stand alone tracker class //
16// Authors: A.Mastroserio C.Terrevoli //
17// e-mail:annalisa.mastroserio@cern.ch //
18// cristina.terrevoli@ba.infn.it //
19// tracks are saved as AliITStrackV2 objects //
20////////////////////////////////////////////////////
21
22#include <TClonesArray.h>
23
24class AliITSclusterTable;
25class AliITStrackSA;
26class AliESDVertex;
27class AliESDEvent;
28class AliITSVertexer;
29class TTree;
30class TArrayD;
31
32class AliITStrackerUpgrade : public AliITStrackerMI {
33
34
35 public:
36
37 AliITStrackerUpgrade();
38 AliITStrackerUpgrade(const AliITStrackerUpgrade& tracker);
39 AliITStrackerUpgrade& operator=(const AliITStrackerUpgrade& source);
40 virtual ~AliITStrackerUpgrade();
41 virtual Int_t Clusters2Tracks(AliESDEvent *event);
42 Int_t FindTracks(AliESDEvent* event,Bool_t useAllClusters=kFALSE);
43 Int_t PropagateBack(AliESDEvent *event);
44 Int_t CorrectForPipeMaterial(AliITStrackMI *t, TString direction="inward");
45 Int_t RefitInward(AliESDEvent *event);//{return 0;}
46 AliCluster *GetCluster(Int_t index) const;
47
48 AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
49 void StoreTrack(AliITStrackV2 *t,AliESDEvent *event, Bool_t pureSA) /*const*/;
50 Int_t FindTrackLowChiSquare() const;
51 Int_t LoadClusters(TTree *clusTree);
52 void SetVertex(AliESDVertex *vtx){fVert = vtx;}
53 void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
54
55 void SetOutwardFinding() {fInwardFlag=kFALSE;}
56 void SetInwardFinding() {fInwardFlag=kTRUE;}
57 void SetOuterStartLayer(Int_t osl = 0) {
58 if(osl>(fNLayer-2)) AliWarning("Minimum Number of layers is 2. OuterStartLayer set to Nlayers-2");
59 fOuterStartLayer = TMath::Min(fNLayer-2,osl);
60 }
61 Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
62 void SetInnerStartLayer(Int_t isl = 5) {
63 if(isl<1) AliWarning("Minimum Number of layers is 2. InnerStartLayer set to 1");
64 fInnerStartLayer = TMath::Max(1,isl);
65 }
66 Int_t GetInnerStartLayer() const {return fInnerStartLayer;}
67 void SetSAFlag(Bool_t fl){fITSStandAlone=fl;} // StandAlone flag setter
68 Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag gett
69 void SetFixedWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
70 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);
71 void UnloadClusters();
72 void SetMinNPoints(Int_t np){fMinNPoints=np;}
73 Int_t GetMinNPoints() const {return fMinNPoints;}
74 void SetMinimumChargeSDDSSD(Float_t minq=0.){fMinQ=minq;}
75 enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
76
77 protected:
78
79 //Initialization
80 void Init();
81 void CreateLayers(Int_t iEvent);
82 void ResetForFinding();
83 void UpdatePoints();
84 Bool_t SetFirstPoint(Int_t lay, Int_t clu, Double_t* primaryVertex);
85 static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
86 Double_t x3,Double_t y3);
87
88 Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp);
89
90 static Int_t FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2,
91 Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
92 static Int_t FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2,
93 Float_t x3, Float_t y3,Float_t& a, Float_t& b,
94 Float_t& c);
95
96 static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
97 static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5,
98 Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
99 Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16,
100 Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
101
102 Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow,
103 AliITStrackSA* trs,Double_t zvertex,Int_t flagp);
104
105 void GetCoorAngles(AliITSRecPoint* cl,Double_t &phi,Double_t &lambda,Double_t &x,Double_t &y,Double_t &z,Double_t* vertex);
106 void GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz);
107 AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
108 void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
109 Bool_t RefitAtBase(Double_t x, AliITStrackMI *track,
110 const Int_t *clusters);
111 Int_t UpdateMI(AliITStrackMI* track, const AliITSRecPoint* cl,Double_t chi2,Int_t layer) const;
112 Int_t CorrectForLayerMaterial(AliITStrackMI *t, Int_t layerindex, Double_t oldGlobXYZ[3], TString direction="inward");
113 Double_t GetPredictedChi2MI(AliITStrackMI* track, const AliITSRecPoint *cluster,Int_t layer);
114 static Int_t GetError(Int_t layer,const AliITSRecPoint*cl,
115 Float_t tgl,Float_t tgphitr,Float_t expQ,
116 Float_t &erry,Float_t &errz,Float_t &covyz,
117 Bool_t addMisalErr=kTRUE);
118
119 static Int_t GetErrorOrigRecPoint(const AliITSRecPoint*cl,
120 Float_t &erry,Float_t &errz,Float_t &covyz);
121 static void GetNTeor(Int_t layer,const AliITSRecPoint* cl,
122 Float_t tgl,Float_t tgphitr,
123 Float_t &ny,Float_t &nz);
124
125 Int_t fNLayer;//number of layer in ITSUpgrade
126 Double_t fPhiEstimate; //Estimation of phi angle on next layer
127 Bool_t fITSStandAlone; //Tracking is performed in the ITS alone if kTRUE
128 Float_t fPoint1[2]; //! coord. of 1-st point to evaluate the curvature
129 Float_t fPoint2[2]; //! coord. of 2-nd point to evaluate the curvature
130 Float_t fPoint3[2]; //! coord. of 3-rd point to evaluate the curvature
131 Float_t fPointc[2]; //! current point coord (for curvature eval.)
132 Double_t fLambdac; //! current value of the Lambda angle in the window
133 Double_t fPhic; //! current value of the Phi angle in the window
134 Float_t fCoef1; //! param. of the equation of the circ. approx a layer
135 Float_t fCoef2; //! param. of the equation of the circ. approx a layer
136 Float_t fCoef3; //! param. of the equation of the circ. approx a layer
137 Int_t fNloop; // Number of iterqations on phi and lambda windows
138 Double_t *fPhiWin; // phi window sizes
139 Double_t *fLambdaWin; // lambda window sizes
140 AliESDVertex *fVert; //! primary vertex
141 AliITSVertexer *fVertexer; //! vertexer
142 TClonesArray *fListOfTracks; //! container for found tracks
143 TClonesArray *fListOfSATracks; //! container for found SA tracks
144 TTree *fITSclusters; //! pointer to ITS tree of clusters
145 Bool_t fInwardFlag; // set to kTRUE for inward track finding
146 Int_t fOuterStartLayer; // Outward search for tracks with <6 points: outer layer to start from
147 Int_t fInnerStartLayer; // Inward search for tracks with <6 points: inner layer to start from
148 Int_t fMinNPoints; // minimum number of clusters for a track
149 Float_t fMinQ; // lower cut on cluster charge (SDD and SSD)
150 AliITSlayerUpgrade** fLayers;
151
152 TClonesArray** fCluLayer; //! array with clusters
153 TClonesArray** fCluCoord; //! array with cluster info
154
155 ClassDef(AliITStrackerUpgrade,1)
156};
157
158#endif
159
160