[examples] add s4u-barrier

[simgrid.git] / examples / s4u / barrier / s4u-barrier.cpp

diff --git a/examples/s4u/barrier/s4u-barrier.cpp b/examples/s4u/barrier/s4u-barrier.cpp
new file mode 100644 (file)
index 0000000..7be2659
--- /dev/null
+++ b/examples/s4u/barrier/s4u-barrier.cpp
@@ -0,0 +1,50 @@
+/* Copyright (c) 2006-2018. 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 <memory>
+#include "simgrid/s4u.hpp"
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "a sample log category");
+
+/// Wait on the barrier then leave
+static void worker(std::shared_ptr<simgrid::s4u::Barrier> barrier)
+{
+    XBT_INFO("Waiting on the barrier");
+    barrier->wait();
+
+    XBT_INFO("Bye");
+}
+
+/// Spawn process_count-1 workers and do a barrier with them
+static void master(int process_count)
+{
+    std::shared_ptr<simgrid::s4u::Barrier> barrier(new simgrid::s4u::Barrier(process_count));
+
+    XBT_INFO("Spawning %d workers", process_count-1);
+    for (int i = 0; i < process_count-1; i++)
+    {
+        simgrid::s4u::Actor::create("worker", simgrid::s4u::Host::by_name("Jupiter"), worker, barrier);
+    }
+
+    XBT_INFO("Waiting on the barrier");
+    barrier->wait();
+
+    XBT_INFO("Bye");
+}
+
+int main(int argc, char **argv)
+{
+  // Parameter: Number of processes in the barrier
+  xbt_assert(argc >= 2, "Usage: %s <process-count>\n", argv[0]);
+  int process_count = std::stoi(argv[1]);
+  xbt_assert(process_count > 0, "<process-count> must be greater than 0");
+
+  simgrid::s4u::Engine e(&argc, argv);
+  e.load_platform("../../platforms/two_hosts.xml");
+  simgrid::s4u::Actor::create("master", simgrid::s4u::Host::by_name("Tremblay"), master, process_count);
+  e.run();
+
+  return 0;
+}