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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
/////////////////////////////////////////////////////////////
// //
// Copyright (c) 2003-2014 by The University of Queensland //
// Centre for Geoscience Computing //
// http://earth.uq.edu.au/centre-geoscience-computing //
// //
// Primary Business: Brisbane, Queensland, Australia //
// Licensed under the Open Software License version 3.0 //
// http://www.apache.org/licenses/LICENSE-2.0 //
// //
/////////////////////////////////////////////////////////////
#ifndef __MPISGBUF_H
#define __MPISGBUF_H
#include <mpi.h>
#include <string>
#include "Parallel/mpibuf.h"
/*!
\class AMPISGBufferRoot
\brief Abstract base class for scatter/gather buffer, root component
\author Steffen Abe
$Revision$
$Date$
*/
class AMPISGBufferRoot
{
protected:
MPI_Comm m_comm; //!< the MPI communicator used for the scatter/gather operations
int m_rank; //!< the rank in this communicator
int m_size; //!< size of the communicator
int m_int_increment,m_dbl_increment; //!< the "packing size" of int/double
MPI_Status m_status;
public:
AMPISGBufferRoot(MPI_Comm);
virtual ~AMPISGBufferRoot(){};
virtual void clear()=0;
virtual void gather()=0;
virtual void scatter()=0;
virtual void append(int,int)=0;
virtual void append(double,int)=0;
virtual void append(const char*,int)=0;
virtual void append(const Vec3 &,int);
virtual int pop_int(int)=0;
virtual double pop_double(int)=0;
virtual void pop_doubles(int,double *,int)=0;
virtual Vec3 pop_vector(int);
const MPI_Status& status(){return m_status;};
};
/*!
\class AMPISGBufferLeaf
\brief Abstract base class for scatter/gather buffer, leaf component
\author Steffen Abe
$Revision$
$Date$
*/
class AMPISGBufferLeaf : public AMPIBuffer
{
protected:
int m_root; //!< rank of the root process
int m_int_increment,m_dbl_increment; //!< the "packing size" of int/double
public:
AMPISGBufferLeaf(MPI_Comm,int);
virtual ~AMPISGBufferLeaf(){};
virtual void clear()=0;
virtual void send()=0;
virtual void receive()=0;
virtual void append(int)=0;
virtual void append(double)=0;
virtual int pop_int()=0;
virtual double pop_double()=0;
virtual void pop_doubles(double *,int)=0;
virtual std::string pop_string()=0;
const MPI_Status& status(){return m_status;};
};
/*!
\class CMPISGBufferRoot
\brief Buffer for MPI scatter/gather, root component
\author Steffen Abe
$Revision$
$Date$
*/
class CMPISGBufferRoot : public AMPISGBufferRoot
{
private:
char* m_buffer;
char* m_dummy_buffer; //!<dummy buffer sent by root to itself
int m_buffersize; //!< the size of the buffer per slice
int *m_position; //!< the current end of the content in each slice
public:
CMPISGBufferRoot(MPI_Comm,int);
virtual ~CMPISGBufferRoot();
virtual void clear();
virtual void gather();
virtual void scatter();
virtual void append(int,int);
virtual void append(double,int);
virtual void append(const char*,int);
virtual int pop_int(int);
virtual double pop_double(int);
virtual void pop_doubles(int,double *,int);
};
/*!
\class CMPISGBufferLeaf
\brief Buffer for MPI scatter/gather, leaf component
\author Steffen Abe
$Revision$
$Date$
*/
class CMPISGBufferLeaf : public AMPISGBufferLeaf
{
private:
char* m_buffer;
int m_buffersize; //!< the size of the buffer
int m_position; //!< the current end of the content
public:
CMPISGBufferLeaf(MPI_Comm,int,int);
virtual ~CMPISGBufferLeaf();
virtual void clear();
virtual void send();
virtual void receive();
virtual void append(int);
virtual void append(double);
virtual void append(const char*);
virtual int pop_int();
virtual double pop_double();
virtual void pop_doubles(double *,int);
virtual std::string pop_string();
};
#endif // __MPISGBUF_H
|
