Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Further reorganize the MSG examples' documentation
[simgrid.git] / examples / msg / trace-categories / trace-categories.c
1 /* Copyright (c) 2010-2016. The SimGrid Team. All rights reserved.          */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include "simgrid/msg.h"
7
8 static int master(int argc, char *argv[])
9 {
10   long number_of_tasks = xbt_str_parse_int(argv[1], "Invalid amount of tasks: %s");
11   long workers_count = xbt_str_parse_int(argv[4], "Invalid amount of workers: %s");
12
13   int i;
14   for (i = 0; i < number_of_tasks; i++) {
15     msg_task_t task = NULL;
16
17     /* creating task and setting its category */
18     if (i % 2) {
19       task = MSG_task_create("task_compute", 10000000, 0, NULL);
20       MSG_task_set_category(task, "compute");
21     } else if (i % 3) {
22       task = MSG_task_create("task_request", 10, 10, NULL);
23       MSG_task_set_category(task, "request");
24     } else {
25       task = MSG_task_create("task_data", 10, 10000000, NULL);
26       MSG_task_set_category(task, "data");
27     }
28     MSG_task_send(task, "master_mailbox");
29   }
30
31   for (i = 0; i < workers_count; i++) {
32     msg_task_t finalize = MSG_task_create("finalize", 0, 1000, 0);
33     MSG_task_set_category(finalize, "finalize");
34     MSG_task_send(finalize, "master_mailbox");
35   }
36
37   return 0;
38 }
39
40 static int worker(int argc, char *argv[])
41 {
42   msg_task_t task = NULL;
43
44   while (1) {
45     MSG_task_receive(&(task), "master_mailbox");
46
47     if (!strcmp(MSG_task_get_name(task), "finalize")) {
48       MSG_task_destroy(task);
49       break;
50     }
51
52     MSG_task_execute(task);
53     MSG_task_destroy(task);
54     task = NULL;
55   }
56   return 0;
57 }
58
59 int main(int argc, char *argv[])
60 {
61   MSG_init(&argc, argv);
62   xbt_assert(argc > 2, "Usage: %s platform_file deployment_file\n"
63              "\tExample: %s msg_platform.xml msg_deployment.xml\n", argv[0], argv[0]);
64
65   MSG_create_environment(argv[1]);
66
67   /* declaring user categories with RGB colors */
68   TRACE_category_with_color ("compute", "1 0 0"); //red
69   TRACE_category_with_color ("request", "0 1 0"); //green
70   TRACE_category_with_color ("data", "0 0 1");    //blue
71   TRACE_category_with_color ("finalize", "0 0 0");//black
72
73   MSG_function_register("master", master);
74   MSG_function_register("worker", worker);
75   MSG_launch_application(argv[2]);
76
77   MSG_main();
78   return 0;
79 }