+/* Copyright (c) 2010. The SimGrid Team.
+ * All rights reserved. */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
+#include "instr/instr_private.h"
+#include "mc/mc.h"
+
+#ifdef HAVE_TRACING
+
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY (instr_msg, instr, "MSG");
+
+void TRACE_msg_set_task_category(msg_task_t task, const char *category)
+{
+ xbt_assert(task->category == NULL, "Task %p(%s) already has a category (%s).",
+ task, task->name, task->category);
+
+ //if user provides a NULL category, task is no longer traced
+ if (category == NULL) {
+ xbt_free (task->category);
+ task->category = NULL;
+ XBT_DEBUG("MSG task %p(%s), category removed", task, task->name);
+ return;
+ }
+
+ //set task category
+ task->category = xbt_strdup (category);
+ XBT_DEBUG("MSG task %p(%s), category %s", task, task->name, task->category);
+}
+
+/* MSG_task_create related function*/
+void TRACE_msg_task_create(msg_task_t task)
+{
+ static long long counter = 0;
+ task->counter = counter++;
+ task->category = NULL;
+
+ if(MC_is_active()){
+ MC_ignore_data_bss(&counter, sizeof(counter));
+ MC_ignore_heap(&(task->counter), sizeof(task->counter));
+ }
+
+ XBT_DEBUG("CREATE %p, %lld", task, task->counter);
+}
+
+/* MSG_task_execute related functions */
+void TRACE_msg_task_execute_start(msg_task_t task)
+{
+ XBT_DEBUG("EXEC,in %p, %lld, %s", task, task->counter, task->category);
+
+ if (TRACE_msg_process_is_enabled()){
+ int len = INSTR_DEFAULT_STR_SIZE;
+ char str[INSTR_DEFAULT_STR_SIZE];
+
+ container_t process_container = PJ_container_get (instr_process_id(MSG_process_self(), str, len));
+ type_t type = PJ_type_get ("MSG_PROCESS_STATE", process_container->type);
+ val_t value = PJ_value_get ("task_execute", type);
+ new_pajePushState (MSG_get_clock(), process_container, type, value);
+ }
+}
+
+void TRACE_msg_task_execute_end(msg_task_t task)
+{
+ XBT_DEBUG("EXEC,out %p, %lld, %s", task, task->counter, task->category);
+
+ if (TRACE_msg_process_is_enabled()){
+ int len = INSTR_DEFAULT_STR_SIZE;
+ char str[INSTR_DEFAULT_STR_SIZE];
+
+ container_t process_container = PJ_container_get (instr_process_id(MSG_process_self(), str, len));
+ type_t type = PJ_type_get ("MSG_PROCESS_STATE", process_container->type);
+ new_pajePopState (MSG_get_clock(), process_container, type);
+ }
+}
+
+/* MSG_task_destroy related functions */
+void TRACE_msg_task_destroy(msg_task_t task)
+{
+ XBT_DEBUG("DESTROY %p, %lld, %s", task, task->counter, task->category);
+
+ //free category
+ xbt_free(task->category);
+ task->category = NULL;
+ return;
+}
+
+/* MSG_task_get related functions */
+void TRACE_msg_task_get_start(void)
+{
+ XBT_DEBUG("GET,in");
+
+ if (TRACE_msg_process_is_enabled()){
+ int len = INSTR_DEFAULT_STR_SIZE;
+ char str[INSTR_DEFAULT_STR_SIZE];
+
+ container_t process_container = PJ_container_get (instr_process_id(MSG_process_self(), str, len));
+ type_t type = PJ_type_get ("MSG_PROCESS_STATE", process_container->type);
+ val_t value = PJ_value_get ("receive", type);
+ new_pajePushState (MSG_get_clock(), process_container, type, value);
+ }
+}
+
+void TRACE_msg_task_get_end(double start_time, msg_task_t task)
+{
+ XBT_DEBUG("GET,out %p, %lld, %s", task, task->counter, task->category);
+
+ if (TRACE_msg_process_is_enabled()){
+ int len = INSTR_DEFAULT_STR_SIZE;
+ char str[INSTR_DEFAULT_STR_SIZE];
+
+ container_t process_container = PJ_container_get (instr_process_id(MSG_process_self(), str, len));
+ type_t type = PJ_type_get ("MSG_PROCESS_STATE", process_container->type);
+ new_pajePopState (MSG_get_clock(), process_container, type);
+
+ char key[INSTR_DEFAULT_STR_SIZE];
+ snprintf (key, INSTR_DEFAULT_STR_SIZE, "p%lld", task->counter);
+ type = PJ_type_get ("MSG_PROCESS_TASK_LINK", PJ_type_get_root());
+ new_pajeEndLink(MSG_get_clock(), PJ_container_get_root(), type, process_container, "SR", key);
+ }
+}
+
+/* MSG_task_put related functions */
+int TRACE_msg_task_put_start(msg_task_t task)
+{
+ XBT_DEBUG("PUT,in %p, %lld, %s", task, task->counter, task->category);
+
+ if (TRACE_msg_process_is_enabled()){
+ int len = INSTR_DEFAULT_STR_SIZE;
+ char str[INSTR_DEFAULT_STR_SIZE];
+
+ container_t process_container = PJ_container_get (instr_process_id(MSG_process_self(), str, len));
+ type_t type = PJ_type_get ("MSG_PROCESS_STATE", process_container->type);
+ val_t value = PJ_value_get ("send", type);
+ new_pajePushState (MSG_get_clock(), process_container, type, value);
+
+ char key[INSTR_DEFAULT_STR_SIZE];
+ snprintf (key, INSTR_DEFAULT_STR_SIZE, "p%lld", task->counter);
+ type = PJ_type_get ("MSG_PROCESS_TASK_LINK", PJ_type_get_root());
+ new_pajeStartLink(MSG_get_clock(), PJ_container_get_root(), type, process_container, "SR", key);
+ }
+
+ return 1;
+}
+
+void TRACE_msg_task_put_end(void)
+{
+ XBT_DEBUG("PUT,out");
+
+ if (TRACE_msg_process_is_enabled()){
+ int len = INSTR_DEFAULT_STR_SIZE;
+ char str[INSTR_DEFAULT_STR_SIZE];
+
+ container_t process_container = PJ_container_get (instr_process_id(MSG_process_self(), str, len));
+ type_t type = PJ_type_get ("MSG_PROCESS_STATE", process_container->type);
+ new_pajePopState (MSG_get_clock(), process_container, type);
+ }
+}
+
+#endif /* HAVE_TRACING */