src/simix/smx_context_ruby.c

   1 /* context_Ruby - implementation of context switching with/for ruby         */
   2
   3 /* Copyright (c) 2010. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   6 /* This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package. */
   8
   9 #include "xbt/function_types.h"
  10 #include "xbt/sysdep.h"
  11 #include "xbt/log.h"
  12 #include "xbt/asserts.h"
  13
  14 #include "smx_context_private.h"
  15 #include "bindings/ruby_bindings.h"
  16
  17 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(ruby);
  18
  19 static smx_context_t
  20 smx_ctx_ruby_create_context(xbt_main_func_t code, int argc, char **argv,
  21                             void_f_pvoid_t cleanup_func,
  22                             void *cleanup_arg);
  23
  24 static void smx_ctx_ruby_stop(smx_context_t context);
  25 static void smx_ctx_ruby_suspend(smx_context_t context);
  26 static void smx_ctx_ruby_resume(smx_context_t new_context);
  27 static void smx_ctx_ruby_wrapper(void);
  28
  29
  30 void SIMIX_ctx_ruby_factory_init(smx_context_factory_t * factory)
  31 {
  32   smx_ctx_base_factory_init(factory);
  33
  34   (*factory)->create_context = smx_ctx_ruby_create_context;
  35   /* Do not overload that method (*factory)->finalize */
  36   /* Do not overload that method (*factory)->free */
  37   (*factory)->stop = smx_ctx_ruby_stop;
  38   (*factory)->suspend = smx_ctx_ruby_suspend;
  39   (*factory)->resume = smx_ctx_ruby_resume;
  40   (*factory)->name = "smx_ruby_context_factory";
  41   ruby_init();
  42   ruby_init_loadpath();
  43 }
  44
  45 static smx_context_t
  46 smx_ctx_ruby_create_context(xbt_main_func_t code, int argc, char **argv,
  47                             void_f_pvoid_t cleanup_func, void *cleanup_arg)
  48 {
  49
  50   smx_ctx_ruby_t context = (smx_ctx_ruby_t)
  51       smx_ctx_base_factory_create_context_sized(sizeof(s_smx_ctx_ruby_t),
  52                                                 code, argc, argv,
  53                                                 cleanup_func, cleanup_arg);
  54
  55   /* if the user provided a function for the process , then use it
  56      Otherwise it's the context for maestro */
  57   if (code) {
  58     context->process = (VALUE) code;
  59
  60     DEBUG1("smx_ctx_ruby_create_context(%s)...Done", argv[0]);
  61   }
  62   return (smx_context_t) context;
  63 }
  64
  65 static void smx_ctx_ruby_stop(smx_context_t context)
  66 {
  67   DEBUG0("smx_ctx_ruby_stop()");
  68   VALUE process = Qnil;
  69   smx_ctx_ruby_t ctx_ruby, current;
  70
  71   smx_ctx_base_stop(context);
  72
  73   ctx_ruby = (smx_ctx_ruby_t) context;
  74
  75   if (simix_global->current_process->iwannadie) {
  76     if (ctx_ruby->process) {
  77
  78       //if the Ruby Process still Alive ,let's Schedule it
  79       if (rb_process_isAlive(ctx_ruby->process)) {
  80
  81         current = (smx_ctx_ruby_t) simix_global->current_process->context;
  82         rb_process_schedule(current->process);
  83         process = ctx_ruby->process;
  84         // interupt/kill The Ruby Process
  85         rb_process_kill_up(process);
  86       }
  87     }
  88   } else {
  89
  90     if (ctx_ruby->process)
  91       ctx_ruby->process = Qnil;
  92
  93   }
  94 }
  95
  96 static void smx_ctx_ruby_suspend(smx_context_t context)
  97 {
  98
  99   DEBUG1("smx_ctx_ruby_suspend(%s)", context->argv[0]);
 100   smx_ctx_ruby_t ctx_ruby = (smx_ctx_ruby_t) context;
 101   if (ctx_ruby->process)
 102     rb_process_unschedule(ctx_ruby->process);
 103 }
 104
 105 static void smx_ctx_ruby_resume(smx_context_t new_context)
 106 {
 107   DEBUG1("smx_ctx_ruby_resume(%s)",
 108          (new_context->argc ? new_context->argv[0] : "maestro"));
 109
 110   smx_ctx_ruby_t ctx_ruby = (smx_ctx_ruby_t) new_context;
 111   rb_process_schedule(ctx_ruby->process);
 112
 113 }