+/* Copyright (c) 2017-2019. The SimGrid Team. All rights reserved. */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
+#include <simgrid/plugins/file_system.h>
+#include <simgrid/s4u.hpp>
+#include <xbt/replay.hpp>
+#include <xbt/str.h>
+
+#include <boost/algorithm/string/join.hpp>
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(replay_io, "Messages specific for this example");
+
+static std::unordered_map<std::string, simgrid::s4u::File*> opened_files;
+
+#define ACT_DEBUG(...) \
+ if (XBT_LOG_ISENABLED(replay_io, xbt_log_priority_verbose)) { \
+ std::string NAME = boost::algorithm::join(action, " "); \
+ XBT_DEBUG(__VA_ARGS__); \
+ } else \
+ ((void)0)
+
+static void log_action(simgrid::xbt::ReplayAction& action, double date)
+{
+ if (XBT_LOG_ISENABLED(replay_io, xbt_log_priority_verbose)) {
+ std::string s = boost::algorithm::join(action, " ");
+ XBT_VERB("%s %f", s.c_str(), date);
+ }
+}
+
+static simgrid::s4u::File* get_file_descriptor(const std::string& file_name)
+{
+ std::string full_name = simgrid::s4u::this_actor::get_name() + ":" + file_name;
+
+ return opened_files.at(full_name);
+}
+
+class Replayer {
+public:
+ explicit Replayer(std::vector<std::string> args)
+ {
+ const char* actor_name = args[0].c_str();
+ simgrid::xbt::replay_runner(actor_name, nullptr);
+ }
+
+ void operator()()
+ {
+ // Nothing to do here
+ }
+
+ /* My actions */
+ static void open(simgrid::xbt::ReplayAction& action)
+ {
+ std::string file_name = action[2];
+ double clock = simgrid::s4u::Engine::get_clock();
+ std::string full_name = simgrid::s4u::this_actor::get_name() + ":" + file_name;
+
+ ACT_DEBUG("Entering Open: %s (filename: %s)", NAME.c_str(), file_name.c_str());
+ simgrid::s4u::File* file = new simgrid::s4u::File(file_name, NULL);
+
+ opened_files.insert({full_name, file});
+
+ log_action(action, simgrid::s4u::Engine::get_clock() - clock);
+ }
+
+ static void read(simgrid::xbt::ReplayAction& action)
+ {
+ std::string file_name = action[2];
+ sg_size_t size = std::stoul(action[3]);
+ double clock = simgrid::s4u::Engine::get_clock();
+
+ simgrid::s4u::File* file = get_file_descriptor(file_name);
+
+ ACT_DEBUG("Entering Read: %s (size: %llu)", NAME.c_str(), size);
+ file->read(size);
+
+ log_action(action, simgrid::s4u::Engine::get_clock() - clock);
+ }
+
+ static void close(simgrid::xbt::ReplayAction& action)
+ {
+ std::string file_name = action[2];
+ double clock = simgrid::s4u::Engine::get_clock();
+
+ simgrid::s4u::File* file = get_file_descriptor(file_name);
+
+ ACT_DEBUG("Entering Close: %s (filename: %s)", NAME.c_str(), file_name.c_str());
+ delete file;
+
+ log_action(action, simgrid::s4u::Engine::get_clock() - clock);
+ }
+};
+
+int main(int argc, char* argv[])
+{
+ simgrid::s4u::Engine e(&argc, argv);
+ sg_storage_file_system_init();
+
+ xbt_assert(argc > 3,
+ "Usage: %s platform_file deployment_file [action_files]\n"
+ "\texample: %s platform.xml deployment.xml actions # if all actions are in the same file\n"
+ "\t# if actions are in separate files, specified in deployment\n"
+ "\texample: %s platform.xml deployment.xml",
+ argv[0], argv[0], argv[0]);
+
+ e.load_platform(argv[1]);
+ e.register_actor<Replayer>("p0");
+ e.load_deployment(argv[2]);
+
+ /* Action registration */
+ xbt_replay_action_register("open", Replayer::open);
+ xbt_replay_action_register("read", Replayer::read);
+ xbt_replay_action_register("close", Replayer::close);
+
+ if (argv[3]) {
+ simgrid::xbt::action_fs = new std::ifstream(argv[3], std::ifstream::in);
+ }
+
+ e.run();
+
+ if (argv[3]) {
+ delete simgrid::xbt::action_fs;
+ simgrid::xbt::action_fs = nullptr;
+ }
+
+ XBT_INFO("Simulation time %g", e.get_clock());
+
+ return 0;
+}