FairMCTrack.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 (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
// -------------------------------------------------------------------------
// -----                      FairMCTrack header file                   -----
// -----                  Created 03/08/04  by V. Friese               -----
// -------------------------------------------------------------------------

#ifndef FAIRMCTRACK_H
#define FAIRMCTRACK_H 1

#include "FairDetectorList.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 <TObject.h>          // for TObject
#include <TVector3.h>         // for TVector3

class TParticle;

class FairMCTrack : public TObject
{
  public:
    FairMCTrack();

    FairMCTrack(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);

    FairMCTrack(const FairMCTrack& track);

    FairMCTrack(TParticle* particle);

    virtual ~FairMCTrack();

#if defined(__clang__)
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Woverloaded-virtual"
#endif

    virtual void Print(Int_t iTrack) const;
#if defined(__clang__)
#pragma clang diagnostic pop
#endif

    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);

    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;

    Double32_t fStartX, fStartY, fStartZ, fStartT;

    Int_t fNPoints;

    ClassDef(FairMCTrack, 2);
};

// ==========   Inline functions   ========================================

inline Double_t FairMCTrack::GetEnergy() const
{
    Double_t mass = GetMass();
    return TMath::Sqrt(mass * mass + fPx * fPx + fPy * fPy + fPz * fPz);
}

inline void FairMCTrack::GetMomentum(TVector3& momentum) { momentum.SetXYZ(fPx, fPy, fPz); }

inline void FairMCTrack::Get4Momentum(TLorentzVector& momentum) { momentum.SetXYZT(fPx, fPy, fPz, GetEnergy()); }

inline void FairMCTrack::GetStartVertex(TVector3& vertex) { vertex.SetXYZ(fStartX, fStartY, fStartZ); }

#endif