EvtGenInterface.h

Go to the documentation of this file.

// Class Based on EvtGenInterface(LHC).
//
// Created March 2014
//
// This class is a modification of the original EvtGenInterface which was developed for EvtGenLHC 9.1.
// The modifications for EvtGen 1.3.0 are implemented by Ian M. Nugent
// I would like to thank the EvtGen developers, in particular John Black, and Mikhail Kirsanov for their assistance.
//

#ifndef gen_EvtGenInterface_h
#define gen_EvtGenInterface_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include <memory>
#include <string>
#include <vector>

#include "EvtGenBase/EvtParticle.hh"
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Utilities/interface/RandomNumberGenerator.h"
#include "GeneratorInterface/EvtGenInterface/interface/EvtGenInterfaceBase.h"

class myEvtRandomEngine;

namespace HepMC {
  class GenParticle;
  class GenEvent;
}  // namespace HepMC

class EvtId;
class EvtGen;

namespace gen {
  class EvtGenInterface : public EvtGenInterfaceBase {
  public:
    // ctor & dtor
    EvtGenInterface(const edm::ParameterSet&);
    ~EvtGenInterface() override;

    void init() override;
    const std::vector<int>& operatesOnParticles() override { return m_PDGs; }
    HepMC::GenEvent* decay(HepMC::GenEvent*) override;
    void setRandomEngine(CLHEP::HepRandomEngine* v) override;
    static double flat();

  private:
    bool addToHepMC(HepMC::GenParticle* partHep, const EvtId& idEvt, HepMC::GenEvent* theEvent, bool del_daug);
    void update_particles(HepMC::GenParticle* partHep, HepMC::GenEvent* theEvent, HepMC::GenParticle* p);
    void SetDefault_m_PDGs();
    bool findLastinChain(HepMC::GenParticle*& p);
    bool hasnoDaughter(HepMC::GenParticle* p);
    void go_through_daughters(EvtParticle* part);

    EvtGen* m_EvtGen;  // EvtGen main  object

    std::vector<EvtId> forced_id;    // EvtGen Id's of particles  which are to be forced by EvtGen
    std::vector<int> forced_pdgids;  // PDG Id's of particles which are to be forced by EvtGen

    std::vector<int> ignore_pdgids;  // HepId's of particles  which are to be ignroed by EvtGen

    // Adding parameters for polarization of spin-1/2 particles
    std::vector<int> polarize_ids;
    std::vector<double> polarize_pol;
    std::map<int, float> polarizations;
    int BmixingOption = 1;
    edm::ParameterSet* fPSet;

    static CLHEP::HepRandomEngine* fRandomEngine;
    myEvtRandomEngine* the_engine;
  };
}  // namespace gen
#endif