Generator.h

Go to the documentation of this file.

#ifndef SimG4Core_Generator_H
#define SimG4Core_Generator_H

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Math/interface/LorentzVector.h"

#include "SimG4Core/Generators/interface/HepMCParticle.h"
#include "SimG4Core/Notification/interface/GenParticleInfo.h"

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

#include <vector>
    
class G4Event;
class G4PrimaryParticle;
class LumiMonitorFilter;

class Generator
{
public:
  Generator(const edm::ParameterSet & p);
  virtual ~Generator();

  void setGenEvent( const HepMC::GenEvent* inpevt ) 
    { evt_ = (HepMC::GenEvent*)inpevt; return ; }
  void HepMC2G4(const HepMC::GenEvent * g,G4Event * e);
  void nonBeamEvent2G4(const HepMC::GenEvent * g,G4Event * e);
  virtual const HepMC::GenEvent*  genEvent() const { return evt_; }
  virtual const math::XYZTLorentzVector* genVertex() const { return vtx_; }
  virtual const double eventWeight() const { return weight_; }

private:

  bool particlePassesPrimaryCuts(const G4ThreeVector& p) const;
  void particleAssignDaughters(G4PrimaryParticle * p, HepMC::GenParticle * hp, 
                   double length);
  void setGenId(G4PrimaryParticle* p, int id) const 
  { p->SetUserInformation(new GenParticleInfo(id));}

private:
  bool   fPCuts;
  bool   fPtransCut;
  bool   fEtaCuts;
  bool   fPhiCuts;
  double theMinPhiCut;
  double theMaxPhiCut;
  double theMinEtaCut;
  double theMaxEtaCut;
  double theMinPCut;
  double theMinPtCut2;
  double theMaxPCut;
  double theDecRCut2;
  double theEtaCutForHector; 
  double theDecLenCut;
  int verbose;
  LumiMonitorFilter* fLumiFilter; 
  HepMC::GenEvent*  evt_;
  math::XYZTLorentzVector* vtx_;
  double weight_;    
  double Z_lmin,Z_lmax,Z_hector;
  std::vector<int> pdgFilter;
  bool pdgFilterSel;
  bool fPDGFilter;
};

#endif