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 void dot_add_task(Agnode_t * dag_node);
18 void dot_add_input_dependencies(SD_task_t current_job, Agedge_t * edge);
19 void dot_add_output_dependencies(SD_task_t current_job, Agedge_t * edge);
20 xbt_dynar_t SD_dotload_FILE(FILE * in_file);
22 static double dot_parse_double(const char *string)
29 ret = sscanf(string, "%lg", &value);
31 WARN1("%s is not a double", string);
35 static int dot_parse_int(const char *string)
42 ret = sscanf(string, "%d", &value);
44 WARN1("%s is not an integer", string);
48 static xbt_dynar_t result;
49 static xbt_dict_t jobs;
50 static xbt_dict_t files;
51 static SD_task_t root_task, end_task;
52 static Agraph_t *dag_dot;
54 static void dump_res()
58 xbt_dynar_foreach(result, cursor, task) {
59 INFO1("Task %d", cursor);
64 static void dot_task_free(void *task)
70 /** @brief loads a DOT file describing a DAG
72 * See http://www.graphviz.org/doc/info/lang.html
74 * To obtain information about transfers and tasks, two attributes are
75 * required : size on task (execution time in Flop) and size on edge
76 * (the amount of data transfer in bit).
77 * if they aren't here, there choose to be equal to zero.
79 xbt_dynar_t SD_dotload(const char *filename)
81 FILE *in_file = fopen(filename, "r");
82 xbt_assert1(in_file, "Unable to open \"%s\"\n", filename);
83 SD_dotload_FILE(in_file);
88 xbt_dynar_t SD_dotload_FILE(FILE * in_file)
90 xbt_assert0(in_file, "Unable to use a null file descriptor\n");
91 dag_dot = agread(in_file, NIL(Agdisc_t *));
93 result = xbt_dynar_new(sizeof(SD_task_t), dot_task_free);
94 files = xbt_dict_new();
95 jobs = xbt_dict_new();
96 root_task = SD_task_create_comp_seq("root", NULL, 0);
97 /* by design the root task is always SCHEDULABLE */
98 __SD_task_set_state(root_task, SD_SCHEDULABLE);
100 xbt_dict_set(jobs, "root", root_task, NULL);
101 xbt_dynar_push(result, &root_task);
102 end_task = SD_task_create_comp_seq("end", NULL, 0);
103 xbt_dict_set(jobs, "end", end_task, NULL);
105 Agnode_t *dag_node = NULL;
106 for (dag_node = agfstnode(dag_dot); dag_node;
107 dag_node = agnxtnode(dag_dot, dag_node)) {
108 dot_add_task(dag_node);
111 xbt_dict_free(&jobs);
113 /* And now, post-process the files.
114 * We want a file task per pair of computation tasks exchanging the file. Duplicate on need
115 * Files not produced in the system are said to be produced by root task (top of DAG).
116 * Files not consumed in the system are said to be consumed by end task (bottom of DAG).
118 xbt_dict_cursor_t cursor;
121 xbt_dict_foreach(files, cursor, name, file) {
122 unsigned int cpt1, cpt2;
123 SD_task_t newfile = NULL;
124 SD_dependency_t depbefore, depafter;
125 if (xbt_dynar_length(file->tasks_before) == 0) {
126 xbt_dynar_foreach(file->tasks_after, cpt2, depafter) {
128 SD_task_create_comm_e2e(file->name, NULL, file->amount);
129 SD_task_dependency_add(NULL, NULL, root_task, newfile);
130 SD_task_dependency_add(NULL, NULL, newfile, depafter->dst);
131 xbt_dynar_push(result, &newfile);
133 } else if (xbt_dynar_length(file->tasks_after) == 0) {
134 xbt_dynar_foreach(file->tasks_before, cpt2, depbefore) {
136 SD_task_create_comm_e2e(file->name, NULL, file->amount);
137 SD_task_dependency_add(NULL, NULL, depbefore->src, newfile);
138 SD_task_dependency_add(NULL, NULL, newfile, end_task);
139 xbt_dynar_push(result, &newfile);
142 xbt_dynar_foreach(file->tasks_before, cpt1, depbefore) {
143 xbt_dynar_foreach(file->tasks_after, cpt2, depafter) {
144 if (depbefore->src == depafter->dst) {
146 ("File %s is produced and consumed by task %s. This loop dependency will prevent the execution of the task.",
147 file->name, depbefore->src->name);
150 SD_task_create_comm_e2e(file->name, NULL, file->amount);
151 SD_task_dependency_add(NULL, NULL, depbefore->src, newfile);
152 SD_task_dependency_add(NULL, NULL, newfile, depafter->dst);
153 xbt_dynar_push(result, &newfile);
159 /* Push end task last */
160 xbt_dynar_push(result, &end_task);
162 /* Free previous copy of the files */
163 xbt_dict_free(&files);
168 /* dot_add_task create a sd_task and all transfers required for this
169 * task. The execution time of the task is given by the attribute size.
170 * The unit of size is the Flop.*/
171 void dot_add_task(Agnode_t * dag_node)
173 char *name = agnameof(dag_node);
174 SD_task_t current_job;
175 double runtime = dot_parse_double(agget(dag_node, (char *) "size"));
177 (long) dot_parse_int((char *) agget(dag_node, (char *) "performer"));
178 INFO3("See <job id=%s runtime=%s %.0f>", name,
179 agget(dag_node, (char *) "size"), runtime);
180 current_job = xbt_dict_get_or_null(jobs, name);
181 if (current_job == NULL) {
183 SD_task_create_comp_seq(name, (void *) performer, runtime);
184 xbt_dict_set(jobs, name, current_job, NULL);
185 xbt_dynar_push(result, ¤t_job);
189 for (e = agfstin(dag_dot, dag_node); e; e = agnxtin(dag_dot, e)) {
190 dot_add_input_dependencies(current_job, e);
193 if (count == 0 && current_job != root_task) {
194 SD_task_dependency_add(NULL, NULL, root_task, current_job);
197 for (e = agfstout(dag_dot, dag_node); e; e = agnxtout(dag_dot, e)) {
198 dot_add_output_dependencies(current_job, e);
201 if (count == 0 && current_job != end_task) {
202 SD_task_dependency_add(NULL, NULL, current_job, end_task);
206 /* dot_add_output_dependencies create the dependencies between a task
207 * and a transfers. This is given by the edges in the dot file.
208 * The amount of data transfers is given by the attribute size on the
210 void dot_add_input_dependencies(SD_task_t current_job, Agedge_t * edge)
215 sprintf(name, "%s->%s", agnameof(agtail(edge)), agnameof(aghead(edge)));
216 double size = dot_parse_double(agget(edge, (char *) "size"));
217 INFO2("size : %e, get size : %s", size, agget(edge, (char *) "size"));
218 //int sender = dot_parse_int(agget(edge,(char*)"sender"));
219 //int reciever = dot_parse_int(agget(edge,(char*)"reciever"));
221 file = xbt_dict_get_or_null(files, name);
223 file = SD_task_create_comm_e2e(name, NULL, size);
224 xbt_dict_set(files, name, file, &dot_task_free);
226 if (SD_task_get_amount(file) != size) {
227 WARN3("Ignoring file %s size redefinition from %.0f to %.0f",
228 name, SD_task_get_amount(file), size);
231 SD_task_dependency_add(NULL, NULL, file, current_job);
233 file = xbt_dict_get_or_null(jobs, agnameof(agtail(edge)));
235 SD_task_dependency_add(NULL, NULL, file, current_job);
240 /* dot_add_output_dependencies create the dependencies between a
241 * transfers and a task. This is given by the edges in the dot file.
242 * The amount of data transfers is given by the attribute size on the
244 void dot_add_output_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 //INFO2("See <uses file=%s %s>",A_dot__uses_file,(is_input?"in":"out"));
256 file = xbt_dict_get_or_null(files, name);
258 file = SD_task_create_comm_e2e(name, NULL, size);
259 xbt_dict_set(files, name, file, &dot_task_free);
261 if (SD_task_get_amount(file) != size) {
262 WARN3("Ignoring file %s size redefinition from %.0f to %.0f",
263 name, SD_task_get_amount(file), size);
266 SD_task_dependency_add(NULL, NULL, current_job, file);
267 if (xbt_dynar_length(file->tasks_before) > 1) {
268 WARN1("File %s created at more than one location...", file->name);
271 file = xbt_dict_get_or_null(jobs, agnameof(aghead(edge)));
273 SD_task_dependency_add(NULL, NULL, current_job, file);