Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
API format unification
[simgrid.git] / examples / s4u / mutex / s4u_mutex.cpp
index 3d0ba54..5e38484 100644 (file)
@@ -3,90 +3,66 @@
 /* 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 <xbt/sysdep.h>
-#include <mutex>
+#include <mutex> /* std::mutex and std::lock_guard */
+#include "simgrid/s4u.hpp" /* All of S4U */
 
-#include "simgrid/s4u.h"
-
-#define NB_ACTOR 2
+#define NB_ACTOR 6
 
 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "a sample log category");
 
-// simgrid::s4u::Mutex mtx; //FIXME generate error -> You must run MSG_init before using MSG
-
-//Create an actor as a c++ functor
-class Worker {
-  simgrid::s4u::Mutex mutex_;
-  int *results_;
-public:
-  Worker(int  *res, simgrid::s4u::Mutex mutex) :
-    mutex_(std::move(mutex)),  results_(res) {};
-  // Define the code of the actor
-  void operator()() {
-    // Do the calculation
-    simgrid::s4u::this_actor::execute(1000);
-
-    // lock the mutex before enter in the critical section
-    std::lock_guard<simgrid::s4u::Mutex> lock(mutex_);
-    XBT_INFO("Hello s4u, I'm ready to compute");
-
-    // And finaly add it to the results
-    *results_ += 1;
-    XBT_INFO("I'm done, good bye");
-  }
-};
-
-// This class is an example of how to use lock_guard with simgrid mutex
-class WorkerLockGuard {
-  simgrid::s4u::Mutex mutex_;
-  int *results_;
-public:
-  WorkerLockGuard(int  *res, simgrid::s4u::Mutex mutex) :
-    mutex_(std::move(mutex)),  results_(res) {};
-  void operator()() {
+/* This worker uses a classical mutex */
+static void worker(simgrid::s4u::MutexPtr mutex, int& result)
+{
+  // lock the mutex before enter in the critical section
+  mutex->lock();
 
-    simgrid::s4u::this_actor::execute(1000);
+  XBT_INFO("Hello s4u, I'm ready to compute after a regular lock");
+  // And finaly add it to the results
+  result += 1;
+  XBT_INFO("I'm done, good bye");
 
-    // Simply use the std::lock_guard like this
-    std::lock_guard<simgrid::s4u::Mutex> lock(mutex_);
+  // You have to unlock the mutex if you locked it manually.
+  // Beware of exceptions preventing your unlock() from being executed!
+  mutex->unlock();
+}
 
-    // then you are in a safe zone
-    XBT_INFO("Hello s4u, I'm ready to compute");
-    // update the results
-    *results_ += 1;
-    XBT_INFO("I'm done, good bye");
-  }
-};
+static void workerLockGuard(simgrid::s4u::MutexPtr mutex, int& result)
+{
+  // Simply use the std::lock_guard like this
+  // It's like a lock() that would do the unlock() automatically when getting out of scope
+  std::lock_guard<simgrid::s4u::Mutex> lock(*mutex);
 
-class MainActor {
-public:
-  void operator()() {
-    int res = 0;
-    simgrid::s4u::Mutex mutex;
-    simgrid::s4u::Actor workers[NB_ACTOR*2];
+  // then you are in a safe zone
+  XBT_INFO("Hello s4u, I'm ready to compute after a lock_guard");
+  // update the results
+  result += 1;
+  XBT_INFO("I'm done, good bye");
 
-    for (int i = 0; i < NB_ACTOR * 2 ; i++) {
-      // To create a worker use the static method simgrid::s4u::Actor.
-      if((i % 2) == 0 )
-        workers[i] = simgrid::s4u::Actor("worker",
-          simgrid::s4u::Host::by_name("Jupiter"),
-          WorkerLockGuard(&res, mutex));
-      else
-        workers[i] = simgrid::s4u::Actor("worker",
-          simgrid::s4u::Host::by_name("Tremblay"),
-          Worker(&res, mutex));
-    }
+  // Nothing specific here: the unlock will be automatic
+}
 
-    simgrid::s4u::this_actor::sleep(10);
-    XBT_INFO("Results is -> %d", res);
+static void master()
+{
+  int result = 0;
+  simgrid::s4u::MutexPtr mutex = simgrid::s4u::Mutex::createMutex();
+
+  for (int i = 0; i < NB_ACTOR * 2 ; i++) {
+    // To create a worker use the static method simgrid::s4u::Actor.
+    if((i % 2) == 0 )
+      simgrid::s4u::Actor::createActor("worker", simgrid::s4u::Host::by_name("Jupiter"),  workerLockGuard, mutex, std::ref(result));
+    else
+      simgrid::s4u::Actor::createActor("worker", simgrid::s4u::Host::by_name("Tremblay"), worker,          mutex, std::ref(result));
   }
-};
 
+  simgrid::s4u::this_actor::sleep_for(10);
+  XBT_INFO("Results is -> %d", result);
+}
 
-int main(int argc, char **argv) {
+int main(int argc, char **argv)
+{
   simgrid::s4u::Engine *e = new simgrid::s4u::Engine(&argc,argv);
   e->loadPlatform("../../platforms/two_hosts.xml");
-  simgrid::s4u::Actor("main", simgrid::s4u::Host::by_name("Tremblay"), 0, MainActor());
+  simgrid::s4u::Actor::createActor("main", simgrid::s4u::Host::by_name("Tremblay"), master);
   e->run();
   return 0;
 }