CbmMCTrack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      CbmMCTrack header file                   -----
// -----                  Created 03/08/04  by V. Friese               -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMMCTRACK_H
#define CBMMCTRACK_H 1
 
#include <Rtypes.h>          // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>      // for Double_t, Int_t, Double32_t, UInt_t
#include <TLorentzVector.h>  // for TLorentzVector
#include <TMath.h>           // for Sqrt
#include <TObject.h>         // for TObject
#include <TVector3.h>        // for TVector3
 
#include <string>  // for string
 
#include "CbmDefs.h"  // for ECbmModuleId, ECbmModuleId::kMvd, ECbmModuleId::kPsd, ECbmModuleId::kSts
 
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();
 
  Int_t AccMVD() const { return GetNPoints(ECbmModuleId::kMvd); }
  Int_t AccSTS() const { return GetNPoints(ECbmModuleId::kSts); }
  Int_t AccPSD() const { return GetNPoints(ECbmModuleId::kPsd); }
 
  UInt_t GetGeantProcessId() const { return fProcessId; }
  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 GetCharge() 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(ECbmModuleId detId) const;
 
 
  void SetMotherId(Int_t id) { fMotherId = id; }
  void SetNPoints(ECbmModuleId iDet, Int_t np);
 
 
  std::string ToString() const;
 
 
private:
  UInt_t fProcessId;
 
  Int_t fPdgCode;
 
  Int_t fMotherId;
 
  Double32_t fPx, fPy, fPz;
 
  Double32_t fE;
 
  Double32_t fStartX, fStartY, fStartZ, fStartT;
 
  Int_t fNPoints;
 
 
  ClassDef(CbmMCTrack, 3)
};
 
 
// ==========   Inline functions   ========================================
 
inline Double_t CbmMCTrack::GetEnergy() const {
  if (fE > 0.) return fE;
  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