MFT/AliMuonForwardTrack.h

   1 #ifndef AliMuonForwardTrack_H
   2 #define AliMuonForwardTrack_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 //====================================================================================================================================================
   8 //
   9 //      Description of an ALICE muon forward track, combining the information of the Muon Spectrometer and the Muon Forward Tracker
  10 //
  11 //      Contact author: antonio.uras@cern.ch
  12 //
  13 //====================================================================================================================================================
  14
  15 #include "AliLog.h"
  16 #include "AliMUONTrack.h"
  17 #include "AliMFTCluster.h"
  18 #include "AliMUONVCluster.h"
  19 #include "AliMUONTrackParam.h"
  20 #include "TMatrixD.h"
  21 #include "TClonesArray.h"
  22 #include "TParticle.h"
  23 #include "AliMFTConstants.h"
  24 #include "TLorentzVector.h"
  25
  26 //====================================================================================================================================================
  27
  28 class AliMuonForwardTrack : public AliMUONTrack {
  29
  30 public:
  31
  32   static const Int_t fgkNParentsMax =  5;   ///< maximum number of parents
  33
  34   AliMuonForwardTrack();
  35   AliMuonForwardTrack(AliMUONTrack *MUONTrack);
  36
  37   AliMuonForwardTrack(const AliMuonForwardTrack&);
  38   AliMuonForwardTrack &operator=(const AliMuonForwardTrack&);
  39
  40   virtual ~AliMuonForwardTrack();
  41   virtual void Clear(const Option_t* /*opt*/);
  42
  43   void SetMUONTrack(AliMUONTrack *MUONTrack);
  44   void SetMCTrackRef(TParticle *MCTrackRef);
  45   AliMUONTrack* GetMUONTrack() { return fMUONTrack; }
  46   TParticle* GetMCTrackRef() { return fMCTrackRef; }
  47
  48   Int_t GetCharge() { return TMath::Nint(GetTrackParamAtMUONCluster(0)->GetCharge()); }
  49
  50   AliMUONVCluster* GetMUONCluster(Int_t iMUONCluster);
  51   AliMFTCluster*   GetMFTCluster(Int_t iMFTCluster);
  52
  53   AliMUONTrackParam* GetTrackParamAtMUONCluster(Int_t iMUONCluster);
  54   AliMUONTrackParam* GetTrackParamAtMFTCluster(Int_t iMFTCluster);
  55
  56   void SetPlaneExists(Int_t iPlane, Bool_t value=kTRUE) { fPlaneExists[iPlane] = value; }
  57   Bool_t PlaneExists(Int_t iPlane) { return fPlaneExists[iPlane]; }
  58
  59   Int_t GetNMUONClusters() { return fMUONTrack->GetNClusters(); }
  60   Int_t GetNMFTClusters()  { return fMFTClusters->GetEntries(); }
  61
  62   Int_t GetMCLabelMUONTrack() { return fMUONTrack->GetMCLabel(); }
  63
  64   void AddTrackParamAtMFTCluster(AliMUONTrackParam &trackParam, AliMFTCluster &mftCluster);
  65
  66   Double_t RunKalmanFilter(AliMUONTrackParam &trackParamAtCluster);
  67
  68   Double_t GetWeightedOffset(Double_t x, Double_t y, Double_t z);
  69   Double_t GetOffset(Double_t x, Double_t y, Double_t z);
  70   Double_t GetDCA(Double_t x, Double_t y, Double_t z);
  71   Double_t GetMomentumSpectrometer(Double_t z);
  72   Double_t GetOffsetX(Double_t x, Double_t z);
  73   Double_t GetOffsetY(Double_t y, Double_t z);
  74   Double_t GetThetaAbs();
  75
  76   void SetParentMCLabel(Int_t iParent, Int_t MClabel) { if (0<=iParent && iParent<fgkNParentsMax) fParentMCLabel[iParent] = MClabel; }
  77   void SetParentPDGCode(Int_t iParent, Int_t PDGCode) { if (0<=iParent && iParent<fgkNParentsMax) fParentPDGCode[iParent] = PDGCode; }
  78
  79   Int_t GetParentMCLabel(Int_t iParent) { if (0<=iParent && iParent<fgkNParentsMax) return fParentMCLabel[iParent]; else return -1; }
  80   Int_t GetParentPDGCode(Int_t iParent) { if (0<=iParent && iParent<fgkNParentsMax) return fParentPDGCode[iParent]; else return  0; }
  81
  82   void SetNWrongClustersMC(Int_t nClusters) { fNWrongClustersMC = nClusters; }
  83   Int_t GetNWrongClustersMC() { return fNWrongClustersMC; }
  84
  85   Double_t Pt()       { return fKinem.Pt(); }
  86   Double_t Eta()      { return fKinem.Eta(); }
  87   Double_t Rapidity() { return fKinem.Rapidity(); }
  88   Double_t Px()       { return fKinem.Px(); }
  89   Double_t Py()       { return fKinem.Py(); }
  90   Double_t Pz()       { return fKinem.Pz(); }
  91   Double_t P()        { return fKinem.P();  }
  92
  93   TMatrixD GetParamCovMatrix() { return fParamCovMatrix; }
  94
  95   void EvalKinem(Double_t z);
  96
  97   void SetTrackMCId(Int_t id) { fTrackMCId = id; }
  98   Int_t GetTrackMCId() { return fTrackMCId; }
  99
 100   Bool_t IsFromResonance();
 101   Bool_t IsDirectCharm();
 102   Bool_t IsDirectBeauty();
 103   Bool_t IsChainBeauty();
 104   Bool_t IsFromCharm()  { return IsDirectCharm(); }
 105   Bool_t IsFromBeauty() { return IsDirectBeauty() || IsChainBeauty(); }
 106   Bool_t IsPDGCharm(Int_t pdg);
 107   Bool_t IsPDGBeauty(Int_t pdg);
 108   Bool_t IsFromBackground();
 109
 110 protected:
 111
 112   static const Int_t fNMaxPlanes = AliMFTConstants::fNMaxPlanes;        // max number of MFT planes
 113
 114   Bool_t fPlaneExists[fNMaxPlanes];
 115
 116   AliMUONTrack *fMUONTrack;
 117   TParticle *fMCTrackRef;
 118
 119   TClonesArray *fMFTClusters;
 120
 121   Int_t fParentMCLabel[fgkNParentsMax];    ///< MC label of parents and grandparents
 122   Int_t fParentPDGCode[fgkNParentsMax];    ///< PDG code of parents and grandparents
 123
 124   Int_t fNWrongClustersMC;    // number of wrong associated MC clusters
 125
 126   Int_t fTrackMCId;   // this number will identify the track within a MC simulation: run, event, MUON track
 127
 128   TLorentzVector fKinem;
 129
 130   TMatrixD fParamCovMatrix;
 131
 132   ClassDef(AliMuonForwardTrack,2)
 133
 134 };
 135
 136 //====================================================================================================================================================
 137
 138 #endif
 139
 140
 141