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() {}

      void init();
      const std::vector<std::string>& specialSettings() { return fSpecialSettings; }
      HepMC::GenEvent* apply( HepMC::GenEvent* );
      void configureOnlyFor( int );
      void avoidTauLeptonicDecays() { fAvoidTauLeptonicDecays=true; return; }
      bool isTauLeptonicDecay( HepMC::GenVertex* );
      void SetDecayRandomEngine(CLHEP::HepRandomEngine* decayRandomEngine);
      
      private: 
            
      struct Scaling {
         HepMC::ThreeVector weights;
     int                flag;
     Scaling( HepMC::ThreeVector vec, int flg ) 
        : weights(HepMC::ThreeVector(1.,1.,1)), flag(1) { weights=vec; flag=flg; }
      };

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

}

#endif