code updates, Add Muon Corrections, KT4index bug fix

[u/mrichter/AliRoot.git] / ITS / AliITStrackMI.h

diff --git a/ITS/AliITStrackMI.h b/ITS/AliITStrackMI.h
index 07c62a126048df1b9d19b609196e3d5cdee87126..7b208b13dc0bfcd7098448168f643dd4eaa98c5e 100644 (file)
--- a/ITS/AliITStrackMI.h
+++ b/ITS/AliITStrackMI.h
@@ -28,7 +28,6 @@
 
 #include <AliKalmanTrack.h>
 
-#include "AliITSRecoParam.h"
 #include "AliITStrackV2.h"
 
 class AliESDtrack;
@@ -37,8 +36,9 @@ class AliESDtrack;
 class AliITStrackMI : public AliITStrackV2 {
 public:
   AliITStrackMI();
-  AliITStrackMI(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
+  AliITStrackMI(AliESDtrack& t,Bool_t c=kFALSE);
   AliITStrackMI(const AliITStrackMI& t);
+  virtual ~AliITStrackMI()  {if (fWinner) fWinner->fWinner = 0;} // release associated seed
   Int_t GetProlongationFast(Double_t alpha, Double_t xr,Double_t &y, Double_t &z);
   Bool_t UpdateMI(const AliCluster *c, Double_t chi2, Int_t i);  
 
@@ -68,8 +68,10 @@ public:
   void SetDnorm(Int_t i, Float_t d) {fDnorm[i]=d;}
   Float_t GetSigmaY(Int_t i) const {return fSigmaY[i];}
   Float_t GetSigmaZ(Int_t i) const {return fSigmaZ[i];}
+  Float_t GetSigmaYZ(Int_t i) const {return fSigmaYZ[i];}
   void SetSigmaY(Int_t i, Float_t s) {fSigmaY[i]=s;}
   void SetSigmaZ(Int_t i, Float_t s) {fSigmaZ[i]=s;}
+  void SetSigmaYZ(Int_t i, Float_t s) {fSigmaYZ[i]=s;}
   Float_t GetNDeadZone() const {return fNDeadZone;}
   void SetNDeadZone(Float_t d) {fNDeadZone=d;}
   Int_t* ClIndex() {return fClIndex;}
@@ -93,17 +95,20 @@ public:
   void SetGoldV0(Bool_t g) {fGoldV0=g;}
   Float_t GetChi22() const {return fChi22;}
   void SetChi22(Float_t c) {fChi22=c;}
-  Float_t GetDeadZoneProbability() const {return fDeadZoneProbability;}
-  void SetDeadZoneProbability(Float_t d) {fDeadZoneProbability=d;}
-
-  Double_t GetPredictedChi2MI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz) const;
+  Float_t GetDeadZoneProbability(Int_t ilayer) const {return fDeadZoneProbability[ilayer];}
+  void SetDeadZoneProbability(Int_t ilayer,Float_t d) {fDeadZoneProbability[ilayer]=d;}
+  //
+  AliITStrackMI*  GetWinner()        const {return fWinner;}
+  void   SetWinner(AliITStrackMI* p);
+  //
+  Double_t GetPredictedChi2MI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz, Double_t covyz=0.) const;
   Bool_t IsGoldPrimary();
 protected:
 
   Float_t fNUsed;                          // number of shared clusters
   Float_t fNSkipped;                       // number of skipped clusters
   Float_t fNDeadZone;                     // number of clusters in dead zone
-  Float_t fDeadZoneProbability;          // probability to cross dead zone
+  Float_t fDeadZoneProbability[6];        // probability to cross dead zone
   Bool_t  fReconstructed;                 // reconstructed - accepted flag
   Float_t fChi2MIP[12];                   // MIP chi squres 
 
@@ -111,6 +116,7 @@ protected:
   Float_t fDz[12];           //dz in layer
   Float_t fSigmaY[12];       //sigma y 
   Float_t fSigmaZ[12];       //sigma z
+  Float_t fSigmaYZ[12];       //covariance of y and z
   Float_t fNy[6];              //expected size of cluster
   Float_t fNz[6];              //expected size of cluster
   Float_t fD[2];            //distance to the vertex
@@ -122,10 +128,23 @@ protected:
   Float_t fdEdxMismatch;    
   Bool_t fConstrain;        //indication of the vertex constrain
   Int_t  fClIndex[6];       //cluster Index
+  AliITStrackMI* fWinner;   //! pointer on winner candidate
   Bool_t fGoldV0;           //corresponding gold V0 found
-  ClassDef(AliITStrackMI,1)   //ITS reconstructed track
+  //
+ private:
+  AliITStrackMI &operator=(const AliITStrackMI &tr);
+
+  ClassDef(AliITStrackMI,4)   //ITS reconstructed track
 };
 
+inline void AliITStrackMI::SetWinner(AliITStrackMI* p) 
+{
+  // connect winning hypothesis with the seed
+  if (fWinner) fWinner->fWinner = 0; // release previous winner
+  fWinner = p; 
+  if (p) p->fWinner = this;
+}
+
 #endif