G4SimEvent.h

Go to the documentation of this file.

#ifndef SimG4Core_G4SimEvent_H
#define SimG4Core_G4SimEvent_H

#include "SimG4Core/Notification/interface/G4SimTrack.h"
#include "SimG4Core/Notification/interface/G4SimVertex.h"

#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"

#include "HepMC/GenEvent.h"

#include <vector>

class G4SimEvent {
public:
  G4SimEvent();
  virtual ~G4SimEvent();
  void load(edm::SimTrackContainer& c) const;
  void load(edm::SimVertexContainer& c) const;
  unsigned int nTracks() const { return g4tracks_.size(); }
  unsigned int nVertices() const { return g4vertices_.size(); }
  unsigned int nGenParts() const { return hepMCEvent_->particles_size(); }
  void hepEvent(const HepMC::GenEvent* r) { hepMCEvent_ = r; }
  const HepMC::GenEvent* hepEvent() const { return hepMCEvent_; }
  void weight(float w) { weight_ = w; }
  float weight() const { return weight_; }
  void collisionPoint(const math::XYZTLorentzVectorD& v) { collisionPoint_ = v; }
  const math::XYZTLorentzVectorD& collisionPoint() const { return collisionPoint_; }
  void nparam(int n) { nparam_ = n; }
  const int nparam() const { return nparam_; }
  void param(const std::vector<float>& p) { param_ = p; }
  const std::vector<float>& param() const { return param_; }
  void add(G4SimTrack* t) { g4tracks_.push_back(t); }
  void add(G4SimVertex* v) { g4vertices_.push_back(v); }
  void clear();

private:
  const HepMC::GenEvent* hepMCEvent_;
  float weight_;
  math::XYZTLorentzVectorD collisionPoint_;
  int nparam_;
  std::vector<float> param_;
  std::vector<G4SimTrack*> g4tracks_;
  std::vector<G4SimVertex*> g4vertices_;
};

#endif