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 }