Create the magnetic field map in the reconstruction macros (Yuri)

[u/mrichter/AliRoot.git] / MUON / AliMUONTrackExtrap.h

diff --git a/MUON/AliMUONTrackExtrap.h b/MUON/AliMUONTrackExtrap.h
index e3312f6b20ef9de89af6150fc881280f95e4df43..53ab61c917d9a9461aea75ebb00cfe6b3f6b9af4 100644 (file)
--- a/MUON/AliMUONTrackExtrap.h
+++ b/MUON/AliMUONTrackExtrap.h
@@ -15,6 +15,7 @@
 //////////////////////////////////////////////////////////////
 
 #include <TObject.h>
+#include <TMatrixD.h>
 
 class AliMagF;
 class AliMUONTrackParam;
@@ -27,18 +28,42 @@ class AliMUONTrackExtrap : public TObject
        /// Destructor
   virtual ~AliMUONTrackExtrap(){};
   
-       /// set field map
-  static void SetField(const AliMagF* magField) {fgkField = magField;}
+  static void SetField(const AliMagF* magField);
+  
+  /// return kTRUE if the field is switched ON
+  static Bool_t IsFieldON() {return fgFieldON;}
   
   static Double_t GetImpactParamFromBendingMomentum(Double_t bendingMomentum);
   static Double_t GetBendingMomentumFromImpactParam(Double_t impactParam);
   
-  static void ExtrapToZ(AliMUONTrackParam *trackParam, Double_t Z);
+  // Linearly extrapolate track parameters and covariances
+  static void LinearExtrapToZ(AliMUONTrackParam* trackParam, Double_t zEnd, Bool_t updatePropagator = kFALSE);
+  
+  // Extrapolate track parameters in magnetic field
+  static void ExtrapToZ(AliMUONTrackParam *trackParam, Double_t zEnd);
+  
+  // Extrapolate track parameters and covariances in magnetic field
   static void ExtrapToZCov(AliMUONTrackParam* trackParam, Double_t zEnd, Bool_t updatePropagator = kFALSE);
-  static void ExtrapToStation(AliMUONTrackParam *trackParamIn, Int_t station, AliMUONTrackParam *trackParamOut);
+  
+  // Extrapolate track parameters to vertex, corrected for multiple scattering and energy loss effects
+  // Add branson correction resolution and energy loss fluctuation to parameter covariances
+  static void ExtrapToVertex(AliMUONTrackParam* trackParam,
+                             Double_t xVtx, Double_t yVtx, Double_t zVtx,
+                             Double_t errXVtx, Double_t errYVtx);
+  
+  // Extrapolate track parameters to vertex, corrected for multiple scattering effects only
+  // Add branson correction resolution to parameter covariances
+  static void ExtrapToVertexWithoutELoss(AliMUONTrackParam* trackParam,
+                                        Double_t xVtx, Double_t yVtx, Double_t zVtx,
+                                        Double_t errXVtx, Double_t errYVtx);
+  
+  // Extrapolate track parameters to vertex, corrected for energy loss effects only
+  // Add dispersion due to multiple scattering and energy loss fluctuation to parameter covariances
+  static void ExtrapToVertexWithoutBranson(AliMUONTrackParam* trackParam, Double_t zVtx);
+  
+  // Extrapolate track parameters to vertex without multiple scattering and energy loss corrections
+  // Add dispersion due to multiple scattering to parameter covariances
   static void ExtrapToVertexUncorrected(AliMUONTrackParam* trackParam, Double_t zVtx);
-  static void ExtrapToVertex(AliMUONTrackParam *trackParam, Double_t xVtx, Double_t yVtx, Double_t zVtx,
-                            Bool_t CorrectForMCS = kTRUE, Bool_t CorrectForEnergyLoss = kTRUE);
   
   static Double_t TotalMomentumEnergyLoss(AliMUONTrackParam* trackParam, Double_t xVtx, Double_t yVtx, Double_t zVtx);
   
@@ -50,6 +75,10 @@ class AliMUONTrackExtrap : public TObject
   
  private:
   static const AliMagF* fgkField;              //!< field map
+  static const Double_t fgkSimpleBPosition;     //!< position of the dipole
+  static const Double_t fgkSimpleBLength;       //!< length of the dipole
+  static       Double_t fgSimpleBValue;         //!< magnetic field value at the centre
+  static       Bool_t   fgFieldON;              //!< kTRUE if the field is switched ON
   static const Bool_t   fgkUseHelix;           //!< Tell whether to use Helix or not (default is Runge-Kutta)
   static const Int_t    fgkMaxStepNumber;      //!< Maximum number of steps for track extrapolation
   static const Double_t fgkHelixStepLength;    //!< Step lenght for track extrapolation (used in Helix)
@@ -68,12 +97,26 @@ class AliMUONTrackExtrap : public TObject
   static void ConvertTrackParamForExtrap(AliMUONTrackParam* trackParam, Double_t forwardBackward, Double_t *v3);
   static void RecoverTrackParam(Double_t *v3, Double_t Charge, AliMUONTrackParam* trackParam);
   
+  static void ExtrapToVertex(AliMUONTrackParam* trackParam,
+                             Double_t xVtx, Double_t yVtx, Double_t zVtx,
+                             Double_t errXVtx, Double_t errYVtx,
+                             Bool_t correctForMCS, Bool_t correctForEnergyLoss);
+  
   static void AddMCSEffectInAbsorber(AliMUONTrackParam* trackParam, Double_t pathLength, Double_t f0, Double_t f1, Double_t f2);
-  static void GetAbsorberCorrectionParam(Double_t trackXYZIn[3], Double_t trackXYZOut[3], Double_t &pathLength,
-                                        Double_t &f0, Double_t &f1, Double_t &f2, Double_t &meanRho);
+  static void CorrectMCSEffectInAbsorber(AliMUONTrackParam* param,
+                                         Double_t xVtx, Double_t yVtx, Double_t zVtx,
+                                         Double_t errXVtx, Double_t errYVtx,
+                                         Double_t absZBeg, Double_t pathLength, Double_t f0, Double_t f1, Double_t f2);
+  static void CorrectELossEffectInAbsorber(AliMUONTrackParam* param, Double_t eLoss, Double_t sigmaELoss2);
+  static Bool_t GetAbsorberCorrectionParam(Double_t trackXYZIn[3], Double_t trackXYZOut[3], Double_t pTotal,
+                                           Double_t &pathLength, Double_t &f0, Double_t &f1, Double_t &f2,
+                                           Double_t &meanRho, Double_t &totalELoss, Double_t &sigmaELoss2);
+  
+  static Double_t BetheBloch(Double_t pTotal, Double_t pathLength, Double_t rho, Double_t atomicA, Double_t atomicZ);
+  static Double_t EnergyLossFluctuation2(Double_t pTotal, Double_t pathLength, Double_t rho, Double_t atomicA, Double_t atomicZ);
   
-  static Double_t TotalMomentumEnergyLoss(Double_t pTotal, Double_t pathLength, Double_t rho);
-  static Double_t ApproximateBetheBloch(Double_t beta2);
+  static void Cov2CovP(const TMatrixD &param, TMatrixD &cov);
+  static void CovP2Cov(const TMatrixD &param, TMatrixD &cov);
   
   static void ExtrapOneStepHelix(Double_t charge, Double_t step, Double_t *vect, Double_t *vout);
   static void ExtrapOneStepHelix3(Double_t field, Double_t step, Double_t *vect, Double_t *vout);