AliMuonForwardTrack class updated

[u/mrichter/AliRoot.git] / MFT / AliMuonForwardTrack.h

diff --git a/MFT/AliMuonForwardTrack.h b/MFT/AliMuonForwardTrack.h
index 3bdc15290ae5afcde6520335534081e916458c15..5ddd1d79c781e9303f2b5e20d78d74e1a3670bbc 100644 (file)
--- a/MFT/AliMuonForwardTrack.h
+++ b/MFT/AliMuonForwardTrack.h
@@ -21,6 +21,7 @@
 #include "TClonesArray.h"
 #include "TParticle.h"
 #include "AliMFTConstants.h"
+#include "TLorentzVector.h"
 
 //====================================================================================================================================================
 
@@ -78,7 +79,17 @@ public:
   void SetNWrongClustersMC(Int_t nClusters) { fNWrongClustersMC = nClusters; }
   Int_t GetNWrongClustersMC() { return fNWrongClustersMC; }
 
-  Double_t Pt() { return TMath::Sqrt(TMath::Power(GetTrackParamAtMFTCluster(0)->Px(),2)+TMath::Power(GetTrackParamAtMFTCluster(0)->Py(),2)); }
+  Double_t Pt()       { return fKinem.Pt(); }
+  Double_t Eta()      { return fKinem.Eta(); }
+  Double_t Rapidity() { return fKinem.Rapidity(); }
+  Double_t Px()       { return fKinem.Px(); }
+  Double_t Py()       { return fKinem.Py(); }
+  Double_t Pz()       { return fKinem.Pz(); }
+  Double_t P()        { return fKinem.P();  }
+
+  TMatrixD GetParamCovMatrix() { return fParamCovMatrix; }
+
+  void EvalKinem(Double_t z);
 
   void SetTrackMCId(Int_t id) { fTrackMCId = id; }
   Int_t GetTrackMCId() { return fTrackMCId; }
@@ -86,6 +97,8 @@ public:
   Bool_t IsFromResonance();
   Bool_t IsFromCharm();
   Bool_t IsFromBeauty();
+  Bool_t IsPDGCharm(Int_t pdg);
+  Bool_t IsPDGBeauty(Int_t pdg);
   Bool_t IsFromBackground();
 
 protected:
@@ -106,7 +119,11 @@ protected:
 
   Int_t fTrackMCId;   // this number will identify the track within a MC simulation: run, event, MUON track
 
-  ClassDef(AliMuonForwardTrack,1)
+  TLorentzVector fKinem;
+
+  TMatrixD fParamCovMatrix;
+
+  ClassDef(AliMuonForwardTrack,2)
     
 };