GeneratorInterface/Pythia6Interface/plugins/Pythia6Hadronizer.h - master - githubmirror/cmssw

GeneratorInterface/Pythia6Interface/plugins/Pythia6Hadronizer.h
120 строк3 KB
Matti Kortelainen
Replace LHEEventProduct forward declaration with include
16 мар 2026, 11:45
16 мар 2026, 11:45acbc5b9
#ifndef gen_Pythia6Hadronizer_h
#define gen_Pythia6Hadronizer_h

// -*- C++ -*-

// class Pythia6Hadronizer is an example of a class that models the
// Hadronizer concept.

#include <memory>
#include <string>
#include <vector>

#include "FWCore/ParameterSet/interface/ParameterSetfwd.h"

#include "GeneratorInterface/Core/interface/BaseHadronizer.h"
#include "SimDataFormats/GeneratorProducts/interface/LHEEventProduct.h"

namespace lhef {
  class LHERunInfo;
  class LHEEvent;
}  // namespace lhef

namespace HepMC {
  class GenEvent;
}

namespace CLHEP {
  class HepRandomEngine;
}

namespace gen {

  class Pythia6Service;
  class JetMatching;

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

    // bool generatePartons();
    bool generatePartonsAndHadronize();
    bool hadronize();
    bool decay();
    bool residualDecay();
    bool readSettings(int);
    bool initializeForExternalPartons();
    bool initializeForInternalPartons();
    bool declareStableParticles(const std::vector<int>&);
    bool declareSpecialSettings(const std::vector<std::string>&);

    static JetMatching* getJetMatching() { return fJetMatching; }

    void finalizeEvent();

    void statistics();

    const char* classname() const;

  private:
    // methods
    //

    void doSetRandomEngine(CLHEP::HepRandomEngine* v) override;
    std::vector<std::string> const& doSharedResources() const override { return theSharedResources; }

    void flushTmpStorage();
    void fillTmpStorage();

    void imposeProperTime();

    // data members
    //

    Pythia6Service* fPy6Service;

    // some of the following params are common for all generators(interfaces)
    // probably better to wrap them up in a class and reuse ?
    // (the event/run pointers are already moved to BaseHadronizer)
    //
    enum { PP, PPbar, ElectronPositron, ElectronProton, PositronProton };

    int fInitialState;  // pp, ppbar, e-e+, or e-p
    double fCOMEnergy;  // this one is irrelevant for setting py6 as hadronizer
                        // or if anything, it should be picked up from LHERunInfoProduct !
    double fBeam1PZ;
    double fBeam2PZ;

    static JetMatching* fJetMatching;

    bool fHepMCVerbosity;
    unsigned int fMaxEventsToPrint;

    // this is the only one specific to Pythia6
    //
    unsigned int fPythiaListVerbosity;
    bool fDisplayPythiaBanner;
    bool fDisplayPythiaCards;

    // these two params control stop- and r-handron features,
    // that are "custom" add-ons to Py6;
    // I doubt they should drag along Py6Int main library...
    //
    bool fStopHadronsEnabled;
    bool fGluinoHadronsEnabled;

    // this is a "trick" to generate enriched mu-samples and the likes
    bool fImposeProperTime;

    // and final touch - conversion of Py6 PID's into PDG convension
    bool fConvertToPDG;

    // tmp stuff, to deal with EvtGen corrupting pyjets
    // int NPartsBeforeDecays;

    static const std::vector<std::string> theSharedResources;
  };
}  // namespace gen

#endif