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, m_task_t task, JNIEnv * env)
32 (*env)->SetLongField(env, jtask, jtask_field_Task_bind, (jlong) (long) (task));
35 m_task_t jtask_to_native_task(jobject jtask, JNIEnv * env)
37 return (m_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 //FIXME: Don't use jxbt_get_sfield directly, it is slower.
52 jtask_field_Task_bind = jxbt_get_sfield(env, "org/simgrid/msg/Task", "bind", "J");
53 jtask_field_Task_name = jxbt_get_jfield(env, jtask_class_Task, "name", "Ljava/lang/String;");
54 jtask_field_Comm_bind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
55 jtask_field_Comm_taskBind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "taskBind", "J");
56 jtask_field_Comm_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
57 if (!jtask_field_Task_bind || !jtask_class_Task || !jtask_field_Comm_bind || !jtask_field_Comm_taskBind ||
58 !jtask_field_Comm_receiving || !jtask_method_Comm_constructor) {
59 jxbt_throw_native(env,bprintf("Can't find some fields in Java class."));
63 JNIEXPORT void JNICALL
64 Java_org_simgrid_msg_Task_create(JNIEnv * env,
65 jobject jtask, jstring jname,
66 jdouble jcomputeDuration,
69 m_task_t task; /* the native task to create */
70 const char *name = NULL; /* the name of the task */
72 if (jcomputeDuration < 0) {
73 jxbt_throw_illegal(env,
75 ("Task ComputeDuration (%f) cannot be negative",
76 (double) jcomputeDuration));
80 if (jmessageSize < 0) {
81 jxbt_throw_illegal(env,
82 bprintf("Task MessageSize (%f) cannot be negative",
83 (double) jmessageSize));
88 /* get the C string from the java string */
89 name = (*env)->GetStringUTFChars(env, jname, 0);
95 MSG_task_create(name, (double) jcomputeDuration,
96 (double) jmessageSize, NULL);
98 (*env)->ReleaseStringUTFChars(env, jname, name);
99 /* sets the task name */
100 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
101 /* bind & store the task */
102 jtask_bind(jtask, task, env);
103 MSG_task_set_data(task, jtask);
106 JNIEXPORT void JNICALL
107 Java_org_simgrid_msg_Task_parallelCreate(JNIEnv * env,
112 jcomputeDurations_arg,
116 m_task_t task; /* the native parallel task to create */
117 const char *name; /* the name of the task */
120 double *computeDurations;
121 double *messageSizes;
122 jdouble *jcomputeDurations;
123 jdouble *jmessageSizes;
129 if (!jcomputeDurations_arg) {
132 ("Parallel task compute durations cannot be null"));
136 if (!jmessageSizes_arg) {
139 ("Parallel task message sizes cannot be null"));
144 jxbt_throw_null(env, xbt_strdup("Parallel task name cannot be null"));
148 host_count = (int) (*env)->GetArrayLength(env, jhosts);
151 hosts = xbt_new0(m_host_t, host_count);
152 computeDurations = xbt_new0(double, host_count);
153 messageSizes = xbt_new0(double, host_count * host_count);
156 (*env)->GetDoubleArrayElements(env, jcomputeDurations_arg, 0);
158 (*env)->GetDoubleArrayElements(env, jmessageSizes_arg, 0);
160 for (index = 0; index < host_count; index++) {
161 jhost = (*env)->GetObjectArrayElement(env, jhosts, index);
162 hosts[index] = jhost_get_native(env, jhost);
163 computeDurations[index] = jcomputeDurations[index];
165 for (index = 0; index < host_count * host_count; index++) {
166 messageSizes[index] = jmessageSizes[index];
169 (*env)->ReleaseDoubleArrayElements(env, jcomputeDurations_arg,
170 jcomputeDurations, 0);
171 (*env)->ReleaseDoubleArrayElements(env, jmessageSizes_arg, jmessageSizes,
175 /* get the C string from the java string */
176 name = (*env)->GetStringUTFChars(env, jname, 0);
179 MSG_parallel_task_create(name, host_count, hosts, computeDurations,
182 (*env)->ReleaseStringUTFChars(env, jname, name);
183 /* sets the task name */
184 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
185 /* associate the java task object and the native task */
186 jtask_bind(jtask, task, env);
188 MSG_task_set_data(task, (void *) jtask);
190 if (!MSG_task_get_data(task))
191 jxbt_throw_jni(env, "global ref allocation failed");
194 JNIEXPORT void JNICALL
195 Java_org_simgrid_msg_Task_cancel(JNIEnv * env,
198 m_task_t ptask = jtask_to_native_task(jtask, env);
201 jxbt_throw_notbound(env, "task", jtask);
205 MSG_error_t rv = MSG_task_cancel(ptask);
207 jxbt_check_res("MSG_task_cancel()", rv, MSG_OK,
208 bprintf("unexpected error , please report this bug"));
211 JNIEXPORT void JNICALL
212 Java_org_simgrid_msg_Task_execute(JNIEnv * env,
215 m_task_t task = jtask_to_native_task(jtask, env);
218 jxbt_throw_notbound(env, "task", jtask);
222 MSG_error_t rv = MSG_task_execute(task);
224 jxbt_check_res("MSG_task_execute()", rv,
225 MSG_HOST_FAILURE | MSG_TASK_CANCELED,
226 bprintf("while executing task %s",
227 MSG_task_get_name(task)));
230 JNIEXPORT jstring JNICALL
231 Java_org_simgrid_msg_Task_getName(JNIEnv * env,
233 m_task_t task = jtask_to_native_task(jtask, env);
236 jxbt_throw_notbound(env, "task", jtask);
240 return (*env)->NewStringUTF(env, MSG_task_get_name(task));
243 JNIEXPORT jobject JNICALL
244 Java_org_simgrid_msg_Task_getSender(JNIEnv * env,
248 m_task_t task = jtask_to_native_task(jtask, env);
251 jxbt_throw_notbound(env, "task", jtask);
255 process = MSG_task_get_sender(task);
256 if (process == NULL) {
259 return (jobject) native_to_java_process(process);
262 JNIEXPORT jobject JNICALL
263 Java_org_simgrid_msg_Task_getSource(JNIEnv * env,
267 m_task_t task = jtask_to_native_task(jtask, env);
270 jxbt_throw_notbound(env, "task", jtask);
274 host = MSG_task_get_source(task);
278 if (!MSG_host_get_data(host)) {
279 jxbt_throw_jni(env, "MSG_task_get_source() failed");
283 return (jobject) MSG_host_get_data(host);
286 JNIEXPORT jdouble JNICALL
287 Java_org_simgrid_msg_Task_getComputeDuration(JNIEnv * env,
290 m_task_t ptask = jtask_to_native_task(jtask, env);
293 jxbt_throw_notbound(env, "task", jtask);
296 return (jdouble) MSG_task_get_compute_duration(ptask);
299 JNIEXPORT jdouble JNICALL
300 Java_org_simgrid_msg_Task_getRemainingDuration(JNIEnv * env, jobject jtask)
302 m_task_t ptask = jtask_to_native_task(jtask, env);
305 jxbt_throw_notbound(env, "task", jtask);
308 return (jdouble) MSG_task_get_remaining_computation(ptask);
310 JNIEXPORT void JNICALL
311 Java_org_simgrid_msg_Task_setName(JNIEnv *env, jobject jtask, jobject jname) {
312 m_task_t task = jtask_to_native_task(jtask, env);
315 jxbt_throw_notbound(env, "task", jtask);
318 const char *name = (*env)->GetStringUTFChars(env, jname, 0);
320 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
321 MSG_task_set_name(task, name);
323 (*env)->ReleaseStringUTFChars(env, jname, name);
325 JNIEXPORT void JNICALL
326 Java_org_simgrid_msg_Task_setPriority(JNIEnv * env,
327 jobject jtask, jdouble priority)
329 m_task_t task = jtask_to_native_task(jtask, env);
332 jxbt_throw_notbound(env, "task", jtask);
335 MSG_task_set_priority(task, (double) priority);
337 JNIEXPORT void JNICALL
338 Java_org_simgrid_msg_Task_setComputeDuration
339 (JNIEnv *env, jobject jtask, jdouble computationAmount) {
340 m_task_t task = jtask_to_native_task(jtask, env);
343 jxbt_throw_notbound(env, "task", jtask);
346 MSG_task_set_compute_duration(task, (double) computationAmount);
348 JNIEXPORT void JNICALL
349 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
354 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
356 m_task_t task = jtask_to_native_task(jtask, env);
360 (*env)->ReleaseStringUTFChars(env, jalias, alias);
361 jxbt_throw_notbound(env, "task", jtask);
365 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
366 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
367 rv = MSG_task_send_with_timeout(task, alias, (double) jtimeout);
369 (*env)->ReleaseStringUTFChars(env, jalias, alias);
371 jxbt_check_res("MSG_task_send_with_timeout()", rv,
372 MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
373 bprintf("while sending task %s to mailbox %s",
374 MSG_task_get_name(task), alias));
377 JNIEXPORT void JNICALL
378 Java_org_simgrid_msg_Task_sendBounded(JNIEnv * env, jobject jtask,
382 m_task_t task = jtask_to_native_task(jtask, env);
387 jxbt_throw_notbound(env, "task", jtask);
391 alias = (*env)->GetStringUTFChars(env, jalias, 0);
393 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
394 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
395 rv = MSG_task_send_bounded(task, alias, (double) jmaxRate);
397 (*env)->ReleaseStringUTFChars(env, jalias, alias);
399 jxbt_check_res("MSG_task_send_bounded()", rv,
400 MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
402 ("while sending task %s to mailbox %s with max rate %f",
403 MSG_task_get_name(task), alias, (double) jmaxRate));
408 JNIEXPORT jobject JNICALL
409 Java_org_simgrid_msg_Task_receive(JNIEnv * env, jclass cls,
410 jstring jalias, jdouble jtimeout,
414 m_task_t task = NULL;
415 m_host_t host = NULL;
416 jobject jtask_global, jtask_local;
420 host = jhost_get_native(env, jhost);
423 jxbt_throw_notbound(env, "host", jhost);
428 alias = (*env)->GetStringUTFChars(env, jalias, 0);
430 rv = MSG_task_receive_ext(&task, alias, (double) jtimeout, host);
434 jxbt_throw_time_out_failure(env,NULL);
436 case MSG_TRANSFER_FAILURE:
437 jxbt_throw_transfer_failure(env,NULL);
439 case MSG_HOST_FAILURE:
440 jxbt_throw_host_failure(env,NULL);
443 jxbt_throw_native(env,bprintf("receive failed"));
447 jtask_global = MSG_task_get_data(task);
449 /* Convert the global ref into a local ref so that the JVM can free the stuff */
450 jtask_local = (*env)->NewLocalRef(env, jtask_global);
451 (*env)->DeleteGlobalRef(env, jtask_global);
452 MSG_task_set_data(task, NULL);
454 (*env)->ReleaseStringUTFChars(env, jalias, alias);
456 jxbt_check_res("MSG_task_receive_ext()", rv,
457 MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
458 bprintf("while receiving from mailbox %s", alias));
460 return (jobject) jtask_local;
464 JNIEXPORT jobject JNICALL
465 Java_org_simgrid_msg_Task_irecv(JNIEnv * env, jclass cls, jstring jmailbox) {
469 //pointer to store the task object pointer.
470 m_task_t *task = xbt_new(m_task_t,1);
472 /* There should be a cache here */
473 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
476 jxbt_throw_native(env,bprintf("fieldID or methodID or class not found."));
480 jobject jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
482 jxbt_throw_native(env,bprintf("Can't create a Comm object."));
486 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
488 comm = MSG_task_irecv(task,mailbox);
490 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
491 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(task));
492 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_TRUE);
494 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
499 JNIEXPORT jobject JNICALL
500 Java_org_simgrid_msg_Task_isend(JNIEnv *env, jobject jtask, jstring jmailbox) {
510 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
512 if (!comm_class) return NULL;
514 jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
515 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
517 task = jtask_to_native_task(jtask, env);
520 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
521 (*env)->DeleteLocalRef(env, jcomm);
522 jxbt_throw_notbound(env, "task", jtask);
526 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
527 comm = MSG_task_isend(task,mailbox);
529 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
530 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
531 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
533 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
538 static void msg_task_cancel_on_failed_dsend(void*t) {
540 JNIEnv *env =get_current_thread_env();
541 jobject jtask_global = MSG_task_get_data(task);
543 /* Destroy the global ref so that the JVM can free the stuff */
544 (*env)->DeleteGlobalRef(env, jtask_global);
545 MSG_task_set_data(task, NULL);
546 MSG_task_destroy(task);
549 JNIEXPORT void JNICALL
550 Java_org_simgrid_msg_Task_dsend(JNIEnv * env, jobject jtask,
553 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
555 m_task_t task = jtask_to_native_task(jtask, env);
559 (*env)->ReleaseStringUTFChars(env, jalias, alias);
560 jxbt_throw_notbound(env, "task", jtask);
564 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
565 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
566 MSG_task_dsend(task, alias, msg_task_cancel_on_failed_dsend);
568 (*env)->ReleaseStringUTFChars(env, jalias, alias);
571 JNIEXPORT jboolean JNICALL
572 Java_org_simgrid_msg_Task_listen(JNIEnv * env, jclass cls,
578 alias = (*env)->GetStringUTFChars(env, jalias, 0);
580 rv = MSG_task_listen(alias);
582 (*env)->ReleaseStringUTFChars(env, jalias, alias);
584 return (jboolean) rv;
587 JNIEXPORT jint JNICALL
588 Java_org_simgrid_msg_Task_listenFromHost(JNIEnv * env, jclass cls,
594 m_host_t host = jhost_get_native(env, jhost);
597 jxbt_throw_notbound(env, "host", jhost);
600 alias = (*env)->GetStringUTFChars(env, jalias, 0);
602 rv = MSG_task_listen_from_host(alias, host);
604 (*env)->ReleaseStringUTFChars(env, jalias, alias);
610 JNIEXPORT jint JNICALL
611 Java_org_simgrid_msg_Task_listenFrom(JNIEnv * env, jclass cls,
615 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
617 rv = MSG_task_listen_from(alias);
619 (*env)->ReleaseStringUTFChars(env, jalias, alias);