Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Use new style logging macros.
[simgrid.git] / src / simix / smx_deployment.c
1 /* Copyright (c) 2007, 2009, 2010. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "private.h"
8 #include "xbt/sysdep.h"
9 #include "xbt/log.h"
10 #include "xbt/dict.h"
11 #include "surf/surfxml_parse_private.h"
12
13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_deployment, simix,
14                                 "Logging specific to SIMIX (deployment)");
15 static int parse_argc = -1;
16 static char **parse_argv = NULL;
17 static xbt_main_func_t parse_code = NULL;
18 static char *parse_host = NULL;
19 static double start_time = 0.0;
20 static double kill_time = -1.0;
21
22 static void parse_process_init(void)
23 {
24   parse_host = xbt_strdup(A_surfxml_process_host);
25   xbt_assert1(SIMIX_host_get_by_name(parse_host),
26               "Host '%s' unknown", parse_host);
27   parse_code = SIMIX_get_registered_function(A_surfxml_process_function);
28   xbt_assert1(parse_code, "Function '%s' unknown",
29               A_surfxml_process_function);
30   parse_argc = 0;
31   parse_argv = NULL;
32   parse_argc++;
33   parse_argv = xbt_realloc(parse_argv, (parse_argc) * sizeof(char *));
34   parse_argv[(parse_argc) - 1] = xbt_strdup(A_surfxml_process_function);
35   surf_parse_get_double(&start_time, A_surfxml_process_start_time);
36   surf_parse_get_double(&kill_time, A_surfxml_process_kill_time);
37
38   current_property_set = xbt_dict_new();
39 }
40
41 static void parse_argument(void)
42 {
43   parse_argc++;
44   parse_argv = xbt_realloc(parse_argv, (parse_argc) * sizeof(char *));
45   parse_argv[(parse_argc) - 1] = xbt_strdup(A_surfxml_argument_value);
46 }
47
48 static void parse_process_finalize(void)
49 {
50   smx_process_arg_t arg = NULL;
51   smx_process_t process = NULL;
52   if (start_time > SIMIX_get_clock()) {
53     arg = xbt_new0(s_smx_process_arg_t, 1);
54     arg->name = parse_argv[0];
55     arg->code = parse_code;
56     arg->data = NULL;
57     arg->hostname = parse_host;
58     arg->argc = parse_argc;
59     arg->argv = parse_argv;
60     arg->kill_time = kill_time;
61     arg->properties = current_property_set;
62
63     XBT_DEBUG("Process %s(%s) will be started at time %f", arg->name,
64            arg->hostname, start_time);
65     SIMIX_timer_set(start_time, &SIMIX_process_create_from_wrapper, arg);
66   } else {                      // start_time <= SIMIX_get_clock()
67     XBT_DEBUG("Starting Process %s(%s) right now", parse_argv[0], parse_host);
68
69     if (simix_global->create_process_function)
70       (*simix_global->create_process_function) (&process,
71                                                 parse_argv[0],
72                                                 parse_code, NULL,
73                                                 parse_host, parse_argc,
74                                                 parse_argv,
75                                                 current_property_set);
76     else
77       SIMIX_req_process_create(&process, parse_argv[0], parse_code, NULL, parse_host, parse_argc, parse_argv,
78                                current_property_set);
79     /* verify if process has been created (won't be the case if the host is currently dead, but that's fine) */
80     if (!process) {
81       xbt_free(parse_host);
82       return;
83     }
84     if (kill_time > SIMIX_get_clock()) {
85       if (simix_global->kill_process_function)
86         SIMIX_timer_set(start_time, simix_global->kill_process_function, process);
87       else
88         SIMIX_timer_set(start_time, &SIMIX_process_kill, process);
89     }
90     xbt_free(parse_host);
91   }
92 }
93
94 /**
95  * \brief An application deployer.
96  *
97  * Creates the process described in \a file.
98  * \param file a filename of a xml description of the application. This file
99  * follows this DTD :
100  *
101  *     \include surfxml.dtd
102  *
103  * Here is a small example of such a platform
104  *
105  *     \include small_deployment.xml
106  *
107  */
108 void SIMIX_launch_application(const char *file)
109 {
110   int parse_status;
111   xbt_assert0(simix_global,
112               "SIMIX_global_init has to be called before SIMIX_launch_application.");
113
114   // Reset callbacks
115   surf_parse_reset_callbacks();
116
117   surfxml_add_callback(STag_surfxml_process_cb_list, parse_process_init);
118   surfxml_add_callback(ETag_surfxml_argument_cb_list, parse_argument);
119   surfxml_add_callback(STag_surfxml_prop_cb_list, parse_properties);
120   surfxml_add_callback(ETag_surfxml_process_cb_list,
121                        parse_process_finalize);
122
123   surf_parse_open(file);
124   parse_status = surf_parse();
125   surf_parse_close();
126   xbt_assert1(!parse_status, "Parse error in %s", file);
127 }
128
129 /**
130  * \brief Registers a #smx_process_code_t code in a global table.
131  *
132  * Registers a code function in a global table.
133  * This table is then used by #SIMIX_launch_application.
134  * \param name the reference name of the function.
135  * \param code the function
136  */
137 XBT_INLINE void SIMIX_function_register(const char *name,
138                                         xbt_main_func_t code)
139 {
140   xbt_assert0(simix_global,
141               "SIMIX_global_init has to be called before SIMIX_function_register.");
142
143   xbt_dict_set(simix_global->registered_functions, name, code, NULL);
144 }
145
146 static xbt_main_func_t default_function = NULL;
147 /**
148  * \brief Registers a #smx_process_code_t code as default value.
149  *
150  * Registers a code function as being the default value. This function will get used by SIMIX_launch_application() when there is no registered function of the requested name in.
151  * \param code the function
152  */
153 void SIMIX_function_register_default(xbt_main_func_t code)
154 {
155   xbt_assert0(simix_global,
156               "SIMIX_global_init has to be called before SIMIX_function_register.");
157
158   default_function = code;
159 }
160
161 /**
162  * \brief Gets a #smx_process_t code from the global table.
163  *
164  * Gets a code function from the global table. Returns NULL if there are no function registered with the name.
165  * This table is then used by #SIMIX_launch_application.
166  * \param name the reference name of the function.
167  * \return The #smx_process_t or NULL.
168  */
169 xbt_main_func_t SIMIX_get_registered_function(const char *name)
170 {
171   xbt_main_func_t res = NULL;
172   xbt_assert0(simix_global,
173               "SIMIX_global_init has to be called before SIMIX_get_registered_function.");
174
175   res = xbt_dict_get_or_null(simix_global->registered_functions, name);
176   return res ? res : default_function;
177 }
178
179
180 /**
181  * \brief Bypass the parser, get arguments, and set function to each process
182  */
183
184 void SIMIX_process_set_function(const char *process_host,
185                                 const char *process_function,
186                                 xbt_dynar_t arguments,
187                                 double process_start_time,
188                                 double process_kill_time)
189 {
190   unsigned int i;
191   char *arg;
192
193   /* init process */
194   parse_host = xbt_strdup(process_host);
195   xbt_assert1(SIMIX_host_get_by_name(parse_host),
196               "Host '%s' unknown", parse_host);
197   parse_code = SIMIX_get_registered_function(process_function);
198   xbt_assert1(parse_code, "Function '%s' unknown", process_function);
199
200   parse_argc = 0;
201   parse_argv = NULL;
202   parse_argc++;
203   parse_argv = xbt_realloc(parse_argv, (parse_argc) * sizeof(char *));
204   parse_argv[(parse_argc) - 1] = xbt_strdup(process_function);
205   start_time = process_start_time;
206   kill_time = process_kill_time;
207   current_property_set = xbt_dict_new();
208
209   /* add arguments */
210   xbt_dynar_foreach(arguments, i, arg) {
211     parse_argc++;
212     parse_argv = xbt_realloc(parse_argv, (parse_argc) * sizeof(char *));
213     parse_argv[(parse_argc) - 1] = xbt_strdup(arg);
214   }
215
216   /* finalize */
217   parse_process_finalize();
218 }