ShipMCTrack.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// SPDX-FileCopyrightText: Copyright CERN for the benefit of the SHiP
// Collaboration
 
// -------------------------------------------------------------------------
// -----                      ShipMCTrack header file                  -----
// -------------------------------------------------------------------------
 
#ifndef SHIPDATA_SHIPMCTRACK_H_
#define SHIPDATA_SHIPMCTRACK_H_
 
#include "Rtypes.h"            // for Double_t, Int_t, Double32_t, etc
#include "ShipDetectorList.h"  // for DetectorId
#include "TLorentzVector.h"    // for TLorentzVector
#include "TMCProcess.h"        // enum with process ids
#include "TMath.h"             // for Sqrt
#include "TObject.h"           // for TObject
#include "TString.h"
#include "TVector3.h"  // for TVector3
class TParticle;
 
class ShipMCTrack : public TObject {
 public:
  ShipMCTrack();
 
  explicit ShipMCTrack(Int_t pdgCode, Int_t motherID, Double_t px, Double_t py,
                       Double_t pz, Double_t E, Double_t x, Double_t y,
                       Double_t z, Double_t t, Int_t nPoints, Int_t eventID,
                       Int_t trackID, Double_t w);
 
  ShipMCTrack(const ShipMCTrack& track);
 
  explicit ShipMCTrack(TParticle* particle);
 
  virtual ~ShipMCTrack();
 
  using TObject::Print;
  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; }
  void SetProcID(Int_t i) { fProcID = i; }
  Int_t GetProcID() const { return fProcID; }
  TString GetProcName() const { return TMCProcessName[fProcID]; }
  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 MultiplyWeight(Double_t w) { fW = fW * w; }
  void SetWeight(Double_t w) { fW = w; }
  Double_t GetWeight() const;
  void GetMomentum(TVector3& momentum);
  void Get4Momentum(TLorentzVector& momentum);
  void GetStartVertex(TVector3& vertex);
 
  Int_t GetNPoints(DetectorId detId) const;
 
  void SetMotherId(Int_t id) { fMotherId = id; }
  void SetNPoints(Int_t iDet, Int_t np);
  void SetEventID(const Int_t& eventID);
  void SetTrackID(const Int_t& trackID);
 
 private:
  Int_t fPdgCode;
 
  Int_t fMotherId;
 
  Double32_t fPx, fPy, fPz, fM;
 
  Double32_t fStartX, fStartY, fStartZ, fStartT;
 
  Double32_t fW;
 
  Int_t fProcID;
 
  Int_t fNPoints;
 
  Int_t fEventID;
 
  Int_t fTrackID;
 
  ClassDef(ShipMCTrack, 9);
};
 
// ==========   Inline functions   ========================================
 
inline void ShipMCTrack::GetMomentum(TVector3& momentum) {
  momentum.SetXYZ(fPx, fPy, fPz);
}
 
inline void ShipMCTrack::Get4Momentum(TLorentzVector& momentum) {
  momentum.SetXYZT(fPx, fPy, fPz, GetEnergy());
}
 
inline void ShipMCTrack::GetStartVertex(TVector3& vertex) {
  vertex.SetXYZ(fStartX, fStartY, fStartZ);
}
 
#endif  // SHIPDATA_SHIPMCTRACK_H_