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 c413e0ed40eef03a8a3917ae797181f7795f869f..7b208b13dc0bfcd7098448168f643dd4eaa98c5e 100644 (file)
--- a/ITS/AliITStrackMI.h
+++ b/ITS/AliITStrackMI.h
@@ -10,6 +10,7 @@
 //     dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
 //-------------------------------------------------------------------------
 
+/* $Id$ */
 
 /*****************************************************************************
  *                          December 18, 2000                                *
@@ -27,7 +28,6 @@
 
 #include <AliKalmanTrack.h>
 
-#include "AliITSrecoV2.h"
 #include "AliITStrackV2.h"
 
 class AliESDtrack;
@@ -36,13 +36,11 @@ 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(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz, Double_t chi2, Int_t i);  
-  Int_t CorrectForMaterial(Double_t d, Double_t x0=21.82);
-
-  void UpdateESDtrack(ULong_t flags);
+  Bool_t UpdateMI(const AliCluster *c, Double_t chi2, Int_t i);  
 
   void SetReconstructed(Bool_t sr=kTRUE){fReconstructed = sr;}  
   Bool_t GetReconstructed() const {return fReconstructed;}
@@ -70,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;}
@@ -95,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 
 
@@ -113,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
@@ -124,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