694ebe504be509429ea4854b210bf1367aab7358
[u/mrichter/AliRoot.git] / MUON / AliMUONTrackParam.h
1 #ifndef ALIMUONTRACKPARAM_H
2 #define ALIMUONTRACKPARAM_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /*$Id$*/
7 // Revision of includes 07/05/2004
8
9 /// \ingroup rec
10 /// \class AliMUONTrackParam
11 /// \brief Track parameters in ALICE dimuon spectrometer
12 ///
13 ////////////////////////////////////////////////////
14 /// Track parameters in ALICE dimuon spectrometer
15 ////////////////////////////////////////////////////
16
17 #include <TObject.h>
18
19 class AliESDMuonTrack;
20 class AliMagF;
21
22 class AliMUONTrackParam : public TObject 
23 {
24  public:
25   AliMUONTrackParam(); // Constructor
26   virtual ~AliMUONTrackParam(){} // Destructor
27   
28   AliMUONTrackParam(const AliMUONTrackParam& rhs);// copy constructor (should be added per default !)
29   AliMUONTrackParam& operator=(const  AliMUONTrackParam& rhs);// (should be added per default !)
30
31   void GetParamFrom(const AliESDMuonTrack& esdMuonTrack);
32   void SetParamFor(AliESDMuonTrack& esdMuonTrack);
33
34
35   // Get and Set methods for data
36   Double_t GetInverseBendingMomentum(void) const {return fInverseBendingMomentum;}
37   void     SetInverseBendingMomentum(Double_t InverseBendingMomentum) {fInverseBendingMomentum = InverseBendingMomentum;}
38   Double_t GetBendingSlope(void) const {return fBendingSlope;}
39   void     SetBendingSlope(Double_t BendingSlope) {fBendingSlope = BendingSlope;}
40   Double_t GetNonBendingSlope(void) const {return fNonBendingSlope;}
41   void     SetNonBendingSlope(Double_t NonBendingSlope) {fNonBendingSlope = NonBendingSlope;}
42   Double_t GetZ(void) const {return fZ;}
43   void     SetZ(Double_t Z) {fZ = Z;}
44   Double_t GetBendingCoor(void) const {return fBendingCoor;}
45   void     SetBendingCoor(Double_t BendingCoor) {fBendingCoor = BendingCoor;}
46   Double_t GetNonBendingCoor(void) const {return fNonBendingCoor;}
47   void     SetNonBendingCoor(Double_t NonBendingCoor) {fNonBendingCoor = NonBendingCoor;}
48   Double_t Px() const;  // return px
49   Double_t Py() const;  // return py
50   Double_t Pz() const;  // return pz
51   Double_t P()  const;  // return total momentum
52
53   void ExtrapToZ(Double_t Z);
54   void ExtrapToStation(Int_t Station, AliMUONTrackParam *TrackParam);
55   void ExtrapToVertex(Double_t xVtx, Double_t yVtx, Double_t zVtx);  // extrapolation to vertex through the absorber (with true vertex) 
56    void BransonCorrection(Double_t xVtx, Double_t yVtx, Double_t zVtx); // makes Branson correction with true vertex  
57    // returns total momentum after energy loss correction in the absorber
58   Double_t TotalMomentumEnergyLoss(Double_t thetaLimit, Double_t pTotal, Double_t theta);
59   void FieldCorrection(Double_t Z); // makes simple magnetic field correction through the absorber 
60
61   void ExtrapOneStepHelix(Double_t charge, Double_t step, 
62                           Double_t *vect, Double_t *vout) const;
63   void ExtrapOneStepHelix3(Double_t field, Double_t step, 
64                            Double_t *vect, Double_t *vout) const;
65
66   void ExtrapOneStepRungekutta(Double_t charge, Double_t step, 
67                                Double_t* vect, Double_t* vout) const;
68   
69   virtual void Print(Option_t* opt="") const;
70  
71   void SetField(const AliMagF* magField) {fkField = magField;}
72
73
74  protected:
75  private:
76   Double_t fInverseBendingMomentum; ///< Inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
77   Double_t fBendingSlope; ///< Bending slope (cm ** -1)
78   Double_t fNonBendingSlope; ///< Non bending slope (cm ** -1)
79   Double_t fZ; ///< Z coordinate (cm)
80   Double_t fBendingCoor; ///< bending coordinate (cm)
81   Double_t fNonBendingCoor; ///< non bending coordinate (cm)
82
83   const AliMagF* fkField;     //!< field map
84
85   void SetGeant3Parameters(Double_t *VGeant3, Double_t ForwardBackward);
86   void GetFromGeant3Parameters(Double_t *VGeant3, Double_t Charge);
87
88   void GetField(Double_t *Position, Double_t *Field) const;
89
90   ClassDef(AliMUONTrackParam, 1) // Track parameters in ALICE dimuon spectrometer
91     };
92         
93 #endif