Page MenuHomeHEPForge
Files F9514795

Run.cc
No OneTemporary
Actions

Run.cc
View Options

// -*- C++ -*-
#include "Rivet/Run.hh"
#include "Rivet/AnalysisHandler.hh"
#include "HepMC/IO_GenEvent.h"
#include "Rivet/Math/MathUtils.hh"
#include <limits>
namespace Rivet {
Run::Run(AnalysisHandler& ah)
: _ah(ah), _fileweight(1.0), _xs(NAN)
{ }
Run::~Run() { }
Run& Run::setCrossSection(const double xs) {
_xs = xs;
return *this;
}
double Run::crossSection() const {
return _ah.crossSection();
}
Run& Run::setListAnalyses(const bool dolist) {
_listAnalyses = dolist;
return *this;
}
// Fill event and check for a bad read state
bool Run::readEvent() {
/// @todo Clear rather than new the GenEvent object per-event?
_evt.reset(new GenEvent());
if (_io->rdstate() != 0 || !_io->fill_next_event(_evt.get()) ) {
Log::getLog("Rivet.Run") << Log::DEBUG << "Read failed. End of file?" << endl;
return false;
}
// Rescale event weights by file-level weight, if scaling is non-trivial
if (!fuzzyEquals(_fileweight, 1.0)) {
for (size_t i = 0; i < (size_t) _evt->weights().size(); ++i) {
_evt->weights()[i] *= _fileweight;
}
}
return true;
}
bool Run::openFile(const std::string& evtfile, double weight) {
// Set current weight-scaling member
_fileweight = weight;
// Set up HepMC input reader objects
if (evtfile == "-") {
_io.reset(new HepMC::IO_GenEvent(std::cin));
} else {
// Ignore the HepMC::IO_GenEvent(filename, ios) constructor, since it's only available from HepMC 2.4
_istr.reset(new std::fstream(evtfile.c_str(), std::ios::in));
_io.reset(new HepMC::IO_GenEvent(*_istr));
}
if (_io->rdstate() != 0) {
Log::getLog("Rivet.Run") << Log::ERROR << "Read error on file " << evtfile << endl;
return false;
}
return true;
}
bool Run::init(const std::string& evtfile, double weight) {
if (!openFile(evtfile, weight)) return false;
// Read first event to define run conditions
bool ok = readEvent();
if (!ok) return false;
if (_evt->particles_size() == 0) {
Log::getLog("Rivet.Run") << Log::ERROR << "Empty first event." << endl;
return false;
}
// Initialise AnalysisHandler with beam information from first event
_ah.init(*_evt);
// Set cross-section from command line
if (_xs >= 0.0) {
Log::getLog("Rivet.Run")
<< Log::DEBUG << "Setting user cross-section = " << _xs << " pb" << endl;
_ah.setCrossSection(_xs);
}
// List the chosen & compatible analyses if requested
if (_listAnalyses) {
foreach (const std::string& ana, _ah.analysisNames()) {
cout << ana << endl;
}
}
return true;
}
bool Run::processEvent() {
// Set cross-section if found in event and not from command line
#ifdef HEPMC_HAS_CROSS_SECTION
if (std::isnan(_xs) && _evt->cross_section()) {
const double xs = _evt->cross_section()->cross_section(); //< in pb
Log::getLog("Rivet.Run")
<< Log::DEBUG << "Setting cross-section = " << xs << " pb" << endl;
_ah.setCrossSection(xs);
}
#endif
// Complain about absence of cross-section if required!
if (_ah.needCrossSection() && !_ah.hasCrossSection()) {
Log::getLog("Rivet.Run")
<< Log::ERROR
<< "Total cross-section needed for at least one of the analyses. "
<< "Please set it (on the command line with '-x' if using the 'rivet' program)" << endl;
return false;
}
// Analyze event
_ah.analyze(*_evt);
return true;
}
bool Run::finalize() {
_evt.reset();
_istr.reset();
_io.reset();
return true;
}
}

File Metadata

Mime Type
text/x-c
Expires
Mon, Feb 24, 6:45 AM (1 d, 7 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
4494245
Default Alt Text
Run.cc (3 KB)

Event Timeline