GeneratorInterface/PhotosInterface/interface/PhotosppInterface.h - master - githubmirror/cmssw

GeneratorInterface/PhotosInterface/interface/PhotosppInterface.h

53 строки1 KB

Mikhail Kirsanov

adjust photospp interface to work with HepMC3

14 янв 2025, 22:37

14 янв 2025, 22:37cec71ee

#ifndef gen_PhotosInterface_PhotosppInterface_h
#define gen_PhotosInterface_PhotosppInterface_h

// #include "HepPDT/ParticleDataTable.hh"

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

#include "HepMC/SimpleVector.h"
#include "GeneratorInterface/PhotosInterface/interface/PhotosInterfaceBase.h"

namespace HepMC {
  class GenEvent;
  class GenVertex;
}  // namespace HepMC

namespace HepMC3 {
  class GenEvent;
}  // namespace HepMC3

namespace gen {
  class PhotosppInterface : public PhotosInterfaceBase {
  public:
    // ctor & dtor
    PhotosppInterface(const edm::ParameterSet& pset);
    ~PhotosppInterface() override {}

    void init() override;
    const std::vector<std::string>& specialSettings() override { return fSpecialSettings; }
    HepMC::GenEvent* apply(HepMC::GenEvent*) override;
    HepMC3::GenEvent* apply(HepMC3::GenEvent*) override;
    void configureOnlyFor(int) override;
    void avoidTauLeptonicDecays() override {
      fAvoidTauLeptonicDecays = true;
      return;
    }
    bool isTauLeptonicDecay(HepMC::GenVertex*);
    void setRandomEngine(CLHEP::HepRandomEngine* decayRandomEngine) override;
    static double flat();
    void statistics() override;

  private:
    int fOnlyPDG;
    bool fAvoidTauLeptonicDecays;
    bool fIsInitialized;
    edm::ParameterSet* fPSet;

    static CLHEP::HepRandomEngine* fRandomEngine;
  };
}  // namespace gen

#endif