X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/4fcf82473390a73a49a33a119ac10ae97cfea44b..e16e0233:/examples/s4u/app-masterworkers/s4u-app-masterworkers-class.cpp diff --git a/examples/s4u/app-masterworkers/s4u-app-masterworkers-class.cpp b/examples/s4u/app-masterworkers/s4u-app-masterworkers-class.cpp index cd172d8f3c..a91f203ad8 100644 --- a/examples/s4u/app-masterworkers/s4u-app-masterworkers-class.cpp +++ b/examples/s4u/app-masterworkers/s4u-app-masterworkers-class.cpp @@ -1,103 +1,102 @@ -/* Copyright (c) 2010-2018. The SimGrid Team. All rights reserved. */ +/* Copyright (c) 2010-2019. The SimGrid Team. All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ -#include "xbt/sysdep.h" +/* ************************************************************************* */ +/* Take this tutorial online: https://simgrid.frama.io/simgrid/tuto_s4u.html */ +/* ************************************************************************* */ + #include -#include XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_app_masterworker, "Messages specific for this s4u example"); class Master { - long number_of_tasks = 0; /* - Number of tasks */ - double comp_size = 0; /* - Task compute cost */ - double comm_size = 0; /* - Task communication size */ - long workers_count = 0; /* - Number of workers */ - simgrid::s4u::MailboxPtr mailbox = nullptr; + long tasks_count = 0; + double compute_cost = 0; + double communicate_cost = 0; + std::vector workers; public: explicit Master(std::vector args) { - xbt_assert(args.size() == 5, "The master function expects 4 arguments from the XML deployment file"); + xbt_assert(args.size() > 4, "The master function expects 3 arguments plus the workers' names"); - number_of_tasks = std::stol(args[1]); - comp_size = std::stod(args[2]); - comm_size = std::stod(args[3]); - workers_count = std::stol(args[4]); + tasks_count = std::stol(args[1]); + compute_cost = std::stod(args[2]); + communicate_cost = std::stod(args[3]); + for (unsigned int i = 4; i < args.size(); i++) + workers.push_back(simgrid::s4u::Mailbox::by_name(args[i])); - XBT_INFO("Got %ld workers and %ld tasks to process", workers_count, number_of_tasks); + XBT_INFO("Got %zu workers and %ld tasks to process", workers.size(), tasks_count); } void operator()() { - for (int i = 0; i < number_of_tasks; i++) { /* For each task to be executed: */ - /* - Select a @ref worker in a round-robin way */ - mailbox = simgrid::s4u::Mailbox::by_name(std::string("worker-") + std::to_string(i % workers_count)); - - if (number_of_tasks < 10000 || (number_of_tasks < 100000 && i % 10000 == 0) || i % 100000 == 0) - XBT_INFO("Sending \"%s\" (of %ld) to mailbox \"%s\"", (std::string("Task_") + std::to_string(i)).c_str(), - number_of_tasks, mailbox->get_cname()); - - /* - Send the computation amount to the @ref worker */ - mailbox->put(new double(comp_size), comm_size); + for (int i = 0; i < tasks_count; i++) { /* For each task to be executed: */ + /* - Select a worker in a round-robin way */ + simgrid::s4u::Mailbox* mailbox = workers[i % workers.size()]; + + /* - Send the computation amount to the worker */ + if (tasks_count < 10000 || (tasks_count < 100000 && i % 10000 == 0) || i % 100000 == 0) + XBT_INFO("Sending task %d of %ld to mailbox '%s'", i, tasks_count, mailbox->get_cname()); + mailbox->put(new double(compute_cost), communicate_cost); } - XBT_INFO("All tasks have been dispatched. Let's tell everybody the computation is over."); - for (int i = 0; i < workers_count; i++) { - /* - Eventually tell all the workers to stop by sending a "finalize" task */ - mailbox = simgrid::s4u::Mailbox::by_name(std::string("worker-") + std::to_string(i % workers_count)); + XBT_INFO("All tasks have been dispatched. Request all workers to stop."); + for (unsigned int i = 0; i < workers.size(); i++) { + /* The workers stop when receiving a negative compute_cost */ + simgrid::s4u::Mailbox* mailbox = workers[i % workers.size()]; mailbox->put(new double(-1.0), 0); } } }; class Worker { - long id = -1; - simgrid::s4u::MailboxPtr mailbox = nullptr; + simgrid::s4u::Mailbox* mailbox = nullptr; public: explicit Worker(std::vector args) { - xbt_assert(args.size() == 2, "The worker expects a single argument from the XML deployment file: " - "its worker ID (its numerical rank)"); - id = std::stol(args[1]); - mailbox = simgrid::s4u::Mailbox::by_name(std::string("worker-") + std::to_string(id)); + xbt_assert(args.size() == 1, "The worker expects to not get any argument"); + + mailbox = simgrid::s4u::Mailbox::by_name(simgrid::s4u::this_actor::get_host()->get_name()); } void operator()() { - while (1) { /* The worker waits in an infinite loop for tasks sent by the @ref master */ - double* task = static_cast(mailbox->get()); - xbt_assert(task != nullptr, "mailbox->get() failed"); - double comp_size = *task; - delete task; - if (comp_size < 0) { /* - Exit when -1.0 is received */ - XBT_INFO("I'm done. See you!"); - break; - } - /* - Otherwise, process the task */ - simgrid::s4u::this_actor::execute(comp_size); - } + double compute_cost; + do { + double* msg = static_cast(mailbox->get()); + compute_cost = *msg; + delete msg; + + if (compute_cost > 0) /* If compute_cost is valid, execute a computation of that cost */ + simgrid::s4u::this_actor::execute(compute_cost); + + } while (compute_cost > 0); /* Stop when receiving an invalid compute_cost */ + + XBT_INFO("Exiting now."); } }; int main(int argc, char* argv[]) { simgrid::s4u::Engine e(&argc, argv); - xbt_assert(argc > 2, - "Usage: %s platform_file deployment_file\n" - "\tExample: %s msg_platform.xml msg_deployment.xml\n", - argv[0], argv[0]); + xbt_assert(argc > 2, "Usage: %s platform_file deployment_file\n", argv[0]); - e.load_platform(argv[1]); /* Load the platform description */ - e.register_actor("master"); /* Register the class representing the actors */ + /* Register the classes representing the actors */ + e.register_actor("master"); e.register_actor("worker"); - e.load_deployment(argv[2]); /* Deploy the application */ - e.run(); /** - Run the simulation */ + /* Load the platform description and then deploy the application */ + e.load_platform(argv[1]); + e.load_deployment(argv[2]); + + /* Run the simulation */ + e.run(); - XBT_INFO("Simulation time %g", e.get_clock()); + XBT_INFO("Simulation is over"); return 0; }