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   // Wrap m_task_t to a Ruby Value
  22   return Data_Wrap_Struct(class, 0, rb_task_free, task);
  23
  24 }
  25
  26 //Get Computation Size
  27 VALUE rb_task_comp(VALUE class,VALUE task) {
  28   double size;
  29   m_task_t tk;
  30   // Wrap Ruby Value to m_task_t struct
  31   Data_Get_Struct(task, s_m_task_t, tk);
  32   size = MSG_task_get_compute_duration(tk);
  33   return rb_float_new(size);
  34 }
  35
  36 //Get Name
  37 VALUE rb_task_name(VALUE class,VALUE task) {
  38
  39   // Wrap Ruby Value to m_task_t struct
  40   m_task_t tk;
  41   Data_Get_Struct(task, s_m_task_t, tk);
  42   return rb_str_new2(MSG_task_get_name(tk));
  43 }
  44
  45 // Execute Task
  46 VALUE rb_task_execute(VALUE class,VALUE task) {
  47
  48   // Wrap Ruby Value to m_task_t struct
  49   m_task_t tk;
  50   Data_Get_Struct(task, s_m_task_t, tk);
  51   return INT2NUM(MSG_task_execute(tk));
  52 }
  53
  54 // Sending Task
  55 void rb_task_send(VALUE class,VALUE task,VALUE mailbox) {
  56
  57   MSG_error_t rv;
  58   // Wrap Ruby Value to m_task_t struct
  59   m_task_t tk;
  60   Data_Get_Struct(task, s_m_task_t, tk);
  61   INFO1("Sending task %p",tk);
  62   rv = MSG_task_send(tk,RSTRING(mailbox)->ptr);
  63   if(rv != MSG_OK)
  64   {
  65     if (rv == MSG_TRANSFER_FAILURE )
  66       rb_raise(rb_eRuntimeError,"Transfer failure while Sending");
  67     else if ( rv == MSG_HOST_FAILURE )
  68       rb_raise(rb_eRuntimeError,"Host failure while Sending");
  69     else if ( rv == MSG_TIMEOUT_FAILURE )
  70       rb_raise(rb_eRuntimeError,"Timeout failure while Sending");
  71     else
  72       rb_raise(rb_eRuntimeError,"MSG_task_send failed");
  73   }
  74 }
  75
  76 // Receiving Task (returns a Task)
  77 VALUE rb_task_receive(VALUE class, VALUE mailbox) {
  78   // We must put the location where we copy the task
  79   // pointer to on the heap, because the stack may move
  80   // during the context switches (damn ruby internals)
  81   m_task_t *ptask = malloc(sizeof(m_task_t));
  82   m_task_t task;
  83   *ptask = NULL;
  84   INFO2("Receiving a task on mailbox '%s', store it into %p",RSTRING(mailbox)->ptr,&task);
  85   MSG_task_receive(ptask,RSTRING(mailbox)->ptr);
  86   task = *ptask;
  87   free(ptask);
  88   return Data_Wrap_Struct(class, 0, rb_task_free, task);
  89 }
  90
  91 // It Return a Native Process ( m_process_t )
  92 VALUE rb_task_sender(VALUE class,VALUE task) {
  93   m_task_t tk;
  94   Data_Get_Struct(task,s_m_task_t,tk);
  95   THROW_UNIMPLEMENTED;
  96   return 0;//MSG_task_get_sender(tk);
  97 }
  98
  99 // it return a Host
 100 VALUE rb_task_source(VALUE class,VALUE task) {
 101   m_task_t tk;
 102   Data_Get_Struct(task,s_m_task_t,tk);
 103
 104   m_host_t host = MSG_task_get_source(tk);
 105   if(!host->data) {
 106     rb_raise(rb_eRuntimeError,"MSG_task_get_source() failed");
 107     return Qnil;
 108   }
 109   THROW_UNIMPLEMENTED;
 110   return 0;//host;
 111 }
 112
 113 // Return Boolean
 114 VALUE rb_task_listen(VALUE class,VALUE task,VALUE alias) {
 115   m_task_t tk;
 116   const char *p_alias;
 117   int rv;
 118
 119   Data_Get_Struct(task,s_m_task_t,tk);
 120   p_alias = RSTRING(alias)->ptr;
 121
 122   rv = MSG_task_listen(p_alias);
 123
 124   if(rv) return Qtrue;
 125
 126   return Qfalse;
 127 }
 128
 129 // return Boolean
 130 VALUE rb_task_listen_host(VALUE class,VALUE task,VALUE alias,VALUE host) {
 131
 132   m_task_t tk;
 133   m_host_t ht;
 134   const char *p_alias;
 135   int rv;
 136
 137   Data_Get_Struct(task,s_m_task_t,tk);
 138   Data_Get_Struct(host,s_m_host_t,ht);
 139   p_alias = RSTRING(alias)->ptr;
 140
 141   rv = MSG_task_listen_from_host(p_alias,ht);
 142
 143   if (rv)
 144     return Qtrue;
 145   return Qfalse;
 146 }