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"
19 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
21 static jmethodID jtask_method_Comm_constructor;
23 static jfieldID jtask_field_Task_bind;
24 static jfieldID jtask_field_Task_name;
25 static jfieldID jtask_field_Comm_bind;
26 static jfieldID jtask_field_Comm_taskBind;
27 static jfieldID jtask_field_Comm_receiving;
29 void jtask_bind(jobject jtask, m_task_t task, JNIEnv * env)
31 (*env)->SetLongField(env, jtask, jtask_field_Task_bind, (jlong) (long) (task));
34 m_task_t jtask_to_native_task(jobject jtask, JNIEnv * env)
36 return (m_task_t) (long) (*env)->GetLongField(env, jtask, jtask_field_Task_bind);
39 jboolean jtask_is_valid(jobject jtask, JNIEnv * env)
41 return (*env)->GetLongField(env, jtask, jtask_field_Task_bind) ? JNI_TRUE : JNI_FALSE;
44 JNIEXPORT void JNICALL
45 Java_org_simgrid_msg_Task_nativeInit(JNIEnv *env, jclass cls) {
46 jclass jtask_class_Comm = (*env)->FindClass(env, "org/simgrid/msg/Comm");
47 jclass jtask_class_Task = (*env)->FindClass(env, "org/simgrid/msg/Task");
49 jtask_method_Comm_constructor = (*env)->GetMethodID(env, jtask_class_Comm, "<init>", "()V");
50 //FIXME: Don't use jxbt_get_sfield directly, it is slower.
51 jtask_field_Task_bind = jxbt_get_sfield(env, "org/simgrid/msg/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_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
54 jtask_field_Comm_taskBind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "taskBind", "J");
55 jtask_field_Comm_receiving = jxbt_get_sfield(env, "org/simgrid/msg/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 m_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);
94 MSG_task_create(name, (double) jcomputeDuration,
95 (double) jmessageSize, NULL);
97 (*env)->ReleaseStringUTFChars(env, jname, name);
98 /* sets the task name */
99 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
100 /* bind & store the task */
101 jtask_bind(jtask, task, env);
102 MSG_task_set_data(task, jtask);
105 JNIEXPORT void JNICALL
106 Java_org_simgrid_msg_Task_parallelCreate(JNIEnv * env,
111 jcomputeDurations_arg,
115 m_task_t task; /* the native parallel task to create */
116 const char *name; /* the name of the task */
119 double *computeDurations;
120 double *messageSizes;
121 jdouble *jcomputeDurations;
122 jdouble *jmessageSizes;
128 if (!jcomputeDurations_arg) {
131 ("Parallel task compute durations cannot be null"));
135 if (!jmessageSizes_arg) {
138 ("Parallel task message sizes cannot be null"));
143 jxbt_throw_null(env, xbt_strdup("Parallel task name cannot be null"));
147 host_count = (int) (*env)->GetArrayLength(env, jhosts);
150 hosts = xbt_new0(m_host_t, host_count);
151 computeDurations = xbt_new0(double, host_count);
152 messageSizes = xbt_new0(double, host_count * host_count);
155 (*env)->GetDoubleArrayElements(env, jcomputeDurations_arg, 0);
157 (*env)->GetDoubleArrayElements(env, jmessageSizes_arg, 0);
159 for (index = 0; index < host_count; index++) {
160 jhost = (*env)->GetObjectArrayElement(env, jhosts, index);
161 hosts[index] = jhost_get_native(env, jhost);
162 computeDurations[index] = jcomputeDurations[index];
164 for (index = 0; index < host_count * host_count; index++) {
165 messageSizes[index] = jmessageSizes[index];
168 (*env)->ReleaseDoubleArrayElements(env, jcomputeDurations_arg,
169 jcomputeDurations, 0);
170 (*env)->ReleaseDoubleArrayElements(env, jmessageSizes_arg, jmessageSizes,
174 /* get the C string from the java string */
175 name = (*env)->GetStringUTFChars(env, jname, 0);
178 MSG_parallel_task_create(name, host_count, hosts, computeDurations,
181 (*env)->ReleaseStringUTFChars(env, jname, name);
182 /* sets the task name */
183 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
184 /* associate the java task object and the native task */
185 jtask_bind(jtask, task, env);
187 MSG_task_set_data(task, (void *) jtask);
189 if (!MSG_task_get_data(task))
190 jxbt_throw_jni(env, "global ref allocation failed");
193 JNIEXPORT void JNICALL
194 Java_org_simgrid_msg_Task_cancel(JNIEnv * env,
197 m_task_t ptask = jtask_to_native_task(jtask, env);
200 jxbt_throw_notbound(env, "task", jtask);
204 MSG_error_t rv = MSG_task_cancel(ptask);
206 jxbt_check_res("MSG_task_cancel()", rv, MSG_OK,
207 bprintf("unexpected error , please report this bug"));
210 JNIEXPORT void JNICALL
211 Java_org_simgrid_msg_Task_execute(JNIEnv * env,
214 m_task_t task = jtask_to_native_task(jtask, env);
217 jxbt_throw_notbound(env, "task", jtask);
221 MSG_error_t rv = MSG_task_execute(task);
223 jxbt_check_res("MSG_task_execute()", rv,
224 MSG_HOST_FAILURE | MSG_TASK_CANCELED,
225 bprintf("while executing task %s",
226 MSG_task_get_name(task)));
229 JNIEXPORT jstring JNICALL
230 Java_org_simgrid_msg_Task_getName(JNIEnv * env,
232 m_task_t task = jtask_to_native_task(jtask, env);
235 jxbt_throw_notbound(env, "task", jtask);
239 return (*env)->NewStringUTF(env, MSG_task_get_name(task));
242 JNIEXPORT jobject JNICALL
243 Java_org_simgrid_msg_Task_getSender(JNIEnv * env,
247 m_task_t task = jtask_to_native_task(jtask, env);
250 jxbt_throw_notbound(env, "task", jtask);
254 process = MSG_task_get_sender(task);
255 if (process == NULL) {
258 return (jobject) native_to_java_process(process);
261 JNIEXPORT jobject JNICALL
262 Java_org_simgrid_msg_Task_getSource(JNIEnv * env,
266 m_task_t task = jtask_to_native_task(jtask, env);
269 jxbt_throw_notbound(env, "task", jtask);
273 host = MSG_task_get_source(task);
277 if (!MSG_host_get_data(host)) {
278 jxbt_throw_jni(env, "MSG_task_get_source() failed");
282 return (jobject) MSG_host_get_data(host);
285 JNIEXPORT jdouble JNICALL
286 Java_org_simgrid_msg_Task_getComputeDuration(JNIEnv * env,
289 m_task_t ptask = jtask_to_native_task(jtask, env);
292 jxbt_throw_notbound(env, "task", jtask);
295 return (jdouble) MSG_task_get_compute_duration(ptask);
298 JNIEXPORT jdouble JNICALL
299 Java_org_simgrid_msg_Task_getRemainingDuration(JNIEnv * env, jobject jtask)
301 m_task_t ptask = jtask_to_native_task(jtask, env);
304 jxbt_throw_notbound(env, "task", jtask);
307 return (jdouble) MSG_task_get_remaining_computation(ptask);
310 JNIEXPORT void JNICALL
311 Java_org_simgrid_msg_Task_setPriority(JNIEnv * env,
312 jobject jtask, jdouble priority)
314 m_task_t task = jtask_to_native_task(jtask, env);
317 jxbt_throw_notbound(env, "task", jtask);
320 MSG_task_set_priority(task, (double) priority);
322 JNIEXPORT void JNICALL
323 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
328 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
330 m_task_t task = jtask_to_native_task(jtask, env);
334 (*env)->ReleaseStringUTFChars(env, jalias, alias);
335 jxbt_throw_notbound(env, "task", jtask);
339 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
340 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
341 rv = MSG_task_send_with_timeout(task, alias, (double) jtimeout);
343 (*env)->ReleaseStringUTFChars(env, jalias, alias);
345 jxbt_check_res("MSG_task_send_with_timeout()", rv,
346 MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
347 bprintf("while sending task %s to mailbox %s",
348 MSG_task_get_name(task), alias));
351 JNIEXPORT void JNICALL
352 Java_org_simgrid_msg_Task_sendBounded(JNIEnv * env, jobject jtask,
356 m_task_t task = jtask_to_native_task(jtask, env);
361 jxbt_throw_notbound(env, "task", jtask);
365 alias = (*env)->GetStringUTFChars(env, jalias, 0);
367 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
368 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
369 rv = MSG_task_send_bounded(task, alias, (double) jmaxRate);
371 (*env)->ReleaseStringUTFChars(env, jalias, alias);
373 jxbt_check_res("MSG_task_send_bounded()", rv,
374 MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
376 ("while sending task %s to mailbox %s with max rate %f",
377 MSG_task_get_name(task), alias, (double) jmaxRate));
382 JNIEXPORT jobject JNICALL
383 Java_org_simgrid_msg_Task_receive(JNIEnv * env, jclass cls,
384 jstring jalias, jdouble jtimeout,
388 m_task_t task = NULL;
389 m_host_t host = NULL;
390 jobject jtask_global, jtask_local;
394 host = jhost_get_native(env, jhost);
397 jxbt_throw_notbound(env, "host", jhost);
402 alias = (*env)->GetStringUTFChars(env, jalias, 0);
404 rv = MSG_task_receive_ext(&task, alias, (double) jtimeout, host);
408 jxbt_throw_time_out_failure(env,NULL);
410 case MSG_TRANSFER_FAILURE:
411 jxbt_throw_transfer_failure(env,NULL);
413 case MSG_HOST_FAILURE:
414 jxbt_throw_host_failure(env,NULL);
417 jxbt_throw_native(env,bprintf("receive failed"));
421 jtask_global = MSG_task_get_data(task);
423 /* Convert the global ref into a local ref so that the JVM can free the stuff */
424 jtask_local = (*env)->NewLocalRef(env, jtask_global);
425 (*env)->DeleteGlobalRef(env, jtask_global);
426 MSG_task_set_data(task, NULL);
428 (*env)->ReleaseStringUTFChars(env, jalias, alias);
430 jxbt_check_res("MSG_task_receive_ext()", rv,
431 MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
432 bprintf("while receiving from mailbox %s", alias));
434 return (jobject) jtask_local;
438 JNIEXPORT jobject JNICALL
439 Java_org_simgrid_msg_Task_irecv(JNIEnv * env, jclass cls, jstring jmailbox) {
443 //pointer to store the task object pointer.
444 m_task_t *task = xbt_new(m_task_t,1);
446 /* There should be a cache here */
447 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
450 jxbt_throw_native(env,bprintf("fieldID or methodID or class not found."));
454 jobject jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
456 jxbt_throw_native(env,bprintf("Can't create a Comm object."));
460 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
462 comm = MSG_task_irecv(task,mailbox);
464 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
465 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(task));
466 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_TRUE);
468 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
473 JNIEXPORT jobject JNICALL
474 Java_org_simgrid_msg_Task_isend(JNIEnv *env, jobject jtask, jstring jmailbox) {
484 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
486 if (!comm_class) return NULL;
488 jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
489 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
491 task = jtask_to_native_task(jtask, env);
494 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
495 (*env)->DeleteLocalRef(env, jcomm);
496 jxbt_throw_notbound(env, "task", jtask);
500 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
501 comm = MSG_task_isend(task,mailbox);
503 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
504 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
505 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
507 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
512 static void msg_task_cancel_on_failed_dsend(void*t) {
514 JNIEnv *env =get_current_thread_env();
515 jobject jtask_global = MSG_task_get_data(task);
517 /* Destroy the global ref so that the JVM can free the stuff */
518 (*env)->DeleteGlobalRef(env, jtask_global);
519 MSG_task_set_data(task, NULL);
520 MSG_task_destroy(task);
523 JNIEXPORT void JNICALL
524 Java_org_simgrid_msg_Task_dsend(JNIEnv * env, jobject jtask,
527 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
529 m_task_t task = jtask_to_native_task(jtask, env);
533 (*env)->ReleaseStringUTFChars(env, jalias, alias);
534 jxbt_throw_notbound(env, "task", jtask);
538 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
539 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
540 MSG_task_dsend(task, alias, msg_task_cancel_on_failed_dsend);
542 (*env)->ReleaseStringUTFChars(env, jalias, alias);
545 JNIEXPORT jboolean JNICALL
546 Java_org_simgrid_msg_Task_listen(JNIEnv * env, jclass cls,
552 alias = (*env)->GetStringUTFChars(env, jalias, 0);
554 rv = MSG_task_listen(alias);
556 (*env)->ReleaseStringUTFChars(env, jalias, alias);
558 return (jboolean) rv;
561 JNIEXPORT jint JNICALL
562 Java_org_simgrid_msg_Task_listenFromHost(JNIEnv * env, jclass cls,
568 m_host_t host = jhost_get_native(env, jhost);
571 jxbt_throw_notbound(env, "host", jhost);
574 alias = (*env)->GetStringUTFChars(env, jalias, 0);
576 rv = MSG_task_listen_from_host(alias, host);
578 (*env)->ReleaseStringUTFChars(env, jalias, alias);
584 JNIEXPORT jint JNICALL
585 Java_org_simgrid_msg_Task_listenFrom(JNIEnv * env, jclass cls,
589 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
591 rv = MSG_task_listen_from(alias);
593 (*env)->ReleaseStringUTFChars(env, jalias, alias);