Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Merge branch 'master' of git+ssh://scm.gforge.inria.fr/gitroot/simgrid/simgrid
[simgrid.git] / examples / msg / trace-categories / trace-categories.c
1 /* Copyright (c) 2010-2015. 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 /** @addtogroup MSG_examples
8  * 
9  * - <b>tracing/categories.c</b> This is a master/worker program. The master creates tasks and send them to workers.
10  * For each task received, the worker executes it and then destroys it. This program declares several tracing categories
11  * that are used to classify tasks. When the program is executed, the tracing mechanism registers the resource
12  * utilization of hosts and links according to these categories. You might want to run this program with the following
13  * parameters:
14  * --cfg=tracing:yes
15  * --cfg=tracing/categorized:yes
16  * --cfg=tracing/uncategorized:yes
17  * --cfg=viva/categorized:viva_cat.plist
18  * --cfg=viva/uncategorized:viva_uncat.plist
19  * (See \ref tracing_tracing_options for details)
20  */
21
22 #include "simgrid/msg.h"
23
24 static int master(int argc, char *argv[])
25 {
26   long number_of_tasks = xbt_str_parse_int(argv[1], "Invalid amount of tasks: %s");
27   long workers_count = xbt_str_parse_int(argv[4], "Invalid amount of workers: %s");
28
29   int i;
30   for (i = 0; i < number_of_tasks; i++) {
31     msg_task_t task = NULL;
32
33     //creating task and setting its category
34     if (i % 2) {
35       task = MSG_task_create("task_compute", 10000000, 0, NULL);
36       MSG_task_set_category(task, "compute");
37     } else if (i % 3) {
38       task = MSG_task_create("task_request", 10, 10, NULL);
39       MSG_task_set_category(task, "request");
40     } else {
41       task = MSG_task_create("task_data", 10, 10000000, NULL);
42       MSG_task_set_category(task, "data");
43     }
44     MSG_task_send(task, "master_mailbox");
45   }
46
47   for (i = 0; i < workers_count; i++) {
48     msg_task_t finalize = MSG_task_create("finalize", 0, 1000, 0);
49     MSG_task_set_category(finalize, "finalize");
50     MSG_task_send(finalize, "master_mailbox");
51   }
52
53   return 0;
54 }
55
56 static int worker(int argc, char *argv[])
57 {
58   msg_task_t task = NULL;
59
60   while (1) {
61     MSG_task_receive(&(task), "master_mailbox");
62
63     if (!strcmp(MSG_task_get_name(task), "finalize")) {
64       MSG_task_destroy(task);
65       break;
66     }
67
68     MSG_task_execute(task);
69     MSG_task_destroy(task);
70     task = NULL;
71   }
72   return 0;
73 }
74
75 int main(int argc, char *argv[])
76 {
77   MSG_init(&argc, argv);
78   xbt_assert(argc > 2, "Usage: %s platform_file deployment_file\n"
79              "\tExample: %s msg_platform.xml msg_deployment.xml\n", argv[0], argv[0]);
80
81   MSG_create_environment(argv[1]);
82
83   //declaring user categories with RGB colors
84   TRACE_category_with_color ("compute", "1 0 0"); //red
85   TRACE_category_with_color ("request", "0 1 0"); //green
86   TRACE_category_with_color ("data", "0 0 1");    //blue
87   TRACE_category_with_color ("finalize", "0 0 0");//black
88
89   MSG_function_register("master", master);
90   MSG_function_register("worker", worker);
91   MSG_launch_application(argv[2]);
92
93   MSG_main();
94   return 0;
95 }