Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
leak --
[simgrid.git] / src / instr / instr_TI_trace.c
1 /* Copyright (c) 2010-2013. 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 #include "instr/instr_private.h"
7 #include "xbt/virtu.h" /* sg_cmdline */
8 #include "xbt/xbt_os_time.h"
9 #include "simgrid/sg_config.h"
10
11 #include <errno.h>
12 #include <string.h>
13 #include <sys/stat.h>
14
15 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(instr_TI_trace, instr_trace, "tracing event system");
16
17 extern FILE *tracing_file;
18 double prefix=0.0;
19
20 xbt_dict_t tracing_files = NULL;
21
22 extern s_instr_trace_writer_t active_writer;
23
24 void TRACE_TI_init(void)
25 {
26   active_writer.print_PushState = print_TIPushState;
27   active_writer.print_CreateContainer=print_TICreateContainer;
28   active_writer.print_DestroyContainer=print_TIDestroyContainer;
29 }
30
31 void TRACE_TI_start(void)
32 {
33   char *filename = TRACE_get_filename();
34   tracing_file = fopen(filename, "w");
35   if (tracing_file == NULL) {
36     THROWF(system_error, 1, "Tracefile %s could not be opened for writing.",
37            filename);
38   }
39
40   XBT_DEBUG("Filename %s is open for writing", filename);
41
42   /* output one line comment */
43   dump_comment(TRACE_get_comment());
44
45   /* output comment file */
46   dump_comment_file(TRACE_get_comment_file());
47
48 }
49
50 void TRACE_TI_end(void)
51 {
52   xbt_dict_free(&tracing_files);
53   fclose(tracing_file);
54   char *filename = TRACE_get_filename();
55   XBT_DEBUG("Filename %s is closed", filename);
56 }
57
58 void print_TICreateContainer(paje_event_t event)
59 {
60   //if we are in the mode with only one file
61   static FILE *temp = NULL;
62
63   if (tracing_files == NULL) {
64     tracing_files = xbt_dict_new_homogeneous(NULL);
65     //generate unique run id with time
66     prefix = xbt_os_time();
67   }
68
69   if (!xbt_cfg_get_boolean(_sg_cfg_set, "tracing/smpi/format/ti_one_file")
70       || temp == NULL) {
71     char *folder_name = bprintf("%s_files", TRACE_get_filename());
72     char *filename = bprintf("%s/%f_%s.txt", folder_name, prefix,
73                              ((createContainer_t) event->data)->container->name);
74     mkdir(folder_name, S_IRWXU | S_IRWXG | S_IRWXO);
75     temp = fopen(filename, "w");
76     if (temp == NULL)
77       xbt_die("Tracefile %s could not be opened for writing: %s",
78               filename, strerror(errno));
79     fprintf(tracing_file, "%s\n", filename);
80
81     xbt_free(folder_name);
82     xbt_free(filename);
83   }
84
85   xbt_dict_set(tracing_files,
86                ((createContainer_t) event->data)->container->name,
87                (void *) temp, NULL);
88 }
89
90 void print_TIDestroyContainer(paje_event_t event)
91 {
92   if (!xbt_cfg_get_boolean(_sg_cfg_set, "tracing/smpi/format/ti_one_file")||
93       xbt_dict_length(tracing_files) == 1) {
94     FILE* f = xbt_dict_get_or_null(tracing_files,
95         ((destroyContainer_t) event->data)->container->name);
96     fclose(f);
97   }
98   xbt_dict_remove(tracing_files, ((destroyContainer_t) event->data)->container->name);
99 }
100
101
102 void print_TIPushState(paje_event_t event)
103 {
104
105
106   int i;
107
108   //char* function=NULL;
109   if (((pushState_t) event->data)->extra == NULL)
110     return;
111   instr_extra_data extra =
112       (instr_extra_data) (((pushState_t) event->data)->extra);
113
114   char *process_id = NULL;
115   //FIXME: dirty extract "rank-" from the name, as we want the bare process id here
116   if (strstr(((pushState_t) event->data)->container->name, "rank-") == NULL)
117     process_id = xbt_strdup(((pushState_t) event->data)->container->name);
118   else
119     process_id = xbt_strdup(((pushState_t) event->data)->container->name + 5);
120
121   FILE* trace_file =  (FILE* )xbt_dict_get(tracing_files, ((pushState_t) event->data)->container->name);
122
123   switch (extra->type) {
124
125   case TRACING_INIT:
126     fprintf(trace_file, "%s init\n", process_id);
127     break;
128   case TRACING_FINALIZE:
129     fprintf(trace_file, "%s finalize\n", process_id);
130     break;
131   case TRACING_SEND:
132     fprintf(trace_file, "%s send %d %d %s\n", process_id, extra->dst,
133             extra->send_size, extra->datatype1);
134     break;
135   case TRACING_ISEND:
136     fprintf(trace_file, "%s isend %d %d %s\n", process_id, extra->dst,
137             extra->send_size, extra->datatype1);
138     break;
139   case TRACING_RECV:
140     fprintf(trace_file, "%s recv %d %d %s\n", process_id, extra->src,
141             extra->send_size, extra->datatype1);
142     break;
143   case TRACING_IRECV:
144     fprintf(trace_file, "%s irecv %d %d %s\n", process_id, extra->src,
145             extra->send_size, extra->datatype1);
146     break;
147   case TRACING_WAIT:
148     fprintf(trace_file, "%s wait\n", process_id);
149     break;
150   case TRACING_WAITALL:
151     fprintf(trace_file, "%s waitall\n", process_id);
152     break;
153   case TRACING_BARRIER:
154     fprintf(trace_file, "%s barrier\n", process_id);
155     break;
156   case TRACING_BCAST:          // rank bcast size (root) (datatype)
157     fprintf(trace_file, "%s bcast %d ", process_id, extra->send_size);
158     if (extra->root != 0 || (extra->datatype1 && strcmp(extra->datatype1, "")))
159       fprintf(trace_file, "%d %s", extra->root, extra->datatype1);
160     fprintf(trace_file, "\n");
161     break;
162   case TRACING_REDUCE:         // rank reduce comm_size comp_size (root) (datatype)
163     fprintf(trace_file, "%s reduce %d %f ", process_id, extra->send_size,
164             extra->comp_size);
165     if (extra->root != 0 || (extra->datatype1 && strcmp(extra->datatype1, "")))
166       fprintf(trace_file, "%d %s", extra->root, extra->datatype1);
167     fprintf(trace_file, "\n");
168     break;
169   case TRACING_ALLREDUCE:      // rank allreduce comm_size comp_size (datatype)
170     fprintf(trace_file, "%s allreduce %d %f %s\n", process_id,
171             extra->send_size, extra->comp_size, extra->datatype1);
172     break;
173   case TRACING_ALLTOALL:       // rank alltoall send_size recv_size (sendtype) (recvtype)
174     fprintf(trace_file, "%s alltoall %d %d %s %s\n", process_id,
175             extra->send_size, extra->recv_size, extra->datatype1,
176             extra->datatype2);
177     break;
178   case TRACING_ALLTOALLV:      // rank alltoallv send_size [sendcounts] recv_size [recvcounts] (sendtype) (recvtype)
179     fprintf(trace_file, "%s alltoallv %d ", process_id, extra->send_size);
180     for (i = 0; i < extra->num_processes; i++)
181       fprintf(trace_file, "%d ", extra->sendcounts[i]);
182     fprintf(trace_file, "%d ", extra->recv_size);
183     for (i = 0; i < extra->num_processes; i++)
184       fprintf(trace_file, "%d ", extra->recvcounts[i]);
185     fprintf(trace_file, "%s %s \n", extra->datatype1, extra->datatype2);
186     break;
187   case TRACING_GATHER:         // rank gather send_size recv_size root (sendtype) (recvtype)
188     fprintf(trace_file, "%s gather %d %d %d %s %s\n", process_id,
189             extra->send_size, extra->recv_size, extra->root, extra->datatype1,
190             extra->datatype2);
191     break;
192   case TRACING_ALLGATHERV:     // rank allgatherv send_size [recvcounts] (sendtype) (recvtype)
193     fprintf(trace_file, "%s allgatherv %d ", process_id, extra->send_size);
194     for (i = 0; i < extra->num_processes; i++)
195       fprintf(trace_file, "%d ", extra->recvcounts[i]);
196     fprintf(trace_file, "%s %s \n", extra->datatype1, extra->datatype2);
197     break;
198   case TRACING_REDUCE_SCATTER: // rank reducescatter [recvcounts] comp_size (sendtype)
199     fprintf(trace_file, "%s reducescatter ", process_id);
200     for (i = 0; i < extra->num_processes; i++)
201       fprintf(trace_file, "%d ", extra->recvcounts[i]);
202     fprintf(trace_file, "%f %s\n", extra->comp_size, extra->datatype1);
203     break;
204   case TRACING_COMPUTING:
205     fprintf(trace_file, "%s compute %f\n", process_id, extra->comp_size);
206     break;
207   case TRACING_GATHERV: // rank gatherv send_size [recvcounts] root (sendtype) (recvtype)
208     fprintf(trace_file, "%s gatherv %d ", process_id, extra->send_size);
209     for (i = 0; i < extra->num_processes; i++)
210       fprintf(trace_file, "%d ", extra->recvcounts[i]);
211     fprintf(trace_file, "%d %s %s\n", extra->root, extra->datatype1, extra->datatype2);
212     break;
213   case TRACING_WAITANY:
214   case TRACING_SENDRECV:
215   case TRACING_SCATTER:
216   case TRACING_SCATTERV:
217   case TRACING_ALLGATHER:
218   case TRACING_SCAN:
219   case TRACING_EXSCAN:
220   case TRACING_COMM_SIZE:
221   case TRACING_COMM_SPLIT:
222   case TRACING_COMM_DUP:
223   case TRACING_SSEND:
224   case TRACING_ISSEND:
225   default:
226
227     XBT_WARN
228         ("Call from %s impossible to translate into replay command : Not implemented (yet)",
229          ((pushState_t) event->data)->value->name);
230     break;
231   }
232
233   if (extra->recvcounts != NULL)
234     xbt_free(extra->recvcounts);
235   if (extra->sendcounts != NULL)
236     xbt_free(extra->sendcounts);
237   xbt_free(process_id);
238   xbt_free(extra);
239 }