examples/msg/gtnets/gtnets.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include "msg/msg.h"
   4 #include "xbt/log.h"
   5 #include "xbt/asserts.h"
   6
   7 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,"Messages specific for this msg example");
   8
   9 int master(int argc, char *argv[]);
  10 int slave(int argc, char *argv[]);
  11 MSG_error_t test_all(const char *platform_file, const char *application_file);
  12
  13 typedef enum {
  14   PORT_22 = 0,
  15   MAX_CHANNEL
  16 } channel_t;
  17
  18 //keep a pointer to all surf running tasks.
  19 #define NTASKS 1500
  20 int bool_printed=0;
  21 double start_time, end_time, elapsed_time;
  22 double   gl_data_size[NTASKS];
  23 m_task_t gl_task_array[NTASKS];
  24 const char * slavenames[NTASKS];
  25 const char * masternames[NTASKS];
  26 int gl_task_array_id=0;
  27
  28 #define FINALIZE ((void*)221297) /* a magic number to tell people to stop working */
  29
  30 /** master */
  31 int master(int argc, char *argv[])
  32 {
  33   char *slavename = NULL;
  34   double task_comm_size = 0;
  35   m_task_t todo;
  36   m_host_t slave;
  37   char id_alias[10];
  38   //unique id to control statistics
  39   int id = -1;
  40
  41   if(argc != 4){
  42     INFO1("Strange number of arguments expected 3 got %d", argc-1 );
  43   }
  44
  45   /* data size */
  46   xbt_assert1(sscanf(argv[1],"%lg", &task_comm_size),
  47               "Invalid argument %s\n", argv[1]);
  48
  49   /* slave name */
  50   slavename = argv[2];
  51   id = atoi(argv[3]);
  52   sprintf(id_alias, "%d",id);
  53   slavenames[id] = slavename;
  54
  55   masternames[id] = MSG_host_get_name(MSG_host_self());
  56
  57   { /*  Task creation.  */
  58     char sprintf_buffer[64] = "Task_0";
  59     todo = MSG_task_create(sprintf_buffer, 0, task_comm_size, NULL);
  60     //keep track of running tasks
  61     gl_task_array[id] = todo;
  62     gl_data_size[id]=task_comm_size;
  63   }
  64
  65   { /* Process organisation */
  66     slave = MSG_get_host_by_name(slavename);
  67   }
  68
  69   /* time measurement */
  70   start_time = MSG_get_clock();
  71   MSG_task_send(todo, id_alias);
  72
  73   end_time = MSG_get_clock();
  74   INFO3("Send completed (to %s). Transfer time: %f\t Agregate bandwidth: %f",
  75         slave->name, (end_time - start_time), task_comm_size/(end_time-start_time));
  76   INFO2("Completed peer: %s time: %f", slave->name, (end_time-start_time));
  77
  78   return 0;
  79 } /* end_of_master */
  80
  81 /** Receiver function  */
  82 int slave(int argc, char *argv[])
  83 {
  84
  85   m_task_t task = NULL;
  86   int a;
  87   int id=0;
  88   double remaining=0;
  89   char id_alias[10];
  90
  91   if(argc != 2){
  92     INFO1("Strange number of arguments expected 1 got %d", argc-1 );
  93   }
  94
  95   id = atoi(argv[1]);
  96   sprintf(id_alias, "%d",id);
  97
  98   a = MSG_task_receive(&(task), id_alias );
  99
 100   if (a != MSG_OK) {
 101     INFO0("Hey?! What's up?");
 102     xbt_assert0(0,"Unexpected behavior.");
 103   }
 104
 105   elapsed_time = MSG_get_clock() - start_time;
 106
 107   if(!bool_printed){
 108     bool_printed=1;
 109     for(id=0; id<NTASKS; id++){
 110       if(gl_task_array[id] == NULL){
 111       }else if(gl_task_array[id] == task){
 112         INFO5("===> Estimated Bw of FLOW[%d] : %f ;  message from %s to %s  with remaining : %f", id,  gl_data_size[id]/elapsed_time, masternames[id], slavenames[id],  0.0);
 113       }else{
 114         remaining = MSG_task_get_remaining_communication(gl_task_array[id]);
 115         INFO5("===> Estimated Bw of FLOW[%d] : %f ;  message from %s to %s  with remaining : %f", id,  (gl_data_size[id]-remaining)/elapsed_time, masternames[id], slavenames[id],  remaining);
 116       }
 117     }
 118     exit(0);
 119   }
 120
 121   for(id=0; id<NTASKS; id++){
 122     if(gl_task_array[id] == task){
 123       MSG_task_destroy(task);
 124       gl_task_array[id] = NULL;
 125       return 0;
 126     }
 127   }
 128
 129   return 0;
 130 } /* end_of_slave */
 131
 132 /** Test function */
 133 MSG_error_t test_all(const char *platform_file,
 134                      const char *application_file)
 135 {
 136   MSG_error_t res = MSG_OK;
 137
 138   /* MSG_config("workstation_model", "GTNETS"); */
 139   /* MSG_config("workstation_model","KCCFLN05"); */
 140   {                             /*  Simulation setting */
 141     MSG_set_channel_number(MAX_CHANNEL);
 142     //MSG_paje_output("msg_test.trace");
 143     MSG_create_environment(platform_file);
 144   }
 145   {                   /*   Application deployment */
 146     MSG_function_register("master", master);
 147     MSG_function_register("slave", slave);
 148     MSG_launch_application(application_file);
 149   }
 150   res = MSG_main();
 151   return res;
 152 } /* end_of_test_all */
 153
 154 /** Main function */
 155 int main(int argc, char *argv[])
 156 {
 157   MSG_error_t res = MSG_OK;
 158   bool_printed = 0;
 159
 160   MSG_global_init(&argc,argv);
 161   if (argc < 3) {
 162      printf ("Usage: %s platform_file deployment_file\n",argv[0]);
 163      exit(1);
 164   }
 165   res = test_all(argv[1],argv[2]);
 166
 167   MSG_clean();
 168
 169   if(res==MSG_OK) return 0;
 170   else return 1;
 171 } /* end_of_main */