#include <rivetoutput.h>

SMASH output to Rivet analyses.

This class passes HepMC events to Rivet for analysis. The event is kept in-memory which significanly speeds up processing (at last a factor 10) as compared to writing the HepMC event to disk or pipe and then decoding in Rivet.

More details of the output format can be found in the User Guide.

Definition at line 36 of file rivetoutput.h.

Inheritance diagram for smash::RivetOutput:

[legend]

Collaboration diagram for smash::RivetOutput:

[legend]

Classes
struct	Proxy
	A proxy object that wraps all Rivet::AnalysisHandler calls in an environment where FP errors are disabled. More...

Public Member Functions
	RivetOutput (const bf::path &path, std::string name, const bool full_event, const OutputParameters &out_par)
	Create Rivet output. More...

	~RivetOutput ()
	Destructor. More...

void	at_eventend (const Particles &particles, const int32_t event_number, const EventInfo &event) override
	Add the final particles information of an event to the central vertex. More...

void	add_analysis (const std::string &name)
	Add an analysis or analyses to Rivet. More...

void	add_path (const std::string &path)
	Add a load path to the Rivet handler. More...

void	add_preload (const std::string &file)
	Add preload to Rivet handler. More...

void	set_ignore_beams (bool ignore=true)
	Do not insist on appropriate beams for analyses. More...

void	set_log_level (const std::string &name, const std::string &level)
	Set log level in Rivet. More...

void	set_cross_section (double xs, double xserr)
	Set X-section. More...

void	setup ()
	Read configuration of Rivet from SMASH configuration. More...

Public Member Functions inherited from smash::HepMcInterface
	HepMcInterface (const std::string &name, const bool full_event)
	Create HepMC particle event in memory. More...

void	at_eventstart (const Particles &particles, const int event_number, const EventInfo &event) override
	Add the initial particles information of an event to the central vertex. More...

void	at_interaction (const Action &action, const double density) override
	Writes collisions to event. More...

void	at_eventend (const Particles &particles, const int32_t event_number, const EventInfo &event) override
	Add the final particles information of an event to the central vertex. More...

Public Member Functions inherited from smash::OutputInterface
	OutputInterface (std::string name)
	Construct output interface. More...

virtual	~OutputInterface ()=default

virtual void	at_eventstart (const std::vector< Particles > &ensembles, int event_number)
	Output launched at event start after initialization, when particles are generated but not yet propagated. More...

virtual void	at_eventstart (const int event_number, const ThermodynamicQuantity tq, const DensityType dens_type, RectangularLattice< DensityOnLattice > lattice)
	Output launched at event start after initialization, when particles are generated but not yet propagated. More...

virtual void	at_eventstart (const int event_number, const ThermodynamicQuantity tq, const DensityType dens_type, RectangularLattice< EnergyMomentumTensor > lattice)
	Output launched atevent start after initialization, when particles are generated but not yet propagated. More...

virtual void	at_eventend (const int event_number, const ThermodynamicQuantity tq, const DensityType dens_type)
	Output launched at event end. More...

virtual void	at_eventend (const ThermodynamicQuantity tq)
	Output launched at event end. More...

virtual void	at_eventend (const Particles &particles, const int event_number, const EventInfo &info)
	Output launched at event end. More...

virtual void	at_eventend (const std::vector< Particles > &ensembles, const int event_number)
	Output launched at event end. More...

virtual void	at_intermediate_time (const Particles &particles, const std::unique_ptr< Clock > &clock, const DensityParameters &dens_param, const EventInfo &info)
	Output launched after every N'th timestep. More...

virtual void	at_intermediate_time (const std::vector< Particles > &ensembles, const std::unique_ptr< Clock > &clock, const DensityParameters &dens_param)
	Output launched after every N'th timestep. More...

virtual void	thermodynamics_output (const ThermodynamicQuantity tq, const DensityType dt, RectangularLattice< DensityOnLattice > &lattice)
	Output to write thermodynamics from the lattice. More...

virtual void	thermodynamics_output (const ThermodynamicQuantity tq, const DensityType dt, RectangularLattice< EnergyMomentumTensor > &lattice)
	Output to write energy-momentum tensor and related quantities from the lattice. More...

virtual void	thermodynamics_lattice_output (RectangularLattice< DensityOnLattice > &lattice, const double current_time)
	Output to write thermodynamics from the lattice. More...

virtual void	thermodynamics_lattice_output (RectangularLattice< DensityOnLattice > &lattice, const double current_time, const std::vector< Particles > &ensembles, const DensityParameters &dens_param)
	Output to write thermodynamics from the lattice. More...

virtual void	thermodynamics_lattice_output (const ThermodynamicQuantity tq, RectangularLattice< EnergyMomentumTensor > &lattice, const double current_time)
	Output to write energy-momentum tensor and related quantities from the lattice. More...

virtual void	thermodynamics_output (const GrandCanThermalizer &gct)
	Output to write energy-momentum tensor and related quantities from the thermalizer class. More...

virtual void	fields_output (const std::string name1, const std::string name2, RectangularLattice< std::pair< ThreeVector, ThreeVector >> &lat)
	Write fields in vtk output. More...

bool	is_dilepton_output () const
	Get, whether this is the dilepton output? More...

bool	is_photon_output () const
	Get, whether this is the photon output? More...

bool	is_IC_output () const
	Get, whether this is the IC output? More...

const char *	to_string (const ThermodynamicQuantity tq)
	Convert thermodynamic quantities to strings. More...

const char *	to_string (const DensityType dens_type)
	Convert density types to strings. More...

Private Member Functions
Proxy	analysis_handler_proxy ()
	Return a proxy that temporarily disables FP exceptions. More...

Private Attributes
std::shared_ptr< Rivet::AnalysisHandler >	handler_
	Rivet analysis handler. More...

bf::path	filename_
	Output file. More...

bool	need_init_
	Whether we need initialisation. More...

Configuration	rivet_confs_
	Configutations for rivet. More...

Additional Inherited Members
Public Types inherited from smash::HepMcInterface
using	AZ = std::pair< int, int >
	Pair of Atomic weight and number. More...

Protected Types inherited from smash::HepMcInterface
enum	Status { beam = 4 , fnal = 1 , dcy = 2 , off = 100 }
	HepMC status codes. More...

using	IdMap = std::map< int, HepMC3::GenParticlePtr >
	Type of mapping from SMASH ID to HepMC ID. More...

using	CollCounter = std::valarray< int >
	Counter of collitions per incoming particle. More...

Protected Member Functions inherited from smash::HepMcInterface
void	clear ()
	Clear before an event. More...

int	get_status (const ProcessType &t) const
	Convert SMASH process type to HepMC status. More...

HepMC3::GenParticlePtr	make_gen (int pid, int status, const smash::FourVector &mom, double mass=-1)
	Make an HepMC particle. More...

HepMC3::GenParticlePtr	make_register (const ParticleData &p, int status=Status::fnal)
	Find particle in mapping or generate it. More...

HepMC3::GenParticlePtr	find_or_make (const ParticleData &p, int status=Status::fnal, bool force_new=false)
	Find particle in mapping or generate it. More...

int	ion_pdg (const AZ &az) const
	Encode ion PDG. More...

Protected Attributes inherited from smash::HepMcInterface
HepMC3::GenEvent	event_
	The event. More...

HepMC3::GenHeavyIonPtr	ion_
	The heavy-ion structure. More...

HepMC3::GenCrossSectionPtr	xs_
	Dummy cross-section. More...

HepMC3::GenVertexPtr	ip_
	The interaction point. More...

IdMap	map_
	Mapping from ID to particle. More...

CollCounter	coll_
	Collision counter. More...

int	ncoll_
	counter of binary collisions (e.g., where both incoming particles are from the beams. More...

int	ncoll_hard_
	counter of hard binary collisions (e.g., where both incoming particles are from the beams. More...

bool	full_event_
	Whether the full event or only final-state particles are in the output. More...

Protected Attributes inherited from smash::OutputInterface
const bool	is_dilepton_output_
	Is this the dilepton output? More...

const bool	is_photon_output_
	Is this the photon output? More...

const bool	is_IC_output_
	Is this the IC output? More...

Constructor & Destructor Documentation

◆ RivetOutput()

smash::RivetOutput::RivetOutput	(	const bf::path &	path,
		std::string	name,
		const bool	full_event,
		const OutputParameters &	out_par
	)

Create Rivet output.

Parameters

[in]	path	Output path.
[in]	name	Name of the output.
[in]	full_event	Whether the full event or only final-state particles are printed in the output
[in]	out_par	Configuration parameters of SMASH

Definition at line 146 of file rivetoutput.cc.

     : HepMcInterface(name, full_event),
       handler_(),
       filename_(path / (name + ".yoda")),
       need_init_(true),
       rivet_confs_(out_par.subcon_for_rivet) {
   handler_ = std::make_shared<Rivet::AnalysisHandler>();
   setup();
 }

Here is the call graph for this function:

◆ ~RivetOutput()

smash::RivetOutput::~RivetOutput ( )

Destructor.

Finalises the analzyses and writes out results to file

Definition at line 157 of file rivetoutput.cc.

                           {
   logg[LOutput].debug() << "Writing Rivet results to " << filename_
                         << std::endl;
   analysis_handler_proxy()->finalize();
   analysis_handler_proxy()->writeData(filename_.string());
 }

Here is the call graph for this function:

Member Function Documentation

◆ at_eventend()

void smash::RivetOutput::at_eventend	(	const Particles &	particles,
		const int32_t	event_number,
		const EventInfo &	event
	)

override

Add the final particles information of an event to the central vertex.

Store impact paramter and write event.

Parameters

[in]	particles	Current list of particles.
[in]	event_number	Number of event.
[in]	event	Event info, see event_info

Definition at line 164 of file rivetoutput.cc.

                                                       {
   HepMcInterface::at_eventend(particles, event_number, event);
  
   // Initialize Rivet on first event
   if (need_init_) {
     logg[LOutput].debug() << "Initialising Rivet" << std::endl;
     need_init_ = false;
     analysis_handler_proxy()->init(event_);
   }
  
   logg[LOutput].debug() << "Analysing event " << event_number << std::endl;
   // Let Rivet analyse the event
   analysis_handler_proxy()->analyze(event_);
 }

Here is the call graph for this function:

◆ add_analysis()

void smash::RivetOutput::add_analysis ( const std::string & name )

Add an analysis or analyses to Rivet.

Parameters

[in] name Name of analysis

Definition at line 181 of file rivetoutput.cc.

                                                     {
   analysis_handler_proxy()->addAnalysis(name);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

◆ add_path()

void smash::RivetOutput::add_path ( const std::string & path )

Add a load path to the Rivet handler.

Definition at line 185 of file rivetoutput.cc.

                                                 {
   Rivet::addAnalysisLibPath(path);
   Rivet::addAnalysisDataPath(path);
 }

Here is the caller graph for this function:

◆ add_preload()

void smash::RivetOutput::add_preload ( const std::string & file )

Add preload to Rivet handler.

Definition at line 190 of file rivetoutput.cc.

                                                    {
   analysis_handler_proxy()->readData(file);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

◆ set_ignore_beams()

void smash::RivetOutput::set_ignore_beams ( bool ignore = true )

Do not insist on appropriate beams for analyses.

Definition at line 194 of file rivetoutput.cc.

                                               {
   logg[LOutput].info() << "Ignore beams? " << (ignore ? "yes" : "no")
                        << std::endl;
   analysis_handler_proxy()->setIgnoreBeams(ignore);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

◆ set_log_level()

void smash::RivetOutput::set_log_level	(	const std::string &	name,
		const std::string &	level
	)

Set log level in Rivet.

Definition at line 200 of file rivetoutput.cc.

                                                         {
   std::string fname(name);
   if (fname.rfind("Rivet", 0) != 0) {
     fname = "Rivet." + fname;
   }
  
   auto upcase = [](const std::string& s) {
     std::string out(s);
     std::transform(out.begin(), out.end(), out.begin(),
                    [](char c) { return std::toupper(c); });
     return out;
   };
  
   try {
     Rivet::Log::setLevel(fname, Rivet::Log::getLevelFromName(upcase(level)));
   } catch (...) {
   }
 }

Here is the caller graph for this function:

◆ set_cross_section()

void smash::RivetOutput::set_cross_section	(	double	xs,
		double	xserr
	)

Set X-section.

Parameters

[in]	xs	Cross-section value in pb
[in]	xserr	Uncertainty on cross-section in pb

Definition at line 220 of file rivetoutput.cc.

                                                            {
   analysis_handler_proxy()->setCrossSection(xs, xserr, true);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

◆ setup()

void smash::RivetOutput::setup ( )

Read configuration of Rivet from SMASH configuration.

Definition at line 224 of file rivetoutput.cc.

                         {
   logg[LOutput].debug() << "Setting up from configuration:\n"
                         << rivet_confs_.to_string() << std::endl;
  
   // Paths to analyses libraries and data
   if (rivet_confs_.has_value({"Paths"})) {
     logg[LOutput].info() << "Processing paths" << std::endl;
     std::vector<std::string> path = rivet_confs_.take({"Paths"});
     for (auto p : path)
       add_path(p);
   }
  
   // Data files to pre-load e.g., for centrality configurations
   if (rivet_confs_.has_value({"Preloads"})) {
     logg[LOutput].info() << "Processing preloads" << std::endl;
     std::vector<std::string> prel = rivet_confs_.take({"Preloads"});
     for (auto p : prel)
       add_preload(p);
   }
  
   // Analyses (including options) to add to run
   if (rivet_confs_.has_value({"Analyses"})) {
     logg[LOutput].info() << "Processing analyses" << std::endl;
     std::vector<std::string> anas = rivet_confs_.take({"Analyses"});
     for (auto p : anas)
       add_analysis(p);
   }
  
   // Whether Rivet should ignore beams
   if (rivet_confs_.has_value({"Ignore_Beams"})) {
     set_ignore_beams(rivet_confs_.take({"Ignore_Beams"}));
   } else {
     // we must explicity tell Rivet, through the handler, to ignore beam checks
     set_ignore_beams(true);
   }
  
   // Cross sections
   if (rivet_confs_.has_value({"Cross_Section"})) {
     std::array<double, 2> xs = rivet_confs_.take({"Cross_Section"});
     set_cross_section(xs[0], xs[1]);
   }
  
   // Logging in Rivet
   if (rivet_confs_.has_value({"Logging"})) {
     std::map<std::string, std::string> logs = rivet_confs_.take({"Logging"});
     for (auto nl : logs)
       set_log_level(nl.first, nl.second);
   }
  
   // Treatment of event weights in Rivet
   if (rivet_confs_.has_value({"Weights"})) {
     auto wconf = rivet_confs_["Weights"];
  
     // Do not care about multi weights - bool
     if (wconf.has_value({"No_Multi"})) {
       analysis_handler_proxy()->skipMultiWeights(wconf.take({"No_Multi"}));
     }
  
     // Set nominal weight name
     if (wconf.has_value({"Nominal"})) {
       analysis_handler_proxy()->setNominalWeightName(wconf.take({"Nominal"}));
     }
  
     // Set cap (maximum) on weights
     if (wconf.has_value({"Cap"})) {
       analysis_handler_proxy()->setWeightCap(wconf.take({"Cap"}));
     }
  
     // Whether to smear for NLO calculations
     if (wconf.has_value({"NLO_Smearing"})) {
       analysis_handler_proxy()->setNLOSmearing(wconf.take({"NLO_Smearing"}));
     }
  
     // Select which weights to enable
     if (wconf.has_value({"Select"})) {
       std::vector<std::string> sel = wconf.take({"Select"});
       std::stringstream s;
       int comma = 0;
       for (auto w : sel)
         s << (comma++ ? "," : "") << w;
       analysis_handler_proxy()->selectMultiWeights(s.str());
     }
  
     // Select weights to disable
     if (wconf.has_value({"Deselect"})) {
       std::vector<std::string> sel = wconf.take({"Deselect"});
       std::stringstream s;
       int comma = 0;
       for (auto w : sel)
         s << (comma++ ? "," : "") << w;
       analysis_handler_proxy()->deselectMultiWeights(s.str());
     }
   }
   logg[LOutput].debug() << "After processing configuration:\n"
                         << rivet_confs_.to_string() << std::endl;
 }

Here is the call graph for this function:

Here is the caller graph for this function:

◆ analysis_handler_proxy()

Proxy smash::RivetOutput::analysis_handler_proxy ( )

inlineprivate

Return a proxy that temporarily disables FP exceptions.

Definition at line 117 of file rivetoutput.h.

117 { return Proxy(handler_); }

Here is the caller graph for this function:

Member Data Documentation

◆ handler_

std::shared_ptr<Rivet::AnalysisHandler> smash::RivetOutput::handler_

private

Rivet analysis handler.

Definition at line 120 of file rivetoutput.h.

◆ filename_

bf::path smash::RivetOutput::filename_

private

Output file.

Definition at line 122 of file rivetoutput.h.

◆ need_init_

bool smash::RivetOutput::need_init_

private

Whether we need initialisation.

Definition at line 124 of file rivetoutput.h.

◆ rivet_confs_

Configuration smash::RivetOutput::rivet_confs_

private

Configutations for rivet.

Definition at line 126 of file rivetoutput.h.

The documentation for this class was generated from the following files:

src/include/smash/rivetoutput.h
src/rivetoutput.cc

Classes

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Constructor & Destructor Documentation

◆ RivetOutput()

◆ ~RivetOutput()

Member Function Documentation

◆ at_eventend()

◆ add_analysis()

◆ add_path()

◆ add_preload()

◆ set_ignore_beams()

◆ set_log_level()

◆ set_cross_section()

◆ setup()

◆ analysis_handler_proxy()

Member Data Documentation

◆ handler_

◆ filename_

◆ need_init_

◆ rivet_confs_