Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
f22c5771141df014fd52318ac6faffbfbbc9dc24
[simgrid.git] / src / instr / jedule / jedule_sd_binding.cpp
1 /* Copyright (c) 2010-2015. 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 "xbt/asserts.h"
8 #include "xbt/dynar.h"
9
10 #include "src/surf/surf_private.h"
11 #include "surf/surf.h"
12
13 #include "simgrid/jedule/jedule_sd_binding.h"
14 #include "simgrid/jedule/jedule_events.h"
15 #include "simgrid/jedule/jedule_platform.h"
16 #include "simgrid/jedule/jedule_output.h"
17
18 #include "simgrid/simdag.h"
19 #include "src/simdag/simdag_private.h"
20
21 #include <stdio.h>
22
23 #ifdef HAVE_JEDULE
24
25 XBT_LOG_NEW_CATEGORY(jedule, "Logging specific to Jedule");
26 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(jed_sd, jedule, "Logging specific to Jedule SD binding");
27
28 jedule_t jedule;
29
30 void jedule_log_sd_event(SD_task_t task)
31 {
32   jed_event_t event;
33
34   xbt_assert(task != NULL);
35
36   xbt_dynar_t host_list = xbt_dynar_new(sizeof(char*), NULL);
37
38   for(int i=0; i<task->host_count; i++) {
39     const char *hostname = sg_host_get_name(task->host_list[i]);
40     xbt_dynar_push(host_list, &hostname);
41   }
42
43   create_jed_event(&event, (char*)SD_task_get_name(task), task->start_time, task->finish_time,"SD");
44
45   jed_event_add_resources(event, host_list);
46   jedule_store_event(event);
47
48   xbt_dynar_free(&host_list);
49 }
50
51 static void create_hierarchy(AS_t current_comp, jed_simgrid_container_t current_container)
52 {
53   xbt_dict_cursor_t cursor = NULL;
54   char *key;
55   AS_t elem;
56   xbt_dict_t routing_sons = surf_AS_get_children(current_comp);
57
58   if (xbt_dict_is_empty(routing_sons)) {
59     // I am no AS
60     // add hosts to jedule platform
61     xbt_dynar_t table = surf_AS_get_hosts(current_comp);
62     xbt_dynar_t hosts;
63     unsigned int dynar_cursor;
64     sg_host_t host_elem;
65
66     hosts = xbt_dynar_new(sizeof(char*), NULL);
67
68     xbt_dynar_foreach(table, dynar_cursor, host_elem) {
69       xbt_dynar_push_as(hosts, const char*, sg_host_get_name(host_elem));
70     }
71
72     jed_simgrid_add_resources(current_container, hosts);
73     xbt_dynar_free(&hosts);
74     xbt_dynar_free(&table);
75   } else {
76     xbt_dict_foreach(routing_sons, cursor, key, elem) {
77       jed_simgrid_container_t child_container;
78       jed_simgrid_create_container(&child_container, surf_AS_get_name(elem));
79       jed_simgrid_add_container(current_container, child_container);
80       XBT_DEBUG("name : %s\n", surf_AS_get_name(elem));
81       create_hierarchy(elem, child_container);
82     }
83   }
84 }
85
86 void jedule_setup_platform()
87 {
88   AS_t root_comp;
89
90   jed_simgrid_container_t root_container;
91
92   jed_create_jedule(&jedule);
93
94   root_comp = surf_AS_get_routing_root();
95   XBT_DEBUG("root name %s\n", surf_AS_get_name(root_comp));
96
97   jed_simgrid_create_container(&root_container, surf_AS_get_name(root_comp));
98   jedule->root_container = root_container;
99
100   create_hierarchy(root_comp, root_container);
101 }
102
103 void jedule_sd_cleanup()
104 {
105   jedule_cleanup_output();
106 }
107
108 void jedule_sd_init()
109 {
110   jedule_init_output();
111 }
112
113 void jedule_sd_exit(void)
114 {
115   if (jedule) {
116     jed_free_jedule(jedule);
117     jedule = NULL;
118   }
119 }
120
121 void jedule_sd_dump(const char * filename)
122 {
123   if (jedule) {
124     char *fname;
125     FILE *fh;
126     if (!filename) {
127       fname = bprintf("%s.jed", xbt_binary_name);
128     } else {
129       fname = xbt_strdup(filename);
130     }
131
132     fh = fopen(fname, "w");
133
134     write_jedule_output(fh, jedule, jedule_event_list, NULL);
135
136     fclose(fh);
137     free(fname);
138   }
139 }
140 #endif