src/jmsg_task.c

   1 /* Functions related to the java task instances.                            */
   2
   3 /* Copyright (c) 2007, 2009, 2010. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   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. */
   8
   9 #include "jmsg.h"
  10
  11 #include "smx_context_java.h"
  12
  13 #include "jmsg_host.h"
  14 #include "jmsg_task.h"
  15
  16 #include "jxbt_utilities.h"
  17
  18 #include <msg/msg.h>
  19 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
  20
  21 static jmethodID jtask_field_Comm_constructor;
  22
  23 static jfieldID jtask_field_Task_bind;
  24 static jfieldID jtask_field_Comm_bind;
  25 static jfieldID jtask_field_Comm_taskBind;
  26 static jfieldID jtask_field_Comm_receiving;
  27
  28 void jtask_bind(jobject jtask, m_task_t task, JNIEnv * env)
  29 {
  30   (*env)->SetLongField(env, jtask, jtask_field_Task_bind, (jlong) (long) (task));
  31 }
  32
  33 m_task_t jtask_to_native_task(jobject jtask, JNIEnv * env)
  34 {
  35   return (m_task_t) (long) (*env)->GetLongField(env, jtask, jtask_field_Task_bind);
  36 }
  37
  38 jboolean jtask_is_valid(jobject jtask, JNIEnv * env)
  39 {
  40   return (*env)->GetLongField(env, jtask, jtask_field_Task_bind) ? JNI_TRUE : JNI_FALSE;
  41 }
  42
  43 JNIEXPORT void JNICALL
  44 Java_org_simgrid_msg_Task_nativeInit(JNIEnv *env, jclass cls) {
  45         jclass jtask_class_Comm = (*env)->FindClass(env, "org/simgrid/msg/Comm");
  46
  47         jtask_field_Comm_constructor = (*env)->GetMethodID(env, jtask_class_Comm, "<init>", "()V");
  48         //FIXME: Don't use jxbt_get_sfield directly.
  49         jtask_field_Task_bind = jxbt_get_sfield(env, "org/simgrid/msg/Task", "bind", "J");
  50         jtask_field_Comm_bind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
  51         jtask_field_Comm_taskBind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "taskBind", "J");
  52         jtask_field_Comm_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
  53 }
  54
  55 JNIEXPORT void JNICALL
  56 Java_org_simgrid_msg_Task_create(JNIEnv * env,
  57                                       jobject jtask, jstring jname,
  58                                       jdouble jcomputeDuration,
  59                                       jdouble jmessageSize)
  60 {
  61   m_task_t task;                /* the native task to create                            */
  62   const char *name = NULL;      /* the name of the task                                 */
  63
  64   if (jcomputeDuration < 0) {
  65     jxbt_throw_illegal(env,
  66                        bprintf
  67                        ("Task ComputeDuration (%f) cannot be negative",
  68                         (double) jcomputeDuration));
  69     return;
  70   }
  71
  72   if (jmessageSize < 0) {
  73     jxbt_throw_illegal(env,
  74                        bprintf("Task MessageSize (%f) cannot be negative",
  75                                (double) jmessageSize));
  76     return;
  77   }
  78
  79   if (jname) {
  80     /* get the C string from the java string */
  81     name = (*env)->GetStringUTFChars(env, jname, 0);
  82   }
  83
  84
  85   /* create the task */
  86   task =
  87       MSG_task_create(name, (double) jcomputeDuration,
  88                       (double) jmessageSize, NULL);
  89   if (jname)
  90     (*env)->ReleaseStringUTFChars(env, jname, name);
  91
  92   /* bind & store the task */
  93   jtask_bind(jtask, task, env);
  94   MSG_task_set_data(task, jtask);
  95 }
  96
  97 JNIEXPORT void JNICALL
  98 Java_org_simgrid_msg_Task_parallelCreate(JNIEnv * env,
  99                                                jobject jtask,
 100                                                jstring jname,
 101                                                jobjectArray jhosts,
 102                                                jdoubleArray
 103                                                jcomputeDurations_arg,
 104                                                jdoubleArray
 105                                                jmessageSizes_arg) {
 106
 107   m_task_t task;                /* the native parallel task to create           */
 108   const char *name;             /* the name of the task                         */
 109   int host_count;
 110   m_host_t *hosts;
 111   double *computeDurations;
 112   double *messageSizes;
 113   jdouble *jcomputeDurations;
 114   jdouble *jmessageSizes;
 115
 116   jobject jhost;
 117   int index;
 118
 119
 120   if (!jcomputeDurations_arg) {
 121     jxbt_throw_null(env,
 122                     xbt_strdup
 123                     ("Parallel task compute durations cannot be null"));
 124     return;
 125   }
 126
 127   if (!jmessageSizes_arg) {
 128     jxbt_throw_null(env,
 129                     xbt_strdup
 130                     ("Parallel task message sizes cannot be null"));
 131     return;
 132   }
 133
 134   if (!jname) {
 135     jxbt_throw_null(env, xbt_strdup("Parallel task name cannot be null"));
 136     return;
 137   }
 138
 139   host_count = (int) (*env)->GetArrayLength(env, jhosts);
 140
 141
 142   hosts = xbt_new0(m_host_t, host_count);
 143   computeDurations = xbt_new0(double, host_count);
 144   messageSizes = xbt_new0(double, host_count * host_count);
 145
 146   jcomputeDurations =
 147       (*env)->GetDoubleArrayElements(env, jcomputeDurations_arg, 0);
 148   jmessageSizes =
 149       (*env)->GetDoubleArrayElements(env, jmessageSizes_arg, 0);
 150
 151   for (index = 0; index < host_count; index++) {
 152     jhost = (*env)->GetObjectArrayElement(env, jhosts, index);
 153     hosts[index] = jhost_get_native(env, jhost);
 154     computeDurations[index] = jcomputeDurations[index];
 155   }
 156   for (index = 0; index < host_count * host_count; index++) {
 157     messageSizes[index] = jmessageSizes[index];
 158   }
 159
 160   (*env)->ReleaseDoubleArrayElements(env, jcomputeDurations_arg,
 161                                      jcomputeDurations, 0);
 162   (*env)->ReleaseDoubleArrayElements(env, jmessageSizes_arg, jmessageSizes,
 163                                      0);
 164
 165
 166   /* get the C string from the java string */
 167   name = (*env)->GetStringUTFChars(env, jname, 0);
 168
 169   task =
 170       MSG_parallel_task_create(name, host_count, hosts, computeDurations,
 171                                messageSizes, NULL);
 172
 173   (*env)->ReleaseStringUTFChars(env, jname, name);
 174
 175   /* associate the java task object and the native task */
 176   jtask_bind(jtask, task, env);
 177
 178   MSG_task_set_data(task, (void *) jtask);
 179
 180   if (!MSG_task_get_data(task))
 181     jxbt_throw_jni(env, "global ref allocation failed");
 182 }
 183
 184 JNIEXPORT jstring JNICALL
 185 Java_org_simgrid_msg_Task_getName(JNIEnv * env,
 186                                        jobject jtask) {
 187   m_task_t task = jtask_to_native_task(jtask, env);
 188
 189   if (!task) {
 190     jxbt_throw_notbound(env, "task", jtask);
 191     return NULL;
 192   }
 193
 194   return (*env)->NewStringUTF(env, MSG_task_get_name(task));
 195 }
 196
 197 JNIEXPORT jobject JNICALL
 198 Java_org_simgrid_msg_Task_getSender(JNIEnv * env,
 199                                          jobject jtask) {
 200   m_process_t process;
 201
 202   m_task_t task = jtask_to_native_task(jtask, env);
 203
 204   if (!task) {
 205     jxbt_throw_notbound(env, "task", jtask);
 206     return NULL;
 207   }
 208
 209   process = MSG_task_get_sender(task);
 210   return (jobject) native_to_java_process(process);
 211 }
 212
 213 JNIEXPORT jobject JNICALL
 214 Java_org_simgrid_msg_Task_irecv(JNIEnv * env, jclass cls, jstring jmailbox) {
 215         msg_comm_t comm;
 216         const char *mailbox;
 217         jclass comm_class;
 218         //pointer to store the task object pointer.
 219         m_task_t *task = xbt_new(m_task_t,1);
 220         *task = NULL;
 221         /* There should be a cache here */
 222         comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
 223
 224         if (!comm_class) {
 225                 jxbt_throw_native(env,bprintf("fieldID or methodID or class not found."));
 226                 return NULL;
 227         }
 228
 229         jobject jcomm = (*env)->NewObject(env, comm_class, jtask_field_Comm_constructor);
 230         if (!jcomm) {
 231                 jxbt_throw_native(env,bprintf("Can't create a Comm object."));
 232                 return NULL;
 233         }
 234
 235         mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
 236
 237         comm = MSG_task_irecv(task,mailbox);
 238
 239         (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
 240         (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(task));
 241         (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_TRUE);
 242
 243         (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
 244
 245         return jcomm;
 246 }
 247
 248 JNIEXPORT jobject JNICALL
 249 Java_org_simgrid_msg_Task_isend(JNIEnv *env, jobject jtask, jstring jmailbox) {
 250         jclass comm_class;
 251
 252         const char *mailbox;
 253
 254         m_task_t task;
 255
 256         jobject jcomm;
 257         msg_comm_t comm;
 258
 259         comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
 260
 261         if (!comm_class) return NULL;
 262
 263         jcomm = (*env)->NewObject(env, comm_class, jtask_field_Comm_constructor);
 264         mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
 265
 266         task = jtask_to_native_task(jtask, env);
 267
 268         if (!task) {
 269     (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
 270     (*env)->DeleteLocalRef(env, jcomm);
 271     jxbt_throw_notbound(env, "task", jtask);
 272                 return NULL;
 273         }
 274
 275   MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
 276         comm = MSG_task_isend(task,mailbox);
 277
 278         (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
 279         (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
 280         (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
 281
 282         (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
 283
 284         return jcomm;
 285 }
 286
 287 static void msg_task_cancel_on_failed_dsend(void*t) {
 288         m_task_t task = t;
 289         JNIEnv *env =get_current_thread_env();
 290         jobject jtask_global = MSG_task_get_data(task);
 291
 292         /* Destroy the global ref so that the JVM can free the stuff */
 293         (*env)->DeleteGlobalRef(env, jtask_global);
 294         MSG_task_set_data(task, NULL);
 295         MSG_task_destroy(task);
 296 }
 297
 298 JNIEXPORT void JNICALL
 299 Java_org_simgrid_msg_Task_dsend(JNIEnv * env, jobject jtask,
 300                                 jstring jalias) {
 301
 302   const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
 303
 304   m_task_t task = jtask_to_native_task(jtask, env);
 305
 306
 307   if (!task) {
 308     (*env)->ReleaseStringUTFChars(env, jalias, alias);
 309     jxbt_throw_notbound(env, "task", jtask);
 310     return;
 311   }
 312
 313   /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
 314   MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
 315   MSG_task_dsend(task, alias, msg_task_cancel_on_failed_dsend);
 316
 317   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 318 }
 319
 320 JNIEXPORT jboolean JNICALL
 321 Java_org_simgrid_msg_Task_listen(JNIEnv * env, jclass cls,
 322                                                                                                            jstring jalias) {
 323
 324   const char *alias;
 325   int rv;
 326
 327   alias = (*env)->GetStringUTFChars(env, jalias, 0);
 328
 329   rv = MSG_task_listen(alias);
 330
 331   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 332
 333   return (jboolean) rv;
 334 }
 335
 336 JNIEXPORT jint JNICALL
 337 Java_org_simgrid_msg_Task_listenFromHost(JNIEnv * env, jclass cls,
 338                                                                                                                                            jstring jalias,
 339                                                                                                                                            jobject jhost) {
 340   int rv;
 341   const char *alias;
 342
 343   m_host_t host = jhost_get_native(env, jhost);
 344
 345   if (!host) {
 346     jxbt_throw_notbound(env, "host", jhost);
 347     return -1;
 348   }
 349   alias = (*env)->GetStringUTFChars(env, jalias, 0);
 350
 351   rv = MSG_task_listen_from_host(alias, host);
 352
 353   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 354
 355   return (jint) rv;
 356 }
 357
 358
 359 JNIEXPORT jint JNICALL
 360 Java_org_simgrid_msg_Task_listenFrom(JNIEnv * env, jclass cls,
 361                                                         jstring jalias) {
 362
 363   int rv;
 364   const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
 365
 366   rv = MSG_task_listen_from(alias);
 367
 368   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 369
 370   return (jint) rv;
 371 }