From 514c85494408cbf2eab157d78843a5db586a57c0 Mon Sep 17 00:00:00 2001
From: Samuel Lepetit <samuel.lepetit@inria.fr>
Date: Wed, 9 May 2012 17:11:17 +0200
Subject: [PATCH] Add setComputeDuration in Task

---
 org/simgrid/msg/Task.java |  6 +++++-
 src/jmsg_task.c           | 12 ++++++++++++
 src/jmsg_task.h           |  9 ++++++++-
 3 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/org/simgrid/msg/Task.java b/org/simgrid/msg/Task.java
index b4890be72c..5242090acb 100644
--- a/org/simgrid/msg/Task.java
+++ b/org/simgrid/msg/Task.java
@@ -132,7 +132,11 @@ public class Task {
 	 * @param priority	The new priority of the task.
 	 */ 
 	public native void setPriority(double priority);
-
+	/**
+	 * Set the computation amount needed to process the task
+	 * @param computationAmount the amount of computation needed to process the task
+	 */
+	public native void setComputeDuration(double computationAmount);
 	/* *                     * *
 	 * * Computation-related * *
 	 * *                     * */
diff --git a/src/jmsg_task.c b/src/jmsg_task.c
index 844ed84f8c..8b52a30806 100644
--- a/src/jmsg_task.c
+++ b/src/jmsg_task.c
@@ -16,6 +16,7 @@
 #include "jxbt_utilities.h"
 
 #include <msg/msg.h>
+
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
 
 static jmethodID jtask_method_Comm_constructor;
@@ -320,6 +321,17 @@ Java_org_simgrid_msg_Task_setPriority(JNIEnv * env,
   MSG_task_set_priority(task, (double) priority);
 }
 JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Task_setComputeDuration
+		(JNIEnv *env, jobject jtask, jdouble computationAmount) {
+	m_task_t task = jtask_to_native_task(jtask, env);
+
+	if (!task) {
+    jxbt_throw_notbound(env, "task", jtask);
+    return;
+	}
+	MSG_task_set_compute_duration(task, (double) computationAmount);
+}
+JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
                                     jstring jalias,
                                     jdouble jtimeout)
diff --git a/src/jmsg_task.h b/src/jmsg_task.h
index 8925bcbf72..d8e7fc7e7b 100644
--- a/src/jmsg_task.h
+++ b/src/jmsg_task.h
@@ -169,7 +169,14 @@ Java_org_simgrid_msg_Task_getRemainingDuration(JNIEnv *, jobject);
  */
 JNIEXPORT void JNICALL Java_org_simgrid_msg_Task_setPriority
     (JNIEnv *, jobject, jdouble);
-
+/**
+ * Class		org_simgrid_msg_Task
+ * Method		setComputationAmount
+ * Signature	(D)V
+ */
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Task_setComputeDuration
+		(JNIEnv *env, jobject jtask, jdouble computationAmount);
 /**
  * Class		org_simgrid_msg_Task
  * Method		send
-- 
2.20.1