FortranCallback.h

Go to the documentation of this file.

#ifndef gen_FortranCallback_h
#define gen_FortranCallback_h

#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 = nullptr;

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

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

  void FortranCallback::fillHeader() {
    if (fRunBlock == nullptr)
      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 == nullptr)
      return;

    const lhef::HEPEUP* hepeup = fEventBlock->getHEPEUP();

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

    lhef::CommonBlocks::fillHEPEUP(hepeup);

    return;
  }

}  // namespace gen

#endif