lhef::LHEReader Class Reference

#include <LHEReader.h>

Classes
class	FileSource
class	Source
class	StringSource
class	XMLHandler

Public Member Functions
	LHEReader (const edm::ParameterSet &params)
	LHEReader (const std::vector< std::string > &fileNames, unsigned int skip=0)
	LHEReader (const std::string &inputs, unsigned int skip=0)
std::shared_ptr< LHEEvent >	next (bool *newFileOpened=nullptr)
	~LHEReader ()

Private Attributes
std::unique_ptr< XMLDocument >	curDoc
unsigned int	curIndex
std::shared_ptr< LHERunInfo >	curRunInfo
std::unique_ptr< Source >	curSource
const std::vector< std::string >	fileURLs
unsigned int	firstEvent
std::unique_ptr< XMLHandler >	handler
int	maxEvents
std::shared_ptr< void >	platform
const std::string	strName
std::vector< std::string >	weightsinconfig

Detailed Description

Definition at line 16 of file LHEReader.h.

Constructor & Destructor Documentation

lhef::LHEReader::LHEReader

(

const edm::ParameterSet &

params

)

Definition at line 415 of file LHEReader.cc.

      : fileURLs(params.getUntrackedParameter<std::vector<std::string> >("fileNames")),
        strName(""),
        firstEvent(params.getUntrackedParameter<unsigned int>("skipEvents", 0)),
        maxEvents(params.getUntrackedParameter<int>("limitEvents", -1)),
        curIndex(0),
        handler(new XMLHandler()) {}

lhef::LHEReader::LHEReader	(	const std::vector< std::string > &	fileNames,
		unsigned int	skip = 0
	)

Definition at line 423 of file LHEReader.cc.

      : fileURLs(fileNames),
        strName(""),
        firstEvent(firstEvent),
        maxEvents(-1),
        curIndex(0),
        handler(new XMLHandler()) {}

lhef::LHEReader::LHEReader	(	const std::string &	inputs,
		unsigned int	skip = 0
	)

Definition at line 431 of file LHEReader.cc.

      : strName(inputs), firstEvent(firstEvent), maxEvents(-1), curIndex(0), handler(new XMLHandler()) {}

lhef::LHEReader::~LHEReader

(

)

Definition at line 434 of file LHEReader.cc.

References curDoc, curSource, and handler.

                        {
    // Explicitly release "orphaned" resources
    // that were created through DOM implementation
    // createXXXX factory method *before* last
    // XMLPlatformUtils::Terminate is called.
    handler.release();
    curDoc.release();
    curSource.release();
  }

Member Function Documentation

std::shared_ptr< LHEEvent > lhef::LHEReader::next

(

bool *

newFileOpened = nullptr

)

Definition at line 444 of file LHEReader.cc.

References lhef::LHERunInfo::addHeader(), curDoc, curIndex, curRunInfo, curSource, data, edmPickEvents::event, fileURLs, first, firstEvent, handler, mps_fire::i, info(), lhef::LHEReader::XMLHandler::kComment, lhef::LHEReader::XMLHandler::kEvent, lhef::LHEReader::XMLHandler::kHeader, lhef::LHEReader::XMLHandler::kInit, lhef::LHEReader::XMLHandler::kNone, lhef::logFileAction(), maxEvents, pileupDistInMC::num, platform, lhef::XMLDocument::platformHandle(), edm::second(), and strName.

Referenced by BeautifulSoup.PageElement::_invert().

                                                             {
    while (curDoc.get() || curIndex < fileURLs.size() || (fileURLs.empty() && !strName.empty())) {
      if (!curDoc.get()) {
        if (!platform) {
          //If we read multiple files, the XercesPlatform must live longer than any one
          // XMLDocument.
          platform = XMLDocument::platformHandle();
        }
        if (!fileURLs.empty()) {
          logFileAction("  Initiating request to open LHE file ", fileURLs[curIndex]);
          curSource = std::make_unique<FileSource>(fileURLs[curIndex]);
          logFileAction("  Successfully opened LHE file ", fileURLs[curIndex]);
          if (newFileOpened != nullptr)
            *newFileOpened = true;
          ++curIndex;
        } else if (!strName.empty()) {
          curSource = std::make_unique<StringSource>(strName);
        }
        handler->reset();
        curDoc.reset(curSource->createReader(*handler));
        curRunInfo.reset();
      }
      handler->skipEvent = firstEvent > 0;

      XMLHandler::Object event = handler->gotObject;
      handler->gotObject = XMLHandler::kNone;

      switch (event) {
        case XMLHandler::kNone:
          if (!curDoc->parse()) {
            curDoc.reset();
            logFileAction("  Closed LHE file ", fileURLs[curIndex - 1]);
            return std::shared_ptr<LHEEvent>();
          }
          break;

        case XMLHandler::kHeader:
          break;

        case XMLHandler::kInit: {
          std::istringstream data;
          data.str(handler->buffer);
          handler->buffer.clear();

          curRunInfo.reset(new LHERunInfo(data));

          std::for_each(handler->headers.begin(),
                        handler->headers.end(),
                        std::bind(&LHERunInfo::addHeader, curRunInfo.get(), std::placeholders::_1));
          handler->headers.clear();
        } break;

        case XMLHandler::kComment:
          break;

        case XMLHandler::kEvent: {
          if (!curRunInfo.get())
            throw cms::Exception("InvalidState") << "Got LHE event without"
                                                    " initialization."
                                                 << std::endl;

          if (firstEvent > 0) {
            firstEvent--;
            continue;
          }

          if (maxEvents == 0)
            return std::shared_ptr<LHEEvent>();
          else if (maxEvents > 0)
            maxEvents--;

          std::istringstream data;
          data.str(handler->buffer);
          handler->buffer.clear();

          std::shared_ptr<LHEEvent> lheevent;
          lheevent.reset(new LHEEvent(curRunInfo, data));
          const XMLHandler::wgt_info &info = handler->weightsinevent;
          for (size_t i = 0; i < info.size(); ++i) {
            double num = -1.0;
            sscanf(info[i].second.c_str(), "%le", &num);
            lheevent->addWeight(gen::WeightsInfo(info[i].first, num));
          }
          lheevent->setNpLO(handler->npLO);
          lheevent->setNpNLO(handler->npNLO);
          //fill scales
          if (!handler->scales.empty()) {
            lheevent->setScales(handler->scales);
          }
          return lheevent;
        }
      }
    }

    return std::shared_ptr<LHEEvent>();
  }

Member Data Documentation

std::unique_ptr<XMLDocument> lhef::LHEReader::curDoc

private

Definition at line 39 of file LHEReader.h.

Referenced by next(), and ~LHEReader().

unsigned int lhef::LHEReader::curIndex

private

Definition at line 35 of file LHEReader.h.

Referenced by next().

std::shared_ptr<LHERunInfo> lhef::LHEReader::curRunInfo

private

Definition at line 40 of file LHEReader.h.

Referenced by next().

std::unique_ptr<Source> lhef::LHEReader::curSource

private

Definition at line 38 of file LHEReader.h.

Referenced by next(), and ~LHEReader().

const std::vector<std::string> lhef::LHEReader::fileURLs

private

Definition at line 29 of file LHEReader.h.

Referenced by next().

unsigned int lhef::LHEReader::firstEvent

private

Definition at line 33 of file LHEReader.h.

Referenced by looper.Looper::loop(), next(), and Printer.Printer::process().

std::unique_ptr<XMLHandler> lhef::LHEReader::handler

private

Definition at line 41 of file LHEReader.h.

Referenced by next(), and ~LHEReader().

int lhef::LHEReader::maxEvents

private

Definition at line 34 of file LHEReader.h.

Referenced by next().

std::shared_ptr<void> lhef::LHEReader::platform

private

Definition at line 42 of file LHEReader.h.

Referenced by next().

const std::string lhef::LHEReader::strName

private

Definition at line 32 of file LHEReader.h.

Referenced by next().

std::vector<std::string> lhef::LHEReader::weightsinconfig

private

Definition at line 36 of file LHEReader.h.