MyProjMCTrack.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"                       *
 ********************************************************************************/

// -------------------------------------------------------------------------
// -----                  MyProjMCTrack header file                  -----
// -----                  M. Al-Turany   June 2014                     -----
// -------------------------------------------------------------------------

#ifndef MyProjMCTrack_H
#define MyProjMCTrack_H 1

#include "MyProjDetectorList.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 MyProjMCTrack : public TObject
{

  public:
    MyProjMCTrack();

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

    MyProjMCTrack(const MyProjMCTrack& track);

    MyProjMCTrack(TParticle* particle);

    virtual ~MyProjMCTrack();

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

    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(MyProjMCTrack, 1);
};

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

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

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

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

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

#endif