1 /* Functions related to the java comm instances */
3 /* Copyright (c) 2012-2015. 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. */
10 #include "jxbt_utilities.h"
13 #include <simgrid/msg.h>
14 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
16 static jfieldID jcomm_field_Comm_bind;
17 static jfieldID jcomm_field_Comm_finished;
18 static jfieldID jcomm_field_Comm_receiving;
19 static jfieldID jtask_field_Comm_task;
20 static jfieldID jcomm_field_Comm_taskBind;
22 void jcomm_bind_task(JNIEnv *env, jobject jcomm) {
23 msg_comm_t comm = (msg_comm_t) (uintptr_t) env->GetLongField(jcomm, jcomm_field_Comm_bind);
24 //test if we are receiving or sending a task.
25 jboolean jreceiving = env->GetBooleanField(jcomm, jcomm_field_Comm_receiving);
26 if (jreceiving == JNI_TRUE) {
27 //bind the task object.
28 msg_task_t task = MSG_comm_get_task(comm);
29 xbt_assert(task != nullptr, "Task is nullptr");
30 jobject jtask_global = static_cast<jobject>(MSG_task_get_data(task));
31 //case where the data has already been retrieved
32 if (jtask_global == nullptr) {
36 //Make sure the data will be correctly gc.
37 jobject jtask_local = env->NewLocalRef(jtask_global);
38 env->DeleteGlobalRef(jtask_global);
40 env->SetObjectField(jcomm, jtask_field_Comm_task, jtask_local);
42 MSG_task_set_data(task, nullptr);
46 JNIEXPORT void JNICALL Java_org_simgrid_msg_Comm_nativeInit(JNIEnv *env, jclass cls) {
47 jclass jfield_class_Comm = env->FindClass("org/simgrid/msg/Comm");
48 if (!jfield_class_Comm) {
49 jxbt_throw_native(env,bprintf("Can't find the org/simgrid/msg/Comm class."));
52 jcomm_field_Comm_bind = jxbt_get_jfield(env, jfield_class_Comm, "bind", "J");
53 jcomm_field_Comm_taskBind = jxbt_get_jfield(env, jfield_class_Comm, "taskBind", "J");
54 jcomm_field_Comm_receiving = jxbt_get_jfield(env, jfield_class_Comm, "receiving", "Z");
55 jtask_field_Comm_task = jxbt_get_jfield(env, jfield_class_Comm, "task", "Lorg/simgrid/msg/Task;");
56 jcomm_field_Comm_finished = jxbt_get_jfield(env, jfield_class_Comm, "finished", "Z");
57 if (!jcomm_field_Comm_bind || !jcomm_field_Comm_taskBind || !jcomm_field_Comm_receiving || !jtask_field_Comm_task ||
58 !jcomm_field_Comm_finished) {
59 jxbt_throw_native(env,bprintf("Can't find some fields in Java class."));
63 JNIEXPORT void JNICALL Java_org_simgrid_msg_Comm_nativeFinalize(JNIEnv *env, jobject jcomm) {
65 msg_task_t *task_received;
67 task_received = (msg_task_t*) (uintptr_t) env->GetLongField(jcomm, jcomm_field_Comm_taskBind);
68 xbt_free(task_received);
70 comm = (msg_comm_t) (uintptr_t) env->GetLongField(jcomm, jcomm_field_Comm_bind);
71 MSG_comm_destroy(comm);
74 JNIEXPORT jboolean JNICALL Java_org_simgrid_msg_Comm_test(JNIEnv *env, jobject jcomm) {
76 comm = (msg_comm_t) (uintptr_t) env->GetLongField(jcomm, jcomm_field_Comm_bind);
78 jboolean finished = env->GetBooleanField(jcomm, jcomm_field_Comm_finished);
79 if (finished == JNI_TRUE) {
84 jxbt_throw_native(env,bprintf("comm is null"));
88 if (MSG_comm_test(comm)) {
89 msg_error_t status = MSG_comm_get_status(comm);
90 if (status == MSG_OK) {
91 jcomm_bind_task(env,jcomm);
94 //send the correct exception
95 jmsg_throw_status(env,status);
101 JNIEXPORT void JNICALL Java_org_simgrid_msg_Comm_waitCompletion(JNIEnv *env, jobject jcomm, jdouble timeout) {
102 msg_comm_t comm = (msg_comm_t) (uintptr_t) env->GetLongField(jcomm, jcomm_field_Comm_bind);
104 jxbt_throw_native(env,bprintf("comm is null"));
108 jboolean finished = env->GetBooleanField(jcomm, jcomm_field_Comm_finished);
109 if (finished == JNI_TRUE) {
114 status = MSG_comm_wait(comm,static_cast<double>(timeout));
115 env->SetBooleanField(jcomm, jcomm_field_Comm_finished, JNI_TRUE);
116 if (status == MSG_OK) {
117 jcomm_bind_task(env,jcomm);
120 jmsg_throw_status(env,status);
124 static msg_comm_t* jarray_to_commArray(JNIEnv *env, jobjectArray jcomms, /* OUT */ int *count)
126 *count = env->GetArrayLength(jcomms);
127 msg_comm_t* comms = xbt_new(msg_comm_t, *count);
129 for (int i=0; i < *count; i++) {
130 jobject jcomm = env->GetObjectArrayElement(jcomms, i);
131 if (env->ExceptionOccurred())
134 comms[i] = (msg_comm_t) (uintptr_t) env->GetLongField(jcomm, jcomm_field_Comm_bind);
136 jxbt_throw_native(env,bprintf("comm at rank %d is null",i));
140 env->DeleteLocalRef(jcomm); // reduce the load on the garbage collector: we don't need that object anymore
144 JNIEXPORT void JNICALL Java_org_simgrid_msg_Comm_waitAll(JNIEnv *env, jclass cls, jobjectArray jcomms, jdouble timeout)
147 msg_comm_t* comms = jarray_to_commArray(env, jcomms, &count);
151 MSG_comm_waitall(comms, count, static_cast<double>(timeout));
154 JNIEXPORT int JNICALL Java_org_simgrid_msg_Comm_waitAny(JNIEnv *env, jclass cls, jobjectArray jcomms)
157 msg_comm_t* comms = jarray_to_commArray(env, jcomms, &count);
160 xbt_dynar_t dyn = xbt_dynar_new(sizeof(msg_comm_t),nullptr);
161 for (int i=0; i<count; i++) {
162 xbt_dynar_push(dyn, &(comms[i]));
165 int rank = MSG_comm_waitany(dyn);
167 xbt_dynar_free(&dyn);