Logo AND Algorithmique Numérique Distribuée

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