src/bindings/ruby/rb_msg_task.c

   1 /* Task-related bindings to ruby  */
   2
   3 /* Copyright (c) 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
  10 #include "bindings/ruby_bindings.h"
  11
  12 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(ruby);
  13
  14 // Free Method
  15 void rb_task_free(m_task_t tk) {
  16   //MSG_task_destroy(tk); ( This cause a bug !! is it really necessary ?!! not really sure !! )
  17 }
  18
  19 // New Method
  20 VALUE rb_task_new(VALUE class, VALUE name,VALUE comp_size,VALUE comm_size) {
  21   m_task_t task = MSG_task_create(RSTRING(name)->ptr,NUM2INT(comp_size),NUM2INT(comm_size),NULL);
  22   rb_data_t data = malloc(sizeof(s_ruby_data_t));
  23   data->ruby_task = NULL;
  24   data->user_data = NULL;
  25   MSG_task_set_data(task,(void*)data);
  26   // Wrap m_task_t to a Ruby Value
  27   return Data_Wrap_Struct(class, 0, rb_task_free, task);
  28 }
  29
  30 //Get Computation Size
  31 VALUE rb_task_comp(VALUE class,VALUE task) {
  32   double size;
  33   m_task_t tk;
  34   // Wrap Ruby Value to m_task_t struct
  35   Data_Get_Struct(task, s_m_task_t, tk);
  36   size = MSG_task_get_compute_duration(tk);
  37   return rb_float_new(size);
  38 }
  39
  40 //Get Name
  41 VALUE rb_task_name(VALUE class,VALUE task) {
  42
  43   // Wrap Ruby Value to m_task_t struct
  44   m_task_t tk;
  45   Data_Get_Struct(task, s_m_task_t, tk);
  46   return rb_str_new2(MSG_task_get_name(tk));
  47 }
  48
  49 // Execute Task
  50 VALUE rb_task_execute(VALUE class,VALUE task) {
  51
  52   // Wrap Ruby Value to m_task_t struct
  53   m_task_t tk;
  54   Data_Get_Struct(task, s_m_task_t, tk);
  55   return INT2NUM(MSG_task_execute(tk));
  56 }
  57
  58 // Sending Task
  59 void rb_task_send(VALUE class,VALUE task,VALUE mailbox) {
  60
  61   MSG_error_t rv;
  62   rb_data_t data;
  63   // Wrap Ruby Value to m_task_t struct
  64   m_task_t tk;
  65   Data_Get_Struct(task, s_m_task_t, tk);
  66   data = MSG_task_get_data(tk);
  67   data->ruby_task =(void*)task;
  68   MSG_task_set_data(tk,(void*)data);
  69   DEBUG1("Sending task %p",tk);
  70   rv = MSG_task_send(tk,RSTRING(mailbox)->ptr);
  71   if(rv != MSG_OK)
  72   {
  73     if (rv == MSG_TRANSFER_FAILURE )
  74       rb_raise(rb_eRuntimeError,"Transfer failure while Sending");
  75     else if ( rv == MSG_HOST_FAILURE )
  76       rb_raise(rb_eRuntimeError,"Host failure while Sending");
  77     else if ( rv == MSG_TIMEOUT )
  78       rb_raise(rb_eRuntimeError,"Timeout failure while Sending");
  79     else
  80       rb_raise(rb_eRuntimeError,"MSG_task_send failed");
  81   }
  82 }
  83
  84 // Receiving Task (returns a Task)
  85 VALUE rb_task_receive(VALUE class, VALUE mailbox) {
  86   // We must put the location where we copy the task
  87   // pointer to on the heap, because the stack may move
  88   // during the context switches (damn ruby internals)
  89   m_task_t *ptask = malloc(sizeof(m_task_t));
  90   m_task_t task;
  91   *ptask = NULL;
  92   rb_data_t data =NULL;
  93   DEBUG2("Receiving a task on mailbox '%s', store it into %p",RSTRING(mailbox)->ptr,&task);
  94   MSG_task_receive(ptask,RSTRING(mailbox)->ptr);
  95   task = *ptask;
  96   free(ptask);
  97   data = MSG_task_get_data(task);
  98   if(data==NULL) printf("Empty task while receving");
  99   return (VALUE)data->ruby_task;
 100 }
 101
 102 // It Return a Native Process ( m_process_t )
 103 VALUE rb_task_sender(VALUE class,VALUE task) {
 104   m_task_t tk;
 105   Data_Get_Struct(task,s_m_task_t,tk);
 106   THROW_UNIMPLEMENTED;
 107   return 0;//MSG_task_get_sender(tk);
 108 }
 109
 110 // it return a Host
 111 VALUE rb_task_source(VALUE class,VALUE task) {
 112   m_task_t tk;
 113   Data_Get_Struct(task,s_m_task_t,tk);
 114
 115   m_host_t host = MSG_task_get_source(tk);
 116   if(!host->data) {
 117     rb_raise(rb_eRuntimeError,"MSG_task_get_source() failed");
 118     return Qnil;
 119   }
 120   THROW_UNIMPLEMENTED;
 121   return 0;//host;
 122 }
 123
 124 // Return Boolean
 125 VALUE rb_task_listen(VALUE class,VALUE task,VALUE alias) {
 126   m_task_t tk;
 127   const char *p_alias;
 128   int rv;
 129
 130   Data_Get_Struct(task,s_m_task_t,tk);
 131   p_alias = RSTRING(alias)->ptr;
 132
 133   rv = MSG_task_listen(p_alias);
 134
 135   if(rv) return Qtrue;
 136
 137   return Qfalse;
 138 }
 139
 140 // return Boolean
 141 VALUE rb_task_listen_host(VALUE class,VALUE task,VALUE alias,VALUE host) {
 142
 143   m_task_t tk;
 144   m_host_t ht;
 145   const char *p_alias;
 146   int rv;
 147
 148   Data_Get_Struct(task,s_m_task_t,tk);
 149   Data_Get_Struct(host,s_m_host_t,ht);
 150   p_alias = RSTRING(alias)->ptr;
 151   rv = MSG_task_listen_from_host(p_alias,ht);
 152   if (rv)
 153     return Qtrue;
 154   return Qfalse;
 155 }
 156
 157
 158 // Set Priority
 159 void rb_task_set_priority(VALUE class,VALUE task,VALUE priority)
 160 {
 161
 162   m_task_t tk;
 163   double prt = NUM2DBL(priority);
 164   Data_Get_Struct(task,s_m_task_t,tk);
 165   MSG_task_set_priority(tk,prt);
 166
 167 }
 168
 169 // Cancel
 170 void rb_task_cancel(VALUE class,VALUE task)
 171 {
 172  m_task_t tk;
 173  Data_Get_Struct(task,s_m_task_t,tk);
 174  MSG_task_cancel(tk);
 175
 176 }
 177
 178 void rb_task_set_data(VALUE class,VALUE task,VALUE data)
 179 {
 180  m_task_t tk;
 181  rb_data_t rb_data;
 182  Data_Get_Struct(task,s_m_task_t,tk);
 183  rb_data = MSG_task_get_data(tk);
 184  rb_data->user_data = (void*)data;
 185  MSG_task_set_data(tk,(void*)rb_data);
 186
 187 }
 188
 189 VALUE rb_task_get_data(VALUE class,VALUE task)
 190 {
 191  m_task_t tk;
 192  Data_Get_Struct(task,s_m_task_t,tk);
 193  rb_data_t rb_data = MSG_task_get_data(tk);
 194  if(!rb_data->user_data)
 195          ERROR1("the task %s contain no user data",MSG_task_get_name(tk));
 196
 197  return (VALUE)rb_data->user_data;
 198 }
 199
 200 VALUE rb_task_has_data(VALUE class,VALUE task)
 201 {
 202  m_task_t tk;
 203  Data_Get_Struct(task,s_m_task_t,tk);
 204  rb_data_t rb_data = MSG_task_get_data(tk);
 205  if(!rb_data->user_data)
 206          return Qfalse;
 207  return Qtrue;
 208 }
 209