#include <RivetAnalyzer.h>

Inheritance diagram for RivetAnalyzer:

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

void	beginJob () override

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

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

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::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

int	_GENweightNumber

edm::EDGetTokenT< edm::HepMCProduct >	_hepmcCollection

bool	_isFirstEvent

edm::EDGetTokenT< LHEEventProduct >	_LHECollection

int	_LHEweightNumber

std::vector< MonitorElement * >	_mes

std::string	_outFileName

bool	_produceDQM

bool	_useExternalWeight

bool	_useGENweights

bool	_useLHEweights

double	_xsection

DQMStore *	dbe

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

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 21 of file RivetAnalyzer.h.

Constructor & Destructor Documentation

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

Definition at line 15 of file RivetAnalyzer.cc.

References _analysisHandler, _genEventInfoCollection, _GENweightNumber, _hepmcCollection, _LHECollection, _LHEweightNumber, _produceDQM, _useExternalWeight, _useGENweights, _useLHEweights, _xsection, dbe, Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), and Utilities::operator.

     : _analysisHandler(),
       _isFirstEvent(true),
       _outFileName(pset.getParameter<std::string>("OutputFile")),
       //decide whether to finlaize tthe 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")),
       _xsection(-1.) {
   usesResource("Rivet");
 
   //retrive the analysis name from paarmeter set
   std::vector<std::string> analysisNames = pset.getParameter<std::vector<std::string> >("AnalysisNames");
 
   _hepmcCollection = consumes<HepMCProduct>(pset.getParameter<edm::InputTag>("HepMCCollection"));
 
   _useExternalWeight = pset.getParameter<bool>("UseExternalWeight");
   if (_useExternalWeight) {
     if (!pset.exists("GenEventInfoCollection")) {
       throw cms::Exception("RivetAnalyzer") << "when using an external event weight you have to specify the "
                                                "GenEventInfoProduct collection from which the weight has to be taken ";
     }
 
     _genEventInfoCollection = consumes<GenEventInfoProduct>(pset.getParameter<edm::InputTag>("GenEventInfoCollection"));
     _useGENweights = pset.getParameter<bool>("useGENweights");
     _GENweightNumber = pset.getParameter<int>("GENweightNumber");
     _LHECollection = consumes<LHEEventProduct>(pset.getParameter<edm::InputTag>("LHECollection"));
     _useLHEweights = pset.getParameter<bool>("useLHEweights");
     _LHEweightNumber = pset.getParameter<int>("LHEweightNumber");
   }
 
   //get the analyses
   _analysisHandler.addAnalyses(analysisNames);
 
   //go through the analyses and check those that need the cross section
   const std::set<AnaHandle, CmpAnaHandle>& analyses = _analysisHandler.analyses();
 
   std::set<AnaHandle, CmpAnaHandle>::const_iterator ibeg = analyses.begin();
   std::set<AnaHandle, CmpAnaHandle>::const_iterator iend = analyses.end();
   std::set<AnaHandle, CmpAnaHandle>::const_iterator iana;
   _xsection = pset.getParameter<double>("CrossSection");
   for (iana = ibeg; iana != iend; ++iana) {
     if ((*iana)->needsCrossSection())
       (*iana)->setCrossSection(_xsection);
   }
   if (_produceDQM) {
     // book stuff needed for DQM
     dbe = nullptr;
     dbe = edm::Service<DQMStore>().operator->();
     dbe->setVerbose(50);
   }
 }

RivetAnalyzer::~RivetAnalyzer ( )

override

Definition at line 68 of file RivetAnalyzer.cc.

68 {}

Member Function Documentation

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

override

Definition at line 94 of file RivetAnalyzer.cc.

References _analysisHandler, _genEventInfoCollection, _GENweightNumber, _hepmcCollection, _isFirstEvent, _LHECollection, _LHEweightNumber, _useExternalWeight, _useGENweights, _useLHEweights, _xsection, Exception, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), LHEEventProduct::originalXWGTUP(), GenEventInfoProduct::weight(), GenEventInfoProduct::weights(), 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 (_useExternalWeight || _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 (_useExternalWeight) {
       if (tmpGenEvtPtr->weights().empty()) {
         throw cms::Exception("RivetAnalyzer") << "Original weight container has 0 size ";
       }
       if (tmpGenEvtPtr->weights().size() > 1) {
         edm::LogWarning("RivetAnalyzer") << "Original event weight size is " << tmpGenEvtPtr->weights().size()
                                          << ". Will change only the first one ";
       }
 
       edm::Handle<GenEventInfoProduct> genEventInfoProduct;
       iEvent.getByToken(_genEventInfoCollection, genEventInfoProduct);
       double weightForRivet = genEventInfoProduct->weight();
 
       if (_useGENweights) {
         weightForRivet = genEventInfoProduct->weights().at(_GENweightNumber);
       }
       if (_useLHEweights) {
         edm::Handle<LHEEventProduct> lheEventHandle;
         iEvent.getByToken(_LHECollection, lheEventHandle);
         weightForRivet *= lheEventHandle->weights().at(_LHEweightNumber).wgt / lheEventHandle->originalXWGTUP();
       }
 
       tmpGenEvtPtr->weights()[0] = weightForRivet;
     }
     myGenEvent = tmpGenEvtPtr.get();
   }
 
   //aaply the beams initialization on the first event
   if (_isFirstEvent) {
     _analysisHandler.init(*myGenEvent);
     _isFirstEvent = false;
   }
 
   //run the analysis
   _analysisHandler.analyze(*myGenEvent);
 }

void RivetAnalyzer::beginJob ( void )

overridevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 70 of file RivetAnalyzer.cc.

References AlCaHLTBitMon_QueryRunRegistry::string.

                              {
   //set the environment, very ugly but rivet is monolithic when it comes to paths
   char* cmsswbase = getenv("CMSSW_BASE");
   char* cmsswrelease = getenv("CMSSW_RELEASE_BASE");
   if (!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 (!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::beginRun	(	const edm::Run &	iRun,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 92 of file RivetAnalyzer.cc.

92 { return; }

void RivetAnalyzer::endJob ( void )

overridevirtual

List of registered analysis data objects.

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 168 of file RivetAnalyzer.cc.

168 {}

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

override

Definition at line 149 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 170 of file RivetAnalyzer.cc.

References _analysisHandler, _mes, _produceDQM, dbe, DEFINE_FWK_MODULE, and nevent.

                                   {
   using namespace YODA;
   std::vector<string> analyses = _analysisHandler.analysisNames();
 
   //tree.ls(".", true);
   const string tmpdir = "/RivetNormalizeTmp";
   //tree.mkdir(tmpdir);
   foreach (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.sumOfWeights());
       _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);
 }