CbmMCTrack.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmMCTrack.h

// -------------------------------------------------------------------------
// -----                      CbmMCTrack header file                   -----
// -----                  Created 03/08/04  by V. Friese               -----
// -------------------------------------------------------------------------


#ifndef CBMMCTRACK_H
#define CBMMCTRACK_H 1


#include "CbmDetectorList.h"

#include "TObject.h"
#include "TMath.h"
#include "TVector3.h"
#include "TLorentzVector.h"

class TParticle;

class CbmMCTrack : public TObject
{

 public:


  CbmMCTrack();


  CbmMCTrack(Int_t pdgCode, Int_t motherID, Double_t px, Double_t py,
       Double_t pz, Double_t x, Double_t y, Double_t z, 
       Double_t t, Int_t nPoints);

  CbmMCTrack(const CbmMCTrack& track);


  CbmMCTrack(TParticle* particle);


  virtual ~CbmMCTrack();


  void Print(Int_t iTrack=0) const;


  Int_t    GetPdgCode()  const { return fPdgCode; }
  Int_t    GetMotherId() const { return fMotherId; }
  Double_t GetPx()       const { return fPx; }
  Double_t GetPy()       const { return fPy; }
  Double_t GetPz()       const { return fPz; }
  Double_t GetStartX()   const { return fStartX; }
  Double_t GetStartY()   const { return fStartY; }
  Double_t GetStartZ()   const { return fStartZ; }
  Double_t GetStartT()   const { return fStartT; }
  Double_t GetMass()     const;
  Double_t GetEnergy()   const;
  Double_t GetPt()       const { return TMath::Sqrt(fPx*fPx+fPy*fPy); }
  Double_t GetP() const { return TMath::Sqrt(fPx*fPx+fPy*fPy+fPz*fPz); }
  Double_t GetRapidity() const;
  void GetMomentum(TVector3& momentum) const; 
  void Get4Momentum(TLorentzVector& momentum) const;
  void GetStartVertex(TVector3& vertex) const;


  Int_t GetNPoints(DetectorId detId)  const;


  void SetMotherId(Int_t id) { fMotherId = id; }
  void SetNPoints(Int_t iDet, Int_t np);



private:

  Int_t  fPdgCode;

  Int_t  fMotherId;

  Double32_t fPx, fPy, fPz;

  Double32_t fStartX, fStartY, fStartZ, fStartT;

  Int_t fNPoints;


  ClassDef(CbmMCTrack,2);

};



// ==========   Inline functions   ========================================

inline Double_t CbmMCTrack::GetEnergy() const {
  Double_t mass = GetMass();
  return TMath::Sqrt(mass*mass + fPx*fPx + fPy*fPy + fPz*fPz );
}


inline void CbmMCTrack::GetMomentum(TVector3& momentum) const { 
  momentum.SetXYZ(fPx,fPy,fPz); 
}


inline void CbmMCTrack::Get4Momentum(TLorentzVector& momentum) const {
  momentum.SetXYZT(fPx,fPy,fPz,GetEnergy()); 
}


inline void CbmMCTrack::GetStartVertex(TVector3& vertex) const { 
  vertex.SetXYZ(fStartX,fStartY,fStartZ); 
}





#endif