X-Git-Url: http://bilbo.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/23f7bd2d6bcf61608b1ace84d6ff26657aec1a1e..6cf3e5a62a7a4a685ddce69bac69d71fe0966a86:/src/instr/instr_config.c

diff --git a/src/instr/instr_config.c b/src/instr/instr_config.c
index 9da8976b8b..7dcb4ab057 100644
--- a/src/instr/instr_config.c
+++ b/src/instr/instr_config.c
@@ -6,6 +6,7 @@
 
 #include "instr/instr_private.h"
 #include "simgrid/sg_config.h"
+#include "surf/surf.h"
 
 #ifdef HAVE_TRACING
 
@@ -48,6 +49,8 @@ static int trace_basic;
 static int trace_configured = 0;
 static int trace_active = 0;
 
+
+
 static void TRACE_getopts(void)
 {
   trace_enabled = xbt_cfg_get_int(_sg_cfg_set, OPT_TRACING);
@@ -65,6 +68,13 @@ static void TRACE_getopts(void)
   trace_basic = xbt_cfg_get_int(_sg_cfg_set, OPT_TRACING_BASIC);
 }
 
+xbt_dynar_t TRACE_start_functions = NULL;
+void TRACE_add_start_function(void (*func)()) {
+  if (TRACE_start_functions==NULL)
+    TRACE_start_functions = xbt_dynar_new(sizeof(void (*)()), NULL);   
+  xbt_dynar_push(TRACE_start_functions, &func);    
+}
+
 int TRACE_start()
 {
   TRACE_getopts();
@@ -93,11 +103,24 @@ int TRACE_start()
   declared_marks = xbt_dict_new_homogeneous (xbt_free);
   user_host_variables = xbt_dict_new_homogeneous (xbt_free);
   user_link_variables = xbt_dict_new_homogeneous (xbt_free);
-  TRACE_surf_alloc();
-  TRACE_smpi_alloc();
+
+  if (TRACE_start_functions!=NULL) {
+    void (*func)();
+    unsigned int iter = xbt_dynar_length(TRACE_start_functions);
+    xbt_dynar_foreach(TRACE_start_functions, iter, func) {
+      func();
+    }
+  }
   return 0;
 }
 
+xbt_dynar_t TRACE_end_functions = NULL;
+void TRACE_add_end_function(void (*func)(void)) {
+  if (TRACE_end_functions==NULL)
+    TRACE_end_functions = xbt_dynar_new(sizeof(void (*) (void)), NULL);   
+  xbt_dynar_push(TRACE_end_functions, &func);    
+}
+
 int TRACE_end()
 {
   if (!trace_active)
@@ -115,8 +138,15 @@ int TRACE_end()
   PJ_type_free_all();
   PJ_container_release();
   PJ_type_release();
-  TRACE_smpi_release();
-  TRACE_surf_release();
+
+  if (TRACE_end_functions!=NULL) {
+    void (*func) (void);
+    unsigned int iter;
+    xbt_dynar_foreach(TRACE_end_functions, iter, func) {
+      func();
+    }
+  }
+
   xbt_dict_free(&user_link_variables);
   xbt_dict_free(&user_host_variables);
   xbt_dict_free(&declared_marks);