examples/smpi/replay/replay.cpp

   1 /* Copyright (c) 2009-2020. 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 "xbt/replay.hpp"
   7 #include "simgrid/s4u/Actor.hpp"
   8 #include "smpi/smpi.h"
   9 #include "xbt/asserts.h"
  10 #include "xbt/str.h"
  11
  12 #include "xbt/log.h"
  13 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test, "Messages specific for this msg example");
  14
  15 /* This shows how to extend the trace format by adding a new kind of events.
  16    This function is registered through xbt_replay_action_register() below. */
  17 static void action_blah(const simgrid::xbt::ReplayAction& /*args*/)
  18 {
  19   /* Add your answer to the blah event here.
  20      args is a strings array containing the blank-separated parameters found in the trace for this event instance. */
  21 }
  22
  23 action_fun previous_send;
  24 static void overriding_send(simgrid::xbt::ReplayAction& args)
  25 {
  26   previous_send(args); // Just call the overridden symbol. That's a toy example.
  27 }
  28
  29 int main(int argc, char* argv[])
  30 {
  31   const char* instance_id = simgrid::s4u::Actor::self()->get_property("instance_id");
  32   const int rank =
  33       static_cast<int>(xbt_str_parse_int(simgrid::s4u::Actor::self()->get_property("rank"), "Cannot parse rank"));
  34   const char* trace_filename = argv[1];
  35   double start_delay_flops   = 0;
  36
  37   if (argc > 2) {
  38     start_delay_flops = xbt_str_parse_double(argv[2], "Cannot parse start_delay_flops");
  39   }
  40
  41   /* Setup things and register default actions */
  42   smpi_replay_init(instance_id, rank, start_delay_flops);
  43
  44   /* Connect your callback function to the "blah" event in the trace files */
  45   xbt_replay_action_register("blah", action_blah);
  46
  47   /* The send action is an override, so we have to first save its previous value in a global */
  48   int new_rank;
  49   MPI_Comm_rank(MPI_COMM_WORLD, &new_rank);
  50   if (new_rank != rank)
  51     XBT_WARN("Rank inconsistency. Got %d, expected %d", new_rank, rank);
  52   if (rank == 0) {
  53     previous_send = xbt_replay_action_get("send");
  54     xbt_replay_action_register("send", overriding_send);
  55   }
  56   /* The regular run of the replayer */
  57   smpi_replay_main(rank, trace_filename);
  58   return 0;
  59 }