convert and simplify network-ns3 example

[simgrid.git] / examples / s4u / network-ns3 / s4u-network-ns3.cpp

diff --git a/examples/s4u/network-ns3/s4u-network-ns3.cpp b/examples/s4u/network-ns3/s4u-network-ns3.cpp
new file mode 100644 (file)
index 0000000..107c85e
--- /dev/null
+++ b/examples/s4u/network-ns3/s4u-network-ns3.cpp
@@ -0,0 +1,111 @@
+/* Copyright (c) 2007-2020. 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/s4u.hpp"
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "Messages specific for this s4u example");
+
+int timer_start; // set as 1 in the master process
+
+#define NTASKS 1500
+double start_time;
+const char* workernames[NTASKS];
+const char* masternames[NTASKS];
+int count_finished = 0;
+
+static void master(int argc, char* argv[])
+{
+  xbt_assert(argc == 4, "Strange number of arguments expected 3 got %d", argc - 1);
+
+  XBT_DEBUG("Master started");
+
+  /* data size */
+  double task_comm_size = std::stod(argv[1]);
+  int id                = std::stoi(argv[3]); // unique id to control statistics
+
+  /* worker name */
+  workernames[id] = xbt_strdup(argv[2]);
+
+  simgrid::s4u::Mailbox* mbox = simgrid::s4u::Mailbox::by_name(argv[3]);
+
+  masternames[id] = simgrid::s4u::Host::current()->get_cname();
+
+  double* payload = new double();
+  *payload        = task_comm_size;
+
+  count_finished++;
+  timer_start = 1;
+
+  /* time measurement */
+  start_time = simgrid::s4u::Engine::get_clock();
+  mbox->put(payload, task_comm_size);
+
+  XBT_DEBUG("Finished");
+}
+
+static void timer(int argc, char* argv[])
+{
+  xbt_assert(argc == 3, "Strange number of arguments expected 2 got %d", argc - 1);
+  double first_sleep = std::stod(argv[1]);
+  double sleep_time  = std::stod(argv[2]);
+
+  XBT_DEBUG("Timer started");
+
+  if (first_sleep)
+    simgrid::s4u::this_actor::sleep_for(first_sleep);
+
+  do {
+    XBT_DEBUG("Get sleep");
+    simgrid::s4u::this_actor::sleep_for(sleep_time);
+  } while (timer_start);
+
+  XBT_DEBUG("Finished");
+}
+
+static void worker(int argc, char* argv[])
+{
+  xbt_assert(argc == 2, "Strange number of arguments expected 1 got %d", argc - 1);
+
+  int id                      = std::stoi(argv[1]);
+  simgrid::s4u::Mailbox* mbox = simgrid::s4u::Mailbox::by_name(argv[1]);
+
+  XBT_DEBUG("Worker started");
+
+  const double* payload = static_cast<double*>(mbox->get());
+
+  count_finished--;
+  if (count_finished == 0) {
+    timer_start = 0;
+  }
+
+  double elapsed_time = simgrid::s4u::Engine::get_clock() - start_time;
+
+  XBT_INFO("FLOW[%d] : Receive %.0f bytes from %s to %s", id, *payload, masternames[id], workernames[id]);
+  XBT_DEBUG("FLOW[%d] : transferred in  %f seconds", id, elapsed_time);
+  delete payload;
+
+  XBT_DEBUG("Finished");
+}
+
+int main(int argc, char* argv[])
+{
+  simgrid::s4u::Engine e(&argc, argv);
+  xbt_assert(argc > 2,
+             "Usage: %s platform_file deployment_file\n"
+             "\tExample: %s platform.xml deployment.xml\n",
+             argv[0], argv[0]);
+
+  e.load_platform(argv[1]);
+
+  e.register_function("master", master);
+  e.register_function("worker", worker);
+  e.register_function("timer", timer);
+
+  e.load_deployment(argv[2]);
+
+  e.run();
+
+  return 0;
+}