GeneratorInterface/ExternalDecays/interface/ExternalDecayDriver.h - master - githubmirror/cmssw

GeneratorInterface/ExternalDecays/interface/ExternalDecayDriver.h

68 строк2 KB

Mikhail Kirsanov

adjust ExternalDecays to work with HepMC3

14 янв 2025, 22:47

14 янв 2025, 22:473d61f17

#ifndef gen_ExternalDecayDriver_h
#define gen_ExternalDecayDriver_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include <string>
#include <vector>

namespace HepMC {
  class GenEvent;
}

namespace HepMC3 {
  class GenEvent;
}

namespace CLHEP {
  class HepRandomEngine;
}

namespace lhef {
  class LHEEvent;
}

namespace gen {

  class EvtGenInterfaceBase;
  class TauolaInterfaceBase;
  class PhotosInterfaceBase;

  class ExternalDecayDriver {
  public:
    // ctor & dtor
    ExternalDecayDriver(const edm::ParameterSet&, edm::ConsumesCollector);
    ~ExternalDecayDriver();

    void init(const edm::EventSetup&);

    const std::vector<int>& operatesOnParticles() { return fPDGs; }
    const std::vector<std::string>& specialSettings() { return fSpecialSettings; }

    HepMC::GenEvent* decay(HepMC::GenEvent* evt);
    HepMC::GenEvent* decay(HepMC::GenEvent* evt, lhef::LHEEvent* lheEvent);

    HepMC3::GenEvent* decay(HepMC3::GenEvent* evt);
    HepMC3::GenEvent* decay(HepMC3::GenEvent* evt, lhef::LHEEvent* lheEvent);

    void statistics() const;

    void setRandomEngine(CLHEP::HepRandomEngine*);
    std::vector<std::string> const& sharedResources() const { return exSharedResources; }

  private:
    bool fIsInitialized;
    std::unique_ptr<TauolaInterfaceBase> fTauolaInterface;
    std::unique_ptr<EvtGenInterfaceBase> fEvtGenInterface;
    std::unique_ptr<PhotosInterfaceBase> fPhotosInterface;
    std::vector<int> fPDGs;
    std::vector<std::string> fSpecialSettings;

    std::vector<std::string> exSharedResources;
  };

}  // namespace gen

#endif