forked from Kitware/VTK
/
vtkMPIUtilities.cxx
115 lines (90 loc) · 2.76 KB
/
vtkMPIUtilities.cxx
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*=========================================================================
Program: Visualization Toolkit
Module: vtkMPIUtilities.h
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#include "vtkMPIUtilities.h"
// VTK includes
#include "vtkMPICommunicator.h"
#include "vtkMPIController.h"
// C/C++ includes
#include <cassert>
#include <cstdarg>
#include <cstdio>
namespace vtkMPIUtilities
{
void Printf(vtkMPIController* comm, const char* format, ...)
{
// Sanity checks
assert("pre: MPI controller is nullptr!" && (comm != nullptr) );
assert("pre: format argument is nullptr!" && (format != nullptr) );
if( comm->GetLocalProcessId() == 0 )
{
va_list argptr;
va_start(argptr,format);
vprintf(format,argptr);
fflush(stdout);
va_end(argptr);
}
comm->Barrier();
}
//------------------------------------------------------------------------------
void SynchronizedPrintf(vtkMPIController* comm, const char* format, ...)
{
// Sanity checks
assert("pre: MPI controller is nullptr!" && (comm != nullptr) );
assert("pre: format argument is nullptr!" && (format != nullptr) );
int rank = comm->GetLocalProcessId();
int numRanks = comm->GetNumberOfProcesses();
vtkMPICommunicator::Request rqst;
int* nullmsg = nullptr;
if(rank == 0)
{
// STEP 0: print message
printf("[%d]: ", rank);
fflush(stdout);
va_list argptr;
va_start(argptr,format);
vprintf(format,argptr);
fflush(stdout);
va_end(argptr);
// STEP 1: signal next process (if any) to print
if( numRanks > 1)
{
comm->NoBlockSend(nullmsg,0,rank+1,0,rqst);
} // END if
} // END first rank
else if( rank == numRanks-1 )
{
// STEP 0: Block until previous process completes
comm->Receive(nullmsg,0,rank-1,0);
// STEP 1: print message
printf("[%d]: ", rank);
va_list argptr;
va_start(argptr,format);
vprintf(format,argptr);
fflush(stdout);
va_end(argptr);
} // END last rank
else
{
// STEP 0: Block until previous process completes
comm->Receive(nullmsg,0,rank-1,0);
// STEP 1: print message
printf("[%d]: ", rank);
va_list argptr;
va_start(argptr,format);
vprintf(format,argptr);
fflush(stdout);
va_end(argptr);
// STEP 2: signal next process to print
comm->NoBlockSend(nullmsg,0,rank+1,0,rqst);
}
comm->Barrier();
}
} // END namespace vtkMPIUtilities