Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
e1a22ee5e22b85ddffb64b81b3c443b959380e65
[simgrid.git] / src / instr / instr_paje.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, instr, "Paje tracing event system (data structures)");
12
13 static type_t rootType = NULL;              /* the root type */
14 static container_t rootContainer = NULL;    /* the root container */
15 static xbt_dict_t allContainers = NULL;     /* all created containers indexed by name */
16 xbt_dynar_t allLinkTypes = NULL;     /* all link types defined */
17 xbt_dynar_t allHostTypes = NULL;     /* all host types defined */
18
19 void instr_paje_init (container_t root)
20 {
21   allContainers = xbt_dict_new ();
22   allLinkTypes = xbt_dynar_new (sizeof(s_type_t), NULL);
23   allHostTypes = xbt_dynar_new (sizeof(s_type_t), NULL);
24   rootContainer = root;
25 }
26
27 static long long int newTypeId ()
28 {
29   static long long int counter = 0;
30   return counter++;
31 }
32
33 static type_t newType (const char *typename, const char *key, e_entity_types kind, type_t father)
34 {
35   type_t ret = xbt_new0(s_type_t, 1);
36   ret->name = xbt_strdup (typename);
37   ret->father = father;
38   ret->kind = kind;
39   ret->children = xbt_dict_new ();
40
41   long long int id = newTypeId();
42   char str_id[INSTR_DEFAULT_STR_SIZE];
43   snprintf (str_id, INSTR_DEFAULT_STR_SIZE, "%lld", id);
44   ret->id = xbt_strdup (str_id);
45
46   if (father != NULL){
47     xbt_dict_set (father->children, key, ret, NULL);
48   }
49   return ret;
50 }
51
52 type_t getRootType ()
53 {
54   return rootType;
55 }
56
57 type_t getContainerType (const char *typename, type_t father)
58 {
59   type_t ret;
60   if (father == NULL){
61     ret = newType (typename, typename, TYPE_CONTAINER, father);
62     if (father) pajeDefineContainerType(ret->id, ret->father->id, ret->name);
63     rootType = ret;
64   }else{
65     //check if my father type already has my typename
66     ret = (type_t)xbt_dict_get_or_null (father->children, typename);
67     if (ret == NULL){
68       ret = newType (typename, typename, TYPE_CONTAINER, father);
69       pajeDefineContainerType(ret->id, ret->father->id, ret->name);
70     }
71   }
72   return ret;
73 }
74
75 type_t getEventType (const char *typename, const char *color, type_t father)
76 {
77   type_t ret = xbt_dict_get_or_null (father->children, typename);
78   if (ret == NULL){
79     ret = newType (typename, typename, TYPE_EVENT, father);
80     //INFO4("EventType %s(%s), child of %s(%s)", ret->name, ret->id, father->name, father->id);
81     if (color){
82       pajeDefineEventTypeWithColor (ret->id, ret->father->id, ret->name, color);
83     }else{
84       pajeDefineEventType(ret->id, ret->father->id, ret->name);
85     }
86   }
87   return ret;
88 }
89
90 type_t getVariableType (const char *typename, const char *color, type_t father)
91 {
92   type_t ret = xbt_dict_get_or_null (father->children, typename);
93   if (ret == NULL){
94     ret = newType (typename, typename, TYPE_VARIABLE, father);
95     //INFO4("VariableType %s(%s), child of %s(%s)", ret->name, ret->id, father->name, father->id);
96     if (color){
97       pajeDefineVariableTypeWithColor(ret->id, ret->father->id, ret->name, color);
98     }else{
99       pajeDefineVariableType(ret->id, ret->father->id, ret->name);
100     }
101   }
102   return ret;
103 }
104
105 char *getVariableTypeIdByName (const char *name, type_t father)
106 {
107   xbt_dict_cursor_t cursor = NULL;
108   type_t type;
109   char *key;
110   xbt_dict_foreach(father->children, cursor, key, type) {
111     if (strcmp (name, type->name) == 0) return type->id;
112   }
113   return NULL;
114 }
115
116 type_t getLinkType (const char *typename, type_t father, type_t source, type_t dest)
117 {
118   //FIXME should check using source and dest here and not by the typename (g5k example)
119   char key[INSTR_DEFAULT_STR_SIZE];
120   snprintf (key, INSTR_DEFAULT_STR_SIZE, "%s-%s-%s", typename, source->id, dest->id);
121   type_t ret = xbt_dict_get_or_null (father->children, key);
122   if (ret == NULL){
123     ret = newType (typename, key, TYPE_LINK, father);
124     //INFO8("LinkType %s(%s), child of %s(%s)  %s(%s)->%s(%s)", ret->name, ret->id, father->name, father->id, source->name, source->id, dest->name, dest->id);
125     pajeDefineLinkType(ret->id, ret->father->id, source->id, dest->id, ret->name);
126   }
127   return ret;
128 }
129
130 type_t getStateType (const char *typename, type_t father)
131 {
132   type_t ret = xbt_dict_get_or_null (father->children, typename);
133   if (ret == NULL){
134     ret = newType (typename, typename, TYPE_STATE, father);
135     //INFO4("StateType %s(%s), child of %s(%s)", ret->name, ret->id, father->name, father->id);
136     pajeDefineStateType(ret->id, ret->father->id, ret->name);
137   }
138   return ret;
139 }
140
141
142 static long long int newContainedId ()
143 {
144   static long long counter = 0;
145   return counter++;
146 }
147
148 container_t newContainer (const char *name, e_container_types kind, container_t father)
149 {
150   long long int counter = newContainedId();
151   char id_str[INSTR_DEFAULT_STR_SIZE];
152   snprintf (id_str, INSTR_DEFAULT_STR_SIZE, "%lld", counter);
153
154   container_t new = xbt_new0(s_container_t, 1);
155   new->name = xbt_strdup (name); // name of the container
156   new->id = xbt_strdup (id_str); // id (or alias) of the container
157   new->father = father;
158   // level depends on level of father
159   if (new->father){
160     new->level = new->father->level+1;
161   }else{
162     new->level = 0;
163   }
164   // type definition (method depends on kind of this new container)
165   new->kind = kind;
166   if (new->kind == INSTR_AS){
167     //if this container is of an AS, its type name depends on its level
168     char as_typename[INSTR_DEFAULT_STR_SIZE];
169     snprintf (as_typename, INSTR_DEFAULT_STR_SIZE, "L%d", new->level);
170     if (new->father){
171       new->type = getContainerType (as_typename, new->father->type);
172     }else{
173       new->type = getContainerType ("0", NULL);
174     }
175   }else{
176     //otherwise, the name is its kind
177     switch (new->kind){
178       case INSTR_HOST: new->type = getContainerType ("HOST", new->father->type); break;
179       case INSTR_LINK: new->type = getContainerType ("LINK", new->father->type); break;
180       case INSTR_ROUTER: new->type = getContainerType ("ROUTER", new->father->type); break;
181       case INSTR_SMPI: new->type = getContainerType ("MPI", new->father->type); break;
182       case INSTR_MSG_PROCESS: new->type = getContainerType ("MSG_PROCESS", new->father->type); break;
183       case INSTR_MSG_TASK: new->type = getContainerType ("MSG_TASK", new->father->type); break;
184       default: xbt_die ("Congratulations, you have found a bug on newContainer function of instr_routing.c"); break;
185     }
186   }
187   new->children = xbt_dict_new();
188   if (new->father){
189     xbt_dict_set(new->father->children, new->name, new, NULL);
190     pajeCreateContainer (SIMIX_get_clock(), new->id, new->type->id, new->father->id, new->name);
191   }
192
193   //register hosts, routers, links containers
194   if (new->kind == INSTR_HOST || new->kind == INSTR_LINK || new->kind == INSTR_ROUTER) {
195     xbt_dict_set (allContainers, new->name, new, NULL);
196   }
197
198   //register the host container types
199   if (new->kind == INSTR_HOST){
200     xbt_dynar_push_as (allHostTypes, type_t, new->type);
201   }
202
203   //register the link container types
204   if (new->kind == INSTR_LINK){
205     xbt_dynar_push_as(allLinkTypes, type_t, new->type);
206   }
207   return new;
208 }
209
210 static container_t recursiveGetContainer (const char *name, container_t root)
211 {
212   if (strcmp (root->name, name) == 0) return root;
213
214   xbt_dict_cursor_t cursor = NULL;
215   container_t child;
216   char *child_name;
217   xbt_dict_foreach(root->children, cursor, child_name, child) {
218     container_t ret = recursiveGetContainer(name, child);
219     if (ret) return ret;
220   }
221   return NULL;
222 }
223
224 container_t getContainer (const char *name)
225 {
226   return recursiveGetContainer(name, rootContainer);
227 }
228
229 container_t getContainerByName (const char *name)
230 {
231   return (container_t)xbt_dict_get (allContainers, name);
232 }
233
234 char *getContainerIdByName (const char *name)
235 {
236   return getContainerByName(name)->id;
237 }
238
239 container_t getRootContainer ()
240 {
241   return rootContainer;
242 }
243
244 static type_t recursiveGetType (const char *name, type_t root)
245 {
246   if (strcmp (root->name, name) == 0) return root;
247
248   xbt_dict_cursor_t cursor = NULL;
249   type_t child;
250   char *child_name;
251   xbt_dict_foreach(root->children, cursor, child_name, child) {
252     type_t ret = recursiveGetType(name, child);
253     if (ret) return ret;
254   }
255   return NULL;
256 }
257
258 type_t getType (const char *name)
259 {
260   return recursiveGetType (name, rootType);
261 }
262
263 void destroyContainer (container_t container)
264 {
265   //remove me from my father
266   if (container->father){
267     xbt_dict_remove(container->father->children, container->name);
268   }
269
270   //trace my destruction
271   pajeDestroyContainer(SIMIX_get_clock(), container->type->id, container->id);
272
273   //free
274   xbt_free (container->name);
275   xbt_free (container->id);
276   xbt_free (container->children);
277   xbt_free (container);
278   container = NULL;
279 }
280
281 static void recursiveDestroyContainer (container_t container)
282 {
283   xbt_dict_cursor_t cursor = NULL;
284   container_t child;
285   char *child_name;
286   xbt_dict_foreach(container->children, cursor, child_name, child) {
287     recursiveDestroyContainer (child);
288   }
289   destroyContainer (container);
290 }
291
292 static void recursiveDestroyType (type_t type)
293 {
294   xbt_dict_cursor_t cursor = NULL;
295   type_t child;
296   char *child_name;
297   xbt_dict_foreach(type->children, cursor, child_name, child) {
298     recursiveDestroyType (child);
299   }
300   xbt_free (type->name);
301   xbt_free (type->id);
302   xbt_free (type->children);
303   xbt_free (type);
304   type = NULL;
305 }
306
307 void destroyAllContainers ()
308 {
309   if (getRootContainer()) recursiveDestroyContainer (getRootContainer());
310   if (getRootType()) recursiveDestroyType (getRootType());
311   rootContainer = NULL;
312   rootType = NULL;
313 }
314
315
316 #endif /* HAVE_TRACING */