FortranCallback.h

Go to the documentation of this file.

#ifndef gen_FortranCallback_h
#define gen_FortranCallback_h

#include <boost/shared_ptr.hpp>

#include "SimDataFormats/GeneratorProducts/interface/LHECommonBlocks.h"

#include "GeneratorInterface/LHEInterface/interface/LHERunInfo.h"
#include "GeneratorInterface/LHEInterface/interface/LHEEvent.h"

namespace HepMC
{
class GenEvent;
}

namespace gen {

  class FortranCallback 
  {

    public:
    
      static FortranCallback* getInstance() ;
    
      //void setLHEEvent(lhef::LHEEvent* lhev) { fPartonLevel = lhev; }
      void setLHERunInfo( lhef::LHERunInfo* lheri ) { fRunBlock=lheri; }
      void setLHEEvent( lhef::LHEEvent* lhee ) { fEventBlock=lhee; }
      
      void resetIterationsPerEvent() { fIterationsPerEvent = 0; }
      
      void fillHeader();
      void fillEvent();  
      
      int getIterationsPerEvent() const { return fIterationsPerEvent; }

   private:
   
     // ctor
     
     FortranCallback();

     // data member(s)
     
     lhef::LHERunInfo*       fRunBlock;
     lhef::LHEEvent*         fEventBlock;
     int                     fIterationsPerEvent;
     
     static FortranCallback* fInstance;

    
  };

// --** Implementation **---

FortranCallback* FortranCallback::fInstance = 0;

FortranCallback::FortranCallback()
//   : fPartonLevel(0)
   : fRunBlock(0), fEventBlock(0), fIterationsPerEvent(0)
{   
}

FortranCallback* FortranCallback::getInstance()
{
   if ( fInstance == 0 ) fInstance = new FortranCallback;
   return fInstance;
}

void FortranCallback::fillHeader()
{
   
   if ( fRunBlock == 0 ) return;
      
   //const lhef::HEPRUP* heprup = &(fRunBlock->heprup());
   const lhef::HEPRUP* heprup = fRunBlock->getHEPRUP();
   
   lhef::CommonBlocks::fillHEPRUP(heprup);   
   
   return;

}

void FortranCallback::fillEvent()
{

   //if ( fPartonLevel == 0 ) return;    
   //const lhef::HEPEUP* hepeup = fPartonLevel->getHEPEUP();
   
   if ( fEventBlock == 0 ) return;
   
   const lhef::HEPEUP* hepeup = fEventBlock->getHEPEUP(); 

        if (fIterationsPerEvent++) 
    {
           hepeup_.nup = 0;
           return;
        }

   lhef::CommonBlocks::fillHEPEUP(hepeup);

   return;
    
}


} // end namespace

#endif