Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
ea770cc9dc338b6404042f87645f57240a211181
[simgrid.git] / 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 }