GeneratorInterface/Pythia8Interface/interface/Py8HMC3InterfaceBase.h - master - githubmirror/cmssw

GeneratorInterface/Pythia8Interface/interface/Py8HMC3InterfaceBase.h

68 строк2 KB

Markus Seidel

Fix Pythia8 HepMC3 random engine ptr

24 май 2024, 18:01

24 май 2024, 18:0122c8d42

#ifndef gen_Py8HMC3InterfaceBase_h
#define gen_Py8HMC3InterfaceBase_h

#include <vector>
#include <string>

#include "GeneratorInterface/Core/interface/ParameterCollector.h"
#include "GeneratorInterface/Pythia8Interface/interface/P8RndmEngine.h"
#include "GeneratorInterface/Core/interface/BaseHadronizer.h"

#include <Pythia8/Pythia.h>
#include <Pythia8Plugins/HepMC3.h>

namespace Pythia8 {
  class EvtGenDecays;
}

namespace CLHEP {
  class HepRandomEngine;
}

namespace gen {

  class Py8HMC3InterfaceBase : public BaseHadronizer {
  public:
    Py8HMC3InterfaceBase(edm::ParameterSet const& ps);
    ~Py8HMC3InterfaceBase() override = default;

    virtual bool generatePartonsAndHadronize() = 0;
    bool decay() { return true; }          // NOT used - let's call it "design imperfection"
    bool readSettings(int);                // common func
    void makeTmpSLHA(const std::string&);  //helper for above
    virtual bool initializeForInternalPartons() = 0;
    bool declareStableParticles(const std::vector<int>&);          // common func
    bool declareSpecialSettings(const std::vector<std::string>&);  // common func
    virtual void finalizeEvent() = 0;
    virtual void statistics();
    virtual const char* classname() const = 0;

    void p8SetRandomEngine(CLHEP::HepRandomEngine* v) { p8RndmEngine_->setRandomEngine(v); }
    P8RndmEngine& randomEngine() { return *p8RndmEngine_; }

  protected:
    std::unique_ptr<Pythia8::Pythia> fMasterGen;
    std::unique_ptr<Pythia8::Pythia> fDecayer;
    HepMC3::Pythia8ToHepMC3 toHepMC;
    edm::ParameterSet fParameters;

    unsigned int pythiaPylistVerbosity;
    bool pythiaHepMCVerbosity;
    bool pythiaHepMCVerbosityParticles;
    unsigned int maxEventsToPrint;

    // EvtGen plugin
    //
    bool useEvtGen;
    std::shared_ptr<Pythia8::EvtGenDecays> evtgenDecays;
    std::string evtgenDecFile;
    std::string evtgenPdlFile;
    std::vector<std::string> evtgenUserFiles;

    std::string slhafile_;

  private:
    std::shared_ptr<P8RndmEngine> p8RndmEngine_;
  };
}  // namespace gen
#endif