[u/mrichter/AliRoot.git] / TEvtGen / Tauola / TauolaHepMCParticle.h

#ifndef _TauolaHepMCParticle_h_included_
#define _TauolaHepMCParticle_h_included_

/**
 * @class TauolaHepMCParticle
 *
 * @brief Interface to HepMC::GenParticle objects
 *
 * This class implements the virtual methods of
 * TauolaParticle. In this way it provides an
 * interface between the generic TauolaParticle class
 * and a HepMC::GenParticle object.
 *
 * @author Nadia Davidson
 * @date 17 June 2008
 *
 * This code is licensed under GNU General Public Licence.
 * For more informations, see: http://www.gnu.org/licenses/
 */

#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:
  /** General constructor */
  TauolaHepMCParticle();

  ~TauolaHepMCParticle();

  /** Constructor which keeps a pointer to the HepMC::GenParticle*/
  TauolaHepMCParticle(HepMC::GenParticle * particle); 

  /** Constructor which creates a new HepMC::GenParticle and
       sets the properties pdg_id, statu and mass. */
  TauolaHepMCParticle(int pdg_id, int status, double mass);
  
  /** Returns the HepMC::GenParticle */
  HepMC::GenParticle * getHepMC();

  /** Remove the decay branch from the event record and reset the particle status code to stable. */
  void undecay();

  /** Set the mothers of this particle via a vector of TauolaParticle*/
  void setMothers(std::vector<TauolaParticle*> mothers);

  /** Set the daughters of this particle via a vector of TauolaParticle*/
  void setDaughters(std::vector<TauolaParticle*> daughters);

  /** Returns the mothers of this particle via a vector of TauolaParticle */
  std::vector<TauolaParticle*> getMothers();

  /** Returns the daughters of this particle via a vector of TauolaParticle */
  std::vector<TauolaParticle*> getDaughters();

  /** Set the PDG ID code of this particle */
  void setPdgID(int pdg_id);

  /** Set the status of this particle */
  void setStatus(int statu);

  /** Set the mass of this particle */
  void setMass(double mass);

  /** Get the PDG ID code of this particle */
  int getPdgID();

  /** Get the status of this particle */
  int getStatus();

  /** Get the barcode of this particle */
  int getBarcode();

  /** Check that the 4 momentum in conserved at the vertices producing
      and ending this particle */
  void checkMomentumConservation();

  /** Overriding of TauolaParticle decayEndgame method.
      Converts the momentum and length units
      and sets the vector (X,T) position */
  void decayEndgame();

  /** Create a new particle of type TauolaHepMCParticle, with the given
      properties. The new particle bares no relations to this
      particle, but it provides a way of creating a instance of
      this derived class. eg. createNewParticle() is used inside
      filhep_() so that a TauolaHepMCParticle can be created without
      the method having explicit knowledge of the TauolaHepMCParticle 
      class */
  TauolaHepMCParticle * createNewParticle(int pdg_id, int status, double mass,
                                          double px, double py,
                                          double pz, double e);

  /** Print some information about this particle to standard output */
  void print();

  /** Returns the px component of the four vector*/
  double getPx();

  /** Returns the py component of the four vector */
  double getPy();

  /** Returns the pz component of the four vector */
  double getPz();

  /** Returns the energy component of the four vector */
  double getE();

  /** Set the px component of the four vector */
  void setPx( double px );

  /** Set the px component of the four vector */
  void setPy( double py );

  /** Set the pz component of the four vector */
  void setPz( double pz );

  /** Set the energy component of the four vector */
  void setE( double e );


private:

  /** Sets the position for whole decay tree starting from given particle */
  void recursiveSetPosition(HepMC::GenParticle *p,HepMC::FourVector pos);

  /** A pointer to the HepMC::GenParticle particle */
  HepMC::GenParticle * m_particle;

  /** A list of mothers */
  std::vector<TauolaParticle*> m_mothers;

  /** A list of daughters */
  std::vector<TauolaParticle*> m_daughters;

  /** List to keep track of new particles which have been
      created from this one, so we can call their destructor later */
  std::vector<TauolaParticle*> m_created_particles;

};

} // namespace Tauolapp
#endif
Commit	Line	Data
0ca57c2f	1	#ifndef _TauolaHepMCParticle_h_included_
	2	#define _TauolaHepMCParticle_h_included_
	3
	4	/**
	5	* @class TauolaHepMCParticle
	6	*
	7	* @brief Interface to HepMC::GenParticle objects
	8	*
	9	* This class implements the virtual methods of
	10	* TauolaParticle. In this way it provides an
	11	* interface between the generic TauolaParticle class
	12	* and a HepMC::GenParticle object.
	13	*
	14	* @author Nadia Davidson
	15	* @date 17 June 2008
	16	*
	17	* This code is licensed under GNU General Public Licence.
	18	* For more informations, see: http://www.gnu.org/licenses/
	19	*/
	20
	21	#include <iostream>
	22	#include <vector>
	23
	24	#include "HepMC/GenParticle.h"
	25	#include "HepMC/SimpleVector.h"
	26	#include "HepMC/GenEvent.h"
	27
	28	//#include "DecayList.h"
	29	#include "TauolaParticle.h"
	30	#include "f_Decay.h"
	31
	32	namespace Tauolapp
	33	{
	34
	35	class TauolaHepMCParticle: public TauolaParticle{
	36
	37	public:
	38	/** General constructor */
	39	TauolaHepMCParticle();
	40
	41	~TauolaHepMCParticle();
	42
	43	/** Constructor which keeps a pointer to the HepMC::GenParticle*/
	44	TauolaHepMCParticle(HepMC::GenParticle * particle);
	45
	46	/** Constructor which creates a new HepMC::GenParticle and
	47	sets the properties pdg_id, statu and mass. */
	48	TauolaHepMCParticle(int pdg_id, int status, double mass);
	49
	50	/** Returns the HepMC::GenParticle */
	51	HepMC::GenParticle * getHepMC();
	52
	53	/** Remove the decay branch from the event record and reset the particle status code to stable. */
	54	void undecay();
	55
	56	/** Set the mothers of this particle via a vector of TauolaParticle*/
	57	void setMothers(std::vector<TauolaParticle*> mothers);
	58
	59	/** Set the daughters of this particle via a vector of TauolaParticle*/
	60	void setDaughters(std::vector<TauolaParticle*> daughters);
	61
	62	/** Returns the mothers of this particle via a vector of TauolaParticle */
	63	std::vector<TauolaParticle*> getMothers();
	64
65	/** Returns the daughters of this particle via a vector of TauolaParticle */
66	std::vector<TauolaParticle*> getDaughters();
67
68	/** Set the PDG ID code of this particle */
69	void setPdgID(int pdg_id);
70
71	/** Set the status of this particle */
72	void setStatus(int statu);
73
74	/** Set the mass of this particle */
75	void setMass(double mass);
76
77	/** Get the PDG ID code of this particle */
78	int getPdgID();
79
80	/** Get the status of this particle */
81	int getStatus();
82
83	/** Get the barcode of this particle */
84	int getBarcode();
85
86	/** Check that the 4 momentum in conserved at the vertices producing
87	and ending this particle */
88	void checkMomentumConservation();
89
90	/** Overriding of TauolaParticle decayEndgame method.
91	Converts the momentum and length units
92	and sets the vector (X,T) position */
93	void decayEndgame();
94
95	/** Create a new particle of type TauolaHepMCParticle, with the given
96	properties. The new particle bares no relations to this
97	particle, but it provides a way of creating a instance of
98	this derived class. eg. createNewParticle() is used inside
99	filhep_() so that a TauolaHepMCParticle can be created without
100	the method having explicit knowledge of the TauolaHepMCParticle
101	class */
102	TauolaHepMCParticle * createNewParticle(int pdg_id, int status, double mass,
103	double px, double py,
104	double pz, double e);
105
106	/** Print some information about this particle to standard output */
107	void print();
108
109	/** Returns the px component of the four vector*/
110	double getPx();
111
112	/** Returns the py component of the four vector */
113	double getPy();
114
115	/** Returns the pz component of the four vector */
116	double getPz();
117
118	/** Returns the energy component of the four vector */
119	double getE();
120
121	/** Set the px component of the four vector */
122	void setPx( double px );
123
124	/** Set the px component of the four vector */
125	void setPy( double py );
126
127	/** Set the pz component of the four vector */
128	void setPz( double pz );
129
130	/** Set the energy component of the four vector */
131	void setE( double e );
132
133
134	private:
135
136	/** Sets the position for whole decay tree starting from given particle */
137	void recursiveSetPosition(HepMC::GenParticle *p,HepMC::FourVector pos);
138
139	/** A pointer to the HepMC::GenParticle particle */
140	HepMC::GenParticle * m_particle;
141
142	/** A list of mothers */
143	std::vector<TauolaParticle*> m_mothers;
144
145	/** A list of daughters */
146	std::vector<TauolaParticle*> m_daughters;
147
148	/** List to keep track of new particles which have been
149	created from this one, so we can call their destructor later */
150	std::vector<TauolaParticle*> m_created_particles;
151
152	};
153
154	} // namespace Tauolapp
155	#endif
156