added a callback functionality at the end of an iteration
[charm.git] / src / ck-com / ComlibStrategy.h
index 8aea3dc18b2fd089223fa7daa6c3454f13a15f0f..1c43478dff1fef7662cf3eab927cab2d47c347ab 100644 (file)
@@ -192,6 +192,7 @@ class CharmStrategy : public Strategy {
     int forwardOnMigration;
     ComlibLearner *learner;
     CmiBool mflag;    //Does this strategy handle point-to-point or 
+    CkCallback onFinish;
 
  public:
     ComlibGroupInfo ginfo;
@@ -222,6 +223,14 @@ class CharmStrategy : public Strategy {
         return mflag;
     }
 
+    inline void setOnFinish (CkCallback of) {
+      onFinish = of;
+    }
+
+    inline CkCallback getOnFinish () {
+      return onFinish;
+    }
+
     //Called for each message
     //Function inserts a Charm++ message
     virtual void insertMessage(CharmMessageHolder *msg) {