X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/d656f9465e7b1c36319f86eeafdcec58fe9551a4..9caf173e476622d309cc5653a83d224d05787cc7:/include/simgrid/s4u/Activity.hpp

diff --git a/include/simgrid/s4u/Activity.hpp b/include/simgrid/s4u/Activity.hpp
index 4568e529ad..c508d9df71 100644
--- a/include/simgrid/s4u/Activity.hpp
+++ b/include/simgrid/s4u/Activity.hpp
@@ -1,4 +1,4 @@
-/* Copyright (c) 2006-2017. The SimGrid Team. All rights reserved.          */
+/* 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. */
@@ -18,10 +18,13 @@ namespace s4u {
  *
  * This class is the ancestor of every activities that an actor can undertake, that is, of the actions that do take time in the simulated world.
  */
-XBT_PUBLIC_CLASS Activity {
+class XBT_PUBLIC Activity {
   friend Comm;
-  friend void intrusive_ptr_release(Comm * c);
-  friend void intrusive_ptr_add_ref(Comm * c);
+  friend XBT_PUBLIC void intrusive_ptr_release(Comm * c);
+  friend XBT_PUBLIC void intrusive_ptr_add_ref(Comm * c);
+  friend Exec;
+  friend XBT_PUBLIC void intrusive_ptr_release(Exec * e);
+  friend XBT_PUBLIC void intrusive_ptr_add_ref(Exec * e);
 
 protected:
   Activity()  = default;
@@ -35,36 +38,40 @@ public:
    *
    * This function is optional: you can call wait() even if you didn't call start()
    */
-  virtual void start()=0;
+  virtual Activity* start() = 0;
   /** Tests whether the given activity is terminated yet. This is a pure function. */
   //virtual bool test()=0;
   /** Blocks until the activity is terminated */
-  virtual void wait()=0;
+  virtual Activity* wait() = 0;
   /** Blocks until the activity is terminated, or until the timeout is elapsed
    *  Raises: timeout exception.*/
-  virtual void wait(double timeout)=0;
+  virtual Activity* wait(double timeout) = 0;
   /** Cancel that activity */
   //virtual void cancel();
   /** Retrieve the current state of the activity */
   e_s4u_activity_state_t getState() {return state_;}
 
   /** Get the remaining amount of work that this Activity entails. When it's 0, it's done. */
-  double getRemains();
+  virtual double getRemains();
   /** Set the [remaining] amount of work that this Activity will entail
    *
    * It is forbidden to change the amount of work once the Activity is started */
-  void setRemains(double remains);
+  Activity* setRemains(double remains);
 
   /** Put some user data onto the Activity */
-  void setUserData(void *data) {userData_=data;}
+  Activity* setUserData(void* data)
+  {
+    user_data_ = data;
+    return this;
+  }
   /** Retrieve the user data of the Activity */
-  void *getUserData() { return userData_; }
+  void* getUserData() { return user_data_; }
 
 private:
   simgrid::kernel::activity::ActivityImplPtr pimpl_ = nullptr;
   e_s4u_activity_state_t state_ = inited;
   double remains_ = 0;
-  void *userData_ = nullptr;
+  void* user_data_                                  = nullptr;
 }; // class
 
 }}; // Namespace simgrid::s4u