include/Tauola/TauolaHEPEVTParticle.h

#ifndef _TauolaHEPEVTParticle_h_included_
#define _TauolaHEPEVTParticle_h_included_

/**
 * @class TauolaHEPEVTParticle
 *
 * @brief Single particle of HEPEVT event record
 *
 * This class implements the virtual methods of
 * TauolaParticle. In this way it provides an
 * interface between the generic TauolaParticle class
 * and information stored in HEPEVT event record.
 *
 * @author Tomasz Przedzinski
 * @date 24 November 2011
 */

#include <iostream>
#include <vector>

#include "f_Decay.h"
#include "TauolaParticle.h"
#include "TauolaHEPEVTEvent.h"

namespace Tauolapp
{

class TauolaHEPEVTEvent;

class TauolaHEPEVTParticle: public TauolaParticle {

 public:
  /** Default destructor */
  ~TauolaHEPEVTParticle();

  /** Default constructor */
  TauolaHEPEVTParticle(int pdgid, int status, double px, double py, double pz, double e, double m, int ms, int me, int ds, int de);

  /** Remove the decay branch from the event record and reset the particle status code to stable.
      WARNING: not implemented for HEPEVT. */
  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();

  /** Check that the 4 momentum in conserved in the decay of this particle */
  void checkMomentumConservation();

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

  /** Check if particle 'p' is daughter of this particle */
  bool isDaughterOf(TauolaHEPEVTParticle *p);

  /** Check if particle 'p' is mother of this particle */
  bool isMotherOf  (TauolaHEPEVTParticle *p);

  /** Print information on this particle into standard output */
  void print();

  /** 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 mass stored (i.e. not calculated from four vector) at generation step */
  double getMass();

  /** 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 );

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

  /** Set barcode (position in  list) of this particle */
  void setBarcode(int barcode);

  /** Set event of this particle */
  void setEvent(TauolaHEPEVTEvent *event);
  
  /** Get index of first mother */
  int getFirstMotherIndex();
  
  /** Get index of second mother */
  int getSecondMotherIndex();

  /** Get index of first daughter */
  int getDaughterRangeStart();

  /** Get index of last daughter */
  int getDaughterRangeEnd();  

private:

  /** Event from which this particle is taken */
  TauolaHEPEVTEvent *m_event;

  /** Position in the event record */
  int m_barcode;

  /** Indexes of mothers (-1 if do not have mothers) */
  int m_first_mother, m_second_mother;

  /** Range of indexes of daughters (-1 if do not have daughters) */
  int m_daughter_start, m_daughter_end;

  /** PDG ID */
  int m_pdgid;

  /** Status (stable, decayed) */
  int m_status;

  /** Momentum */
  double m_px, m_py, m_pz, m_e;

  /** Mass saved at generation step */
  double m_generated_mass;

  /** List of created particles - if they are not in the event, they
      will be deleted when no longer needed */
  vector<TauolaHEPEVTParticle*> cache;
};

} // namespace Tauolapp
#endif