src/bindings/java/jmsg_host.cpp

   1 /* Functions related to the java host instances.                            */
   2
   3 /* Copyright (c) 2007-2018. The SimGrid Team. All rights reserved.          */
   4
   5 /* This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the license (GNU LGPL) which comes with this package. */
   7
   8 #include "simgrid/plugins/energy.h"
   9 #include "simgrid/s4u/Host.hpp"
  10 #include "simgrid/s4u/Storage.hpp"
  11
  12 #include "jmsg.hpp"
  13 #include "jmsg_host.h"
  14 #include "jmsg_storage.h"
  15 #include "jxbt_utilities.hpp"
  16
  17 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
  18
  19 static jmethodID jhost_method_Host_constructor;
  20 static jfieldID jhost_field_Host_bind;
  21 static jfieldID jhost_field_Host_name;
  22
  23 jobject jhost_new_instance(JNIEnv * env) {
  24   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Host");
  25   return env->NewObject(cls, jhost_method_Host_constructor);
  26 }
  27
  28 jobject jhost_ref(JNIEnv * env, jobject jhost) {
  29   return env->NewGlobalRef(jhost);
  30 }
  31
  32 void jhost_unref(JNIEnv * env, jobject jhost) {
  33   env->DeleteGlobalRef(jhost);
  34 }
  35
  36 void jhost_bind(jobject jhost, msg_host_t host, JNIEnv * env) {
  37   env->SetLongField(jhost, jhost_field_Host_bind, (jlong) (uintptr_t) (host));
  38 }
  39
  40 msg_host_t jhost_get_native(JNIEnv * env, jobject jhost) {
  41   return (msg_host_t) (uintptr_t) env->GetLongField(jhost, jhost_field_Host_bind);
  42 }
  43
  44 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_nativeInit(JNIEnv *env, jclass cls) {
  45   jclass class_Host = env->FindClass("org/simgrid/msg/Host");
  46   jhost_method_Host_constructor = env->GetMethodID(class_Host, "<init>", "()V");
  47   jhost_field_Host_bind = jxbt_get_jfield(env,class_Host, "bind", "J");
  48   jhost_field_Host_name = jxbt_get_jfield(env, class_Host, "name", "Ljava/lang/String;");
  49   xbt_assert(class_Host && jhost_field_Host_name && jhost_method_Host_constructor && jhost_field_Host_bind,
  50              "Native initialization of msg/Host failed. Please report that bug");
  51 }
  52
  53 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Host_getByName(JNIEnv * env, jclass cls, jstring jname) {
  54
  55   /* get the C string from the java string */
  56   if (jname == nullptr) {
  57     jxbt_throw_null(env, "No host can have a null name");
  58     return nullptr;
  59   }
  60   const char *name = env->GetStringUTFChars(jname, 0);
  61   /* get the host by name       (the hosts are created during the grid resolution) */
  62   msg_host_t host = MSG_host_by_name(name);
  63
  64   if (not host) { /* invalid name */
  65     jxbt_throw_host_not_found(env, name);
  66     env->ReleaseStringUTFChars(jname, name);
  67     return nullptr;
  68   }
  69   env->ReleaseStringUTFChars(jname, name);
  70
  71   if (not host->extension(JAVA_HOST_LEVEL)) { /* native host not associated yet with java host */
  72     /* Instantiate a new java host */
  73     jobject jhost = jhost_new_instance(env);
  74
  75     if (not jhost) {
  76       jxbt_throw_jni(env, "java host instantiation failed");
  77       return nullptr;
  78     }
  79
  80     /* get a global reference to the newly created host */
  81     jhost = jhost_ref(env, jhost);
  82
  83     if (not jhost) {
  84       jxbt_throw_jni(env, "new global ref allocation failed");
  85       return nullptr;
  86     }
  87     /* Sets the java host name */
  88     env->SetObjectField(jhost, jhost_field_Host_name, jname);
  89     /* bind the java host and the native host */
  90     jhost_bind(jhost, host, env);
  91
  92     /* the native host data field is set with the global reference to the java host returned by this function */
  93     host->extension_set(JAVA_HOST_LEVEL, (void *)jhost);
  94   }
  95
  96   /* return the global reference to the java host instance */
  97   return (jobject) host->extension(JAVA_HOST_LEVEL);
  98 }
  99
 100 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Host_currentHost(JNIEnv * env, jclass cls) {
 101   jobject jhost;
 102
 103   msg_host_t host = MSG_host_self();
 104
 105   if (not host->extension(JAVA_HOST_LEVEL)) {
 106     /* the native host not yet associated with the java host instance */
 107
 108     /* instanciate a new java host instance */
 109     jhost = jhost_new_instance(env);
 110
 111     if (not jhost) {
 112       jxbt_throw_jni(env, "java host instantiation failed");
 113       return nullptr;
 114     }
 115
 116     /* get a global reference to the newly created host */
 117     jhost = jhost_ref(env, jhost);
 118
 119     if (not jhost) {
 120       jxbt_throw_jni(env, "global ref allocation failed");
 121       return nullptr;
 122     }
 123     /* Sets the host name */
 124     jobject jname = env->NewStringUTF(host->get_cname());
 125     env->SetObjectField(jhost, jhost_field_Host_name, jname);
 126     /* Bind & store it */
 127     jhost_bind(jhost, host, env);
 128     host->extension_set(JAVA_HOST_LEVEL, (void *) jhost);
 129   } else {
 130     jhost = (jobject) host->extension(JAVA_HOST_LEVEL);
 131   }
 132
 133   return jhost;
 134 }
 135
 136 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_on(JNIEnv *env, jobject jhost) {
 137   msg_host_t host = jhost_get_native(env, jhost);
 138   MSG_host_on(host);
 139 }
 140
 141 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_off(JNIEnv *env, jobject jhost) {
 142   msg_host_t host = jhost_get_native(env, jhost);
 143   MSG_host_off(host);
 144 }
 145
 146 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Host_getCount(JNIEnv * env, jclass cls) {
 147   xbt_dynar_t hosts =  MSG_hosts_as_dynar();
 148   int nb_host = xbt_dynar_length(hosts);
 149   xbt_dynar_free(&hosts);
 150   return (jint) nb_host;
 151 }
 152
 153 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getSpeed(JNIEnv * env, jobject jhost) {
 154   msg_host_t host = jhost_get_native(env, jhost);
 155
 156   if (not host) {
 157     jxbt_throw_notbound(env, "host", jhost);
 158     return -1;
 159   }
 160
 161   return (jdouble) MSG_host_get_speed(host);
 162 }
 163
 164 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getCoreNumber(JNIEnv * env, jobject jhost) {
 165   msg_host_t host = jhost_get_native(env, jhost);
 166
 167   if (not host) {
 168     jxbt_throw_notbound(env, "host", jhost);
 169     return -1;
 170   }
 171
 172   return (jdouble) MSG_host_get_core_number(host);
 173 }
 174
 175 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Host_getProperty(JNIEnv *env, jobject jhost, jobject jname) {
 176   msg_host_t host = jhost_get_native(env, jhost);
 177
 178   if (not host) {
 179     jxbt_throw_notbound(env, "host", jhost);
 180     return nullptr;
 181   }
 182   const char *name = env->GetStringUTFChars((jstring) jname, 0);
 183
 184   const char *property = MSG_host_get_property_value(host, name);
 185   if (not property) {
 186     return nullptr;
 187   }
 188
 189   jobject jproperty = env->NewStringUTF(property);
 190
 191   env->ReleaseStringUTFChars((jstring) jname, name);
 192
 193   return jproperty;
 194 }
 195
 196 JNIEXPORT void JNICALL
 197 Java_org_simgrid_msg_Host_setProperty(JNIEnv *env, jobject jhost, jobject jname, jobject jvalue) {
 198   msg_host_t host = jhost_get_native(env, jhost);
 199
 200   if (not host) {
 201     jxbt_throw_notbound(env, "host", jhost);
 202     return;
 203   }
 204   const char *name = env->GetStringUTFChars((jstring) jname, 0);
 205   const char *value_java = env->GetStringUTFChars((jstring) jvalue, 0);
 206   char *value = xbt_strdup(value_java);
 207
 208   MSG_host_set_property_value(host, name, value);
 209
 210   env->ReleaseStringUTFChars((jstring) jvalue, value_java);
 211   env->ReleaseStringUTFChars((jstring) jname, name);
 212 }
 213
 214 JNIEXPORT jboolean JNICALL Java_org_simgrid_msg_Host_isOn(JNIEnv * env, jobject jhost)
 215 {
 216   msg_host_t host = jhost_get_native(env, jhost);
 217
 218   if (not host) {
 219     jxbt_throw_notbound(env, "host", jhost);
 220     return 0;
 221   }
 222
 223   return (jboolean) MSG_host_is_on(host);
 224 }
 225
 226 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_getMountedStorage(JNIEnv * env, jobject jhost)
 227 {
 228   msg_host_t host = jhost_get_native(env, jhost);
 229   jobject jstorage;
 230   jstring jname;
 231
 232   if (not host) {
 233     jxbt_throw_notbound(env, "host", jhost);
 234     return 0;
 235   }
 236
 237   int index = 0;
 238   jobjectArray jtable;
 239   std::unordered_map<std::string, msg_storage_t> mounted_storages = host->get_mounted_storages();
 240   int count  = mounted_storages.size();
 241   jclass cls = env->FindClass("org/simgrid/msg/Storage");
 242
 243   jtable = env->NewObjectArray((jsize) count, cls, nullptr);
 244
 245   if (not jtable) {
 246     jxbt_throw_jni(env, "Storages table allocation failed");
 247     return nullptr;
 248   }
 249
 250   for (auto const& elm : mounted_storages) {
 251     jname    = env->NewStringUTF(elm.second->get_cname());
 252     jstorage = Java_org_simgrid_msg_Storage_getByName(env,cls,jname);
 253     env->SetObjectArrayElement(jtable, index, jstorage);
 254     index++;
 255   }
 256   return jtable;
 257 }
 258
 259 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_getAttachedStorage(JNIEnv * env, jobject jhost)
 260 {
 261   msg_host_t host = jhost_get_native(env, jhost);
 262
 263   if (not host) {
 264     jxbt_throw_notbound(env, "host", jhost);
 265     return 0;
 266   }
 267   jobjectArray jtable;
 268
 269   xbt_dynar_t dyn = sg_host_get_attached_storage_list(host);
 270   jclass cls      = jxbt_get_class(env, "java/lang/String");
 271   jtable          = env->NewObjectArray(static_cast<jsize>(xbt_dynar_length(dyn)), cls, nullptr);
 272   unsigned int index;
 273   const char* storage_name;
 274   jstring jstorage_name;
 275   xbt_dynar_foreach (dyn, index, storage_name) {
 276     jstorage_name = env->NewStringUTF(storage_name);
 277     env->SetObjectArrayElement(jtable, index, jstorage_name);
 278   }
 279   xbt_dynar_free_container(&dyn);
 280   return jtable;
 281 }
 282
 283 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_getStorageContent(JNIEnv * env, jobject jhost)
 284 {
 285   msg_host_t host = jhost_get_native(env, jhost);
 286
 287   if (not host) {
 288     jxbt_throw_notbound(env, "host", jhost);
 289     return 0;
 290   }
 291   return (jobjectArray)MSG_host_get_storage_content(host);
 292 }
 293
 294 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_all(JNIEnv * env, jclass cls_arg)
 295 {
 296
 297   xbt_dynar_t table =  MSG_hosts_as_dynar();
 298   int count = xbt_dynar_length(table);
 299
 300   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Host");
 301   if (not cls)
 302     return nullptr;
 303
 304   jobjectArray jtable = env->NewObjectArray((jsize)count, cls, nullptr);
 305
 306   if (not jtable) {
 307     jxbt_throw_jni(env, "Hosts table allocation failed");
 308     return nullptr;
 309   }
 310
 311   for (int index = 0; index < count; index++) {
 312     msg_host_t host = xbt_dynar_get_as(table, index, msg_host_t);
 313     jobject jhost   = static_cast<jobject>(host->extension(JAVA_HOST_LEVEL));
 314
 315     if (not jhost) {
 316       jstring jname = env->NewStringUTF(host->get_cname());
 317       jhost         = Java_org_simgrid_msg_Host_getByName(env, cls_arg, jname);
 318     }
 319
 320     env->SetObjectArrayElement(jtable, index, jhost);
 321   }
 322   xbt_dynar_free(&table);
 323   return jtable;
 324 }
 325
 326 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_setAsyncMailbox(JNIEnv * env, jclass cls_arg, jobject jname)
 327 {
 328   const char *name = env->GetStringUTFChars((jstring) jname, 0);
 329   sg_mailbox_set_receiver(name);
 330   env->ReleaseStringUTFChars((jstring) jname, name);
 331 }
 332
 333 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_updateAllEnergyConsumptions(JNIEnv* env, jclass cls)
 334 {
 335   sg_host_energy_update_all();
 336 }
 337
 338 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getConsumedEnergy (JNIEnv *env, jobject jhost)
 339 {
 340   msg_host_t host = jhost_get_native(env, jhost);
 341
 342   if (not host) {
 343     jxbt_throw_notbound(env, "host", jhost);
 344     return 0;
 345   }
 346
 347   return MSG_host_get_consumed_energy(host);
 348 }
 349
 350 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_setPstate(JNIEnv* env, jobject jhost, jint pstate)
 351 {
 352   msg_host_t host = jhost_get_native(env, jhost);
 353   MSG_host_set_pstate(host, pstate);
 354 }
 355 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Host_getPstate(JNIEnv* env, jobject jhost)
 356 {
 357   msg_host_t host = jhost_get_native(env, jhost);
 358   return MSG_host_get_pstate(host);
 359 }
 360 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Host_getPstatesCount(JNIEnv* env, jobject jhost)
 361 {
 362   msg_host_t host = jhost_get_native(env, jhost);
 363   return MSG_host_get_nb_pstates(host);
 364 }
 365 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getCurrentPowerPeak(JNIEnv* env, jobject jhost)
 366 {
 367   msg_host_t host = jhost_get_native(env, jhost);
 368   return MSG_host_get_speed(host);
 369 }
 370 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getPowerPeakAt(JNIEnv* env, jobject jhost, jint pstate)
 371 {
 372   msg_host_t host = jhost_get_native(env, jhost);
 373   return MSG_host_get_power_peak_at(host, pstate);
 374 }