CbmMCTrack.h

Go to the documentation of this file.

/* Copyright (C) 2006-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Denis Bertini [committer], Volker Friese, Florian Uhlig */
 
// -------------------------------------------------------------------------
// -----                      CbmMCTrack header file                   -----
// -----                  Created 03/08/04  by V. Friese               -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMMCTRACK_H
#define CBMMCTRACK_H 1
 
#include "CbmDefs.h"  // for ECbmModuleId, ECbmModuleId::kMvd, ECbmModuleId::kPsd, ECbmModuleId::kSts
 
#include <Rtypes.h>          // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>      // for Double32_t
#include <TLorentzVector.h>  // for TLorentzVector
#include <TObject.h>         // for TObject
#include <TVector3.h>        // for TVector3
 
#include <cstdint>
#include <string>  // for string
 
#include <cmath>
 
class TParticle;
 
class CbmMCTrack : public TObject {
 
public:
  CbmMCTrack();
 
 
  CbmMCTrack(int32_t pdgCode, int32_t motherID, double px, double py, double pz, double x, double y, double z, double t,
             int32_t nPoints);
 
  CbmMCTrack(const CbmMCTrack& track);
 
 
  CbmMCTrack(TParticle* particle);
 
 
  virtual ~CbmMCTrack();
 
  int32_t AccMVD() const { return GetNPoints(ECbmModuleId::kMvd); }
  int32_t AccSTS() const { return GetNPoints(ECbmModuleId::kSts); }
  int32_t AccPSD() const { return GetNPoints(ECbmModuleId::kPsd); }
 
  uint32_t GetGeantProcessId() const { return fProcessId; }
  int32_t GetPdgCode() const { return fPdgCode; }
  int32_t GetMotherId() const { return fMotherId; }
  double GetPx() const { return fPx; }
  double GetPy() const { return fPy; }
  double GetPz() const { return fPz; }
  double GetStartX() const { return fStartX; }
  double GetStartY() const { return fStartY; }
  double GetStartZ() const { return fStartZ; }
  double GetStartT() const { return fStartT; }
 
  double GetMass() const;
 
  double GetCharge() const;
 
  double GetEnergy() const;
  double GetPt() const { return sqrt(fPx * fPx + fPy * fPy); }
  double GetP() const { return sqrt(fPx * fPx + fPy * fPy + fPz * fPz); }
  double GetRapidity() const;
  void GetMomentum(TVector3& momentum) const;
  void Get4Momentum(TLorentzVector& momentum) const;
  void GetStartVertex(TVector3& vertex) const;
 
 
  int32_t GetNPoints(ECbmModuleId detId) const;
 
 
  void SetMotherId(int32_t id) { fMotherId = id; }
  void SetNPoints(ECbmModuleId iDet, int32_t np);
 
 
  std::string ToString() const;
 
 
private:
  uint32_t fProcessId;
 
  int32_t fPdgCode;
 
  int32_t fMotherId;
 
  Double32_t fPx, fPy, fPz;
 
  Double32_t fE;
 
  Double32_t fStartX, fStartY, fStartZ, fStartT;
 
  int32_t fNPoints;
 
 
  ClassDef(CbmMCTrack, 3)
};
 
 
// ==========   Inline functions   ========================================
 
inline double CbmMCTrack::GetEnergy() const
{
  if (fE > 0.) return fE;
  double mass = GetMass();
  return 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