src/bindings/ruby/simgrid_ruby.c

   1 /* SimGrid Ruby bindings                                                    */
   2
   3 /* Copyright (c) 2010, the SimGrid team. All right 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.h"
   9 #include "bindings/ruby_bindings.h"
  10
  11 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(ruby);
  12
  13 // MSG Module
  14 VALUE rb_msg;
  15 // MSG Classes
  16 VALUE rb_task;
  17 VALUE rb_host;
  18
  19 //Init Msg  From Ruby
  20 static void msg_init(VALUE Class,VALUE args)
  21 {
  22   char **argv=NULL;
  23   const char *tmp;
  24   int argc,type,i;
  25   VALUE *ptr ;
  26   // Testing The Args Type
  27   type =  TYPE(args);
  28   if (type != T_ARRAY ) {
  29     rb_raise(rb_eRuntimeError,"Bad arguments to msg_init (expecting an array)");
  30     return;
  31   }
  32   ptr= RARRAY(args)->ptr;
  33   argc= RARRAY(args)->len;
  34   //  Create C Array to Hold Data_Get_Struct
  35   argc++;
  36   argv = xbt_new0(char *, argc);
  37   argv[0] = strdup("ruby");
  38   for (i=0;i<argc-1;i++) {
  39     VALUE value = ptr[i];
  40     type = TYPE(value);
  41     //  if (type == T_STRING)
  42     tmp = RSTRING(value)->ptr;
  43     argv[i+1] = strdup(tmp);
  44   }
  45   // Calling C Msg_Init Method
  46   MSG_global_init(&argc,argv);
  47
  48   // Cleanups
  49   for (i=0;i<argc;i++)
  50     free(argv[i]) ;
  51   free (argv);
  52 }
  53 //Init Msg_Run From Ruby
  54 static void msg_run(VALUE class) {
  55   DEBUG0("Start Running...");
  56   m_host_t *hosts;
  57   int cpt,host_count;
  58   VALUE rbHost;
  59   // Let's Run
  60   //printf("msg_run3\n");
  61   if (MSG_OK != MSG_main()){
  62     rb_raise(rb_eRuntimeError,"MSG_main() failed");
  63   }
  64
  65   DEBUG0
  66   ("MSG_main finished. Bail out before cleanup since there is a bug in this part.");
  67   /* Cleanup Ruby hosts */
  68   DEBUG0("Clean Ruby World  ");
  69   hosts = MSG_get_host_table();
  70   host_count = MSG_get_host_number();
  71   for (cpt=0;cpt<host_count;cpt++) {
  72     rbHost = (VALUE)((hosts[cpt])->data);
  73   }
  74   return;
  75 }
  76
  77 static void msg_clean(VALUE class)
  78 {
  79    if (MSG_OK != MSG_clean())
  80     rb_raise(rb_eRuntimeError,"MSG_clean() failed");
  81
  82 }
  83 static void msg_createEnvironment(VALUE class,VALUE plateformFile) {
  84
  85   int type = TYPE(plateformFile);
  86   if ( type != T_STRING )
  87     rb_raise(rb_eRuntimeError,"Bad Argument's Type");
  88   const char * platform =  RSTRING(plateformFile)->ptr;
  89   MSG_create_environment(platform);
  90   DEBUG1("Create Environment (%s)...Done",platform);
  91 }
  92
  93 //deploy Application
  94 static void msg_deployApplication(VALUE class,VALUE deploymentFile ) {
  95
  96   int type = TYPE(deploymentFile);
  97   if ( type != T_STRING )
  98     rb_raise(rb_eRuntimeError,"Bad Argument's Type for deployApplication ");
  99   const char *dep_file = RSTRING(deploymentFile)->ptr;
 100   surf_parse_reset_parser();
 101   surfxml_add_callback(STag_surfxml_process_cb_list,
 102       rb_application_handler_on_begin_process);
 103   surfxml_add_callback(ETag_surfxml_argument_cb_list,
 104       rb_application_handler_on_process_arg);
 105
 106   surfxml_add_callback(STag_surfxml_prop_cb_list,
 107       rb_application_handler_on_property);
 108
 109   surfxml_add_callback(ETag_surfxml_process_cb_list,
 110       rb_application_handler_on_end_process);
 111
 112   surf_parse_open(dep_file);
 113   rb_application_handler_on_start_document();
 114   if(surf_parse())
 115     rb_raise(rb_eRuntimeError,"surf_parse() failed");
 116   surf_parse_close();
 117
 118   rb_application_handler_on_end_document();
 119
 120   DEBUG1("Deploy Application(%s)...Done",dep_file);
 121 }
 122
 123 // INFO
 124 static void msg_info(VALUE class,VALUE msg) {
 125   const char *s = RSTRING(msg)->ptr;
 126   INFO1("%s",s);
 127 }
 128 static void msg_debug(VALUE class,VALUE msg) {
 129   const char *s = RSTRING(msg)->ptr;
 130   DEBUG1("%s",s);
 131 }
 132
 133 // get Clock
 134 static VALUE msg_get_clock(VALUE class) {
 135   return rb_float_new(MSG_get_clock());
 136
 137 }
 138
 139 extern const char*xbt_ctx_factory_to_use; /*Hack: let msg load directly the right factory */
 140
 141 typedef VALUE(*rb_meth)(ANYARGS);
 142 void Init_simgrid_ruby() {
 143   xbt_ctx_factory_to_use = "ruby";
 144
 145   // Modules
 146   rb_msg = rb_define_module("MSG");
 147   //Associated Environment Methods
 148   rb_define_module_function(rb_msg,"init",(rb_meth)msg_init,1);
 149   rb_define_module_function(rb_msg,"run",(rb_meth)msg_run,0);
 150   rb_define_module_function(rb_msg,"createEnvironment",(rb_meth)msg_createEnvironment,1);
 151   rb_define_module_function(rb_msg,"deployApplication",(rb_meth)msg_deployApplication,1);
 152   rb_define_module_function(rb_msg,"info",(rb_meth)msg_info,1);
 153   rb_define_module_function(rb_msg,"debug",(rb_meth)msg_debug,1);
 154   rb_define_module_function(rb_msg,"getClock",(rb_meth)msg_get_clock,0);
 155   rb_define_module_function(rb_msg,"exit",(rb_meth)msg_clean,0);
 156
 157   //Associated Process Methods
 158   rb_define_method(rb_msg,"processSuspend",(rb_meth)rb_process_suspend,1);
 159   rb_define_method(rb_msg,"processResume",(rb_meth)rb_process_resume,1);
 160   rb_define_method(rb_msg,"processIsSuspend",(rb_meth)rb_process_isSuspended,1);
 161   rb_define_method(rb_msg,"processKill",(rb_meth)rb_process_kill_up,1);
 162   rb_define_method(rb_msg,"processKillDown",(rb_meth)rb_process_kill_down,1);
 163   rb_define_method(rb_msg,"processGetHost",(rb_meth)rb_process_getHost,1);
 164   rb_define_method(rb_msg,"processExit",(rb_meth)rb_process_exit,1);
 165
 166   //Classes
 167   rb_task = rb_define_class_under(rb_msg,"RbTask",rb_cObject);
 168   rb_host = rb_define_class_under(rb_msg,"RbHost",rb_cObject);
 169
 170   //Task Methods
 171   rb_define_module_function(rb_task,"new",(rb_meth)rb_task_new,3);
 172   rb_define_module_function(rb_task,"setData",(rb_meth)rb_task_set_data,2);
 173   rb_define_module_function(rb_task,"data",(rb_meth)rb_task_get_data,1);
 174   rb_define_module_function(rb_task,"compSize",(rb_meth)rb_task_comp,1);
 175   rb_define_module_function(rb_task,"name",(rb_meth)rb_task_name,1);
 176   rb_define_module_function(rb_task,"execute",(rb_meth)rb_task_execute,1);
 177   rb_define_module_function(rb_task,"send",(rb_meth)rb_task_send,2);
 178   rb_define_module_function(rb_task,"receive",(rb_meth)rb_task_receive,1);
 179   rb_define_module_function(rb_task,"sender",(rb_meth)rb_task_sender,1);
 180   rb_define_module_function(rb_task,"source",(rb_meth)rb_task_source,1);
 181   rb_define_module_function(rb_task,"listen",(rb_meth)rb_task_listen,2);
 182   rb_define_module_function(rb_task,"listenFromHost",(rb_meth)rb_task_listen_host,3);
 183   rb_define_module_function(rb_task,"setPriority",(rb_meth)rb_task_set_priority,2);
 184   rb_define_module_function(rb_task,"cancel",(rb_meth)rb_task_cancel,1);
 185
 186   //Host Methods
 187   rb_define_module_function(rb_host,"getByName",(rb_meth)rb_host_get_by_name,1);
 188   rb_define_module_function(rb_host,"name",(rb_meth)rb_host_name,1);
 189   rb_define_module_function(rb_host,"speed",(rb_meth)rb_host_speed,1);
 190   rb_define_module_function(rb_host,"number",(rb_meth)rb_host_number,0);
 191   rb_define_module_function(rb_host,"isAvail",(rb_meth)rb_host_is_avail,1);
 192   rb_define_module_function(rb_host,"getHostProcess",(rb_meth)rb_host_process,1);
 193 }