ParticleMCS.h

Go to the documentation of this file.

#ifndef INCLUDE_EICSMEAR_SMEAR_PARTICLEMCS_H_
#define INCLUDE_EICSMEAR_SMEAR_PARTICLEMCS_H_

#include <cmath>

#include <TLorentzVector.h>

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

namespace Smear {

class Event;

class ParticleMCS : public erhic::VirtualParticle {
 public:
  virtual ~ParticleMCS();

  ParticleMCS();

  ParticleMCS(const TLorentzVector&, int pdg, int status);

  virtual Double_t GetPx() const;

  virtual Double_t GetPy() const;

  virtual Double_t GetPz() const;

  virtual Double_t GetE() const;

  virtual TLorentzVector Get4Vector() const;

  virtual TLorentzVector PxPyPzE() const { return Get4Vector(); }

  virtual Double_t GetM() const;

  virtual Double_t GetPt() const;

  virtual TVector3 GetVertex() const;

  virtual Double_t GetP() const;

  virtual Double_t GetTheta() const;

  virtual Double_t GetPhi() const;

  virtual Double_t GetRapidity() const;

  virtual Double_t GetEta() const;

  virtual UShort_t GetStatus() const;

  virtual ::erhic::Pid Id() const;

  virtual void SetE(Double_t);

  virtual void SetP(Double_t);

  virtual void SetPt(Double_t);

  virtual void SetPz(Double_t);

  virtual void SetPhi(Double_t);

  virtual void SetTheta(Double_t);

  void Set4Vector(const TLorentzVector&) { }

  virtual void SetId(Int_t);

  virtual void SetStatus(Int_t);

  virtual UShort_t GetParentIndex() const { return 0; }

  virtual void Print(Option_t* = "") const;

  virtual void SetVertex(const TVector3&) { }

// protected:
  UShort_t   status;      
  Int_t      id;          
  Double32_t px;          
  Double32_t py;          
  Double32_t pz;          
  Double32_t E;           
  Double32_t pt;          
  Double32_t p;           
  Double32_t theta;       
  Double32_t phi;         

  ClassDef(Smear::ParticleMCS, 1)
};

inline Double_t ParticleMCS::GetPx() const {
  return p * sin(theta) * cos(phi);
}

inline Double_t ParticleMCS::GetPy() const {
  return p * sin(theta) * sin(phi);
}

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

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

inline Double_t ParticleMCS::GetM() const {
  return sqrt(pow(E, 2.) - pow(p, 2.));
}

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

inline TVector3 ParticleMCS::GetVertex() const {
  return TVector3();
}

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

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

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

inline UShort_t ParticleMCS::GetStatus() const {
  return status;
}

inline void ParticleMCS::SetE(Double_t e) {
  E = e;
}

inline void ParticleMCS::SetP(Double_t momentum) {
  p = momentum;
}

inline void ParticleMCS::SetPt(Double_t momentum) {
  pt = momentum;
}

inline void ParticleMCS::SetPz(Double_t momentum) {
  pz = momentum;
}

inline void ParticleMCS::SetPhi(Double_t value) {
  phi = value;
}

inline void ParticleMCS::SetTheta(Double_t value) {
  theta = value;
}

inline void ParticleMCS::SetId(Int_t i) {
  id = i;
}

inline void ParticleMCS::SetStatus(Int_t i) {
  status = i;
}

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

}  // namespace Smear

#endif  // INCLUDE_EICSMEAR_SMEAR_PARTICLEMCS_H_