From 61bd0c289f356c79890ba0f17797a29987b53716 Mon Sep 17 00:00:00 2001
From: Jonathan Rouzaud-Cornabas <jonathan.rouzaud-cornabas@ens-lyon.fr>
Date: Wed, 30 Jan 2013 12:30:16 +0100
Subject: [PATCH 1/1] Add MSG_task_set_data_size function binding

---
 org/simgrid/msg/Task.java |  5 +++++
 src/jmsg_task.c           | 12 ++++++++++++
 src/jmsg_task.h           | 10 ++++++++++
 3 files changed, 27 insertions(+)

diff --git a/org/simgrid/msg/Task.java b/org/simgrid/msg/Task.java
index 022961d490..1b2e0cdac1 100644
--- a/org/simgrid/msg/Task.java
+++ b/org/simgrid/msg/Task.java
@@ -152,6 +152,11 @@ public class Task {
 	 * @param computationAmount the amount of computation needed to process the task
 	 */
 	public native void setComputeDuration(double computationAmount);
+	/**
+	 * Set the data size of the task
+	 * @param dataSize the size of the task
+	 */
+	public native void setDataSize(double dataSize);
 	/* *                     * *
 	 * * Computation-related * *
 	 * *                     * */
diff --git a/src/jmsg_task.c b/src/jmsg_task.c
index d7a9fea6b7..a2e4b9281d 100644
--- a/src/jmsg_task.c
+++ b/src/jmsg_task.c
@@ -347,6 +347,18 @@ Java_org_simgrid_msg_Task_setComputeDuration
 	}
 	MSG_task_set_compute_duration(task, (double) computationAmount);
 }
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Task_setDataSize
+		(JNIEnv *env, jobject jtask, jdouble dataSize) {
+	msg_task_t task = jtask_to_native_task(jtask, env);
+
+	if (!task) {
+    jxbt_throw_notbound(env, "task", jtask);
+    return;
+	}
+	MSG_task_set_data_size(task, (double) dataSize);
+}
+
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
                                     jstring jalias,
diff --git a/src/jmsg_task.h b/src/jmsg_task.h
index 99b83e6192..2bae18d6f3 100644
--- a/src/jmsg_task.h
+++ b/src/jmsg_task.h
@@ -183,6 +183,16 @@ JNIEXPORT void JNICALL Java_org_simgrid_msg_Task_setPriority
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Task_setComputeDuration
 		(JNIEnv *env, jobject jtask, jdouble computationAmount);
+		
+		/**
+ * Class		org_simgrid_msg_Task
+ * Method		setDataSize
+ * Signature	(D)V
+ */
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Task_setDataSize
+		(JNIEnv *env, jobject jtask, jdouble dataSize);
+		
 /**
  * Class		org_simgrid_msg_Task
  * Method		send
-- 
2.20.1