src/bindings/ruby/rb_msg_task.c

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