Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
fix the logic of refcount in msg_task_t by ... killing it. It's now a boolean indicat...
[simgrid.git] / src / msg / task.c
1 /* Copyright (c) 2004, 2005, 2006, 2007, 2008, 2009, 2010. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "msg/private.h"
8 #include "xbt/sysdep.h"
9 #include "xbt/log.h"
10
11 /** \defgroup m_task_management Managing functions of Tasks
12  *  \brief This section describes the task structure of MSG
13  *  (#m_task_t) and the functions for managing it.
14  */
15 /** @addtogroup m_task_management
16  *    \htmlonly <!-- DOXYGEN_NAVBAR_LABEL="Tasks" --> \endhtmlonly
17  * 
18  *  Since most scheduling algorithms rely on a concept of task
19  *  that can be either <em>computed</em> locally or
20  *  <em>transferred</em> on another processor, it seems to be the
21  *  right level of abstraction for our purposes. A <em>task</em>
22  *  may then be defined by a <em>computing amount</em>, a
23  *  <em>message size</em> and some <em>private data</em>.
24  */
25
26 #ifdef HAVE_RUBY                /* FIXME: KILLME */
27 XBT_LOG_EXTERNAL_CATEGORY(ruby);
28 #endif
29
30
31
32 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(msg_task, msg,
33                                 "Logging specific to MSG (task)");
34
35 /********************************* Task **************************************/
36 /** \ingroup m_task_management
37  * \brief Creates a new #m_task_t.
38  *
39  * A constructor for #m_task_t taking four arguments and returning the 
40    corresponding object.
41  * \param name a name for the object. It is for user-level information
42    and can be NULL.
43  * \param compute_duration a value of the processing amount (in flop)
44    needed to process this new task. If 0, then it cannot be executed with
45    MSG_task_execute(). This value has to be >=0.
46  * \param message_size a value of the amount of data (in bytes) needed to
47    transfer this new task. If 0, then it cannot be transfered with
48    MSG_task_get() and MSG_task_put(). This value has to be >=0.
49  * \param data a pointer to any data may want to attach to the new
50    object.  It is for user-level information and can be NULL. It can
51    be retrieved with the function \ref MSG_task_get_data.
52  * \see m_task_t
53  * \return The new corresponding object.
54  */
55 m_task_t MSG_task_create(const char *name, double compute_duration,
56                          double message_size, void *data)
57 {
58   m_task_t task = xbt_new(s_m_task_t, 1);
59   simdata_task_t simdata = xbt_new(s_simdata_task_t, 1);
60   task->simdata = simdata;
61   /* Task structure */
62   task->name = xbt_strdup(name);
63   task->data = data;
64
65   /* Simulator Data */
66   simdata->host_nb = 0;
67   simdata->computation_amount = compute_duration;
68   simdata->message_size = message_size;
69   simdata->rate = -1.0;
70   simdata->priority = 1.0;
71   simdata->isused = 0;
72   simdata->sender = NULL;
73   simdata->receiver = NULL;
74   simdata->compute = NULL;
75   simdata->comm = NULL;
76
77   simdata->host_list = NULL;
78   simdata->comp_amount = NULL;
79   simdata->comm_amount = NULL;
80 #ifdef HAVE_TRACING
81   TRACE_msg_task_create(task);
82 #endif
83
84   return task;
85 }
86
87 /** \ingroup m_task_management
88  * \brief Return the user data of a #m_task_t.
89  *
90  * This function checks whether \a task is a valid pointer or not and return
91    the user data associated to \a task if it is possible.
92  */
93 void *MSG_task_get_data(m_task_t task)
94 {
95   xbt_assert0((task != NULL), "Invalid parameter");
96
97   return (task->data);
98 }
99
100 /** \ingroup m_task_management
101  * \brief Sets the user data of a #m_task_t.
102  *
103  * This function allows to associate a new pointer to
104    the user data associated of \a task.
105  */
106 void MSG_task_set_data(m_task_t task, void *data)
107 {
108   xbt_assert0((task != NULL), "Invalid parameter");
109
110   task->data = data;
111 }
112
113 /** \ingroup m_task_management
114  * \brief Return the sender of a #m_task_t.
115  *
116  * This functions returns the #m_process_t which sent this task
117  */
118 m_process_t MSG_task_get_sender(m_task_t task)
119 {
120   xbt_assert0(task, "Invalid parameters");
121   return ((simdata_task_t) task->simdata)->sender;
122 }
123
124 /** \ingroup m_task_management
125  * \brief Return the source of a #m_task_t.
126  *
127  * This functions returns the #m_host_t from which this task was sent
128  */
129 m_host_t MSG_task_get_source(m_task_t task)
130 {
131   xbt_assert0(task, "Invalid parameters");
132   return ((simdata_task_t) task->simdata)->source;
133 }
134
135 /** \ingroup m_task_management
136  * \brief Return the name of a #m_task_t.
137  *
138  * This functions returns the name of a #m_task_t as specified on creation
139  */
140 const char *MSG_task_get_name(m_task_t task)
141 {
142   xbt_assert0(task, "Invalid parameters");
143   return task->name;
144 }
145
146 /** \ingroup m_task_management
147  * \brief Return the name of a #m_task_t.
148  *
149  * This functions allows to associate a name to a task
150  */
151 void MSG_task_set_name(m_task_t task, const char *name)
152 {
153   xbt_assert0(task, "Invalid parameters");
154   task->name = xbt_strdup(name);
155 }
156
157 /** \ingroup m_task_management
158  * \brief Destroy a #m_task_t.
159  *
160  * Destructor for #m_task_t. Note that you should free user data, if any, \b 
161    before calling this function.
162  */
163 MSG_error_t MSG_task_destroy(m_task_t task)
164 {
165   smx_action_t action = NULL;
166   xbt_assert0((task != NULL), "Invalid parameter");
167
168   /* why? if somebody is using, then you can't free! ok... but will return MSG_OK? when this task will be destroyed? isn't the user code wrong? */
169   if (task->simdata->isused > 0)
170     return MSG_OK;
171 #ifdef HAVE_TRACING
172   TRACE_msg_task_destroy(task);
173 #endif
174
175   if (task->name)
176     free(task->name);
177
178   action = task->simdata->compute;
179   if (action)
180     SIMIX_req_host_execution_destroy(action);
181
182   /* parallel tasks only */
183   if (task->simdata->host_list)
184     xbt_free(task->simdata->host_list);
185
186   /* free main structures */
187   xbt_free(task->simdata);
188   xbt_free(task);
189
190   return MSG_OK;
191 }
192
193
194 /** \ingroup m_task_management
195  * \brief Cancel a #m_task_t.
196  * \param task the taskt to cancel. If it was executed or transfered, it 
197           stops the process that were working on it.
198  */
199 MSG_error_t MSG_task_cancel(m_task_t task)
200 {
201   xbt_assert0((task != NULL), "Invalid parameter");
202
203   if (task->simdata->compute) {
204     SIMIX_req_host_execution_cancel(task->simdata->compute);
205     return MSG_OK;
206   }
207   if (task->simdata->comm) {
208     SIMIX_req_comm_cancel(task->simdata->comm);
209     return MSG_OK;
210   }
211   THROW_IMPOSSIBLE;
212 }
213
214 /** \ingroup m_task_management
215  * \brief Returns the computation amount needed to process a task #m_task_t.
216  *        Once a task has been processed, this amount is thus set to 0...
217  */
218 double MSG_task_get_compute_duration(m_task_t task)
219 {
220   xbt_assert0((task != NULL)
221               && (task->simdata != NULL), "Invalid parameter");
222
223   return task->simdata->computation_amount;
224 }
225
226
227 /** \ingroup m_task_management
228  * \brief set the computation amount needed to process a task #m_task_t.
229  */
230
231 void MSG_task_set_compute_duration(m_task_t task,
232                                    double computation_amount)
233 {
234   xbt_assert0(task, "Invalid parameter");
235   task->simdata->computation_amount = computation_amount;
236
237 }
238
239 /** \ingroup m_task_management
240  * \brief Returns the remaining computation amount of a task #m_task_t.
241  *
242  */
243 double MSG_task_get_remaining_computation(m_task_t task)
244 {
245   xbt_assert0((task != NULL)
246               && (task->simdata != NULL), "Invalid parameter");
247
248   if (task->simdata->compute) {
249     return SIMIX_req_host_execution_get_remains(task->simdata->compute);
250   } else {
251     return task->simdata->computation_amount;
252   }
253 }
254
255 /** \ingroup m_task_management
256  * \brief Returns the total amount received by a task #m_task_t.
257  *        If the communication does not exist it will return 0.
258  *        So, if the communication has FINISHED or FAILED it returns
259  *        zero.
260  */
261 double MSG_task_get_remaining_communication(m_task_t task)
262 {
263   xbt_assert0((task != NULL)
264               && (task->simdata != NULL), "Invalid parameter");
265   DEBUG1("calling SIMIX_req_communication_get_remains(%p)",
266          task->simdata->comm);
267   return SIMIX_req_comm_get_remains(task->simdata->comm);
268 }
269
270 #ifdef HAVE_LATENCY_BOUND_TRACKING
271 /** \ingroup m_task_management
272  * \brief Return 1 if communication task is limited by latency, 0 otherwise
273  *
274  */
275 int MSG_task_is_latency_bounded(m_task_t task)
276 {
277   xbt_assert0((task != NULL)
278               && (task->simdata != NULL), "Invalid parameter");
279   DEBUG1("calling SIMIX_req_communication_is_latency_bounded(%p)",
280          task->simdata->comm);
281   return SIMIX_req_comm_is_latency_bounded(task->simdata->comm);
282 }
283 #endif
284
285 /** \ingroup m_task_management
286  * \brief Returns the size of the data attached to a task #m_task_t.
287  *
288  */
289 double MSG_task_get_data_size(m_task_t task)
290 {
291   xbt_assert0((task != NULL)
292               && (task->simdata != NULL), "Invalid parameter");
293
294   return task->simdata->message_size;
295 }
296
297
298
299 /** \ingroup m_task_management
300  * \brief Changes the priority of a computation task. This priority doesn't affect 
301  *        the transfer rate. A priority of 2 will make a task receive two times more
302  *        cpu power than the other ones.
303  *
304  */
305 void MSG_task_set_priority(m_task_t task, double priority)
306 {
307   xbt_assert0((task != NULL)
308               && (task->simdata != NULL), "Invalid parameter");
309
310   task->simdata->priority = 1 / priority;
311   if (task->simdata->compute)
312     SIMIX_req_host_execution_set_priority(task->simdata->compute,
313                                       task->simdata->priority);
314 }