1 /* Functions related to the java task instances. */
3 /* Copyright (c) 2007, 2009, 2010. The SimGrid Team.
4 * All rights reserved. */
6 /* This program is free software; you can redistribute it and/or modify it
7 * under the terms of the license (GNU LGPL) which comes with this package. */
11 #include "smx_context_java.h"
13 #include "jmsg_host.h"
14 #include "jmsg_task.h"
16 #include "jxbt_utilities.h"
20 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
22 static jmethodID jtask_method_Comm_constructor;
24 static jfieldID jtask_field_Task_bind;
25 static jfieldID jtask_field_Task_name;
26 static jfieldID jtask_field_Comm_bind;
27 static jfieldID jtask_field_Comm_taskBind;
28 static jfieldID jtask_field_Comm_receiving;
30 void jtask_bind(jobject jtask, msg_task_t task, JNIEnv * env)
32 (*env)->SetLongField(env, jtask, jtask_field_Task_bind, (jlong) (long) (task));
35 msg_task_t jtask_to_native_task(jobject jtask, JNIEnv * env)
37 return (msg_task_t) (long) (*env)->GetLongField(env, jtask, jtask_field_Task_bind);
40 jboolean jtask_is_valid(jobject jtask, JNIEnv * env)
42 return (*env)->GetLongField(env, jtask, jtask_field_Task_bind) ? JNI_TRUE : JNI_FALSE;
45 JNIEXPORT void JNICALL
46 Java_org_simgrid_msg_Task_nativeInit(JNIEnv *env, jclass cls) {
47 jclass jtask_class_Comm = (*env)->FindClass(env, "org/simgrid/msg/Comm");
48 jclass jtask_class_Task = (*env)->FindClass(env, "org/simgrid/msg/Task");
50 jtask_method_Comm_constructor = (*env)->GetMethodID(env, jtask_class_Comm, "<init>", "()V");
51 jtask_field_Task_bind = jxbt_get_jfield(env, jtask_class_Task, "bind", "J");
52 jtask_field_Task_name = jxbt_get_jfield(env, jtask_class_Task, "name", "Ljava/lang/String;");
53 jtask_field_Comm_bind = jxbt_get_jfield(env, jtask_class_Comm, "bind", "J");
54 jtask_field_Comm_taskBind = jxbt_get_jfield(env, jtask_class_Comm, "taskBind", "J");
55 jtask_field_Comm_receiving = jxbt_get_jfield(env, jtask_class_Comm, "receiving", "Z");
56 if (!jtask_field_Task_bind || !jtask_class_Task || !jtask_field_Comm_bind || !jtask_field_Comm_taskBind ||
57 !jtask_field_Comm_receiving || !jtask_method_Comm_constructor) {
58 jxbt_throw_native(env,bprintf("Can't find some fields in Java class."));
62 JNIEXPORT void JNICALL
63 Java_org_simgrid_msg_Task_create(JNIEnv * env,
64 jobject jtask, jstring jname,
65 jdouble jcomputeDuration,
68 msg_task_t task; /* the native task to create */
69 const char *name = NULL; /* the name of the task */
71 if (jcomputeDuration < 0) {
72 jxbt_throw_illegal(env,
74 ("Task ComputeDuration (%f) cannot be negative",
75 (double) jcomputeDuration));
79 if (jmessageSize < 0) {
80 jxbt_throw_illegal(env,
81 bprintf("Task MessageSize (%f) cannot be negative",
82 (double) jmessageSize));
87 /* get the C string from the java string */
88 name = (*env)->GetStringUTFChars(env, jname, 0);
93 MSG_task_create(name, (double) jcomputeDuration,
94 (double) jmessageSize, NULL);
96 (*env)->ReleaseStringUTFChars(env, jname, name);
97 /* sets the task name */
98 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
99 /* bind & store the task */
100 jtask_bind(jtask, task, env);
101 MSG_task_set_data(task, jtask);
104 JNIEXPORT void JNICALL
105 Java_org_simgrid_msg_Task_parallelCreate(JNIEnv * env,
110 jcomputeDurations_arg,
114 msg_task_t task; /* the native parallel task to create */
115 const char *name; /* the name of the task */
118 double *computeDurations;
119 double *messageSizes;
120 jdouble *jcomputeDurations;
121 jdouble *jmessageSizes;
126 if (!jcomputeDurations_arg) {
129 ("Parallel task compute durations cannot be null"));
133 if (!jmessageSizes_arg) {
136 ("Parallel task message sizes cannot be null"));
141 jxbt_throw_null(env, xbt_strdup("Parallel task name cannot be null"));
145 host_count = (int) (*env)->GetArrayLength(env, jhosts);
148 hosts = xbt_new0(msg_host_t, host_count);
149 computeDurations = xbt_new0(double, host_count);
150 messageSizes = xbt_new0(double, host_count * host_count);
153 (*env)->GetDoubleArrayElements(env, jcomputeDurations_arg, 0);
155 (*env)->GetDoubleArrayElements(env, jmessageSizes_arg, 0);
157 for (index = 0; index < host_count; index++) {
158 jhost = (*env)->GetObjectArrayElement(env, jhosts, index);
159 hosts[index] = jhost_get_native(env, jhost);
160 computeDurations[index] = jcomputeDurations[index];
162 for (index = 0; index < host_count * host_count; index++) {
163 messageSizes[index] = jmessageSizes[index];
166 (*env)->ReleaseDoubleArrayElements(env, jcomputeDurations_arg,
167 jcomputeDurations, 0);
168 (*env)->ReleaseDoubleArrayElements(env, jmessageSizes_arg, jmessageSizes,
172 /* get the C string from the java string */
173 name = (*env)->GetStringUTFChars(env, jname, 0);
176 MSG_parallel_task_create(name, host_count, hosts, computeDurations,
179 (*env)->ReleaseStringUTFChars(env, jname, name);
180 /* sets the task name */
181 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
182 /* associate the java task object and the native task */
183 jtask_bind(jtask, task, env);
185 MSG_task_set_data(task, (void *) jtask);
187 if (!MSG_task_get_data(task))
188 jxbt_throw_jni(env, "global ref allocation failed");
191 JNIEXPORT void JNICALL
192 Java_org_simgrid_msg_Task_cancel(JNIEnv * env,
195 msg_task_t ptask = jtask_to_native_task(jtask, env);
198 jxbt_throw_notbound(env, "task", jtask);
202 msg_error_t rv = MSG_task_cancel(ptask);
204 jxbt_check_res("MSG_task_cancel()", rv, MSG_OK,
205 bprintf("unexpected error , please report this bug"));
208 JNIEXPORT void JNICALL
209 Java_org_simgrid_msg_Task_execute(JNIEnv * env, jobject jtask)
211 msg_task_t task = jtask_to_native_task(jtask, env);
214 jxbt_throw_notbound(env, "task", jtask);
218 rv = MSG_task_execute(task);
220 jmsg_throw_status(env, rv);
224 JNIEXPORT jstring JNICALL
225 Java_org_simgrid_msg_Task_getName(JNIEnv * env,
227 msg_task_t task = jtask_to_native_task(jtask, env);
230 jxbt_throw_notbound(env, "task", jtask);
234 return (*env)->NewStringUTF(env, MSG_task_get_name(task));
237 JNIEXPORT jobject JNICALL
238 Java_org_simgrid_msg_Task_getSender(JNIEnv * env,
240 msg_process_t process;
242 msg_task_t task = jtask_to_native_task(jtask, env);
245 jxbt_throw_notbound(env, "task", jtask);
249 process = MSG_task_get_sender(task);
250 if (process == NULL) {
253 return (jobject) native_to_java_process(process);
256 JNIEXPORT jobject JNICALL
257 Java_org_simgrid_msg_Task_getSource(JNIEnv * env,
261 msg_task_t task = jtask_to_native_task(jtask, env);
264 jxbt_throw_notbound(env, "task", jtask);
268 host = MSG_task_get_source(task);
272 if (!MSG_host_get_data(host)) {
273 jxbt_throw_jni(env, "MSG_task_get_source() failed");
277 return (jobject) MSG_host_get_data(host);
280 JNIEXPORT jdouble JNICALL
281 Java_org_simgrid_msg_Task_getComputeDuration(JNIEnv * env,
284 msg_task_t ptask = jtask_to_native_task(jtask, env);
287 jxbt_throw_notbound(env, "task", jtask);
290 return (jdouble) MSG_task_get_compute_duration(ptask);
293 JNIEXPORT jdouble JNICALL
294 Java_org_simgrid_msg_Task_getRemainingDuration(JNIEnv * env, jobject jtask)
296 msg_task_t ptask = jtask_to_native_task(jtask, env);
299 jxbt_throw_notbound(env, "task", jtask);
302 return (jdouble) MSG_task_get_remaining_computation(ptask);
304 JNIEXPORT void JNICALL
305 Java_org_simgrid_msg_Task_setName(JNIEnv *env, jobject jtask, jobject jname) {
306 msg_task_t task = jtask_to_native_task(jtask, env);
309 jxbt_throw_notbound(env, "task", jtask);
312 const char *name = (*env)->GetStringUTFChars(env, jname, 0);
314 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
315 MSG_task_set_name(task, name);
317 (*env)->ReleaseStringUTFChars(env, jname, name);
319 JNIEXPORT void JNICALL
320 Java_org_simgrid_msg_Task_setPriority(JNIEnv * env,
321 jobject jtask, jdouble priority)
323 msg_task_t task = jtask_to_native_task(jtask, env);
326 jxbt_throw_notbound(env, "task", jtask);
329 MSG_task_set_priority(task, (double) priority);
331 JNIEXPORT void JNICALL
332 Java_org_simgrid_msg_Task_setComputeDuration
333 (JNIEnv *env, jobject jtask, jdouble computationAmount) {
334 msg_task_t task = jtask_to_native_task(jtask, env);
337 jxbt_throw_notbound(env, "task", jtask);
340 MSG_task_set_compute_duration(task, (double) computationAmount);
342 JNIEXPORT void JNICALL
343 Java_org_simgrid_msg_Task_setDataSize
344 (JNIEnv *env, jobject jtask, jdouble dataSize) {
345 msg_task_t task = jtask_to_native_task(jtask, env);
348 jxbt_throw_notbound(env, "task", jtask);
351 MSG_task_set_data_size(task, (double) dataSize);
354 JNIEXPORT void JNICALL
355 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
360 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
362 msg_task_t task = jtask_to_native_task(jtask, env);
366 (*env)->ReleaseStringUTFChars(env, jalias, alias);
367 jxbt_throw_notbound(env, "task", jtask);
371 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
372 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
373 rv = MSG_task_send_with_timeout(task, alias, (double) jtimeout);
374 (*env)->ReleaseStringUTFChars(env, jalias, alias);
377 jmsg_throw_status(env, rv);
381 JNIEXPORT void JNICALL
382 Java_org_simgrid_msg_Task_sendBounded(JNIEnv * env,jobject jtask,
388 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
390 msg_task_t task = jtask_to_native_task(jtask, env);
394 (*env)->ReleaseStringUTFChars(env, jalias, alias);
395 jxbt_throw_notbound(env, "task", jtask);
399 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
400 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
401 rv = MSG_task_send_with_timeout_bounded(task, alias, (double) jtimeout, (double) maxrate);
402 (*env)->ReleaseStringUTFChars(env, jalias, alias);
405 jmsg_throw_status(env, rv);
409 JNIEXPORT jobject JNICALL
410 Java_org_simgrid_msg_Task_receive(JNIEnv * env, jclass cls,
411 jstring jalias, jdouble jtimeout,
415 msg_task_t *task = xbt_new(msg_task_t,1);
418 msg_host_t host = NULL;
419 jobject jtask_global, jtask_local;
423 host = jhost_get_native(env, jhost);
426 jxbt_throw_notbound(env, "host", jhost);
431 alias = (*env)->GetStringUTFChars(env, jalias, 0);
432 rv = MSG_task_receive_ext(task, alias, (double) jtimeout, host);
434 jmsg_throw_status(env,rv);
437 jtask_global = MSG_task_get_data(*task);
439 /* Convert the global ref into a local ref so that the JVM can free the stuff */
440 jtask_local = (*env)->NewLocalRef(env, jtask_global);
441 (*env)->DeleteGlobalRef(env, jtask_global);
442 MSG_task_set_data(*task, NULL);
444 (*env)->ReleaseStringUTFChars(env, jalias, alias);
448 return (jobject) jtask_local;
452 JNIEXPORT jobject JNICALL
453 Java_org_simgrid_msg_Task_irecv(JNIEnv * env, jclass cls, jstring jmailbox) {
457 //pointer to store the task object pointer.
458 msg_task_t *task = xbt_new(msg_task_t,1);
460 /* There should be a cache here */
461 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
464 jxbt_throw_native(env,bprintf("fieldID or methodID or class not found."));
468 jobject jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
470 jxbt_throw_native(env,bprintf("Can't create a Comm object."));
474 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
476 comm = MSG_task_irecv(task,mailbox);
478 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
479 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(task));
480 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_TRUE);
482 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
487 JNIEXPORT jobject JNICALL
488 Java_org_simgrid_msg_Task_isend(JNIEnv *env, jobject jtask, jstring jmailbox) {
498 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
500 if (!comm_class) return NULL;
502 jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
503 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
505 task = jtask_to_native_task(jtask, env);
508 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
509 (*env)->DeleteLocalRef(env, jcomm);
510 jxbt_throw_notbound(env, "task", jtask);
514 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
515 comm = MSG_task_isend(task,mailbox);
517 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
518 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
519 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
521 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
526 JNIEXPORT jobject JNICALL
527 Java_org_simgrid_msg_Task_isendBounded(JNIEnv *env, jobject jtask, jstring jmailbox, jdouble maxrate) {
537 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
539 if (!comm_class) return NULL;
541 jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
542 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
544 task = jtask_to_native_task(jtask, env);
547 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
548 (*env)->DeleteLocalRef(env, jcomm);
549 jxbt_throw_notbound(env, "task", jtask);
553 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
554 comm = MSG_task_isend_bounded(task,mailbox,maxrate);
556 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
557 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
558 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
560 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
567 static void msg_task_cancel_on_failed_dsend(void*t) {
569 JNIEnv *env =get_current_thread_env();
570 jobject jtask_global = MSG_task_get_data(task);
572 /* Destroy the global ref so that the JVM can free the stuff */
573 (*env)->DeleteGlobalRef(env, jtask_global);
574 MSG_task_set_data(task, NULL);
575 MSG_task_destroy(task);
578 JNIEXPORT void JNICALL
579 Java_org_simgrid_msg_Task_dsend(JNIEnv * env, jobject jtask,
581 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
583 msg_task_t task = jtask_to_native_task(jtask, env);
587 (*env)->ReleaseStringUTFChars(env, jalias, alias);
588 jxbt_throw_notbound(env, "task", jtask);
592 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
593 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
594 MSG_task_dsend(task, alias, msg_task_cancel_on_failed_dsend);
596 (*env)->ReleaseStringUTFChars(env, jalias, alias);
599 JNIEXPORT void JNICALL
600 Java_org_simgrid_msg_Task_dsendBounded(JNIEnv * env, jobject jtask,
601 jstring jalias, jdouble maxrate) {
602 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
604 msg_task_t task = jtask_to_native_task(jtask, env);
608 (*env)->ReleaseStringUTFChars(env, jalias, alias);
609 jxbt_throw_notbound(env, "task", jtask);
613 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
614 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
615 MSG_task_dsend_bounded(task, alias, msg_task_cancel_on_failed_dsend,(double)maxrate);
617 (*env)->ReleaseStringUTFChars(env, jalias, alias);
622 JNIEXPORT jboolean JNICALL
623 Java_org_simgrid_msg_Task_listen(JNIEnv * env, jclass cls, jstring jalias)
628 alias = (*env)->GetStringUTFChars(env, jalias, 0);
629 rv = MSG_task_listen(alias);
630 (*env)->ReleaseStringUTFChars(env, jalias, alias);
632 return (jboolean) rv;
635 JNIEXPORT jint JNICALL
636 Java_org_simgrid_msg_Task_listenFromHost(JNIEnv * env, jclass cls, jstring jalias, jobject jhost)
641 msg_host_t host = jhost_get_native(env, jhost);
644 jxbt_throw_notbound(env, "host", jhost);
647 alias = (*env)->GetStringUTFChars(env, jalias, 0);
648 rv = MSG_task_listen_from_host(alias, host);
649 (*env)->ReleaseStringUTFChars(env, jalias, alias);
655 JNIEXPORT jint JNICALL
656 Java_org_simgrid_msg_Task_listenFrom(JNIEnv * env, jclass cls, jstring jalias)
659 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
660 rv = MSG_task_listen_from(alias);
661 (*env)->ReleaseStringUTFChars(env, jalias, alias);