Analysis code updated
[u/mrichter/AliRoot.git] / MFT / AliMuonForwardTrack.h
CommitLineData
820b4d9e 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"
d4643a10 23#include "AliMFTConstants.h"
ec7c7172 24#include "TLorentzVector.h"
820b4d9e 25
26//====================================================================================================================================================
27
28class AliMuonForwardTrack : public AliMUONTrack {
29
30public:
31
d4643a10 32 static const Int_t fgkNParentsMax = 5; ///< maximum number of parents
33
820b4d9e 34 AliMuonForwardTrack();
35 AliMuonForwardTrack(AliMUONTrack *MUONTrack);
36
37 AliMuonForwardTrack(const AliMuonForwardTrack&);
38 AliMuonForwardTrack &operator=(const AliMuonForwardTrack&);
39
274c2dce 40 virtual ~AliMuonForwardTrack();
41 virtual void Clear(const Option_t* /*opt*/);
820b4d9e 42
43 void SetMUONTrack(AliMUONTrack *MUONTrack);
44 void SetMCTrackRef(TParticle *MCTrackRef);
45 AliMUONTrack* GetMUONTrack() { return fMUONTrack; }
46 TParticle* GetMCTrackRef() { return fMCTrackRef; }
47
a6f7b1e5 48 Int_t GetCharge() { return TMath::Nint(GetTrackParamAtMUONCluster(0)->GetCharge()); }
49
820b4d9e 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(); }
d4643a10 60 Int_t GetNMFTClusters() { return fMFTClusters->GetEntries(); }
820b4d9e 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 GetOffsetX(Double_t x, Double_t z);
71 Double_t GetOffsetY(Double_t y, Double_t z);
72
d4643a10 73 void SetParentMCLabel(Int_t iParent, Int_t MClabel) { if (0<=iParent && iParent<fgkNParentsMax) fParentMCLabel[iParent] = MClabel; }
74 void SetParentPDGCode(Int_t iParent, Int_t PDGCode) { if (0<=iParent && iParent<fgkNParentsMax) fParentPDGCode[iParent] = PDGCode; }
75
76 Int_t GetParentMCLabel(Int_t iParent) { if (0<=iParent && iParent<fgkNParentsMax) return fParentMCLabel[iParent]; else return -1; }
77 Int_t GetParentPDGCode(Int_t iParent) { if (0<=iParent && iParent<fgkNParentsMax) return fParentPDGCode[iParent]; else return 0; }
78
79 void SetNWrongClustersMC(Int_t nClusters) { fNWrongClustersMC = nClusters; }
80 Int_t GetNWrongClustersMC() { return fNWrongClustersMC; }
81
ec7c7172 82 Double_t Pt() { return fKinem.Pt(); }
83 Double_t Eta() { return fKinem.Eta(); }
84 Double_t Rapidity() { return fKinem.Rapidity(); }
85 Double_t Px() { return fKinem.Px(); }
86 Double_t Py() { return fKinem.Py(); }
87 Double_t Pz() { return fKinem.Pz(); }
88 Double_t P() { return fKinem.P(); }
89
90 TMatrixD GetParamCovMatrix() { return fParamCovMatrix; }
91
92 void EvalKinem(Double_t z);
7e3dd1af 93
94 void SetTrackMCId(Int_t id) { fTrackMCId = id; }
95 Int_t GetTrackMCId() { return fTrackMCId; }
d4643a10 96
3a775f35 97 Bool_t IsFromResonance();
98 Bool_t IsFromCharm();
99 Bool_t IsFromBeauty();
100 Bool_t IsFromBackground();
101
820b4d9e 102protected:
103
d4643a10 104 static const Int_t fNMaxPlanes = AliMFTConstants::fNMaxPlanes; // max number of MFT planes
820b4d9e 105
d4643a10 106 Bool_t fPlaneExists[fNMaxPlanes];
820b4d9e 107
108 AliMUONTrack *fMUONTrack;
109 TParticle *fMCTrackRef;
110
111 TClonesArray *fMFTClusters;
112
d4643a10 113 Int_t fParentMCLabel[fgkNParentsMax]; ///< MC label of parents and grandparents
114 Int_t fParentPDGCode[fgkNParentsMax]; ///< PDG code of parents and grandparents
115
116 Int_t fNWrongClustersMC; // number of wrong associated MC clusters
117
7e3dd1af 118 Int_t fTrackMCId; // this number will identify the track within a MC simulation: run, event, MUON track
119
ec7c7172 120 TLorentzVector fKinem;
121
122 TMatrixD fParamCovMatrix;
123
ca5d8012 124 ClassDef(AliMuonForwardTrack,2)
820b4d9e 125
126};
127
128//====================================================================================================================================================
129
130#endif
131
132
133