ShipMCTrack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      ShipMCTrack header file                  -----
// -------------------------------------------------------------------------
 
 
#ifndef ShipMCTrack_H
#define ShipMCTrack_H 1
 
#include "TObject.h"                    // for TObject
#include "ShipDetectorList.h"           // for DetectorId
#include "Rtypes.h"                     // for Double_t, Int_t, Double32_t, etc
#include "TLorentzVector.h"             // for TLorentzVector
#include "TMath.h"                      // for Sqrt
#include "TVector3.h"                   // for TVector3
#include "TMCProcess.h"                   // enum with process ids
#include "TString.h"
class TParticle;
 
class ShipMCTrack : public TObject
{
 
  public:
 
 
    ShipMCTrack();
 
 
    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, Double_t w);
 
    ShipMCTrack(const ShipMCTrack& track);
 
 
    ShipMCTrack(TParticle* particle);
 
 
    virtual ~ShipMCTrack();
 
 
    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);
 
 
 
  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;
 
 
    ClassDef(ShipMCTrack,8);
 
};
 
 
 
// ==========   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