Fixed potential memory leak
[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 #include <TMatrixD.h>
19
20 class AliMUONVCluster;
21 class AliESDMuonTrack;
22
23 class AliMUONTrackParam : public TObject 
24 {
25  public:
26   AliMUONTrackParam(); // Constructor
27   virtual ~AliMUONTrackParam(); // Destructor
28   
29   AliMUONTrackParam(const AliMUONTrackParam& theMUONTrackParam);
30   AliMUONTrackParam& operator=(const  AliMUONTrackParam& theMUONTrackParam);
31
32   void GetParamFrom(const AliESDMuonTrack& esdMuonTrack);
33   void SetParamFor(AliESDMuonTrack& esdMuonTrack) const;
34   void GetParamFromDCA(const AliESDMuonTrack& esdMuonTrack);
35   void SetParamForDCA(AliESDMuonTrack& esdMuonTrack) const;
36   void GetParamFromUncorrected(const AliESDMuonTrack& esdMuonTrack);
37   void SetParamForUncorrected(AliESDMuonTrack& esdMuonTrack) const;
38   
39   void GetCovFrom(const AliESDMuonTrack& esdMuonTrack);
40   void SetCovFor(AliESDMuonTrack& esdMuonTrack) const;
41
42   // Get and Set methods for data
43         /// return Z coordinate (cm)
44   Double_t GetZ() const {return fZ;}
45         /// set Z coordinate (cm)
46   void     SetZ(Double_t z) {fZ = z;}
47         /// return non bending coordinate (cm)
48   Double_t GetNonBendingCoor() const {return fParameters(0,0);}
49         /// set non bending coordinate (cm)
50   void     SetNonBendingCoor(Double_t nonBendingCoor) {fParameters(0,0) = nonBendingCoor;}
51         /// return non bending slope (cm ** -1)
52   Double_t GetNonBendingSlope() const {return fParameters(1,0);}
53         /// set non bending slope (cm ** -1)
54   void     SetNonBendingSlope(Double_t nonBendingSlope) {fParameters(1,0) = nonBendingSlope;}
55         /// return bending coordinate (cm)
56   Double_t GetBendingCoor() const {return fParameters(2,0);}
57         /// set bending coordinate (cm)
58   void     SetBendingCoor(Double_t bendingCoor) {fParameters(2,0) = bendingCoor;}
59         /// return bending slope (cm ** -1)
60   Double_t GetBendingSlope() const {return fParameters(3,0);}
61         /// set bending slope (cm ** -1)
62   void     SetBendingSlope(Double_t bendingSlope) {fParameters(3,0) = bendingSlope;}
63         /// return inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
64   Double_t GetInverseBendingMomentum() const {return fParameters(4,0);}
65         /// set inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
66   void     SetInverseBendingMomentum(Double_t inverseBendingMomentum) {fParameters(4,0) = inverseBendingMomentum;}
67         /// return the charge (assumed forward motion)
68   Double_t GetCharge() const {return TMath::Sign(1.,fParameters(4,0));}
69         /// set the charge (assumed forward motion)
70   void     SetCharge(Double_t charge) {if (charge*fParameters(4,0) < 0.) fParameters(4,0) *= -1.;}
71   
72         /// return track parameters
73   const TMatrixD& GetParameters() const {return fParameters;}
74         /// set track parameters
75   void            SetParameters(const TMatrixD& parameters) {fParameters = parameters;}
76         /// add track parameters
77   void            AddParameters(const TMatrixD& parameters) {fParameters += parameters;}
78   
79   Double_t Px() const;  // return px
80   Double_t Py() const;  // return py
81   Double_t Pz() const;  // return pz
82   Double_t P()  const;  // return total momentum
83
84         /// return kTRUE if the covariance matrix exist, kFALSE if not
85   Bool_t    CovariancesExist() const {return (fCovariances) ? kTRUE : kFALSE;}
86   
87   const TMatrixD& GetCovariances() const;
88   void            SetCovariances(const TMatrixD& covariances);
89   void            SetCovariances(const Double_t matrix[5][5]);
90   void            SetVariances(const Double_t matrix[5][5]);
91   void            DeleteCovariances();
92   
93   const TMatrixD& GetPropagator() const;
94   void            ResetPropagator();
95   void            UpdatePropagator(const TMatrixD& propagator);
96   
97   const TMatrixD& GetExtrapParameters() const;
98   void            SetExtrapParameters(const TMatrixD& parameters);
99   
100   const TMatrixD& GetExtrapCovariances() const;
101   void            SetExtrapCovariances(const TMatrixD& covariances);
102   
103   const TMatrixD& GetSmoothParameters() const;
104   void            SetSmoothParameters(const TMatrixD& parameters);
105   
106   const TMatrixD& GetSmoothCovariances() const;
107   void            SetSmoothCovariances(const TMatrixD& covariances);
108   
109         /// get pointeur to associated cluster
110   AliMUONVCluster* GetClusterPtr() const {return fClusterPtr;}
111   void             SetClusterPtr(AliMUONVCluster* cluster, Bool_t owner = kFALSE);
112   
113         /// return kTRUE if the associated cluster can be removed from the track it belongs to
114   Bool_t IsRemovable() const {return fRemovable;}
115         /// set the flag telling whether the associated cluster can be removed from the track it belongs to or not
116   void   SetRemovable(Bool_t removable) {fRemovable = removable;}
117   
118         /// return kTRUE if the associated cluster alone in its chamber
119   Bool_t IsAloneInChamber() const {return fAloneInChamber;}
120         /// set the flag telling whether the associated hi alone in its chamber or not
121   void   SetAloneInChamber(Bool_t aloneInChamber) {fAloneInChamber = aloneInChamber;}
122   
123   /// return the chi2 of the track when the associated cluster was attached
124   Double_t GetTrackChi2() const {return fTrackChi2;}
125         /// set the chi2 of the track when the associated cluster was attached
126   void     SetTrackChi2(Double_t chi2) {fTrackChi2 = chi2;}
127         /// return the local chi2 of the associated cluster with respect to the track
128   Double_t GetLocalChi2() const {return fLocalChi2;}
129         /// set the local chi2 of the associated cluster with respect to the track
130   void     SetLocalChi2(Double_t chi2) {fLocalChi2 = chi2;}
131   
132         /// necessary for sorting TClonesArray of AliMUONTrackParam
133   Bool_t IsSortable () const {return kTRUE;}
134   Int_t Compare(const TObject* trackParam) const;
135
136   Bool_t CompatibleTrackParam(const AliMUONTrackParam &trackParam, Double_t sigma2Cut, Double_t &normChi2) const;
137
138   virtual void Print(Option_t* opt="") const;
139  
140   virtual void Clear(Option_t* opt="");
141
142  private:
143   
144   Double_t fZ; ///< Z coordinate (cm)
145   
146   /// Track parameters ordered as follow:      <pre>
147   /// X       = Non bending coordinate   (cm)
148   /// SlopeX  = Non bending slope        (cm ** -1)
149   /// Y       = Bending coordinate       (cm)
150   /// SlopeY  = Bending slope            (cm ** -1)
151   /// InvP_yz = Inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)  </pre>
152   TMatrixD fParameters; ///< \brief Track parameters 
153   
154   /// Covariance matrix of track parameters, ordered as follow:      <pre>
155   ///    <X,X>      <X,SlopeX>        <X,Y>      <X,SlopeY>       <X,InvP_yz>
156   /// <X,SlopeX>  <SlopeX,SlopeX>  <Y,SlopeX>  <SlopeX,SlopeY>  <SlopeX,InvP_yz>
157   ///    <X,Y>      <Y,SlopeX>        <Y,Y>      <Y,SlopeY>       <Y,InvP_yz>
158   /// <X,SlopeY>  <SlopeX,SlopeY>  <Y,SlopeY>  <SlopeY,SlopeY>  <SlopeY,InvP_yz>
159   /// <X,InvP_yz> <SlopeX,InvP_yz> <Y,InvP_yz> <SlopeY,InvP_yz> <InvP_yz,InvP_yz>  </pre>
160   mutable TMatrixD *fCovariances; ///< \brief Covariance matrix of track parameters 
161   
162   mutable TMatrixD *fPropagator;        //!< Jacobian used to extrapolate the track parameters and covariances to the actual z position
163   mutable TMatrixD *fExtrapParameters;  //!< Track parameters extrapolated to the actual z position (not filtered by Kalman)
164   mutable TMatrixD *fExtrapCovariances; //!< Covariance matrix extrapolated to the actual z position (not filtered by Kalman)
165   
166   mutable TMatrixD *fSmoothParameters;  //!< Track parameters obtained using smoother
167   mutable TMatrixD *fSmoothCovariances; //!< Covariance matrix obtained using smoother
168   
169   AliMUONVCluster *fClusterPtr; //!< Pointer to associated cluster if any
170   Bool_t           fOwnCluster; //!< Ownership of the associated cluster
171   
172   Bool_t fRemovable; //!< kTRUE if the associated cluster can be removed from the track it belongs to
173   
174   Bool_t fAloneInChamber; //!< kTRUE if the associated cluster is alone in its chamber
175
176   Double_t fTrackChi2; //!< Chi2 of the track when the associated cluster was attached
177   Double_t fLocalChi2; //!< Local chi2 of the associated cluster with respect to the track
178   
179   ClassDef(AliMUONTrackParam, 4) // Track parameters in ALICE dimuon spectrometer
180 };
181         
182 #endif