TauolaHepMCParticle.h

#ifndef _TauolaHepMCParticle_h_included_
#define _TauolaHepMCParticle_h_included_

#include <iostream>
#include <vector>

#include "HepMC/GenParticle.h"
#include "HepMC/SimpleVector.h"
#include "HepMC/GenEvent.h"

//#include "DecayList.h"
#include "TauolaParticle.h"
#include "f_Decay.h"

namespace Tauolapp
{

class TauolaHepMCParticle: public TauolaParticle{

 public:
  TauolaHepMCParticle();

  ~TauolaHepMCParticle();

  TauolaHepMCParticle(HepMC::GenParticle * particle); 

  TauolaHepMCParticle(int pdg_id, int status, double mass);
  
  HepMC::GenParticle * getHepMC();

  void undecay();

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

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

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

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

  void setPdgID(int pdg_id);

  void setStatus(int statu);

  void setMass(double mass);

  int getPdgID();

  int getStatus();

  int getBarcode();

  void checkMomentumConservation();

  void decayEndgame();

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

  void print();

  double getPx();

  double getPy();

  double getPz();

  double getE();

  void setPx( double px );

  void setPy( double py );

  void setPz( double pz );

  void setE( double e );


private:

  void recursiveSetPosition(HepMC::GenParticle *p,HepMC::FourVector pos);

  HepMC::GenParticle * m_particle;

  std::vector<TauolaParticle*> m_mothers;

  std::vector<TauolaParticle*> m_daughters;

  std::vector<TauolaParticle*> m_created_particles;

};

} // namespace Tauolapp
#endif