+
+namespace simgrid {
+namespace instr {
+static bool trace_active = false;
+TraceFormat trace_format = TraceFormat::Paje;
+int trace_precision;
+
+/*************
+ * Callbacks *
+ *************/
+xbt::signal<void(Container const&)> Container::on_creation;
+xbt::signal<void(Container const&)> Container::on_destruction;
+xbt::signal<void(Type const&, PajeEventType)> Type::on_creation;
+xbt::signal<void(LinkType const&, Type const&, Type const&)> LinkType::on_creation;
+xbt::signal<void(PajeEvent&)> PajeEvent::on_creation;
+xbt::signal<void(PajeEvent const&)> PajeEvent::on_destruction;
+xbt::signal<void(StateEvent const&)> StateEvent::on_destruction;
+xbt::signal<void(EntityValue const&)> EntityValue::on_creation;
+
+static void on_container_creation_paje(const Container& c)
+{
+ double timestamp = simgrid_get_clock();
+ std::stringstream stream;
+
+ XBT_DEBUG("%s: event_type=%u, timestamp=%f", __func__, static_cast<unsigned>(PajeEventType::CreateContainer),
+ timestamp);
+
+ stream << std::fixed << std::setprecision(trace_precision) << PajeEventType::CreateContainer << " ";
+ stream << timestamp << " " << c.get_id() << " " << c.type_->get_id() << " " << c.parent_->get_id() << " \"";
+ if (c.get_name().find("rank-") != 0)
+ stream << c.get_name() << "\"";
+ else
+ /* Subtract -1 because this is the process id and we transform it to the rank id */
+ stream << "rank-" << stoi(c.get_name().substr(5)) - 1 << "\"";
+
+ XBT_DEBUG("Dump %s", stream.str().c_str());
+ tracing_file << stream.str() << std::endl;
+}
+
+static void on_container_destruction_paje(const Container& c)
+{
+ // trace my destruction, but not if user requests so or if the container is root
+ if (not trace_disable_destroy && &c != Container::get_root()) {
+ std::stringstream stream;
+ double timestamp = simgrid_get_clock();
+
+ XBT_DEBUG("%s: event_type=%u, timestamp=%f", __func__, static_cast<unsigned>(PajeEventType::DestroyContainer),
+ timestamp);
+
+ stream << std::fixed << std::setprecision(trace_precision) << PajeEventType::DestroyContainer << " ";
+ stream << timestamp << " " << c.type_->get_id() << " " << c.get_id();
+ XBT_DEBUG("Dump %s", stream.str().c_str());
+ tracing_file << stream.str() << std::endl;
+ }
+}
+
+static void on_container_creation_ti(const Container& c)
+{
+ XBT_DEBUG("%s: event_type=%u, timestamp=%f", __func__, static_cast<unsigned>(PajeEventType::CreateContainer),
+ simgrid_get_clock());
+ // if we are in the mode with only one file
+ static std::ofstream* ti_unique_file = nullptr;
+ static double prefix = 0.0;
+
+ if (tracing_files.empty()) {
+ // generate unique run id with time
+ prefix = xbt_os_time();
+ }
+
+ if (not simgrid::config::get_value<bool>("tracing/smpi/format/ti-one-file") || ti_unique_file == nullptr) {
+ std::string folder_name = simgrid::config::get_value<std::string>("tracing/filename") + "_files";
+ std::string filename = folder_name + "/" + std::to_string(prefix) + "_" + c.get_name() + ".txt";
+#ifdef WIN32
+ _mkdir(folder_name.c_str());
+#else
+ mkdir(folder_name.c_str(), S_IRWXU | S_IRWXG | S_IRWXO);
+#endif
+ ti_unique_file = new std::ofstream(filename.c_str(), std::ofstream::out);
+ xbt_assert(not ti_unique_file->fail(), "Tracefile %s could not be opened for writing", filename.c_str());
+ tracing_file << filename << std::endl;
+ }
+ tracing_files.insert({&c, ti_unique_file});
+}
+
+static void on_container_destruction_ti(const Container& c)
+{
+ if (not trace_disable_destroy && &c != Container::get_root()) {
+ if (not simgrid::config::get_value<bool>("tracing/smpi/format/ti-one-file") || tracing_files.size() == 1) {
+ tracing_files.at(&c)->close();
+ delete tracing_files.at(&c);
+ }
+ tracing_files.erase(&c);
+ }
+}
+
+static void on_entity_value_creation(const EntityValue& value)
+{
+ std::stringstream stream;
+ XBT_DEBUG("%s: event_type=%u", __func__, static_cast<unsigned>(PajeEventType::DefineEntityValue));
+ stream << std::fixed << std::setprecision(trace_precision) << PajeEventType::DefineEntityValue;
+ stream << " " << value.get_id() << " " << value.get_parent()->get_id() << " " << value.get_name();
+ if (not value.get_color().empty())
+ stream << " \"" << value.get_color() << "\"";
+ XBT_DEBUG("Dump %s", stream.str().c_str());
+ tracing_file << stream.str() << std::endl;
+}
+
+static void on_event_creation(PajeEvent& event)
+{
+ XBT_DEBUG("%s: event_type=%u, timestamp=%.*f", __func__, static_cast<unsigned>(event.eventType_), trace_precision,
+ event.timestamp_);
+ event.stream_ << std::fixed << std::setprecision(trace_precision);
+ event.stream_ << event.eventType_ << " " << event.timestamp_ << " ";
+ event.stream_ << event.get_type()->get_id() << " " << event.get_container()->get_id();
+}
+
+static void on_event_destruction(const PajeEvent& event)
+{
+ XBT_DEBUG("Dump %s", event.stream_.str().c_str());
+ tracing_file << event.stream_.str() << std::endl;
+}
+
+static void on_state_event_destruction(const StateEvent& event)
+{
+ if (event.has_extra())
+ *tracing_files.at(event.get_container()) << event.stream_.str() << std::endl;
+}
+
+static void on_type_creation(const Type& type, PajeEventType event_type)
+{
+ if (event_type == PajeEventType::DefineLinkType)
+ return; // this kind of type has to be handled differently
+
+ std::stringstream stream;
+ stream << std::fixed << std::setprecision(trace_precision);
+ XBT_DEBUG("%s: event_type=%u, timestamp=%.*f", __func__, static_cast<unsigned>(event_type), trace_precision, 0.);
+ stream << event_type << " " << type.get_id() << " " << type.get_parent()->get_id() << " " << type.get_name();
+ if (type.is_colored())
+ stream << " \"" << type.get_color() << "\"";
+ XBT_DEBUG("Dump %s", stream.str().c_str());
+ tracing_file << stream.str() << std::endl;
+}
+
+static void on_link_type_creation(const Type& type, const Type& source, const Type& dest)
+{
+ std::stringstream stream;
+ XBT_DEBUG("%s: event_type=%u, timestamp=%.*f", __func__, static_cast<unsigned>(PajeEventType::DefineLinkType),
+ trace_precision, 0.);
+ stream << PajeEventType::DefineLinkType << " " << type.get_id() << " " << type.get_parent()->get_id();
+ stream << " " << source.get_id() << " " << dest.get_id() << " " << type.get_name();
+ XBT_DEBUG("Dump %s", stream.str().c_str());
+ tracing_file << stream.str() << std::endl;
+}
+
+static void on_simulation_start()
+{
+ if (trace_active || not TRACE_is_enabled())
+ return;
+
+ define_callbacks();
+
+ XBT_DEBUG("Tracing starts");
+ trace_precision = config::get_value<int>("tracing/precision");
+
+ /* init the tracing module to generate the right output */
+ std::string format = config::get_value<std::string>("tracing/smpi/format");
+ XBT_DEBUG("Tracing format %s", format.c_str());
+
+ /* open the trace file(s) */
+ std::string filename = simgrid::config::get_value<std::string>("tracing/filename");
+ tracing_file.open(filename.c_str(), std::ofstream::out);
+ if (tracing_file.fail()) {
+ throw TracingError(XBT_THROW_POINT,
+ xbt::string_printf("Tracefile %s could not be opened for writing.", filename.c_str()));
+ }
+
+ XBT_DEBUG("Filename %s is open for writing", filename.c_str());
+
+ if (format == "Paje") {
+ Container::on_creation.connect(on_container_creation_paje);
+ Container::on_destruction.connect(on_container_destruction_paje);
+ EntityValue::on_creation.connect(on_entity_value_creation);
+ Type::on_creation.connect(on_type_creation);
+ LinkType::on_creation.connect(on_link_type_creation);
+ PajeEvent::on_creation.connect(on_event_creation);
+ PajeEvent::on_destruction.connect(on_event_destruction);
+
+ paje::dump_generator_version();
+
+ /* output one line comment */
+ std::string comment = simgrid::config::get_value<std::string>("tracing/comment");
+ if (not comment.empty())
+ tracing_file << "# " << comment << std::endl;
+
+ /* output comment file */
+ paje::dump_comment_file(config::get_value<std::string>(OPT_TRACING_COMMENT_FILE));
+ paje::dump_header(trace_basic, TRACE_display_sizes());
+ } else {
+ trace_format = TraceFormat::Ti;
+ Container::on_creation.connect(on_container_creation_ti);
+ Container::on_destruction.connect(on_container_destruction_ti);
+ StateEvent::on_destruction.connect(on_state_event_destruction);
+ }
+
+ trace_active = true;
+ XBT_DEBUG("Tracing is on");
+}
+
+static void on_simulation_end()
+{
+ if (not trace_active)
+ return;
+
+ /* dump trace buffer */
+ last_timestamp_to_dump = surf_get_clock();
+ dump_buffer(true);
+
+ const Type* root_type = Container::get_root()->type_;
+ /* destroy all data structures of tracing (and free) */
+ delete Container::get_root();
+ delete root_type;
+
+ /* close the trace files */
+ tracing_file.close();
+ XBT_DEBUG("Filename %s is closed", config::get_value<std::string>("tracing/filename").c_str());
+
+ /* de-activate trace */
+ trace_active = false;
+ XBT_DEBUG("Tracing is off");
+ XBT_DEBUG("Tracing system is shutdown");
+}
+
+void init()
+{
+ static bool is_initialized = false;
+ if (is_initialized)
+ return;
+
+ is_initialized = true;
+
+ /* name of the tracefile */
+ config::declare_flag<std::string>("tracing/filename", "Trace file created by the instrumented SimGrid.",
+ "simgrid.trace");
+ config::declare_flag<std::string>("tracing/smpi/format",
+ "Select trace output format used by SMPI. The default is the 'Paje' format. "
+ "The 'TI' (Time-Independent) format allows for trace replay.",
+ "Paje");
+
+ config::declare_flag<bool>(OPT_TRACING_FORMAT_TI_ONEFILE,
+ "(smpi only) For replay format only : output to one file only", false);
+ config::declare_flag<std::string>("tracing/comment", "Add a comment line to the top of the trace file.", "");
+ config::declare_flag<std::string>(OPT_TRACING_COMMENT_FILE,
+ "Add the contents of a file as comments to the top of the trace.", "");
+ config::declare_flag<int>("tracing/precision",
+ "Numerical precision used when timestamping events "
+ "(expressed in number of digits after decimal point)",
+ 6);
+
+ /* Connect Engine callbacks */
+ s4u::Engine::on_platform_creation.connect(on_simulation_start);
+ s4u::Engine::on_time_advance.connect([](double /*time_delta*/) { dump_buffer(false); });
+ s4u::Engine::on_deadlock.connect(on_simulation_end);
+ s4u::Engine::on_simulation_end.connect(on_simulation_end);
+}
+} // namespace instr
+} // namespace simgrid