#include <RivetAnalyzer.h>

Inheritance diagram for RivetAnalyzer:

Public Types
typedef dqm::legacy::DQMStore	DQMStore

typedef dqm::legacy::MonitorElement	MonitorElement

Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Public Member Functions
void	analyze (const edm::Event &, const edm::EventSetup &) override

void	beginJob () override

void	beginLuminosityBlock (const edm::LuminosityBlock &, const edm::EventSetup &) override

void	beginRun (const edm::Run &, const edm::EventSetup &) override

void	endJob () override
	List of registered analysis data objects. More...

void	endLuminosityBlock (const edm::LuminosityBlock &, const edm::EventSetup &) override

void	endRun (const edm::Run &, const edm::EventSetup &) override

	RivetAnalyzer (const edm::ParameterSet &)

	~RivetAnalyzer () override

Public Member Functions inherited from edm::one::EDAnalyzer< edm::one::WatchRuns, edm::one::WatchLuminosityBlocks, edm::one::SharedResources >
	EDAnalyzer ()=default

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDAnalyzerBase () override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	normalizeTree ()

Private Attributes
Rivet::AnalysisHandler	_analysisHandler

bool	_doFinalize

edm::EDGetTokenT< GenEventInfoProduct >	_genEventInfoCollection

edm::EDGetTokenT< GenLumiInfoHeader >	_genLumiInfoToken

int	_GENweightNumber

edm::EDGetTokenT< edm::HepMCProduct >	_hepmcCollection

bool	_isFirstEvent

edm::EDGetTokenT< LHEEventProduct >	_LHECollection

const edm::InputTag	_lheLabel

edm::EDGetTokenT< LHERunInfoProduct >	_lheRunInfoToken

std::vector< std::string >	_lheWeightNames

int	_LHEweightNumber

std::vector< MonitorElement * >	_mes

std::string	_outFileName

bool	_produceDQM

bool	_useExternalWeight

bool	_useGENweights

bool	_useLHEweights

std::vector< std::string >	_weightNames

double	_xsection

DQMStore *	dbe

Additional Inherited Members
Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 22 of file RivetAnalyzer.h.

Member Typedef Documentation

typedef dqm::legacy::DQMStore RivetAnalyzer::DQMStore

Definition at line 25 of file RivetAnalyzer.h.

typedef dqm::legacy::MonitorElement RivetAnalyzer::MonitorElement

Definition at line 26 of file RivetAnalyzer.h.

Constructor & Destructor Documentation

RivetAnalyzer::RivetAnalyzer ( const edm::ParameterSet & pset )

Definition at line 19 of file RivetAnalyzer.cc.

References _analysisHandler, _genLumiInfoToken, _hepmcCollection, _LHECollection, _lheLabel, _lheRunInfoToken, _produceDQM, _useLHEweights, _xsection, dbe, edm::ParameterSet::getParameter(), and Utilities::operator.

     : _analysisHandler(),
       _isFirstEvent(true),
       _outFileName(pset.getParameter<std::string>("OutputFile")),
       //decide whether to finalize the plots or not.
       //deciding not to finalize them can be useful for further harvesting of many jobs
       _doFinalize(pset.getParameter<bool>("DoFinalize")),
       _produceDQM(pset.getParameter<bool>("ProduceDQMOutput")),
       _lheLabel(pset.getParameter<edm::InputTag>("LHECollection")),
       _xsection(-1.) {
   usesResource("Rivet");
 
   //retrive the analysis name from parameter set
   std::vector<std::string> analysisNames = pset.getParameter<std::vector<std::string> >("AnalysisNames");
 
   _hepmcCollection = consumes<HepMCProduct>(pset.getParameter<edm::InputTag>("HepMCCollection"));
   _genLumiInfoToken = consumes<GenLumiInfoHeader, edm::InLumi>(pset.getParameter<edm::InputTag>("genLumiInfo"));
 
   _useLHEweights = pset.getParameter<bool>("useLHEweights");
   if (_useLHEweights) {
     _lheRunInfoToken = consumes<LHERunInfoProduct, edm::InRun>(_lheLabel);
     _LHECollection = consumes<LHEEventProduct>(_lheLabel);
   }
 
   //get the analyses
   _analysisHandler.addAnalyses(analysisNames);
 
   //set user cross section if needed
   _xsection = pset.getParameter<double>("CrossSection");
 
   if (_produceDQM) {
     // book stuff needed for DQM
     dbe = nullptr;
     dbe = edm::Service<DQMStore>().operator->();
   }
 }

RivetAnalyzer::~RivetAnalyzer ( )

override

Definition at line 56 of file RivetAnalyzer.cc.

56 {}

Member Function Documentation

void RivetAnalyzer::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 115 of file RivetAnalyzer.cc.

References _analysisHandler, _hepmcCollection, _isFirstEvent, _LHECollection, _lheWeightNames, _useLHEweights, _weightNames, _xsection, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), mps_fire::i, LHEEventProduct::originalXWGTUP(), AlCaHLTBitMon_QueryRunRegistry::string, and LHEEventProduct::weights().

                                                                                {
   //get the hepmc product from the event
   edm::Handle<HepMCProduct> evt;
   iEvent.getByToken(_hepmcCollection, evt);
 
   // get HepMC GenEvent
   const HepMC::GenEvent* myGenEvent = evt->GetEvent();
   std::unique_ptr<HepMC::GenEvent> tmpGenEvtPtr;
   //if you want to use an external weight or set the cross section we have to clone the GenEvent and change the weight
   if (_useLHEweights || _xsection > 0) {
     tmpGenEvtPtr = std::make_unique<HepMC::GenEvent>(*(evt->GetEvent()));
 
     if (_xsection > 0) {
       HepMC::GenCrossSection xsec;
       xsec.set_cross_section(_xsection);
       tmpGenEvtPtr->set_cross_section(xsec);
     }
 
     if (_useLHEweights) {
       std::vector<double> mergedWeights;
       for (unsigned int i = 0; i < tmpGenEvtPtr->weights().size(); i++) {
         mergedWeights.push_back(tmpGenEvtPtr->weights()[i]);
       }
 
       edm::Handle<LHEEventProduct> lheEventHandle;
       iEvent.getByToken(_LHECollection, lheEventHandle);
       for (unsigned int i = 0; i < _lheWeightNames.size(); i++) {
         mergedWeights.push_back(tmpGenEvtPtr->weights()[0] * lheEventHandle->weights().at(i).wgt /
                                 lheEventHandle->originalXWGTUP());
       }
 
       tmpGenEvtPtr->weights() = mergedWeights;
     }
     myGenEvent = tmpGenEvtPtr.get();
   }
 
   //apply the beams initialization on the first event
   if (_isFirstEvent) {
     if (_useLHEweights) {
       _weightNames.insert(_weightNames.end(), _lheWeightNames.begin(), _lheWeightNames.end());
     }
     // clean weight names to be accepted by Rivet plotting
     std::vector<std::string> cleanedWeightNames;
     for (std::string wn : _weightNames) {
       cleanedWeightNames.push_back(std::regex_replace(wn, std::regex("[^A-Za-z\\d\\._=]"), "_"));
     }
     _analysisHandler.init(*myGenEvent, cleanedWeightNames);
     const HepMC::GenCrossSection* xs = myGenEvent->cross_section();
     _analysisHandler.setCrossSection(make_pair(xs->cross_section(), xs->cross_section_error()));
 
     _isFirstEvent = false;
   }
 
   //run the analysis
   _analysisHandler.analyze(*myGenEvent);
 }

void RivetAnalyzer::beginJob ( void )

overridevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 58 of file RivetAnalyzer.cc.

References SiStripCommissioningSource_FromEDM_cfg::cmsswbase, and AlCaHLTBitMon_QueryRunRegistry::string.

                              {
   //set the environment, very ugly but rivet is monolithic when it comes to paths
   char* cmsswbase = std::getenv("CMSSW_BASE");
   char* cmsswrelease = std::getenv("CMSSW_RELEASE_BASE");
   if (!std::getenv("RIVET_REF_PATH")) {
     const std::string rivetref = "RIVET_REF_PATH=" + string(cmsswbase) +
                                  "/src/GeneratorInterface/RivetInterface/data:" + string(cmsswrelease) +
                                  "/src/GeneratorInterface/RivetInterface/data";
     char* rivetrefCstr = strdup(rivetref.c_str());
     putenv(rivetrefCstr);
     free(rivetrefCstr);
   }
   if (!std::getenv("RIVET_INFO_PATH")) {
     const std::string rivetinfo = "RIVET_INFO_PATH=" + string(cmsswbase) +
                                   "/src/GeneratorInterface/RivetInterface/data:" + string(cmsswrelease) +
                                   "/src/GeneratorInterface/RivetInterface/data";
     char* rivetinfoCstr = strdup(rivetinfo.c_str());
     putenv(rivetinfoCstr);
     free(rivetinfoCstr);
   }
 }

void RivetAnalyzer::beginLuminosityBlock	(	const edm::LuminosityBlock &	iLumi,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 101 of file RivetAnalyzer.cc.

References _genLumiInfoToken, _weightNames, edm::LuminosityBlock::getByToken(), and GenLumiInfoHeader::weightNames().

                                                                                                      {
   edm::Handle<GenLumiInfoHeader> genLumiInfoHandle;
   iLumi.getByToken(_genLumiInfoToken, genLumiInfoHandle);
 
   _weightNames = genLumiInfoHandle->weightNames();
 
   // need to reset the default weight name (or plotting will fail)
   if (!_weightNames.empty()) {
     _weightNames[0] = "";
   }
 }

void RivetAnalyzer::beginRun	(	const edm::Run &	iRun,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 80 of file RivetAnalyzer.cc.

References _lheLabel, _lheWeightNames, _useLHEweights, edm::Run::getByLabel(), LHERunInfoProduct::headers_begin(), LHERunInfoProduct::headers_end(), groupFilesInBlocks::lines, and match().

                                                                             {
   if (_useLHEweights) {
     edm::Handle<LHERunInfoProduct> lheRunInfoHandle;
     iRun.getByLabel(_lheLabel, lheRunInfoHandle);
     typedef std::vector<LHERunInfoProduct::Header>::const_iterator headers_const_iterator;
 
     std::regex reg("<weight.*> ?(.*?) ?<\\/weight>");
     for (headers_const_iterator iter = lheRunInfoHandle->headers_begin(); iter != lheRunInfoHandle->headers_end();
          iter++) {
       std::vector<std::string> lines = iter->lines();
       for (unsigned int iLine = 0; iLine < lines.size(); iLine++) {
         std::smatch match;
         std::regex_search(lines.at(iLine), match, reg);
         if (!match.empty()) {
           _lheWeightNames.push_back(match[1]);
         }
       }
     }
   }
 }

void RivetAnalyzer::endJob ( void )

overridevirtual

List of registered analysis data objects.

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 191 of file RivetAnalyzer.cc.

191 {}

void RivetAnalyzer::endLuminosityBlock	(	const edm::LuminosityBlock &	iLumi,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 113 of file RivetAnalyzer.cc.

113 { return; }

void RivetAnalyzer::endRun	(	const edm::Run &	iRun,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 172 of file RivetAnalyzer.cc.

References _analysisHandler, _doFinalize, and _outFileName.

                                                                           {
   if (_doFinalize)
     _analysisHandler.finalize();
   else {
     //if we don't finalize we just want to do the transformation from histograms to DPS
     //normalizeTree();
   }
   _analysisHandler.writeData(_outFileName);
 
   return;
 }

void RivetAnalyzer::normalizeTree ( )

private

Definition at line 193 of file RivetAnalyzer.cc.

References _analysisHandler, _mes, _produceDQM, hltExoticaValidator_cfi::analyses, dqm::dqmstoreimpl::DQMStore::book1D(), dbe, DEFINE_FWK_MODULE, nevent, and dqm::dqmstoreimpl::DQMStore::setCurrentFolder().

                                   {
   using namespace YODA;
   std::vector<string> analyses = _analysisHandler.analysisNames();
 
   //tree.ls(".", true);
   const string tmpdir = "/RivetNormalizeTmp";
   //tree.mkdir(tmpdir);
   for (const string& analysis : analyses) {
     if (_produceDQM) {
       dbe->setCurrentFolder("Rivet/" + analysis);
       //global variables that are always present
       //sumOfWeights
       TH1F nevent("nEvt", "n analyzed Events", 1, 0., 1.);
       nevent.SetBinContent(1, _analysisHandler.sumW());
       _mes.push_back(dbe->book1D("nEvt", &nevent));
     }
     //cross section
     //TH1F xsection("xSection", "Cross Section", 1, 0., 1.);
     //xsection.SetBinContent(1,_analysisHandler.crossSection());
     //_mes.push_back(dbe->book1D("xSection",&xsection));
     //now loop over the histograms
 
     /*
     const vector<string> paths = tree.listObjectNames("/"+analysis, true); // args set recursive listing
     std::cout << "Number of objects in YODA tree for analysis " << analysis << " = " << paths.size() << std::endl;
     foreach (const string& path, paths) {
       IManagedObject* hobj = tree.find(path);
       if (hobj) {
         // Weird seg fault on SLC4 when trying to dyn cast an IProfile ptr to a IHistogram
         // Fix by attempting to cast to IProfile first, only try IHistogram if it fails.
         IHistogram1D* histo = 0;
         IProfile1D* prof = dynamic_cast<IProfile1D*>(hobj);
         if (!prof) histo = dynamic_cast<IHistogram1D*>(hobj);
 
         std::cout << "Converting histo " << path << " to DPS" << std::endl;
         tree.mv(path, tmpdir);
         const size_t lastslash = path.find_last_of("/");
         const string basename = path.substr(lastslash+1, path.length() - (lastslash+1));
         const string tmppath = tmpdir + "/" + basename;
 
         // If it's a normal histo:
         if (histo) {
           IHistogram1D* tmphisto = dynamic_cast<IHistogram1D*>(tree.find(tmppath));
           if (tmphisto) {
             _analysisHandler.datapointsetFactory().create(path, *tmphisto);
           }
           //now convert to root and then ME
     //need aida2flat (from Rivet 1.X) & flat2root here
           TH1F* h = aida2root<IHistogram1D, TH1F>(histo, basename);
           if (_produceDQM)
             _mes.push_back(dbe->book1D(h->GetName(), h));
           delete h;
           tree.rm(tmppath);
         }
         // If it's a profile histo:
         else if (prof) {
           IProfile1D* tmpprof = dynamic_cast<IProfile1D*>(tree.find(tmppath));
           if (tmpprof) {
             _analysisHandler.datapointsetFactory().create(path, *tmpprof);
           }
           //now convert to root and then ME
     //need aida2flat (from Rivet 1.X) & flat2root here
           TProfile* p = aida2root<IProfile1D, TProfile>(prof, basename);
           if (_produceDQM)
             _mes.push_back(dbe->bookProfile(p->GetName(), p));
           delete p;
           tree.rm(tmppath);
         }
       }
     }
     */
   }
   //tree.rmdir(tmpdir);
 }

Member Data Documentation

Rivet::AnalysisHandler RivetAnalyzer::_analysisHandler

private

Definition at line 59 of file RivetAnalyzer.h.

Referenced by analyze(), endRun(), normalizeTree(), and RivetAnalyzer().

bool RivetAnalyzer::_doFinalize

private

Definition at line 62 of file RivetAnalyzer.h.

Referenced by endRun().

edm::EDGetTokenT<GenEventInfoProduct> RivetAnalyzer::_genEventInfoCollection

private

Definition at line 56 of file RivetAnalyzer.h.

edm::EDGetTokenT<GenLumiInfoHeader> RivetAnalyzer::_genLumiInfoToken

private

Definition at line 57 of file RivetAnalyzer.h.

Referenced by beginLuminosityBlock(), and RivetAnalyzer().

int RivetAnalyzer::_GENweightNumber

private

Definition at line 54 of file RivetAnalyzer.h.

edm::EDGetTokenT<edm::HepMCProduct> RivetAnalyzer::_hepmcCollection

private

Definition at line 49 of file RivetAnalyzer.h.

Referenced by analyze(), and RivetAnalyzer().

bool RivetAnalyzer::_isFirstEvent

private

Definition at line 60 of file RivetAnalyzer.h.

Referenced by analyze().

edm::EDGetTokenT<LHEEventProduct> RivetAnalyzer::_LHECollection

private

Definition at line 55 of file RivetAnalyzer.h.

Referenced by analyze(), and RivetAnalyzer().

const edm::InputTag RivetAnalyzer::_lheLabel

private

Definition at line 64 of file RivetAnalyzer.h.

Referenced by beginRun(), and RivetAnalyzer().

edm::EDGetTokenT<LHERunInfoProduct> RivetAnalyzer::_lheRunInfoToken

private

Definition at line 58 of file RivetAnalyzer.h.

Referenced by RivetAnalyzer().

std::vector<std::string> RivetAnalyzer::_lheWeightNames

private

Definition at line 67 of file RivetAnalyzer.h.

Referenced by analyze(), and beginRun().

int RivetAnalyzer::_LHEweightNumber

private

Definition at line 52 of file RivetAnalyzer.h.

std::vector<MonitorElement *> RivetAnalyzer::_mes

private

Definition at line 70 of file RivetAnalyzer.h.

Referenced by normalizeTree().

std::string RivetAnalyzer::_outFileName

private

Definition at line 61 of file RivetAnalyzer.h.

Referenced by endRun().

bool RivetAnalyzer::_produceDQM

private

Definition at line 63 of file RivetAnalyzer.h.

Referenced by normalizeTree(), and RivetAnalyzer().

bool RivetAnalyzer::_useExternalWeight

private

Definition at line 50 of file RivetAnalyzer.h.

bool RivetAnalyzer::_useGENweights

private

Definition at line 53 of file RivetAnalyzer.h.

bool RivetAnalyzer::_useLHEweights

private

Definition at line 51 of file RivetAnalyzer.h.

Referenced by analyze(), beginRun(), and RivetAnalyzer().

std::vector<std::string> RivetAnalyzer::_weightNames

private

Definition at line 66 of file RivetAnalyzer.h.

Referenced by analyze(), and beginLuminosityBlock().

double RivetAnalyzer::_xsection

private

Definition at line 65 of file RivetAnalyzer.h.

Referenced by analyze(), and RivetAnalyzer().

DQMStore* RivetAnalyzer::dbe

private

Definition at line 69 of file RivetAnalyzer.h.

Referenced by normalizeTree(), and RivetAnalyzer().

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation