PhotosInterface.h

Go to the documentation of this file.

#ifndef gen_PhotosInterface_PhotosInterface_h
#define gen_PhotosInterface_PhotosInterface_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 gen {
  class PhotosInterface : public PhotosInterfaceBase {
  public:
    // ctor & dtor
    PhotosInterface();
    PhotosInterface(const edm::ParameterSet&);
    ~PhotosInterface() override;
 
    void init() override;
    const std::vector<std::string>& specialSettings() override { return fSpecialSettings; }
    HepMC::GenEvent* apply(HepMC::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();
 
  private:
    int fOnlyPDG;
    bool fAvoidTauLeptonicDecays;
    std::vector<int> fBarcodes;
    std::vector<int> fSecVtxStore;
    bool fIsInitialized;
 
    void applyToVertex(HepMC::GenEvent*, int);
    void applyToBranch(HepMC::GenEvent*, int);
    void attachParticles(HepMC::GenEvent*, HepMC::GenVertex*, int);
 
    static CLHEP::HepRandomEngine* fRandomEngine;
  };
}  // namespace gen
 
#endif