src/surf/workstation.c

   1 /*      $Id$     */
   2
   3 /* Copyright (c) 2004 Arnaud Legrand. All rights reserved.                  */
   4
   5 /* This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the license (GNU LGPL) which comes with this package. */
   7
   8 #include "xbt/ex.h"
   9 #include "xbt/dict.h"
  10 #include "workstation_private.h"
  11 #include "cpu_private.h"
  12 #include "network_private.h"
  13
  14 surf_workstation_resource_t surf_workstation_resource = NULL;
  15
  16 xbt_dict_t workstation_set = NULL;
  17
  18 static workstation_CLM03_t workstation_new(const char *name,
  19                                      void *cpu, void *card)
  20 {
  21   workstation_CLM03_t workstation = xbt_new0(s_workstation_CLM03_t, 1);
  22
  23   workstation->resource = (surf_resource_t) surf_workstation_resource;
  24   workstation->name = xbt_strdup(name);
  25   workstation->cpu = cpu;
  26   workstation->network_card = card;
  27
  28   return workstation;
  29 }
  30
  31 static void workstation_free(void *workstation)
  32 {
  33   free(((workstation_CLM03_t)workstation)->name);
  34   free(workstation);
  35 }
  36
  37 static void create_workstations(void)
  38 {
  39   xbt_dict_cursor_t cursor = NULL;
  40   char *name = NULL;
  41   void *cpu = NULL;
  42   void *nw_card = NULL;
  43
  44   xbt_dict_foreach(cpu_set, cursor, name, cpu) {
  45     nw_card = xbt_dict_get_or_null(network_card_set, name);
  46     xbt_assert1(nw_card,
  47                 "No corresponding card found for %s",name);
  48
  49     xbt_dict_set(workstation_set, name,
  50                  workstation_new(name, cpu, nw_card), workstation_free);
  51   }
  52 }
  53
  54 static void *name_service(const char *name)
  55 {
  56   return xbt_dict_get_or_null(workstation_set, name);
  57 }
  58
  59 static const char *get_resource_name(void *resource_id)
  60 {
  61   return ((workstation_CLM03_t) resource_id)->name;
  62 }
  63
  64 static int resource_used(void *resource_id)
  65 {
  66   xbt_assert0(0,
  67               "Workstation is a virtual resource. I should not be there!");
  68   return 0;
  69 }
  70
  71 static int parallel_action_free(surf_action_t action)
  72 {
  73   action->using--;
  74   if(!action->using) {
  75     xbt_swag_remove(action, action->state_set);
  76     if(((surf_action_parallel_task_CSL05_t)action)->variable)
  77       lmm_variable_free(maxmin_system, ((surf_action_parallel_task_CSL05_t)action)->variable);
  78     free(action);
  79     return 1;
  80   }
  81   return 0;
  82 }
  83
  84 static void parallel_action_use(surf_action_t action)
  85 {
  86   action->using++;
  87 }
  88
  89 static int action_free(surf_action_t action)
  90 {
  91   if(action->resource_type==(surf_resource_t)surf_network_resource)
  92     return surf_network_resource->common_public->action_free(action);
  93   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
  94     return surf_cpu_resource->common_public->action_free(action);
  95   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
  96     return parallel_action_free(action);
  97   else DIE_IMPOSSIBLE;
  98   return 0;
  99 }
 100
 101 static void action_use(surf_action_t action)
 102 {
 103   if(action->resource_type==(surf_resource_t)surf_network_resource)
 104     surf_network_resource->common_public->action_use(action);
 105   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 106     surf_cpu_resource->common_public->action_use(action);
 107   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
 108     parallel_action_use(action);
 109   else DIE_IMPOSSIBLE;
 110   return;
 111 }
 112
 113 static void action_cancel(surf_action_t action)
 114 {
 115   if(action->resource_type==(surf_resource_t)surf_network_resource)
 116     surf_network_resource->common_public->action_cancel(action);
 117   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 118     surf_cpu_resource->common_public->action_cancel(action);
 119   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
 120     parallel_action_use(action);
 121   else DIE_IMPOSSIBLE;
 122   return;
 123 }
 124
 125 static void action_recycle(surf_action_t action)
 126 {
 127   DIE_IMPOSSIBLE;
 128   return;
 129 }
 130
 131 static void action_change_state(surf_action_t action,
 132                                 e_surf_action_state_t state)
 133 {
 134   if(action->resource_type==(surf_resource_t)surf_network_resource)
 135     surf_network_resource->common_public->action_change_state(action,state);
 136   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 137     surf_cpu_resource->common_public->action_change_state(action,state);
 138   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
 139     surf_action_change_state(action, state);
 140   else DIE_IMPOSSIBLE;
 141   return;
 142 }
 143
 144 static double share_resources(double now)
 145 {
 146   s_surf_action_parallel_task_CSL05_t action;
 147   return generic_maxmin_share_resources(surf_workstation_resource->common_public->
 148                                         states.running_action_set,
 149                                         xbt_swag_offset(action, variable));
 150 }
 151
 152 static void update_actions_state(double now, double delta)
 153 {
 154   surf_action_parallel_task_CSL05_t action = NULL;
 155   surf_action_parallel_task_CSL05_t next_action = NULL;
 156   xbt_swag_t running_actions =
 157       surf_workstation_resource->common_public->states.running_action_set;
 158   /* FIXME: unused
 159   xbt_swag_t failed_actions =
 160       surf_workstation_resource->common_public->states.failed_action_set;
 161   */
 162
 163   xbt_swag_foreach_safe(action, next_action, running_actions) {
 164     surf_double_update(&(action->generic_action.remains),
 165         lmm_variable_getvalue(action->variable) * delta);
 166     if (action->generic_action.max_duration != NO_MAX_DURATION)
 167       surf_double_update(&(action->generic_action.max_duration), delta);
 168     if ((action->generic_action.remains <= 0) &&
 169         (lmm_get_variable_weight(action->variable)>0)) {
 170       action->generic_action.finish = surf_get_clock();
 171       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
 172     } else if ((action->generic_action.max_duration != NO_MAX_DURATION) &&
 173                (action->generic_action.max_duration <= 0)) {
 174       action->generic_action.finish = surf_get_clock();
 175       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
 176     } else {                    /* Need to check that none of the resource has failed */
 177       lmm_constraint_t cnst = NULL;
 178       int i = 0;
 179       surf_resource_t resource = NULL;
 180
 181       while ((cnst =
 182               lmm_get_cnst_from_var(maxmin_system, action->variable,
 183                                     i++))) {
 184         resource = (surf_resource_t) lmm_constraint_id(cnst);
 185         if(resource== (surf_resource_t) surf_cpu_resource) {
 186           cpu_Cas01_t cpu = lmm_constraint_id(cnst);
 187           if (cpu->state_current == SURF_CPU_OFF) {
 188             action->generic_action.finish = surf_get_clock();
 189             action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
 190             break;
 191           }
 192         } else if (resource== (surf_resource_t) surf_network_resource) {
 193           network_link_CM02_t nw_link = lmm_constraint_id(cnst);
 194
 195           if (nw_link->state_current == SURF_NETWORK_LINK_OFF) {
 196             action->generic_action.finish = surf_get_clock();
 197             action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
 198             break;
 199           }
 200         }
 201       }
 202     }
 203   }
 204
 205   return;
 206 }
 207
 208 static void update_resource_state(void *id,
 209                                   tmgr_trace_event_t event_type,
 210                                   double value)
 211 {
 212   return;
 213 }
 214
 215 static surf_action_t execute(void *workstation, double size)
 216 {
 217   return surf_cpu_resource->extension_public->
 218       execute(((workstation_CLM03_t) workstation)->cpu, size);
 219 }
 220
 221 static surf_action_t action_sleep(void *workstation, double duration)
 222 {
 223   return surf_cpu_resource->extension_public->
 224       sleep(((workstation_CLM03_t) workstation)->cpu, duration);
 225 }
 226
 227 static void action_suspend(surf_action_t action)
 228 {
 229   if(action->resource_type==(surf_resource_t)surf_network_resource)
 230     surf_network_resource->common_public->suspend(action);
 231   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 232     surf_cpu_resource->common_public->suspend(action);
 233   else DIE_IMPOSSIBLE;
 234 }
 235
 236 static void action_resume(surf_action_t action)
 237 {
 238   if(action->resource_type==(surf_resource_t)surf_network_resource)
 239     surf_network_resource->common_public->resume(action);
 240   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 241     surf_cpu_resource->common_public->resume(action);
 242   else DIE_IMPOSSIBLE;
 243 }
 244
 245 static int action_is_suspended(surf_action_t action)
 246 {
 247   if(action->resource_type==(surf_resource_t)surf_network_resource)
 248     return surf_network_resource->common_public->is_suspended(action);
 249   if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 250     return surf_cpu_resource->common_public->is_suspended(action);
 251   DIE_IMPOSSIBLE;
 252 }
 253
 254 static void action_set_max_duration(surf_action_t action, double duration)
 255 {
 256   if(action->resource_type==(surf_resource_t)surf_network_resource)
 257     surf_network_resource->common_public->set_max_duration(action,duration);
 258   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 259     surf_cpu_resource->common_public->set_max_duration(action,duration);
 260   else  DIE_IMPOSSIBLE;
 261 }
 262
 263 static void action_set_priority(surf_action_t action, double priority)
 264 {
 265   if(action->resource_type==(surf_resource_t)surf_network_resource)
 266     surf_network_resource->common_public->set_priority(action,priority);
 267   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 268     surf_cpu_resource->common_public->set_priority(action,priority);
 269   else  DIE_IMPOSSIBLE;
 270 }
 271
 272 static surf_action_t communicate(void *workstation_src,
 273                                  void *workstation_dst, double size,
 274                                  double rate)
 275 {
 276   return surf_network_resource->extension_public->
 277       communicate(((workstation_CLM03_t) workstation_src)->network_card,
 278                   ((workstation_CLM03_t) workstation_dst)->network_card, size, rate);
 279 }
 280
 281 static e_surf_cpu_state_t get_state(void *workstation)
 282 {
 283   return surf_cpu_resource->extension_public->
 284       get_state(((workstation_CLM03_t) workstation)->cpu);
 285 }
 286
 287 static double get_speed(void *workstation, double load)
 288 {
 289   return surf_cpu_resource->extension_public->
 290       get_speed(((workstation_CLM03_t) workstation)->cpu, load);
 291 }
 292
 293 static double get_available_speed(void *workstation)
 294 {
 295   return surf_cpu_resource->extension_public->
 296       get_available_speed(((workstation_CLM03_t) workstation)->cpu);
 297 }
 298
 299 static surf_action_t execute_parallel_task (int workstation_nb,
 300                                             void **workstation_list,
 301                                             double *computation_amount,
 302                                             double *communication_amount,
 303                                             double amount,
 304                                             double rate)
 305 {
 306   surf_action_parallel_task_CSL05_t action = NULL;
 307   int i, j, k;
 308   static xbt_dict_t network_link_set;
 309   static int first_run = 1;
 310   int nb_link = 0;
 311   int nb_host = 0;
 312
 313   if (first_run) {
 314     network_link_set = xbt_dict_new_ext(workstation_nb * workstation_nb * 10);
 315     first_run = 0;
 316   }
 317   else {
 318     xbt_dict_reset(network_link_set);
 319   }
 320
 321   /* Compute the number of affected resources... */
 322   for(i=0; i< workstation_nb; i++) {
 323     for(j=0; j< workstation_nb; j++) {
 324       network_card_CM02_t card_src = ((workstation_CLM03_t*)workstation_list)[i]->network_card;
 325       network_card_CM02_t card_dst = ((workstation_CLM03_t*)workstation_list)[j]->network_card;
 326       int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
 327       network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
 328
 329       if(communication_amount[i*workstation_nb+j]>0)
 330         for(k=0; k< route_size; k++) {
 331           xbt_dict_set(network_link_set, route[k]->name, route[k], NULL);
 332         }
 333     }
 334   }
 335
 336   nb_link = xbt_dict_length(network_link_set);
 337
 338   for (i = 0; i<workstation_nb; i++)
 339     if(computation_amount[i]>0) nb_host++;
 340
 341   if(nb_link + workstation_nb == 0)
 342     return NULL;
 343
 344   action = xbt_new0(s_surf_action_parallel_task_CSL05_t, 1);
 345   action->generic_action.using = 1;
 346   action->generic_action.cost = amount;
 347   action->generic_action.remains = amount;
 348   action->generic_action.max_duration = NO_MAX_DURATION;
 349   action->generic_action.start = surf_get_clock();
 350   action->generic_action.finish = -1.0;
 351   action->generic_action.resource_type =
 352       (surf_resource_t) surf_workstation_resource;
 353   action->suspended = 0;  /* Should be useless because of the
 354                              calloc but it seems to help valgrind... */
 355   action->generic_action.state_set =
 356       surf_workstation_resource->common_public->states.running_action_set;
 357
 358   xbt_swag_insert(action, action->generic_action.state_set);
 359   action->rate = rate;
 360
 361   if(action->rate>0)
 362     action->variable = lmm_variable_new(maxmin_system, action, 1.0, -1.0,
 363                                         nb_host + nb_link);
 364   else
 365     action->variable = lmm_variable_new(maxmin_system, action, 1.0, action->rate,
 366                                         nb_host + nb_link);
 367
 368   for (i = 0; i<workstation_nb; i++)
 369     if(computation_amount[i]>0)
 370       lmm_expand(maxmin_system, ((cpu_Cas01_t) ((workstation_CLM03_t) workstation_list[i])->cpu)->constraint,
 371                  action->variable, computation_amount[i]);
 372
 373   for (i=0; i<workstation_nb; i++) {
 374     for(j=0; j< workstation_nb; j++) {
 375       network_card_CM02_t card_src = ((workstation_CLM03_t*)workstation_list)[i]->network_card;
 376       network_card_CM02_t card_dst = ((workstation_CLM03_t*)workstation_list)[j]->network_card;
 377       int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
 378       network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
 379
 380       for(k=0; k< route_size; k++) {
 381         if(communication_amount[i*workstation_nb+j]>0) {
 382           lmm_expand_add(maxmin_system, route[k]->constraint,
 383                        action->variable, communication_amount[i*workstation_nb+j]);
 384         }
 385       }
 386     }
 387   }
 388
 389   return (surf_action_t) action;
 390 }
 391
 392 /* returns an array of network_link_CM02_t */
 393 static const void** get_route(void *src, void *dst) {
 394   workstation_CLM03_t workstation_src = (workstation_CLM03_t) src;
 395   workstation_CLM03_t workstation_dst = (workstation_CLM03_t) dst;
 396   return surf_network_resource->extension_public->get_route(workstation_src->network_card, workstation_dst->network_card);
 397 }
 398
 399 static int get_route_size(void *src, void *dst) {
 400   workstation_CLM03_t workstation_src = (workstation_CLM03_t) src;
 401   workstation_CLM03_t workstation_dst = (workstation_CLM03_t) dst;
 402   return surf_network_resource->extension_public->get_route_size(workstation_src->network_card, workstation_dst->network_card);
 403 }
 404
 405 static const char *get_link_name(const void *link) {
 406   return surf_network_resource->extension_public->get_link_name(link);
 407 }
 408
 409 static double get_link_bandwidth(const void *link) {
 410   return surf_network_resource->extension_public->get_link_bandwidth(link);
 411 }
 412
 413 static double get_link_latency(const void *link) {
 414   return surf_network_resource->extension_public->get_link_latency(link);
 415 }
 416
 417 static void finalize(void)
 418 {
 419   xbt_dict_free(&workstation_set);
 420   xbt_swag_free(surf_workstation_resource->common_public->states.ready_action_set);
 421   xbt_swag_free(surf_workstation_resource->common_public->states.
 422                 running_action_set);
 423   xbt_swag_free(surf_workstation_resource->common_public->states.
 424                 failed_action_set);
 425   xbt_swag_free(surf_workstation_resource->common_public->states.done_action_set);
 426
 427   free(surf_workstation_resource->common_public);
 428   free(surf_workstation_resource->common_private);
 429   free(surf_workstation_resource->extension_public);
 430
 431   free(surf_workstation_resource);
 432   surf_workstation_resource = NULL;
 433 }
 434
 435 static void surf_workstation_resource_init_internal(void)
 436 {
 437   s_surf_action_t action;
 438
 439   surf_workstation_resource = xbt_new0(s_surf_workstation_resource_t, 1);
 440
 441   surf_workstation_resource->common_private =
 442       xbt_new0(s_surf_resource_private_t, 1);
 443   surf_workstation_resource->common_public =
 444       xbt_new0(s_surf_resource_public_t, 1);
 445 /*   surf_workstation_resource->extension_private = xbt_new0(s_surf_workstation_resource_extension_private_t,1); */
 446   surf_workstation_resource->extension_public =
 447       xbt_new0(s_surf_workstation_resource_extension_public_t, 1);
 448
 449   surf_workstation_resource->common_public->states.ready_action_set =
 450       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 451   surf_workstation_resource->common_public->states.running_action_set =
 452       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 453   surf_workstation_resource->common_public->states.failed_action_set =
 454       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 455   surf_workstation_resource->common_public->states.done_action_set =
 456       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 457
 458   surf_workstation_resource->common_public->name_service = name_service;
 459   surf_workstation_resource->common_public->get_resource_name =
 460       get_resource_name;
 461   surf_workstation_resource->common_public->action_get_state =
 462       surf_action_get_state;
 463   surf_workstation_resource->common_public->action_get_start_time =
 464       surf_action_get_start_time;
 465   surf_workstation_resource->common_public->action_get_finish_time =
 466       surf_action_get_finish_time;
 467   surf_workstation_resource->common_public->action_free = action_free;
 468   surf_workstation_resource->common_public->action_use = action_use;
 469   surf_workstation_resource->common_public->action_cancel = action_cancel;
 470   surf_workstation_resource->common_public->action_recycle =
 471       action_recycle;
 472   surf_workstation_resource->common_public->action_change_state =
 473       action_change_state;
 474   surf_workstation_resource->common_public->action_set_data = surf_action_set_data;
 475   surf_workstation_resource->common_public->name = "Workstation";
 476
 477   surf_workstation_resource->common_private->resource_used = resource_used;
 478   surf_workstation_resource->common_private->share_resources =
 479       share_resources;
 480   surf_workstation_resource->common_private->update_actions_state =
 481       update_actions_state;
 482   surf_workstation_resource->common_private->update_resource_state =
 483       update_resource_state;
 484   surf_workstation_resource->common_private->finalize = finalize;
 485
 486   surf_workstation_resource->common_public->suspend = action_suspend;
 487   surf_workstation_resource->common_public->resume = action_resume;
 488   surf_workstation_resource->common_public->is_suspended = action_is_suspended;
 489   surf_workstation_resource->common_public->set_max_duration = action_set_max_duration;
 490   surf_workstation_resource->common_public->set_priority = action_set_priority;
 491
 492   surf_workstation_resource->extension_public->execute = execute;
 493   surf_workstation_resource->extension_public->sleep = action_sleep;
 494   surf_workstation_resource->extension_public->get_state = get_state;
 495   surf_workstation_resource->extension_public->get_speed = get_speed;
 496   surf_workstation_resource->extension_public->get_available_speed = get_available_speed;
 497   surf_workstation_resource->extension_public->communicate = communicate;
 498   surf_workstation_resource->extension_public->execute_parallel_task =
 499     execute_parallel_task;
 500   surf_workstation_resource->extension_public->get_route = get_route;
 501   surf_workstation_resource->extension_public->get_route_size = get_route_size;
 502   surf_workstation_resource->extension_public->get_link_name = get_link_name;
 503   surf_workstation_resource->extension_public->get_link_bandwidth = get_link_bandwidth;
 504   surf_workstation_resource->extension_public->get_link_latency = get_link_latency;
 505   workstation_set = xbt_dict_new();
 506
 507   xbt_assert0(maxmin_system, "surf_init has to be called first!");
 508 }
 509
 510 /********************************************************************/
 511 /* The model used in MSG and presented at CCGrid03                  */
 512 /********************************************************************/
 513 /* @InProceedings{Casanova.CLM_03, */
 514 /*   author = {Henri Casanova and Arnaud Legrand and Loris Marchal}, */
 515 /*   title = {Scheduling Distributed Applications: the SimGrid Simulation Framework}, */
 516 /*   booktitle = {Proceedings of the third IEEE International Symposium on Cluster Computing and the Grid (CCGrid'03)}, */
 517 /*   publisher = {"IEEE Computer Society Press"}, */
 518 /*   month = {may}, */
 519 /*   year = {2003} */
 520 /* } */
 521 void surf_workstation_resource_init_CLM03(const char *filename)
 522 {
 523 /*   int i ; */
 524 /*   surf_resource_t resource =  NULL; */
 525
 526   surf_workstation_resource_init_internal();
 527   surf_cpu_resource_init_Cas01(filename);
 528   surf_network_resource_init_CM02(filename);
 529   create_workstations();
 530   xbt_dynar_push(resource_list, &surf_workstation_resource);
 531 /*   xbt_dynar_foreach(resource_list, i, resource) { */
 532 /*     if(resource==surf_cpu_resource) { */
 533 /*       xbt_dynar_remove_at(resource_list, i, NULL); */
 534 /*       i--;  */
 535 /*       continue; */
 536 /*     } */
 537 /*     if(resource==surf_network_resource) { */
 538 /*       xbt_dynar_remove_at(resource_list, i, NULL); */
 539 /*       i--;  */
 540 /*       continue; */
 541 /*     } */
 542 /*   } */
 543 }