Print removed
[u/mrichter/AliRoot.git] / ITS / AliITSNeuralTrack.h
1 #ifndef ALIITSNEURALTRACK_H
2 #define ALIITSNEURALTRACK_H
3
4 #include <TMatrixD.h>
5
6 class TObjArray;
7 class AliITSNeuralPoint;
8 //class AliITSVertex;
9 class AliITSIOTrack;
10
11 class AliITSNeuralTrack : public TObject {
12
13 public:
14                  AliITSNeuralTrack();
15                                 AliITSNeuralTrack(AliITSNeuralTrack &track);
16         virtual ~AliITSNeuralTrack();
17                 
18         // Points insertion and goodness evaluation
19
20         void     AssignLabel();
21         void     CleanSlot(Int_t i, Bool_t del = kFALSE);
22         void     CleanAllSlots(Bool_t del = kFALSE)        {Int_t i; for(i=0;i<6;i++) CleanSlot(i,del);}
23         void     GetModuleData(Int_t i, Int_t &mod, Int_t &pos);
24         void     Insert(AliITSNeuralPoint *point);
25         Bool_t   IsGood(Int_t min)                         {return (fCount >= min);}
26         Int_t    OccupationMask();
27         void     PrintLabels();
28                 
29         // Fit procedures
30                 
31         Bool_t   AddEL(Int_t layer, Double_t sign);
32         Bool_t   AddMS(Int_t layer);
33         void     ForceSign(Double_t sign)            { fC *= sign; }  // externally imposed trach charge
34         void     ResetChi2()                         { fChi2 = fNSteps = 0.0; }
35         Bool_t   SeedCovariance();
36         Int_t    PropagateTo(Double_t r);
37         Bool_t   Filter(AliITSNeuralPoint *test);
38         Bool_t   KalmanFit();
39         Bool_t   RiemannFit();
40         void     PrintState(Bool_t matrix);
41
42         // Getters
43
44         Int_t    GetLabel() const  {return fLabel;}
45         Int_t    GetCount() const  {return fCount;}
46         Double_t GetDt()    const  {return fDt;}
47         Double_t GetDz()    const;          
48         Double_t GetC()     const  {return fC;}
49         Double_t GetR()     const  {return fR;}
50         Double_t GetXC()    const  {return fXC;}
51         Double_t GetYC()    const  {return fYC;}
52         Double_t GetTanL()  const  {return fTanL;}
53         Double_t GetGamma() const;       
54         Double_t GetChi2()  const  {return fChi2;}
55         Double_t GetStateR() const {return fStateR;}
56         Double_t GetStatePhi() const {return fStatePhi;}
57         Double_t GetStateZ() const {return fStateZ;}
58         Double_t GetCovElement(Int_t i, Int_t j) const {return fMatrix(i,j);}
59         Double_t GetPhi(Double_t r) const;  // phi = gamma0 + asin(argphi(rho))
60         Double_t GetZ(Double_t r) const;    // z = dz + (tanl / C) * asin(argz(rho))
61
62         Double_t GetP()       {return GetPt() * (1.0 + fTanL * fTanL);}
63         Double_t GetPt()      {return 0.299792658 * 0.2 * fField * TMath::Abs(1./fC/100.);}
64         Double_t GetPz()      {return GetPt() * fTanL;}
65         Double_t GetE()       {return TMath::Sqrt(fMass*fMass + GetPt()*GetPt());}
66         Double_t GetLambda()  {return TMath::ATan(fTanL);}
67         Int_t    GetPDGcode() const {return fPDG;}
68         Double_t GetdEdX();
69
70         // Setters
71
72         void     SetFieldFactor(Double_t fact) {fField=fact;}
73         void     SetMass(Double_t mass)        {fMass=mass;}
74         void     SetPDGcode(Int_t code)        {fPDG=code;}
75         void     SetVertex(Double_t *pos, Double_t *err);
76         /*
77         void     SetRho(Double_t a)  {fStateR=a;}
78         void     SetPhi(Double_t a)  {fStatePhi=a;}
79         void     SetZ(Double_t a)    {fStateZ=a;}
80         void     SetDt(Double_t a)   {fDt=a;}
81         void     SetTanL(Double_t a) {fTanL=a;}
82         void     SetC(Double_t a)    {fC=a;}
83         void     SetChi2(Double_t a) {fChi2=a;}
84         void     SetGamma(Double_t a){fG0=a;}
85         void     SetDz(Double_t a)   {fDz=a;}
86         void     SetCovElement(Int_t i, Int_t j, Double_t a) {fMatrix(i,j)=a; if(i!=j) fMatrix(j,i)=a;}
87         */
88         AliITSIOTrack* ExportIOtrack(Int_t min);
89
90 private:
91         
92         Double_t ArgPhi(Double_t r) const;
93         Double_t ArgZ  (Double_t r) const;
94         Double_t ArgB  (Double_t r) const;
95         Double_t ArgC  (Double_t r) const;
96
97         Double_t fXC;       // X ofcurvature center
98         Double_t fYC;       // Y of curvature center
99         Double_t fR;        // curvature radius
100         Double_t fC;        // semi-curvature of the projected circle (signed)
101         Double_t fTanL;     // tangent of dip angle
102         Double_t fG0;       // phase coefficient
103         Double_t fDt;       // transverse impact parameter
104         Double_t fDz;       // longitudinal impact parameter
105
106         Double_t fStateR;   // state vector coordinates
107         Double_t fStatePhi; // state vector coordinates
108         Double_t fStateZ;   // state vector coordinates
109         TMatrixD fMatrix;   // covariance matrix
110         Double_t fChi2;     // square chi (calculated by Kalman filter)
111         Double_t fNSteps;   // number of Kalman steps
112
113         Double_t fMass;     // the particle mass
114         Double_t fField;    // B field = 0.2 * fField (Tesla)
115                 
116         Int_t    fPDG;      // PDG code of the recognized particle
117         Int_t    fLabel;    // the GEANT label most appearing among track recpoints
118         Int_t    fCount;    // number of counts of above label
119                 
120         AliITSNeuralPoint  fVertex;   // vertex position data
121         AliITSNeuralPoint *fPoint[6]; // track points
122
123         ClassDef(AliITSNeuralTrack, 1)
124 };
125
126 #endif