Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
move one more tracing test
[simgrid.git] / examples / msg / trace-user-variables / trace-user-variables.c
diff --git a/examples/msg/trace-user-variables/trace-user-variables.c b/examples/msg/trace-user-variables/trace-user-variables.c
new file mode 100644 (file)
index 0000000..5e6a057
--- /dev/null
@@ -0,0 +1,125 @@
+/* Copyright (c) 2010, 2012-2015. 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. */
+
+/** @addtogroup MSG_examples
+ * 
+ * - <b>tracing/user_variables.c</b>: This program demonstrates how to trace user variables associated to the hosts of
+ * the platform file. You might want to run this program with the following parameters:
+ * --cfg=tracing:yes
+ * --cfg=tracing/platform:yes
+ * (See \ref tracing_tracing_options for details)
+ */
+#include <stdio.h>
+#include "simgrid/msg.h"
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test, "Messages specific for this msg example");
+
+static int master(int argc, char *argv[])
+{
+  const char *hostname = MSG_host_get_name(MSG_host_self());
+  int i;
+
+  //the hostname has an empty HDD with a capacity of 100000 (bytes)
+  TRACE_host_variable_set(hostname, "HDD_capacity", 100000);
+  TRACE_host_variable_set(hostname, "HDD_utilization", 0);
+
+  for (i = 0; i < 10; i++) {
+    //create and execute a task just to make the simulated time advance
+    msg_task_t task = MSG_task_create("task", 10000, 0, NULL);
+    MSG_task_execute (task);
+    MSG_task_destroy (task);
+
+    //ADD: after the execution of this task, the HDD utilization increases by 100 (bytes)
+    TRACE_host_variable_add(hostname, "HDD_utilization", 100);
+  }
+
+  for (i = 0; i < 10; i++) {
+    //create and execute a task just to make the simulated time advance
+    msg_task_t task = MSG_task_create("task", 10000, 0, NULL);
+    MSG_task_execute (task);
+    MSG_task_destroy (task);
+
+    //SUB: after the execution of this task, the HDD utilization decreases by 100 (bytes)
+    TRACE_host_variable_sub(hostname, "HDD_utilization", 100);
+  }
+  return 0;
+}
+
+int main(int argc, char *argv[])
+{
+  MSG_init(&argc, argv);
+  xbt_assert(argc > 2, "Usage: %s platform_file deployment_file\n"
+             "\tExample: %s msg_platform.xml msg_deployment.xml\n", argv[0], argv[0]);
+
+  MSG_create_environment(argv[1]);
+
+  //declaring user variables
+  TRACE_host_variable_declare("HDD_capacity");
+  TRACE_host_variable_declare("HDD_utilization");
+
+  //register functions and launch deployment
+  MSG_function_register("master", master);
+  MSG_function_register("slave", master);
+  MSG_launch_application(argv[2]);
+
+  MSG_main();
+
+  //get user declared variables
+  unsigned int cursor;
+  char *variable;
+  xbt_dynar_t host_variables = TRACE_get_host_variables ();
+  if (host_variables){
+    XBT_INFO ("Declared host variables:");
+    xbt_dynar_foreach (host_variables, cursor, variable){
+      XBT_INFO ("%s", variable);
+    }
+    xbt_dynar_free (&host_variables);
+  }
+  xbt_dynar_t link_variables = TRACE_get_link_variables ();
+  if (link_variables){
+    XBT_INFO ("Declared link variables:");
+    xbt_dynar_foreach (link_variables, cursor, variable){
+      XBT_INFO ("%s", variable);
+    }
+    xbt_dynar_free (&link_variables);
+  }
+
+  //create a customized viva graph configuration file
+  FILE *fp;
+  fp = fopen ("viva_graph.plist", "w");
+  if (!fp){
+    return 1;
+  }
+  fprintf (fp, "{\n node = (");
+  xbt_dynar_t nodes_type = TRACE_get_node_types ();
+  if (nodes_type){
+    XBT_INFO ("Node types in the trace:");
+    char *node_type;
+    xbt_dynar_foreach (nodes_type, cursor, node_type){
+      XBT_INFO ("%s", node_type);
+      fprintf (fp, "%s, ", node_type);
+    }
+    xbt_dynar_free (&nodes_type);
+  }
+  fprintf (fp, ");\n edge = (");
+  xbt_dynar_t edges_type = TRACE_get_edge_types ();
+  if (edges_type){
+    XBT_INFO ("Node types in the trace:");
+    char *edge_type;
+    xbt_dynar_foreach (edges_type, cursor, edge_type){
+      XBT_INFO ("%s", edge_type);
+      fprintf (fp, "%s, ", edge_type);
+    }
+    xbt_dynar_free (&edges_type);
+  }
+  fprintf (fp, ");\n");
+  fprintf (fp, " host = {\n  type = square;\n  size = HDD_capacity; \n  values = (HDD_utilization);\n };\n");
+  fprintf (fp, " link = {\n  type = rhombus;\n  size = bandwidth;\n };\n");
+  fprintf (fp, "}\n");
+  fclose (fp);
+
+  return 0;
+}