1 /* Copyright (c) 2009, 2010. The SimGrid Team.
2 * All rights reserved. */
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. */
8 #include "simdag/simdag.h"
12 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(sd_dotparse, sd, "Parsing DOT files");
17 #include <graphviz/cgraph.h>
19 #include <graphviz/agraph.h>
22 void dot_add_task(Agnode_t * dag_node);
23 void dot_add_input_dependencies(SD_task_t current_job, Agedge_t * edge);
24 void dot_add_output_dependencies(SD_task_t current_job, Agedge_t * edge);
25 xbt_dynar_t SD_dotload_FILE(FILE * in_file);
27 static double dot_parse_double(const char *string)
34 ret = sscanf(string, "%lg", &value);
36 WARN1("%s is not a double", string);
40 static int dot_parse_int(const char *string)
47 ret = sscanf(string, "%d", &value);
49 WARN1("%s is not an integer", string);
53 static xbt_dynar_t result;
54 static xbt_dict_t jobs;
55 static xbt_dict_t files;
56 static SD_task_t root_task, end_task;
57 static Agraph_t *dag_dot;
59 static void dump_res()
63 xbt_dynar_foreach(result, cursor, task) {
64 INFO1("Task %d", cursor);
69 static void dot_task_free(void *task)
75 static void TRACE_sd_dotloader (SD_task_t task, const char *category)
78 if (strlen (category) != 0){
79 TRACE_category (category);
80 TRACE_sd_set_task_category (task, category);
85 /** @brief loads a DOT file describing a DAG
87 * See http://www.graphviz.org/doc/info/lang.html
89 * To obtain information about transfers and tasks, two attributes are
90 * required : size on task (execution time in Flop) and size on edge
91 * (the amount of data transfer in bit).
92 * if they aren't here, there choose to be equal to zero.
94 xbt_dynar_t SD_dotload(const char *filename)
96 FILE *in_file = fopen(filename, "r");
97 xbt_assert1(in_file, "Unable to open \"%s\"\n", filename);
98 SD_dotload_FILE(in_file);
103 xbt_dynar_t SD_dotload_FILE(FILE * in_file)
105 xbt_assert0(in_file, "Unable to use a null file descriptor\n");
106 dag_dot = agread(in_file, NIL(Agdisc_t *));
108 result = xbt_dynar_new(sizeof(SD_task_t), dot_task_free);
109 files = xbt_dict_new();
110 jobs = xbt_dict_new();
111 root_task = SD_task_create_comp_seq("root", NULL, 0);
112 /* by design the root task is always SCHEDULABLE */
113 __SD_task_set_state(root_task, SD_SCHEDULABLE);
115 xbt_dict_set(jobs, "root", root_task, NULL);
116 xbt_dynar_push(result, &root_task);
117 end_task = SD_task_create_comp_seq("end", NULL, 0);
118 xbt_dict_set(jobs, "end", end_task, NULL);
120 Agnode_t *dag_node = NULL;
121 for (dag_node = agfstnode(dag_dot); dag_node;
123 dag_node = agnxtnode(dag_dot, dag_node)) {
125 dag_node = agnxtnode(dag_node)) {
128 dot_add_task(dag_node);
131 xbt_dict_free(&jobs);
133 /* And now, post-process the files.
134 * We want a file task per pair of computation tasks exchanging the file. Duplicate on need
135 * Files not produced in the system are said to be produced by root task (top of DAG).
136 * Files not consumed in the system are said to be consumed by end task (bottom of DAG).
138 xbt_dict_cursor_t cursor;
141 xbt_dict_foreach(files, cursor, name, file) {
142 unsigned int cpt1, cpt2;
143 SD_task_t newfile = NULL;
144 SD_dependency_t depbefore, depafter;
145 if (xbt_dynar_length(file->tasks_before) == 0) {
146 xbt_dynar_foreach(file->tasks_after, cpt2, depafter) {
148 SD_task_create_comm_e2e(file->name, NULL, file->amount);
149 SD_task_dependency_add(NULL, NULL, root_task, newfile);
150 SD_task_dependency_add(NULL, NULL, newfile, depafter->dst);
151 xbt_dynar_push(result, &newfile);
153 } else if (xbt_dynar_length(file->tasks_after) == 0) {
154 xbt_dynar_foreach(file->tasks_before, cpt2, depbefore) {
156 SD_task_create_comm_e2e(file->name, NULL, file->amount);
157 SD_task_dependency_add(NULL, NULL, depbefore->src, newfile);
158 SD_task_dependency_add(NULL, NULL, newfile, end_task);
159 xbt_dynar_push(result, &newfile);
162 xbt_dynar_foreach(file->tasks_before, cpt1, depbefore) {
163 xbt_dynar_foreach(file->tasks_after, cpt2, depafter) {
164 if (depbefore->src == depafter->dst) {
166 ("File %s is produced and consumed by task %s. This loop dependency will prevent the execution of the task.",
167 file->name, depbefore->src->name);
170 SD_task_create_comm_e2e(file->name, NULL, file->amount);
171 SD_task_dependency_add(NULL, NULL, depbefore->src, newfile);
172 SD_task_dependency_add(NULL, NULL, newfile, depafter->dst);
173 xbt_dynar_push(result, &newfile);
179 /* Push end task last */
180 xbt_dynar_push(result, &end_task);
182 /* Free previous copy of the files */
183 xbt_dict_free(&files);
188 /* dot_add_task create a sd_task and all transfers required for this
189 * task. The execution time of the task is given by the attribute size.
190 * The unit of size is the Flop.*/
191 void dot_add_task(Agnode_t * dag_node)
193 char *name = agnameof(dag_node);
194 SD_task_t current_job;
195 double runtime = dot_parse_double(agget(dag_node, (char *) "size"));
197 (long) dot_parse_int((char *) agget(dag_node, (char *) "performer"));
198 INFO3("See <job id=%s runtime=%s %.0f>", name,
199 agget(dag_node, (char *) "size"), runtime);
200 current_job = xbt_dict_get_or_null(jobs, name);
201 if (current_job == NULL) {
203 SD_task_create_comp_seq(name, (void *) performer, runtime);
205 TRACE_sd_dotloader (current_job, agget (dag_node, (char*)"category"));
207 xbt_dict_set(jobs, name, current_job, NULL);
208 xbt_dynar_push(result, ¤t_job);
214 for (e = agfstin(dag_dot, dag_node); e; e = agnxtin(dag_dot, e)) {
216 for (e = agfstin(dag_node); e; e = agnxtin(e)) {
218 dot_add_input_dependencies(current_job, e);
221 if (count == 0 && current_job != root_task) {
222 SD_task_dependency_add(NULL, NULL, root_task, current_job);
226 for (e = agfstout(dag_dot, dag_node); e; e = agnxtout(dag_dot, e)) {
228 for (e = agfstout(dag_node); e; e = agnxtout(e)) {
231 dot_add_output_dependencies(current_job, e);
234 if (count == 0 && current_job != end_task) {
235 SD_task_dependency_add(NULL, NULL, current_job, end_task);
239 /* dot_add_output_dependencies create the dependencies between a task
240 * and a transfers. This is given by the edges in the dot file.
241 * The amount of data transfers is given by the attribute size on the
243 void dot_add_input_dependencies(SD_task_t current_job, Agedge_t * edge)
248 sprintf(name, "%s->%s", agnameof(agtail(edge)), agnameof(aghead(edge)));
249 double size = dot_parse_double(agget(edge, (char *) "size"));
250 INFO2("size : %e, get size : %s", size, agget(edge, (char *) "size"));
251 //int sender = dot_parse_int(agget(edge,(char*)"sender"));
252 //int reciever = dot_parse_int(agget(edge,(char*)"reciever"));
254 file = xbt_dict_get_or_null(files, name);
256 file = SD_task_create_comm_e2e(name, NULL, size);
258 TRACE_sd_dotloader (file, agget (edge, (char*)"category"));
260 xbt_dict_set(files, name, file, &dot_task_free);
262 if (SD_task_get_amount(file) != size) {
263 WARN3("Ignoring file %s size redefinition from %.0f to %.0f",
264 name, SD_task_get_amount(file), size);
267 SD_task_dependency_add(NULL, NULL, file, current_job);
269 file = xbt_dict_get_or_null(jobs, agnameof(agtail(edge)));
271 SD_task_dependency_add(NULL, NULL, file, current_job);
276 /* dot_add_output_dependencies create the dependencies between a
277 * transfers and a task. This is given by the edges in the dot file.
278 * The amount of data transfers is given by the attribute size on the
280 void dot_add_output_dependencies(SD_task_t current_job, Agedge_t * edge)
284 sprintf(name, "%s->%s", agnameof(agtail(edge)), agnameof(aghead(edge)));
285 double size = dot_parse_double(agget(edge, (char *) "size"));
286 INFO2("size : %e, get size : %s", size, agget(edge, (char *) "size"));
287 //int sender = dot_parse_int(agget(edge,(char*)"sender"));
288 //int reciever = dot_parse_int(agget(edge,(char*)"reciever"));
290 //INFO2("See <uses file=%s %s>",A_dot__uses_file,(is_input?"in":"out"));
292 file = xbt_dict_get_or_null(files, name);
294 file = SD_task_create_comm_e2e(name, NULL, size);
296 TRACE_sd_dotloader (file, agget (edge, (char*)"category"));
298 xbt_dict_set(files, name, file, &dot_task_free);
300 if (SD_task_get_amount(file) != size) {
301 WARN3("Ignoring file %s size redefinition from %.0f to %.0f",
302 name, SD_task_get_amount(file), size);
305 SD_task_dependency_add(NULL, NULL, current_job, file);
306 if (xbt_dynar_length(file->tasks_before) > 1) {
307 WARN1("File %s created at more than one location...", file->name);
310 file = xbt_dict_get_or_null(jobs, agnameof(aghead(edge)));
312 SD_task_dependency_add(NULL, NULL, current_job, file);