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