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

#ifndef AliMuonForwardTrack_H
#define AliMuonForwardTrack_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//====================================================================================================================================================
//
//      Description of an ALICE muon forward track, combining the information of the Muon Spectrometer and the Muon Forward Tracker
//
//      Contact author: antonio.uras@cern.ch
//
//====================================================================================================================================================

#include "AliLog.h"
#include "AliMUONTrack.h"
#include "AliMFTCluster.h"
#include "AliMUONVCluster.h"
#include "AliMUONTrackParam.h"
#include "TMatrixD.h"
#include "TClonesArray.h"
#include "TParticle.h"
#include "AliMFTConstants.h"
#include "TLorentzVector.h"

//====================================================================================================================================================

class AliMuonForwardTrack : public AliMUONTrack {

public:

  static const Int_t fgkNParentsMax =  5;   ///< maximum number of parents

  AliMuonForwardTrack();
  AliMuonForwardTrack(AliMUONTrack *MUONTrack);

  AliMuonForwardTrack(const AliMuonForwardTrack&);
  AliMuonForwardTrack &operator=(const AliMuonForwardTrack&);
  
  virtual ~AliMuonForwardTrack(); 
  virtual void Clear(const Option_t* /*opt*/);

  void SetMUONTrack(AliMUONTrack *MUONTrack);
  void SetMCTrackRef(TParticle *MCTrackRef);
  AliMUONTrack* GetMUONTrack() { return fMUONTrack; }
  TParticle* GetMCTrackRef() { return fMCTrackRef; }

  Int_t GetCharge() { return TMath::Nint(GetTrackParamAtMUONCluster(0)->GetCharge()); }

  AliMUONVCluster* GetMUONCluster(Int_t iMUONCluster);
  AliMFTCluster*   GetMFTCluster(Int_t iMFTCluster);
  
  AliMUONTrackParam* GetTrackParamAtMUONCluster(Int_t iMUONCluster);
  AliMUONTrackParam* GetTrackParamAtMFTCluster(Int_t iMFTCluster);

  void SetPlaneExists(Int_t iPlane, Bool_t value=kTRUE) { fPlaneExists[iPlane] = value; }
  Bool_t PlaneExists(Int_t iPlane) { return fPlaneExists[iPlane]; }

  Int_t GetNMUONClusters() { return fMUONTrack->GetNClusters(); }
  Int_t GetNMFTClusters()  { return fMFTClusters->GetEntries(); }

  Int_t GetMCLabelMUONTrack() { return fMUONTrack->GetMCLabel(); }

  void AddTrackParamAtMFTCluster(AliMUONTrackParam &trackParam, AliMFTCluster &mftCluster);
  
  Double_t RunKalmanFilter(AliMUONTrackParam &trackParamAtCluster);

  Double_t GetWeightedOffset(Double_t x, Double_t y, Double_t z);
  Double_t GetOffset(Double_t x, Double_t y, Double_t z);
  Double_t GetDCA(Double_t x, Double_t y, Double_t z);
  Double_t GetMomentumSpectrometer(Double_t z);
  Double_t GetOffsetX(Double_t x, Double_t z);
  Double_t GetOffsetY(Double_t y, Double_t z);
  Double_t GetThetaAbs();

  void SetParentMCLabel(Int_t iParent, Int_t MClabel) { if (0<=iParent && iParent<fgkNParentsMax) fParentMCLabel[iParent] = MClabel; }
  void SetParentPDGCode(Int_t iParent, Int_t PDGCode) { if (0<=iParent && iParent<fgkNParentsMax) fParentPDGCode[iParent] = PDGCode; }

  Int_t GetParentMCLabel(Int_t iParent) { if (0<=iParent && iParent<fgkNParentsMax) return fParentMCLabel[iParent]; else return -1; }
  Int_t GetParentPDGCode(Int_t iParent) { if (0<=iParent && iParent<fgkNParentsMax) return fParentPDGCode[iParent]; else return  0; }

  void SetNWrongClustersMC(Int_t nClusters) { fNWrongClustersMC = nClusters; }
  Int_t GetNWrongClustersMC() { return fNWrongClustersMC; }

  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; }
  
  Bool_t IsFromDirectResonance();
  Bool_t IsFromDirectCharm();
  Bool_t IsFromDirectBeauty();
  Bool_t IsFromChainResonance();
  Bool_t IsFromChainCharm();
  Bool_t IsFromChainBeauty();
  Bool_t IsFromCharm()  { return IsFromDirectCharm()  || IsFromChainCharm();  }
  Bool_t IsFromBeauty() { return IsFromDirectBeauty() || IsFromChainBeauty(); }
  Bool_t IsMuon();
  Bool_t IsFake();
  Bool_t IsPDGResonance(Int_t pdg);
  Bool_t IsPDGCharm(Int_t pdg);
  Bool_t IsPDGBeauty(Int_t pdg);
  Bool_t IsMuonFromBackground();

  Int_t GetFirstMotherID();

  void PrintHistory();

protected:

  static const Int_t fNMaxPlanes = AliMFTConstants::fNMaxPlanes;        // max number of MFT planes

  Bool_t fPlaneExists[fNMaxPlanes];

  AliMUONTrack *fMUONTrack;
  TParticle *fMCTrackRef;

  TClonesArray *fMFTClusters;

  Int_t fParentMCLabel[fgkNParentsMax];    ///< MC label of parents and grandparents
  Int_t fParentPDGCode[fgkNParentsMax];    ///< PDG code of parents and grandparents 

  Int_t fNWrongClustersMC;    // number of wrong associated MC clusters

  Int_t fTrackMCId;   // this number will identify the track within a MC simulation: run, event, MUON track

  TLorentzVector fKinem;

  TMatrixD fParamCovMatrix;

  ClassDef(AliMuonForwardTrack,2)
    
};

//====================================================================================================================================================

#endif
Commit	Line	Data
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
	28	class AliMuonForwardTrack : public AliMUONTrack {
	29
	30	public:
	31
d4643a10	32	static const Int_t fgkNParentsMax = 5; ///< maximum number of parents
d4643a10	33
820b4d9e	34	AliMuonForwardTrack();
	35	AliMuonForwardTrack(AliMUONTrack *MUONTrack);
	36
	37	AliMuonForwardTrack(const AliMuonForwardTrack&);
	38	AliMuonForwardTrack &operator=(const AliMuonForwardTrack&);
	39
274c2dce	40	virtual ~AliMuonForwardTrack();
274c2dce	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()); }
a6f7b1e5	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);
94bd88d5	70	Double_t GetDCA(Double_t x, Double_t y, Double_t z);
94bd88d5	71	Double_t GetMomentumSpectrometer(Double_t z);
820b4d9e	72	Double_t GetOffsetX(Double_t x, Double_t z);
820b4d9e	73	Double_t GetOffsetY(Double_t y, Double_t z);
94bd88d5	74	Double_t GetThetaAbs();
820b4d9e	75
d4643a10	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
ec7c7172	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);
7e3dd1af	96
	97	void SetTrackMCId(Int_t id) { fTrackMCId = id; }
	98	Int_t GetTrackMCId() { return fTrackMCId; }
d4643a10	99
45c2d1ca	100	Bool_t IsFromDirectResonance();
	101	Bool_t IsFromDirectCharm();
	102	Bool_t IsFromDirectBeauty();
	103	Bool_t IsFromChainResonance();
	104	Bool_t IsFromChainCharm();
	105	Bool_t IsFromChainBeauty();
	106	Bool_t IsFromCharm() { return IsFromDirectCharm() \|\| IsFromChainCharm(); }
	107	Bool_t IsFromBeauty() { return IsFromDirectBeauty() \|\| IsFromChainBeauty(); }
	108	Bool_t IsMuon();
	109	Bool_t IsFake();
	110	Bool_t IsPDGResonance(Int_t pdg);
64185694	111	Bool_t IsPDGCharm(Int_t pdg);
64185694	112	Bool_t IsPDGBeauty(Int_t pdg);
45c2d1ca	113	Bool_t IsMuonFromBackground();
	114
	115	Int_t GetFirstMotherID();
	116
	117	void PrintHistory();
3a775f35	118
820b4d9e	119	protected:
820b4d9e	120
d4643a10	121	static const Int_t fNMaxPlanes = AliMFTConstants::fNMaxPlanes; // max number of MFT planes
820b4d9e	122
d4643a10	123	Bool_t fPlaneExists[fNMaxPlanes];
820b4d9e	124
	125	AliMUONTrack *fMUONTrack;
	126	TParticle *fMCTrackRef;
	127
	128	TClonesArray *fMFTClusters;
	129
d4643a10	130	Int_t fParentMCLabel[fgkNParentsMax]; ///< MC label of parents and grandparents
	131	Int_t fParentPDGCode[fgkNParentsMax]; ///< PDG code of parents and grandparents
	132
	133	Int_t fNWrongClustersMC; // number of wrong associated MC clusters
	134
7e3dd1af	135	Int_t fTrackMCId; // this number will identify the track within a MC simulation: run, event, MUON track
7e3dd1af	136
ec7c7172	137	TLorentzVector fKinem;
	138
	139	TMatrixD fParamCovMatrix;
	140
ca5d8012	141	ClassDef(AliMuonForwardTrack,2)
820b4d9e	142
	143	};
	144
	145	//====================================================================================================================================================
	146
	147	#endif
	148
	149
	150