summaryrefslogtreecommitdiffstats
path: root/src/boost/libs/mpi/test/debugger.cpp
blob: 90093e79f6ca22d228a51d1877e4b07522391a7a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
//          Copyright AlainMiniussi 20014 - 20015.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          http://www.boost.org/LICENSE_1_0.txt)

#include <cstdlib>

#include "debugger.hpp"

std::vector<int> extract_paused_ranks(int argc, char** argv) {
  std::vector<int> paused;
  for (int i=1; i < argc; ++i) {
    paused.push_back(std::atoi(argv[i]));
  }
  return paused;
}

void wait_for_debugger(std::vector<int> const& processes, boost::mpi::communicator const& comm)
{
  int i = 1;
  bool waiting = std::find(processes.begin(), processes.end(), comm.rank()) != processes.end();
  for (int r = 0; r < comm.size(); ++r) {
    if (comm.rank() == r) {
      std::cout << "Rank " << comm.rank() << " has PID " << getpid();
      if (waiting) {
        std::cout << " and is waiting.";
      }
      std::cout << std::endl;
    }
    comm.barrier();
  }
  if (std::find(processes.begin(), processes.end(), comm.rank()) != processes.end()) {
    while (i != 0) {
      sleep(5);
    }
  }
  std::cout << "Rank " << comm.rank() << " will proceed.\n";
}

void wait_for_debugger(boost::mpi::communicator const& comm)
{
  std::vector<int> all;
  for (int r = 0; r < comm.size(); ++r) {
    all.push_back(r);
  }
  wait_for_debugger(all, comm);
}