Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
647b34eccf2e5769cc725e63600aecb190a5263f
[simgrid.git] / src / smx_context_java.c
1 /* context_java - implementation of context switching for java threads */
2
3 /* Copyright (c) 2009, 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
10 #include <xbt/function_types.h>
11 #include <simgrid/simix.h>
12 #include "smx_context_java.h"
13 #include "xbt/dynar.h"
14
15 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(jmsg, bindings, "MSG for Java(TM)");
16
17 static smx_context_t my_current_context = NULL;
18
19 static smx_context_t smx_ctx_java_self(void);
20 static smx_context_t
21 smx_ctx_java_factory_create_context(xbt_main_func_t code, int argc,
22                                     char **argv,
23                                     void_pfn_smxprocess_t cleanup_func,
24                                     void *data);
25
26 static void smx_ctx_java_free(smx_context_t context);
27 static void smx_ctx_java_start(smx_context_t context);
28 static void smx_ctx_java_suspend(smx_context_t context);
29 static void smx_ctx_java_resume(smx_context_t new_context);
30 static void smx_ctx_java_runall(void);
31
32 void SIMIX_ctx_java_factory_init(smx_context_factory_t * factory)
33 {
34   /* instantiate the context factory */
35   smx_ctx_base_factory_init(factory);
36
37   (*factory)->create_context = smx_ctx_java_factory_create_context;
38   /* Leave default behavior of (*factory)->finalize */
39   (*factory)->free = smx_ctx_java_free;
40   (*factory)->stop = smx_ctx_java_stop;
41   (*factory)->suspend = smx_ctx_java_suspend;
42   (*factory)->runall = smx_ctx_java_runall;
43   (*factory)->name = "ctx_java_factory";
44   //(*factory)->finalize = smx_ctx_base_factory_finalize;
45   (*factory)->self = smx_ctx_java_self;
46   (*factory)->get_data = smx_ctx_base_get_data;
47 }
48
49 static smx_context_t smx_ctx_java_self(void)
50 {
51         return my_current_context;
52 }
53
54 static smx_context_t
55 smx_ctx_java_factory_create_context(xbt_main_func_t code, int argc,
56                                     char **argv,
57                                     void_pfn_smxprocess_t cleanup_func,
58                                     void* data)
59 {
60   XBT_DEBUG("XXXX Create Context\n");
61   smx_ctx_java_t context = xbt_new0(s_smx_ctx_java_t, 1);
62
63   /* If the user provided a function for the process then use it
64      otherwise is the context for maestro */
65   if (code) {
66     context->super.cleanup_func = cleanup_func;
67     context->jprocess = (jobject) code;
68     context->jenv = get_current_thread_env();
69     jprocess_start(((smx_ctx_java_t) context)->jprocess,
70                    get_current_thread_env());
71   }else{
72     my_current_context = (smx_context_t)context;
73   }
74   context->super.data = data;
75   
76   return (smx_context_t) context;
77 }
78
79 static void smx_ctx_java_free(smx_context_t context)
80 {
81   if (context) {
82     smx_ctx_java_t ctx_java = (smx_ctx_java_t) context;
83
84     if (ctx_java->jprocess) { /* the java process still exists */
85       jobject jprocess = ctx_java->jprocess;
86       ctx_java->jprocess = NULL;
87
88       /* stop it */
89       XBT_DEBUG("The process still exists, making it exit now");
90       jprocess_exit(jprocess, get_current_thread_env());
91
92       /* it's dead now, remove it from the JVM */
93       jprocess_delete_global_ref(jprocess, get_current_thread_env());
94     }
95   }
96
97   smx_ctx_base_free(context);
98 }
99
100
101 void smx_ctx_java_stop(smx_context_t context)
102 {
103  xbt_assert(context == my_current_context,
104      "The context to stop must be the current one");
105   /* I am the current process and I am dying */
106   
107   smx_ctx_base_stop(context);
108
109   XBT_DEBUG("I am dying");
110
111   /* suspend myself again, smx_ctx_java_free() will destroy me later
112    * from maeastro */
113   jprocess_unschedule(context);
114   XBT_INFO("Java stop finished");
115 }
116
117 static void smx_ctx_java_suspend(smx_context_t context)
118 {
119   jprocess_unschedule(context);
120 }
121
122 // FIXME: inline those functions
123 static void smx_ctx_java_resume(smx_context_t new_context)
124 {
125   XBT_DEBUG("XXXX Context Resume\n");
126   jprocess_schedule(new_context);
127 }
128
129 static void smx_ctx_java_runall(void)
130 {
131   xbt_dynar_t processes = SIMIX_process_get_runnable();
132   XBT_DEBUG("XXXX Run all\n");
133   smx_process_t process;
134   smx_context_t old_context;
135   unsigned int cursor;
136   xbt_dynar_foreach(processes, cursor, process) {
137     old_context = my_current_context;
138     my_current_context = SIMIX_process_get_context(process);
139     smx_ctx_java_resume(my_current_context);
140     my_current_context = old_context;
141   }
142
143   XBT_DEBUG("XXXX End of run all\n");
144 }