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 2
  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 int gl_task_array_id=0;
  25
  26 #define FINALIZE ((void*)221297) /* a magic number to tell people to stop working */
  27
  28 /** master */
  29 int master(int argc, char *argv[])
  30 {
  31   char *slavename = NULL;
  32   double task_comm_size = 0;
  33   m_task_t todo;
  34   m_host_t slave;
  35   //unique id to control statistics
  36   int id=gl_task_array_id++;
  37
  38   /* data size */
  39   xbt_assert1(sscanf(argv[1],"%lg", &task_comm_size),
  40               "Invalid argument %s\n", argv[1]);
  41
  42   /* slave name */
  43   slavename = argv[2];
  44
  45   { /*  Task creation.  */
  46     char sprintf_buffer[64] = "Task_0";
  47     todo = MSG_task_create(sprintf_buffer, 0, task_comm_size, NULL);
  48     //keep track of running tasks
  49     gl_task_array[id] = todo;
  50     gl_data_size[id]=task_comm_size;
  51   }
  52
  53   { /* Process organisation */
  54     slave = MSG_get_host_by_name(slavename);
  55   }
  56
  57   /* time measurement */
  58   start_time = MSG_get_clock();
  59   MSG_task_put(todo, slave, PORT_22);
  60   end_time = MSG_get_clock();
  61   INFO3("Send completed (to %s). Transfer time: %f\t Agregate bandwidth: %f",
  62         slave->name, (end_time - start_time), task_comm_size/(end_time-start_time));
  63   INFO2("Completed peer: %s time: %f", slave->name, (end_time-start_time));
  64
  65   return 0;
  66 } /* end_of_master */
  67
  68 /** Receiver function  */
  69 int slave(int argc, char *argv[])
  70 {
  71   m_task_t task = NULL;
  72   int a;
  73   int id=0;
  74   double remaining=0;
  75   a = MSG_task_get(&(task), PORT_22);
  76   if (a != MSG_OK) {
  77     INFO0("Hey?! What's up?");
  78     xbt_assert0(0,"Unexpected behavior.");
  79   }
  80
  81   elapsed_time = MSG_get_clock() - start_time;
  82
  83   if(!bool_printed){
  84     bool_printed=1;
  85     for(id=0; id<NTASKS; id++){
  86       if(gl_task_array[id] == NULL){
  87         INFO0("===> Task already done, skipping print statistics");
  88       }else if(gl_task_array[id] == task){
  89         INFO1("===> Estimated Bw of FLOWA : %f", gl_data_size[id]/elapsed_time);
  90       }else{
  91         remaining = MSG_task_get_remaining_communication(gl_task_array[id]);
  92         INFO1("===> Estimated Bw of FLOWB : %f", (gl_data_size[id]-remaining)/elapsed_time);
  93       }
  94     }
  95   }
  96
  97   MSG_task_destroy(task);
  98
  99   return 0;
 100 } /* end_of_slave */
 101
 102 /** Test function */
 103 MSG_error_t test_all(const char *platform_file,
 104                      const char *application_file)
 105 {
 106   MSG_error_t res = MSG_OK;
 107
 108   /* MSG_config("workstation_model", "GTNETS"); */
 109   /* MSG_config("workstation_model","KCCFLN05"); */
 110   {                             /*  Simulation setting */
 111     MSG_set_channel_number(MAX_CHANNEL);
 112     MSG_paje_output("msg_test.trace");
 113     MSG_create_environment(platform_file);
 114   }
 115   {                   /*   Application deployment */
 116     MSG_function_register("master", master);
 117     MSG_function_register("slave", slave);
 118     MSG_launch_application(application_file);
 119   }
 120   res = MSG_main();
 121   return res;
 122 } /* end_of_test_all */
 123
 124 /** Main function */
 125 int main(int argc, char *argv[])
 126 {
 127   MSG_error_t res = MSG_OK;
 128   bool_printed = 0;
 129
 130   MSG_global_init(&argc,argv);
 131   if (argc < 3) {
 132      printf ("Usage: %s platform_file deployment_file\n",argv[0]);
 133      exit(1);
 134   }
 135   res = test_all(argv[1],argv[2]);
 136
 137   MSG_clean();
 138
 139   if(res==MSG_OK) return 0;
 140   else return 1;
 141 } /* end_of_main */