set magnetic field for tracks

[u/mrichter/AliRoot.git] / TRD / AliTRDtrack.h

diff --git a/TRD/AliTRDtrack.h b/TRD/AliTRDtrack.h
index 6a92c3981221ad9a7c6345f932772b4597c2e014..d192fd5041d6d760ba91eadc63d7838f72c2eddb 100644 (file)
--- a/TRD/AliTRDtrack.h
+++ b/TRD/AliTRDtrack.h
@@ -13,6 +13,7 @@
 
 class AliTRDcluster;
 class AliTPCtrack;
+class AliESDtrack;
 
 const unsigned kMAX_CLUSTERS_PER_TRACK=210; 
 
@@ -27,6 +28,7 @@ public:
                const Double_t cc[15], Double_t xr, Double_t alpha);  
    AliTRDtrack(const AliTRDtrack& t);    
    AliTRDtrack(const AliKalmanTrack& t, Double_t alpha); 
+   AliTRDtrack(const AliESDtrack& t);    
 
    Int_t    Compare(const TObject *o) const;
    void     CookdEdx(Double_t low=0.05, Double_t up=0.70);   
@@ -42,6 +44,7 @@ public:
 
    void     GetCovariance(Double_t cov[15]) const;  
    Double_t GetdEdx()  const {return fdEdx;}
+   Double_t GetPIDsignal()  const {return GetdEdx();}
    Double_t GetEta()   const {return fE;}
 
    void     GetExternalCovariance(Double_t cov[15]) const ;   
@@ -67,9 +70,26 @@ public:
    Double_t GetX()    const {return fX;}
    Double_t GetY()    const {return fY;}
    Double_t GetZ()    const {return fZ;}
+   UInt_t * GetBackupIndexes()  {return fIndexBackup;}
+   UInt_t * GetIndexes()  {return fIndex;}
+  
+
+   Double_t GetYat(Double_t xk) const {     
+//-----------------------------------------------------------------
+// This function calculates the Y-coordinate of a track at the plane x=xk.
+// Needed for matching with the TOF (I.Belikov)
+//-----------------------------------------------------------------
+      Double_t c1=fC*fX - fE, r1=TMath::Sqrt(1.- c1*c1);
+      Double_t c2=fC*xk - fE, r2=TMath::Sqrt(1.- c2*c2);
+      return fY + (xk-fX)*(c1+c2)/(r1+r2);
+   }
+   void SetStop(Bool_t stop) {fStopped=stop;}
+   Bool_t GetStop() const {return fStopped;}
 
    Int_t    PropagateTo(Double_t xr, Double_t x0=8.72, Double_t rho=5.86e-3);
    void     ResetCovariance();   
+   void     ResetCovariance(Float_t mult);   
+   void ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
    Int_t    Rotate(Double_t angle);
 
    void     SetdEdx(Float_t dedx) {fdEdx=dedx;}  
@@ -85,6 +105,8 @@ public:
 
    Int_t    Update(const AliTRDcluster* c, Double_t chi2, UInt_t i, 
                    Double_t h01);
+   Int_t    UpdateMI(const AliTRDcluster* c, Double_t chi2, UInt_t i, 
+                   Double_t h01, Int_t plane);
 
   //
   void GetBarrelTrack(AliBarrelTrack *track);
@@ -102,7 +124,7 @@ protected:
 
    Double_t fAlpha;       // rotation angle
    Double_t fX;           // running local X-coordinate of the track (time bin)
-
+   Bool_t   fStopped;     // track stop indication
 
    Double_t fY;             // Y-coordinate of the track
    Double_t fZ;             // Z-coordinate of the track
@@ -117,6 +139,7 @@ protected:
    Double_t fCcy, fCcz, fCce, fCct, fCcc; // parameters   
    
    UInt_t  fIndex[kMAX_CLUSTERS_PER_TRACK];  // global indexes of clusters  
+   UInt_t  fIndexBackup[kMAX_CLUSTERS_PER_TRACK]; //backup indexes of clusters - used in iterations
    Float_t fdQdl[kMAX_CLUSTERS_PER_TRACK];   // cluster amplitudes corrected 
                                              // for track angles