-/* Copyright (c) 2010-2015. The SimGrid Team.
- * All rights reserved. */
+/* Copyright (c) 2010-2021. The SimGrid Team. All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
* under the terms of the license (GNU LGPL) which comes with this package. */
-#include "simgrid/jedule/jedule_platform.h"
-
+#include "simgrid/jedule/jedule.hpp"
+#include "simgrid/config.h"
+#include "simgrid/host.h"
+#include "simgrid/s4u/NetZone.hpp"
#include "xbt/asserts.h"
-#include "xbt/dynar.h"
-#include "xbt/str.h"
-
-#include <stdlib.h>
-#include <string.h>
-
-#ifdef HAVE_JEDULE
-
-/********************************************************************/
-
-static xbt_dict_t host2_simgrid_parent_container;
-static xbt_dict_t container_name2container;
+#include <algorithm>
-/********************************************************************/
+#if SIMGRID_HAVE_JEDULE
-static void add_subset_to(xbt_dynar_t subset_list, int start, int end,
- jed_simgrid_container_t parent);
-
-static void add_subsets_to(xbt_dynar_t subset_list, xbt_dynar_t hostgroup,
- jed_simgrid_container_t parent);
-
-static void jed_free_container(jed_simgrid_container_t container);
+namespace {
+std::unordered_map<const char*, jed_container_t> host2_simgrid_parent_container;
+std::unordered_map<std::string, jed_container_t> container_name2container;
+}
-/********************************************************************/
+namespace simgrid {
+namespace jedule {
+Subset::Subset(int start_idx, int end_idx, Container* parent)
+ : start_idx(start_idx), nres(end_idx - start_idx + 1), parent(parent)
+{
+}
-static int compare_ids(const void *num1, const void *num2) {
- int *i1 = (int*) num1;
- int *i2 = (int*) num2;
- return *i1 - *i2;
+Container::Container(const std::string& name) : name(name)
+{
+ container_name2container.insert({this->name, this});
}
-static void jed_free_container(jed_simgrid_container_t container) {
+void Container::add_child(jed_container_t child)
+{
+ xbt_assert(child != nullptr);
+ children_.emplace_back(child);
+ child->set_parent(this);
+}
- xbt_dict_free(&container->name2id);
- xbt_dynar_free(&container->resource_list);
+void Container::add_resources(const std::vector<sg_host_t>& hosts)
+{
+ children_.clear();
+ last_id_ = 0;
+
+ for (auto const& host : hosts) {
+ const char *host_name = sg_host_get_name(host);
+ this->name2id.insert({host_name, this->last_id_});
+ (this->last_id_)++;
+ host2_simgrid_parent_container.insert({host_name, this});
+ this->resource_list.push_back(host);
+ }
+}
- if( container->container_children != NULL ) {
- unsigned int iter;
- jed_simgrid_container_t child_container;
- xbt_dynar_foreach(container->container_children, iter, child_container) {
- jed_free_container(child_container);
+void Container::create_hierarchy(const_sg_netzone_t from_as)
+{
+ if (from_as->get_children().empty()) {
+ // I am no AS
+ // add hosts to jedule platform
+ std::vector<sg_host_t> table = from_as->get_all_hosts();
+ this->add_resources(table);
+ } else {
+ for (auto const& nz : from_as->get_children()) {
+ auto* child_container = new simgrid::jedule::Container(nz->get_name());
+ this->add_child(child_container);
+ child_container->create_hierarchy(nz);
}
- xbt_dynar_free(&container->container_children);
}
-
- xbt_free(container->name);
- xbt_free(container);
}
-void jed_simgrid_create_container(jed_simgrid_container_t *container,
- const char *name)
+int Container::get_child_position(const Container* child) const
{
- xbt_assert( name != NULL );
-
- *container = xbt_new0(s_jed_simgrid_container_t,1);
- (*container)->name = xbt_strdup(name);
- (*container)->is_lowest = 0;
- (*container)->container_children = xbt_dynar_new(sizeof(jed_simgrid_container_t), NULL);
- (*container)->parent = NULL;
-
- xbt_dict_set(container_name2container, (*container)->name, *container, NULL);
+ auto it = std::find_if(begin(children_), end(children_),
+ [&child](const std::unique_ptr<Container>& c) { return c.get() == child; });
+ return it == end(children_) ? -1 : static_cast<int>(std::distance(begin(children_), it));
}
-
-void jed_simgrid_add_container(jed_simgrid_container_t parent,
- jed_simgrid_container_t child) {
- xbt_assert(parent != NULL);
- xbt_assert(child != NULL);
- xbt_dynar_push(parent->container_children, &child);
- child->parent = parent;
+std::vector<int> Container::get_hierarchy()
+{
+ if (parent_ == nullptr) {
+ int top_level = 0;
+ std::vector<int> hier_list = {top_level};
+ return hier_list;
+ } else if (parent_->has_children()) {
+ int child_nb = parent_->get_child_position(this);
+ xbt_assert(child_nb > -1);
+ std::vector<int> hier_list = parent_->get_hierarchy();
+ hier_list.insert(hier_list.begin(), child_nb);
+ return hier_list;
+ } else {
+ // we are in the last level
+ return parent_->get_hierarchy();
+ }
}
-void jed_simgrid_add_resources(jed_simgrid_container_t parent,
- xbt_dynar_t host_names) {
-
- unsigned int iter;
- char *host_name;
- char *buf;
-
- parent->is_lowest = 1;
- xbt_dynar_free(&parent->container_children);
- parent->container_children = NULL;
- parent->name2id = xbt_dict_new_homogeneous(xbt_free_f);
- parent->last_id = 0;
- parent->resource_list = xbt_dynar_new(sizeof(char *), NULL);
+std::string Container::get_hierarchy_as_string()
+{
+ std::string output("");
- xbt_dynar_sort_strings(host_names);
+ std::vector<int> hier_list = this->get_hierarchy();
- xbt_dynar_foreach(host_names, iter, host_name) {
- buf = bprintf("%d", parent->last_id);
- (parent->last_id)++;
- xbt_dict_set(parent->name2id, host_name, buf, NULL);
- xbt_dict_set(host2_simgrid_parent_container, host_name, parent, NULL);
- xbt_dynar_push(parent->resource_list, &host_name);
+ bool sep = false;
+ for (auto const& id : hier_list) {
+ if (sep)
+ output += '.';
+ else
+ sep = true;
+ output += std::to_string(id);
}
+ return output;
}
-static void add_subset_to(xbt_dynar_t subset_list, int start, int end,
- jed_simgrid_container_t parent) {
-
- jed_res_subset_t subset;
-
- xbt_assert( subset_list != NULL );
- xbt_assert( parent != NULL );
-
- // printf(">>> start=%d end=%d\n", start, end);
-
- subset = xbt_new0(s_jed_res_subset_t,1);
- subset->start_idx = start;
- subset->nres = end-start+1;
- subset->parent = parent;
-
- xbt_dynar_push(subset_list, &subset);
-
+void Container::print_resources(FILE* jed_file)
+{
+ xbt_assert(not this->resource_list.empty());
+
+ std::string resid = this->get_hierarchy_as_string();
+
+ fprintf(jed_file, " <rset id=\"%s\" nb=\"%zu\" names=\"", resid.c_str(), this->resource_list.size());
+ bool sep = false;
+ for (auto const& res : this->resource_list) {
+ if (sep)
+ putc('|', jed_file);
+ else
+ sep = true;
+ const char * res_name = sg_host_get_name(res);
+ fprintf(jed_file, "%s", res_name);
+ }
+ fprintf(jed_file, "\" />\n");
}
-static void add_subsets_to(xbt_dynar_t subset_list, xbt_dynar_t hostgroup,
- jed_simgrid_container_t parent) {
+void Container::print(FILE* jed_file)
+{
+ fprintf(jed_file, " <res name=\"%s\">\n", this->name.c_str());
+ if (not children_.empty()) {
+ for (auto const& child : children_) {
+ child->print(jed_file);
+ }
+ } else {
+ this->print_resources(jed_file);
+ }
+ fprintf(jed_file, " </res>\n");
+}
- unsigned int iter;
- char *host_name;
- xbt_dynar_t id_list;
- int *id_ar;
- int nb_ids;
- char *id_str;
+} // namespace jedule
+} // namespace simgrid
+static void add_subsets_to(std::vector<simgrid::jedule::Subset>& subset_list, const std::vector<const char*>& hostgroup,
+ jed_container_t parent)
+{
// get ids for each host
// sort ids
// compact ids
// create subset for each id group
- xbt_assert( host2_simgrid_parent_container != NULL );
- xbt_assert( subset_list != NULL );
- xbt_assert( hostgroup != NULL );
- xbt_assert( parent != NULL );
-
- id_list = xbt_dynar_new(sizeof(char *), NULL);
+ xbt_assert( parent != nullptr );
- xbt_dynar_foreach(hostgroup, iter, host_name) {
- jed_simgrid_container_t parent;
- xbt_assert( host_name != NULL );
- parent = (jed_simgrid_container_t)xbt_dict_get(host2_simgrid_parent_container, host_name);
- id_str = (char*)xbt_dict_get(parent->name2id, host_name);
- xbt_dynar_push(id_list, &id_str);
- }
+ std::vector<unsigned int> id_list;
- nb_ids = xbt_dynar_length(id_list);
- id_ar = xbt_new0(int,nb_ids);
- xbt_dynar_foreach(id_list, iter, id_str) {
- id_ar[iter] = xbt_str_parse_int(id_str, "Parse error: not a number: %s");
+ for (auto const& host_name : hostgroup) {
+ xbt_assert( host_name != nullptr );
+ const simgrid::jedule::Container* parent_cont = host2_simgrid_parent_container.at(host_name);
+ unsigned int id = parent_cont->get_id_by_name(host_name);
+ id_list.push_back(id);
}
-
- qsort (id_ar, nb_ids, sizeof(int), &compare_ids);
-
- if( nb_ids > 0 ) {
- int start = 0;
- int pos;
- int i;
-
- pos = start;
- for(i=0; i<nb_ids; i++) {
-
- if( id_ar[i] - id_ar[pos] > 1 ) {
-
- add_subset_to( subset_list, id_ar[start], id_ar[pos], parent );
- start = i;
-
- if( i == nb_ids-1 ) {
- add_subset_to( subset_list, id_ar[i], id_ar[i], parent );
- }
-
- } else {
- if( i == nb_ids-1 ) {
- add_subset_to( subset_list, id_ar[start], id_ar[i], parent );
- }
+ std::sort(id_list.begin(), id_list.end());
+
+ size_t nb_ids = id_list.size();
+ size_t start = 0;
+ size_t pos = start;
+ for (size_t i = 0; i < nb_ids; i++) {
+ if (id_list[i] - id_list[pos] > 1) {
+ subset_list.emplace_back(id_list[start], id_list[pos], parent);
+ start = i;
+
+ if (i == nb_ids - 1) {
+ subset_list.emplace_back(id_list[i], id_list[i], parent);
+ }
+ } else {
+ if (i == nb_ids - 1) {
+ subset_list.emplace_back(id_list[start], id_list[i], parent);
}
-
- pos = i;
}
+ pos = i;
}
-
- free(id_ar);
- xbt_dynar_free(&id_list);
-
-
}
-void jed_simgrid_get_resource_selection_by_hosts(xbt_dynar_t subset_list,
- xbt_dynar_t host_names) {
-
- char *host_name;
- unsigned int iter;
- xbt_dict_t parent2hostgroup; // group hosts by parent
-
- parent2hostgroup = xbt_dict_new_homogeneous(NULL);
-
- xbt_assert( host_names != NULL );
-
+void get_resource_selection_by_hosts(std::vector<simgrid::jedule::Subset>& subset_list,
+ const std::vector<sg_host_t>& host_list)
+{
// for each host name
// find parent container
// group by parent container
-
- xbt_dynar_foreach(host_names, iter, host_name) {
- //printf("checking %s \n", host_name);
-
- jed_simgrid_container_t parent = (jed_simgrid_container_t)xbt_dict_get(host2_simgrid_parent_container, host_name);
- xbt_assert( parent != NULL );
-
- xbt_dynar_t hostgroup = (xbt_dynar_t)xbt_dict_get_or_null (parent2hostgroup, parent->name);
- if( hostgroup == NULL ) {
- hostgroup = xbt_dynar_new(sizeof(char*), NULL);
- xbt_dict_set(parent2hostgroup, parent->name, hostgroup, NULL);
- }
-
- xbt_dynar_push(hostgroup, &host_name);
+ std::unordered_map<const char*, std::vector<const char*>> parent2hostgroup;
+ for (auto const& host : host_list) {
+ const char *host_name = sg_host_get_name(host);
+ const simgrid::jedule::Container* parent = host2_simgrid_parent_container.at(host_name);
+ xbt_assert( parent != nullptr );
+
+ auto host_group = parent2hostgroup.find(parent->get_cname());
+ if (host_group == parent2hostgroup.end())
+ parent2hostgroup.insert({parent->get_cname(), std::vector<const char*>(1, host_name)});
+ else
+ host_group->second.push_back(host_name);
}
- {
- xbt_dict_cursor_t cursor=NULL;
- char *parent_name;
- xbt_dynar_t hostgroup;
- jed_simgrid_container_t parent;
-
- xbt_dict_foreach(parent2hostgroup,cursor,parent_name,hostgroup) {
- parent = (jed_simgrid_container_t)xbt_dict_get(container_name2container, parent_name);
- // printf("subset parent >>> %s\n", parent->name);
- add_subsets_to(subset_list, hostgroup, parent);
- }
- xbt_dynar_free(&hostgroup);
+ for (auto const& elm : parent2hostgroup) {
+ jed_container_t parent = container_name2container.at(elm.first);
+ add_subsets_to(subset_list, elm.second, parent);
}
-
- xbt_dict_free(&parent2hostgroup);
-
-}
-
-
-void jedule_add_meta_info(jedule_t jedule, char *key, char *value) {
-
- char *val_cp;
-
- xbt_assert(key != NULL);
- xbt_assert(value != NULL);
-
- val_cp = xbt_strdup(value);
- xbt_dict_set(jedule->jedule_meta_info, key, val_cp, NULL);
-}
-
-void jed_create_jedule(jedule_t *jedule) {
- *jedule = xbt_new0(s_jedule_t,1);
- host2_simgrid_parent_container = xbt_dict_new_homogeneous(NULL);
- container_name2container = xbt_dict_new_homogeneous(NULL);
- (*jedule)->jedule_meta_info = xbt_dict_new_homogeneous(NULL);
-}
-
-void jed_free_jedule(jedule_t jedule) {
-
- jed_free_container(jedule->root_container);
-
- xbt_dict_free(&jedule->jedule_meta_info);
- xbt_free(jedule);
-
- xbt_dict_free(&host2_simgrid_parent_container);
- xbt_dict_free(&container_name2container);
}
#endif