examples/msg/gtnets/gtnets_kayo.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   end_time = MSG_get_clock();
  73
  74   INFO5("===> Estimated Bw of FLOW[%d] : %f ;  message from %s to %s  with remaining : %f", id+1,  task_comm_size/(end_time-start_time), masternames[id], slavenames[id],  0.0);
  75
  76
  77   return 0;
  78 } /* end_of_master */
  79
  80 /** Receiver function  */
  81 int slave(int argc, char *argv[])
  82 {
  83
  84   m_task_t task = NULL;
  85   int a;
  86   int id=0;
  87   char id_alias[10];
  88
  89   if(argc != 2){
  90     INFO1("Strange number of arguments expected 1 got %d", argc-1 );
  91   }
  92   id = atoi(argv[1]);
  93   sprintf(id_alias, "%d",id);
  94
  95   a = MSG_task_receive(&(task), id_alias );
  96   return 0;
  97 } /* end_of_slave */
  98
  99 /** Test function */
 100 MSG_error_t test_all(const char *platform_file,
 101                      const char *application_file){
 102   MSG_error_t res = MSG_OK;
 103   MSG_set_channel_number(MAX_CHANNEL);
 104   MSG_create_environment(platform_file);
 105   MSG_function_register("master", master);
 106   MSG_function_register("slave", slave);
 107   MSG_launch_application(application_file);
 108   res = MSG_main();
 109   return res;
 110 } /* end_of_test_all */
 111
 112 /** Main function */
 113 int main(int argc, char *argv[])
 114 {
 115   MSG_error_t res = MSG_OK;
 116   bool_printed = 0;
 117
 118   MSG_global_init(&argc,argv);
 119   if (argc < 3) {
 120      printf ("Usage: %s platform_file deployment_file\n",argv[0]);
 121      exit(1);
 122   }
 123   res = test_all(argv[1],argv[2]);
 124
 125   MSG_clean();
 126
 127   if(res==MSG_OK) return 0;
 128   else return 1;
 129 } /* end_of_main */