Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
1c4ad33d17eed91d5967077ee5c5922dd85c709b
[simgrid.git] / src / instr / jedule / jedule_platform.cpp
1 /* Copyright (c) 2010-2018. 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 "simgrid/jedule/jedule.hpp"
7 #include "simgrid/host.h"
8 #include "simgrid/s4u/NetZone.hpp"
9 #include "xbt/asserts.h"
10 #include <algorithm>
11
12 #if SIMGRID_HAVE_JEDULE
13
14 namespace {
15 std::unordered_map<const char*, jed_container_t> host2_simgrid_parent_container;
16 std::unordered_map<std::string, jed_container_t> container_name2container;
17 }
18
19 namespace simgrid {
20 namespace jedule {
21 Subset::Subset(int start_idx, int end_idx, Container* parent)
22 : start_idx(start_idx), parent(parent)
23 {
24   nres=end_idx-start_idx+1;
25 }
26
27
28 Container::Container(std::string name): name(name)
29 {
30   container_name2container.insert({this->name, this});
31 }
32
33 Container::~Container()
34 {
35   if (not this->children.empty())
36     for (auto const& child : this->children)
37       delete child;
38 }
39
40 void Container::add_child(jed_container_t child)
41 {
42   xbt_assert(child != nullptr);
43   this->children.push_back(child);
44   child->parent = this;
45 }
46
47 void Container::add_resources(std::vector<sg_host_t> hosts)
48 {
49   this->is_lowest = 1;
50   this->children.clear();
51   this->last_id = 0;
52
53   //FIXME do we need to sort?: xbt_dynar_sort_strings(host_names);
54
55   for (auto const& host : hosts) {
56     const char *host_name = sg_host_get_name(host);
57     this->name2id.insert({host_name, this->last_id});
58     (this->last_id)++;
59     host2_simgrid_parent_container.insert({host_name, this});
60     this->resource_list.push_back(host);
61   }
62 }
63
64 void Container::create_hierarchy(sg_netzone_t from_as)
65 {
66
67   if (from_as->get_children()->empty()) {
68     // I am no AS
69     // add hosts to jedule platform
70     std::vector<sg_host_t> table = from_as->get_all_hosts();
71     this->add_resources(table);
72   } else {
73     for (auto const& nz : *from_as->get_children()) {
74       jed_container_t child_container = new simgrid::jedule::Container(std::string(nz->get_cname()));
75       this->add_child(child_container);
76       child_container->create_hierarchy(nz);
77     }
78   }
79 }
80
81 std::vector<int> Container::get_hierarchy()
82 {
83   if(this->parent != nullptr ) {
84
85     if (not this->parent->children.empty()) {
86       // we are in the last level
87       return this->parent->get_hierarchy();
88     } else {
89       unsigned int i =0;
90       int child_nb = -1;
91
92       for (auto const& child : this->parent->children) {
93         if( child == this) {
94           child_nb = i;
95           break;
96         }
97         i++;
98       }
99
100       xbt_assert( child_nb > - 1);
101       std::vector<int> heir_list = this->parent->get_hierarchy();
102       heir_list.insert(heir_list.begin(), child_nb);
103       return heir_list;
104     }
105   } else {
106     int top_level = 0;
107     std::vector<int> heir_list = {top_level};
108     return heir_list;
109   }
110 }
111
112 std::string Container::get_hierarchy_as_string()
113 {
114   std::string output("");
115
116   std::vector<int> heir_list = this->get_hierarchy();
117
118   unsigned int length = heir_list.size();
119   unsigned int i = 0;
120   for (auto const& id : heir_list) {
121     output += std::to_string(id);
122     if( i != length-1 ) {
123       output += ".";
124     }
125   }
126
127   return output;
128 }
129
130 void Container::print_resources(FILE* jed_file)
131 {
132   unsigned int i=0;
133   xbt_assert(not this->resource_list.empty());
134
135   unsigned int res_nb = this->resource_list.size();
136   std::string resid   = this->get_hierarchy_as_string();
137
138   fprintf(jed_file, "      <rset id=\"%s\" nb=\"%u\" names=\"", resid.c_str(), res_nb);
139   for (auto const& res : this->resource_list) {
140     const char * res_name = sg_host_get_name(res);
141     fprintf(jed_file, "%s", res_name);
142     if( i != res_nb-1 ) {
143       fprintf(jed_file, "|");
144     }
145     i++;
146   }
147   fprintf(jed_file, "\" />\n");
148 }
149
150 void Container::print(FILE* jed_file)
151 {
152   fprintf(jed_file, "    <res name=\"%s\">\n", this->name.c_str());
153   if (not this->children.empty()) {
154     for (auto const& child : this->children) {
155       child->print(jed_file);
156     }
157   } else {
158     this->print_resources(jed_file);
159   }
160   fprintf(jed_file, "    </res>\n");
161 }
162
163 }
164 }
165
166 static void add_subsets_to(std::vector<jed_subset_t> *subset_list, std::vector<const char*> hostgroup, jed_container_t parent)
167 {
168   // get ids for each host
169   // sort ids
170   // compact ids
171   // create subset for each id group
172
173   xbt_assert( parent != nullptr );
174
175   std::vector<unsigned int> id_list;
176
177   for (auto const& host_name : hostgroup) {
178     xbt_assert( host_name != nullptr );
179     jed_container_t parent = host2_simgrid_parent_container.at(host_name);
180     unsigned int id = parent->name2id.at(host_name);
181     id_list.push_back(id);
182   }
183   unsigned int nb_ids = id_list.size();
184   std::sort(id_list.begin(), id_list.end());
185
186   if( nb_ids > 0 ) {
187     int start = 0;
188     int pos = start;
189     for(unsigned int i=0; i<nb_ids; i++) {
190       if( id_list[i] - id_list[pos] > 1 ) {
191         subset_list->push_back(new simgrid::jedule::Subset(id_list[start], id_list[pos], parent));
192         start = i;
193
194         if( i == nb_ids-1 ) {
195          subset_list->push_back(new simgrid::jedule::Subset(id_list[i], id_list[i], parent));
196         }
197       } else {
198         if( i == nb_ids-1 ) {
199           subset_list->push_back(new simgrid::jedule::Subset(id_list[start], id_list[i], parent));
200         }
201       }
202       pos = i;
203     }
204   }
205
206 }
207
208 void get_resource_selection_by_hosts(std::vector<jed_subset_t> *subset_list, std::vector<sg_host_t> *host_list)
209 {
210   xbt_assert( host_list != nullptr );
211   // for each host name
212   //  find parent container
213   //  group by parent container
214   std::unordered_map<const char*, std::vector<const char*>> parent2hostgroup;
215   for (auto const& host : *host_list) {
216     const char *host_name = sg_host_get_name(host);
217     jed_container_t parent = host2_simgrid_parent_container.at(host_name);
218     xbt_assert( parent != nullptr );
219
220     auto host_group = parent2hostgroup.find(parent->name.c_str());
221     if (host_group == parent2hostgroup.end())
222       parent2hostgroup.insert({parent->name.c_str(), std::vector<const char*>(1,host_name)});
223     else
224       host_group->second.push_back(host_name);
225   }
226
227   for (auto const& elm : parent2hostgroup) {
228     jed_container_t parent = container_name2container.at(elm.first);
229     add_subsets_to(subset_list, elm.second, parent);
230   }
231 }
232
233 #endif