- /* Sanity check */
- xbt_assert0(task,"Null pointer for the task\n");
-
- if (*task)
- CRITICAL0("MSG_task_get() was asked to write in a non empty task struct.");
-
- /* Get the task */
- h = MSG_host_self();
- h_simdata = h->simdata;
-
- DEBUG2("Waiting for a task on channel %d (%s)", channel,h->name);
-
- while ((t = xbt_fifo_shift(h_simdata->mbox[channel])) == NULL) {
- xbt_assert2(!(h_simdata->sleeping[channel]),
- "A process (%s(%d)) is already blocked on this channel",
- h_simdata->sleeping[channel]->name,
- h_simdata->sleeping[channel]->simdata->PID);
- h_simdata->sleeping[channel] = process; /* I'm waiting. Wake me up when you're ready */
- __MSG_process_block();
- if(surf_workstation_resource->extension_public->get_state(h_simdata->host)
- == SURF_CPU_OFF)
- MSG_RETURN(MSG_HOST_FAILURE);
- h_simdata->sleeping[channel] = NULL;
- /* OK, we should both be ready now. Are you there ? */
+#ifdef HAVE_TRACING
+ TRACE_msg_task_execute_start (task);
+#endif
+
+ simdata = task->simdata;
+
+ xbt_assert1((!simdata->compute) && (task->simdata->refcount == 1),
+ "This task is executed somewhere else. Go fix your code! %d", task->simdata->refcount);
+
+ DEBUG1("Computing on %s", MSG_process_self()->simdata->m_host->name);
+
+ if (simdata->comp_amount == 0)
+ return MSG_OK;
+
+ simdata->refcount++;
+ SIMIX_mutex_lock(simdata->mutex);
+ simdata->compute =
+ SIMIX_action_execute(SIMIX_host_self(), task->name,
+ simdata->computation_amount);
+ SIMIX_action_set_priority(simdata->compute, simdata->priority);
+
+ /* changed to waiting action since we are always waiting one action (execute, communicate or sleep) */
+ self->simdata->waiting_action = simdata->compute;
+ SIMIX_register_action_to_condition(simdata->compute, simdata->cond);
+ do {
+ SIMIX_cond_wait(simdata->cond, simdata->mutex);
+ state = SIMIX_action_get_state(simdata->compute);
+ } while (state == SURF_ACTION_READY || state == SURF_ACTION_RUNNING);
+ SIMIX_unregister_action_to_condition(simdata->compute, simdata->cond);
+ self->simdata->waiting_action = NULL;
+
+ SIMIX_mutex_unlock(simdata->mutex);
+ simdata->refcount--;
+
+ if (SIMIX_action_get_state(task->simdata->compute) == SURF_ACTION_DONE) {
+ /* action ended, set comm and compute = NULL, the actions is already destroyed in the main function */
+ SIMIX_action_destroy(task->simdata->compute);
+ simdata->computation_amount = 0.0;
+ simdata->comm = NULL;
+ simdata->compute = NULL;
+#ifdef HAVE_TRACING
+ TRACE_msg_task_execute_end (task);
+#endif
+ MSG_RETURN(MSG_OK);
+ } else if (SIMIX_host_get_state(SIMIX_host_self()) == 0) {
+ /* action ended, set comm and compute = NULL, the actions is already destroyed in the main function */
+ SIMIX_action_destroy(task->simdata->compute);
+ simdata->comm = NULL;
+ simdata->compute = NULL;
+#ifdef HAVE_TRACING
+ TRACE_msg_task_execute_end (task);
+#endif
+ MSG_RETURN(MSG_HOST_FAILURE);
+ } else {
+ /* action ended, set comm and compute = NULL, the actions is already destroyed in the main function */
+ SIMIX_action_destroy(task->simdata->compute);
+ simdata->comm = NULL;
+ simdata->compute = NULL;
+#ifdef HAVE_TRACING
+ TRACE_msg_task_execute_end (task);
+#endif
+ MSG_RETURN(MSG_TASK_CANCELLED);