1 /* Copyright (c) 2010-2015. 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. */
9 #include "xbt/asserts.h"
10 #include "xbt/dynar.h"
12 #include "instr/jedule/jedule_platform.h"
16 /********************************************************************/
18 static xbt_dict_t host2_simgrid_parent_container;
19 static xbt_dict_t container_name2container;
21 /********************************************************************/
23 static void add_subset_to(xbt_dynar_t subset_list, int start, int end,
24 jed_simgrid_container_t parent);
26 static void add_subsets_to(xbt_dynar_t subset_list, xbt_dynar_t hostgroup,
27 jed_simgrid_container_t parent);
29 static void jed_free_container(jed_simgrid_container_t container);
31 /********************************************************************/
33 static int compare_hostnames(const void *host1, const void *host2) {
34 const char *hp1 = *((const char**) host1);
35 const char *hp2 = *((const char**) host2);
36 return strcmp (hp1, hp2);
39 static int compare_ids(const void *num1, const void *num2) {
40 int *i1 = (int*) num1;
41 int *i2 = (int*) num2;
45 static void jed_free_container(jed_simgrid_container_t container) {
47 xbt_dict_free(&container->name2id);
48 xbt_dynar_free(&container->resource_list);
50 if( container->container_children != NULL ) {
52 jed_simgrid_container_t child_container;
53 xbt_dynar_foreach(container->container_children, iter, child_container) {
54 jed_free_container(child_container);
56 xbt_dynar_free(&container->container_children);
59 xbt_free(container->name);
63 void jed_simgrid_create_container(jed_simgrid_container_t *container,
66 xbt_assert( name != NULL );
68 *container = xbt_new0(s_jed_simgrid_container_t,1);
69 (*container)->name = xbt_strdup(name);
70 (*container)->is_lowest = 0;
71 (*container)->container_children = xbt_dynar_new(sizeof(jed_simgrid_container_t), NULL);
72 (*container)->parent = NULL;
74 xbt_dict_set(container_name2container, (*container)->name, *container, NULL);
78 void jed_simgrid_add_container(jed_simgrid_container_t parent,
79 jed_simgrid_container_t child) {
80 xbt_assert(parent != NULL);
81 xbt_assert(child != NULL);
82 xbt_dynar_push(parent->container_children, &child);
83 child->parent = parent;
86 void jed_simgrid_add_resources(jed_simgrid_container_t parent,
87 xbt_dynar_t host_names) {
93 parent->is_lowest = 1;
94 xbt_dynar_free(&parent->container_children);
95 parent->container_children = NULL;
96 parent->name2id = xbt_dict_new_homogeneous(xbt_free_f);
98 parent->resource_list = xbt_dynar_new(sizeof(char *), NULL);
100 xbt_dynar_sort (host_names, &compare_hostnames);
102 xbt_dynar_foreach(host_names, iter, host_name) {
103 buf = bprintf("%d", parent->last_id);
105 xbt_dict_set(parent->name2id, host_name, buf, NULL);
106 xbt_dict_set(host2_simgrid_parent_container, host_name, parent, NULL);
107 xbt_dynar_push(parent->resource_list, &host_name);
112 static void add_subset_to(xbt_dynar_t subset_list, int start, int end,
113 jed_simgrid_container_t parent) {
115 jed_res_subset_t subset;
117 xbt_assert( subset_list != NULL );
118 xbt_assert( parent != NULL );
120 // printf(">>> start=%d end=%d\n", start, end);
122 subset = xbt_new0(s_jed_res_subset_t,1);
123 subset->start_idx = start;
124 subset->nres = end-start+1;
125 subset->parent = parent;
127 xbt_dynar_push(subset_list, &subset);
131 static void add_subsets_to(xbt_dynar_t subset_list, xbt_dynar_t hostgroup,
132 jed_simgrid_container_t parent) {
141 // get ids for each host
144 // create subset for each id group
146 xbt_assert( host2_simgrid_parent_container != NULL );
147 xbt_assert( subset_list != NULL );
148 xbt_assert( hostgroup != NULL );
149 xbt_assert( parent != NULL );
151 id_list = xbt_dynar_new(sizeof(char *), NULL);
153 xbt_dynar_foreach(hostgroup, iter, host_name) {
154 jed_simgrid_container_t parent;
155 xbt_assert( host_name != NULL );
156 parent = xbt_dict_get(host2_simgrid_parent_container, host_name);
157 id_str = xbt_dict_get(parent->name2id, host_name);
158 xbt_dynar_push(id_list, &id_str);
161 nb_ids = xbt_dynar_length(id_list);
162 id_ar = xbt_new0(int,nb_ids);
163 xbt_dynar_foreach(id_list, iter, id_str) {
164 id_ar[iter] = atoi(id_str);
167 qsort (id_ar, nb_ids, sizeof(int), &compare_ids);
175 for(i=0; i<nb_ids; i++) {
177 if( id_ar[i] - id_ar[pos] > 1 ) {
179 add_subset_to( subset_list, id_ar[start], id_ar[pos], parent );
182 if( i == nb_ids-1 ) {
183 add_subset_to( subset_list, id_ar[i], id_ar[i], parent );
187 if( i == nb_ids-1 ) {
188 add_subset_to( subset_list, id_ar[start], id_ar[i], parent );
197 xbt_dynar_free(&id_list);
202 void jed_simgrid_get_resource_selection_by_hosts(xbt_dynar_t subset_list,
203 xbt_dynar_t host_names) {
207 xbt_dict_t parent2hostgroup; // group hosts by parent
209 parent2hostgroup = xbt_dict_new_homogeneous(NULL);
211 xbt_assert( host_names != NULL );
213 // for each host name
214 // find parent container
215 // group by parent container
217 xbt_dynar_foreach(host_names, iter, host_name) {
218 jed_simgrid_container_t parent;
219 xbt_dynar_t hostgroup;
221 //printf("checking %s \n", host_name);
223 parent = xbt_dict_get(host2_simgrid_parent_container, host_name);
224 xbt_assert( parent != NULL );
226 hostgroup = xbt_dict_get_or_null (parent2hostgroup, parent->name);
227 if( hostgroup == NULL ) {
228 hostgroup = xbt_dynar_new(sizeof(char*), NULL);
229 xbt_dict_set(parent2hostgroup, parent->name, hostgroup, NULL);
232 xbt_dynar_push(hostgroup, &host_name);
236 xbt_dict_cursor_t cursor=NULL;
238 xbt_dynar_t hostgroup;
239 jed_simgrid_container_t parent;
241 xbt_dict_foreach(parent2hostgroup,cursor,parent_name,hostgroup) {
242 parent = xbt_dict_get(container_name2container, parent_name);
243 // printf("subset parent >>> %s\n", parent->name);
244 add_subsets_to(subset_list, hostgroup, parent);
246 xbt_dynar_free(&hostgroup);
249 xbt_dict_free(&parent2hostgroup);
254 void jedule_add_meta_info(jedule_t jedule, char *key, char *value) {
258 xbt_assert(key != NULL);
259 xbt_assert(value != NULL);
261 val_cp = xbt_strdup(value);
262 xbt_dict_set(jedule->jedule_meta_info, key, val_cp, NULL);
265 void jed_create_jedule(jedule_t *jedule) {
266 *jedule = xbt_new0(s_jedule_t,1);
267 host2_simgrid_parent_container = xbt_dict_new_homogeneous(NULL);
268 container_name2container = xbt_dict_new_homogeneous(NULL);
269 (*jedule)->jedule_meta_info = xbt_dict_new_homogeneous(NULL);
272 void jed_free_jedule(jedule_t jedule) {
274 jed_free_container(jedule->root_container);
276 xbt_dict_free(&jedule->jedule_meta_info);
279 xbt_dict_free(&host2_simgrid_parent_container);
280 xbt_dict_free(&container_name2container);