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 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;
    
  };
}

#endif