LHESource.h

Go to the documentation of this file.

#ifndef GeneratorInterface_LHEInterface_LHESource_h
#define GeneratorInterface_LHEInterface_LHESource_h

#include <memory>

#include <boost/shared_ptr.hpp>
#include <boost/ptr_container/ptr_deque.hpp>

#include "DataFormats/Provenance/interface/ProcessHistoryID.h"
#include "GeneratorInterface/LHEInterface/plugins/LHEProvenanceHelper.h"
#include "FWCore/Sources/interface/ProducerSourceFromFiles.h"

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

namespace lhef {
  class LHERunInfo;
  class LHEEvent;
  class LHEReader;
}

namespace edm {
  class EventPrincipal;
  class LuminosityBlockAuxiliary;
  class LuminosityBlockPrincipal;
  class ParameterSet;
  class Run;
  class RunAuxiliary;
  class RunPrincipal;
}

class LHERunInfoProduct;

class LHESource : public edm::ProducerSourceFromFiles {
public:
  explicit LHESource(const edm::ParameterSet &params,
                     const edm::InputSourceDescription &desc);
  ~LHESource() override;

private:
  void endJob() override;
  void beginRun(edm::Run &run) override;
  void endRun(edm::Run &run) override;
  bool setRunAndEventInfo(edm::EventID&, edm::TimeValue_t&, edm::EventAuxiliary::ExperimentType&) override;
  void readRun_(edm::RunPrincipal& runPrincipal) override;
  void readLuminosityBlock_(edm::LuminosityBlockPrincipal& lumiPrincipal) override;
  void readEvent_(edm::EventPrincipal& eventPrincipal) override;
  void produce(edm::Event&) override {}
  std::shared_ptr<edm::RunAuxiliary> readRunAuxiliary_() override;
  std::shared_ptr<edm::LuminosityBlockAuxiliary> readLuminosityBlockAuxiliary_() override;

  void nextEvent();

  std::auto_ptr<lhef::LHEReader>        reader;

  boost::shared_ptr<lhef::LHERunInfo>   runInfoLast;
  boost::shared_ptr<lhef::LHERunInfo>   runInfo;
  boost::shared_ptr<lhef::LHEEvent> partonLevel;

  boost::ptr_deque<LHERunInfoProduct>   runInfoProducts;
  bool                  wasMerged;
  edm::LHEProvenanceHelper      lheProvenanceHelper_;
  edm::ProcessHistoryID         phid_;
  edm::RunPrincipal*                    runPrincipal_;
};

#endif // GeneratorInterface_LHEInterface_LHESource_h