1 /* Copyright (c) 2010. The SimGrid Team.
2 * All rights reserved. */
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. */
9 #include "xbt/sysdep.h" /* calloc, printf */
11 /* Create a log channel to have nice outputs. */
13 #include "xbt/asserts.h"
14 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
15 "Messages specific for this msg example");
17 int master(int argc, char *argv[]);
18 int slave(int argc, char *argv[]);
19 MSG_error_t test_all(const char *platform_file,
20 const char *application_file);
22 /** Emitter function */
23 int master(int argc, char *argv[])
25 long number_of_tasks = atol(argv[1]);
26 double task_comp_size = atof(argv[2]);
27 double task_comm_size = atof(argv[3]);
28 long slaves_count = atol(argv[4]);
30 //setting the variable "is_master" (previously declared) to value 1
31 TRACE_host_variable_set(MSG_host_self()->name, "is_master", 1);
33 TRACE_mark("msmark", "start_send_tasks");
35 for (i = 0; i < number_of_tasks; i++) {
37 task = MSG_task_create("task", task_comp_size, task_comm_size, NULL);
39 //setting the variable "task_creation" to value i
40 TRACE_host_variable_set(MSG_host_self()->name, "task_creation", i);
42 //setting the category of task to "compute"
43 //the category of a task must be defined before it is sent or executed
44 TRACE_msg_set_task_category(task, "compute");
45 MSG_task_send(task, "master_mailbox");
47 TRACE_mark("msmark", "finish_send_tasks");
49 for (i = 0; i < slaves_count; i++) {
50 m_task_t finalize = MSG_task_create("finalize", 0, 0, 0);
51 TRACE_msg_set_task_category(finalize, "finalize");
52 MSG_task_send(finalize, "master_mailbox");
58 /** Receiver function */
59 int slave(int argc, char *argv[])
64 TRACE_host_variable_set(MSG_host_self()->name, "is_slave", 1);
66 res = MSG_task_receive(&(task), "master_mailbox");
68 if (!strcmp(MSG_task_get_name(task), "finalize")) {
69 MSG_task_destroy(task);
72 //adding the value returned by MSG_task_get_compute_duration(task)
73 //to the variable "task_computation"
74 TRACE_host_variable_add(MSG_host_self()->name,
76 MSG_task_get_compute_duration(task));
77 MSG_task_execute(task);
78 MSG_task_destroy(task);
85 MSG_error_t test_all(const char *platform_file,
86 const char *application_file)
88 MSG_error_t res = MSG_OK;
90 { /* Simulation setting */
91 MSG_set_channel_number(0);
92 MSG_create_environment(platform_file);
95 /* declaring tracing categories */
97 //declaring user variables
98 TRACE_host_variable_declare("is_slave");
99 TRACE_host_variable_declare("is_master");
100 TRACE_host_variable_declare("task_creation");
101 TRACE_host_variable_declare("task_computation");
103 //declaring user markers
104 TRACE_declare_mark("msmark");
106 //declaring user categories with RGB colors (values from 0 to 1)
107 TRACE_category_with_color ("compute", "1 0 0"); //compute is red
108 TRACE_category_with_color ("finalize", "0 1 0"); //finalize is green
109 //categories without user-defined colors receive
110 //random colors generated by the tracing system
111 TRACE_category ("request");
112 TRACE_category_with_color ("report", NULL);
114 { /* Application deployment */
115 MSG_function_register("master", master);
116 MSG_function_register("slave", slave);
117 MSG_launch_application(application_file);
121 XBT_INFO("Simulation time %g", MSG_get_clock());
127 int main(int argc, char *argv[])
129 MSG_error_t res = MSG_OK;
131 MSG_global_init(&argc, argv);
133 printf("Usage: %s platform_file deployment_file\n", argv[0]);
134 printf("example: %s msg_platform.xml msg_deployment.xml\n", argv[0]);
138 res = test_all(argv[1], argv[2]);