MCDBSource.cc File Reference

#include <iostream>
#include <utility>
#include <string>
#include <boost/regex.hpp>
#include "mcdb.hpp"
#include "FWCore/Framework/interface/InputSourceMacros.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Run.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "SimDataFormats/GeneratorProducts/interface/LesHouches.h"
#include "GeneratorInterface/LHEInterface/interface/LHEEvent.h"
#include "GeneratorInterface/LHEInterface/interface/LHEReader.h"
#include "LHESource.h"

Go to the source code of this file.

Classes
class	MCDBSource

Functions
static edm::ParameterSet	augmentPSetFromMCDB (const edm::ParameterSet &params)
	DEFINE_FWK_INPUT_SOURCE (MCDBSource)
static std::pair< std::vector < std::string >, unsigned int >	getFileURLs (const edm::ParameterSet &params)

Function Documentation

static edm::ParameterSet augmentPSetFromMCDB ( const edm::ParameterSet &  params )

static

Definition at line 111 of file MCDBSource.cc.

References edm::ParameterSet::addUntrackedParameter(), getFileURLs(), and query::result.

{
    // note that this is inherently ugly, but the only way to
    // pass the file URLs to ExternalInputSource, as the MCDB client
    // is nothing but an MCDB to URL converter
    // all modified parameters are untracked, so the provenance is
    // unchanged

    std::pair<std::vector<std::string>, unsigned int> result =
                            getFileURLs(params);

    edm::ParameterSet newParams = params;
    newParams.addUntrackedParameter<std::vector<std::string> >(
                        "fileNames", result.first);
    newParams.addUntrackedParameter<unsigned int>(
                        "skipEvents", result.second);

    return newParams;
}

DEFINE_FWK_INPUT_SOURCE

(

MCDBSource

)

static std::pair<std::vector<std::string>, unsigned int> getFileURLs ( const edm::ParameterSet &  params )

static

Definition at line 32 of file MCDBSource.cc.

References mergeVDriftHistosByStation::file, alcazmumu_cfi::filter, create_complex_test_file0_cfg::firstEvent, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), nEvents, and scaleCards::path.

Referenced by augmentPSetFromMCDB().

{
    unsigned int articleId = params.getParameter<unsigned int>("articleID");

    edm::LogInfo("Generator|LHEInterface")
        << "Reading article id " << articleId << " from MCDB."
        << std::endl;

    mcdb::MCDB mcdb;

    mcdb::Article article = mcdb.getArticle(articleId);

    edm::LogInfo("Generator|LHEInterface")
        << "Title: " << article.title() << std::endl
        << "First author: " << article.authors()[0].firstName() << " "
        << article.authors()[0].lastName() << std::endl
        << "Number of authors: " << article.authors().size() << std::endl
        << "Abstract: " << article.abstract() << std::endl
        << "Generator: " << article.generator().name()
        << ", " << article.generator().version() << std::endl
        << "Number of files: " << article.files().size() << std::endl
        << "Files: " << std::endl;

    std::vector<std::string> supportedProtocols = 
        params.getUntrackedParameter< std::vector<std::string> >(
                            "supportedProtocols");

    boost::regex filter(params.getUntrackedParameter<std::string>(
                            "filter", "\\.lhef?$"),
                        boost::regex_constants::normal |
                        boost::regex_constants::icase);

    unsigned int firstEvent =
        params.getUntrackedParameter<unsigned int>("skipEvents", 0);

    unsigned int fcount = 0;
    std::vector<std::string> fileURLs;
    for(std::vector<mcdb::File>::iterator file = article.files().begin();
        file != article.files().end(); ++file) {
        std::string fileURL;
        for(std::vector<std::string>::const_iterator prot =
                        supportedProtocols.begin();
            prot != supportedProtocols.end(); ++prot) {
            for(std::vector<std::string>::const_iterator path =
                            file->paths().begin();
                path != file->paths().end(); ++path) {
                if (path->substr(0, prot->length() + 1) ==
                    *prot + ":") {
                    fileURL = *path;
                    break;
                }
            }
            if (!fileURL.empty())
                break;
        }

        if (fileURL.empty())
            throw cms::Exception("Generator|LHEInterface")
                << "MCDB did not contain any URLs with"
                   " supported protocols for at least one"
                   " file." << std::endl;

        if (!boost::regex_search(fileURL, filter))
            continue;

        int nEvents = file->eventsNumber();
        if (nEvents > 0 && (int)firstEvent >= nEvents) {
            firstEvent -= nEvents;
            continue;
        }

        fileURLs.push_back(fileURL);
        fcount++;
        edm::LogInfo("Generator|LHEInterface") << "Adding file n. " << fcount << " " << fileURL;
    }

    return std::make_pair(fileURLs, firstEvent);
}