CbmMCTrack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      CbmMCTrack header file                   -----
// -----                  should be replaced by BmnMCTrack             -----
// -------------------------------------------------------------------------
 
#ifndef CBMMCTRACK_H
#define CBMMCTRACK_H 1
 
#include "BmnDetectorList.h"
#include "TLorentzVector.h"
#include "TMath.h"
#include "TObject.h"
#include "TParticle.h"
#include "TVector3.h"
 
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,
               Long64_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;
 
    Long64_t GetNPoints(DetectorId detId) const;
    Double_t GetPolar(Int_t indx) { return fPolar[indx]; }
 
    void SetMotherId(Int_t id) { fMotherId = id; }
    void SetNPoints(Int_t iDet, Long64_t np);
    void SetPolar(Int_t indx, Double_t polar) { fPolar[indx] = polar; }   // AZ-310822
 
  private:
    Int_t fPdgCode;
 
    Int_t fMotherId;
 
    Double32_t fPx, fPy, fPz;
 
    Double32_t fStartX, fStartY, fStartZ, fStartT;
 
    Long64_t fNPoints;
    Double32_t fPolar[3];   // AZ-310822 - polarization
 
    // AZ-310822 ClassDef(CbmMCTrack,2);
    ClassDef(CbmMCTrack, 3);   // AZ-310822
};
 
// ==========   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