2 #ifndef RIVET_RivetHandler_HH 3 #define RIVET_RivetHandler_HH 5 #include "Rivet/Config/RivetCommon.hh" 6 #include "Rivet/Particle.hh" 7 #include "Rivet/AnalysisLoader.hh" 8 #include "Rivet/Tools/RivetYODA.hh" 15 typedef std::shared_ptr<Analysis> AnaHandle;
61 double sumW()
const {
return _eventCounter->sumW(); }
63 double sumW2()
const {
return _eventCounter->sumW2(); }
66 const vector<string>&
weightNames()
const {
return _weightNames; }
69 size_t numWeights()
const {
return _weightNames.size(); }
81 void setWeightCap(
const double maxWeight) { _weightCap = maxWeight; }
93 void setCrossSection(pair<double, double> xsec,
bool isUserSupplied=
false);
101 _xs.get()->setActiveWeightIdx(_defaultWeightIdx);
102 const YODA::Scatter1D::Points& ps = _xs->points();
103 if (ps.size() != 1) {
104 string errMsg =
"cross section missing when requesting nominal cross section";
107 double xs = ps[0].x();
108 _xs.get()->unsetActiveWeight();
121 MSG_DEBUG(
"Setting run beams = " << beams <<
" @ " <<
sqrtS()/GeV <<
" GeV");
126 const ParticlePair&
beams()
const {
return _beams; }
134 double sqrtS()
const;
158 std::vector<AnaHandle> rtn;
159 rtn.reserve(_analyses.size());
160 for (
const auto& apair : _analyses) rtn.push_back(apair.second);
165 AnaHandle
analysis(
const std::string& analysisname) {
166 if ( _analyses.find(analysisname) == _analyses.end() )
167 throw LookupError(
"No analysis named '" + analysisname +
"' registered in AnalysisHandler");
169 return _analyses[analysisname];
171 throw LookupError(
"No analysis named '" + analysisname +
"' registered in AnalysisHandler");
212 void init(
const GenEvent& event);
218 void analyze(
const GenEvent& event);
224 void analyze(
const GenEvent* event);
241 void readData(
const std::string& filename);
248 const YODA::AnalysisObjectPtr
getPreload(
string path)
const {
249 auto it = _preloads.find(path);
250 if ( it == _preloads.end() )
return nullptr;
255 void writeData(
const std::string& filename)
const;
260 void dump(
string dumpfile,
int period) {
261 _dumpPeriod = period;
262 _dumpFile = dumpfile;
279 void mergeYodas(
const vector<string> & aofiles,
280 const vector<string> & delopts = vector<string>(),
285 const vector<string> & delopts)
const;
292 enum class Stage { OTHER, INIT, FINALIZE };
301 Stage _stage = Stage::OTHER;
304 std::map<std::string, AnaHandle> _analyses;
308 map<string,YODA::AnalysisObjectPtr> _preloads;
312 vector<YODA::AnalysisObjectPtr> _finalizedAOs;
319 std::vector<std::string> _weightNames;
320 std::vector<std::valarray<double> > _subEventWeights;
324 std::string _runname;
327 mutable CounterPtr _eventCounter;
333 std::pair<double,double> _userxs;
354 size_t _defaultWeightIdx;
Definition: MC_Cent_pPb.hh:10
void setCrossSection(pair< double, double > xsec, bool isUserSupplied=false)
Set the cross-section for the process being generated.
void mergeYodas(const vector< string > &aofiles, const vector< string > &delopts=vector< string >(), bool equiv=false)
const ParticlePair & beams() const
Get the beam particles for this run, usually determined from the first event.
Definition: AnalysisHandler.hh:126
double nominalCrossSection() const
Get the nominal cross-section.
Definition: AnalysisHandler.hh:100
AnalysisHandler & addAnalysis(Analysis *analysis)
Add an analysis to the run list by object.
bool haveNamedWeights() const
Are any of the weights non-numeric?
void analyze(const GenEvent &event)
Analyze the given event by reference.
void init(const GenEvent &event)
Initialize a run, with the run beams taken from the example event.
double sumW2() const
Access to the sum of squared-weights.
Definition: AnalysisHandler.hh:63
void setWeightCap(const double maxWeight)
Set the weight cap.
Definition: AnalysisHandler.hh:81
void skipMultiWeights(bool ignore=false)
Setter for _skipWeights.
std::vector< AnaHandle > analyses() const
Get the collection of currently registered analyses.
Definition: AnalysisHandler.hh:157
std::vector< std::string > analysisNames() const
Get a list of the currently registered analyses' names.
void setCrossSection(double xsec, double xsecerr, bool isUserSupplied=false)
Set the cross-section for the process being generated (alternative signature)
Definition: AnalysisHandler.hh:95
Scatter1DPtr crossSection() const
Get the cross-section known to the handler.
Definition: AnalysisHandler.hh:90
PdgIdPair beamIds() const
AnalysisHandler(const string &runname="")
Preferred constructor, with optional run name.
Stage
Definition: AnalysisHandler.hh:292
void readData(const std::string &filename)
Read analysis plots into the histo collection (via addData) from the named file.
Logging system for controlled & formatted writing to stdout.
Definition: Logging.hh:10
void setWeightNames(const GenEvent &ge)
Set the weight names from a GenEvent.
AnalysisHandler & removeAnalyses(const std::vector< std::string > &analysisnames)
Remove analyses from the run list using their names.
~AnalysisHandler()
Destructor The destructor is not virtual, as this class should not be inherited from.
Generic runtime Rivet error.
Definition: Exceptions.hh:12
AnalysisHandler & addAnalyses(const std::vector< std::string > &analysisnames)
Add analyses to the run list using their names.
This is the base class of all analysis classes in Rivet.
Definition: Analysis.hh:64
size_t numWeights() const
Are any of the weights non-numeric?
Definition: AnalysisHandler.hh:69
AnalysisHandler & removeAnalysis(const std::string &analysisname)
Remove an analysis from the run list using its name.
const vector< string > & weightNames() const
Names of event weight categories.
Definition: AnalysisHandler.hh:66
Stage stage() const
Which stage are we in?
Definition: AnalysisHandler.hh:295
Error relating to looking up analysis objects in the register.
Definition: Exceptions.hh:61
The key class for coordination of Analysis objects and the event loop.
Definition: AnalysisHandler.hh:23
void dump(string dumpfile, int period)
Definition: AnalysisHandler.hh:260
const std::map< std::string, AnaHandle > & analysesMap() const
Get the collection of currently registered analyses.
Definition: AnalysisHandler.hh:152
double sumW() const
Access the sum of the event weights seen.
Definition: AnalysisHandler.hh:61
const YODA::AnalysisObjectPtr getPreload(string path) const
Definition: AnalysisHandler.hh:248
string runName() const
Get the name of this run.
void stripOptions(YODA::AnalysisObjectPtr ao, const vector< string > &delopts) const
Helper function to strip specific options from data object paths.
void writeData(const std::string &filename) const
Write all analyses' plots (via getData) to the named file.
vector< MultiweightAOPtr > getRivetAOs() const
Get all multi-weight Rivet analysis object wrappers.
size_t defaultWeightIndex() const
Get the index of the nominal weight-stream.
Definition: AnalysisHandler.hh:78
void setIgnoreBeams(bool ignore=true)
Setter for _ignoreBeams.
AnaHandle analysis(const std::string &analysisname)
Get a registered analysis by name.
Definition: AnalysisHandler.hh:165
AnalysisHandler & setRunBeams(const ParticlePair &beams)
Set the beam particles for this run.
Definition: AnalysisHandler.hh:119