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_Task_messageSize;
27 static jfieldID jtask_field_Comm_bind;
28 static jfieldID jtask_field_Comm_taskBind;
29 static jfieldID jtask_field_Comm_receiving;
31 void jtask_bind(jobject jtask, msg_task_t task, JNIEnv * env)
33 (*env)->SetLongField(env, jtask, jtask_field_Task_bind, (jlong) (long) (task));
36 msg_task_t jtask_to_native_task(jobject jtask, JNIEnv * env)
38 return (msg_task_t) (long) (*env)->GetLongField(env, jtask, jtask_field_Task_bind);
41 jboolean jtask_is_valid(jobject jtask, JNIEnv * env)
43 return (*env)->GetLongField(env, jtask, jtask_field_Task_bind) ? JNI_TRUE : JNI_FALSE;
46 JNIEXPORT void JNICALL
47 Java_org_simgrid_msg_Task_nativeInit(JNIEnv *env, jclass cls) {
48 jclass jtask_class_Comm = (*env)->FindClass(env, "org/simgrid/msg/Comm");
49 jclass jtask_class_Task = (*env)->FindClass(env, "org/simgrid/msg/Task");
51 jtask_method_Comm_constructor = (*env)->GetMethodID(env, jtask_class_Comm, "<init>", "()V");
52 jtask_field_Task_bind = jxbt_get_jfield(env, jtask_class_Task, "bind", "J");
53 jtask_field_Task_name = jxbt_get_jfield(env, jtask_class_Task, "name", "Ljava/lang/String;");
54 jtask_field_Task_messageSize = jxbt_get_jfield(env, jtask_class_Task, "messageSize", "D");
55 jtask_field_Comm_bind = jxbt_get_jfield(env, jtask_class_Comm, "bind", "J");
56 jtask_field_Comm_taskBind = jxbt_get_jfield(env, jtask_class_Comm, "taskBind", "J");
57 jtask_field_Comm_receiving = jxbt_get_jfield(env, jtask_class_Comm, "receiving", "Z");
58 if (!jtask_field_Task_bind || !jtask_class_Task || !jtask_field_Comm_bind || !jtask_field_Comm_taskBind ||
59 !jtask_field_Comm_receiving || !jtask_method_Comm_constructor) {
60 jxbt_throw_native(env,bprintf("Can't find some fields in Java class."));
64 JNIEXPORT void JNICALL
65 Java_org_simgrid_msg_Task_create(JNIEnv * env,
66 jobject jtask, jstring jname,
67 jdouble jcomputeDuration,
70 msg_task_t task; /* the native task to create */
71 const char *name = NULL; /* the name of the task */
73 if (jcomputeDuration < 0) {
74 jxbt_throw_illegal(env,
76 ("Task ComputeDuration (%f) cannot be negative",
77 (double) jcomputeDuration));
81 if (jmessageSize < 0) {
82 jxbt_throw_illegal(env,
83 bprintf("Task MessageSize (%f) cannot be negative",
84 (double) jmessageSize));
89 /* get the C string from the java string */
90 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 msg_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;
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(msg_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 msg_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, jobject jtask)
213 msg_task_t task = jtask_to_native_task(jtask, env);
216 jxbt_throw_notbound(env, "task", jtask);
220 rv = MSG_task_execute(task);
222 jmsg_throw_status(env, rv);
226 JNIEXPORT jstring JNICALL
227 Java_org_simgrid_msg_Task_getName(JNIEnv * env,
229 msg_task_t task = jtask_to_native_task(jtask, env);
232 jxbt_throw_notbound(env, "task", jtask);
236 return (*env)->NewStringUTF(env, MSG_task_get_name(task));
239 JNIEXPORT jobject JNICALL
240 Java_org_simgrid_msg_Task_getSender(JNIEnv * env,
242 msg_process_t process;
244 msg_task_t task = jtask_to_native_task(jtask, env);
247 jxbt_throw_notbound(env, "task", jtask);
251 process = MSG_task_get_sender(task);
252 if (process == NULL) {
255 return (jobject) native_to_java_process(process);
258 JNIEXPORT jobject JNICALL
259 Java_org_simgrid_msg_Task_getSource(JNIEnv * env,
263 msg_task_t task = jtask_to_native_task(jtask, env);
266 jxbt_throw_notbound(env, "task", jtask);
270 host = MSG_task_get_source(task);
274 if (!MSG_host_get_data(host)) {
275 jxbt_throw_jni(env, "MSG_task_get_source() failed");
279 return (jobject) MSG_host_get_data(host);
282 JNIEXPORT jdouble JNICALL
283 Java_org_simgrid_msg_Task_getComputeDuration(JNIEnv * env,
286 msg_task_t ptask = jtask_to_native_task(jtask, env);
289 jxbt_throw_notbound(env, "task", jtask);
292 return (jdouble) MSG_task_get_compute_duration(ptask);
295 JNIEXPORT jdouble JNICALL
296 Java_org_simgrid_msg_Task_getRemainingDuration(JNIEnv * env, jobject jtask)
298 msg_task_t ptask = jtask_to_native_task(jtask, env);
301 jxbt_throw_notbound(env, "task", jtask);
304 return (jdouble) MSG_task_get_remaining_computation(ptask);
306 JNIEXPORT void JNICALL
307 Java_org_simgrid_msg_Task_setName(JNIEnv *env, jobject jtask, jobject jname) {
308 msg_task_t task = jtask_to_native_task(jtask, env);
311 jxbt_throw_notbound(env, "task", jtask);
314 const char *name = (*env)->GetStringUTFChars(env, jname, 0);
316 (*env)->SetObjectField(env, jtask, jtask_field_Task_name, jname);
317 MSG_task_set_name(task, name);
319 (*env)->ReleaseStringUTFChars(env, jname, name);
321 JNIEXPORT void JNICALL
322 Java_org_simgrid_msg_Task_setPriority(JNIEnv * env,
323 jobject jtask, jdouble priority)
325 msg_task_t task = jtask_to_native_task(jtask, env);
328 jxbt_throw_notbound(env, "task", jtask);
331 MSG_task_set_priority(task, (double) priority);
333 JNIEXPORT void JNICALL
334 Java_org_simgrid_msg_Task_setComputeDuration
335 (JNIEnv *env, jobject jtask, jdouble computationAmount) {
336 msg_task_t task = jtask_to_native_task(jtask, env);
339 jxbt_throw_notbound(env, "task", jtask);
342 MSG_task_set_compute_duration(task, (double) computationAmount);
344 JNIEXPORT void JNICALL
345 Java_org_simgrid_msg_Task_setDataSize
346 (JNIEnv *env, jobject jtask, jdouble dataSize) {
347 msg_task_t task = jtask_to_native_task(jtask, env);
350 jxbt_throw_notbound(env, "task", jtask);
353 (*env)->SetDoubleField(env, jtask, jtask_field_Task_messageSize, dataSize);
354 MSG_task_set_data_size(task, (double) dataSize);
357 JNIEXPORT void JNICALL
358 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
363 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
365 msg_task_t task = jtask_to_native_task(jtask, env);
369 (*env)->ReleaseStringUTFChars(env, jalias, alias);
370 jxbt_throw_notbound(env, "task", jtask);
374 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
375 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
376 rv = MSG_task_send_with_timeout(task, alias, (double) jtimeout);
377 (*env)->ReleaseStringUTFChars(env, jalias, alias);
380 jmsg_throw_status(env, rv);
384 JNIEXPORT void JNICALL
385 Java_org_simgrid_msg_Task_sendBounded(JNIEnv * env,jobject jtask,
391 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
393 msg_task_t task = jtask_to_native_task(jtask, env);
397 (*env)->ReleaseStringUTFChars(env, jalias, alias);
398 jxbt_throw_notbound(env, "task", jtask);
402 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
403 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
404 rv = MSG_task_send_with_timeout_bounded(task, alias, (double) jtimeout, (double) maxrate);
405 (*env)->ReleaseStringUTFChars(env, jalias, alias);
408 jmsg_throw_status(env, rv);
412 JNIEXPORT jobject JNICALL
413 Java_org_simgrid_msg_Task_receive(JNIEnv * env, jclass cls,
414 jstring jalias, jdouble jtimeout,
418 msg_task_t *task = xbt_new(msg_task_t,1);
421 msg_host_t host = NULL;
422 jobject jtask_global, jtask_local;
426 host = jhost_get_native(env, jhost);
429 jxbt_throw_notbound(env, "host", jhost);
434 alias = (*env)->GetStringUTFChars(env, jalias, 0);
435 rv = MSG_task_receive_ext(task, alias, (double) jtimeout, host);
437 jmsg_throw_status(env,rv);
440 jtask_global = MSG_task_get_data(*task);
442 /* Convert the global ref into a local ref so that the JVM can free the stuff */
443 jtask_local = (*env)->NewLocalRef(env, jtask_global);
444 (*env)->DeleteGlobalRef(env, jtask_global);
445 MSG_task_set_data(*task, NULL);
447 (*env)->ReleaseStringUTFChars(env, jalias, alias);
451 return (jobject) jtask_local;
455 JNIEXPORT jobject JNICALL
456 Java_org_simgrid_msg_Task_irecv(JNIEnv * env, jclass cls, jstring jmailbox) {
460 //pointer to store the task object pointer.
461 msg_task_t *task = xbt_new(msg_task_t,1);
463 /* There should be a cache here */
464 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
467 jxbt_throw_native(env,bprintf("fieldID or methodID or class not found."));
471 jobject jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
473 jxbt_throw_native(env,bprintf("Can't create a Comm object."));
477 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
479 comm = MSG_task_irecv(task,mailbox);
481 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
482 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(task));
483 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_TRUE);
485 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
490 JNIEXPORT jobject JNICALL
491 Java_org_simgrid_msg_Task_isend(JNIEnv *env, jobject jtask, jstring jmailbox) {
501 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
503 if (!comm_class) return NULL;
505 jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
506 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
508 task = jtask_to_native_task(jtask, env);
511 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
512 (*env)->DeleteLocalRef(env, jcomm);
513 jxbt_throw_notbound(env, "task", jtask);
517 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
518 comm = MSG_task_isend(task,mailbox);
520 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
521 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
522 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
524 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
529 JNIEXPORT jobject JNICALL
530 Java_org_simgrid_msg_Task_isendBounded(JNIEnv *env, jobject jtask, jstring jmailbox, jdouble maxrate) {
540 comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
542 if (!comm_class) return NULL;
544 jcomm = (*env)->NewObject(env, comm_class, jtask_method_Comm_constructor);
545 mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
547 task = jtask_to_native_task(jtask, env);
550 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
551 (*env)->DeleteLocalRef(env, jcomm);
552 jxbt_throw_notbound(env, "task", jtask);
556 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
557 comm = MSG_task_isend_bounded(task,mailbox,maxrate);
559 (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
560 (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
561 (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
563 (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
570 static void msg_task_cancel_on_failed_dsend(void*t) {
572 JNIEnv *env =get_current_thread_env();
573 jobject jtask_global = MSG_task_get_data(task);
575 /* Destroy the global ref so that the JVM can free the stuff */
576 (*env)->DeleteGlobalRef(env, jtask_global);
577 MSG_task_set_data(task, NULL);
578 MSG_task_destroy(task);
581 JNIEXPORT void JNICALL
582 Java_org_simgrid_msg_Task_dsend(JNIEnv * env, jobject jtask,
584 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
586 msg_task_t task = jtask_to_native_task(jtask, env);
590 (*env)->ReleaseStringUTFChars(env, jalias, alias);
591 jxbt_throw_notbound(env, "task", jtask);
595 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
596 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
597 MSG_task_dsend(task, alias, msg_task_cancel_on_failed_dsend);
599 (*env)->ReleaseStringUTFChars(env, jalias, alias);
602 JNIEXPORT void JNICALL
603 Java_org_simgrid_msg_Task_dsendBounded(JNIEnv * env, jobject jtask,
604 jstring jalias, jdouble maxrate) {
605 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
607 msg_task_t task = jtask_to_native_task(jtask, env);
611 (*env)->ReleaseStringUTFChars(env, jalias, alias);
612 jxbt_throw_notbound(env, "task", jtask);
616 /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
617 MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
618 MSG_task_dsend_bounded(task, alias, msg_task_cancel_on_failed_dsend,(double)maxrate);
620 (*env)->ReleaseStringUTFChars(env, jalias, alias);
625 JNIEXPORT jboolean JNICALL
626 Java_org_simgrid_msg_Task_listen(JNIEnv * env, jclass cls, jstring jalias)
631 alias = (*env)->GetStringUTFChars(env, jalias, 0);
632 rv = MSG_task_listen(alias);
633 (*env)->ReleaseStringUTFChars(env, jalias, alias);
635 return (jboolean) rv;
638 JNIEXPORT jint JNICALL
639 Java_org_simgrid_msg_Task_listenFromHost(JNIEnv * env, jclass cls, jstring jalias, jobject jhost)
644 msg_host_t host = jhost_get_native(env, jhost);
647 jxbt_throw_notbound(env, "host", jhost);
650 alias = (*env)->GetStringUTFChars(env, jalias, 0);
651 rv = MSG_task_listen_from_host(alias, host);
652 (*env)->ReleaseStringUTFChars(env, jalias, alias);
658 JNIEXPORT jint JNICALL
659 Java_org_simgrid_msg_Task_listenFrom(JNIEnv * env, jclass cls, jstring jalias)
662 const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
663 rv = MSG_task_listen_from(alias);
664 (*env)->ReleaseStringUTFChars(env, jalias, alias);