PhotosHepMCParticle.h

Go to the documentation of this file.

#ifndef _PhotosHepMCParticle_h_included_
#define _PhotosHepMCParticle_h_included_

#include <vector>

#include "HepMC/GenParticle.h"
#include "PhotosParticle.h"

using namespace std;

namespace Photospp {

class PhotosHepMCParticle : public PhotosParticle {

 public:
  PhotosHepMCParticle();

  PhotosHepMCParticle(HepMC::GenParticle *particle);

  PhotosHepMCParticle(int pdg_id, int status, double mass);

  ~PhotosHepMCParticle();

  HepMC::GenParticle *getHepMC();

  void setMothers(std::vector<PhotosParticle *> mothers);

  void setDaughters(std::vector<PhotosParticle *> daughters);

  void addDaughter(PhotosParticle *daughter);

  std::vector<PhotosParticle *> getMothers();

  std::vector<PhotosParticle *> getDaughters();

  std::vector<PhotosParticle *> getAllDecayProducts();

  void setPdgID(int pdg_id);

  void setStatus(int statu);

  void setMass(double mass);

  int getPdgID();

  int getStatus();

  int getBarcode();

  bool checkMomentumConservation();

  PhotosHepMCParticle *createNewParticle(int pdg_id, int status, double mass, double px, double py, double pz, double e);

  void createHistoryEntry();

  void createSelfDecayVertex(PhotosParticle *out);

  void print();

  double getPx();

  double getPy();

  double getPz();

  double getE();

  double getMass();

  void setPx(double px);

  void setPy(double py);

  void setPz(double pz);

  void setE(double e);

 private:
  void clear(std::vector<PhotosParticle *> v);

  HepMC::GenParticle *m_particle;

  std::vector<PhotosParticle *> m_mothers;

  std::vector<PhotosParticle *> m_daughters;

  std::vector<PhotosParticle *> m_decay_products;

  std::vector<PhotosParticle *> m_created_particles;
};

} // namespace Photospp
#endif