ParticleMC.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ParticleMC.h

#ifndef INCLUDE_EICSMEAR_HADRONIC_PARTICLEMC_H_
#define INCLUDE_EICSMEAR_HADRONIC_PARTICLEMC_H_

#include <iostream>

#include <TLorentzVector.h>
#include <TVector3.h>

#include "eicsmear/erhic/Pid.h"
#include "eicsmear/erhic/VirtualParticle.h"

class TMCParticle;  // ROOT Monte Carlo particle class

namespace erhic {
namespace hadronic {

class ParticleMC : public erhic::VirtualParticle {
 public:
  virtual ~ParticleMC() { }

  ParticleMC();

  explicit ParticleMC(const TMCParticle&);

  ParticleMC(const TLorentzVector&, const TVector3&, int, int, int);

  virtual erhic::Pid Id() const;

  virtual TLorentzVector Get4Vector() const;

  virtual Double_t GetPx() const;

  virtual Double_t GetPy() const;

  virtual Double_t GetPz() const;

  virtual Double_t GetE() const;

  virtual Double_t GetP() const;

  virtual Double_t GetM() const;

  virtual Double_t GetPt() const;

  virtual Double_t GetTheta() const;

  virtual Double_t GetPhi() const;

  virtual Double_t GetRapidity() const;

  virtual Double_t GetEta() const;

  virtual TVector3 GetVertex() const;

  virtual UShort_t GetStatus() const;

  virtual UShort_t GetParentIndex() const;

  virtual Double_t GetXFeynman() const;

  virtual void SetStatus(UShort_t);

  virtual void SetParentIndex(UShort_t);

  virtual void SetXFeynman(double xf);

  virtual void Set4Vector(const TLorentzVector&);

  virtual void SetVertex(const TVector3&);

 protected:
  UShort_t KS;  
  UShort_t orig;  
  Int_t id;  
  Double32_t px;  
  Double32_t py;  
  Double32_t pz;  
  Double32_t E;  
  Double32_t p;  
  Double32_t m;  
  Double32_t pt;  
  Double32_t theta;  
  Double32_t phi;  
  Double32_t rapidity;  
  Double32_t eta;  
  Double32_t xFeynman;  
  Double32_t xv;  
  Double32_t yv;  
  Double32_t zv;  

  ClassDef(erhic::hadronic::ParticleMC, 1)
};

inline Pid ParticleMC::Id() const {
  return Pid(id);
}

inline TLorentzVector ParticleMC::Get4Vector() const {
  return TLorentzVector(GetPx(), GetPy(), GetPz(), GetE());
}

inline Double_t ParticleMC::GetPx() const {
  return px;
}

inline Double_t ParticleMC::GetPy() const {
  return py;
}

inline Double_t ParticleMC::GetPz() const {
  return pz;
}

inline Double_t ParticleMC::GetE() const {
  return E;
}

inline Double_t ParticleMC::GetP() const {
  return p;
}

inline Double_t ParticleMC::GetM() const {
  return m;
}

inline Double_t ParticleMC::GetPt() const {
  return pt;
}

inline Double_t ParticleMC::GetTheta() const {
  return theta;
}

inline Double_t ParticleMC::GetPhi() const {
  return phi;
}

inline Double_t ParticleMC::GetRapidity() const {
  return rapidity;
}

inline Double_t ParticleMC::GetEta() const {
  return eta;
}

inline TVector3 ParticleMC::GetVertex() const {
  return TVector3(xv, yv, zv);
}

inline UShort_t ParticleMC::GetStatus() const {
  return KS;
}

inline UShort_t ParticleMC::GetParentIndex() const {
  return orig;
}

inline Double_t ParticleMC::GetXFeynman() const {
  return xFeynman;
}

inline void ParticleMC::SetStatus(UShort_t i) {
  KS = i;
}

inline void ParticleMC::SetXFeynman(double xf) {
  xFeynman = xf;
}

inline void ParticleMC::SetVertex(const TVector3& v) {
  xv = v.x();
  yv = v.y();
  zv = v.z();
}

}  // namespace hadronic
}  // namespace erhic

#endif  // INCLUDE_EICSMEAR_HADRONIC_PARTICLEMC_H_