PHHepMCGenEvent.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHHepMCGenEvent.h

#ifndef PHHEPMC_PHHEPMCGENEVENT_H
#define PHHEPMC_PHHEPMCGENEVENT_H

#include <phool/PHObject.h>

#include <phool/phool.h>

#include <HepMC/SimpleVector.h>

#include <CLHEP/Vector/LorentzRotation.h>

#include <iostream>  // for cout, ostream

namespace HepMC
{
  class GenEvent;
}

class PHHepMCGenEvent : public PHObject
{
 public:
  PHHepMCGenEvent();

  PHHepMCGenEvent(const PHHepMCGenEvent& event);
  PHHepMCGenEvent& operator=(const PHHepMCGenEvent& event);
  ~PHHepMCGenEvent() override;

  void identify(std::ostream& os = std::cout) const override;
  void Reset() override;
  int isValid() const override
  {
    return (getEvent() != nullptr) ? 1 : 0;
  }
  PHObject* CloneMe() const override { return new PHHepMCGenEvent(*this); }

  HepMC::GenEvent* getEvent();
  const HepMC::GenEvent* getEvent() const;

  int get_embedding_id() const { return _embedding_id; }

  void set_embedding_id(int id) { _embedding_id = id; }

  bool is_simulated() const { return _isSimulated; }

  void is_simulated(bool v) { _isSimulated = v; }

  const HepMC::FourVector& get_collision_vertex() const { return _collisionVertex; }

  void set_collision_vertex(const HepMC::FourVector& v) { _collisionVertex = v; }

  virtual const HepMC::ThreeVector& get_boost_beta_vector() const
  {
    PHOOL_VIRTUAL_WARNING;
    static HepMC::ThreeVector dummy_vec(0, 0, 0);
    return dummy_vec;
  }

  virtual void set_boost_beta_vector(const HepMC::ThreeVector&) { PHOOL_VIRTUAL_WARNING; }

  virtual const HepMC::ThreeVector& get_rotation_vector() const
  {
    PHOOL_VIRTUAL_WARNING;
    static HepMC::ThreeVector dummy_vec(0, 0, 1);
    return dummy_vec;
  }

  virtual void set_rotation_vector(const HepMC::ThreeVector&) { PHOOL_VIRTUAL_WARNING; }

  virtual double get_rotation_angle() const
  {
    PHOOL_VIRTUAL_WARNING;
    return 0;
  }

  virtual void set_rotation_angle(const double) { PHOOL_VIRTUAL_WARNING; }

  virtual CLHEP::HepLorentzRotation get_LorentzRotation_EvtGen2Lab() const { return CLHEP::HepLorentzRotation::IDENTITY; }

  virtual CLHEP::HepLorentzRotation get_LorentzRotation_Lab2EvtGen() const { return CLHEP::HepLorentzRotation::IDENTITY; }

  bool addEvent(HepMC::GenEvent* evt);
  bool addEvent(HepMC::GenEvent& evt);
  bool swapEvent(HepMC::GenEvent*& evt);
  void clearEvent();

  virtual void moveVertex(double x, double y, double z, double t = 0);

  // the number of entries in the array of particles
  virtual int size(void) const;
  virtual int vertexSize(void) const;

  void print(std::ostream& os = std::cout) const;

  void PrintEvent();

 protected:
  int _embedding_id;

  bool _isSimulated;

  HepMC::FourVector _collisionVertex;

  HepMC::GenEvent* _theEvt;

  ClassDefOverride(PHHepMCGenEvent, 5)
};

#endif  // PHHEPMC_PHHEPMCEVENT_H