GeneratorInterface/Core/interface/BaseHadronizer.h - master - githubmirror/cmssw

GeneratorInterface/Core/interface/BaseHadronizer.h
131 строка5 KB
Manuel González Hernández
hepID override to use weights in LO
09 фев 2026, 19:01
09 фев 2026, 19:01a4b7b6c
// -*- C++ -*-
//
//

// class BaseHadronizer meant as base class for hadronizers
// implements a few common methods concerning communication with the
// gen::HadronizerFilter<...> and gen::GeneratorFilter<...> templates,
// mostly memory management regarding the GenEvent pointers and such

#ifndef gen_BaseHadronizer_h
#define gen_BaseHadronizer_h

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

#include <memory>

#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMC3Product.h"

#include "SimDataFormats/GeneratorProducts/interface/GenRunInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct3.h"
#include "SimDataFormats/GeneratorProducts/interface/GenLumiInfoHeader.h"

#include "SimDataFormats/GeneratorProducts/interface/LHERunInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/LHEEventProduct.h"

#include "GeneratorInterface/LHEInterface/interface/LHERunInfo.h"
#include "GeneratorInterface/LHEInterface/interface/LHEEvent.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"

#include "CLHEP/Random/RandomEngine.h"

// foward declarations
namespace edm {
  class Event;
}

namespace CLHEP {
  class HepRandomEngine;
}

namespace gen {

  class BaseHadronizer {
  public:
    BaseHadronizer(edm::ParameterSet const& ps);
    virtual ~BaseHadronizer() noexcept(false) {}

    // GenRunInfo and GenEvent passing
    GenRunInfoProduct& getGenRunInfo() { return genRunInfo_; }
    std::unique_ptr<HepMC::GenEvent> getGenEvent() { return std::move(genEvent_); }
    std::unique_ptr<HepMC3::GenEvent> getGenEvent3() { return std::move(genEvent3_); }
    std::unique_ptr<GenEventInfoProduct> getGenEventInfo() { return std::move(genEventInfo_); }
    std::unique_ptr<GenEventInfoProduct3> getGenEventInfo3() { return std::move(genEventInfo3_); }
    virtual std::unique_ptr<GenLumiInfoHeader> getGenLumiInfoHeader() const;
    std::unique_ptr<lhef::LHEEvent> getLHEEvent() { return std::move(lheEvent_); }

    void resetEvent(std::unique_ptr<HepMC::GenEvent> event) { genEvent_ = std::move(event); }
    void resetEvent3(std::unique_ptr<HepMC3::GenEvent> event3) { genEvent3_ = std::move(event3); }
    void resetEventInfo(std::unique_ptr<GenEventInfoProduct> eventInfo) { genEventInfo_ = std::move(eventInfo); }
    void resetEventInfo3(std::unique_ptr<GenEventInfoProduct3> eventInfo) { genEventInfo3_ = std::move(eventInfo); }

    // LHERunInfo and LHEEvent passing
    const std::shared_ptr<lhef::LHERunInfo>& getLHERunInfo() const { return lheRunInfo_; }

    void setLHERunInfo(std::unique_ptr<lhef::LHERunInfo> runInfo) { lheRunInfo_ = std::move(runInfo); }
    void setLHEEvent(std::unique_ptr<lhef::LHEEvent> event) { lheEvent_ = std::move(event); }

    // interface for accessing the EDM information from the hadronizer
    void setEDMEvent(edm::Event& event) { edmEvent_ = &event; }
    edm::Event& getEDMEvent() const { return *edmEvent_; }
    virtual bool select(HepMC::GenEvent*) const { return true; }

    void setRandomEngine(CLHEP::HepRandomEngine* v) { doSetRandomEngine(v); }

    std::vector<std::string> const& sharedResources() const { return doSharedResources(); }

    int randomIndex() const { return randomIndex_; }
    const std::string& randomInitConfigDescription() const { return randomInitConfigDescriptions_[randomIndex_]; }
    const std::string& gridpackPath() const { return gridpackPaths_[std::max(randomIndex_, 0)]; }

    void randomizeIndex(edm::LuminosityBlock const& lumi, CLHEP::HepRandomEngine* rengine);
    void generateLHE(edm::LuminosityBlock const& lumi, CLHEP::HepRandomEngine* rengine, unsigned int ncpu);
    void cleanLHE();
    unsigned int getVHepMC() { return ivhepmc; }
    virtual int getOverrideHEPIDWTUP() { return -999; }

  protected:
    unsigned int ivhepmc = 2;
    GenRunInfoProduct& runInfo() { return genRunInfo_; }
    std::unique_ptr<HepMC::GenEvent>& event() { return genEvent_; }
    std::unique_ptr<GenEventInfoProduct>& eventInfo() { return genEventInfo_; }
    //HepMC3:
    std::unique_ptr<HepMC3::GenEvent>& event3() { return genEvent3_; }
    std::unique_ptr<GenEventInfoProduct3>& eventInfo3() { return genEventInfo3_; }

    lhef::LHEEvent* lheEvent() { return lheEvent_.get(); }
    lhef::LHERunInfo* lheRunInfo() { return lheRunInfo_.get(); }
    int randomIndex_;
    std::string lheFile_;

  private:
    virtual void doSetRandomEngine(CLHEP::HepRandomEngine* v) {}

    virtual std::vector<std::string> const& doSharedResources() const { return theSharedResources; }

    GenRunInfoProduct genRunInfo_;
    std::unique_ptr<HepMC::GenEvent> genEvent_;
    std::unique_ptr<HepMC3::GenEvent> genEvent3_;
    std::unique_ptr<GenEventInfoProduct> genEventInfo_;
    std::unique_ptr<GenEventInfoProduct3> genEventInfo3_;

    std::shared_ptr<lhef::LHERunInfo> lheRunInfo_;
    std::unique_ptr<lhef::LHEEvent> lheEvent_;

    edm::Event* edmEvent_;

    static const std::vector<std::string> theSharedResources;

    std::vector<double> randomInitWeights_;
    std::vector<std::string> randomInitConfigDescriptions_;
    std::vector<std::string> gridpackPaths_;
  };

}  // namespace gen

#endif  // gen_BaseHadronizer_h