AtMCTrack.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
 
// -------------------------------------------------------------------------
// -----                  AtMCTrack header file                  -----
// -----                  M. Al-Turany   June 2014                     -----
// -------------------------------------------------------------------------
 
#ifndef AtMCTrack_H
#define AtMCTrack_H 1
 
#include <Rtypes.h>         // for Double_t, Int_t, Double32_t, etc
#include <TLorentzVector.h> // for TLorentzVector
#include <TMath.h>          // for Sqrt
#include <TObject.h>        // for TObject
#include <TVector3.h>       // for TVector3
 
class TParticle;
class TBuffer;
class TClass;
class TMemberInspector;
 
class AtMCTrack : public TObject {
 
public:
   AtMCTrack();
 
   AtMCTrack(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);
 
   AtMCTrack(const AtMCTrack &track);
 
   AtMCTrack(TParticle *particle);
 
   virtual ~AtMCTrack();
 
   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);
   void Get4Momentum(TLorentzVector &momentum);
   void GetStartVertex(TVector3 &vertex);
 
   void SetMotherId(Int_t id) { fMotherId = id; }
 
private:
   Int_t fPdgCode;
 
   Int_t fMotherId;
 
   Double32_t fPx, fPy, fPz;
 
   Double32_t fStartX, fStartY, fStartZ, fStartT;
 
   Int_t fNPoints;
 
   ClassDef(AtMCTrack, 1);
};
 
// ==========   Inline functions   ========================================
 
inline Double_t AtMCTrack::GetEnergy() const
{
   Double_t mass = GetMass();
   return TMath::Sqrt(mass * mass + fPx * fPx + fPy * fPy + fPz * fPz);
}
 
inline void AtMCTrack::GetMomentum(TVector3 &momentum)
{
   momentum.SetXYZ(fPx, fPy, fPz);
}
 
inline void AtMCTrack::Get4Momentum(TLorentzVector &momentum)
{
   momentum.SetXYZT(fPx, fPy, fPz, GetEnergy());
}
 
inline void AtMCTrack::GetStartVertex(TVector3 &vertex)
{
   vertex.SetXYZ(fStartX, fStartY, fStartZ);
}
 
#endif