ERMCTrack.h

/********************************************************************************
 *              Copyright (C) Joint Institute for Nuclear Research              *
 *                                                                              *
 *              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"                       *
 ********************************************************************************/

#ifndef ERMCTRACK_H_
#define ERMCTRACK_H_

#include "ERDetectorList.h"

#include "TObject.h"
#include "TMath.h"
#include "TVector3.h"
#include "TLorentzVector.h"
#include "TParticle.h"
 
enum ExpertTrackingStatus{
    Entering,
    Exiting,
    Inside,
    Stop
};

class TParticle;

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

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

  ERMCTrack(const ERMCTrack& track);

  ERMCTrack(TParticle* particle);


  virtual ~ERMCTrack();

  void Print(Int_t iTrack=0) const;

  //Int_t AccNeuRad() const {return GetNPoints(kNEURAD);}
  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 GetCharge()   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) const;
  void Get4Momentum(TLorentzVector& momentum) const;
  void GetStartVertex(TVector3& vertex) const;
  TLorentzVector GetVector();
  Double_t GetTheta();
  Double_t GetPhi();


  Int_t GetNPoints(DetectorId detId)  const;


  void SetMotherId(Int_t id) { fMotherId = id; }
  void SetNPoints(Int_t iDet, Int_t np);

private:
  Double_t CalculateMass();
  Double_t CalculateEnergy();

  Int_t fID;

  Int_t  fPdgCode;

  Int_t  fMotherId;

  Double32_t fPx, fPy, fPz;

  Double32_t fStartX, fStartY, fStartZ, fStartT;

  Int_t fNPoints;
  Double_t fMass;
  Double_t fEnergy;
  Double_t fTheta,fPhi;

  TLorentzVector fMomentum;

  ClassDef(ERMCTrack,2);

};

inline Double_t ERMCTrack::GetEnergy() const {
  return fEnergy;
}

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

inline void ERMCTrack::Get4Momentum(TLorentzVector& momentum) const {
  momentum.SetXYZT(fPx,fPy,fPz,fEnergy);
}

inline void ERMCTrack::GetStartVertex(TVector3& vertex) const {
  vertex.SetXYZ(fStartX,fStartY,fStartZ);
}
#endif /* ERMCTRACK_H_ */