Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
- implemented MPI_MAXLOC & MPI_MINLOC in operations
[simgrid.git] / src / simix / smx_context_thread.c
1 /* $Id$ */
2
3 /* context_thread - implementation of context switching with native threads */
4
5 /* Copyright (c) 2004-2008 the SimGrid team. All right reserved */
6
7 /* This program is free software; you can redistribute it and/or modify it
8  * under the terms of the license (GNU LGPL) which comes with this package. */
9
10 #include "xbt/function_types.h"
11 #include "private.h"
12
13 #include "portable.h"           /* loads context system definitions */
14 #include "xbt/swag.h"
15 #include "xbt/xbt_os_thread.h"
16 #include "xbt_modinter.h"       /* prototype of os thread module's init/exit in XBT */
17
18 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_context);
19
20 typedef struct s_smx_ctx_thread {
21   s_smx_ctx_base_t super;       /* Fields of super implementation */
22   xbt_os_thread_t thread;       /* a plain dumb thread (portable to posix or windows) */
23   xbt_os_sem_t begin;           /* this semaphore is used to schedule/yield the process  */
24   xbt_os_sem_t end;             /* this semaphore is used to schedule/unschedule the process   */
25 } s_smx_ctx_thread_t, *smx_ctx_thread_t;
26
27 static smx_context_t
28 smx_ctx_thread_factory_create_context(xbt_main_func_t code, int argc, char** argv, 
29                                       void_f_pvoid_t cleanup_func, void* cleanup_arg);
30
31 static void smx_ctx_thread_free(smx_context_t context);
32 static void smx_ctx_thread_stop(smx_context_t context);
33 static void smx_ctx_thread_suspend(smx_context_t context);
34 static void smx_ctx_thread_resume(smx_context_t new_context);
35
36 static void *smx_ctx_thread_wrapper(void *param);
37
38 void SIMIX_ctx_thread_factory_init(smx_context_factory_t * factory) {
39
40   smx_ctx_base_factory_init(factory);
41
42   (*factory)->create_context = smx_ctx_thread_factory_create_context;
43   /* Do not overload that method (*factory)->finalize */
44   (*factory)->free = smx_ctx_thread_free;
45   (*factory)->stop = smx_ctx_thread_stop;
46   (*factory)->suspend = smx_ctx_thread_suspend;
47   (*factory)->resume = smx_ctx_thread_resume;
48   (*factory)->name = "ctx_thread_factory";
49 }
50
51 static smx_context_t 
52 smx_ctx_thread_factory_create_context(xbt_main_func_t code, int argc, char** argv, 
53                                       void_f_pvoid_t cleanup_func, void* cleanup_arg)
54 {
55   smx_ctx_thread_t context = (smx_ctx_thread_t)smx_ctx_base_factory_create_context_sized
56       (sizeof(s_smx_ctx_thread_t), code,argc,argv,cleanup_func,cleanup_arg);
57
58   /* If the user provided a function for the process then use it
59      otherwise is the context for maestro */
60   if(code){
61     context->begin = xbt_os_sem_init(0);
62     context->end = xbt_os_sem_init(0);
63
64
65     /* create and start the process */
66     /* NOTE: The first argument to xbt_os_thread_create used to be the process *
67      * name, but now the name is stored at SIMIX level, so we pass a null      */
68     context->thread =
69       xbt_os_thread_create(NULL, smx_ctx_thread_wrapper, context);
70
71     /* wait the starting of the newly created process */
72     xbt_os_sem_acquire(context->end);
73   }
74     
75   return (smx_context_t)context;
76 }
77
78 static void smx_ctx_thread_free(smx_context_t pcontext)
79 {
80   int i;
81   smx_ctx_thread_t context = (smx_ctx_thread_t)pcontext;
82
83   /* check if this is the context of maestro (it doesn't has a real thread) */  
84   if (context->thread) {
85     /* wait about the thread terminason */
86     xbt_os_thread_join(context->thread, NULL);
87     
88     /* destroy the synchronisation objects */
89     xbt_os_sem_destroy(context->begin);
90     xbt_os_sem_destroy(context->end);
91   }
92   
93   smx_ctx_base_free(pcontext);
94 }
95
96 static void smx_ctx_thread_stop(smx_context_t pcontext)
97 {
98
99   smx_ctx_thread_t context = (smx_ctx_thread_t)pcontext;
100   
101   /* please no debug here: our procdata was already free'd */
102   smx_ctx_base_stop(pcontext);
103
104   /* signal to the maestro that it has finished */
105   xbt_os_sem_release(((smx_ctx_thread_t) context)->end);
106
107   /* exit */
108   /* We should provide return value in case other wants it */
109   xbt_os_thread_exit(NULL);     
110 }
111
112 static void *smx_ctx_thread_wrapper(void *param)
113 {
114   smx_ctx_thread_t context = (smx_ctx_thread_t) param;
115
116   /* Tell the maestro we are starting, and wait for its green light */
117   xbt_os_sem_release(context->end);
118   xbt_os_sem_acquire(context->begin);
119
120   (context->super.code) (context->super.argc, context->super.argv);
121
122   smx_ctx_thread_stop((smx_context_t)context);
123   return NULL;
124 }
125
126 static void smx_ctx_thread_suspend(smx_context_t context) {
127   xbt_os_sem_release(((smx_ctx_thread_t) context)->end);
128   xbt_os_sem_acquire(((smx_ctx_thread_t) context)->begin);
129 }
130
131 static void smx_ctx_thread_resume(smx_context_t new_context) {
132   xbt_os_sem_release(((smx_ctx_thread_t) new_context)->begin);
133   xbt_os_sem_acquire(((smx_ctx_thread_t) new_context)->end);
134 }