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

GeneratorInterface/Pythia8Interface/interface/Py8InterfaceBase.h

73 строки2 KB

Markus Seidel

Adaptions for Pythia 8.310

27 мар 2024, 13:58

27 мар 2024, 13:5874695b3

#ifndef gen_Py8InterfaceBase_h
#define gen_Py8InterfaceBase_h

#include <vector>
#include <string>

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

#include "HepMC/IO_AsciiParticles.h"

#include <Pythia8/Pythia.h>
#include <Pythia8Plugins/HepMC2.h>
//#include <Dire/Dire.h>

namespace Pythia8 {
  class EvtGenDecays;
}

namespace CLHEP {
  class HepRandomEngine;
}

namespace gen {

  class Py8InterfaceBase : public BaseHadronizer {
  public:
    Py8InterfaceBase(edm::ParameterSet const& ps);
    ~Py8InterfaceBase() override {}

    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;
    HepMC::Pythia8ToHepMC toHepMC;
    // 	 ParameterCollector	        fParameters;
    edm::ParameterSet fParameters;

    unsigned int pythiaPylistVerbosity;
    bool pythiaHepMCVerbosity;
    bool pythiaHepMCVerbosityParticles;
    unsigned int maxEventsToPrint;
    HepMC::IO_AsciiParticles* ascii_io;

    // 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