Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Modify instr_paje_containers.c to include MSG_VM trace
[simgrid.git] / src / instr / instr_paje_containers.c
1 /* Copyright (c) 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 "instr/instr_private.h"
8
9 #ifdef HAVE_TRACING
10
11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY (instr_paje_containers, instr, "Paje tracing event system (containers)");
12
13 static container_t rootContainer = NULL;    /* the root container */
14 static xbt_dict_t allContainers = NULL;     /* all created containers indexed by name */
15 xbt_dict_t trivaNodeTypes = NULL;     /* all host types defined */
16 xbt_dict_t trivaEdgeTypes = NULL;     /* all link types defined */
17
18 long long int instr_new_paje_id (void)
19 {
20   static long long int type_id = 0;
21   return type_id++;
22 }
23
24 void PJ_container_alloc (void)
25 {
26   allContainers = xbt_dict_new_homogeneous(NULL);
27   trivaNodeTypes = xbt_dict_new_homogeneous(xbt_free);
28   trivaEdgeTypes = xbt_dict_new_homogeneous(xbt_free);
29 }
30
31 void PJ_container_release (void)
32 {
33   xbt_dict_free (&allContainers);
34   xbt_dict_free (&trivaNodeTypes);
35   xbt_dict_free (&trivaEdgeTypes);
36 }
37
38 void PJ_container_set_root (container_t root)
39 {
40   rootContainer = root;
41 }
42
43 container_t PJ_container_new (const char *name, e_container_types kind, container_t father)
44 {
45   if (name == NULL){
46     THROWF (tracing_error, 0, "can't create a container with a NULL name");
47   }
48
49   static long long int container_id = 0;
50   char id_str[INSTR_DEFAULT_STR_SIZE];
51   snprintf (id_str, INSTR_DEFAULT_STR_SIZE, "%lld", container_id++);
52
53   container_t new = xbt_new0(s_container_t, 1);
54   new->name = xbt_strdup (name); // name of the container
55   new->id = xbt_strdup (id_str); // id (or alias) of the container
56   new->father = father;
57
58   //Search for network_element_t
59   switch (kind){
60     case INSTR_HOST:
61       new->net_elm = xbt_lib_get_or_null(host_lib,name,ROUTING_HOST_LEVEL);
62       if(!new->net_elm) xbt_die("Element '%s' not found",name);
63       break;
64     case INSTR_ROUTER:
65       new->net_elm = xbt_lib_get_or_null(as_router_lib,name,ROUTING_ASR_LEVEL);
66       if(!new->net_elm) xbt_die("Element '%s' not found",name);
67       break;
68     case INSTR_AS:
69       new->net_elm = xbt_lib_get_or_null(as_router_lib,name,ROUTING_ASR_LEVEL);
70       if(!new->net_elm) xbt_die("Element '%s' not found",name);
71       break;
72     default:
73       new->net_elm = NULL;
74       break;
75   }
76
77   // level depends on level of father
78   if (new->father){
79     new->level = new->father->level+1;
80     XBT_DEBUG("new container %s, child of %s", name, father->name);
81   }else{
82     new->level = 0;
83   }
84   // type definition (method depends on kind of this new container)
85   new->kind = kind;
86   if (new->kind == INSTR_AS){
87     //if this container is of an AS, its type name depends on its level
88     char as_typename[INSTR_DEFAULT_STR_SIZE];
89     snprintf (as_typename, INSTR_DEFAULT_STR_SIZE, "L%d", new->level);
90     if (new->father){
91       new->type = PJ_type_get_or_null (as_typename, new->father->type);
92       if (new->type == NULL){
93         new->type = PJ_type_container_new (as_typename, new->father->type);
94       }
95     }else{
96       new->type = PJ_type_container_new ("0", NULL);
97     }
98   }else{
99     //otherwise, the name is its kind
100     char typename[INSTR_DEFAULT_STR_SIZE];
101     switch (new->kind){
102       case INSTR_HOST:        snprintf (typename, INSTR_DEFAULT_STR_SIZE, "HOST");        break;
103       case INSTR_LINK:        snprintf (typename, INSTR_DEFAULT_STR_SIZE, "LINK");        break;
104       case INSTR_ROUTER:      snprintf (typename, INSTR_DEFAULT_STR_SIZE, "ROUTER");      break;
105       case INSTR_SMPI:        snprintf (typename, INSTR_DEFAULT_STR_SIZE, "MPI");         break;
106       case INSTR_MSG_PROCESS: snprintf (typename, INSTR_DEFAULT_STR_SIZE, "MSG_PROCESS"); break;
107       case INSTR_MSG_VM: snprintf (typename, INSTR_DEFAULT_STR_SIZE, "MSG_VM"); break;
108       case INSTR_MSG_TASK:    snprintf (typename, INSTR_DEFAULT_STR_SIZE, "MSG_TASK");    break;
109       default: THROWF (tracing_error, 0, "new container kind is unknown."); break;
110     }
111     type_t type = PJ_type_get_or_null (typename, new->father->type);
112     if (type == NULL){
113       new->type = PJ_type_container_new (typename, new->father->type);
114     }else{
115       new->type = type;
116     }
117   }
118   new->children = xbt_dict_new_homogeneous(NULL);
119   if (new->father){
120     xbt_dict_set(new->father->children, new->name, new, NULL);
121     new_pajeCreateContainer (new);
122   }
123
124   //register all kinds by name
125   if (xbt_dict_get_or_null(allContainers, new->name) != NULL){
126     THROWF(tracing_error, 1, "container %s already present in allContainers data structure", new->name);
127   }
128
129   xbt_dict_set (allContainers, new->name, new, NULL);
130   XBT_DEBUG("Add container name '%s'",new->name);
131
132   //register NODE types for triva configuration
133   if (new->kind == INSTR_HOST || new->kind == INSTR_LINK || new->kind == INSTR_ROUTER) {
134     xbt_dict_set (trivaNodeTypes, new->type->name, xbt_strdup("1"), NULL);
135   }
136
137   return new;
138 }
139
140 container_t PJ_container_get (const char *name)
141 {
142   container_t ret = PJ_container_get_or_null (name);
143   if (ret == NULL){
144     THROWF(tracing_error, 1, "container with name %s not found", name);
145   }
146   return ret;
147 }
148
149 container_t PJ_container_get_or_null (const char *name)
150 {
151   return name ? xbt_dict_get_or_null(allContainers, name) : NULL;
152 }
153
154 container_t PJ_container_get_root ()
155 {
156   return rootContainer;
157 }
158
159 void PJ_container_remove_from_parent (container_t child)
160 {
161   if (child == NULL){
162     THROWF (tracing_error, 0, "can't remove from parent with a NULL child");
163   }
164
165   container_t parent = child->father;
166   if (parent){
167     XBT_DEBUG("removeChildContainer (%s) FromContainer (%s) ",
168         child->name,
169         parent->name);
170     xbt_dict_remove (parent->children, child->name);
171   }
172 }
173
174 void PJ_container_free (container_t container)
175 {
176   if (container == NULL){
177     THROWF (tracing_error, 0, "trying to free a NULL container");
178   }
179   XBT_DEBUG("destroy container %s", container->name);
180
181   //obligation to dump previous events because they might
182   //reference the container that is about to be destroyed
183   TRACE_last_timestamp_to_dump = surf_get_clock();
184   TRACE_paje_dump_buffer(1);
185
186   //trace my destruction
187   if (!TRACE_disable_destroy() && container != PJ_container_get_root()){
188     //do not trace the container destruction if user requests
189     //or if the container is root
190     new_pajeDestroyContainer(container);
191   }
192
193   //remove it from allContainers data structure
194   xbt_dict_remove (allContainers, container->name);
195
196   //free
197   xbt_free (container->name);
198   xbt_free (container->id);
199   xbt_dict_free (&container->children);
200   xbt_free (container);
201   container = NULL;
202 }
203
204 static void recursiveDestroyContainer (container_t container)
205 {
206   if (container == NULL){
207     THROWF (tracing_error, 0, "trying to recursively destroy a NULL container");
208   }
209   XBT_DEBUG("recursiveDestroyContainer %s", container->name);
210   xbt_dict_cursor_t cursor = NULL;
211   container_t child;
212   char *child_name;
213   xbt_dict_foreach(container->children, cursor, child_name, child) {
214     recursiveDestroyContainer (child);
215   }
216   PJ_container_free (container);
217 }
218
219 void PJ_container_free_all ()
220 {
221   container_t root = PJ_container_get_root();
222   if (root == NULL){
223     THROWF (tracing_error, 0, "trying to free all containers, but root is NULL");
224   }
225   recursiveDestroyContainer (root);
226   rootContainer = NULL;
227
228   //checks
229   if (xbt_dict_length(allContainers) != 0){
230     THROWF(tracing_error, 0, "some containers still present even after destroying all of them");
231   }
232 }
233
234 #endif /* HAVE_TRACING */