PhotosHepMCParticle.h

#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